]> git.saurik.com Git - wxWidgets.git/blob - src/common/ftp.cpp
Removed wxProp source files
[wxWidgets.git] / src / common / ftp.cpp
1 /////////////////////////////////////////////////////////////////////////////
2 // Name: ftp.cpp
3 // Purpose: FTP protocol
4 // Author: Guilhem Lavaux
5 // Modified by:
6 // Created: 07/07/1997
7 // RCS-ID: $Id$
8 // Copyright: (c) 1997, 1998 Guilhem Lavaux
9 // Licence: wxWindows license
10 /////////////////////////////////////////////////////////////////////////////
11
12 #ifdef __GNUG__
13 #pragma implementation "ftp.h"
14 #endif
15
16 // For compilers that support precompilation, includes "wx.h".
17 #include "wx/wxprec.h"
18
19 #ifdef __BORLANDC__
20 #pragma hdrstop
21 #endif
22
23 #ifndef __MWERKS__
24 #include <memory.h>
25 #endif
26 #include <stdlib.h>
27 #include "wx/string.h"
28 #include "wx/utils.h"
29 // #include "wx/data.h"
30 #define WXSOCK_INTERNAL
31 #include "wx/sckaddr.h"
32 #undef WXSOCK_INTERNAL
33 #include "wx/socket.h"
34 #include "wx/url.h"
35 #include "wx/sckstrm.h"
36 #include "wx/protocol/protocol.h"
37 #include "wx/protocol/ftp.h"
38
39 #ifdef __BORLANDC__
40 #pragma hdrstop
41 #endif
42
43 #define FTP_BSIZE 1024
44
45 #if !USE_SHARED_LIBRARY
46 IMPLEMENT_DYNAMIC_CLASS(wxFTP, wxProtocol)
47 IMPLEMENT_PROTOCOL(wxFTP, "ftp", "ftp", TRUE)
48 #endif
49
50 ////////////////////////////////////////////////////////////////
51 ////// wxFTP constructor and destructor ////////////////////////
52 ////////////////////////////////////////////////////////////////
53
54 wxFTP::wxFTP()
55 : wxProtocol()
56 {
57 char tmp[256];
58
59 m_lastError = wxPROTO_NOERR;
60 m_streaming = FALSE;
61
62 m_user = "anonymous";
63 wxGetUserName(tmp, 256);
64 m_passwd.sprintf("%s@",tmp);
65 wxGetHostName(tmp, 256);
66 m_passwd += tmp;
67
68 SetNotify(0);
69 }
70
71 wxFTP::~wxFTP()
72 {
73 SendCommand("QUIT", '2');
74 }
75
76 ////////////////////////////////////////////////////////////////
77 ////// wxFTP connect and login methods /////////////////////////
78 ////////////////////////////////////////////////////////////////
79 bool wxFTP::Connect(wxSockAddress& addr)
80 {
81 if (!m_handler) {
82 m_lastError = wxPROTO_NOHNDLR;
83 return FALSE;
84 }
85
86 if (!wxProtocol::Connect(addr)) {
87 m_lastError = wxPROTO_NETERR;
88 return FALSE;
89 }
90
91 if (!m_user || !m_passwd) {
92 m_lastError = wxPROTO_CONNERR;
93 return FALSE;
94 }
95
96 wxString command;
97
98 if (!GetResult('2')) {
99 Close();
100 return FALSE;
101 }
102
103 command.sprintf("USER %s", (const char *)m_user);
104 if (!SendCommand(command, '3')) {
105 Close();
106 return FALSE;
107 }
108
109 command.sprintf("PASS %s", (const char *)m_passwd);
110 if (!SendCommand(command, '2')) {
111 Close();
112 return FALSE;
113 }
114
115 return TRUE;
116 }
117
118 bool wxFTP::Connect(const wxString& host)
119 {
120 wxIPV4address addr;
121 wxString my_host = host;
122
123 addr.Hostname(my_host);
124 addr.Service("ftp");
125
126 return Connect(addr);
127 }
128
129 bool wxFTP::Close()
130 {
131 if (m_streaming) {
132 m_lastError = wxPROTO_STREAMING;
133 return FALSE;
134 }
135 if (m_connected)
136 SendCommand(wxString("QUIT"), '2');
137 return wxSocketClient::Close();
138 }
139
140 ////////////////////////////////////////////////////////////////
141 ////// wxFTP low-level methods /////////////////////////////////
142 ////////////////////////////////////////////////////////////////
143 bool wxFTP::SendCommand(const wxString& command, char exp_ret)
144 {
145 wxString tmp_str;
146
147 if (m_streaming) {
148 m_lastError = wxPROTO_STREAMING;
149 return FALSE;
150 }
151 tmp_str = command + "\r\n";
152 if (Write((char *)tmp_str.GetData(), tmp_str.Length()).Error()) {
153 m_lastError = wxPROTO_NETERR;
154 return FALSE;
155 }
156 return GetResult(exp_ret);
157 }
158
159 bool wxFTP::GetResult(char exp)
160 {
161 if ((m_lastError = GetLine(this, m_lastResult)))
162 return FALSE;
163 if (m_lastResult.GetChar(0) != exp) {
164 m_lastError = wxPROTO_PROTERR;
165 return FALSE;
166 }
167
168 if (m_lastResult.GetChar(3) == '-') {
169 wxString key = m_lastResult.Left((size_t)3);
170
171 key += ' ';
172
173 while (m_lastResult.Index(key) != 0) {
174 if ((m_lastError = GetLine(this, m_lastResult)))
175 return FALSE;
176 }
177 }
178 return TRUE;
179 }
180
181 ////////////////////////////////////////////////////////////////
182 ////// wxFTP low-level methods /////////////////////////////////
183 ////////////////////////////////////////////////////////////////
184 bool wxFTP::ChDir(const wxString& dir)
185 {
186 wxString str = dir;
187
188 str.Prepend("CWD ");
189 return SendCommand(str, '2');
190 }
191
192 bool wxFTP::MkDir(const wxString& dir)
193 {
194 wxString str = dir;
195 str.Prepend("MKD ");
196 return SendCommand(str, '2');
197 }
198
199 bool wxFTP::RmDir(const wxString& dir)
200 {
201 wxString str = dir;
202
203 str.Prepend("PWD ");
204 return SendCommand(str, '2');
205 }
206
207 wxString wxFTP::Pwd()
208 {
209 int beg, end;
210
211 if (!SendCommand("PWD", '2'))
212 return wxString((char *)NULL);
213
214 beg = m_lastResult.Find('\"',FALSE);
215 end = m_lastResult.Find('\"',TRUE);
216
217 return wxString(beg+1, end);
218 }
219
220 bool wxFTP::Rename(const wxString& src, const wxString& dst)
221 {
222 wxString str;
223
224 str = "RNFR " + src;
225 if (!SendCommand(str, '3'))
226 return FALSE;
227
228 str = "RNTO " + dst;
229 return SendCommand(str, '2');
230 }
231
232 bool wxFTP::RmFile(const wxString& path)
233 {
234 wxString str;
235
236 str = "DELE ";
237 str += path;
238 return SendCommand(str, '2');
239 }
240
241 ////////////////////////////////////////////////////////////////
242 ////// wxFTP download*upload ///////////////////////////////////
243 ////////////////////////////////////////////////////////////////
244
245 class wxInputFTPStream : public wxSocketInputStream {
246 public:
247 wxFTP *m_ftp;
248
249 wxInputFTPStream(wxFTP *ftp_clt, wxSocketBase *sock)
250 : wxSocketInputStream(*sock), m_ftp(ftp_clt) {}
251 virtual ~wxInputFTPStream(void)
252 {
253 if (LastError() != wxStream_NOERROR)
254 m_ftp->GetResult('2');
255 else
256 m_ftp->Abort();
257 delete m_i_socket;
258 }
259 };
260
261 class wxOutputFTPStream : public wxSocketOutputStream {
262 public:
263 wxFTP *m_ftp;
264
265 wxOutputFTPStream(wxFTP *ftp_clt, wxSocketBase *sock)
266 : wxSocketOutputStream(*sock), m_ftp(ftp_clt) {}
267 virtual ~wxOutputFTPStream(void)
268 {
269 if (LastError() != wxStream_NOERROR)
270 m_ftp->GetResult('2');
271 else
272 m_ftp->Abort();
273 delete m_o_socket;
274 }
275 };
276
277 wxSocketClient *wxFTP::GetPort()
278 {
279 wxIPV4address addr;
280 wxSocketClient *client;
281 struct sockaddr sin;
282 int a[6];
283 wxString straddr;
284 int addr_pos;
285
286 if (!SendCommand("PASV", '2'))
287 return NULL;
288
289 sin.sa_family = AF_INET;
290 addr_pos = m_lastResult.Find('(');
291 if (addr_pos == -1) {
292 m_lastError = wxPROTO_PROTERR;
293 return NULL;
294 }
295 straddr = m_lastResult(addr_pos+1, m_lastResult.Length());
296 sscanf((const char *)straddr,"%d,%d,%d,%d,%d,%d",&a[2],&a[3],&a[4],&a[5],&a[0],&a[1]);
297 sin.sa_data[2] = (char)a[2];
298 sin.sa_data[3] = (char)a[3];
299 sin.sa_data[4] = (char)a[4];
300 sin.sa_data[5] = (char)a[5];
301 sin.sa_data[0] = (char)a[0];
302 sin.sa_data[1] = (char)a[1];
303
304 addr.Disassemble(&sin, sizeof(sin));
305
306 client = m_handler->CreateClient();
307 if (!client->Connect(addr)) {
308 delete client;
309 return NULL;
310 }
311 client->Notify(FALSE);
312
313 return client;
314 }
315
316 bool wxFTP::Abort(void)
317 {
318 m_streaming = FALSE;
319 if (!SendCommand("ABOR", '4'))
320 return FALSE;
321 return GetResult('2');
322 }
323
324 wxInputStream *wxFTP::GetInputStream(const wxString& path)
325 {
326 wxString tmp_str;
327
328 if (!SendCommand("TYPE I", '2'))
329 return NULL;
330
331 wxSocketClient *sock = GetPort();
332
333 if (!sock) {
334 m_lastError = wxPROTO_NETERR;
335 return NULL;
336 }
337
338 tmp_str = "RETR " + path;
339 if (!SendCommand(tmp_str, '1'))
340 return NULL;
341
342 return new wxInputFTPStream(this, sock);
343 }
344
345 wxOutputStream *wxFTP::GetOutputStream(const wxString& path)
346 {
347 wxString tmp_str;
348
349 if (!SendCommand("TYPE I", '2'))
350 return NULL;
351
352 wxSocketClient *sock = GetPort();
353
354 tmp_str = "STOR " + path;
355 if (!SendCommand(tmp_str, '1'))
356 return FALSE;
357
358 return new wxOutputFTPStream(this, sock);
359 }
360
361 wxList *wxFTP::GetList(const wxString& wildcard)
362 {
363 wxList *file_list = new wxList;
364 wxSocketBase *sock = GetPort();
365 wxString tmp_str = "NLST";
366
367 if (!wildcard.IsNull())
368 tmp_str += wildcard;
369
370 if (!SendCommand(tmp_str, '1')) {
371 delete sock;
372 delete file_list;
373 return NULL;
374 }
375
376 while (GetLine(sock, tmp_str) == wxPROTO_NOERR) {
377 file_list->Append((wxObject *)(new wxString(tmp_str)));
378 }
379
380 if (!GetResult('2')) {
381 delete sock;
382 file_list->DeleteContents(TRUE);
383 delete file_list;
384 return NULL;
385 }
386
387 return file_list;
388 }