blob: e836b72cd756b8f05725fdfc481437a91c3dabbe [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.'''
250 line = 'ABOR' + 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)
Guido van Rossumd3166071993-05-24 14:16:22 +0000256
Tim Peters88869f92001-01-14 23:36:06 +0000257 def sendcmd(self, cmd):
258 '''Send a command and return the response.'''
259 self.putcmd(cmd)
260 return self.getresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000261
Tim Peters88869f92001-01-14 23:36:06 +0000262 def voidcmd(self, cmd):
263 """Send a command and expect a response beginning with '2'."""
264 self.putcmd(cmd)
265 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000266
Tim Peters88869f92001-01-14 23:36:06 +0000267 def sendport(self, host, port):
268 '''Send a PORT command with the current host and the given
269 port number.
270 '''
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000271 hbytes = host.split('.')
Benjamin Peterson3a53fbb2008-09-27 22:04:16 +0000272 pbytes = [repr(port//256), repr(port%256)]
Tim Peters88869f92001-01-14 23:36:06 +0000273 bytes = hbytes + pbytes
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000274 cmd = 'PORT ' + ','.join(bytes)
Tim Peters88869f92001-01-14 23:36:06 +0000275 return self.voidcmd(cmd)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000276
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000277 def sendeprt(self, host, port):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000278 '''Send a EPRT command with the current host and the given port number.'''
279 af = 0
280 if self.af == socket.AF_INET:
281 af = 1
282 if self.af == socket.AF_INET6:
283 af = 2
284 if af == 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000285 raise error_proto('unsupported address family')
Walter Dörwald70a6b492004-02-12 17:35:32 +0000286 fields = ['', repr(af), host, repr(port), '']
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000287 cmd = 'EPRT ' + '|'.join(fields)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000288 return self.voidcmd(cmd)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000289
Tim Peters88869f92001-01-14 23:36:06 +0000290 def makeport(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000291 '''Create a new socket and send a PORT command for it.'''
Martin v. Löwis2ad25692001-07-31 08:40:21 +0000292 msg = "getaddrinfo returns an empty list"
Martin v. Löwis322c0d12001-10-07 08:53:32 +0000293 sock = None
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000294 for res in socket.getaddrinfo(None, 0, self.af, socket.SOCK_STREAM, 0, socket.AI_PASSIVE):
295 af, socktype, proto, canonname, sa = res
296 try:
297 sock = socket.socket(af, socktype, proto)
298 sock.bind(sa)
Guido van Rossumb940e112007-01-10 16:19:56 +0000299 except socket.error as msg:
Martin v. Löwis322c0d12001-10-07 08:53:32 +0000300 if sock:
301 sock.close()
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000302 sock = None
303 continue
304 break
305 if not sock:
Collin Winterce36ad82007-08-30 01:19:48 +0000306 raise socket.error(msg)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000307 sock.listen(1)
308 port = sock.getsockname()[1] # Get proper port
309 host = self.sock.getsockname()[0] # Get proper host
310 if self.af == socket.AF_INET:
311 resp = self.sendport(host, port)
312 else:
313 resp = self.sendeprt(host, port)
Giampaolo RodolĂ 5fb313b2010-04-19 22:05:54 +0000314 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
315 sock.settimeout(self.timeout)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000316 return sock
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000317
318 def makepasv(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000319 if self.af == socket.AF_INET:
320 host, port = parse227(self.sendcmd('PASV'))
321 else:
322 host, port = parse229(self.sendcmd('EPSV'), self.sock.getpeername())
323 return host, port
Guido van Rossum1115ab21992-11-04 15:51:30 +0000324
Tim Peters88869f92001-01-14 23:36:06 +0000325 def ntransfercmd(self, cmd, rest=None):
326 """Initiate a transfer over the data connection.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000327
Tim Peters88869f92001-01-14 23:36:06 +0000328 If the transfer is active, send a port command and the
329 transfer command, and accept the connection. If the server is
330 passive, send a pasv command, connect to it, and start the
331 transfer command. Either way, return the socket for the
332 connection and the expected size of the transfer. The
333 expected size may be None if it could not be determined.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000334
Tim Peters88869f92001-01-14 23:36:06 +0000335 Optional `rest' argument can be a string that is sent as the
Christian Heimes1af737c2008-01-23 08:24:23 +0000336 argument to a REST command. This is essentially a server
Tim Peters88869f92001-01-14 23:36:06 +0000337 marker used to tell the server to skip over any data up to the
338 given marker.
339 """
340 size = None
341 if self.passiveserver:
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000342 host, port = self.makepasv()
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000343 conn = socket.create_connection((host, port), self.timeout,
344 source_address=self.source_address)
Tim Peters88869f92001-01-14 23:36:06 +0000345 if rest is not None:
346 self.sendcmd("REST %s" % rest)
347 resp = self.sendcmd(cmd)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000348 # Some servers apparently send a 200 reply to
349 # a LIST or STOR command, before the 150 reply
350 # (and way before the 226 reply). This seems to
351 # be in violation of the protocol (which only allows
352 # 1xx or error messages for LIST), so we just discard
353 # this response.
354 if resp[0] == '2':
Thomas Wouters9fe394c2007-02-05 01:24:16 +0000355 resp = self.getresp()
Tim Peters88869f92001-01-14 23:36:06 +0000356 if resp[0] != '1':
Collin Winterce36ad82007-08-30 01:19:48 +0000357 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000358 else:
359 sock = self.makeport()
360 if rest is not None:
361 self.sendcmd("REST %s" % rest)
362 resp = self.sendcmd(cmd)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000363 # See above.
364 if resp[0] == '2':
Thomas Wouters9fe394c2007-02-05 01:24:16 +0000365 resp = self.getresp()
Tim Peters88869f92001-01-14 23:36:06 +0000366 if resp[0] != '1':
Collin Winterce36ad82007-08-30 01:19:48 +0000367 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000368 conn, sockaddr = sock.accept()
Giampaolo RodolĂ 5fb313b2010-04-19 22:05:54 +0000369 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
370 conn.settimeout(self.timeout)
Benjamin Peterson1bd93a72010-10-31 19:58:07 +0000371 sock.close()
Tim Peters88869f92001-01-14 23:36:06 +0000372 if resp[:3] == '150':
373 # this is conditional in case we received a 125
374 size = parse150(resp)
375 return conn, size
Fred Drake4de02d91997-01-10 18:26:09 +0000376
Tim Peters88869f92001-01-14 23:36:06 +0000377 def transfercmd(self, cmd, rest=None):
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000378 """Like ntransfercmd() but returns only the socket."""
Tim Peters88869f92001-01-14 23:36:06 +0000379 return self.ntransfercmd(cmd, rest)[0]
Guido van Rossumc567c601992-11-05 22:22:37 +0000380
Tim Peters88869f92001-01-14 23:36:06 +0000381 def login(self, user = '', passwd = '', acct = ''):
382 '''Login, default anonymous.'''
383 if not user: user = 'anonymous'
384 if not passwd: passwd = ''
385 if not acct: acct = ''
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000386 if user == 'anonymous' and passwd in {'', '-'}:
Tim Peterse4418602002-02-16 07:34:19 +0000387 # If there is no anonymous ftp password specified
388 # then we'll just use anonymous@
389 # We don't send any other thing because:
390 # - We want to remain anonymous
391 # - We want to stop SPAM
392 # - We don't want to let ftp sites to discriminate by the user,
393 # host or country.
Guido van Rossumc33e0772001-12-28 20:54:28 +0000394 passwd = passwd + 'anonymous@'
Tim Peters88869f92001-01-14 23:36:06 +0000395 resp = self.sendcmd('USER ' + user)
396 if resp[0] == '3': resp = self.sendcmd('PASS ' + passwd)
397 if resp[0] == '3': resp = self.sendcmd('ACCT ' + acct)
398 if resp[0] != '2':
Collin Winterce36ad82007-08-30 01:19:48 +0000399 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000400 return resp
Guido van Rossumc567c601992-11-05 22:22:37 +0000401
Tim Peters88869f92001-01-14 23:36:06 +0000402 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000403 """Retrieve data in binary mode. A new port is created for you.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000404
Christian Heimes1af737c2008-01-23 08:24:23 +0000405 Args:
406 cmd: A RETR command.
407 callback: A single parameter callable to be called on each
408 block of data read.
409 blocksize: The maximum number of bytes to read from the
410 socket at one time. [default: 8192]
411 rest: Passed to transfercmd(). [default: None]
Guido van Rossum1115ab21992-11-04 15:51:30 +0000412
Christian Heimes1af737c2008-01-23 08:24:23 +0000413 Returns:
414 The response code.
Tim Peters88869f92001-01-14 23:36:06 +0000415 """
416 self.voidcmd('TYPE I')
Benjamin Petersona8071372010-10-31 18:21:16 +0000417 with self.transfercmd(cmd, rest) as conn:
418 while 1:
419 data = conn.recv(blocksize)
420 if not data:
421 break
422 callback(data)
Tim Peters88869f92001-01-14 23:36:06 +0000423 return self.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000424
Tim Peters88869f92001-01-14 23:36:06 +0000425 def retrlines(self, cmd, callback = None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000426 """Retrieve data in line mode. A new port is created for you.
427
428 Args:
429 cmd: A RETR, LIST, NLST, or MLSD command.
430 callback: An optional single parameter callable that is called
431 for each line with the trailing CRLF stripped.
432 [default: print_line()]
433
434 Returns:
435 The response code.
436 """
Raymond Hettingere874fc32002-05-12 05:53:51 +0000437 if callback is None: callback = print_line
Tim Peters88869f92001-01-14 23:36:06 +0000438 resp = self.sendcmd('TYPE A')
Benjamin Petersona8071372010-10-31 18:21:16 +0000439 with self.transfercmd(cmd) as conn, \
440 conn.makefile('r', encoding=self.encoding) as fp:
441 while 1:
442 line = fp.readline()
443 if self.debugging > 2: print('*retr*', repr(line))
444 if not line:
445 break
446 if line[-2:] == CRLF:
447 line = line[:-2]
448 elif line[-1:] == '\n':
449 line = line[:-1]
450 callback(line)
Tim Peters88869f92001-01-14 23:36:06 +0000451 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000452
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000453 def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000454 """Store a file in binary mode. A new port is created for you.
455
456 Args:
457 cmd: A STOR command.
458 fp: A file-like object with a read(num_bytes) method.
459 blocksize: The maximum data size to read from fp and send over
460 the connection at once. [default: 8192]
461 callback: An optional single parameter callable that is called on
462 on each block of data after it is sent. [default: None]
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000463 rest: Passed to transfercmd(). [default: None]
Christian Heimes1af737c2008-01-23 08:24:23 +0000464
465 Returns:
466 The response code.
467 """
Tim Peters88869f92001-01-14 23:36:06 +0000468 self.voidcmd('TYPE I')
Benjamin Petersona8071372010-10-31 18:21:16 +0000469 with self.transfercmd(cmd, rest) as conn:
470 while 1:
471 buf = fp.read(blocksize)
472 if not buf: break
473 conn.sendall(buf)
474 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000475 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000476
Christian Heimes1af737c2008-01-23 08:24:23 +0000477 def storlines(self, cmd, fp, callback=None):
478 """Store a file in line mode. A new port is created for you.
479
480 Args:
481 cmd: A STOR command.
482 fp: A file-like object with a readline() method.
483 callback: An optional single parameter callable that is called on
484 on each line after it is sent. [default: None]
485
486 Returns:
487 The response code.
488 """
Tim Peters88869f92001-01-14 23:36:06 +0000489 self.voidcmd('TYPE A')
Benjamin Petersona8071372010-10-31 18:21:16 +0000490 with self.transfercmd(cmd) as conn:
491 while 1:
492 buf = fp.readline()
493 if not buf: break
494 if buf[-2:] != B_CRLF:
495 if buf[-1] in B_CRLF: buf = buf[:-1]
496 buf = buf + B_CRLF
497 conn.sendall(buf)
498 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000499 return self.voidresp()
Guido van Rossum0eaa74b1996-01-25 18:37:21 +0000500
Tim Peters88869f92001-01-14 23:36:06 +0000501 def acct(self, password):
502 '''Send new account name.'''
503 cmd = 'ACCT ' + password
504 return self.voidcmd(cmd)
Guido van Rossumc567c601992-11-05 22:22:37 +0000505
Tim Peters88869f92001-01-14 23:36:06 +0000506 def nlst(self, *args):
507 '''Return a list of files in a given directory (default the current).'''
508 cmd = 'NLST'
509 for arg in args:
510 cmd = cmd + (' ' + arg)
511 files = []
512 self.retrlines(cmd, files.append)
513 return files
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000514
Tim Peters88869f92001-01-14 23:36:06 +0000515 def dir(self, *args):
516 '''List a directory in long form.
517 By default list current directory to stdout.
518 Optional last argument is callback function; all
519 non-empty arguments before it are concatenated to the
520 LIST command. (This *should* only be used for a pathname.)'''
521 cmd = 'LIST'
522 func = None
523 if args[-1:] and type(args[-1]) != type(''):
524 args, func = args[:-1], args[-1]
525 for arg in args:
526 if arg:
527 cmd = cmd + (' ' + arg)
528 self.retrlines(cmd, func)
Guido van Rossumc567c601992-11-05 22:22:37 +0000529
Tim Peters88869f92001-01-14 23:36:06 +0000530 def rename(self, fromname, toname):
531 '''Rename a file.'''
532 resp = self.sendcmd('RNFR ' + fromname)
533 if resp[0] != '3':
Collin Winterce36ad82007-08-30 01:19:48 +0000534 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000535 return self.voidcmd('RNTO ' + toname)
Guido van Rossuma61bdeb1995-10-11 17:36:31 +0000536
Tim Peters88869f92001-01-14 23:36:06 +0000537 def delete(self, filename):
538 '''Delete a file.'''
539 resp = self.sendcmd('DELE ' + filename)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000540 if resp[:3] in {'250', '200'}:
Tim Peters88869f92001-01-14 23:36:06 +0000541 return resp
Tim Peters88869f92001-01-14 23:36:06 +0000542 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000543 raise error_reply(resp)
Guido van Rossum02cf5821993-05-17 08:00:02 +0000544
Tim Peters88869f92001-01-14 23:36:06 +0000545 def cwd(self, dirname):
546 '''Change to a directory.'''
547 if dirname == '..':
548 try:
549 return self.voidcmd('CDUP')
Guido van Rossumb940e112007-01-10 16:19:56 +0000550 except error_perm as msg:
Martin v. Löwisb5255112002-03-10 15:59:58 +0000551 if msg.args[0][:3] != '500':
552 raise
Tim Peters88869f92001-01-14 23:36:06 +0000553 elif dirname == '':
554 dirname = '.' # does nothing, but could return error
555 cmd = 'CWD ' + dirname
556 return self.voidcmd(cmd)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000557
Tim Peters88869f92001-01-14 23:36:06 +0000558 def size(self, filename):
559 '''Retrieve the size of a file.'''
Christian Heimes1af737c2008-01-23 08:24:23 +0000560 # The SIZE command is defined in RFC-3659
Tim Peters88869f92001-01-14 23:36:06 +0000561 resp = self.sendcmd('SIZE ' + filename)
562 if resp[:3] == '213':
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000563 s = resp[3:].strip()
564 try:
565 return int(s)
Guido van Rossum1f74cb32001-10-17 17:21:47 +0000566 except (OverflowError, ValueError):
Guido van Rossume2a383d2007-01-15 16:59:06 +0000567 return int(s)
Guido van Rossumc567c601992-11-05 22:22:37 +0000568
Tim Peters88869f92001-01-14 23:36:06 +0000569 def mkd(self, dirname):
570 '''Make a directory, return its full pathname.'''
Giampaolo RodolĂ bbc47822010-08-23 22:10:32 +0000571 resp = self.voidcmd('MKD ' + dirname)
572 # fix around non-compliant implementations such as IIS shipped
573 # with Windows server 2003
574 if not resp.startswith('257'):
575 return ''
Tim Peters88869f92001-01-14 23:36:06 +0000576 return parse257(resp)
Guido van Rossum98245091998-02-19 21:15:44 +0000577
Tim Peters88869f92001-01-14 23:36:06 +0000578 def rmd(self, dirname):
579 '''Remove a directory.'''
580 return self.voidcmd('RMD ' + dirname)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000581
Tim Peters88869f92001-01-14 23:36:06 +0000582 def pwd(self):
583 '''Return current working directory.'''
Giampaolo RodolĂ bbc47822010-08-23 22:10:32 +0000584 resp = self.voidcmd('PWD')
585 # fix around non-compliant implementations such as IIS shipped
586 # with Windows server 2003
587 if not resp.startswith('257'):
588 return ''
Tim Peters88869f92001-01-14 23:36:06 +0000589 return parse257(resp)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000590
Tim Peters88869f92001-01-14 23:36:06 +0000591 def quit(self):
592 '''Quit, and close the connection.'''
593 resp = self.voidcmd('QUIT')
594 self.close()
595 return resp
596
597 def close(self):
598 '''Close the connection without assuming anything about it.'''
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000599 if self.file is not None:
Fred Drake9c98a422001-02-28 21:46:37 +0000600 self.file.close()
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000601 if self.sock is not None:
Fred Drake9c98a422001-02-28 21:46:37 +0000602 self.sock.close()
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000603 self.file = self.sock = None
Guido van Rossumc567c601992-11-05 22:22:37 +0000604
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000605try:
606 import ssl
607except ImportError:
608 pass
609else:
610 class FTP_TLS(FTP):
611 '''A FTP subclass which adds TLS support to FTP as described
612 in RFC-4217.
613
614 Connect as usual to port 21 implicitly securing the FTP control
615 connection before authenticating.
616
617 Securing the data connection requires user to explicitly ask
618 for it by calling prot_p() method.
619
620 Usage example:
621 >>> from ftplib import FTP_TLS
622 >>> ftps = FTP_TLS('ftp.python.org')
Ezio Melotti4969f702011-03-15 05:59:46 +0200623 >>> ftps.login() # login anonymously previously securing control channel
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000624 '230 Guest login ok, access restrictions apply.'
625 >>> ftps.prot_p() # switch to secure data connection
626 '200 Protection level set to P'
627 >>> ftps.retrlines('LIST') # list directory content securely
628 total 9
629 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 .
630 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 ..
631 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 bin
632 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 etc
633 d-wxrwxr-x 2 ftp wheel 1024 Sep 5 13:43 incoming
634 drwxr-xr-x 2 root wheel 1024 Nov 17 1993 lib
635 drwxr-xr-x 6 1094 wheel 1024 Sep 13 19:07 pub
636 drwxr-xr-x 3 root wheel 1024 Jan 3 1994 usr
637 -rw-r--r-- 1 root root 312 Aug 1 1994 welcome.msg
638 '226 Transfer complete.'
639 >>> ftps.quit()
640 '221 Goodbye.'
641 >>>
642 '''
643 ssl_version = ssl.PROTOCOL_TLSv1
644
645 def __init__(self, host='', user='', passwd='', acct='', keyfile=None,
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000646 certfile=None, context=None,
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000647 timeout=_GLOBAL_DEFAULT_TIMEOUT, source_address=None):
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000648 if context is not None and keyfile is not None:
649 raise ValueError("context and keyfile arguments are mutually "
650 "exclusive")
651 if context is not None and certfile is not None:
652 raise ValueError("context and certfile arguments are mutually "
653 "exclusive")
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000654 self.keyfile = keyfile
655 self.certfile = certfile
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000656 self.context = context
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000657 self._prot_p = False
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000658 FTP.__init__(self, host, user, passwd, acct, timeout, source_address)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000659
660 def login(self, user='', passwd='', acct='', secure=True):
661 if secure and not isinstance(self.sock, ssl.SSLSocket):
662 self.auth()
663 return FTP.login(self, user, passwd, acct)
664
665 def auth(self):
666 '''Set up secure control connection by using TLS/SSL.'''
667 if isinstance(self.sock, ssl.SSLSocket):
668 raise ValueError("Already using TLS")
669 if self.ssl_version == ssl.PROTOCOL_TLSv1:
670 resp = self.voidcmd('AUTH TLS')
671 else:
672 resp = self.voidcmd('AUTH SSL')
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000673 if self.context is not None:
674 self.sock = self.context.wrap_socket(self.sock)
675 else:
676 self.sock = ssl.wrap_socket(self.sock, self.keyfile,
677 self.certfile,
678 ssl_version=self.ssl_version)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000679 self.file = self.sock.makefile(mode='r', encoding=self.encoding)
680 return resp
681
682 def prot_p(self):
683 '''Set up secure data connection.'''
684 # PROT defines whether or not the data channel is to be protected.
685 # Though RFC-2228 defines four possible protection levels,
686 # RFC-4217 only recommends two, Clear and Private.
687 # Clear (PROT C) means that no security is to be used on the
688 # data-channel, Private (PROT P) means that the data-channel
689 # should be protected by TLS.
690 # PBSZ command MUST still be issued, but must have a parameter of
691 # '0' to indicate that no buffering is taking place and the data
692 # connection should not be encapsulated.
693 self.voidcmd('PBSZ 0')
694 resp = self.voidcmd('PROT P')
695 self._prot_p = True
696 return resp
697
698 def prot_c(self):
699 '''Set up clear text data connection.'''
700 resp = self.voidcmd('PROT C')
701 self._prot_p = False
702 return resp
703
704 # --- Overridden FTP methods
705
706 def ntransfercmd(self, cmd, rest=None):
707 conn, size = FTP.ntransfercmd(self, cmd, rest)
708 if self._prot_p:
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000709 if self.context is not None:
710 conn = self.context.wrap_socket(conn)
711 else:
712 conn = ssl.wrap_socket(conn, self.keyfile, self.certfile,
713 ssl_version=self.ssl_version)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000714 return conn, size
715
716 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
717 self.voidcmd('TYPE I')
718 conn = self.transfercmd(cmd, rest)
719 try:
720 while 1:
721 data = conn.recv(blocksize)
722 if not data:
723 break
724 callback(data)
725 # shutdown ssl layer
726 if isinstance(conn, ssl.SSLSocket):
727 conn.unwrap()
728 finally:
729 conn.close()
730 return self.voidresp()
731
732 def retrlines(self, cmd, callback = None):
733 if callback is None: callback = print_line
734 resp = self.sendcmd('TYPE A')
735 conn = self.transfercmd(cmd)
736 fp = conn.makefile('r', encoding=self.encoding)
737 try:
738 while 1:
739 line = fp.readline()
740 if self.debugging > 2: print('*retr*', repr(line))
741 if not line:
742 break
743 if line[-2:] == CRLF:
744 line = line[:-2]
745 elif line[-1:] == '\n':
746 line = line[:-1]
747 callback(line)
748 # shutdown ssl layer
749 if isinstance(conn, ssl.SSLSocket):
750 conn.unwrap()
751 finally:
752 fp.close()
753 conn.close()
754 return self.voidresp()
755
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000756 def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000757 self.voidcmd('TYPE I')
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000758 conn = self.transfercmd(cmd, rest)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000759 try:
760 while 1:
761 buf = fp.read(blocksize)
762 if not buf: break
763 conn.sendall(buf)
764 if callback: callback(buf)
765 # shutdown ssl layer
766 if isinstance(conn, ssl.SSLSocket):
767 conn.unwrap()
768 finally:
769 conn.close()
770 return self.voidresp()
771
772 def storlines(self, cmd, fp, callback=None):
773 self.voidcmd('TYPE A')
774 conn = self.transfercmd(cmd)
775 try:
776 while 1:
777 buf = fp.readline()
778 if not buf: break
Giampaolo RodolĂ f96482e2010-08-04 10:36:18 +0000779 if buf[-2:] != B_CRLF:
780 if buf[-1] in B_CRLF: buf = buf[:-1]
781 buf = buf + B_CRLF
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000782 conn.sendall(buf)
783 if callback: callback(buf)
784 # shutdown ssl layer
785 if isinstance(conn, ssl.SSLSocket):
786 conn.unwrap()
787 finally:
788 conn.close()
789 return self.voidresp()
790
791 __all__.append('FTP_TLS')
792 all_errors = (Error, IOError, EOFError, ssl.SSLError)
793
794
Guido van Rossumacfb82a1997-10-22 20:49:52 +0000795_150_re = None
Fred Drake4de02d91997-01-10 18:26:09 +0000796
797def parse150(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000798 '''Parse the '150' response for a RETR request.
799 Returns the expected transfer size or None; size is not guaranteed to
800 be present in the 150 message.
801 '''
802 if resp[:3] != '150':
Collin Winterce36ad82007-08-30 01:19:48 +0000803 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000804 global _150_re
805 if _150_re is None:
806 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000807 _150_re = re.compile(
808 "150 .* \((\d+) bytes\)", re.IGNORECASE | re.ASCII)
Tim Peters88869f92001-01-14 23:36:06 +0000809 m = _150_re.match(resp)
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000810 if not m:
811 return None
812 s = m.group(1)
813 try:
814 return int(s)
Guido van Rossum1f74cb32001-10-17 17:21:47 +0000815 except (OverflowError, ValueError):
Guido van Rossume2a383d2007-01-15 16:59:06 +0000816 return int(s)
Fred Drake4de02d91997-01-10 18:26:09 +0000817
818
Guido van Rossum70297d32001-08-17 17:24:29 +0000819_227_re = None
820
Guido van Rossumd2560b01996-05-28 23:41:25 +0000821def parse227(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000822 '''Parse the '227' response for a PASV request.
823 Raises error_proto if it does not contain '(h1,h2,h3,h4,p1,p2)'
824 Return ('host.addr.as.numbers', port#) tuple.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000825
Tim Peters88869f92001-01-14 23:36:06 +0000826 if resp[:3] != '227':
Collin Winterce36ad82007-08-30 01:19:48 +0000827 raise error_reply(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000828 global _227_re
829 if _227_re is None:
830 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000831 _227_re = re.compile(r'(\d+),(\d+),(\d+),(\d+),(\d+),(\d+)', re.ASCII)
Guido van Rossum70297d32001-08-17 17:24:29 +0000832 m = _227_re.search(resp)
833 if not m:
Collin Winterce36ad82007-08-30 01:19:48 +0000834 raise error_proto(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000835 numbers = m.groups()
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000836 host = '.'.join(numbers[:4])
837 port = (int(numbers[4]) << 8) + int(numbers[5])
Tim Peters88869f92001-01-14 23:36:06 +0000838 return host, port
Guido van Rossumd2560b01996-05-28 23:41:25 +0000839
840
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000841def parse229(resp, peer):
842 '''Parse the '229' response for a EPSV request.
843 Raises error_proto if it does not contain '(|||port|)'
844 Return ('host.addr.as.numbers', port#) tuple.'''
845
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000846 if resp[:3] != '229':
Collin Winterce36ad82007-08-30 01:19:48 +0000847 raise error_reply(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000848 left = resp.find('(')
Collin Winterce36ad82007-08-30 01:19:48 +0000849 if left < 0: raise error_proto(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000850 right = resp.find(')', left + 1)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000851 if right < 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000852 raise error_proto(resp) # should contain '(|||port|)'
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000853 if resp[left + 1] != resp[right - 1]:
Collin Winterce36ad82007-08-30 01:19:48 +0000854 raise error_proto(resp)
Walter Dörwalda401ae42002-06-03 10:41:45 +0000855 parts = resp[left + 1:right].split(resp[left+1])
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000856 if len(parts) != 5:
Collin Winterce36ad82007-08-30 01:19:48 +0000857 raise error_proto(resp)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000858 host = peer[0]
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000859 port = int(parts[3])
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000860 return host, port
861
862
Guido van Rossumc567c601992-11-05 22:22:37 +0000863def parse257(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000864 '''Parse the '257' response for a MKD or PWD request.
865 This is a response to a MKD or PWD request: a directory name.
866 Returns the directoryname in the 257 reply.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000867
Tim Peters88869f92001-01-14 23:36:06 +0000868 if resp[:3] != '257':
Collin Winterce36ad82007-08-30 01:19:48 +0000869 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000870 if resp[3:5] != ' "':
871 return '' # Not compliant to RFC 959, but UNIX ftpd does this
872 dirname = ''
873 i = 5
874 n = len(resp)
875 while i < n:
876 c = resp[i]
877 i = i+1
878 if c == '"':
879 if i >= n or resp[i] != '"':
880 break
881 i = i+1
882 dirname = dirname + c
883 return dirname
Guido van Rossum1115ab21992-11-04 15:51:30 +0000884
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000885
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000886def print_line(line):
Tim Peters88869f92001-01-14 23:36:06 +0000887 '''Default retrlines callback to print a line.'''
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000888 print(line)
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000889
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000890
Guido van Rossumd2560b01996-05-28 23:41:25 +0000891def ftpcp(source, sourcename, target, targetname = '', type = 'I'):
Tim Peters88869f92001-01-14 23:36:06 +0000892 '''Copy file from one FTP-instance to another.'''
893 if not targetname: targetname = sourcename
894 type = 'TYPE ' + type
895 source.voidcmd(type)
896 target.voidcmd(type)
897 sourcehost, sourceport = parse227(source.sendcmd('PASV'))
898 target.sendport(sourcehost, sourceport)
899 # RFC 959: the user must "listen" [...] BEFORE sending the
900 # transfer request.
901 # So: STOR before RETR, because here the target is a "user".
902 treply = target.sendcmd('STOR ' + targetname)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000903 if treply[:3] not in {'125', '150'}: raise error_proto # RFC 959
Tim Peters88869f92001-01-14 23:36:06 +0000904 sreply = source.sendcmd('RETR ' + sourcename)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000905 if sreply[:3] not in {'125', '150'}: raise error_proto # RFC 959
Tim Peters88869f92001-01-14 23:36:06 +0000906 source.voidresp()
907 target.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000908
Tim Peters88869f92001-01-14 23:36:06 +0000909
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000910class Netrc:
Tim Peters88869f92001-01-14 23:36:06 +0000911 """Class to parse & provide access to 'netrc' format files.
Fred Drake475d51d1997-06-24 22:02:54 +0000912
Tim Peters88869f92001-01-14 23:36:06 +0000913 See the netrc(4) man page for information on the file format.
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000914
Tim Peters88869f92001-01-14 23:36:06 +0000915 WARNING: This class is obsolete -- use module netrc instead.
Guido van Rossumc822a451998-12-22 16:49:16 +0000916
Tim Peters88869f92001-01-14 23:36:06 +0000917 """
918 __defuser = None
919 __defpasswd = None
920 __defacct = None
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000921
Tim Peters88869f92001-01-14 23:36:06 +0000922 def __init__(self, filename=None):
Raymond Hettinger094662a2002-06-01 01:29:16 +0000923 if filename is None:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000924 if "HOME" in os.environ:
Tim Peters88869f92001-01-14 23:36:06 +0000925 filename = os.path.join(os.environ["HOME"],
926 ".netrc")
927 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000928 raise IOError("specify file to load or set $HOME")
Tim Peters88869f92001-01-14 23:36:06 +0000929 self.__hosts = {}
930 self.__macros = {}
931 fp = open(filename, "r")
932 in_macro = 0
933 while 1:
934 line = fp.readline()
935 if not line: break
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000936 if in_macro and line.strip():
Tim Peters88869f92001-01-14 23:36:06 +0000937 macro_lines.append(line)
938 continue
939 elif in_macro:
940 self.__macros[macro_name] = tuple(macro_lines)
941 in_macro = 0
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000942 words = line.split()
Tim Peters88869f92001-01-14 23:36:06 +0000943 host = user = passwd = acct = None
944 default = 0
945 i = 0
946 while i < len(words):
947 w1 = words[i]
948 if i+1 < len(words):
949 w2 = words[i + 1]
950 else:
951 w2 = None
952 if w1 == 'default':
953 default = 1
954 elif w1 == 'machine' and w2:
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000955 host = w2.lower()
Tim Peters88869f92001-01-14 23:36:06 +0000956 i = i + 1
957 elif w1 == 'login' and w2:
958 user = w2
959 i = i + 1
960 elif w1 == 'password' and w2:
961 passwd = w2
962 i = i + 1
963 elif w1 == 'account' and w2:
964 acct = w2
965 i = i + 1
966 elif w1 == 'macdef' and w2:
967 macro_name = w2
968 macro_lines = []
969 in_macro = 1
970 break
971 i = i + 1
972 if default:
973 self.__defuser = user or self.__defuser
974 self.__defpasswd = passwd or self.__defpasswd
975 self.__defacct = acct or self.__defacct
976 if host:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000977 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +0000978 ouser, opasswd, oacct = \
979 self.__hosts[host]
980 user = user or ouser
981 passwd = passwd or opasswd
982 acct = acct or oacct
983 self.__hosts[host] = user, passwd, acct
984 fp.close()
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000985
Tim Peters88869f92001-01-14 23:36:06 +0000986 def get_hosts(self):
987 """Return a list of hosts mentioned in the .netrc file."""
988 return self.__hosts.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +0000989
Tim Peters88869f92001-01-14 23:36:06 +0000990 def get_account(self, host):
991 """Returns login information for the named host.
Guido van Rossum8ca84201998-03-26 20:56:10 +0000992
Tim Peters88869f92001-01-14 23:36:06 +0000993 The return value is a triple containing userid,
994 password, and the accounting field.
Guido van Rossum8ca84201998-03-26 20:56:10 +0000995
Tim Peters88869f92001-01-14 23:36:06 +0000996 """
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000997 host = host.lower()
Tim Peters88869f92001-01-14 23:36:06 +0000998 user = passwd = acct = None
Raymond Hettinger54f02222002-06-01 14:18:47 +0000999 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +00001000 user, passwd, acct = self.__hosts[host]
1001 user = user or self.__defuser
1002 passwd = passwd or self.__defpasswd
1003 acct = acct or self.__defacct
1004 return user, passwd, acct
Guido van Rossum8ca84201998-03-26 20:56:10 +00001005
Tim Peters88869f92001-01-14 23:36:06 +00001006 def get_macros(self):
1007 """Return a list of all defined macro names."""
1008 return self.__macros.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +00001009
Tim Peters88869f92001-01-14 23:36:06 +00001010 def get_macro(self, macro):
1011 """Return a sequence of lines which define a named macro."""
1012 return self.__macros[macro]
Guido van Rossum56d1e3a1997-03-14 04:16:54 +00001013
Fred Drake475d51d1997-06-24 22:02:54 +00001014
Tim Peters88869f92001-01-14 23:36:06 +00001015
Guido van Rossum1115ab21992-11-04 15:51:30 +00001016def test():
Tim Peters88869f92001-01-14 23:36:06 +00001017 '''Test program.
Raymond Hettingerc88a6c72005-04-05 04:31:09 +00001018 Usage: ftp [-d] [-r[file]] host [-l[dir]] [-d[dir]] [-p] [file] ...
1019
1020 -d dir
1021 -l list
1022 -p password
1023 '''
1024
1025 if len(sys.argv) < 2:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001026 print(test.__doc__)
Raymond Hettingerc88a6c72005-04-05 04:31:09 +00001027 sys.exit(0)
Guido van Rossumd2560b01996-05-28 23:41:25 +00001028
Tim Peters88869f92001-01-14 23:36:06 +00001029 debugging = 0
1030 rcfile = None
1031 while sys.argv[1] == '-d':
1032 debugging = debugging+1
1033 del sys.argv[1]
1034 if sys.argv[1][:2] == '-r':
1035 # get name of alternate ~/.netrc file:
1036 rcfile = sys.argv[1][2:]
1037 del sys.argv[1]
1038 host = sys.argv[1]
1039 ftp = FTP(host)
1040 ftp.set_debuglevel(debugging)
1041 userid = passwd = acct = ''
1042 try:
1043 netrc = Netrc(rcfile)
1044 except IOError:
1045 if rcfile is not None:
1046 sys.stderr.write("Could not open account file"
1047 " -- using anonymous login.")
1048 else:
1049 try:
1050 userid, passwd, acct = netrc.get_account(host)
1051 except KeyError:
1052 # no account for host
1053 sys.stderr.write(
1054 "No account -- using anonymous login.")
1055 ftp.login(userid, passwd, acct)
1056 for file in sys.argv[2:]:
1057 if file[:2] == '-l':
1058 ftp.dir(file[2:])
1059 elif file[:2] == '-d':
1060 cmd = 'CWD'
1061 if file[2:]: cmd = cmd + ' ' + file[2:]
1062 resp = ftp.sendcmd(cmd)
1063 elif file == '-p':
1064 ftp.set_pasv(not ftp.passiveserver)
1065 else:
1066 ftp.retrbinary('RETR ' + file, \
1067 sys.stdout.write, 1024)
1068 ftp.quit()
Guido van Rossum221ec0b1995-08-04 04:39:30 +00001069
1070
1071if __name__ == '__main__':
Tim Peters88869f92001-01-14 23:36:06 +00001072 test()