TransferSocket.cpp 37 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460
  1. // FileZilla - a Windows ftp client
  2. // Copyright (C) 2002-2004 - Tim Kosse <[email protected]>
  3. // This program is free software; you can redistribute it and/or
  4. // modify it under the terms of the GNU General Public License
  5. // as published by the Free Software Foundation; either version 2
  6. // of the License, or (at your option) any later version.
  7. // This program is distributed in the hope that it will be useful,
  8. // but WITHOUT ANY WARRANTY; without even the implied warranty of
  9. // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  10. // GNU General Public License for more details.
  11. // You should have received a copy of the GNU General Public License
  12. // along with this program; if not, write to the Free Software
  13. // Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  14. // TransferSocket.cpp: Implementierungsdatei
  15. //
  16. #include "stdafx.h"
  17. #ifndef MPEXT
  18. #include "filezilla.h"
  19. #endif
  20. #include "TransferSocket.h"
  21. #include "mainthread.h"
  22. #include "AsyncProxySocketLayer.h"
  23. #ifndef MPEXT_NO_GSS
  24. #include "AsyncGssSocketLayer.h"
  25. #endif
  26. #ifdef _DEBUG
  27. #define new DEBUG_NEW
  28. #undef THIS_FILE
  29. static char THIS_FILE[] = __FILE__;
  30. #endif
  31. #define BUFSIZE 16384
  32. #define STATE_WAITING 0
  33. #define STATE_STARTING 1
  34. #define STATE_STARTED 2
  35. /////////////////////////////////////////////////////////////////////////////
  36. // CTransferSocket
  37. CTransferSocket::CTransferSocket(CFtpControlSocket *pOwner, int nMode)
  38. {
  39. ASSERT(pOwner);
  40. InitLog(pOwner);
  41. m_pOwner = pOwner;
  42. m_nMode = nMode;
  43. m_nTransferState = STATE_WAITING;
  44. m_bCheckTimeout = FALSE;
  45. m_pBuffer = 0;
  46. #ifndef MPEXT_NO_ZLIB
  47. m_pBuffer2 = 0;
  48. #endif
  49. m_bufferpos = 0;
  50. m_pFile = 0;
  51. m_bListening = FALSE;
  52. m_bSentClose = FALSE;
  53. m_nInternalMessageID = 0;
  54. m_transferdata.transfersize = 0;
  55. m_transferdata.transferleft = 0;
  56. m_transferdata.nTransferStart = 0;
  57. m_nNotifyWaiting = 0;
  58. m_bShutDown = FALSE;
  59. UpdateStatusBar(true);
  60. for (int i = 0; i < SPEED_SECONDS; i++)
  61. {
  62. m_Transfered[i] = 0;
  63. m_UsedForTransfer[i] = 0;
  64. }
  65. m_pProxyLayer = NULL;
  66. #ifndef MPEXT_NO_SSL
  67. m_pSslLayer = NULL;
  68. #endif
  69. #ifndef MPEXT_NO_GSS
  70. m_pGssLayer = NULL;
  71. #endif
  72. if (m_nMode & CSMODE_LIST)
  73. {
  74. m_pListResult = new CFtpListResult(pOwner->m_CurrentServer, &pOwner->m_bUTF8);
  75. m_pListResult->InitLog(this);
  76. }
  77. else
  78. m_pListResult = 0;
  79. m_LastUpdateTime.QuadPart = 0;
  80. #ifndef MPEXT_NO_ZLIB
  81. memset(&m_zlibStream, 0, sizeof(m_zlibStream));
  82. m_useZlib = false;
  83. #endif
  84. }
  85. CTransferSocket::~CTransferSocket()
  86. {
  87. LogMessage(__FILE__, __LINE__, this,FZ_LOG_DEBUG, _T("~CTransferSocket()"));
  88. delete [] m_pBuffer;
  89. #ifndef MPEXT_NO_ZLIB
  90. delete [] m_pBuffer2;
  91. #endif
  92. PostMessage(m_pOwner->m_pOwner->m_hOwnerWnd, m_pOwner->m_pOwner->m_nReplyMessageID, FZ_MSG_MAKEMSG(FZ_MSG_TRANSFERSTATUS, 0), 0);
  93. Close();
  94. RemoveAllLayers();
  95. delete m_pProxyLayer;
  96. #ifndef MPEXT_NO_SSL
  97. delete m_pSslLayer;
  98. #endif
  99. #ifndef MPEXT_NO_GSS
  100. delete m_pGssLayer;
  101. #endif
  102. m_pOwner->RemoveActiveTransfer();
  103. delete m_pListResult;
  104. #ifndef MPEXT_NO_ZLIB
  105. if (m_useZlib)
  106. {
  107. if (m_nMode & CSMODE_UPLOAD)
  108. deflateEnd(&m_zlibStream);
  109. else
  110. inflateEnd(&m_zlibStream);
  111. }
  112. #endif
  113. }
  114. /////////////////////////////////////////////////////////////////////////////
  115. // Member-Funktion CTransferSocket
  116. void CTransferSocket::OnReceive(int nErrorCode)
  117. {
  118. if (GetState() != connected && GetState() != attached && GetState() != closed)
  119. return;
  120. if (m_nTransferState == STATE_WAITING)
  121. {
  122. m_nNotifyWaiting |= FD_READ;
  123. return;
  124. }
  125. if (m_bSentClose)
  126. return;
  127. if (m_bListening)
  128. return;
  129. if (m_nMode&CSMODE_LIST)
  130. {
  131. if (m_nTransferState == STATE_STARTING)
  132. OnConnect(0);
  133. char *buffer = new char[BUFSIZE];
  134. int numread = CAsyncSocketEx::Receive(buffer, BUFSIZE);
  135. if (numread != SOCKET_ERROR && numread)
  136. {
  137. m_LastActiveTime = CTime::GetCurrentTime();
  138. UpdateRecvLed();
  139. #ifndef MPEXT_NO_ZLIB
  140. if (m_useZlib)
  141. {
  142. m_zlibStream.next_in = (Bytef *)buffer;
  143. m_zlibStream.avail_in = numread;
  144. char *out = new char[BUFSIZE];
  145. m_zlibStream.next_out = (Bytef *)out;
  146. m_zlibStream.avail_out = BUFSIZE;
  147. int res = inflate(&m_zlibStream, 0);
  148. while (res == Z_OK)
  149. {
  150. m_pListResult->AddData(out, BUFSIZE - m_zlibStream.avail_out);
  151. out = new char[BUFSIZE];
  152. m_zlibStream.next_out = (Bytef *)out;
  153. m_zlibStream.avail_out = BUFSIZE;
  154. res = inflate(&m_zlibStream, 0);
  155. }
  156. delete [] buffer;
  157. if (res == Z_STREAM_END)
  158. m_pListResult->AddData(out, BUFSIZE - m_zlibStream.avail_out);
  159. else if (res != Z_OK && res != Z_BUF_ERROR)
  160. {
  161. delete [] out;
  162. Close();
  163. if (!m_bSentClose)
  164. {
  165. m_nMode |= CSMODE_TRANSFERERROR;
  166. m_bSentClose = TRUE;
  167. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  168. }
  169. return;
  170. }
  171. else
  172. delete [] out;
  173. }
  174. else
  175. #endif
  176. m_pListResult->AddData(buffer, numread);
  177. m_transferdata.transfersize += numread;
  178. CTimeSpan timespan = CTime::GetCurrentTime() - m_StartTime;
  179. int elapsed = (int)timespan.GetTotalSeconds();
  180. //TODO
  181. //There are servers which report the total number of
  182. //bytes in the list response message, but yet it is not supported by FZ.
  183. /*double leftmodifier=(transfersize-transferstart-transferleft);
  184. leftmodifier*=100;
  185. leftmodifier/=(transfersize-transferstart);
  186. if (leftmodifier==0)
  187. leftmodifier=1;
  188. double leftmodifier2=100-leftmodifier;
  189. int left=(int)((elapsed/leftmodifier)*leftmodifier2);
  190. int percent=MulDiv(100,transfersize-transferleft,transfersize);*/
  191. int transferrate=static_cast<int>( (elapsed && m_transferdata.transfersize)?m_transferdata.transfersize/elapsed:0 );
  192. t_ffam_transferstatus *status = new t_ffam_transferstatus;
  193. status->bFileTransfer = FALSE;
  194. #ifdef MPEXT
  195. status->transfersize = -1;
  196. #endif
  197. status->bytes = m_transferdata.transfersize;
  198. status->percent = -1;
  199. status->timeelapsed = elapsed;
  200. status->timeleft = -1;
  201. status->transferrate = transferrate;
  202. PostMessage(m_pOwner->m_pOwner->m_hOwnerWnd, m_pOwner->m_pOwner->m_nReplyMessageID, FZ_MSG_MAKEMSG(FZ_MSG_TRANSFERSTATUS, 0), (LPARAM)status);
  203. }
  204. else
  205. delete [] buffer;
  206. if (!numread)
  207. {
  208. Close();
  209. if (!m_bSentClose)
  210. {
  211. m_bSentClose = TRUE;
  212. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  213. }
  214. }
  215. if (numread == SOCKET_ERROR)
  216. {
  217. int nError = GetLastError();
  218. if (nError == WSAENOTCONN)
  219. {
  220. //Not yet connected
  221. return;
  222. }
  223. #ifndef MPEXT_NO_SSL
  224. else if (m_pSslLayer && nError == WSAESHUTDOWN)
  225. {
  226. // Do nothing, wait for shutdown complete notification.
  227. return;
  228. }
  229. #endif
  230. else if (nError != WSAEWOULDBLOCK)
  231. {
  232. Close();
  233. if (!m_bSentClose)
  234. {
  235. m_nMode |= CSMODE_TRANSFERERROR;
  236. m_bSentClose = TRUE;
  237. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  238. }
  239. }
  240. }
  241. }
  242. else if (m_nMode & CSMODE_DOWNLOAD)
  243. {
  244. if (m_nTransferState == STATE_STARTING)
  245. OnConnect(0);
  246. bool beenWaiting = false;
  247. _int64 ableToRead;
  248. if (GetState() != closed)
  249. ableToRead = m_pOwner->GetAbleToTransferSize(CControlSocket::download, beenWaiting);
  250. else
  251. ableToRead = BUFSIZE;
  252. if (!beenWaiting)
  253. ASSERT(ableToRead);
  254. else if (!ableToRead)
  255. {
  256. TriggerEvent(FD_READ);
  257. return;
  258. }
  259. if (!m_pBuffer)
  260. m_pBuffer = new char[BUFSIZE];
  261. int numread = CAsyncSocketEx::Receive(m_pBuffer, static_cast<int>(ableToRead));
  262. if (numread!=SOCKET_ERROR)
  263. {
  264. Transfered( numread, CTime::GetCurrentTime());
  265. m_pOwner->SpeedLimitAddTransferredBytes(CControlSocket::download, numread);
  266. }
  267. if (!numread)
  268. {
  269. Close();
  270. if (!m_bSentClose)
  271. {
  272. m_bSentClose = TRUE;
  273. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  274. }
  275. return;
  276. }
  277. if (numread == SOCKET_ERROR)
  278. {
  279. int nError = GetLastError();
  280. if (nError == WSAENOTCONN)
  281. {
  282. //Not yet connected
  283. return;
  284. }
  285. #ifndef MPEXT_NO_SSL
  286. else if (m_pSslLayer && nError == WSAESHUTDOWN)
  287. {
  288. // Do nothing, wait for shutdown complete notification.
  289. return;
  290. }
  291. #endif
  292. else if (nError != WSAEWOULDBLOCK)
  293. {
  294. Close();
  295. if (!m_bSentClose)
  296. {
  297. m_nMode |= CSMODE_TRANSFERERROR;
  298. m_bSentClose = TRUE;
  299. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  300. }
  301. }
  302. UpdateStatusBar(false);
  303. return;
  304. }
  305. int written = 0;
  306. m_LastActiveTime = CTime::GetCurrentTime();
  307. UpdateRecvLed();
  308. TRY
  309. {
  310. #ifndef MPEXT_NO_ZLIB
  311. if (m_useZlib)
  312. {
  313. if (!m_pBuffer2)
  314. m_pBuffer2 = new char[BUFSIZE];
  315. m_zlibStream.next_in = (Bytef *)m_pBuffer;
  316. m_zlibStream.avail_in = numread;
  317. m_zlibStream.next_out = (Bytef *)m_pBuffer2;
  318. m_zlibStream.avail_out = BUFSIZE;
  319. int res = inflate(&m_zlibStream, 0);
  320. while (res == Z_OK)
  321. {
  322. m_pFile->Write(m_pBuffer2, BUFSIZE - m_zlibStream.avail_out);
  323. written += BUFSIZE - m_zlibStream.avail_out;
  324. m_zlibStream.next_out = (Bytef *)m_pBuffer2;
  325. m_zlibStream.avail_out = BUFSIZE;
  326. res = inflate(&m_zlibStream, 0);
  327. }
  328. if (res == Z_STREAM_END)
  329. {
  330. m_pFile->Write(m_pBuffer2, BUFSIZE - m_zlibStream.avail_out);
  331. written += BUFSIZE - m_zlibStream.avail_out;
  332. }
  333. else if (res != Z_OK && res != Z_BUF_ERROR)
  334. {
  335. Close();
  336. if (!m_bSentClose)
  337. {
  338. m_nMode |= CSMODE_TRANSFERERROR;
  339. m_bSentClose = TRUE;
  340. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  341. }
  342. return;
  343. }
  344. }
  345. else
  346. #endif
  347. {
  348. m_pFile->Write(m_pBuffer, numread);
  349. written = numread;
  350. }
  351. }
  352. CATCH(CFileException,e)
  353. {
  354. LPTSTR msg = new TCHAR[BUFSIZE];
  355. if (e->GetErrorMessage(msg, BUFSIZE))
  356. m_pOwner->ShowStatus(msg, 1);
  357. delete [] msg;
  358. Close();
  359. if (!m_bSentClose)
  360. {
  361. m_nMode |= CSMODE_TRANSFERERROR;
  362. m_bSentClose = TRUE;
  363. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  364. }
  365. return;
  366. }
  367. END_CATCH;
  368. m_transferdata.transferleft -= written;
  369. UpdateStatusBar(false);
  370. }
  371. }
  372. void CTransferSocket::OnAccept(int nErrorCode)
  373. {
  374. LogMessage(__FILE__, __LINE__, this,FZ_LOG_DEBUG, _T("OnAccept(%d)"), nErrorCode);
  375. m_bListening=FALSE;
  376. CAsyncSocketEx tmp;
  377. Accept(tmp);
  378. SOCKET socket=tmp.Detach();
  379. CAsyncSocketEx::Close();
  380. Attach(socket);
  381. /* Set internal socket send buffer
  382. * this should fix the speed problems some users have reported
  383. */
  384. DWORD value;
  385. int len = sizeof(value);
  386. GetSockOpt(SO_SNDBUF, &value, &len);
  387. // MPEXT
  388. int sndbuf = COptions::GetOptionVal(OPTION_MPEXT_SNDBUF);
  389. if (value < sndbuf)
  390. {
  391. value = sndbuf;
  392. SetSockOpt(SO_SNDBUF, &value, sizeof(value));
  393. }
  394. if (m_nTransferState == STATE_STARTING)
  395. {
  396. m_nTransferState = STATE_STARTED;
  397. #ifndef MPEXT_NO_SSL
  398. if (m_pSslLayer)
  399. {
  400. AddLayer(m_pSslLayer);
  401. int res = m_pSslLayer->InitSSLConnection(true, m_pOwner->m_pSslLayer, COptions::GetOptionVal(OPTION_MPEXT_SSLSESSIONREUSE));
  402. #ifndef MPEXT_NO_SSLDLL
  403. if (res == SSL_FAILURE_LOADDLLS)
  404. m_pOwner->ShowStatus(IDS_ERRORMSG_CANTLOADSSLDLLS, 1);
  405. else
  406. #endif
  407. if (res == SSL_FAILURE_INITSSL)
  408. m_pOwner->ShowStatus(IDS_ERRORMSG_CANTINITSSL, 1);
  409. if (res)
  410. {
  411. Close();
  412. if (!m_bSentClose)
  413. {
  414. m_nMode |= CSMODE_TRANSFERERROR;
  415. m_bSentClose = TRUE;
  416. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  417. }
  418. return;
  419. }
  420. }
  421. #endif
  422. #ifndef MPEXT_NO_GSS
  423. if (m_pGssLayer)
  424. {
  425. AddLayer(m_pGssLayer);
  426. }
  427. #endif
  428. m_TransferedFirst = m_StartTime = CTime::GetCurrentTime();
  429. m_LastActiveTime = CTime::GetCurrentTime();
  430. }
  431. }
  432. void CTransferSocket::OnConnect(int nErrorCode)
  433. {
  434. LogMessage(__FILE__, __LINE__, this,FZ_LOG_DEBUG, _T("OnConnect(%d)"), nErrorCode);
  435. if (nErrorCode)
  436. {
  437. TCHAR buffer[1000];
  438. memset(buffer, 0, sizeof(buffer));
  439. FormatMessage(FORMAT_MESSAGE_FROM_SYSTEM, 0, nErrorCode, 0, buffer, 999, 0);
  440. CString str;
  441. str.Format(IDS_ERRORMSG_CANTOPENTRANSFERCHANNEL,buffer);
  442. str.Replace( _T("\n"), _T("\0") );
  443. str.Replace( _T("\r"), _T("\0") );
  444. m_pOwner->ShowStatus(str, 1);
  445. Close();
  446. if (!m_bSentClose)
  447. {
  448. m_nMode|=CSMODE_TRANSFERERROR;
  449. m_bSentClose=TRUE;
  450. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  451. }
  452. }
  453. else
  454. {
  455. /* Set internal socket send buffer
  456. * this should fix the speed problems some users have reported
  457. */
  458. DWORD value;
  459. int len = sizeof(value);
  460. GetSockOpt(SO_SNDBUF, &value, &len);
  461. // MPEXT
  462. int sndbuf = COptions::GetOptionVal(OPTION_MPEXT_SNDBUF);
  463. if (value < sndbuf)
  464. {
  465. value = sndbuf;
  466. SetSockOpt(SO_SNDBUF, &value, sizeof(value));
  467. }
  468. }
  469. if (m_nTransferState == STATE_STARTING)
  470. {
  471. m_nTransferState = STATE_STARTED;
  472. m_TransferedFirst = m_StartTime = CTime::GetCurrentTime();
  473. m_LastActiveTime=CTime::GetCurrentTime();
  474. #ifndef MPEXT_NO_SSL
  475. if (m_pSslLayer)
  476. {
  477. AddLayer(m_pSslLayer);
  478. int res = m_pSslLayer->InitSSLConnection(true, m_pOwner->m_pSslLayer, COptions::GetOptionVal(OPTION_MPEXT_SSLSESSIONREUSE));
  479. #ifndef MPEXT_NO_SSLDLL
  480. if (res == SSL_FAILURE_LOADDLLS)
  481. m_pOwner->ShowStatus(IDS_ERRORMSG_CANTLOADSSLDLLS, 1);
  482. else
  483. #endif
  484. if (res == SSL_FAILURE_INITSSL)
  485. m_pOwner->ShowStatus(IDS_ERRORMSG_CANTINITSSL, 1);
  486. if (res)
  487. {
  488. Close();
  489. if (!m_bSentClose)
  490. {
  491. m_nMode |= CSMODE_TRANSFERERROR;
  492. m_bSentClose = TRUE;
  493. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  494. }
  495. return;
  496. }
  497. }
  498. #endif
  499. #ifndef MPEXT_NO_GSS
  500. if (m_pGssLayer)
  501. {
  502. AddLayer(m_pGssLayer);
  503. }
  504. #endif
  505. }
  506. }
  507. void CTransferSocket::OnClose(int nErrorCode)
  508. {
  509. LogMessage(__FILE__, __LINE__, this, FZ_LOG_DEBUG, _T("OnClose(%d)"), nErrorCode);
  510. if (m_nTransferState == STATE_WAITING)
  511. {
  512. m_nNotifyWaiting |= FD_CLOSE;
  513. return;
  514. }
  515. OnReceive(0);
  516. Close();
  517. if (!m_bSentClose)
  518. {
  519. m_bSentClose=TRUE;
  520. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  521. }
  522. }
  523. int CTransferSocket::CheckForTimeout(int delay)
  524. {
  525. UpdateStatusBar(false);
  526. if (!m_bCheckTimeout)
  527. {
  528. // we are closed, so make sure the FTP control socket is itself checking for
  529. // timeout as we are not
  530. return 0;
  531. }
  532. CTimeSpan span = CTime::GetCurrentTime()-m_LastActiveTime;
  533. if (span.GetTotalSeconds()>=delay)
  534. {
  535. m_pOwner->ShowStatus(IDS_ERRORMSG_TIMEOUT, 1);
  536. Close();
  537. if (!m_bSentClose)
  538. {
  539. m_nMode |= CSMODE_TRANSFERTIMEOUT;
  540. m_bSentClose = TRUE;
  541. VERIFY(m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode));
  542. }
  543. return 2;
  544. }
  545. return 1;
  546. }
  547. void CTransferSocket::SetActive()
  548. {
  549. LogMessage(__FILE__, __LINE__, this, FZ_LOG_DEBUG, _T("SetActive()"));
  550. if (m_nTransferState == STATE_WAITING)
  551. m_nTransferState = STATE_STARTING;
  552. m_bCheckTimeout = TRUE;
  553. m_LastActiveTime = CTime::GetCurrentTime();
  554. if (m_nNotifyWaiting & FD_READ)
  555. OnReceive(0);
  556. if (m_nNotifyWaiting & FD_WRITE)
  557. OnSend(0);
  558. if (m_nNotifyWaiting & FD_CLOSE)
  559. OnClose(0);
  560. }
  561. void CTransferSocket::OnSend(int nErrorCode)
  562. {
  563. if (m_nTransferState == STATE_WAITING)
  564. {
  565. m_nNotifyWaiting |= FD_WRITE;
  566. return;
  567. }
  568. if (m_bSentClose)
  569. {
  570. return;
  571. }
  572. if (m_bListening)
  573. {
  574. return;
  575. }
  576. if (!(m_nMode&CSMODE_UPLOAD))
  577. {
  578. return;
  579. }
  580. if (m_nTransferState == STATE_STARTING)
  581. {
  582. OnConnect(0);
  583. }
  584. #ifndef MPEXT_NO_ZLIB
  585. if (m_useZlib)
  586. {
  587. if (!m_pBuffer)
  588. {
  589. m_pBuffer = new char[BUFSIZE];
  590. m_bufferpos = 0;
  591. m_zlibStream.next_out = (Bytef *)m_pBuffer;
  592. m_zlibStream.avail_out = BUFSIZE;
  593. }
  594. if (!m_pBuffer2)
  595. {
  596. m_pBuffer2 = new char[BUFSIZE];
  597. m_zlibStream.next_in = (Bytef *)m_pBuffer2;
  598. }
  599. bool beenWaiting = false;
  600. while (true)
  601. {
  602. int numsend;
  603. if (!m_zlibStream.avail_in)
  604. {
  605. if (m_pFile)
  606. {
  607. DWORD numread;
  608. numread = ReadDataFromFile(m_pBuffer2, BUFSIZE);
  609. if (numread < 0)
  610. {
  611. return;
  612. }
  613. m_transferdata.transferleft -= numread;
  614. m_zlibStream.next_in = (Bytef *)m_pBuffer2;
  615. m_zlibStream.avail_in = numread;
  616. if (numread < BUFSIZE)
  617. m_pFile = 0;
  618. }
  619. }
  620. if (!m_zlibStream.avail_out)
  621. {
  622. if (m_bufferpos >= BUFSIZE)
  623. {
  624. m_bufferpos = 0;
  625. m_zlibStream.next_out = (Bytef *)m_pBuffer;
  626. m_zlibStream.avail_out = BUFSIZE;
  627. }
  628. }
  629. int res = Z_OK;
  630. if (m_zlibStream.avail_out)
  631. {
  632. res = deflate(&m_zlibStream, m_pFile ? 0 : Z_FINISH);
  633. if (res != Z_OK && (!m_pFile && res != Z_STREAM_END))
  634. {
  635. m_pOwner->ShowStatus("ZLib error", 1);
  636. Close();
  637. if (!m_bSentClose)
  638. {
  639. m_nMode |= CSMODE_TRANSFERERROR;
  640. m_bSentClose = TRUE;
  641. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  642. }
  643. return;
  644. }
  645. }
  646. numsend = BUFSIZE;
  647. int len = BUFSIZE - m_bufferpos - m_zlibStream.avail_out;
  648. if (!len && !m_pFile)
  649. {
  650. break;
  651. }
  652. if (len < BUFSIZE)
  653. numsend = len;
  654. int nLimit = (int)m_pOwner->GetAbleToTransferSize(CControlSocket::upload, beenWaiting);
  655. if (nLimit != -1 && GetState() != closed && numsend > nLimit)
  656. numsend = nLimit;
  657. if (!numsend)
  658. {
  659. TriggerEvent(FD_WRITE);
  660. return;
  661. }
  662. int numsent = Send(m_pBuffer + m_bufferpos, numsend);
  663. if (numsent == SOCKET_ERROR)
  664. {
  665. int nError = GetLastError();
  666. if (nError == WSAENOTCONN)
  667. {
  668. //Not yet connected
  669. return;
  670. }
  671. #ifndef MPEXT_NO_SSL
  672. else if (m_pSslLayer && nError == WSAESHUTDOWN)
  673. {
  674. // Do nothing, wait for shutdown complete notification.
  675. return;
  676. }
  677. #endif
  678. else if (nError != WSAEWOULDBLOCK)
  679. {
  680. if (ShutDown() || GetLastError() != WSAEWOULDBLOCK)
  681. {
  682. Close();
  683. if (!m_bSentClose)
  684. {
  685. m_nMode |= CSMODE_TRANSFERERROR;
  686. m_bSentClose = TRUE;
  687. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  688. }
  689. }
  690. }
  691. UpdateStatusBar(false);
  692. return;
  693. }
  694. Transfered( numsent, CTime::GetCurrentTime());
  695. m_pOwner->SpeedLimitAddTransferredBytes(CControlSocket::upload, numsent);
  696. m_LastActiveTime = CTime::GetCurrentTime();
  697. UpdateSendLed();
  698. m_bufferpos += numsent;
  699. UpdateStatusBar(false);
  700. if (!m_zlibStream.avail_in && !m_pFile && m_zlibStream.avail_out &&
  701. m_zlibStream.avail_out + m_bufferpos == BUFSIZE && res == Z_STREAM_END)
  702. {
  703. if (ShutDown() || GetLastError() != WSAEWOULDBLOCK)
  704. {
  705. Close();
  706. if (!m_bSentClose)
  707. {
  708. m_bSentClose = TRUE;
  709. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  710. }
  711. }
  712. return;
  713. }
  714. //Check if there are other commands in the command queue.
  715. MSG msg;
  716. if (PeekMessage(&msg,0, 0, 0, PM_NOREMOVE))
  717. {
  718. TriggerEvent(FD_WRITE);
  719. return;
  720. }
  721. }
  722. }
  723. else
  724. #endif
  725. {
  726. if (!m_pFile)
  727. {
  728. return;
  729. }
  730. if (!m_pBuffer)
  731. m_pBuffer = new char[BUFSIZE];
  732. int numread;
  733. bool beenWaiting = false;
  734. _int64 currentBufferSize;
  735. if (GetState() != closed)
  736. currentBufferSize = m_pOwner->GetAbleToTransferSize(CControlSocket::upload, beenWaiting);
  737. else
  738. currentBufferSize = BUFSIZE;
  739. if (!currentBufferSize && !m_bufferpos)
  740. {
  741. // Not allowed to send yet, try later
  742. TriggerEvent(FD_WRITE);
  743. return;
  744. }
  745. else if (m_bufferpos < currentBufferSize)
  746. {
  747. numread = ReadDataFromFile(m_pBuffer + m_bufferpos, static_cast<int>(currentBufferSize - m_bufferpos));
  748. if (numread < 0 )
  749. {
  750. return;
  751. }
  752. else if (!numread && !m_bufferpos)
  753. {
  754. if (ShutDown() || GetLastError() != WSAEWOULDBLOCK)
  755. {
  756. Close();
  757. if (!m_bSentClose)
  758. {
  759. m_bSentClose = TRUE;
  760. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  761. }
  762. }
  763. return;
  764. }
  765. }
  766. else
  767. numread = 0;
  768. ASSERT((numread+m_bufferpos) <= BUFSIZE);
  769. ASSERT(numread>=0);
  770. ASSERT(m_bufferpos>=0);
  771. if (numread+m_bufferpos <= 0)
  772. {
  773. if (ShutDown() || GetLastError()!=WSAEWOULDBLOCK)
  774. {
  775. Close();
  776. if (!m_bSentClose)
  777. {
  778. m_bSentClose=TRUE;
  779. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  780. }
  781. }
  782. return;
  783. }
  784. int numsent = Send(m_pBuffer, numread + m_bufferpos);
  785. while (TRUE)
  786. {
  787. if (numsent != SOCKET_ERROR)
  788. {
  789. Transfered(numsent, CTime::GetCurrentTime());
  790. m_pOwner->SpeedLimitAddTransferredBytes(CControlSocket::upload, numsent);
  791. m_LastActiveTime = CTime::GetCurrentTime();
  792. UpdateSendLed();
  793. m_transferdata.transferleft -= numsent;
  794. }
  795. if (numsent==SOCKET_ERROR || !numsent)
  796. {
  797. int nError = GetLastError();
  798. if (nError == WSAENOTCONN)
  799. {
  800. //Not yet connected
  801. m_bufferpos += numread;
  802. return;
  803. }
  804. else if (nError == WSAEWOULDBLOCK)
  805. m_bufferpos += numread;
  806. #ifndef MPEXT_NO_SSL
  807. else if (m_pSslLayer && nError == WSAESHUTDOWN)
  808. {
  809. m_bufferpos += numread;
  810. // Do nothing, wait for shutdown complete notification.
  811. return;
  812. }
  813. #endif
  814. else
  815. {
  816. if (ShutDown() || GetLastError() != WSAEWOULDBLOCK)
  817. {
  818. Close();
  819. if (!m_bSentClose)
  820. {
  821. m_nMode |= CSMODE_TRANSFERERROR;
  822. m_bSentClose = TRUE;
  823. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  824. }
  825. }
  826. }
  827. UpdateStatusBar(false);
  828. return;
  829. }
  830. else
  831. {
  832. int pos = numread + m_bufferpos - numsent;
  833. if (pos < 0 || (numsent + pos) > BUFSIZE)
  834. {
  835. LogMessage(__FILE__, __LINE__, this, FZ_LOG_WARNING, _T("Index out of range"));
  836. if (ShutDown() || GetLastError() != WSAEWOULDBLOCK)
  837. {
  838. Close();
  839. if (!m_bSentClose)
  840. {
  841. m_nMode |= CSMODE_TRANSFERERROR;
  842. m_bSentClose = TRUE;
  843. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  844. }
  845. }
  846. return;
  847. }
  848. else if (!pos && numread < (currentBufferSize-m_bufferpos) && m_bufferpos != currentBufferSize)
  849. {
  850. if (ShutDown() || GetLastError() != WSAEWOULDBLOCK)
  851. {
  852. Close();
  853. if (!m_bSentClose)
  854. {
  855. m_bSentClose = TRUE;
  856. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  857. }
  858. }
  859. return;
  860. }
  861. else if (!pos)
  862. m_bufferpos = 0;
  863. else
  864. {
  865. memmove(m_pBuffer, m_pBuffer+numsent, pos);
  866. m_bufferpos=pos;
  867. }
  868. }
  869. //Check if there are other commands in the command queue.
  870. MSG msg;
  871. if (PeekMessage(&msg, 0, m_nInternalMessageID, m_nInternalMessageID, PM_NOREMOVE))
  872. {
  873. //Send resume message
  874. LogMessage(__FILE__, __LINE__, this, FZ_LOG_DEBUG, _T("Message waiting in queue, resuming later"));
  875. TriggerEvent(FD_WRITE);
  876. UpdateStatusBar(false);
  877. return;
  878. }
  879. UpdateStatusBar(false);
  880. if (GetState() != closed)
  881. currentBufferSize = m_pOwner->GetAbleToTransferSize(CControlSocket::upload, beenWaiting);
  882. else
  883. currentBufferSize = BUFSIZE;
  884. if (m_bufferpos < currentBufferSize)
  885. {
  886. numread = ReadDataFromFile(m_pBuffer + m_bufferpos, static_cast<int>(currentBufferSize - m_bufferpos));
  887. if (numread < 0 )
  888. {
  889. return;
  890. }
  891. else if (!numread && !m_bufferpos)
  892. {
  893. if (ShutDown() || GetLastError() != WSAEWOULDBLOCK)
  894. {
  895. Close();
  896. if (!m_bSentClose)
  897. {
  898. m_bSentClose = TRUE;
  899. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  900. }
  901. }
  902. return;
  903. }
  904. }
  905. else
  906. numread = 0;
  907. if (!currentBufferSize && !m_bufferpos)
  908. {
  909. // Not allowed to send yet, try later
  910. TriggerEvent(FD_WRITE);
  911. return;
  912. }
  913. ASSERT(numread>=0);
  914. ASSERT(m_bufferpos>=0);
  915. numsent = Send(m_pBuffer, numread+m_bufferpos);
  916. }
  917. }
  918. }
  919. void CTransferSocket::UpdateStatusBar(bool forceUpdate)
  920. {
  921. if (m_nTransferState != STATE_STARTED)
  922. return;
  923. if (!forceUpdate)
  924. {
  925. //Don't flood the main window with messages
  926. //Else performance would be really low
  927. LARGE_INTEGER curtime;
  928. LARGE_INTEGER freq;
  929. QueryPerformanceFrequency(&freq);
  930. QueryPerformanceCounter(&curtime);
  931. if (((curtime.QuadPart-m_LastUpdateTime.QuadPart) < (freq.QuadPart/15) ) )
  932. return;
  933. m_LastUpdateTime = curtime;
  934. }
  935. //Update the statusbar
  936. CTimeSpan timespan=CTime::GetCurrentTime()-m_StartTime;
  937. int elapsed=(int)timespan.GetTotalSeconds();
  938. t_ffam_transferstatus *status=new t_ffam_transferstatus;
  939. status->bFileTransfer = m_nMode & (CSMODE_DOWNLOAD | CSMODE_UPLOAD);
  940. #ifdef MPEXT
  941. status->transfersize = m_transferdata.transfersize;
  942. #endif
  943. status->timeelapsed=elapsed;
  944. status->bytes=m_transferdata.transfersize-m_transferdata.transferleft;
  945. if (m_transferdata.transfersize>0 && !(m_nMode&CSMODE_LIST))
  946. {
  947. double leftmodifier=static_cast<double>(m_transferdata.transfersize-m_transferdata.nTransferStart-m_transferdata.transferleft);
  948. leftmodifier*=100;
  949. if (m_transferdata.transfersize-m_transferdata.nTransferStart)
  950. leftmodifier /= (m_transferdata.transfersize-m_transferdata.nTransferStart);
  951. else
  952. leftmodifier = 1;
  953. if (leftmodifier == 0)
  954. leftmodifier = 1;
  955. double leftmodifier2 = 100 - leftmodifier;
  956. int left=static_cast<int>((elapsed/leftmodifier)*leftmodifier2);
  957. double percent=100*static_cast<double>(m_transferdata.transfersize-m_transferdata.transferleft);
  958. percent/=m_transferdata.transfersize;
  959. status->percent=static_cast<int>(percent);
  960. if (status->percent>100)
  961. status->percent=100;
  962. if (left < 0)
  963. left = -1;
  964. status->timeleft=left;
  965. }
  966. else
  967. {
  968. status->percent=-1;
  969. status->timeleft=-1;
  970. }
  971. int count = 0;
  972. status->transferrate = 0;
  973. for ( int i = 0; i < SPEED_SECONDS; i++)
  974. {
  975. if ( m_UsedForTransfer[ i])
  976. {
  977. status->transferrate += m_Transfered[ i];
  978. count++;
  979. }
  980. }
  981. if ( count > 0)
  982. status->transferrate = status->transferrate / count;
  983. else if (m_Transfered[0])
  984. status->transferrate = m_Transfered[0];
  985. else
  986. status->timeleft=-1;
  987. PostMessage(m_pOwner->m_pOwner->m_hOwnerWnd, m_pOwner->m_pOwner->m_nReplyMessageID, FZ_MSG_MAKEMSG(FZ_MSG_TRANSFERSTATUS, 0), (LPARAM)status);
  988. }
  989. void CTransferSocket::UpdateSendLed()
  990. {
  991. //Don't flood the main window with messages
  992. //Else performance would be really low
  993. LARGE_INTEGER curtime;
  994. LARGE_INTEGER freq;
  995. QueryPerformanceFrequency(&freq);
  996. QueryPerformanceCounter(&curtime);
  997. static LARGE_INTEGER oldtime={0};
  998. if ( ( (curtime.QuadPart-oldtime.QuadPart) < (freq.QuadPart/15) ) )
  999. return;
  1000. oldtime=curtime;
  1001. PostMessage(m_pOwner->m_pOwner->m_hOwnerWnd, m_pOwner->m_pOwner->m_nReplyMessageID, FZ_MSG_MAKEMSG(FZ_MSG_SOCKETSTATUS, FZ_SOCKETSTATUS_SEND), 0);
  1002. }
  1003. void CTransferSocket::UpdateRecvLed()
  1004. {
  1005. //Don't flood the main window with messages
  1006. //Else performance would be really low
  1007. LARGE_INTEGER curtime;
  1008. LARGE_INTEGER freq;
  1009. QueryPerformanceFrequency(&freq);
  1010. QueryPerformanceCounter(&curtime);
  1011. static LARGE_INTEGER oldtime={0};
  1012. if ( ( (curtime.QuadPart-oldtime.QuadPart) < (freq.QuadPart/15) ) )
  1013. return;
  1014. oldtime=curtime;
  1015. PostMessage(m_pOwner->m_pOwner->m_hOwnerWnd, m_pOwner->m_pOwner->m_nReplyMessageID, FZ_MSG_MAKEMSG(FZ_MSG_SOCKETSTATUS, FZ_SOCKETSTATUS_RECV), 0);
  1016. }
  1017. BOOL CTransferSocket::Create(
  1018. #ifndef MPEXT_NO_SSL
  1019. BOOL bUseSsl
  1020. #endif
  1021. )
  1022. {
  1023. #ifndef MPEXT_NO_SSL
  1024. if (bUseSsl)
  1025. m_pSslLayer = new CAsyncSslSocketLayer;
  1026. #endif
  1027. if (!m_pOwner->m_CurrentServer.fwbypass)
  1028. {
  1029. int nProxyType = COptions::GetOptionVal(OPTION_PROXYTYPE);
  1030. if (nProxyType != PROXYTYPE_NOPROXY)
  1031. {
  1032. USES_CONVERSION;
  1033. m_pProxyLayer = new CAsyncProxySocketLayer;
  1034. if (nProxyType == PROXYTYPE_SOCKS4)
  1035. m_pProxyLayer->SetProxy(PROXYTYPE_SOCKS4, T2CA(COptions::GetOption(OPTION_PROXYHOST)), COptions::GetOptionVal(OPTION_PROXYPORT));
  1036. else if (nProxyType == PROXYTYPE_SOCKS4A)
  1037. m_pProxyLayer->SetProxy(PROXYTYPE_SOCKS4A, T2CA(COptions::GetOption(OPTION_PROXYHOST)), COptions::GetOptionVal(OPTION_PROXYPORT));
  1038. else if (nProxyType == PROXYTYPE_SOCKS5)
  1039. if (COptions::GetOptionVal(OPTION_PROXYUSELOGON))
  1040. m_pProxyLayer->SetProxy(PROXYTYPE_SOCKS5, T2CA(COptions::GetOption(OPTION_PROXYHOST)),
  1041. COptions::GetOptionVal(OPTION_PROXYPORT),
  1042. T2CA(COptions::GetOption(OPTION_PROXYUSER)),
  1043. T2CA(CCrypt::decrypt(COptions::GetOption(OPTION_PROXYPASS))));
  1044. else
  1045. m_pProxyLayer->SetProxy(PROXYTYPE_SOCKS5, T2CA(COptions::GetOption(OPTION_PROXYHOST)),
  1046. COptions::GetOptionVal(OPTION_PROXYPORT));
  1047. else if (nProxyType == PROXYTYPE_HTTP11)
  1048. if (COptions::GetOptionVal(OPTION_PROXYUSELOGON))
  1049. m_pProxyLayer->SetProxy(PROXYTYPE_HTTP11, T2CA(COptions::GetOption(OPTION_PROXYHOST)), COptions::GetOptionVal(OPTION_PROXYPORT),
  1050. T2CA(COptions::GetOption(OPTION_PROXYUSER)),
  1051. T2CA(CCrypt::decrypt(COptions::GetOption(OPTION_PROXYPASS))));
  1052. else
  1053. m_pProxyLayer->SetProxy(PROXYTYPE_HTTP11, T2CA(COptions::GetOption(OPTION_PROXYHOST)), COptions::GetOptionVal(OPTION_PROXYPORT));
  1054. else
  1055. ASSERT(FALSE);
  1056. AddLayer(m_pProxyLayer);
  1057. }
  1058. }
  1059. if (!COptions::GetOptionVal(OPTION_LIMITPORTRANGE))
  1060. {
  1061. if (!CAsyncSocketEx::Create(0, SOCK_STREAM, FD_READ | FD_WRITE | FD_OOB | FD_ACCEPT | FD_CONNECT | FD_CLOSE, 0, GetFamily()))
  1062. return FALSE;
  1063. return TRUE;
  1064. }
  1065. else
  1066. {
  1067. int min=COptions::GetOptionVal(OPTION_PORTRANGELOW);
  1068. int max=COptions::GetOptionVal(OPTION_PORTRANGEHIGH);
  1069. if (min>=max)
  1070. {
  1071. m_pOwner->ShowStatus(IDS_ERRORMSG_CANTCREATEDUETOPORTRANGE,1);
  1072. return FALSE;
  1073. }
  1074. int startport=static_cast<int>(min+((double)rand()*(max-min))/(RAND_MAX+1));
  1075. int port=startport;
  1076. while (!CAsyncSocketEx::Create(port, SOCK_STREAM, FD_READ | FD_WRITE | FD_OOB | FD_ACCEPT | FD_CONNECT | FD_CLOSE, 0, GetFamily()))
  1077. {
  1078. port++;
  1079. if (port>max)
  1080. port=min;
  1081. if (port==startport)
  1082. {
  1083. m_pOwner->ShowStatus(IDS_ERRORMSG_CANTCREATEDUETOPORTRANGE,1);
  1084. return FALSE;
  1085. }
  1086. }
  1087. }
  1088. return TRUE;
  1089. }
  1090. void CTransferSocket::Close()
  1091. {
  1092. LogMessage(__FILE__, __LINE__, this,FZ_LOG_DEBUG, _T("Close()"));
  1093. m_bCheckTimeout = FALSE;
  1094. CAsyncSocketEx::Close();
  1095. }
  1096. int CTransferSocket::OnLayerCallback(std::list<t_callbackMsg>& callbacks)
  1097. {
  1098. for (std::list<t_callbackMsg>::iterator iter = callbacks.begin(); iter != callbacks.end(); iter++)
  1099. {
  1100. if (iter->nType == LAYERCALLBACK_STATECHANGE)
  1101. {
  1102. if (iter->pLayer == m_pProxyLayer)
  1103. LogMessage(__FILE__, __LINE__, this, FZ_LOG_INFO, _T("m_pProxyLayer changed state from %d to %d"), iter->nParam2, iter->nParam1);
  1104. #ifndef MPEXT_NO_SSL
  1105. else if (iter->pLayer == m_pSslLayer)
  1106. LogMessage(__FILE__, __LINE__, this, FZ_LOG_INFO, _T("m_pSslLayer changed state from %d to %d"), iter->nParam2, iter->nParam1);
  1107. #endif
  1108. #ifndef MPEXT_NO_GSS
  1109. else if (iter->pLayer == m_pGssLayer)
  1110. LogMessage(__FILE__, __LINE__, this, FZ_LOG_INFO, _T("m_pGssLayer changed state from %d to %d"), iter->nParam2, iter->nParam1);
  1111. #endif
  1112. else
  1113. LogMessage(__FILE__, __LINE__, this, FZ_LOG_INFO, _T("Layer @ %d changed state from %d to %d"), iter->pLayer, iter->nParam2, iter->nParam1);
  1114. }
  1115. else if (iter->nType == LAYERCALLBACK_LAYERSPECIFIC)
  1116. {
  1117. if (iter->pLayer == m_pProxyLayer)
  1118. {
  1119. switch (iter->nParam1)
  1120. {
  1121. case PROXYERROR_NOCONN:
  1122. m_pOwner->ShowStatus(IDS_ERRORMSG_PROXY_NOCONN, 1);
  1123. break;
  1124. case PROXYERROR_REQUESTFAILED:
  1125. m_pOwner->ShowStatus(IDS_ERRORMSG_PROXY_REQUESTFAILED, 1);
  1126. break;
  1127. case PROXYERROR_AUTHTYPEUNKNOWN:
  1128. m_pOwner->ShowStatus(IDS_ERRORMSG_PROXY_AUTHTYPEUNKNOWN, 1);
  1129. break;
  1130. case PROXYERROR_AUTHFAILED:
  1131. m_pOwner->ShowStatus(IDS_ERRORMSG_PROXY_AUTHFAILED, 1);
  1132. break;
  1133. case PROXYERROR_AUTHNOLOGON:
  1134. m_pOwner->ShowStatus(IDS_ERRORMSG_PROXY_AUTHNOLOGON, 1);
  1135. break;
  1136. case PROXYERROR_CANTRESOLVEHOST:
  1137. m_pOwner->ShowStatus(IDS_ERRORMSG_PROXY_CANTRESOLVEHOST, 1);
  1138. break;
  1139. default:
  1140. LogMessage(__FILE__, __LINE__, this, FZ_LOG_WARNING, _T("Unknown proxy error"));
  1141. }
  1142. }
  1143. #ifndef MPEXT_NO_SSL
  1144. else if (iter->pLayer == m_pSslLayer)
  1145. {
  1146. switch (iter->nParam1)
  1147. {
  1148. case SSL_INFO:
  1149. switch(iter->nParam2)
  1150. {
  1151. case SSL_INFO_SHUTDOWNCOMPLETE:
  1152. Close();
  1153. if (!m_bSentClose)
  1154. {
  1155. m_bSentClose=TRUE;
  1156. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  1157. }
  1158. break;
  1159. case SSL_INFO_ESTABLISHED:
  1160. m_pOwner->ShowStatus(IDS_STATUSMSG_SSLESTABLISHEDTRANSFER, 0);
  1161. TriggerEvent(FD_FORCEREAD);
  1162. break;
  1163. }
  1164. break;
  1165. case SSL_FAILURE:
  1166. switch (iter->nParam2)
  1167. {
  1168. case SSL_FAILURE_ESTABLISH:
  1169. m_pOwner->ShowStatus(IDS_ERRORMSG_CANTESTABLISHSSLCONNECTION, 1);
  1170. break;
  1171. #ifndef MPEXT_NO_SSLDLL
  1172. case SSL_FAILURE_LOADDLLS:
  1173. m_pOwner->ShowStatus(IDS_ERRORMSG_CANTLOADSSLDLLS, 1);
  1174. break;
  1175. #endif
  1176. case SSL_FAILURE_INITSSL:
  1177. m_pOwner->ShowStatus(IDS_ERRORMSG_CANTINITSSL, 1);
  1178. break;
  1179. }
  1180. if (!m_bSentClose)
  1181. {
  1182. m_nMode |= CSMODE_TRANSFERERROR;
  1183. m_bSentClose = TRUE;
  1184. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  1185. }
  1186. break;
  1187. case SSL_VERIFY_CERT:
  1188. t_SslCertData data;
  1189. if (m_pSslLayer->GetPeerCertificateData(data))
  1190. m_pSslLayer->SetNotifyReply(data.priv_data, SSL_VERIFY_CERT, 1);
  1191. else
  1192. {
  1193. Close();
  1194. if (!m_bSentClose)
  1195. {
  1196. m_nMode |= CSMODE_TRANSFERERROR;
  1197. m_bSentClose = TRUE;
  1198. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  1199. }
  1200. }
  1201. break;
  1202. }
  1203. }
  1204. #endif
  1205. #ifndef MPEXT_NO_GSS
  1206. else if (iter->pLayer == m_pGssLayer)
  1207. {
  1208. USES_CONVERSION;
  1209. switch (iter->nParam1)
  1210. {
  1211. case GSS_INFO:
  1212. LogMessageRaw(FZ_LOG_INFO, A2CT(iter->str));
  1213. break;
  1214. case GSS_ERROR:
  1215. LogMessageRaw(FZ_LOG_APIERROR, A2CT(iter->str));
  1216. break;
  1217. case GSS_SHUTDOWN_COMPLETE:
  1218. Close();
  1219. if (!m_bSentClose)
  1220. {
  1221. m_bSentClose = TRUE;
  1222. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  1223. }
  1224. break;
  1225. }
  1226. }
  1227. #endif
  1228. }
  1229. delete [] iter->str;
  1230. }
  1231. return 0;
  1232. }
  1233. void CTransferSocket::Transfered(int count, CTime time)
  1234. {
  1235. CTimeSpan ts = time - m_TransferedFirst;
  1236. int diff = (int)ts.GetTotalSeconds();
  1237. if (diff < 0)
  1238. diff = 0;
  1239. if ( diff >= SPEED_SECONDS)
  1240. {
  1241. int move = diff - SPEED_SECONDS + 1;
  1242. int start = SPEED_SECONDS - move;
  1243. if ( start <= 0)
  1244. start = 0;
  1245. else
  1246. {
  1247. for ( int i = 0; i < SPEED_SECONDS - move; i++)
  1248. {
  1249. m_Transfered[ i] = m_Transfered[ i + move];
  1250. m_UsedForTransfer[ i] = m_UsedForTransfer[ i + move];
  1251. }
  1252. }
  1253. for ( int i = start; i < SPEED_SECONDS; i++)
  1254. {
  1255. m_Transfered[ i] = 0;
  1256. m_UsedForTransfer[ i] = false;
  1257. }
  1258. if (move >= SPEED_SECONDS)
  1259. {
  1260. m_TransferedFirst = time;
  1261. diff = 0;
  1262. }
  1263. else
  1264. {
  1265. m_TransferedFirst += CTimeSpan( move);
  1266. ts = time - m_TransferedFirst;
  1267. diff = (int)(ts.GetTotalSeconds() % 60);
  1268. }
  1269. }
  1270. m_Transfered[ diff] += count;
  1271. for ( int i = 0; i < diff - 1; i++)
  1272. m_UsedForTransfer[ i] = true;
  1273. }
  1274. #ifndef MPEXT_NO_GSS
  1275. void CTransferSocket::UseGSS(CAsyncGssSocketLayer *pGssLayer)
  1276. {
  1277. m_pGssLayer = new CAsyncGssSocketLayer;
  1278. m_pGssLayer->InitTransferChannel(pGssLayer);
  1279. }
  1280. #endif
  1281. #ifndef MPEXT_NO_ZLIB
  1282. bool CTransferSocket::InitZlib(int level)
  1283. {
  1284. int res;
  1285. if (m_nMode & CSMODE_UPLOAD)
  1286. res = deflateInit2(&m_zlibStream, level, Z_DEFLATED, 15, 8, Z_DEFAULT_STRATEGY);
  1287. else
  1288. res = inflateInit2(&m_zlibStream, 15);
  1289. if (res == Z_OK)
  1290. m_useZlib = true;
  1291. return res == Z_OK;
  1292. }
  1293. #endif
  1294. int CTransferSocket::ReadDataFromFile(char *buffer, int len)
  1295. {
  1296. TRY
  1297. {
  1298. // Comparing to Filezilla 2, we do not do any translation locally,
  1299. // leaving it onto the server (what Filezilla 3 seems to do too)
  1300. const char Bom[3] = "\xEF\xBB\xBF";
  1301. int read = m_pFile->Read(buffer, len);
  1302. if (m_transferdata.bType && (read >= sizeof(Bom)) && (memcmp(buffer, Bom, sizeof(Bom)) == 0))
  1303. {
  1304. memcpy(buffer, buffer + sizeof(Bom), read - sizeof(Bom));
  1305. read -= sizeof(Bom);
  1306. int read2 = m_pFile->Read(buffer + read, sizeof(Bom));
  1307. if (read2 > 0)
  1308. {
  1309. read += read2;
  1310. }
  1311. }
  1312. return read;
  1313. }
  1314. CATCH_ALL(e)
  1315. {
  1316. TCHAR error[BUFSIZE];
  1317. if (e->GetErrorMessage(error, BUFSIZE))
  1318. m_pOwner->ShowStatus(error, 1);
  1319. if (ShutDown() || GetLastError() != WSAEWOULDBLOCK)
  1320. {
  1321. Close();
  1322. if (!m_bSentClose)
  1323. {
  1324. m_nMode |= CSMODE_TRANSFERERROR;
  1325. m_bSentClose = TRUE;
  1326. m_pOwner->m_pOwner->PostThreadMessage(m_nInternalMessageID, FZAPI_THREADMSG_TRANSFEREND, m_nMode);
  1327. }
  1328. }
  1329. return -1;
  1330. }
  1331. END_CATCH_ALL;
  1332. }
  1333. void CTransferSocket::LogSocketMessage(int nMessageType, LPCTSTR pMsgFormat)
  1334. {
  1335. LogMessage(nMessageType, pMsgFormat);
  1336. }