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

源码开发语言/平台
当前位置：首页 > 源码/资料 > Internet/网络编程 > Ftp服务器 > 查看源码
FtpdPi.cpp
资源名称：FTP.rar　[点击查看]
上传用户：szlsqc
上传日期：2015-05-05
资源大小：149k
文件大小：36k
源码类别：

Ftp服务器

开发平台：

C/C++

FtpdPi.cpp：源码内容
  1. #include "stdafx.h"
  2. #include "CFtpd.h"
  3. #include "FtpdPi.h"
  5. #define CRLF "rn"
  6. #ifdef _DEBUG
  7. #define new DEBUG_NEW
  8. #undef THIS_FILE
  9. static char THIS_FILE[] = __FILE__;
  10. #endif
  12. char sadr2[32];
  14. bool canclePasvListenSocket(CFtpPasvSrv* sck)
  15. {
  16. try
  17. {
  18. sck->Close();
  19. delete sck;
  20. return true;
  21. }
  22. catch(...)
  23. {
  24. }
  25. return false;
  26. }
  28. bool deleFtpPI(CFtpdPi* sck)
  29. {
  30. try
  31. {
  32. CFtpd* K = (CFtpd*) sck->pFtpd;
  33. K->RemoveClient(sck);
  34. }
  35. catch(...)
  36. {
  37. }
  38. return true;
  39. }
  41. UINT thFtpCmd( LPVOID pParam )
  42. {
  43. ((CFtpdPi*)pParam)->ProcessMsg(((CFtpdPi*)pParam)->bufcmd);
  44. return 1;
  45. }
  47. UINT thSendData( LPVOID pParam)
  48. {
  49. CFtpdPi* pFtpBind = (CFtpdPi*) pParam;
  50. CFile* inFile = NULL;
  51. try
  52. {
  53. char path[256] = "";
  54. strcpy(path,pFtpBind->RemoveLeadingSpace(pFtpBind->curCmdth+5));
  55. char srcPath[300] = "";
  56. if(strcmp(pFtpBind->vCurrentDir,"/")==0)
  57. wsprintf(srcPath,"%s%s%s",pFtpBind->vRealRoot,pFtpBind->vCurrentDir,path);
  58. else
  59. wsprintf(srcPath,"%s%s\%s",pFtpBind->vRealRoot,pFtpBind->vCurrentDir,path);
  60. pFtpBind->slashFix(srcPath);
  61. if(pFtpBind->isFilePresent(path))
  62. {
  63. bool b_opened = true;
  64. try
  65. {
  66. if(pFtpBind->type == 1)
  67. inFile = new CFile(srcPath,CFile::modeRead|CFile::shareDenyNone|CFile::typeBinary);//CStdioFile(srcPath,CFile::modeRead|CFile::shareDenyNone|CFile::typeText);
  68. else if(pFtpBind->type == 3)
  69. inFile = new CFile(srcPath,CFile::modeRead|CFile::shareDenyNone|CFile::typeBinary);
  70. }
  71. catch(...)
  72. {
  73. b_opened = false;
  74. }
  75. if(b_opened)
  76. {
  77. long flen = inFile->GetLength(); 
  78. if(pFtpBind->restart_marker > 0)
  79. {
  80. inFile->Seek(pFtpBind->restart_marker,CFile::begin);
  81. pFtpBind->restart_marker = 0;
  82. }
  83. long nCount = 0;
  84. if(pFtpBind->type == 1)
  85. wsprintf(pFtpBind->str,"150 Sending %s (%u bytes). Mode STREAM Type ASCII%s",path,flen,CRLF);
  86. else if(pFtpBind->type == 3)
  87. wsprintf(pFtpBind->str,"150 Sending %s (%u bytes). Mode STREAM Type BINARY%s",path,flen,CRLF);
  88. pFtpBind->Send(pFtpBind->str,strlen(pFtpBind->str),0);
  89. if(flen)
  90. {
  91. long bytesent = 0;
  92. long offset = 0;
  93. DWORD start,end;
  94. do
  95. {
  96. if(flen > pFtpBind->m_buf_length_out)
  97. nCount = inFile->Read(pFtpBind->lpDataBuf_out,pFtpBind->m_buf_length_out);
  98. else
  99. nCount = inFile->Read(pFtpBind->lpDataBuf_out,flen);
  100. flen = flen - nCount;
  101. if(pFtpBind->b_speed_limit_out)
  102. {
  103. bytesent = 0; 
  104. offset = 0;
  105. HANDLE h = CreateEvent(NULL,false,false,"noname");
  106. do
  107. {
  108. start = GetTickCount();
  109. if((nCount-offset) > pFtpBind->m_buf_length_out)
  110. bytesent = pFtpBind->dbsocket->Send((char*)pFtpBind->lpDataBuf_out+offset,pFtpBind->m_buf_length_out,60);
  111. else
  112. bytesent = pFtpBind->dbsocket->Send((char*)pFtpBind->lpDataBuf_out+offset,nCount-offset,60);
  113. offset = offset + bytesent;
  114. end = GetTickCount();
  115. if((end-start) < 1000)
  116. {
  117. int time = end-start;
  118. WaitForSingleObject(h,1000-time);
  119. }
  120. }while(offset < nCount);
  121. }
  122. else
  123. {
  124. bytesent = 0; 
  125. offset = 0;
  126. do
  127. {
  128. if((nCount-offset) > pFtpBind->m_buf_length_out)
  129. bytesent = pFtpBind->dbsocket->Send((char*)pFtpBind->lpDataBuf_out+offset,pFtpBind->m_buf_length_out,60);
  130. else
  131. bytesent = pFtpBind->dbsocket->Send((char*)pFtpBind->lpDataBuf_out+offset,nCount-offset,60);
  132. offset = offset + bytesent;
  133. }while(offset < nCount);
  134. if(pFtpBind->type == 1)
  135. {
  136. pFtpBind->dbsocket->Send(CRLF,2,60);
  137. }
  138. }
  139. }while(flen != 0);
  140. }
  141. pFtpBind->dbsocket->Close();
  142. if(pFtpBind->b_passive)
  143. pFtpBind->svSocketSrv->Close();
  144. delete inFile;
  145. wsprintf(pFtpBind->str,"226 Transfer finished successfully. Data connection closed.%s",CRLF);
  146. pFtpBind->Send(pFtpBind->str,strlen(pFtpBind->str),0);
  147. SetEvent(pFtpBind->h_inprogress);
  148. return true;
  149. }else
  150. {
  151. pFtpBind->dbsocket->Close();
  152. if(pFtpBind->b_passive)
  153. pFtpBind->svSocketSrv->Close();
  154. if(inFile != NULL)
  155. delete inFile;
  156. wsprintf(pFtpBind->str,"426 Connection closed , file opening failed%s",CRLF);
  157. pFtpBind->Send(pFtpBind->str,strlen(pFtpBind->str),0);
  158. SetEvent(pFtpBind->h_inprogress);
  159. }
  160. }
  161. }catch(...)
  162. {
  163. inFile->Close();
  164. pFtpBind->dbsocket->Close();
  165. canclePasvListenSocket(pFtpBind->svSocketSrv);
  166. delete inFile;
  167. wsprintf(pFtpBind->str,"426 Connection closed; transfer aborted (transfer failed)%s",CRLF);
  168. pFtpBind->Send(pFtpBind->str,strlen(pFtpBind->str),0);
  169. SetEvent(pFtpBind->h_inprogress);
  170. return false;
  171. }
  172. return true;
  173. }
  177. CFtpdPi::CFtpdPi()
  178. {
  179. clientIpCount = 0;
  180. type = 3; //1 = ascii 2 = ebcdic 3 = binary
  181. ipRoller = 0;
  182. b_use_bsock = true;
  183. b_verbose = true;
  184. b_passive = false;
  185. b_bell = false;
  186. b_prompt = true;
  187. b_globbing = true;
  188. b_debugging = false;
  189. b_hashmark = false;
  190. bLOGINOK = false;
  191. bUSER = false;
  192. bPASS = false;
  193. b_speed_limit_out = false;
  194. b_speed_limit_in = false;
  195. restart_marker = 0;
  196. strcpy(vCurrentDir,"/");
  197. strcpy(old_filename,"");
  198. strcpy(new_filename,"");
  199. strcpy(m_group,"nogroup");
  200. m_buf_length_out = 32768;
  201. m_buf_length_in = 32768;
  202. lpDataBuf_out = (unsigned char*) malloc(m_buf_length_out);
  203. memset(lpDataBuf_out,0,m_buf_length_out);
  204. lpDataBuf_in = (unsigned char*) malloc(m_buf_length_in);
  205. memset(lpDataBuf_in,0,m_buf_length_in);
  206. strcpy(monthStr[1],"Jan");
  207. strcpy(monthStr[2],"Feb");
  208. strcpy(monthStr[3],"Mar");
  209. strcpy(monthStr[4],"Apr");
  210. strcpy(monthStr[5],"May");
  211. strcpy(monthStr[6],"Jun");
  212. strcpy(monthStr[7],"Jul");
  213. strcpy(monthStr[8],"Aug");
  214. strcpy(monthStr[9],"Sep");
  215. strcpy(monthStr[10],"Oct");
  216. strcpy(monthStr[11],"Dec");
  217. strcpy(monthStr[12],"Nov");
  218. sl = sizeof(SOCKADDR_IN);
  219. GetLocalTime(&MyTime);
  220. if(FileExists(ipfile))
  221. LoadIpRange();
  222. h_inprogress = CreateEvent(NULL,false,true,"in_progress");
  223. h_dbsocket_ready_to_send = CreateEvent(NULL,false,false,"in_progress");
  224. dbsocket = NULL;
  225. svSocketSrv = NULL;
  226. }
  228. CFtpdPi::~CFtpdPi()
  229. {
  230. free(lpDataBuf_in);
  231. free(lpDataBuf_out);
  232. if(dbsocket != NULL)
  233. {
  234. dbsocket->Close();
  235. delete dbsocket;
  236. }
  237. if(svSocketSrv != NULL)
  238. {
  239. svSocketSrv->Close();
  240. delete svSocketSrv;
  241. }
  242. }
  245. // Do not edit the following lines, which are needed by ClassWizard.
  246. #if 0
  247. BEGIN_MESSAGE_MAP(CFtpdPi, CSocket)
  248. //{{AFX_MSG_MAP(CFtpdPi)
  249. //}}AFX_MSG_MAP
  250. END_MESSAGE_MAP()
  251. #endif // 0
  253. void CFtpdPi::OnReceive(int nErrorCode) 
  254. {
  255. unsigned char* bufnow = (unsigned char*)malloc(256);
  256. try
  257. {
  258. memset(bufnow,NULL,256);
  259. result = Receive(bufnow,256,0);
  260. bool cont = true;
  261. int currentPos = 0;
  262. if(*(bufnow+1)==0xf2)
  263. currentPos = 1;
  264. for(int i=0;i<result;i++)
  265. {
  266. if(*(bufnow+i)==0x0d&&*(bufnow+i+1)==0x0a)
  267. {
  268. strncpy(bufcmd,(char*)bufnow+currentPos,i-currentPos);
  269. *(bufcmd+i-currentPos) = NULL;
  270. strcpy(bufcmd_history,bufcmd);
  271. AfxBeginThread(&thFtpCmd,this,THREAD_PRIORITY_NORMAL,0,0,NULL);
  272. currentPos = i+2;
  273. i = i + 1;
  274. }
  275. }
  276. free(bufnow);
  277. }
  278. catch (...)
  279. {
  280. delete bufnow;
  281. char lpMsgBuf[128]="";
  282. DWORD eRRR = GetLastError();
  283. FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS,NULL,eRRR,0,lpMsgBuf,128,NULL);
  284. char ecp[256]="";
  285. wsprintf(ecp,"%s\core.ftpd",AppDir);
  286. CStdioFile ofs((LPCTSTR) ecp ,CFile::modeWrite|CFile::modeCreate);
  287. ofs.WriteString("*---------------------------------------------------------------------------n");
  288. ofs.WriteString("Exception occurs while processing socket msgn");
  289. GetLocalTime(&MyTime);
  290. wsprintf(ecp,"Thread crashed at %.4d-%.2d-%.2d-%.2d-%.2dn",MyTime.wYear,MyTime.wMonth,MyTime.wDay,MyTime.wHour,MyTime.wMinute);
  291. ofs.WriteString(ecp);
  292. ofs.WriteString(lpMsgBuf);
  293. ofs.WriteString("*---------------------------------------------------------------------------n");
  294. ofs.Close();
  295. wsprintf(ecp,"Socket Exception occurs , aboarting current thread");
  296. }
  297. CSocket::OnReceive(nErrorCode);
  298. }
  300. void CFtpdPi::ProcessMsg(char *cmd)
  301. {
  302. char curCmd[256] = "";
  303. strcpy(curCmd,cmd);
  304. for(int k=0;k<=3;k++)
  305. {
  306. if(curCmd[k] != ' ')
  307. {
  308. curCmd[k] = toupper(curCmd[k]);
  309. }else
  310. k = 3;
  311. }
  312. if(!bLOGINOK)
  313. {
  314. if(strncmp("USER ",curCmd,5) == 0)
  315. {
  316. char username[256] = "";
  317. strcpy(username,RemoveLeadingSpace(curCmd+5));
  318. if(validuser(username) && strlen(username) <= 32)
  319. {
  320. strcpy(m_username,username);
  321. bUSER = true;
  322. wsprintf(str,"331 Password required for %s .%s",(LPCSTR)m_username,CRLF);
  323. SafeSend(str);
  324. }else
  325. {
  326. wsprintf(str,"530 Login incorrect for %s .%s",username,CRLF);
  327. SafeSend(str);
  328. }
  329. }else if(strncmp("PASS ",curCmd,5) == 0 && bUSER == true)
  330. {
  331. char password[256] = "";
  332. strcpy(password,RemoveLeadingSpace(curCmd+5));
  333. if(matchpass(password,(char*)(LPCSTR)m_username) && strlen(password) <= 32)
  334. {
  335. strcpy(m_password,password);
  336. bPASS = true;
  337. }else
  338. {
  339. wsprintf(str,"530 Password incorrect.%s",CRLF);
  340. SafeSend(str);
  341. }
  342. }
  343. if(bUSER && bPASS)
  344. {
  345. bLOGINOK = true;
  346. wsprintf(str,"230 User %s logged in , proceed%s",(LPCSTR)m_username,CRLF);
  347. SafeSend(str);
  348. }
  349. }else
  350. {
  351. if(stricmp("XPWD",curCmd) == 0 || stricmp("PWD",curCmd) == 0)
  352. {
  353. if(strlen(vCurrentDir) > 500)
  354. wsprintf(str,"503 current directory is too long.%s",CRLF);
  355. else
  356. wsprintf(str,"257 "%s" is current directory%s",vCurrentDir,CRLF);
  357. SafeSend(str);
  358. return;
  359. }
  360. else if(strncmp("CWD ",curCmd,4) == 0)
  361. {
  362. char path[512] = "";
  363. strcpy(path,RemoveLeadingSpace(curCmd+4));
  364. if(strcmp(path,"..")==0)
  365. {
  366. if(goParent(vCurrentDir))
  367. wsprintf(str,"250 "%s" is current directory.%s",vCurrentDir,CRLF);
  368. else
  369. wsprintf(str,"550 Permission denied.%s",CRLF);
  370. }else if(strcmp(path,".")==0)
  371. wsprintf(str,"250 "%s" is current directory.%s",vCurrentDir,CRLF);
  372. else
  373. {
  374. fixVpath(path);
  375. if(isVpathPresent(path))
  376. {
  377. strcpy(vCurrentDir,path);
  378. wsprintf(str,"250 "%s" is current directory.%s",vCurrentDir,CRLF);
  379. }else
  380. wsprintf(str,"550 Permission denied.%s",CRLF);
  381. }
  382. SafeSend(str);
  383. return;
  384. }
  385. else if(strncmp("CDUP",curCmd,4) == 0)
  386. {
  387. if(goParent(vCurrentDir))
  388. wsprintf(str,"250 "%s" is current directory.%s",vCurrentDir,CRLF);
  389. else
  390. wsprintf(str,"550 Permission denied.%s",CRLF);
  391. SafeSend(str);
  392. return;
  393. }
  394. else if(strcmp("REIN",curCmd) == 0)
  395. {
  396. bUSER = false;
  397. bPASS = false;
  398. bLOGINOK = false;
  399. wsprintf(str,"550 Relogin required , connection reinitialized.%s",CRLF);
  400. SafeSend(str);
  401. return;
  402. }
  403. else if(strncmp("DELE ",curCmd,5) == 0)
  404. {
  405. char path[512] = "";
  406. strcpy(path,RemoveLeadingSpace(curCmd+4));
  407. if(vRemove(path))
  408. wsprintf(str,"250 Requested file action ok , completed%s",CRLF);
  409. else
  410. {
  411. char lpMsgBuf[128]="";
  412. DWORD eRRR = GetLastError();
  413. FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS,NULL,eRRR,0,lpMsgBuf,128,NULL);
  414. wsprintf(str,"553 Requested action not taken , %s",lpMsgBuf);
  415. }
  416. SafeSend(str);
  417. return;
  418. }
  419. else if(strncmp("XMKD ",curCmd,5) == 0)
  420. {
  421. char path[512] = "";
  422. strcpy(path,RemoveLeadingSpace(curCmd+5));
  423. if(vMkdir(path))
  424. wsprintf(str,"257 "%s" created%s",path,CRLF);
  425. else
  426. {
  427. char lpMsgBuf[128]="";
  428. DWORD eRRR = GetLastError();
  429. FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS,NULL,eRRR,0,lpMsgBuf,128,NULL);
  430. wsprintf(str,"553 Requested action not taken , %s",lpMsgBuf);
  431. }
  432. SafeSend(str);
  433. return;
  434. }
  435. else if(strncmp("RETR ",curCmd,5) == 0)
  436. {
  437. strcpy(curCmdth,curCmd);
  438. AfxBeginThread(&thSendData,this,THREAD_PRIORITY_TIME_CRITICAL,0,0,NULL);
  439. return;
  440. }
  441. else if(strcmp("ABOR",curCmd) == 0)
  442. {
  443. try
  444. {
  445. dbsocket->Close();
  446. }
  447. catch(...)
  448. {
  449. }
  450. wsprintf(str,"426 Transfer aborted. Data connection closed.%s",CRLF);
  451. SafeSend(str);
  452. SetEvent(h_inprogress);
  453. wsprintf(str,"226 Abort successful.%s",CRLF);
  454. SafeSend(str);
  455. return;
  456. }
  457. else if(strncmp("STOR ",curCmd,5) == 0 || strncmp("STOU ",curCmd,5) == 0)
  458. {
  459. CFile* outFile;
  460. try
  461. {
  462. char path[256] = "";
  463. strcpy(path,RemoveLeadingSpace(curCmd+5));
  464. char srcPath[300] = "";
  465. if(strcmp(vCurrentDir,"/")==0)
  466. wsprintf(srcPath,"%s%s%s",vRealRoot,vCurrentDir,path);
  467. else
  468. wsprintf(srcPath,"%s%s\%s",vRealRoot,vCurrentDir,path);
  469. slashFix(srcPath);
  470. if(isFilePresent(path) && strncmp("STOU ",curCmd,5) == 0)
  471. {
  472. dbsocket->Close();
  473. wsprintf(str,"426 Connection closed; transfer aborted (target file already existed)%s",CRLF);
  474. SafeSend(str);
  475. }
  476. else
  477. {
  478. bool b_opened = true;
  479. try
  480. {
  481. if(type == 1)
  482. outFile = new CStdioFile(srcPath,CFile::modeWrite|CFile::modeNoTruncate|CFile::typeText|CFile::modeCreate);
  483. else if(type == 3)
  484. outFile = new CFile(srcPath,CFile::modeWrite|CFile::modeNoTruncate|CFile::typeBinary|CFile::modeCreate);
  485. }
  486. catch(...)
  487. {
  488. b_opened = false;
  489. }
  490. if(b_opened)
  491. {
  492. long flen = 0;
  493. long nCount = 0;
  494. if(type == 1)
  495. wsprintf(str,"150 Opening ASCII data connection for %s%s",path,CRLF);
  496. else if(type == 3)
  497. wsprintf(str,"150 Opening BINARY data connection for %s%s",path,CRLF);
  498. SafeSend(str);
  499. DWORD start,end;
  500. HANDLE h = CreateEvent(NULL,false,false,"noname");
  501. do
  502. {
  503. start = GetTickCount();
  504. nCount = dbsocket->Receive((char*)lpDataBuf_in,m_buf_length_in,60);
  505. outFile->Write(lpDataBuf_in,nCount);
  506. end = GetTickCount();
  507. if(b_speed_limit_in)
  508. {
  509. if((end-start) < 1000)
  510. {
  511. int time = end-start;
  512. WaitForSingleObject(h,((1000-time)*nCount)/m_buf_length_in);
  513. }
  514. }
  515. }while(nCount > 0);
  516. dbsocket->Close();
  517. if(b_passive)
  518. svSocketSrv->Close();
  519. delete outFile;
  520. wsprintf(str,"226 Transfer finished successfully. Data connection closed.%s",CRLF);
  521. SafeSend(str);
  522. }else
  523. {
  524. outFile->Close();
  525. dbsocket->Close();
  526. if(b_passive)
  527. svSocketSrv->Close();
  528. delete outFile;
  529. wsprintf(str,"426 Connection closed; file opening failed%s",CRLF);
  530. SafeSend(str);
  531. }
  532. }
  533. }
  534. catch(...)
  535. {
  536. outFile->Close();
  537. dbsocket->Close();
  538. if(b_passive)
  539. svSocketSrv->Close();
  540. delete outFile;
  541. wsprintf(str,"426 Connection closed; transfer aborted (transfer failed)%s",CRLF);
  542. SafeSend(str);
  543. }
  544. SetEvent(h_inprogress);
  545. return;
  546. }
  547. else if(strncmp("APPE ",curCmd,5) == 0)
  548. {
  549. CFile* outFile;
  550. try
  551. {
  552. char path[256] = "";
  553. strcpy(path,RemoveLeadingSpace(curCmd+5));
  554. char srcPath[300] = "";
  555. if(strcmp(vCurrentDir,"/")==0)
  556. wsprintf(srcPath,"%s%s%s",vRealRoot,vCurrentDir,path);
  557. else
  558. wsprintf(srcPath,"%s%s\%s",vRealRoot,vCurrentDir,path);
  559. slashFix(srcPath);
  560. bool b_opened = true;
  561. try
  562. {
  563. if(type == 1)
  564. outFile = new CStdioFile(srcPath,CFile::modeWrite|CFile::modeNoTruncate|CFile::typeText|CFile::modeCreate);
  565. else if(type == 3)
  566. outFile = new CFile(srcPath,CFile::modeWrite|CFile::modeNoTruncate|CFile::typeBinary|CFile::modeCreate);
  567. }
  568. catch(...)
  569. {
  570. b_opened = false;
  571. }
  572. if(b_opened)
  573. {
  575. outFile->Seek(0,CFile::end);
  576. long nCount = 0;
  577. if(type == 1)
  578. wsprintf(str,"150 Opening ASCII data connection for %s%s",path,CRLF);
  579. else if(type == 3)
  580. wsprintf(str,"150 Opening BINARY data connection for %s%s",path,CRLF);
  581. SafeSend(str);
  582. DWORD start,end;
  583. HANDLE h = CreateEvent(NULL,false,false,"noname");
  584. do
  585. {
  586. start = GetTickCount();
  587. nCount = dbsocket->Receive((char*)lpDataBuf_in,m_buf_length_in,60);
  588. end = GetTickCount();
  589. outFile->Write(lpDataBuf_in,nCount);
  590. if(b_speed_limit_in)
  591. {
  592. if((end-start) < 1000)
  593. {
  594. int time = end-start;
  595. WaitForSingleObject(h,1000-time);
  596. }
  597. }
  598. }while(nCount > 0);
  599. dbsocket->Close();
  600. if(b_passive)
  601. svSocketSrv->Close();
  602. delete outFile;
  603. wsprintf(str,"226 Transfer finished successfully. Data connection closed.%s",CRLF);
  604. SafeSend(str);
  605. }else
  606. {
  607. outFile->Close();
  608. dbsocket->Close();
  609. if(b_passive)
  610. svSocketSrv->Close();
  611. delete outFile;
  612. wsprintf(str,"426 Connection closed file opening failed %s",CRLF);
  613. SafeSend(str);
  614. }
  615. }catch(...)
  616. {
  617. outFile->Close();
  618. dbsocket->Close();
  619. if(b_passive)
  620. svSocketSrv->Close();
  621. delete outFile;
  622. wsprintf(str,"426 Connection closed; transfer aborted (transfer failed)%s",CRLF);
  623. SafeSend(str);
  624. }
  625. SetEvent(h_inprogress);
  626. return;
  627. }
  628. else if(strncmp("SIZE ",curCmd,5) == 0)
  629. {
  630. try
  631. {
  632. char path[256] = "";
  633. strcpy(path,RemoveLeadingSpace(curCmd+5));
  634. char srcPath[300] = "";
  635. if(strcmp(vCurrentDir,"/")==0)
  636. wsprintf(srcPath,"%s%s%s",vRealRoot,vCurrentDir,path);
  637. else
  638. wsprintf(srcPath,"%s%s\%s",vRealRoot,vCurrentDir,path);
  639. slashFix(srcPath);
  640. HANDLE Hfound;
  641. WIN32_FIND_DATA FoundFile;
  642. Hfound=FindFirstFile(srcPath,&FoundFile);
  643. DWORD size_t = FoundFile.nFileSizeHigh*MAXDWORD+FoundFile.nFileSizeLow;
  644. if(Hfound != INVALID_HANDLE_VALUE)
  645. wsprintf(str,"213 %lu%s",size_t,CRLF);
  646. else
  647. wsprintf(str,"213 0%s",CRLF);
  648. SafeSend(str);
  649. FindClose(Hfound);
  650. }catch(...)
  651. {
  652. wsprintf(str,"213 0%s",CRLF);
  653. SafeSend(str);
  654. }
  655. return;
  656. }
  657. else if(strncmp("MDTM ",curCmd,5) == 0)
  658. {
  659. try
  660. {
  661. char path[256] = "";
  662. strcpy(path,RemoveLeadingSpace(curCmd+5));
  663. char srcPath[300] = "";
  664. if(strcmp(vCurrentDir,"/")==0)
  665. wsprintf(srcPath,"%s%s%s",vRealRoot,vCurrentDir,path);
  666. else
  667. wsprintf(srcPath,"%s%s\%s",vRealRoot,vCurrentDir,path);
  668. slashFix(srcPath);
  669. HANDLE Hfound;
  670. WIN32_FIND_DATA FoundFile;
  671. FILETIME lft;
  672. SYSTEMTIME st;
  673. Hfound=FindFirstFile(srcPath,&FoundFile);
  674. FileTimeToLocalFileTime(&FoundFile.ftLastWriteTime,&lft);
  675. FileTimeToSystemTime(&lft,&st);
  676. wsprintf(str,"213 %.4d%.2d%.2d%.2d%.2d%.2d%s",st.wYear,st.wMonth,st.wDay,st.wHour,st.wMinute,st.wSecond,CRLF);
  677. SafeSend(str);
  678. FindClose(Hfound);
  679. }catch(...)
  680. {
  681. wsprintf(str,"213 0%s",CRLF);
  682. SafeSend(str);
  683. }
  684. return;
  685. }
  686. else if(strncmp("MKD ",curCmd,4) == 0)
  687. {
  688. char path[512] = "";
  689. strcpy(path,RemoveLeadingSpace(curCmd+4));
  690. if(vMkdir(path))
  691. wsprintf(str,"250 Requested file action ok , completed%s",CRLF);
  692. else
  693. {
  694. char lpMsgBuf[128]="";
  695. DWORD eRRR = GetLastError();
  696. FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS,NULL,eRRR,0,lpMsgBuf,128,NULL);
  697. wsprintf(str,"553 Requested action not taken , %s",lpMsgBuf);
  698. }
  699. SafeSend(str);
  700. return;
  701. }
  702. else if(strncmp("RMD ",curCmd,4) == 0)
  703. {
  704. char path[512] = "";
  705. strcpy(path,RemoveLeadingSpace(curCmd+4));
  706. if(vRmdir(path))
  707. wsprintf(str,"250 Requested file action ok , completed%s",CRLF);
  708. else
  709. {
  710. char lpMsgBuf[128]="";
  711. DWORD eRRR = GetLastError();
  712. FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS,NULL,eRRR,0,lpMsgBuf,128,NULL);
  713. wsprintf(str,"553 Requested action not taken , %s",lpMsgBuf);
  714. }
  715. SafeSend(str);
  716. return;
  717. }
  718. else if(strncmp("TYPE ",curCmd,5) == 0)
  719. {
  720. char path[8] = "";
  721. strcpy(path,RemoveLeadingSpace(curCmd+5));
  722. if(strcmp("I",path)==0)
  723. {
  724. type = 3;
  725. wsprintf(str,"200 Type set to IMAGE.%s",CRLF);
  726. }else if(strcmp("A",path)==0)
  727. {
  728. type = 1;
  729. wsprintf(str,"200 Type set to ASCII.%s",CRLF);
  730. }
  731. SafeSend(str);
  732. return;
  733. }
  734. else if(stricmp("SYST",curCmd) == 0)
  735. {
  736. // wsprintf(str,"215 %s%s",sinfo.OSVER,CRLF);
  737. wsprintf(str,"215 %s%s","UNIX",CRLF);
  738. SafeSend(str);
  739. return;
  740. }
  741. else if(stricmp("QUIT",curCmd) == 0)
  742. {
  743. wsprintf(str,"221 Goodbye.%s",CRLF);
  744. SafeSend(str);
  745. deleFtpPI(this);
  746. return;
  747. }
  748. else if(stricmp("REST ",curCmd) == 0)
  749. {
  750. char marker[32] = "";
  751. strcpy(marker,RemoveLeadingSpace(curCmd+5));
  752. restart_marker = atol(marker);
  753. wsprintf(str,"110 Command Ok , restarting position set to %u%s",restart_marker,CRLF);
  754. SafeSend(str);
  755. return;
  756. }
  757. else if(strncmp("RNFR ",curCmd,5) == 0)
  758. {
  759. strcpy(old_filename,RemoveLeadingSpace(curCmd+5));
  760. wsprintf(str,"350 Requested file action pending further information.%s",CRLF);
  761. SafeSend(str);
  762. return;
  763. }
  764. else if(strncmp("PORT ",curCmd,5) == 0)
  765. {
  766. char path[32] = "";
  767. int pHi=0;
  768. int pLo=0;
  769. int strOffset=0;
  770. int portOffset=0;
  771. int commaCnt=0;
  772. char ipStr[32]="";
  773. strcpy(path,RemoveLeadingSpace(curCmd+5));
  774. for(unsigned int i=0;i<strlen(path);i++)
  775. {
  776. if(path[i] == ',')
  777. {
  778. commaCnt++;
  779. path[i] = '.';
  780. }
  781. if(commaCnt == 4)
  782. {
  783. strOffset = i;
  784. i = strlen(path);
  785. }
  786. }
  787. strncpy(ipStr,path,strOffset);
  788. for(i=0;i<strlen(path+strOffset);i++)
  789. {
  790. if(path[strOffset+i] == ',')
  791. {
  792. portOffset = i;
  793. path[strOffset+i] = NULL;
  794. }
  795. }
  796. pHi = atoi(path+strOffset+1);
  797.             pLo = atoi(path+strOffset+portOffset+1);
  798. data_port = pHi*256 + pLo;
  799. strcpy(data_ip,ipStr);
  800. WaitForSingleObject(h_inprogress,10000);
  801. try
  802. {
  803. ResetEvent(h_inprogress);
  804. if(dbsocket != NULL)
  805. {
  806. dbsocket->Close();
  807. delete dbsocket;
  808. }
  809. dbsocket = new CBlockingSocket();
  810. if(!dbsocket->Create(SOCK_STREAM))
  811. wsprintf(str,"425 Cannot open data connection (%s %d)%s",data_ip,data_port,CRLF);
  812. else
  813. {
  814. CSockAddr tSockAdr(data_ip,data_port);
  815. if(dbsocket->Connect(tSockAdr))
  816. {
  817. SetEvent(h_dbsocket_ready_to_send);
  818. wsprintf(str,"200 Port command okay.%s",CRLF);
  819. }
  820. else
  821. wsprintf(str,"425 Cannot open data connection (%s %d)%s",data_ip,data_port,CRLF);
  822. }
  823. }catch(...)
  824. {
  825. wsprintf(str,"425 Cannot open data connection (%s %d)%s",data_ip,data_port,CRLF);
  826. }
  827. SafeSend(str);
  828. return;
  829. }
  830. else if(strcmp("NLST",curCmd) == 0)
  831. {
  832. ListEngine("*.*",false);
  833. return;
  834. }
  835. else if(strcmp("LIST",curCmd) == 0)
  836. {
  837. ListEngine("*.*",true);
  838. return;
  839. }
  840. else if(strncmp("NLST ",curCmd,5) == 0)
  841. {
  842. char path[256] = "";
  843. strcpy(path,RemoveLeadingSpace(curCmd+5));
  844. ListEngine(path,false);
  845. return;
  846. }
  847. else if(strncmp("LIST ",curCmd,5) == 0)
  848. {
  849. char path[256] = "";
  850. strcpy(path,RemoveLeadingSpace(curCmd+5));
  851. ListEngine(path,true);
  852. return;
  853. }
  854. else if(strncmp("RNTO ",curCmd,5) == 0)
  855. {
  856. strcpy(new_filename,RemoveLeadingSpace(curCmd+5));
  857. if(vMoveFile(old_filename,new_filename))
  858. wsprintf(str,"250 Requested file action ok , completed%s",CRLF);
  859. else
  860. {
  861. char lpMsgBuf[128]="";
  862. DWORD eRRR = GetLastError();
  863. FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_IGNORE_INSERTS,NULL,eRRR,0,lpMsgBuf,128,NULL);
  864. wsprintf(str,"553 Requested action not taken , %s%s",lpMsgBuf,CRLF);
  865. }
  866. SafeSend(str);
  867. return;
  868. }
  869. else if(stricmp("NOOP",curCmd) == 0)
  870. {
  871. wsprintf(str,"200 Command Ok , NOOP%s",CRLF);
  872. SafeSend(str);
  873. return;
  874. }
  875. else if(strcmp("PASV",curCmd) == 0)
  876. {
  877. try
  878. {
  879. int pHi=0;
  880. int pLo=0;
  881. char ipStr[32]="";
  882. WaitForSingleObject(h_inprogress,10000);
  883. ResetEvent(h_inprogress);
  884. if(dbsocket != NULL)
  885. {
  886. dbsocket->Close();
  887. delete dbsocket;
  888. }
  889. dbsocket = new CBlockingSocket();
  890. if(svSocketSrv != NULL)
  891. {
  892. svSocketSrv->Close();
  893. delete svSocketSrv;
  894. }
  895. svSocketSrv = new CFtpPasvSrv();
  896. if(1)//dbsocket->Create(SOCK_STREAM))
  897. {
  898. svSocketSrv->Create(SOCK_STREAM);
  899. m_port_from++;
  900. if(m_port_from >= m_port_to)
  901. m_port_from = m_port_from_preset;
  902. for(UINT i=m_port_from;i<=m_port_to;i++)
  903. {
  904. CSockAddr tSockAdr(INADDR_ANY,i);
  905. if(svSocketSrv->Bind(tSockAdr))
  906. {
  907. svSocketSrv->m_current_port = i;
  908. i = 65536;
  909. } 
  910. }
  911. if(svSocketSrv->m_current_port && svSocketSrv->Listen())
  912. {
  913. pHi = (svSocketSrv->m_current_port-svSocketSrv->m_current_port%256)/256 ;
  914. pLo = svSocketSrv->m_current_port%256;
  915. b_passive = true;
  916. CString ip;
  917. if(!b_switch_pasv_ip)
  918. ip = ipList.GetTail();
  919. else
  920. {
  921. pos = ipList.GetHeadPosition();
  922. for(int j=0;j<=ipRoller;j++)
  923. {
  924. ip = ipList.GetNext(pos);
  925. }
  926. if(ipRoller+1 < ipList.GetCount())
  927. ipRoller++;
  928. else
  929. ipRoller=0;
  930. }
  931. for(int i=0;i<ip.GetLength();i++)
  932. {
  933. if(ip.GetAt(i) == '.')
  934. ip.SetAt(i,',');
  935. }
  936. wsprintf(str,"227 Entering Passive Mode (%s,%d,%d)%s",(LPCSTR)ip,pHi,pLo,CRLF);
  937. SafeSend(str);
  938. svSocketSrv->Accept(*dbsocket,(SOCKADDR*)&lpsockftpd);
  939. wsprintf(str,"%d.%d.%d.%d",lpsockftpd.sin_addr.S_un.S_un_b.s_b1,lpsockftpd.sin_addr.S_un.S_un_b.s_b2,lpsockftpd.sin_addr.S_un.S_un_b.s_b3,lpsockftpd.sin_addr.S_un.S_un_b.s_b4);
  940. if(!isAllowedIp(str))
  941. {
  942. #ifdef _TEST
  943. MessageBox(NULL,"ERROR","ERROR_PASV_NOT_ALLOWED_IP",MB_OK);
  944. #endif
  945. canclePasvListenSocket(svSocketSrv);
  946. wsprintf(str,"425 Cannot open data connection (client ip denied)%s",CRLF);
  947. SafeSend(str);
  948. }
  949. else
  950. SetEvent(h_dbsocket_ready_to_send);
  951. return;
  952. }
  953. else
  954. {
  955. wsprintf(str,"425 Cannot open data connection (%s %d create failed)%s",(LPCSTR)ipList.GetTail(),svSocketSrv->m_current_port,CRLF);
  956. SafeSend(str);
  957. }
  958. }else
  959. {
  960. wsprintf(str,"425 Cannot open data connection (%s %d create failed)%s",(LPCSTR)ipList.GetTail(),svSocketSrv->m_current_port,CRLF);
  961. SafeSend(str);
  962. }
  963. }
  964. catch(...)
  965. {
  966. #ifdef _TEST
  967. MessageBox(NULL,"ERROR","ERROR_PASV",MB_OK);
  968. #endif
  969. }
  970. return;
  971. }
  972. else
  973. {
  974. if(strcmp(curCmd,"") !=0)
  975. {
  976. wsprintf(str,"502 Command %s not implemented%s",curCmd,CRLF);
  977. SafeSend(str);
  978. }
  979. return;
  980. }
  981. }
  982. }
  984. char* CFtpdPi::RemoveLeadingSpace(char *str)
  985. {
  986. try
  987. {
  988. int breakat = 0;
  989. for(unsigned int i=0;i<strlen(str);i++)
  990. {
  991. if(*(str+i) != ' ')
  992. {
  993. breakat = i;
  994. i = strlen(str);
  995. }
  996. }
  997. return str+breakat;
  998. }
  999. catch(...)
  1000. {
  1001. return "";
  1002. }
  1003. }
  1005. bool CFtpdPi::vMoveFile(LPCTSTR lpExistingFileName, LPCTSTR lpNewFileName)
  1006. {
  1007. char srcPath[512] = "";
  1008. char tgtPath[512] = "";
  1009. if(strcmp(vCurrentDir,"/")==0)
  1010. {
  1011. wsprintf(srcPath,"%s%s%s",vRealRoot,vCurrentDir,lpExistingFileName);
  1012. wsprintf(tgtPath,"%s%s%s",vRealRoot,vCurrentDir,lpNewFileName);
  1013. }else
  1014. {
  1015. wsprintf(srcPath,"%s%s\%s",vRealRoot,vCurrentDir,lpExistingFileName);
  1016. wsprintf(tgtPath,"%s%s\%s",vRealRoot,vCurrentDir,lpNewFileName);
  1017. }
  1018. if(MoveFile(srcPath,tgtPath))
  1019. return true;
  1020. else
  1021. {
  1022. return false;
  1023. }
  1024. }
  1026. bool CFtpdPi::isFilePresent(char *path)
  1027. {
  1028. WIN32_FIND_DATA MyFoundFile;
  1029. char srcPath[512] = "";
  1030. if(strcmp(vCurrentDir,"/")==0)
  1031. wsprintf(srcPath,"%s\%s",vRealRoot,path);
  1032. else
  1033. wsprintf(srcPath,"%s%s\%s",vRealRoot,vCurrentDir,path);
  1034. slashFix(srcPath);
  1035. HANDLE Hfound=FindFirstFile(srcPath,&MyFoundFile);
  1036. if(Hfound == INVALID_HANDLE_VALUE)
  1037. {
  1038. FindClose(Hfound);
  1039. return false;
  1040. }else
  1041. {
  1042. if(MyFoundFile.dwFileAttributes == FILE_ATTRIBUTE_DIRECTORY)
  1043. {
  1044. FindClose(Hfound);
  1045. return false;
  1046. }else
  1047. FindClose(Hfound);
  1048. return true;
  1049. }
  1050. }
  1052. bool CFtpdPi::isVpathPresent(char *vpath)
  1053. {
  1054. WIN32_FIND_DATA MyFoundFile;
  1055. char srcPath[512] = "";
  1056. if(strcmp(vpath,"/")==0)
  1057. wsprintf(srcPath,"%s",vRealRoot);
  1058. else
  1059. wsprintf(srcPath,"%s%s",vRealRoot,vpath);
  1060. slashFix(srcPath);
  1061. HANDLE Hfound=FindFirstFile(srcPath,&MyFoundFile);
  1062. if(Hfound == INVALID_HANDLE_VALUE)
  1063. {
  1064. FindClose(Hfound);
  1065. return false;
  1066. }else
  1067. {
  1068. if(MyFoundFile.dwFileAttributes == FILE_ATTRIBUTE_DIRECTORY)
  1069. {
  1070. FindClose(Hfound);
  1071. return true;
  1072. }else
  1073. FindClose(Hfound);
  1074. return false;
  1075. }
  1076. }
  1078. char* CFtpdPi::fixVpath(char *vPath)
  1079. {
  1080. char retPath[512]="";
  1081. strcpy(retPath,replaceDotInVpath(vPath));
  1082. if(strncmp("/",retPath,1)==0)
  1083. {
  1084. return vPath;
  1085. }
  1086. else
  1087. {
  1088. if(strcmp("/",vCurrentDir) == 0)
  1089. {
  1090. wsprintf(vPath,"/%s",retPath);
  1091. return vPath;
  1092. }
  1093. else
  1094. {
  1095. wsprintf(vPath,"%s/%s",vCurrentDir,retPath);
  1096. return vPath;
  1097. }
  1098. }
  1099. }
  1101. char* CFtpdPi::replaceDotInVpath(char *vPath)
  1102. {
  1103. char vCurDirTmp[512] = "";
  1104. strcpy(vCurDirTmp,vCurrentDir);
  1105. char* outPut = (char*) malloc(512);
  1106. int outPos = 0;
  1107. bool dotAgain =  false;
  1108. memset(outPut,NULL,512);
  1109. for(unsigned int i =0;i<=strlen(vPath)-1;i++)
  1110. {
  1111. if(*(vPath+i) == '.')
  1112. {
  1113. if(*(vPath+i+1) == '/' || *(vPath+i+1) == NULL)
  1114. {
  1115. i=i+2;
  1116. }
  1117. else if(*(vPath+i+1) == '.' && (*(vPath+i+2) == '/' || *(vPath+i+2) == NULL))
  1118. {
  1119. if(dotAgain)
  1120. strcpy(vCurDirTmp,outPut);
  1121. if(!goParent(vCurDirTmp))
  1122. {
  1124. }else
  1125. {
  1126. strcpy(outPut,vCurDirTmp);
  1127. if(strcmp(outPut,"/") == 0)
  1128. {
  1129. outPos = 1;
  1130. i=i+2;
  1131. }else
  1132. {
  1133. i=i+2;
  1134. if(i <= strlen(vPath)-1)
  1135. {
  1136. int a = strlen(outPut);
  1137. *(outPut+a) = '/';
  1138. *(outPut+a+1) = NULL;
  1139. }
  1140. outPos = strlen(outPut);
  1141. }
  1142. dotAgain = true;
  1143. }
  1144. }
  1145. else
  1146. {
  1147. *(outPut+outPos) = *(vPath+i);
  1148. outPos++;
  1149. }
  1150. }else
  1151. {
  1152. *(outPut+outPos) = *(vPath+i);
  1153. outPos++;
  1154. }
  1155. }
  1156. *(outPut+outPos) = NULL;
  1157. strcpy(vPath,outPut);
  1158. free(outPut);
  1159. return vPath;
  1160. }
  1162. bool CFtpdPi::goParent(char *path)
  1163. {
  1164. for(unsigned int i=strlen(path)-1;i>=0;i--)
  1165. {
  1166. if( *(path+i) == '/' && i != strlen(path)-1)
  1167. {
  1168. if(i>0)
  1169. {
  1170. *(path+i) = NULL;
  1171. i=0;
  1172. }else
  1173. *(path+1) = NULL;
  1174. return true;
  1175. }
  1176. }
  1177. return false;
  1178. }
  1180. bool CFtpdPi::vRemove(char *file)
  1181. {
  1182. char srcPath[512] = "";
  1183. if(strcmp(vCurrentDir,"/")==0)
  1184. wsprintf(srcPath,"%s%s%s",vRealRoot,vCurrentDir,file);
  1185. else
  1186. wsprintf(srcPath,"%s%s\%s",vRealRoot,vCurrentDir,file);
  1187. slashFix(srcPath);
  1188. if(remove(srcPath)==0)
  1189. return true;
  1190. else
  1191. {
  1192. return false;
  1193. }
  1194. }
  1196. bool CFtpdPi::vMkdir(char *dir)
  1197. {
  1198. char srcPath[512] = "";
  1199. if(strcmp(vCurrentDir,"/")==0)
  1200. wsprintf(srcPath,"%s%s%s",vRealRoot,vCurrentDir,dir);
  1201. else
  1202. wsprintf(srcPath,"%s%s\%s",vRealRoot,vCurrentDir,dir);
  1203. slashFix(srcPath);
  1204. if(CreateDirectory(srcPath,NULL))
  1205. return true;
  1206. else
  1207. {
  1208. return false;
  1209. }
  1210. }
  1212. void CFtpdPi::slashFix(char *path)
  1213. {
  1214. for(unsigned int i=0;i<strlen(path);i++)
  1215. {
  1216. if(*(path+i) == '/')
  1217. *(path+i) = '\';
  1218. }
  1219. return;
  1220. }
  1222. bool CFtpdPi::vRmdir(char *path)
  1223. {
  1224. char srcPath[512] = "";
  1225. if(strcmp(vCurrentDir,"/")==0)
  1226. wsprintf(srcPath,"%s%s%s",vRealRoot,vCurrentDir,path);
  1227. else
  1228. wsprintf(srcPath,"%s%s\%s",vRealRoot,vCurrentDir,path);
  1229. slashFix(srcPath);
  1230. if(RemoveDirectory(srcPath))
  1231. return true;
  1232. else
  1233. {
  1234. return false;
  1235. }
  1236. }
  1239. void CFtpdPi::slashFixUnix(char *path)
  1240. {
  1241. for(unsigned int i=0;i<strlen(path);i++)
  1242. {
  1243. if(*(path+i) == '\')
  1244. *(path+i) = '/';
  1245. }
  1246. return;
  1247. }
  1249. bool CFtpdPi::validuser(char *buf)
  1250. {
  1251. return true;
  1252. }
  1254. bool CFtpdPi::matchpass(char *buf, char *uname)
  1255. {
  1256. return true;
  1257. }
  1260. void CFtpdPi::OnClose(int nErrorCode) 
  1261. {
  1262. // TODO: Add your specialized code here and/or call the base class
  1263. CFtpd* K = (CFtpd*) pFtpd;
  1264. K->RemoveClient(this);
  1265. CSocket::OnClose(nErrorCode);
  1266. }
  1268. void CFtpdPi::SetOutStreamSpeed(int kbs)
  1269. {
  1270. free(lpDataBuf_out);
  1271. lpDataBuf_out = (unsigned char*) malloc(kbs*1024);
  1272. m_buf_length_out = kbs*1024;
  1273. b_speed_limit_out = true;
  1274. return;
  1275. }
  1277. void CFtpdPi::SetInStreamSpeed(int kbs)
  1278. {
  1279. free(lpDataBuf_in);
  1280. lpDataBuf_in = (unsigned char*) malloc(kbs*1024);
  1281. m_buf_length_in = kbs*1024;
  1282. b_speed_limit_in = true;
  1283. return;
  1284. }
  1286. void CFtpdPi::SafeSend(char *str)
  1287. {
  1288. int len = strlen(str);
  1289. try
  1290. {
  1291. int result = Send(str,len,0);
  1292. if(result < len)
  1293. {
  1294. #ifdef _TEST
  1295. MessageBox(NULL,"ERROR","ERROR_SEND_TWICE",MB_OK);
  1296. #endif
  1297. result = Send(str+result,len-result,0);
  1298. }
  1299. }catch(...)
  1300. {
  1301. #ifdef _TEST
  1302. MessageBox(NULL,"ERROR","ERROR_SEND_EXCEPTION",MB_OK);
  1303. #endif
  1304. }
  1305. }
  1307. bool CFtpdPi::isAllowedIp(char *ip)
  1308. {
  1309. for(int i = 0;i < clientIpCount;i++)
  1310. {
  1311. if(strncmp(ip,clientIp[i],strlen(clientIp[i]))==0)
  1312. return true;
  1313. }
  1314. return false;
  1315. }
  1317. void CFtpdPi::addClientIp(char *ip)
  1318. {
  1319. strcpy(clientIp[clientIpCount],ip);
  1320. if(clientIpCount < 32)
  1321. clientIpCount++;
  1322. return;
  1323. }
  1325. void CFtpdPi::LoadIpRange()
  1326. {
  1327. char buf[32] = "";
  1328. CStdioFile ifs(ipfile , CFile::modeRead);
  1329. if(ifs.ReadString(buf,32) != NULL)
  1330. {
  1331. do
  1332. {
  1333. int l = strlen(buf) - 1;
  1334. if(l >= 0)
  1335. {
  1336. if(*(buf+l) == 'n')
  1337. *(buf+l) = NULL;
  1338. }
  1339. addClientIp(buf);
  1340. }while(ifs.ReadString(buf,32) != NULL);
  1341. }
  1342. return;
  1343. }
  1345. void CFtpdPi::ListEngine(char *vpath, bool longList)
  1346. {
  1347. char path[256] = "";
  1348. HANDLE Hfound;
  1349. try
  1350. {
  1351. strcpy(path,vpath);
  1352. WaitForSingleObject(h_dbsocket_ready_to_send,5000);
  1353. ResetEvent(h_dbsocket_ready_to_send);
  1354. char pattern[256] = "";
  1355. if(strncmp(path,"-la",3)==0 || strncmp(path,"-l",2)==0 || strncmp(path,"-a",2)==0 || strcmp(path,"*")==0 || strcmp(path,".")==0)
  1356. strcpy(pattern,"*.*");
  1357. else if(strstr(path,"*.")!=NULL || strstr(path,"/*")!=NULL)
  1358. strcpy(pattern,path);
  1359. else
  1360. {
  1361. strcpy(pattern,path);
  1362. if(strcmp(pattern,"/")==0)
  1363. strcat(pattern,"*.*");
  1364. else
  1365. strcat(pattern,"/*.*");
  1366. }
  1367. fixVpath(pattern);
  1368. WIN32_FIND_DATA FoundFile;
  1369. char dbyte = '-';
  1370. char perm[4] = "rwx";
  1371. char temp1[300]="";
  1372. FILETIME lft;
  1373. SYSTEMTIME st;
  1374. strcpy(perm,"rwx");
  1375. if(strcmp(pattern,"/")==0)
  1376. wsprintf(temp1,"%s\*.*",vRealRoot);
  1377. else
  1378. wsprintf(temp1,"%s%s",vRealRoot,pattern);
  1379. slashFix(temp1);
  1380. wsprintf(str,"150 Opening ASCII NO-PRINT mode data connection for ls -l.%s",CRLF);
  1381. SafeSend(str);
  1382. Hfound=FindFirstFile(temp1,&FoundFile);
  1383. if(FoundFile.dwFileAttributes == FILE_ATTRIBUTE_DIRECTORY)
  1384. dbyte = 'd';
  1385. else
  1386. dbyte = '-';
  1387. FileTimeToLocalFileTime(&FoundFile.ftLastWriteTime,&lft);
  1388. FileTimeToSystemTime(&lft,&st);
  1389. char timeStr2[16] = "";
  1390. if(st.wYear != MyTime.wYear)
  1391. wsprintf(timeStr2,"%5d",st.wYear);
  1392. else
  1393. wsprintf(timeStr2,"%.2d:%.2d",st.wHour,st.wMinute);
  1394. if(!longList)
  1395. {
  1396. if(strcmp(FoundFile.cFileName,".") !=0 && strcmp(FoundFile.cFileName,"..") !=0)
  1397. {
  1398. wsprintf(str,"%s%s",FoundFile.cFileName,CRLF);
  1399. dbsocket->Send(str,strlen(str),20);
  1400. }
  1401. }
  1402. else
  1403. {
  1404. wsprintf(str,"%c%s%s%s  1 %-9s %-9s %10u %s %d %s %s%s",dbyte,perm,perm,perm,(LPCSTR)m_username,(LPCSTR)m_group,(FoundFile.nFileSizeHigh*MAXDWORD)+FoundFile.nFileSizeLow,monthStr[st.wMonth],st.wDay,timeStr2,FoundFile.cFileName,CRLF);
  1405. dbsocket->Send(str,strlen(str),20);
  1406. }
  1407. while(FindNextFile(Hfound,&FoundFile))
  1408. {
  1409. if(FoundFile.dwFileAttributes == FILE_ATTRIBUTE_DIRECTORY)
  1410. dbyte = 'd';
  1411. else
  1412. dbyte = '-';
  1413. FileTimeToLocalFileTime(&FoundFile.ftLastWriteTime,&lft);
  1414. FileTimeToSystemTime(&lft,&st);
  1415. if(st.wYear != MyTime.wYear)
  1416. wsprintf(timeStr2,"%5d",st.wYear);
  1417. else
  1418. wsprintf(timeStr2,"%.2d:%.2d",st.wHour,st.wMinute);
  1419. if(!longList)
  1420. {
  1421. if(strcmp(FoundFile.cFileName,".") !=0 && strcmp(FoundFile.cFileName,"..") !=0)
  1422. {
  1423. wsprintf(str,"%s%s",FoundFile.cFileName,CRLF);
  1424. dbsocket->Send(str,strlen(str),20);
  1425. }
  1426. }
  1427. else
  1428. {
  1429. wsprintf(str,"%c%s%s%s  1 %-9s %-9s %10u %s %d %s %s%s",dbyte,perm,perm,perm,(LPCSTR)m_username,(LPCSTR)m_group,(FoundFile.nFileSizeHigh*MAXDWORD)+FoundFile.nFileSizeLow,monthStr[st.wMonth],st.wDay,timeStr2,FoundFile.cFileName,CRLF);
  1430. dbsocket->Send(str,strlen(str),20);
  1431. }
  1432. }
  1433. FindClose(Hfound);
  1434. dbsocket->Close();
  1435. if(b_passive)
  1436. svSocketSrv->Close();
  1437. wsprintf(str,"226 Transfer finished successfully. Data connection closed.%s",CRLF);
  1438. SafeSend(str);
  1439. SetEvent(h_inprogress);
  1440. }
  1441. catch(...)
  1442. {
  1443. FindClose(Hfound);
  1444. dbsocket->Close();
  1445. if(b_passive)
  1446. svSocketSrv->Close();
  1447. wsprintf(str,"426 Connection closed; transfer aborted (transfer failed)%s",CRLF);
  1448. SafeSend(str);
  1449. SetEvent(h_inprogress);
  1450. }
  1451. }