FTPclient.cpp
上传用户:tjfeida
上传日期:2013-03-10
资源大小:1917k
文件大小:12k
源码类别:

Ftp客户端

开发平台:

Visual C++

  1. /*/////////////////////////////////////////////////////////////////////
  2. FTPclient.cpp (c) GDI 1999
  3. V1.0.0 (10/4/99)
  4. Phil Anderson. philip@gd-ind.com
  5. Simple FTP client functionality. If you have any problems with it,
  6. please tell me about them (or better still e-mail me the fixed
  7. code). Please feel free to use this code however you wish, although
  8. if you make changes please put your name in the source & comment what
  9. you did.
  10. Nothing awesome going on here at all (all sockets are used in
  11. synchronous blocking mode), but it does the following
  12. things WinInet doesn't seem to:
  13. * Supports loads of different firewalls (I think, I don't
  14.   have access to all types so they haven't all been fully
  15.   tested yet)
  16. * Allows you to execute any command on the FTP server
  17. * Adds 10K to your app install rather than 1Mb #;-)
  18. Functions return TRUE if everything went OK, FALSE if there was an,
  19. error. A message describing the outcome (normally the one returned
  20. from the server) will be in m_retmsg on return from the function.
  21. There are a few error msgs in the app's string table that you'll
  22. need to paste into your app, along with this file & FTPclient.h
  23. If you created your app without checking the "Use Windows Sockets"
  24. checkbox in AppWizard, you'll need to add the following bit of code
  25. to you app's InitInstance()
  26. if(!AfxSocketInit())
  27. {
  28. AfxMessageBox("Could not initialize Windows Sockets!");
  29. return FALSE;
  30. }
  31. To use:
  32. 1/ Create an object of CFTPclient.
  33. 2/ Use LogOnToServer() to connect to the server. Any arguments
  34. not used (e.g. if you're not using a firewall), pass an empty
  35. string or zero for numeric args. You must pass a server port
  36. number, use the FTP default of 21 if you don't know what it is.
  37. 3/ Use MoveFile() to upload/download a file, 1st arg is local file
  38. path, 2nd arg is remote file path, 3rd arg is TRUE for a PASV
  39. connection (required by some firewalls), FALSE otherwise, 4th arg
  40. is TRUE to upload, FALSE to download file. MoveFile only works in
  41. synchronous mode (ie the function will not return 'till the transfer
  42. is finished). File transfers are always of type BINARY.
  43. 4/ You can use FTPcommand() to execute FTP commands (eg
  44. FTPcommand("CWD /home/mydir") to change directory on the server),
  45. note that this function will return FALSE unless the server response
  46. is a 200 series code. This should work fine for most FTP commands, 
  47. otherwise you can use WriteStr() and ReadStr() to send commands & 
  48. interpret the response yourself. Use LogOffServer() to disconnect
  49. when done.
  50. /////////////////////////////////////////////////////////////////////*/
  51. #include "stdafx.h"
  52. #include "GFTPTest.h"
  53. #include "FTPclient.h"
  54. #ifdef _DEBUG
  55. #undef THIS_FILE
  56. static char THIS_FILE[]=__FILE__;
  57. #define new DEBUG_NEW
  58. #endif
  59. //////////////////////////////////////////////////////////////////////
  60. // Construction/Destruction
  61. //////////////////////////////////////////////////////////////////////
  62. CFTPclient::CFTPclient()
  63. {
  64. m_pCtrlsokfile=NULL;
  65. m_pCtrlTxarch=NULL;
  66. m_pCtrlRxarch=NULL;
  67. m_Ctrlsok=NULL;
  68. }
  69. CFTPclient::~CFTPclient()
  70. {
  71. CloseControlChannel();
  72. }
  73. //////////////////////////////////////////////////////////////////////
  74. // Public Functions
  75. //////////////////////////////////////////////////////////////////////
  76. // function to connect & log on to FTP server
  77. BOOL CFTPclient::LogOnToServer(CString hostname,int hostport,CString username, CString password, CString acct, CString fwhost,CString fwusername, CString fwpassword,int fwport,int logontype) {
  78. int port,logonpoint=0;
  79. const int LO=-2, ER=-1;
  80. CString buf,temp;
  81. //登陆方式数目
  82. const int NUMLOGIN=9;
  83. //登陆方式数组
  84. int logonseq[NUMLOGIN][100] = 
  85. {
  86. {0,LO,3, 1,LO,6, 2,LO,ER}, // no firewall
  87. {3,6,3, 4,6,ER, 5,ER,9, 0,LO,12, 1,LO,15, 2,LO,ER}, // SITE hostname
  88. {3,6,3, 4,6,ER, 6,LO,9, 1,LO,12, 2,LO,ER}, // USER after logon
  89. {7,3,3, 0,LO,6, 1,LO,9, 2,LO,ER}, //proxy OPEN
  90. {3,6,3, 4,6,ER, 0,LO,9, 1,LO,12, 2,LO,ER}, // Transparent
  91. {6,LO,3, 1,LO,6, 2,LO,ER}, // USER with no logon
  92. {8,6,3, 4,6,ER, 0,LO,9, 1,LO,12, 2,LO,ER}, //USER fireID@remotehost
  93. {9,ER,3, 1,LO,6, 2,LO,ER}, //USER remoteID@remotehost fireID
  94. {10,LO,3, 11,LO,6, 2,LO,ER} // USER remoteID@fireID@remotehost
  95. };
  96. //连接方式无效
  97. if(logontype<0||logontype>=NUMLOGIN) 
  98. return FALSE; 
  99. //确定直接连接还是通过防火墙连接
  100. if(!logontype) 
  101. {
  102. temp=hostname;
  103. port=hostport;
  104. }
  105. else 
  106. {
  107. temp=fwhost;
  108. port=fwport;
  109. }
  110. if(hostport!=21) 
  111. hostname.Format(hostname+":%d",hostport);
  112. //打开控制连接
  113. if(!OpenControlChannel(temp,port)) 
  114. return false;
  115. //获取初始的服务器信息
  116. if(!FTPcommand("")) 
  117. return FALSE; 
  118. // 测试连接,便于选择适合的登陆方式
  119. while(1) 
  120. {
  121. switch(logonseq[logontype][logonpoint]) 
  122. {
  123. case 0:
  124. temp="USER "+username;
  125. break;
  126. case 1:
  127. temp="PASS "+password;
  128. break;
  129. case 2:
  130. temp="ACCT "+acct;
  131. break;
  132. case 3:
  133. temp="USER "+fwusername;
  134. break;
  135. case 4:
  136. temp="PASS "+fwpassword;
  137. break;
  138. case 5:
  139. temp="SITE "+hostname;
  140. break;
  141. case 6:
  142. temp="USER "+username+"@"+hostname;
  143. break;
  144. case 7:
  145. temp="OPEN "+hostname;
  146. break;
  147. case 8:
  148. temp="USER "+fwusername+"@"+hostname;
  149. break;
  150. case 9:
  151. temp="USER "+username+"@"+hostname+" "+fwusername;
  152. break;
  153. case 10:
  154. temp="USER "+username+"@"+fwusername+"@"+hostname;
  155. break;
  156. case 11:
  157. temp="PASS "+password+"@"+fwpassword;
  158. break;
  159. }
  160. //发送命令
  161. if(!WriteStr(temp)) 
  162. return FALSE;
  163. if(!ReadStr()) 
  164. return FALSE;
  165. if(m_fc!=2&&m_fc!=3) 
  166. return FALSE;
  167. //获取下一条命令
  168. logonpoint=logonseq[logontype][logonpoint+m_fc-1]; 
  169. switch(logonpoint)
  170. {
  171. //登陆错误
  172. case ER: 
  173. m_retmsg.LoadString(IDS_FTPMSG1);
  174. return FALSE;
  175. //完全登陆
  176. case LO: 
  177. return TRUE;
  178. }
  179. }
  180. }
  181. // function to log off & close connection to FTP server
  182. void CFTPclient::LogOffServer() {
  183. WriteStr("QUIT");
  184. CloseControlChannel();
  185. }
  186. // function to execute commands on the FTP server
  187. BOOL CFTPclient::FTPcommand(CString command) {
  188. if(command!=""&&!WriteStr(command)) return FALSE;
  189. if((!ReadStr())||(m_fc!=2)) return FALSE;
  190. return TRUE;
  191. }
  192. // function to upload/download files
  193. BOOL CFTPclient::MoveFile(CString remotefile, CString localfile,BOOL pasv,BOOL get) 
  194. {
  195. CString lhost,temp,rhost;
  196. UINT localsock,serversock,i,j;
  197. CFile datafile;
  198. //侦听服务器的连接
  199. CSocket sockSrvr;
  200. //数据连接通道
  201. CAsyncSocket datachannel;
  202. //接收、发送字节
  203. int num,numread,numsent;
  204. //缓冲区大小
  205. const int BUFSIZE=4096;
  206. //缓冲区
  207. char cbuf[BUFSIZE];
  208. DWORD lpArgument=0;
  209. // 打开本地文件
  210. if(!datafile.Open(localfile,(get?CFile::modeWrite|CFile::modeCreate:CFile::modeRead))) 
  211. {
  212. m_retmsg.LoadString(IDS_FTPMSG4);
  213. return FALSE;
  214. }
  215. //二进制传输模式
  216. if(!FTPcommand("TYPE I")) 
  217. return FALSE; 
  218. //被动接收
  219. if(pasv) 
  220. {
  221. if(!FTPcommand("PASV")) 
  222. return FALSE;
  223. //获取服务器IP
  224. if((i=m_retmsg.Find("("))==-1||(j=m_retmsg.Find(")"))==-1) 
  225. return FALSE;
  226. temp=m_retmsg.Mid(i+1,(j-i)-1);
  227. i=temp.ReverseFind(',');
  228. serversock=atol(temp.Right(temp.GetLength()-(i+1))); 
  229. temp=temp.Left(i);
  230. i=temp.ReverseFind(',');
  231. serversock+=256*atol(temp.Right(temp.GetLength()-(i+1))); // add ms byte to server socket
  232. rhost=temp.Left(i);
  233. while(1)
  234. //转换,
  235. if((i=rhost.Find(","))==-1) 
  236. break;
  237. rhost.SetAt(i,'.');
  238. }
  239. }
  240. //主动传输模式
  241. else 
  242. m_retmsg.LoadString(IDS_FTPMSG6);
  243. //获取本地控制连接IP
  244. if(!m_Ctrlsok->GetSockName(lhost,localsock)) 
  245. return FALSE;
  246. while(1) 
  247. //转换.
  248. if((i=lhost.Find("."))==-1) break;
  249. lhost.SetAt(i,',');
  250. }
  251. //创建侦听Socket
  252. if((!sockSrvr.Create(0,SOCK_STREAM,NULL))||(!sockSrvr.Listen())) 
  253. return FALSE;
  254. if(!sockSrvr.GetSockName(temp,localsock)) 
  255. return FALSE;
  256. lhost.Format(lhost+",%d,%d",localsock/256,localsock%256);
  257. //发送Port命令
  258. if(!FTPcommand("PORT "+lhost)) return FALSE;
  259. }
  260. if(!WriteStr((get?"RETR ":"STOR ")+remotefile)) return FALSE;
  261. if(pasv) 
  262. {
  263. if(!datachannel.Create()) {
  264. m_retmsg.LoadString(IDS_FTPMSG6);
  265. return FALSE;
  266. }
  267. //数据通道连接
  268. datachannel.Connect(rhost,serversock); 
  269. }
  270. //获取响应
  271. if(!ReadStr()||m_fc!=1) 
  272. return FALSE; 
  273. //通道建立
  274. if(!pasv&&!sockSrvr.Accept(datachannel)) 
  275. return FALSE; 
  276. if((!datachannel.AsyncSelect(0))||(!datachannel.IOCtl(FIONBIO,&lpArgument))) 
  277. {
  278. m_retmsg.LoadString(IDS_FTPMSG6);
  279. return FALSE;
  280. }
  281. //开始数据传输
  282. while(1) 
  283. TRY 
  284. {
  285. //获取数据
  286. if(get) 
  287. {
  288. //传输完毕
  289. if(!(num=datachannel.Receive(cbuf,BUFSIZE,0))||num==SOCKET_ERROR) 
  290. break;
  291. else 
  292. datafile.Write(cbuf,num);
  293. }
  294. //上传数据
  295. else 
  296. {
  297. //传输完毕
  298. if(!(numread=datafile.Read(cbuf,BUFSIZE))) 
  299. break; 
  300. if((numsent=datachannel.Send(cbuf,numread,0))==SOCKET_ERROR) 
  301. break;
  302. //发送的数据少于实际文件,重新定位文件指针
  303. if(numread!=numsent) 
  304. datafile.Seek(numsent-numread,CFile::current);
  305. }
  306. }
  307. CATCH (CException,e) {
  308. m_retmsg.LoadString(IDS_FTPMSG5);
  309. return FALSE;
  310. }
  311. END_CATCH
  312. }
  313. //数据通道关闭
  314. datachannel.Close();
  315. datafile.Close();
  316. //获取响应
  317. if(!FTPcommand("")) 
  318. return FALSE; 
  319. return TRUE; 
  320. }
  321. // function to send a command string on the server control channel
  322. BOOL CFTPclient::WriteStr(CString outputstring) 
  323. {
  324. //载入错误信息,便于后面显示
  325. m_retmsg.LoadString(IDS_FTPMSG6);
  326. //发送信息
  327. TRY {
  328. m_pCtrlTxarch->WriteString(outputstring+"rn");
  329. m_pCtrlTxarch->Flush();
  330. }
  331. //捕获异常
  332. CATCH(CException,e) {
  333. return FALSE;
  334. }
  335. END_CATCH
  336. return TRUE;
  337. }
  338. // this function gets the server response line
  339. BOOL CFTPclient::ReadStr() 
  340. {
  341. //响应码
  342. int retcode;
  343. //读取信息
  344. if(!ReadStr2()) 
  345. return FALSE;
  346. //根据获取的信息确定是否成功
  347. if(m_retmsg.GetLength()<4||m_retmsg.GetAt(3)!='-')
  348. return TRUE;
  349. //获取响应码
  350. retcode=atol(m_retmsg);
  351. while(1) 
  352. //处理多行响应
  353. if(m_retmsg.GetLength()>3&&(m_retmsg.GetAt(3)==' '&&atol(m_retmsg)==retcode)) 
  354. return TRUE;
  355. if(!ReadStr2()) 
  356. return FALSE;
  357. }
  358. }
  359. //////////////////////////////////////////////////////////////////////
  360. // Private functions
  361. //////////////////////////////////////////////////////////////////////
  362. // read a single response line from the server control channel
  363. BOOL CFTPclient::ReadStr2() 
  364. {
  365. TRY 
  366. {
  367. //利用序列化读取信息
  368. if(!m_pCtrlRxarch->ReadString(m_retmsg)) 
  369. {
  370. //载入失败信息
  371. m_retmsg.LoadString(IDS_FTPMSG6);
  372. return FALSE;
  373. }
  374. }
  375. //捕获异常
  376. CATCH(CException,e) 
  377. {
  378. m_retmsg.LoadString(IDS_FTPMSG6);
  379. return FALSE;
  380. }
  381. END_CATCH
  382. //获取响应信息的第一个字节
  383. if(m_retmsg.GetLength()>0) 
  384. m_fc=m_retmsg.GetAt(0)-48; 
  385. return TRUE;
  386. }
  387. // open the control channel to the FTP server
  388. BOOL CFTPclient::OpenControlChannel(CString serverhost,int serverport) {
  389. m_retmsg.LoadString(IDS_FTPMSG2);
  390. if(!(m_Ctrlsok=new CSocket)) 
  391. return FALSE;
  392. if(!(m_Ctrlsok->Create())) 
  393. return FALSE;
  394. m_retmsg.LoadString(IDS_FTPMSG3);
  395. //控制通道Socket连接服务器
  396. if(!(m_Ctrlsok->Connect(serverhost,serverport))) 
  397. return FALSE;
  398. m_retmsg.LoadString(IDS_FTPMSG2);
  399. //连接成功后创建序列化对象
  400. if(!(m_pCtrlsokfile=new CSocketFile(m_Ctrlsok))) 
  401. return FALSE;
  402. if(!(m_pCtrlRxarch=new CArchive(m_pCtrlsokfile,CArchive::load))) 
  403. return FALSE;
  404. if(!(m_pCtrlTxarch=new CArchive(m_pCtrlsokfile,CArchive::store))) 
  405. return FALSE;
  406. return TRUE;
  407. }
  408. // close the control channel to the FTP server
  409. void CFTPclient::CloseControlChannel() {
  410. if(m_pCtrlTxarch) delete m_pCtrlTxarch;
  411. m_pCtrlTxarch=NULL;
  412. if(m_pCtrlRxarch) delete m_pCtrlRxarch;
  413. m_pCtrlRxarch=NULL;
  414. if(m_pCtrlsokfile) delete m_pCtrlsokfile;
  415. m_pCtrlsokfile=NULL;
  416. if(m_Ctrlsok) delete m_Ctrlsok;
  417. m_Ctrlsok=NULL;
  418. return;
  419. }