blob: eaaa6fd69c0938949a33050e47a88b3782808545 [file] [log] [blame]
Guido van Rossum4b8c6ea2000-02-04 15:39:30 +00001"""An FTP client class and some helper functions.
2
Barry Warsaw100d81e2000-09-01 06:09:23 +00003Based on RFC 959: File Transfer Protocol (FTP), by J. Postel and J. Reynolds
Guido van Rossum1115ab21992-11-04 15:51:30 +00004
Guido van Rossumd2560b01996-05-28 23:41:25 +00005Example:
6
7>>> from ftplib import FTP
8>>> ftp = FTP('ftp.python.org') # connect to host, default port
Guido van Rossum24a64342001-12-28 20:54:55 +00009>>> ftp.login() # default, i.e.: user anonymous, passwd anonymous@
Guido van Rossum2f3941d1997-10-07 14:49:56 +000010'230 Guest login ok, access restrictions apply.'
Guido van Rossumd2560b01996-05-28 23:41:25 +000011>>> ftp.retrlines('LIST') # list directory contents
12total 9
13drwxr-xr-x 8 root wheel 1024 Jan 3 1994 .
14drwxr-xr-x 8 root wheel 1024 Jan 3 1994 ..
15drwxr-xr-x 2 root wheel 1024 Jan 3 1994 bin
16drwxr-xr-x 2 root wheel 1024 Jan 3 1994 etc
17d-wxrwxr-x 2 ftp wheel 1024 Sep 5 13:43 incoming
18drwxr-xr-x 2 root wheel 1024 Nov 17 1993 lib
19drwxr-xr-x 6 1094 wheel 1024 Sep 13 19:07 pub
20drwxr-xr-x 3 root wheel 1024 Jan 3 1994 usr
21-rw-r--r-- 1 root root 312 Aug 1 1994 welcome.msg
Guido van Rossum2f3941d1997-10-07 14:49:56 +000022'226 Transfer complete.'
Guido van Rossumd2560b01996-05-28 23:41:25 +000023>>> ftp.quit()
Guido van Rossum2f3941d1997-10-07 14:49:56 +000024'221 Goodbye.'
Tim Peters88869f92001-01-14 23:36:06 +000025>>>
Guido van Rossumd2560b01996-05-28 23:41:25 +000026
27A nice test that reveals some of the network dialogue would be:
28python ftplib.py -d localhost -l -p -l
Guido van Rossum4b8c6ea2000-02-04 15:39:30 +000029"""
Guido van Rossumc567c601992-11-05 22:22:37 +000030
Tim Peters88869f92001-01-14 23:36:06 +000031#
Guido van Rossum98d9fd32000-02-28 15:12:25 +000032# Changes and improvements suggested by Steve Majewski.
33# Modified by Jack to work on the mac.
34# Modified by Siebren to support docstrings and PASV.
Christian Heimes1af737c2008-01-23 08:24:23 +000035# Modified by Phil Schwartz to add storbinary and storlines callbacks.
Antoine Pitrouf988cd02009-11-17 20:21:14 +000036# Modified by Giampaolo Rodola' to add TLS support.
Guido van Rossum98d9fd32000-02-28 15:12:25 +000037#
Guido van Rossumc567c601992-11-05 22:22:37 +000038
Guido van Rossum1115ab21992-11-04 15:51:30 +000039import os
40import sys
Giampaolo RodolĂ 6545ed62010-08-21 20:34:37 +000041import socket
Georg Brandlf78e02b2008-06-10 17:40:04 +000042from socket import _GLOBAL_DEFAULT_TIMEOUT
Guido van Rossumb6775db1994-08-01 11:34:53 +000043
Skip Montanaroeccd02a2001-01-20 23:34:12 +000044__all__ = ["FTP","Netrc"]
Guido van Rossum1115ab21992-11-04 15:51:30 +000045
Guido van Rossumd3166071993-05-24 14:16:22 +000046# Magic number from <socket.h>
Tim Peters88869f92001-01-14 23:36:06 +000047MSG_OOB = 0x1 # Process data out of band
Guido van Rossumd3166071993-05-24 14:16:22 +000048
49
Guido van Rossumc567c601992-11-05 22:22:37 +000050# The standard FTP server control port
Guido van Rossum1115ab21992-11-04 15:51:30 +000051FTP_PORT = 21
Guido van Rossum1115ab21992-11-04 15:51:30 +000052
53
Guido van Rossum21974791992-11-06 13:34:17 +000054# Exception raised when an error or invalid response is received
Fred Drake227b1202000-08-17 05:06:49 +000055class Error(Exception): pass
Tim Peters88869f92001-01-14 23:36:06 +000056class error_reply(Error): pass # unexpected [123]xx reply
57class error_temp(Error): pass # 4xx errors
58class error_perm(Error): pass # 5xx errors
59class error_proto(Error): pass # response does not begin with [1-5]
Guido van Rossum1115ab21992-11-04 15:51:30 +000060
61
Guido van Rossum21974791992-11-06 13:34:17 +000062# All exceptions (hopefully) that may be raised here and that aren't
63# (always) programming errors on our side
Christian Heimes33fe8092008-04-13 13:53:33 +000064all_errors = (Error, IOError, EOFError)
Guido van Rossum21974791992-11-06 13:34:17 +000065
66
Guido van Rossum1115ab21992-11-04 15:51:30 +000067# Line terminators (we always output CRLF, but accept any of CRLF, CR, LF)
68CRLF = '\r\n'
Benjamin Petersonbe17a112008-09-27 21:49:47 +000069B_CRLF = b'\r\n'
Guido van Rossum1115ab21992-11-04 15:51:30 +000070
Guido van Rossum1115ab21992-11-04 15:51:30 +000071# The class itself
72class FTP:
73
Tim Peters88869f92001-01-14 23:36:06 +000074 '''An FTP client class.
Guido van Rossumd2560b01996-05-28 23:41:25 +000075
Guido van Rossumd8faa362007-04-27 19:54:29 +000076 To create a connection, call the class using these arguments:
77 host, user, passwd, acct, timeout
78
79 The first four arguments are all strings, and have default value ''.
80 timeout must be numeric and defaults to None if not passed,
81 meaning that no timeout will be set on any ftp socket(s)
82 If a timeout is passed, then this is now the default timeout for all ftp
83 socket operations for this instance.
84
Tim Peters88869f92001-01-14 23:36:06 +000085 Then use self.connect() with optional host and port argument.
Guido van Rossumd2560b01996-05-28 23:41:25 +000086
Tim Peters88869f92001-01-14 23:36:06 +000087 To download a file, use ftp.retrlines('RETR ' + filename),
88 or ftp.retrbinary() with slightly different arguments.
89 To upload a file, use ftp.storlines() or ftp.storbinary(),
90 which have an open file as argument (see their definitions
91 below for details).
92 The download/upload functions first issue appropriate TYPE
93 and PORT or PASV commands.
Guido van Rossum52b89762007-07-17 20:45:57 +000094 '''
Guido van Rossumd2560b01996-05-28 23:41:25 +000095
Fred Drake9c98a422001-02-28 21:46:37 +000096 debugging = 0
97 host = ''
98 port = FTP_PORT
99 sock = None
100 file = None
101 welcome = None
102 passiveserver = 1
Marc-André Lemburg8f36af72011-02-25 15:42:01 +0000103 encoding = "latin-1"
Fred Drake9c98a422001-02-28 21:46:37 +0000104
Tim Peters88869f92001-01-14 23:36:06 +0000105 # Initialization method (called by class instantiation).
106 # Initialize host to localhost, port to standard ftp port
107 # Optional arguments are host (for connect()),
108 # and user, passwd, acct (for login())
Georg Brandlf78e02b2008-06-10 17:40:04 +0000109 def __init__(self, host='', user='', passwd='', acct='',
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000110 timeout=_GLOBAL_DEFAULT_TIMEOUT, source_address=None):
111 self.source_address = source_address
Guido van Rossumd8faa362007-04-27 19:54:29 +0000112 self.timeout = timeout
Tim Peters88869f92001-01-14 23:36:06 +0000113 if host:
Fred Drake9c98a422001-02-28 21:46:37 +0000114 self.connect(host)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000115 if user:
116 self.login(user, passwd, acct)
Guido van Rossum52fc1f61993-06-17 12:38:10 +0000117
Giampaolo RodolĂ bd576b72010-05-10 14:53:29 +0000118 def __enter__(self):
119 return self
120
121 # Context management protocol: try to quit() if active
122 def __exit__(self, *args):
123 if self.sock is not None:
124 try:
125 self.quit()
126 except (socket.error, EOFError):
127 pass
128 finally:
129 if self.sock is not None:
130 self.close()
131
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000132 def connect(self, host='', port=0, timeout=-999, source_address=None):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000133 '''Connect to host. Arguments are:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000134 - host: hostname to connect to (string, default previous host)
135 - port: port to connect to (integer, default previous port)
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000136 - source_address: a 2-tuple (host, port) for the socket to bind
137 to as its source address before connecting.
Guido van Rossumd8faa362007-04-27 19:54:29 +0000138 '''
139 if host != '':
140 self.host = host
141 if port > 0:
142 self.port = port
Georg Brandlf78e02b2008-06-10 17:40:04 +0000143 if timeout != -999:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000144 self.timeout = timeout
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000145 if source_address is not None:
146 self.source_address = source_address
147 self.sock = socket.create_connection((self.host, self.port), self.timeout,
148 source_address=self.source_address)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000149 self.af = self.sock.family
Guido van Rossum52b89762007-07-17 20:45:57 +0000150 self.file = self.sock.makefile('r', encoding=self.encoding)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000151 self.welcome = self.getresp()
152 return self.welcome
Guido van Rossum1115ab21992-11-04 15:51:30 +0000153
Tim Peters88869f92001-01-14 23:36:06 +0000154 def getwelcome(self):
155 '''Get the welcome message from the server.
156 (this is read and squirreled away by connect())'''
157 if self.debugging:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000158 print('*welcome*', self.sanitize(self.welcome))
Tim Peters88869f92001-01-14 23:36:06 +0000159 return self.welcome
Guido van Rossum1115ab21992-11-04 15:51:30 +0000160
Tim Peters88869f92001-01-14 23:36:06 +0000161 def set_debuglevel(self, level):
162 '''Set the debugging level.
163 The required argument level means:
164 0: no debugging output (default)
165 1: print commands and responses but not body text etc.
166 2: also print raw lines read and sent before stripping CR/LF'''
167 self.debugging = level
168 debug = set_debuglevel
Guido van Rossum1115ab21992-11-04 15:51:30 +0000169
Tim Peters88869f92001-01-14 23:36:06 +0000170 def set_pasv(self, val):
171 '''Use passive or active mode for data transfers.
172 With a false argument, use the normal PORT mode,
173 With a true argument, use the PASV command.'''
174 self.passiveserver = val
Guido van Rossumd2560b01996-05-28 23:41:25 +0000175
Tim Peters88869f92001-01-14 23:36:06 +0000176 # Internal: "sanitize" a string for printing
177 def sanitize(self, s):
178 if s[:5] == 'pass ' or s[:5] == 'PASS ':
179 i = len(s)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000180 while i > 5 and s[i-1] in {'\r', '\n'}:
Tim Peters88869f92001-01-14 23:36:06 +0000181 i = i-1
182 s = s[:5] + '*'*(i-5) + s[i:]
Walter Dörwald70a6b492004-02-12 17:35:32 +0000183 return repr(s)
Guido van Rossumebaf1041995-05-05 15:54:14 +0000184
Tim Peters88869f92001-01-14 23:36:06 +0000185 # Internal: send one line to the server, appending CRLF
186 def putline(self, line):
187 line = line + CRLF
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000188 if self.debugging > 1: print('*put*', self.sanitize(line))
Guido van Rossum52b89762007-07-17 20:45:57 +0000189 self.sock.sendall(line.encode(self.encoding))
Guido van Rossum1115ab21992-11-04 15:51:30 +0000190
Tim Peters88869f92001-01-14 23:36:06 +0000191 # Internal: send one command to the server (through putline())
192 def putcmd(self, line):
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000193 if self.debugging: print('*cmd*', self.sanitize(line))
Tim Peters88869f92001-01-14 23:36:06 +0000194 self.putline(line)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000195
Tim Peters88869f92001-01-14 23:36:06 +0000196 # Internal: return one line from the server, stripping CRLF.
197 # Raise EOFError if the connection is closed
198 def getline(self):
199 line = self.file.readline()
200 if self.debugging > 1:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000201 print('*get*', self.sanitize(line))
Tim Peters88869f92001-01-14 23:36:06 +0000202 if not line: raise EOFError
203 if line[-2:] == CRLF: line = line[:-2]
204 elif line[-1:] in CRLF: line = line[:-1]
205 return line
Guido van Rossum1115ab21992-11-04 15:51:30 +0000206
Tim Peters88869f92001-01-14 23:36:06 +0000207 # Internal: get a response from the server, which may possibly
208 # consist of multiple lines. Return a single string with no
209 # trailing CRLF. If the response consists of multiple lines,
210 # these are separated by '\n' characters in the string
211 def getmultiline(self):
212 line = self.getline()
213 if line[3:4] == '-':
214 code = line[:3]
215 while 1:
216 nextline = self.getline()
217 line = line + ('\n' + nextline)
218 if nextline[:3] == code and \
219 nextline[3:4] != '-':
220 break
221 return line
Guido van Rossum1115ab21992-11-04 15:51:30 +0000222
Tim Peters88869f92001-01-14 23:36:06 +0000223 # Internal: get a response from the server.
224 # Raise various errors if the response indicates an error
225 def getresp(self):
226 resp = self.getmultiline()
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000227 if self.debugging: print('*resp*', self.sanitize(resp))
Tim Peters88869f92001-01-14 23:36:06 +0000228 self.lastresp = resp[:3]
229 c = resp[:1]
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000230 if c in {'1', '2', '3'}:
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000231 return resp
Tim Peters88869f92001-01-14 23:36:06 +0000232 if c == '4':
Collin Winterce36ad82007-08-30 01:19:48 +0000233 raise error_temp(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000234 if c == '5':
Collin Winterce36ad82007-08-30 01:19:48 +0000235 raise error_perm(resp)
236 raise error_proto(resp)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000237
Tim Peters88869f92001-01-14 23:36:06 +0000238 def voidresp(self):
239 """Expect a response beginning with '2'."""
240 resp = self.getresp()
Benjamin Petersond23f8222009-04-05 19:13:16 +0000241 if resp[:1] != '2':
Collin Winterce36ad82007-08-30 01:19:48 +0000242 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000243 return resp
Guido van Rossumc567c601992-11-05 22:22:37 +0000244
Tim Peters88869f92001-01-14 23:36:06 +0000245 def abort(self):
246 '''Abort a file transfer. Uses out-of-band data.
247 This does not follow the procedure from the RFC to send Telnet
248 IP and Synch; that doesn't seem to work with the servers I've
249 tried. Instead, just send the ABOR command as OOB data.'''
Giampaolo Rodola'0b5c21f2011-05-07 19:03:47 +0200250 line = b'ABOR' + B_CRLF
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000251 if self.debugging > 1: print('*put urgent*', self.sanitize(line))
Martin v. Löwise12454f2002-02-16 23:06:19 +0000252 self.sock.sendall(line, MSG_OOB)
Tim Peters88869f92001-01-14 23:36:06 +0000253 resp = self.getmultiline()
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000254 if resp[:3] not in {'426', '225', '226'}:
Collin Winterce36ad82007-08-30 01:19:48 +0000255 raise error_proto(resp)
Giampaolo Rodola'0b5c21f2011-05-07 19:03:47 +0200256 return resp
Guido van Rossumd3166071993-05-24 14:16:22 +0000257
Tim Peters88869f92001-01-14 23:36:06 +0000258 def sendcmd(self, cmd):
259 '''Send a command and return the response.'''
260 self.putcmd(cmd)
261 return self.getresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000262
Tim Peters88869f92001-01-14 23:36:06 +0000263 def voidcmd(self, cmd):
264 """Send a command and expect a response beginning with '2'."""
265 self.putcmd(cmd)
266 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000267
Tim Peters88869f92001-01-14 23:36:06 +0000268 def sendport(self, host, port):
269 '''Send a PORT command with the current host and the given
270 port number.
271 '''
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000272 hbytes = host.split('.')
Benjamin Peterson3a53fbb2008-09-27 22:04:16 +0000273 pbytes = [repr(port//256), repr(port%256)]
Tim Peters88869f92001-01-14 23:36:06 +0000274 bytes = hbytes + pbytes
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000275 cmd = 'PORT ' + ','.join(bytes)
Tim Peters88869f92001-01-14 23:36:06 +0000276 return self.voidcmd(cmd)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000277
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000278 def sendeprt(self, host, port):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000279 '''Send a EPRT command with the current host and the given port number.'''
280 af = 0
281 if self.af == socket.AF_INET:
282 af = 1
283 if self.af == socket.AF_INET6:
284 af = 2
285 if af == 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000286 raise error_proto('unsupported address family')
Walter Dörwald70a6b492004-02-12 17:35:32 +0000287 fields = ['', repr(af), host, repr(port), '']
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000288 cmd = 'EPRT ' + '|'.join(fields)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000289 return self.voidcmd(cmd)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000290
Tim Peters88869f92001-01-14 23:36:06 +0000291 def makeport(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000292 '''Create a new socket and send a PORT command for it.'''
Martin v. Löwis2ad25692001-07-31 08:40:21 +0000293 msg = "getaddrinfo returns an empty list"
Martin v. Löwis322c0d12001-10-07 08:53:32 +0000294 sock = None
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000295 for res in socket.getaddrinfo(None, 0, self.af, socket.SOCK_STREAM, 0, socket.AI_PASSIVE):
296 af, socktype, proto, canonname, sa = res
297 try:
298 sock = socket.socket(af, socktype, proto)
299 sock.bind(sa)
Guido van Rossumb940e112007-01-10 16:19:56 +0000300 except socket.error as msg:
Martin v. Löwis322c0d12001-10-07 08:53:32 +0000301 if sock:
302 sock.close()
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000303 sock = None
304 continue
305 break
306 if not sock:
Collin Winterce36ad82007-08-30 01:19:48 +0000307 raise socket.error(msg)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000308 sock.listen(1)
309 port = sock.getsockname()[1] # Get proper port
310 host = self.sock.getsockname()[0] # Get proper host
311 if self.af == socket.AF_INET:
312 resp = self.sendport(host, port)
313 else:
314 resp = self.sendeprt(host, port)
Giampaolo RodolĂ 5fb313b2010-04-19 22:05:54 +0000315 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
316 sock.settimeout(self.timeout)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000317 return sock
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000318
319 def makepasv(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000320 if self.af == socket.AF_INET:
321 host, port = parse227(self.sendcmd('PASV'))
322 else:
323 host, port = parse229(self.sendcmd('EPSV'), self.sock.getpeername())
324 return host, port
Guido van Rossum1115ab21992-11-04 15:51:30 +0000325
Tim Peters88869f92001-01-14 23:36:06 +0000326 def ntransfercmd(self, cmd, rest=None):
327 """Initiate a transfer over the data connection.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000328
Tim Peters88869f92001-01-14 23:36:06 +0000329 If the transfer is active, send a port command and the
330 transfer command, and accept the connection. If the server is
331 passive, send a pasv command, connect to it, and start the
332 transfer command. Either way, return the socket for the
333 connection and the expected size of the transfer. The
334 expected size may be None if it could not be determined.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000335
Tim Peters88869f92001-01-14 23:36:06 +0000336 Optional `rest' argument can be a string that is sent as the
Christian Heimes1af737c2008-01-23 08:24:23 +0000337 argument to a REST command. This is essentially a server
Tim Peters88869f92001-01-14 23:36:06 +0000338 marker used to tell the server to skip over any data up to the
339 given marker.
340 """
341 size = None
342 if self.passiveserver:
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000343 host, port = self.makepasv()
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000344 conn = socket.create_connection((host, port), self.timeout,
345 source_address=self.source_address)
Tim Peters88869f92001-01-14 23:36:06 +0000346 if rest is not None:
347 self.sendcmd("REST %s" % rest)
348 resp = self.sendcmd(cmd)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000349 # Some servers apparently send a 200 reply to
350 # a LIST or STOR command, before the 150 reply
351 # (and way before the 226 reply). This seems to
352 # be in violation of the protocol (which only allows
353 # 1xx or error messages for LIST), so we just discard
354 # this response.
355 if resp[0] == '2':
Thomas Wouters9fe394c2007-02-05 01:24:16 +0000356 resp = self.getresp()
Tim Peters88869f92001-01-14 23:36:06 +0000357 if resp[0] != '1':
Collin Winterce36ad82007-08-30 01:19:48 +0000358 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000359 else:
360 sock = self.makeport()
361 if rest is not None:
362 self.sendcmd("REST %s" % rest)
363 resp = self.sendcmd(cmd)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000364 # See above.
365 if resp[0] == '2':
Thomas Wouters9fe394c2007-02-05 01:24:16 +0000366 resp = self.getresp()
Tim Peters88869f92001-01-14 23:36:06 +0000367 if resp[0] != '1':
Collin Winterce36ad82007-08-30 01:19:48 +0000368 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000369 conn, sockaddr = sock.accept()
Giampaolo RodolĂ 5fb313b2010-04-19 22:05:54 +0000370 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
371 conn.settimeout(self.timeout)
Benjamin Peterson1bd93a72010-10-31 19:58:07 +0000372 sock.close()
Tim Peters88869f92001-01-14 23:36:06 +0000373 if resp[:3] == '150':
374 # this is conditional in case we received a 125
375 size = parse150(resp)
376 return conn, size
Fred Drake4de02d91997-01-10 18:26:09 +0000377
Tim Peters88869f92001-01-14 23:36:06 +0000378 def transfercmd(self, cmd, rest=None):
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000379 """Like ntransfercmd() but returns only the socket."""
Tim Peters88869f92001-01-14 23:36:06 +0000380 return self.ntransfercmd(cmd, rest)[0]
Guido van Rossumc567c601992-11-05 22:22:37 +0000381
Tim Peters88869f92001-01-14 23:36:06 +0000382 def login(self, user = '', passwd = '', acct = ''):
383 '''Login, default anonymous.'''
384 if not user: user = 'anonymous'
385 if not passwd: passwd = ''
386 if not acct: acct = ''
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000387 if user == 'anonymous' and passwd in {'', '-'}:
Tim Peterse4418602002-02-16 07:34:19 +0000388 # If there is no anonymous ftp password specified
389 # then we'll just use anonymous@
390 # We don't send any other thing because:
391 # - We want to remain anonymous
392 # - We want to stop SPAM
393 # - We don't want to let ftp sites to discriminate by the user,
394 # host or country.
Guido van Rossumc33e0772001-12-28 20:54:28 +0000395 passwd = passwd + 'anonymous@'
Tim Peters88869f92001-01-14 23:36:06 +0000396 resp = self.sendcmd('USER ' + user)
397 if resp[0] == '3': resp = self.sendcmd('PASS ' + passwd)
398 if resp[0] == '3': resp = self.sendcmd('ACCT ' + acct)
399 if resp[0] != '2':
Collin Winterce36ad82007-08-30 01:19:48 +0000400 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000401 return resp
Guido van Rossumc567c601992-11-05 22:22:37 +0000402
Tim Peters88869f92001-01-14 23:36:06 +0000403 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000404 """Retrieve data in binary mode. A new port is created for you.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000405
Christian Heimes1af737c2008-01-23 08:24:23 +0000406 Args:
407 cmd: A RETR command.
408 callback: A single parameter callable to be called on each
409 block of data read.
410 blocksize: The maximum number of bytes to read from the
411 socket at one time. [default: 8192]
412 rest: Passed to transfercmd(). [default: None]
Guido van Rossum1115ab21992-11-04 15:51:30 +0000413
Christian Heimes1af737c2008-01-23 08:24:23 +0000414 Returns:
415 The response code.
Tim Peters88869f92001-01-14 23:36:06 +0000416 """
417 self.voidcmd('TYPE I')
Benjamin Petersona8071372010-10-31 18:21:16 +0000418 with self.transfercmd(cmd, rest) as conn:
419 while 1:
420 data = conn.recv(blocksize)
421 if not data:
422 break
423 callback(data)
Tim Peters88869f92001-01-14 23:36:06 +0000424 return self.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000425
Tim Peters88869f92001-01-14 23:36:06 +0000426 def retrlines(self, cmd, callback = None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000427 """Retrieve data in line mode. A new port is created for you.
428
429 Args:
Giampaolo Rodola'd78def92011-05-06 19:49:08 +0200430 cmd: A RETR, LIST, or NLST command.
Christian Heimes1af737c2008-01-23 08:24:23 +0000431 callback: An optional single parameter callable that is called
432 for each line with the trailing CRLF stripped.
433 [default: print_line()]
434
435 Returns:
436 The response code.
437 """
Raymond Hettingere874fc32002-05-12 05:53:51 +0000438 if callback is None: callback = print_line
Tim Peters88869f92001-01-14 23:36:06 +0000439 resp = self.sendcmd('TYPE A')
Benjamin Petersona8071372010-10-31 18:21:16 +0000440 with self.transfercmd(cmd) as conn, \
441 conn.makefile('r', encoding=self.encoding) as fp:
442 while 1:
443 line = fp.readline()
444 if self.debugging > 2: print('*retr*', repr(line))
445 if not line:
446 break
447 if line[-2:] == CRLF:
448 line = line[:-2]
449 elif line[-1:] == '\n':
450 line = line[:-1]
451 callback(line)
Tim Peters88869f92001-01-14 23:36:06 +0000452 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000453
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000454 def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000455 """Store a file in binary mode. A new port is created for you.
456
457 Args:
458 cmd: A STOR command.
459 fp: A file-like object with a read(num_bytes) method.
460 blocksize: The maximum data size to read from fp and send over
461 the connection at once. [default: 8192]
462 callback: An optional single parameter callable that is called on
463 on each block of data after it is sent. [default: None]
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000464 rest: Passed to transfercmd(). [default: None]
Christian Heimes1af737c2008-01-23 08:24:23 +0000465
466 Returns:
467 The response code.
468 """
Tim Peters88869f92001-01-14 23:36:06 +0000469 self.voidcmd('TYPE I')
Benjamin Petersona8071372010-10-31 18:21:16 +0000470 with self.transfercmd(cmd, rest) as conn:
471 while 1:
472 buf = fp.read(blocksize)
473 if not buf: break
474 conn.sendall(buf)
475 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000476 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000477
Christian Heimes1af737c2008-01-23 08:24:23 +0000478 def storlines(self, cmd, fp, callback=None):
479 """Store a file in line mode. A new port is created for you.
480
481 Args:
482 cmd: A STOR command.
483 fp: A file-like object with a readline() method.
484 callback: An optional single parameter callable that is called on
485 on each line after it is sent. [default: None]
486
487 Returns:
488 The response code.
489 """
Tim Peters88869f92001-01-14 23:36:06 +0000490 self.voidcmd('TYPE A')
Benjamin Petersona8071372010-10-31 18:21:16 +0000491 with self.transfercmd(cmd) as conn:
492 while 1:
493 buf = fp.readline()
494 if not buf: break
495 if buf[-2:] != B_CRLF:
496 if buf[-1] in B_CRLF: buf = buf[:-1]
497 buf = buf + B_CRLF
498 conn.sendall(buf)
499 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000500 return self.voidresp()
Guido van Rossum0eaa74b1996-01-25 18:37:21 +0000501
Tim Peters88869f92001-01-14 23:36:06 +0000502 def acct(self, password):
503 '''Send new account name.'''
504 cmd = 'ACCT ' + password
505 return self.voidcmd(cmd)
Guido van Rossumc567c601992-11-05 22:22:37 +0000506
Tim Peters88869f92001-01-14 23:36:06 +0000507 def nlst(self, *args):
508 '''Return a list of files in a given directory (default the current).'''
509 cmd = 'NLST'
510 for arg in args:
511 cmd = cmd + (' ' + arg)
512 files = []
513 self.retrlines(cmd, files.append)
514 return files
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000515
Tim Peters88869f92001-01-14 23:36:06 +0000516 def dir(self, *args):
517 '''List a directory in long form.
518 By default list current directory to stdout.
519 Optional last argument is callback function; all
520 non-empty arguments before it are concatenated to the
521 LIST command. (This *should* only be used for a pathname.)'''
522 cmd = 'LIST'
523 func = None
524 if args[-1:] and type(args[-1]) != type(''):
525 args, func = args[:-1], args[-1]
526 for arg in args:
527 if arg:
528 cmd = cmd + (' ' + arg)
529 self.retrlines(cmd, func)
Guido van Rossumc567c601992-11-05 22:22:37 +0000530
Giampaolo Rodola'd78def92011-05-06 19:49:08 +0200531 def mlsd(self, path="", facts=[]):
532 '''List a directory in a standardized format by using MLSD
533 command (RFC-3659). If path is omitted the current directory
534 is assumed. "facts" is a list of strings representing the type
535 of information desired (e.g. ["type", "size", "perm"]).
536
537 Return a generator object yielding a tuple of two elements
538 for every file found in path.
539 First element is the file name, the second one is a dictionary
540 including a variable number of "facts" depending on the server
541 and whether "facts" argument has been provided.
542 '''
543 if facts:
544 self.sendcmd("OPTS MLST " + ";".join(facts) + ";")
545 if path:
546 cmd = "MLSD %s" % path
547 else:
548 cmd = "MLSD"
549 lines = []
550 self.retrlines(cmd, lines.append)
551 for line in lines:
552 facts_found, _, name = line.rstrip(CRLF).partition(' ')
553 entry = {}
554 for fact in facts_found[:-1].split(";"):
555 key, _, value = fact.partition("=")
556 entry[key.lower()] = value
557 yield (name, entry)
558
Tim Peters88869f92001-01-14 23:36:06 +0000559 def rename(self, fromname, toname):
560 '''Rename a file.'''
561 resp = self.sendcmd('RNFR ' + fromname)
562 if resp[0] != '3':
Collin Winterce36ad82007-08-30 01:19:48 +0000563 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000564 return self.voidcmd('RNTO ' + toname)
Guido van Rossuma61bdeb1995-10-11 17:36:31 +0000565
Tim Peters88869f92001-01-14 23:36:06 +0000566 def delete(self, filename):
567 '''Delete a file.'''
568 resp = self.sendcmd('DELE ' + filename)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000569 if resp[:3] in {'250', '200'}:
Tim Peters88869f92001-01-14 23:36:06 +0000570 return resp
Tim Peters88869f92001-01-14 23:36:06 +0000571 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000572 raise error_reply(resp)
Guido van Rossum02cf5821993-05-17 08:00:02 +0000573
Tim Peters88869f92001-01-14 23:36:06 +0000574 def cwd(self, dirname):
575 '''Change to a directory.'''
576 if dirname == '..':
577 try:
578 return self.voidcmd('CDUP')
Guido van Rossumb940e112007-01-10 16:19:56 +0000579 except error_perm as msg:
Martin v. Löwisb5255112002-03-10 15:59:58 +0000580 if msg.args[0][:3] != '500':
581 raise
Tim Peters88869f92001-01-14 23:36:06 +0000582 elif dirname == '':
583 dirname = '.' # does nothing, but could return error
584 cmd = 'CWD ' + dirname
585 return self.voidcmd(cmd)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000586
Tim Peters88869f92001-01-14 23:36:06 +0000587 def size(self, filename):
588 '''Retrieve the size of a file.'''
Christian Heimes1af737c2008-01-23 08:24:23 +0000589 # The SIZE command is defined in RFC-3659
Tim Peters88869f92001-01-14 23:36:06 +0000590 resp = self.sendcmd('SIZE ' + filename)
591 if resp[:3] == '213':
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000592 s = resp[3:].strip()
593 try:
594 return int(s)
Guido van Rossum1f74cb32001-10-17 17:21:47 +0000595 except (OverflowError, ValueError):
Guido van Rossume2a383d2007-01-15 16:59:06 +0000596 return int(s)
Guido van Rossumc567c601992-11-05 22:22:37 +0000597
Tim Peters88869f92001-01-14 23:36:06 +0000598 def mkd(self, dirname):
599 '''Make a directory, return its full pathname.'''
Giampaolo RodolĂ bbc47822010-08-23 22:10:32 +0000600 resp = self.voidcmd('MKD ' + dirname)
601 # fix around non-compliant implementations such as IIS shipped
602 # with Windows server 2003
603 if not resp.startswith('257'):
604 return ''
Tim Peters88869f92001-01-14 23:36:06 +0000605 return parse257(resp)
Guido van Rossum98245091998-02-19 21:15:44 +0000606
Tim Peters88869f92001-01-14 23:36:06 +0000607 def rmd(self, dirname):
608 '''Remove a directory.'''
609 return self.voidcmd('RMD ' + dirname)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000610
Tim Peters88869f92001-01-14 23:36:06 +0000611 def pwd(self):
612 '''Return current working directory.'''
Giampaolo RodolĂ bbc47822010-08-23 22:10:32 +0000613 resp = self.voidcmd('PWD')
614 # fix around non-compliant implementations such as IIS shipped
615 # with Windows server 2003
616 if not resp.startswith('257'):
617 return ''
Tim Peters88869f92001-01-14 23:36:06 +0000618 return parse257(resp)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000619
Tim Peters88869f92001-01-14 23:36:06 +0000620 def quit(self):
621 '''Quit, and close the connection.'''
622 resp = self.voidcmd('QUIT')
623 self.close()
624 return resp
625
626 def close(self):
627 '''Close the connection without assuming anything about it.'''
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000628 if self.file is not None:
Fred Drake9c98a422001-02-28 21:46:37 +0000629 self.file.close()
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000630 if self.sock is not None:
Fred Drake9c98a422001-02-28 21:46:37 +0000631 self.sock.close()
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000632 self.file = self.sock = None
Guido van Rossumc567c601992-11-05 22:22:37 +0000633
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000634try:
635 import ssl
636except ImportError:
637 pass
638else:
639 class FTP_TLS(FTP):
640 '''A FTP subclass which adds TLS support to FTP as described
641 in RFC-4217.
642
643 Connect as usual to port 21 implicitly securing the FTP control
644 connection before authenticating.
645
646 Securing the data connection requires user to explicitly ask
647 for it by calling prot_p() method.
648
649 Usage example:
650 >>> from ftplib import FTP_TLS
651 >>> ftps = FTP_TLS('ftp.python.org')
Ezio Melotti4969f702011-03-15 05:59:46 +0200652 >>> ftps.login() # login anonymously previously securing control channel
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000653 '230 Guest login ok, access restrictions apply.'
654 >>> ftps.prot_p() # switch to secure data connection
655 '200 Protection level set to P'
656 >>> ftps.retrlines('LIST') # list directory content securely
657 total 9
658 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 .
659 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 ..
660 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 bin
661 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 etc
662 d-wxrwxr-x 2 ftp wheel 1024 Sep 5 13:43 incoming
663 drwxr-xr-x 2 root wheel 1024 Nov 17 1993 lib
664 drwxr-xr-x 6 1094 wheel 1024 Sep 13 19:07 pub
665 drwxr-xr-x 3 root wheel 1024 Jan 3 1994 usr
666 -rw-r--r-- 1 root root 312 Aug 1 1994 welcome.msg
667 '226 Transfer complete.'
668 >>> ftps.quit()
669 '221 Goodbye.'
670 >>>
671 '''
672 ssl_version = ssl.PROTOCOL_TLSv1
673
674 def __init__(self, host='', user='', passwd='', acct='', keyfile=None,
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000675 certfile=None, context=None,
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000676 timeout=_GLOBAL_DEFAULT_TIMEOUT, source_address=None):
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000677 if context is not None and keyfile is not None:
678 raise ValueError("context and keyfile arguments are mutually "
679 "exclusive")
680 if context is not None and certfile is not None:
681 raise ValueError("context and certfile arguments are mutually "
682 "exclusive")
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000683 self.keyfile = keyfile
684 self.certfile = certfile
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000685 self.context = context
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000686 self._prot_p = False
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000687 FTP.__init__(self, host, user, passwd, acct, timeout, source_address)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000688
689 def login(self, user='', passwd='', acct='', secure=True):
690 if secure and not isinstance(self.sock, ssl.SSLSocket):
691 self.auth()
692 return FTP.login(self, user, passwd, acct)
693
694 def auth(self):
695 '''Set up secure control connection by using TLS/SSL.'''
696 if isinstance(self.sock, ssl.SSLSocket):
697 raise ValueError("Already using TLS")
698 if self.ssl_version == ssl.PROTOCOL_TLSv1:
699 resp = self.voidcmd('AUTH TLS')
700 else:
701 resp = self.voidcmd('AUTH SSL')
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000702 if self.context is not None:
703 self.sock = self.context.wrap_socket(self.sock)
704 else:
705 self.sock = ssl.wrap_socket(self.sock, self.keyfile,
706 self.certfile,
707 ssl_version=self.ssl_version)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000708 self.file = self.sock.makefile(mode='r', encoding=self.encoding)
709 return resp
710
Giampaolo Rodola'096dcb12011-06-27 11:17:51 +0200711 def ccc(self):
712 '''Switch back to a clear-text control connection.'''
713 if not isinstance(self.sock, ssl.SSLSocket):
714 raise ValueError("not using TLS")
715 resp = self.voidcmd('CCC')
716 self.sock = self.sock.unwrap()
717 return resp
718
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000719 def prot_p(self):
720 '''Set up secure data connection.'''
721 # PROT defines whether or not the data channel is to be protected.
722 # Though RFC-2228 defines four possible protection levels,
723 # RFC-4217 only recommends two, Clear and Private.
724 # Clear (PROT C) means that no security is to be used on the
725 # data-channel, Private (PROT P) means that the data-channel
726 # should be protected by TLS.
727 # PBSZ command MUST still be issued, but must have a parameter of
728 # '0' to indicate that no buffering is taking place and the data
729 # connection should not be encapsulated.
730 self.voidcmd('PBSZ 0')
731 resp = self.voidcmd('PROT P')
732 self._prot_p = True
733 return resp
734
735 def prot_c(self):
736 '''Set up clear text data connection.'''
737 resp = self.voidcmd('PROT C')
738 self._prot_p = False
739 return resp
740
741 # --- Overridden FTP methods
742
743 def ntransfercmd(self, cmd, rest=None):
744 conn, size = FTP.ntransfercmd(self, cmd, rest)
745 if self._prot_p:
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000746 if self.context is not None:
747 conn = self.context.wrap_socket(conn)
748 else:
749 conn = ssl.wrap_socket(conn, self.keyfile, self.certfile,
750 ssl_version=self.ssl_version)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000751 return conn, size
752
753 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
754 self.voidcmd('TYPE I')
755 conn = self.transfercmd(cmd, rest)
756 try:
757 while 1:
758 data = conn.recv(blocksize)
759 if not data:
760 break
761 callback(data)
762 # shutdown ssl layer
763 if isinstance(conn, ssl.SSLSocket):
764 conn.unwrap()
765 finally:
766 conn.close()
767 return self.voidresp()
768
769 def retrlines(self, cmd, callback = None):
770 if callback is None: callback = print_line
771 resp = self.sendcmd('TYPE A')
772 conn = self.transfercmd(cmd)
773 fp = conn.makefile('r', encoding=self.encoding)
774 try:
775 while 1:
776 line = fp.readline()
777 if self.debugging > 2: print('*retr*', repr(line))
778 if not line:
779 break
780 if line[-2:] == CRLF:
781 line = line[:-2]
782 elif line[-1:] == '\n':
783 line = line[:-1]
784 callback(line)
785 # shutdown ssl layer
786 if isinstance(conn, ssl.SSLSocket):
787 conn.unwrap()
788 finally:
789 fp.close()
790 conn.close()
791 return self.voidresp()
792
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000793 def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000794 self.voidcmd('TYPE I')
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000795 conn = self.transfercmd(cmd, rest)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000796 try:
797 while 1:
798 buf = fp.read(blocksize)
799 if not buf: break
800 conn.sendall(buf)
801 if callback: callback(buf)
802 # shutdown ssl layer
803 if isinstance(conn, ssl.SSLSocket):
804 conn.unwrap()
805 finally:
806 conn.close()
807 return self.voidresp()
808
809 def storlines(self, cmd, fp, callback=None):
810 self.voidcmd('TYPE A')
811 conn = self.transfercmd(cmd)
812 try:
813 while 1:
814 buf = fp.readline()
815 if not buf: break
Giampaolo RodolĂ f96482e2010-08-04 10:36:18 +0000816 if buf[-2:] != B_CRLF:
817 if buf[-1] in B_CRLF: buf = buf[:-1]
818 buf = buf + B_CRLF
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000819 conn.sendall(buf)
820 if callback: callback(buf)
821 # shutdown ssl layer
822 if isinstance(conn, ssl.SSLSocket):
823 conn.unwrap()
824 finally:
825 conn.close()
826 return self.voidresp()
827
Giampaolo Rodola'24befa82011-05-07 19:09:34 +0200828 def abort(self):
829 # overridden as we can't pass MSG_OOB flag to sendall()
830 line = b'ABOR' + B_CRLF
831 self.sock.sendall(line)
832 resp = self.getmultiline()
833 if resp[:3] not in {'426', '225', '226'}:
834 raise error_proto(resp)
835 return resp
836
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000837 __all__.append('FTP_TLS')
838 all_errors = (Error, IOError, EOFError, ssl.SSLError)
839
840
Guido van Rossumacfb82a1997-10-22 20:49:52 +0000841_150_re = None
Fred Drake4de02d91997-01-10 18:26:09 +0000842
843def parse150(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000844 '''Parse the '150' response for a RETR request.
845 Returns the expected transfer size or None; size is not guaranteed to
846 be present in the 150 message.
847 '''
848 if resp[:3] != '150':
Collin Winterce36ad82007-08-30 01:19:48 +0000849 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000850 global _150_re
851 if _150_re is None:
852 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000853 _150_re = re.compile(
854 "150 .* \((\d+) bytes\)", re.IGNORECASE | re.ASCII)
Tim Peters88869f92001-01-14 23:36:06 +0000855 m = _150_re.match(resp)
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000856 if not m:
857 return None
858 s = m.group(1)
859 try:
860 return int(s)
Guido van Rossum1f74cb32001-10-17 17:21:47 +0000861 except (OverflowError, ValueError):
Guido van Rossume2a383d2007-01-15 16:59:06 +0000862 return int(s)
Fred Drake4de02d91997-01-10 18:26:09 +0000863
864
Guido van Rossum70297d32001-08-17 17:24:29 +0000865_227_re = None
866
Guido van Rossumd2560b01996-05-28 23:41:25 +0000867def parse227(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000868 '''Parse the '227' response for a PASV request.
869 Raises error_proto if it does not contain '(h1,h2,h3,h4,p1,p2)'
870 Return ('host.addr.as.numbers', port#) tuple.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000871
Tim Peters88869f92001-01-14 23:36:06 +0000872 if resp[:3] != '227':
Collin Winterce36ad82007-08-30 01:19:48 +0000873 raise error_reply(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000874 global _227_re
875 if _227_re is None:
876 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000877 _227_re = re.compile(r'(\d+),(\d+),(\d+),(\d+),(\d+),(\d+)', re.ASCII)
Guido van Rossum70297d32001-08-17 17:24:29 +0000878 m = _227_re.search(resp)
879 if not m:
Collin Winterce36ad82007-08-30 01:19:48 +0000880 raise error_proto(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000881 numbers = m.groups()
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000882 host = '.'.join(numbers[:4])
883 port = (int(numbers[4]) << 8) + int(numbers[5])
Tim Peters88869f92001-01-14 23:36:06 +0000884 return host, port
Guido van Rossumd2560b01996-05-28 23:41:25 +0000885
886
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000887def parse229(resp, peer):
888 '''Parse the '229' response for a EPSV request.
889 Raises error_proto if it does not contain '(|||port|)'
890 Return ('host.addr.as.numbers', port#) tuple.'''
891
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000892 if resp[:3] != '229':
Collin Winterce36ad82007-08-30 01:19:48 +0000893 raise error_reply(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000894 left = resp.find('(')
Collin Winterce36ad82007-08-30 01:19:48 +0000895 if left < 0: raise error_proto(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000896 right = resp.find(')', left + 1)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000897 if right < 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000898 raise error_proto(resp) # should contain '(|||port|)'
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000899 if resp[left + 1] != resp[right - 1]:
Collin Winterce36ad82007-08-30 01:19:48 +0000900 raise error_proto(resp)
Walter Dörwalda401ae42002-06-03 10:41:45 +0000901 parts = resp[left + 1:right].split(resp[left+1])
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000902 if len(parts) != 5:
Collin Winterce36ad82007-08-30 01:19:48 +0000903 raise error_proto(resp)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000904 host = peer[0]
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000905 port = int(parts[3])
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000906 return host, port
907
908
Guido van Rossumc567c601992-11-05 22:22:37 +0000909def parse257(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000910 '''Parse the '257' response for a MKD or PWD request.
911 This is a response to a MKD or PWD request: a directory name.
912 Returns the directoryname in the 257 reply.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000913
Tim Peters88869f92001-01-14 23:36:06 +0000914 if resp[:3] != '257':
Collin Winterce36ad82007-08-30 01:19:48 +0000915 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000916 if resp[3:5] != ' "':
917 return '' # Not compliant to RFC 959, but UNIX ftpd does this
918 dirname = ''
919 i = 5
920 n = len(resp)
921 while i < n:
922 c = resp[i]
923 i = i+1
924 if c == '"':
925 if i >= n or resp[i] != '"':
926 break
927 i = i+1
928 dirname = dirname + c
929 return dirname
Guido van Rossum1115ab21992-11-04 15:51:30 +0000930
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000931
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000932def print_line(line):
Tim Peters88869f92001-01-14 23:36:06 +0000933 '''Default retrlines callback to print a line.'''
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000934 print(line)
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000935
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000936
Guido van Rossumd2560b01996-05-28 23:41:25 +0000937def ftpcp(source, sourcename, target, targetname = '', type = 'I'):
Tim Peters88869f92001-01-14 23:36:06 +0000938 '''Copy file from one FTP-instance to another.'''
939 if not targetname: targetname = sourcename
940 type = 'TYPE ' + type
941 source.voidcmd(type)
942 target.voidcmd(type)
943 sourcehost, sourceport = parse227(source.sendcmd('PASV'))
944 target.sendport(sourcehost, sourceport)
945 # RFC 959: the user must "listen" [...] BEFORE sending the
946 # transfer request.
947 # So: STOR before RETR, because here the target is a "user".
948 treply = target.sendcmd('STOR ' + targetname)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000949 if treply[:3] not in {'125', '150'}: raise error_proto # RFC 959
Tim Peters88869f92001-01-14 23:36:06 +0000950 sreply = source.sendcmd('RETR ' + sourcename)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000951 if sreply[:3] not in {'125', '150'}: raise error_proto # RFC 959
Tim Peters88869f92001-01-14 23:36:06 +0000952 source.voidresp()
953 target.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000954
Tim Peters88869f92001-01-14 23:36:06 +0000955
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000956class Netrc:
Tim Peters88869f92001-01-14 23:36:06 +0000957 """Class to parse & provide access to 'netrc' format files.
Fred Drake475d51d1997-06-24 22:02:54 +0000958
Tim Peters88869f92001-01-14 23:36:06 +0000959 See the netrc(4) man page for information on the file format.
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000960
Tim Peters88869f92001-01-14 23:36:06 +0000961 WARNING: This class is obsolete -- use module netrc instead.
Guido van Rossumc822a451998-12-22 16:49:16 +0000962
Tim Peters88869f92001-01-14 23:36:06 +0000963 """
964 __defuser = None
965 __defpasswd = None
966 __defacct = None
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000967
Tim Peters88869f92001-01-14 23:36:06 +0000968 def __init__(self, filename=None):
Raymond Hettinger094662a2002-06-01 01:29:16 +0000969 if filename is None:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000970 if "HOME" in os.environ:
Tim Peters88869f92001-01-14 23:36:06 +0000971 filename = os.path.join(os.environ["HOME"],
972 ".netrc")
973 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000974 raise IOError("specify file to load or set $HOME")
Tim Peters88869f92001-01-14 23:36:06 +0000975 self.__hosts = {}
976 self.__macros = {}
977 fp = open(filename, "r")
978 in_macro = 0
979 while 1:
980 line = fp.readline()
981 if not line: break
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000982 if in_macro and line.strip():
Tim Peters88869f92001-01-14 23:36:06 +0000983 macro_lines.append(line)
984 continue
985 elif in_macro:
986 self.__macros[macro_name] = tuple(macro_lines)
987 in_macro = 0
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000988 words = line.split()
Tim Peters88869f92001-01-14 23:36:06 +0000989 host = user = passwd = acct = None
990 default = 0
991 i = 0
992 while i < len(words):
993 w1 = words[i]
994 if i+1 < len(words):
995 w2 = words[i + 1]
996 else:
997 w2 = None
998 if w1 == 'default':
999 default = 1
1000 elif w1 == 'machine' and w2:
Eric S. Raymondc95bf692001-02-09 10:06:47 +00001001 host = w2.lower()
Tim Peters88869f92001-01-14 23:36:06 +00001002 i = i + 1
1003 elif w1 == 'login' and w2:
1004 user = w2
1005 i = i + 1
1006 elif w1 == 'password' and w2:
1007 passwd = w2
1008 i = i + 1
1009 elif w1 == 'account' and w2:
1010 acct = w2
1011 i = i + 1
1012 elif w1 == 'macdef' and w2:
1013 macro_name = w2
1014 macro_lines = []
1015 in_macro = 1
1016 break
1017 i = i + 1
1018 if default:
1019 self.__defuser = user or self.__defuser
1020 self.__defpasswd = passwd or self.__defpasswd
1021 self.__defacct = acct or self.__defacct
1022 if host:
Raymond Hettinger54f02222002-06-01 14:18:47 +00001023 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +00001024 ouser, opasswd, oacct = \
1025 self.__hosts[host]
1026 user = user or ouser
1027 passwd = passwd or opasswd
1028 acct = acct or oacct
1029 self.__hosts[host] = user, passwd, acct
1030 fp.close()
Guido van Rossum56d1e3a1997-03-14 04:16:54 +00001031
Tim Peters88869f92001-01-14 23:36:06 +00001032 def get_hosts(self):
1033 """Return a list of hosts mentioned in the .netrc file."""
1034 return self.__hosts.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +00001035
Tim Peters88869f92001-01-14 23:36:06 +00001036 def get_account(self, host):
1037 """Returns login information for the named host.
Guido van Rossum8ca84201998-03-26 20:56:10 +00001038
Tim Peters88869f92001-01-14 23:36:06 +00001039 The return value is a triple containing userid,
1040 password, and the accounting field.
Guido van Rossum8ca84201998-03-26 20:56:10 +00001041
Tim Peters88869f92001-01-14 23:36:06 +00001042 """
Eric S. Raymondc95bf692001-02-09 10:06:47 +00001043 host = host.lower()
Tim Peters88869f92001-01-14 23:36:06 +00001044 user = passwd = acct = None
Raymond Hettinger54f02222002-06-01 14:18:47 +00001045 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +00001046 user, passwd, acct = self.__hosts[host]
1047 user = user or self.__defuser
1048 passwd = passwd or self.__defpasswd
1049 acct = acct or self.__defacct
1050 return user, passwd, acct
Guido van Rossum8ca84201998-03-26 20:56:10 +00001051
Tim Peters88869f92001-01-14 23:36:06 +00001052 def get_macros(self):
1053 """Return a list of all defined macro names."""
1054 return self.__macros.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +00001055
Tim Peters88869f92001-01-14 23:36:06 +00001056 def get_macro(self, macro):
1057 """Return a sequence of lines which define a named macro."""
1058 return self.__macros[macro]
Guido van Rossum56d1e3a1997-03-14 04:16:54 +00001059
Fred Drake475d51d1997-06-24 22:02:54 +00001060
Tim Peters88869f92001-01-14 23:36:06 +00001061
Guido van Rossum1115ab21992-11-04 15:51:30 +00001062def test():
Tim Peters88869f92001-01-14 23:36:06 +00001063 '''Test program.
Raymond Hettingerc88a6c72005-04-05 04:31:09 +00001064 Usage: ftp [-d] [-r[file]] host [-l[dir]] [-d[dir]] [-p] [file] ...
1065
1066 -d dir
1067 -l list
1068 -p password
1069 '''
1070
1071 if len(sys.argv) < 2:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001072 print(test.__doc__)
Raymond Hettingerc88a6c72005-04-05 04:31:09 +00001073 sys.exit(0)
Guido van Rossumd2560b01996-05-28 23:41:25 +00001074
Tim Peters88869f92001-01-14 23:36:06 +00001075 debugging = 0
1076 rcfile = None
1077 while sys.argv[1] == '-d':
1078 debugging = debugging+1
1079 del sys.argv[1]
1080 if sys.argv[1][:2] == '-r':
1081 # get name of alternate ~/.netrc file:
1082 rcfile = sys.argv[1][2:]
1083 del sys.argv[1]
1084 host = sys.argv[1]
1085 ftp = FTP(host)
1086 ftp.set_debuglevel(debugging)
1087 userid = passwd = acct = ''
1088 try:
1089 netrc = Netrc(rcfile)
1090 except IOError:
1091 if rcfile is not None:
1092 sys.stderr.write("Could not open account file"
1093 " -- using anonymous login.")
1094 else:
1095 try:
1096 userid, passwd, acct = netrc.get_account(host)
1097 except KeyError:
1098 # no account for host
1099 sys.stderr.write(
1100 "No account -- using anonymous login.")
1101 ftp.login(userid, passwd, acct)
1102 for file in sys.argv[2:]:
1103 if file[:2] == '-l':
1104 ftp.dir(file[2:])
1105 elif file[:2] == '-d':
1106 cmd = 'CWD'
1107 if file[2:]: cmd = cmd + ' ' + file[2:]
1108 resp = ftp.sendcmd(cmd)
1109 elif file == '-p':
1110 ftp.set_pasv(not ftp.passiveserver)
1111 else:
1112 ftp.retrbinary('RETR ' + file, \
1113 sys.stdout.write, 1024)
1114 ftp.quit()
Guido van Rossum221ec0b1995-08-04 04:39:30 +00001115
1116
1117if __name__ == '__main__':
Tim Peters88869f92001-01-14 23:36:06 +00001118 test()