TransferSocket.cpp 37 KB

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