blob: 5efae95a60fb8b4a9afca385293f7098d5234f8c [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):
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200178 if s[:5] in {'pass ', 'PASS '}:
179 i = len(s.rstrip('\r\n'))
Tim Peters88869f92001-01-14 23:36:06 +0000180 s = s[:5] + '*'*(i-5) + s[i:]
Walter Dörwald70a6b492004-02-12 17:35:32 +0000181 return repr(s)
Guido van Rossumebaf1041995-05-05 15:54:14 +0000182
Tim Peters88869f92001-01-14 23:36:06 +0000183 # Internal: send one line to the server, appending CRLF
184 def putline(self, line):
185 line = line + CRLF
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000186 if self.debugging > 1: print('*put*', self.sanitize(line))
Guido van Rossum52b89762007-07-17 20:45:57 +0000187 self.sock.sendall(line.encode(self.encoding))
Guido van Rossum1115ab21992-11-04 15:51:30 +0000188
Tim Peters88869f92001-01-14 23:36:06 +0000189 # Internal: send one command to the server (through putline())
190 def putcmd(self, line):
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000191 if self.debugging: print('*cmd*', self.sanitize(line))
Tim Peters88869f92001-01-14 23:36:06 +0000192 self.putline(line)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000193
Tim Peters88869f92001-01-14 23:36:06 +0000194 # Internal: return one line from the server, stripping CRLF.
195 # Raise EOFError if the connection is closed
196 def getline(self):
197 line = self.file.readline()
198 if self.debugging > 1:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000199 print('*get*', self.sanitize(line))
Tim Peters88869f92001-01-14 23:36:06 +0000200 if not line: raise EOFError
201 if line[-2:] == CRLF: line = line[:-2]
202 elif line[-1:] in CRLF: line = line[:-1]
203 return line
Guido van Rossum1115ab21992-11-04 15:51:30 +0000204
Tim Peters88869f92001-01-14 23:36:06 +0000205 # Internal: get a response from the server, which may possibly
206 # consist of multiple lines. Return a single string with no
207 # trailing CRLF. If the response consists of multiple lines,
208 # these are separated by '\n' characters in the string
209 def getmultiline(self):
210 line = self.getline()
211 if line[3:4] == '-':
212 code = line[:3]
213 while 1:
214 nextline = self.getline()
215 line = line + ('\n' + nextline)
216 if nextline[:3] == code and \
217 nextline[3:4] != '-':
218 break
219 return line
Guido van Rossum1115ab21992-11-04 15:51:30 +0000220
Tim Peters88869f92001-01-14 23:36:06 +0000221 # Internal: get a response from the server.
222 # Raise various errors if the response indicates an error
223 def getresp(self):
224 resp = self.getmultiline()
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000225 if self.debugging: print('*resp*', self.sanitize(resp))
Tim Peters88869f92001-01-14 23:36:06 +0000226 self.lastresp = resp[:3]
227 c = resp[:1]
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000228 if c in {'1', '2', '3'}:
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000229 return resp
Tim Peters88869f92001-01-14 23:36:06 +0000230 if c == '4':
Collin Winterce36ad82007-08-30 01:19:48 +0000231 raise error_temp(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000232 if c == '5':
Collin Winterce36ad82007-08-30 01:19:48 +0000233 raise error_perm(resp)
234 raise error_proto(resp)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000235
Tim Peters88869f92001-01-14 23:36:06 +0000236 def voidresp(self):
237 """Expect a response beginning with '2'."""
238 resp = self.getresp()
Benjamin Petersond23f8222009-04-05 19:13:16 +0000239 if resp[:1] != '2':
Collin Winterce36ad82007-08-30 01:19:48 +0000240 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000241 return resp
Guido van Rossumc567c601992-11-05 22:22:37 +0000242
Tim Peters88869f92001-01-14 23:36:06 +0000243 def abort(self):
244 '''Abort a file transfer. Uses out-of-band data.
245 This does not follow the procedure from the RFC to send Telnet
246 IP and Synch; that doesn't seem to work with the servers I've
247 tried. Instead, just send the ABOR command as OOB data.'''
Giampaolo Rodola'0b5c21f2011-05-07 19:03:47 +0200248 line = b'ABOR' + B_CRLF
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000249 if self.debugging > 1: print('*put urgent*', self.sanitize(line))
Martin v. Löwise12454f2002-02-16 23:06:19 +0000250 self.sock.sendall(line, MSG_OOB)
Tim Peters88869f92001-01-14 23:36:06 +0000251 resp = self.getmultiline()
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000252 if resp[:3] not in {'426', '225', '226'}:
Collin Winterce36ad82007-08-30 01:19:48 +0000253 raise error_proto(resp)
Giampaolo Rodola'0b5c21f2011-05-07 19:03:47 +0200254 return resp
Guido van Rossumd3166071993-05-24 14:16:22 +0000255
Tim Peters88869f92001-01-14 23:36:06 +0000256 def sendcmd(self, cmd):
257 '''Send a command and return the response.'''
258 self.putcmd(cmd)
259 return self.getresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000260
Tim Peters88869f92001-01-14 23:36:06 +0000261 def voidcmd(self, cmd):
262 """Send a command and expect a response beginning with '2'."""
263 self.putcmd(cmd)
264 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000265
Tim Peters88869f92001-01-14 23:36:06 +0000266 def sendport(self, host, port):
267 '''Send a PORT command with the current host and the given
268 port number.
269 '''
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000270 hbytes = host.split('.')
Benjamin Peterson3a53fbb2008-09-27 22:04:16 +0000271 pbytes = [repr(port//256), repr(port%256)]
Tim Peters88869f92001-01-14 23:36:06 +0000272 bytes = hbytes + pbytes
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000273 cmd = 'PORT ' + ','.join(bytes)
Tim Peters88869f92001-01-14 23:36:06 +0000274 return self.voidcmd(cmd)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000275
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000276 def sendeprt(self, host, port):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000277 '''Send a EPRT command with the current host and the given port number.'''
278 af = 0
279 if self.af == socket.AF_INET:
280 af = 1
281 if self.af == socket.AF_INET6:
282 af = 2
283 if af == 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000284 raise error_proto('unsupported address family')
Walter Dörwald70a6b492004-02-12 17:35:32 +0000285 fields = ['', repr(af), host, repr(port), '']
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000286 cmd = 'EPRT ' + '|'.join(fields)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000287 return self.voidcmd(cmd)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000288
Tim Peters88869f92001-01-14 23:36:06 +0000289 def makeport(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000290 '''Create a new socket and send a PORT command for it.'''
Giampaolo Rodola'fc327882012-12-17 14:20:27 +0100291 err = None
Martin v. Löwis322c0d12001-10-07 08:53:32 +0000292 sock = None
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000293 for res in socket.getaddrinfo(None, 0, self.af, socket.SOCK_STREAM, 0, socket.AI_PASSIVE):
294 af, socktype, proto, canonname, sa = res
295 try:
296 sock = socket.socket(af, socktype, proto)
297 sock.bind(sa)
Giampaolo Rodola'57f34872012-12-17 20:46:16 +0100298 except socket.error as _:
299 err = _
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
Giampaolo Rodola'fc327882012-12-17 14:20:27 +0100305 if sock is None:
306 if err is not None:
307 raise err
308 else:
309 raise socket.error("getaddrinfo returns an empty list")
Collin Winterce36ad82007-08-30 01:19:48 +0000310 raise socket.error(msg)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000311 sock.listen(1)
312 port = sock.getsockname()[1] # Get proper port
313 host = self.sock.getsockname()[0] # Get proper host
314 if self.af == socket.AF_INET:
315 resp = self.sendport(host, port)
316 else:
317 resp = self.sendeprt(host, port)
Giampaolo RodolĂ 5fb313b2010-04-19 22:05:54 +0000318 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
319 sock.settimeout(self.timeout)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000320 return sock
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000321
322 def makepasv(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000323 if self.af == socket.AF_INET:
324 host, port = parse227(self.sendcmd('PASV'))
325 else:
326 host, port = parse229(self.sendcmd('EPSV'), self.sock.getpeername())
327 return host, port
Guido van Rossum1115ab21992-11-04 15:51:30 +0000328
Tim Peters88869f92001-01-14 23:36:06 +0000329 def ntransfercmd(self, cmd, rest=None):
330 """Initiate a transfer over the data connection.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000331
Tim Peters88869f92001-01-14 23:36:06 +0000332 If the transfer is active, send a port command and the
333 transfer command, and accept the connection. If the server is
334 passive, send a pasv command, connect to it, and start the
335 transfer command. Either way, return the socket for the
336 connection and the expected size of the transfer. The
337 expected size may be None if it could not be determined.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000338
Tim Peters88869f92001-01-14 23:36:06 +0000339 Optional `rest' argument can be a string that is sent as the
Christian Heimes1af737c2008-01-23 08:24:23 +0000340 argument to a REST command. This is essentially a server
Tim Peters88869f92001-01-14 23:36:06 +0000341 marker used to tell the server to skip over any data up to the
342 given marker.
343 """
344 size = None
345 if self.passiveserver:
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000346 host, port = self.makepasv()
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000347 conn = socket.create_connection((host, port), self.timeout,
348 source_address=self.source_address)
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +0200349 try:
350 if rest is not None:
351 self.sendcmd("REST %s" % rest)
352 resp = self.sendcmd(cmd)
353 # Some servers apparently send a 200 reply to
354 # a LIST or STOR command, before the 150 reply
355 # (and way before the 226 reply). This seems to
356 # be in violation of the protocol (which only allows
357 # 1xx or error messages for LIST), so we just discard
358 # this response.
359 if resp[0] == '2':
360 resp = self.getresp()
361 if resp[0] != '1':
362 raise error_reply(resp)
363 except:
364 conn.close()
365 raise
Tim Peters88869f92001-01-14 23:36:06 +0000366 else:
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100367 with self.makeport() as sock:
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +0200368 if rest is not None:
369 self.sendcmd("REST %s" % rest)
370 resp = self.sendcmd(cmd)
371 # See above.
372 if resp[0] == '2':
373 resp = self.getresp()
374 if resp[0] != '1':
375 raise error_reply(resp)
376 conn, sockaddr = sock.accept()
377 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
378 conn.settimeout(self.timeout)
Tim Peters88869f92001-01-14 23:36:06 +0000379 if resp[:3] == '150':
380 # this is conditional in case we received a 125
381 size = parse150(resp)
382 return conn, size
Fred Drake4de02d91997-01-10 18:26:09 +0000383
Tim Peters88869f92001-01-14 23:36:06 +0000384 def transfercmd(self, cmd, rest=None):
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000385 """Like ntransfercmd() but returns only the socket."""
Tim Peters88869f92001-01-14 23:36:06 +0000386 return self.ntransfercmd(cmd, rest)[0]
Guido van Rossumc567c601992-11-05 22:22:37 +0000387
Tim Peters88869f92001-01-14 23:36:06 +0000388 def login(self, user = '', passwd = '', acct = ''):
389 '''Login, default anonymous.'''
390 if not user: user = 'anonymous'
391 if not passwd: passwd = ''
392 if not acct: acct = ''
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000393 if user == 'anonymous' and passwd in {'', '-'}:
Tim Peterse4418602002-02-16 07:34:19 +0000394 # If there is no anonymous ftp password specified
395 # then we'll just use anonymous@
396 # We don't send any other thing because:
397 # - We want to remain anonymous
398 # - We want to stop SPAM
399 # - We don't want to let ftp sites to discriminate by the user,
400 # host or country.
Guido van Rossumc33e0772001-12-28 20:54:28 +0000401 passwd = passwd + 'anonymous@'
Tim Peters88869f92001-01-14 23:36:06 +0000402 resp = self.sendcmd('USER ' + user)
403 if resp[0] == '3': resp = self.sendcmd('PASS ' + passwd)
404 if resp[0] == '3': resp = self.sendcmd('ACCT ' + acct)
405 if resp[0] != '2':
Collin Winterce36ad82007-08-30 01:19:48 +0000406 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000407 return resp
Guido van Rossumc567c601992-11-05 22:22:37 +0000408
Tim Peters88869f92001-01-14 23:36:06 +0000409 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000410 """Retrieve data in binary mode. A new port is created for you.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000411
Christian Heimes1af737c2008-01-23 08:24:23 +0000412 Args:
413 cmd: A RETR command.
414 callback: A single parameter callable to be called on each
415 block of data read.
416 blocksize: The maximum number of bytes to read from the
417 socket at one time. [default: 8192]
418 rest: Passed to transfercmd(). [default: None]
Guido van Rossum1115ab21992-11-04 15:51:30 +0000419
Christian Heimes1af737c2008-01-23 08:24:23 +0000420 Returns:
421 The response code.
Tim Peters88869f92001-01-14 23:36:06 +0000422 """
423 self.voidcmd('TYPE I')
Benjamin Petersona8071372010-10-31 18:21:16 +0000424 with self.transfercmd(cmd, rest) as conn:
425 while 1:
426 data = conn.recv(blocksize)
427 if not data:
428 break
429 callback(data)
Tim Peters88869f92001-01-14 23:36:06 +0000430 return self.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000431
Tim Peters88869f92001-01-14 23:36:06 +0000432 def retrlines(self, cmd, callback = None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000433 """Retrieve data in line mode. A new port is created for you.
434
435 Args:
Giampaolo Rodola'd78def92011-05-06 19:49:08 +0200436 cmd: A RETR, LIST, or NLST command.
Christian Heimes1af737c2008-01-23 08:24:23 +0000437 callback: An optional single parameter callable that is called
438 for each line with the trailing CRLF stripped.
439 [default: print_line()]
440
441 Returns:
442 The response code.
443 """
Raymond Hettingere874fc32002-05-12 05:53:51 +0000444 if callback is None: callback = print_line
Tim Peters88869f92001-01-14 23:36:06 +0000445 resp = self.sendcmd('TYPE A')
Benjamin Petersona8071372010-10-31 18:21:16 +0000446 with self.transfercmd(cmd) as conn, \
447 conn.makefile('r', encoding=self.encoding) as fp:
448 while 1:
449 line = fp.readline()
450 if self.debugging > 2: print('*retr*', repr(line))
451 if not line:
452 break
453 if line[-2:] == CRLF:
454 line = line[:-2]
455 elif line[-1:] == '\n':
456 line = line[:-1]
457 callback(line)
Tim Peters88869f92001-01-14 23:36:06 +0000458 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000459
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000460 def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000461 """Store a file in binary mode. A new port is created for you.
462
463 Args:
464 cmd: A STOR command.
465 fp: A file-like object with a read(num_bytes) method.
466 blocksize: The maximum data size to read from fp and send over
467 the connection at once. [default: 8192]
468 callback: An optional single parameter callable that is called on
469 on each block of data after it is sent. [default: None]
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000470 rest: Passed to transfercmd(). [default: None]
Christian Heimes1af737c2008-01-23 08:24:23 +0000471
472 Returns:
473 The response code.
474 """
Tim Peters88869f92001-01-14 23:36:06 +0000475 self.voidcmd('TYPE I')
Benjamin Petersona8071372010-10-31 18:21:16 +0000476 with self.transfercmd(cmd, rest) as conn:
477 while 1:
478 buf = fp.read(blocksize)
479 if not buf: break
480 conn.sendall(buf)
481 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000482 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000483
Christian Heimes1af737c2008-01-23 08:24:23 +0000484 def storlines(self, cmd, fp, callback=None):
485 """Store a file in line mode. A new port is created for you.
486
487 Args:
488 cmd: A STOR command.
489 fp: A file-like object with a readline() method.
490 callback: An optional single parameter callable that is called on
491 on each line after it is sent. [default: None]
492
493 Returns:
494 The response code.
495 """
Tim Peters88869f92001-01-14 23:36:06 +0000496 self.voidcmd('TYPE A')
Benjamin Petersona8071372010-10-31 18:21:16 +0000497 with self.transfercmd(cmd) as conn:
498 while 1:
499 buf = fp.readline()
500 if not buf: break
501 if buf[-2:] != B_CRLF:
502 if buf[-1] in B_CRLF: buf = buf[:-1]
503 buf = buf + B_CRLF
504 conn.sendall(buf)
505 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000506 return self.voidresp()
Guido van Rossum0eaa74b1996-01-25 18:37:21 +0000507
Tim Peters88869f92001-01-14 23:36:06 +0000508 def acct(self, password):
509 '''Send new account name.'''
510 cmd = 'ACCT ' + password
511 return self.voidcmd(cmd)
Guido van Rossumc567c601992-11-05 22:22:37 +0000512
Tim Peters88869f92001-01-14 23:36:06 +0000513 def nlst(self, *args):
514 '''Return a list of files in a given directory (default the current).'''
515 cmd = 'NLST'
516 for arg in args:
517 cmd = cmd + (' ' + arg)
518 files = []
519 self.retrlines(cmd, files.append)
520 return files
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000521
Tim Peters88869f92001-01-14 23:36:06 +0000522 def dir(self, *args):
523 '''List a directory in long form.
524 By default list current directory to stdout.
525 Optional last argument is callback function; all
526 non-empty arguments before it are concatenated to the
527 LIST command. (This *should* only be used for a pathname.)'''
528 cmd = 'LIST'
529 func = None
530 if args[-1:] and type(args[-1]) != type(''):
531 args, func = args[:-1], args[-1]
532 for arg in args:
533 if arg:
534 cmd = cmd + (' ' + arg)
535 self.retrlines(cmd, func)
Guido van Rossumc567c601992-11-05 22:22:37 +0000536
Giampaolo Rodola'd78def92011-05-06 19:49:08 +0200537 def mlsd(self, path="", facts=[]):
538 '''List a directory in a standardized format by using MLSD
539 command (RFC-3659). If path is omitted the current directory
540 is assumed. "facts" is a list of strings representing the type
541 of information desired (e.g. ["type", "size", "perm"]).
542
543 Return a generator object yielding a tuple of two elements
544 for every file found in path.
545 First element is the file name, the second one is a dictionary
546 including a variable number of "facts" depending on the server
547 and whether "facts" argument has been provided.
548 '''
549 if facts:
550 self.sendcmd("OPTS MLST " + ";".join(facts) + ";")
551 if path:
552 cmd = "MLSD %s" % path
553 else:
554 cmd = "MLSD"
555 lines = []
556 self.retrlines(cmd, lines.append)
557 for line in lines:
558 facts_found, _, name = line.rstrip(CRLF).partition(' ')
559 entry = {}
560 for fact in facts_found[:-1].split(";"):
561 key, _, value = fact.partition("=")
562 entry[key.lower()] = value
563 yield (name, entry)
564
Tim Peters88869f92001-01-14 23:36:06 +0000565 def rename(self, fromname, toname):
566 '''Rename a file.'''
567 resp = self.sendcmd('RNFR ' + fromname)
568 if resp[0] != '3':
Collin Winterce36ad82007-08-30 01:19:48 +0000569 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000570 return self.voidcmd('RNTO ' + toname)
Guido van Rossuma61bdeb1995-10-11 17:36:31 +0000571
Tim Peters88869f92001-01-14 23:36:06 +0000572 def delete(self, filename):
573 '''Delete a file.'''
574 resp = self.sendcmd('DELE ' + filename)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000575 if resp[:3] in {'250', '200'}:
Tim Peters88869f92001-01-14 23:36:06 +0000576 return resp
Tim Peters88869f92001-01-14 23:36:06 +0000577 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000578 raise error_reply(resp)
Guido van Rossum02cf5821993-05-17 08:00:02 +0000579
Tim Peters88869f92001-01-14 23:36:06 +0000580 def cwd(self, dirname):
581 '''Change to a directory.'''
582 if dirname == '..':
583 try:
584 return self.voidcmd('CDUP')
Guido van Rossumb940e112007-01-10 16:19:56 +0000585 except error_perm as msg:
Martin v. Löwisb5255112002-03-10 15:59:58 +0000586 if msg.args[0][:3] != '500':
587 raise
Tim Peters88869f92001-01-14 23:36:06 +0000588 elif dirname == '':
589 dirname = '.' # does nothing, but could return error
590 cmd = 'CWD ' + dirname
591 return self.voidcmd(cmd)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000592
Tim Peters88869f92001-01-14 23:36:06 +0000593 def size(self, filename):
594 '''Retrieve the size of a file.'''
Christian Heimes1af737c2008-01-23 08:24:23 +0000595 # The SIZE command is defined in RFC-3659
Tim Peters88869f92001-01-14 23:36:06 +0000596 resp = self.sendcmd('SIZE ' + filename)
597 if resp[:3] == '213':
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000598 s = resp[3:].strip()
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200599 return int(s)
Guido van Rossumc567c601992-11-05 22:22:37 +0000600
Tim Peters88869f92001-01-14 23:36:06 +0000601 def mkd(self, dirname):
602 '''Make a directory, return its full pathname.'''
Giampaolo RodolĂ bbc47822010-08-23 22:10:32 +0000603 resp = self.voidcmd('MKD ' + dirname)
604 # fix around non-compliant implementations such as IIS shipped
605 # with Windows server 2003
606 if not resp.startswith('257'):
607 return ''
Tim Peters88869f92001-01-14 23:36:06 +0000608 return parse257(resp)
Guido van Rossum98245091998-02-19 21:15:44 +0000609
Tim Peters88869f92001-01-14 23:36:06 +0000610 def rmd(self, dirname):
611 '''Remove a directory.'''
612 return self.voidcmd('RMD ' + dirname)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000613
Tim Peters88869f92001-01-14 23:36:06 +0000614 def pwd(self):
615 '''Return current working directory.'''
Giampaolo RodolĂ bbc47822010-08-23 22:10:32 +0000616 resp = self.voidcmd('PWD')
617 # fix around non-compliant implementations such as IIS shipped
618 # with Windows server 2003
619 if not resp.startswith('257'):
620 return ''
Tim Peters88869f92001-01-14 23:36:06 +0000621 return parse257(resp)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000622
Tim Peters88869f92001-01-14 23:36:06 +0000623 def quit(self):
624 '''Quit, and close the connection.'''
625 resp = self.voidcmd('QUIT')
626 self.close()
627 return resp
628
629 def close(self):
630 '''Close the connection without assuming anything about it.'''
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000631 if self.file is not None:
Fred Drake9c98a422001-02-28 21:46:37 +0000632 self.file.close()
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000633 if self.sock is not None:
Fred Drake9c98a422001-02-28 21:46:37 +0000634 self.sock.close()
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000635 self.file = self.sock = None
Guido van Rossumc567c601992-11-05 22:22:37 +0000636
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000637try:
638 import ssl
639except ImportError:
640 pass
641else:
642 class FTP_TLS(FTP):
643 '''A FTP subclass which adds TLS support to FTP as described
644 in RFC-4217.
645
646 Connect as usual to port 21 implicitly securing the FTP control
647 connection before authenticating.
648
649 Securing the data connection requires user to explicitly ask
650 for it by calling prot_p() method.
651
652 Usage example:
653 >>> from ftplib import FTP_TLS
654 >>> ftps = FTP_TLS('ftp.python.org')
Ezio Melotti4969f702011-03-15 05:59:46 +0200655 >>> ftps.login() # login anonymously previously securing control channel
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000656 '230 Guest login ok, access restrictions apply.'
657 >>> ftps.prot_p() # switch to secure data connection
658 '200 Protection level set to P'
659 >>> ftps.retrlines('LIST') # list directory content securely
660 total 9
661 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 .
662 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 ..
663 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 bin
664 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 etc
665 d-wxrwxr-x 2 ftp wheel 1024 Sep 5 13:43 incoming
666 drwxr-xr-x 2 root wheel 1024 Nov 17 1993 lib
667 drwxr-xr-x 6 1094 wheel 1024 Sep 13 19:07 pub
668 drwxr-xr-x 3 root wheel 1024 Jan 3 1994 usr
669 -rw-r--r-- 1 root root 312 Aug 1 1994 welcome.msg
670 '226 Transfer complete.'
671 >>> ftps.quit()
672 '221 Goodbye.'
673 >>>
674 '''
675 ssl_version = ssl.PROTOCOL_TLSv1
676
677 def __init__(self, host='', user='', passwd='', acct='', keyfile=None,
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000678 certfile=None, context=None,
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000679 timeout=_GLOBAL_DEFAULT_TIMEOUT, source_address=None):
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000680 if context is not None and keyfile is not None:
681 raise ValueError("context and keyfile arguments are mutually "
682 "exclusive")
683 if context is not None and certfile is not None:
684 raise ValueError("context and certfile arguments are mutually "
685 "exclusive")
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000686 self.keyfile = keyfile
687 self.certfile = certfile
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000688 self.context = context
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000689 self._prot_p = False
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000690 FTP.__init__(self, host, user, passwd, acct, timeout, source_address)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000691
692 def login(self, user='', passwd='', acct='', secure=True):
693 if secure and not isinstance(self.sock, ssl.SSLSocket):
694 self.auth()
695 return FTP.login(self, user, passwd, acct)
696
697 def auth(self):
698 '''Set up secure control connection by using TLS/SSL.'''
699 if isinstance(self.sock, ssl.SSLSocket):
700 raise ValueError("Already using TLS")
701 if self.ssl_version == ssl.PROTOCOL_TLSv1:
702 resp = self.voidcmd('AUTH TLS')
703 else:
704 resp = self.voidcmd('AUTH SSL')
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000705 if self.context is not None:
706 self.sock = self.context.wrap_socket(self.sock)
707 else:
708 self.sock = ssl.wrap_socket(self.sock, self.keyfile,
709 self.certfile,
710 ssl_version=self.ssl_version)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000711 self.file = self.sock.makefile(mode='r', encoding=self.encoding)
712 return resp
713
Giampaolo Rodola'096dcb12011-06-27 11:17:51 +0200714 def ccc(self):
715 '''Switch back to a clear-text control connection.'''
716 if not isinstance(self.sock, ssl.SSLSocket):
717 raise ValueError("not using TLS")
718 resp = self.voidcmd('CCC')
719 self.sock = self.sock.unwrap()
720 return resp
721
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000722 def prot_p(self):
723 '''Set up secure data connection.'''
724 # PROT defines whether or not the data channel is to be protected.
725 # Though RFC-2228 defines four possible protection levels,
726 # RFC-4217 only recommends two, Clear and Private.
727 # Clear (PROT C) means that no security is to be used on the
728 # data-channel, Private (PROT P) means that the data-channel
729 # should be protected by TLS.
730 # PBSZ command MUST still be issued, but must have a parameter of
731 # '0' to indicate that no buffering is taking place and the data
732 # connection should not be encapsulated.
733 self.voidcmd('PBSZ 0')
734 resp = self.voidcmd('PROT P')
735 self._prot_p = True
736 return resp
737
738 def prot_c(self):
739 '''Set up clear text data connection.'''
740 resp = self.voidcmd('PROT C')
741 self._prot_p = False
742 return resp
743
744 # --- Overridden FTP methods
745
746 def ntransfercmd(self, cmd, rest=None):
747 conn, size = FTP.ntransfercmd(self, cmd, rest)
748 if self._prot_p:
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000749 if self.context is not None:
750 conn = self.context.wrap_socket(conn)
751 else:
752 conn = ssl.wrap_socket(conn, self.keyfile, self.certfile,
753 ssl_version=self.ssl_version)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000754 return conn, size
755
756 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
757 self.voidcmd('TYPE I')
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100758 with self.transfercmd(cmd, rest) as conn:
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000759 while 1:
760 data = conn.recv(blocksize)
761 if not data:
762 break
763 callback(data)
764 # shutdown ssl layer
765 if isinstance(conn, ssl.SSLSocket):
766 conn.unwrap()
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000767 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)
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100774 with fp, conn:
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000775 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()
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000788 return self.voidresp()
789
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000790 def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000791 self.voidcmd('TYPE I')
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100792 with self.transfercmd(cmd, rest) as conn:
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000793 while 1:
794 buf = fp.read(blocksize)
795 if not buf: break
796 conn.sendall(buf)
797 if callback: callback(buf)
798 # shutdown ssl layer
799 if isinstance(conn, ssl.SSLSocket):
800 conn.unwrap()
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000801 return self.voidresp()
802
803 def storlines(self, cmd, fp, callback=None):
804 self.voidcmd('TYPE A')
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100805 with self.transfercmd(cmd) as conn:
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000806 while 1:
807 buf = fp.readline()
808 if not buf: break
Giampaolo RodolĂ f96482e2010-08-04 10:36:18 +0000809 if buf[-2:] != B_CRLF:
810 if buf[-1] in B_CRLF: buf = buf[:-1]
811 buf = buf + B_CRLF
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000812 conn.sendall(buf)
813 if callback: callback(buf)
814 # shutdown ssl layer
815 if isinstance(conn, ssl.SSLSocket):
816 conn.unwrap()
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000817 return self.voidresp()
818
Giampaolo Rodola'24befa82011-05-07 19:09:34 +0200819 def abort(self):
820 # overridden as we can't pass MSG_OOB flag to sendall()
821 line = b'ABOR' + B_CRLF
822 self.sock.sendall(line)
823 resp = self.getmultiline()
824 if resp[:3] not in {'426', '225', '226'}:
825 raise error_proto(resp)
826 return resp
827
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000828 __all__.append('FTP_TLS')
829 all_errors = (Error, IOError, EOFError, ssl.SSLError)
830
831
Guido van Rossumacfb82a1997-10-22 20:49:52 +0000832_150_re = None
Fred Drake4de02d91997-01-10 18:26:09 +0000833
834def parse150(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000835 '''Parse the '150' response for a RETR request.
836 Returns the expected transfer size or None; size is not guaranteed to
837 be present in the 150 message.
838 '''
839 if resp[:3] != '150':
Collin Winterce36ad82007-08-30 01:19:48 +0000840 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000841 global _150_re
842 if _150_re is None:
843 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000844 _150_re = re.compile(
845 "150 .* \((\d+) bytes\)", re.IGNORECASE | re.ASCII)
Tim Peters88869f92001-01-14 23:36:06 +0000846 m = _150_re.match(resp)
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000847 if not m:
848 return None
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200849 return int(m.group(1))
Fred Drake4de02d91997-01-10 18:26:09 +0000850
851
Guido van Rossum70297d32001-08-17 17:24:29 +0000852_227_re = None
853
Guido van Rossumd2560b01996-05-28 23:41:25 +0000854def parse227(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000855 '''Parse the '227' response for a PASV request.
856 Raises error_proto if it does not contain '(h1,h2,h3,h4,p1,p2)'
857 Return ('host.addr.as.numbers', port#) tuple.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000858
Tim Peters88869f92001-01-14 23:36:06 +0000859 if resp[:3] != '227':
Collin Winterce36ad82007-08-30 01:19:48 +0000860 raise error_reply(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000861 global _227_re
862 if _227_re is None:
863 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000864 _227_re = re.compile(r'(\d+),(\d+),(\d+),(\d+),(\d+),(\d+)', re.ASCII)
Guido van Rossum70297d32001-08-17 17:24:29 +0000865 m = _227_re.search(resp)
866 if not m:
Collin Winterce36ad82007-08-30 01:19:48 +0000867 raise error_proto(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000868 numbers = m.groups()
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000869 host = '.'.join(numbers[:4])
870 port = (int(numbers[4]) << 8) + int(numbers[5])
Tim Peters88869f92001-01-14 23:36:06 +0000871 return host, port
Guido van Rossumd2560b01996-05-28 23:41:25 +0000872
873
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000874def parse229(resp, peer):
875 '''Parse the '229' response for a EPSV request.
876 Raises error_proto if it does not contain '(|||port|)'
877 Return ('host.addr.as.numbers', port#) tuple.'''
878
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000879 if resp[:3] != '229':
Collin Winterce36ad82007-08-30 01:19:48 +0000880 raise error_reply(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000881 left = resp.find('(')
Collin Winterce36ad82007-08-30 01:19:48 +0000882 if left < 0: raise error_proto(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000883 right = resp.find(')', left + 1)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000884 if right < 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000885 raise error_proto(resp) # should contain '(|||port|)'
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000886 if resp[left + 1] != resp[right - 1]:
Collin Winterce36ad82007-08-30 01:19:48 +0000887 raise error_proto(resp)
Walter Dörwalda401ae42002-06-03 10:41:45 +0000888 parts = resp[left + 1:right].split(resp[left+1])
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000889 if len(parts) != 5:
Collin Winterce36ad82007-08-30 01:19:48 +0000890 raise error_proto(resp)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000891 host = peer[0]
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000892 port = int(parts[3])
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000893 return host, port
894
895
Guido van Rossumc567c601992-11-05 22:22:37 +0000896def parse257(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000897 '''Parse the '257' response for a MKD or PWD request.
898 This is a response to a MKD or PWD request: a directory name.
899 Returns the directoryname in the 257 reply.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000900
Tim Peters88869f92001-01-14 23:36:06 +0000901 if resp[:3] != '257':
Collin Winterce36ad82007-08-30 01:19:48 +0000902 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000903 if resp[3:5] != ' "':
904 return '' # Not compliant to RFC 959, but UNIX ftpd does this
905 dirname = ''
906 i = 5
907 n = len(resp)
908 while i < n:
909 c = resp[i]
910 i = i+1
911 if c == '"':
912 if i >= n or resp[i] != '"':
913 break
914 i = i+1
915 dirname = dirname + c
916 return dirname
Guido van Rossum1115ab21992-11-04 15:51:30 +0000917
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000918
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000919def print_line(line):
Tim Peters88869f92001-01-14 23:36:06 +0000920 '''Default retrlines callback to print a line.'''
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000921 print(line)
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000922
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000923
Guido van Rossumd2560b01996-05-28 23:41:25 +0000924def ftpcp(source, sourcename, target, targetname = '', type = 'I'):
Tim Peters88869f92001-01-14 23:36:06 +0000925 '''Copy file from one FTP-instance to another.'''
926 if not targetname: targetname = sourcename
927 type = 'TYPE ' + type
928 source.voidcmd(type)
929 target.voidcmd(type)
930 sourcehost, sourceport = parse227(source.sendcmd('PASV'))
931 target.sendport(sourcehost, sourceport)
932 # RFC 959: the user must "listen" [...] BEFORE sending the
933 # transfer request.
934 # So: STOR before RETR, because here the target is a "user".
935 treply = target.sendcmd('STOR ' + targetname)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000936 if treply[:3] not in {'125', '150'}: raise error_proto # RFC 959
Tim Peters88869f92001-01-14 23:36:06 +0000937 sreply = source.sendcmd('RETR ' + sourcename)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000938 if sreply[:3] not in {'125', '150'}: raise error_proto # RFC 959
Tim Peters88869f92001-01-14 23:36:06 +0000939 source.voidresp()
940 target.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000941
Tim Peters88869f92001-01-14 23:36:06 +0000942
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000943class Netrc:
Tim Peters88869f92001-01-14 23:36:06 +0000944 """Class to parse & provide access to 'netrc' format files.
Fred Drake475d51d1997-06-24 22:02:54 +0000945
Tim Peters88869f92001-01-14 23:36:06 +0000946 See the netrc(4) man page for information on the file format.
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000947
Tim Peters88869f92001-01-14 23:36:06 +0000948 WARNING: This class is obsolete -- use module netrc instead.
Guido van Rossumc822a451998-12-22 16:49:16 +0000949
Tim Peters88869f92001-01-14 23:36:06 +0000950 """
951 __defuser = None
952 __defpasswd = None
953 __defacct = None
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000954
Tim Peters88869f92001-01-14 23:36:06 +0000955 def __init__(self, filename=None):
Raymond Hettinger094662a2002-06-01 01:29:16 +0000956 if filename is None:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000957 if "HOME" in os.environ:
Tim Peters88869f92001-01-14 23:36:06 +0000958 filename = os.path.join(os.environ["HOME"],
959 ".netrc")
960 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000961 raise IOError("specify file to load or set $HOME")
Tim Peters88869f92001-01-14 23:36:06 +0000962 self.__hosts = {}
963 self.__macros = {}
964 fp = open(filename, "r")
965 in_macro = 0
966 while 1:
967 line = fp.readline()
968 if not line: break
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000969 if in_macro and line.strip():
Tim Peters88869f92001-01-14 23:36:06 +0000970 macro_lines.append(line)
971 continue
972 elif in_macro:
973 self.__macros[macro_name] = tuple(macro_lines)
974 in_macro = 0
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000975 words = line.split()
Tim Peters88869f92001-01-14 23:36:06 +0000976 host = user = passwd = acct = None
977 default = 0
978 i = 0
979 while i < len(words):
980 w1 = words[i]
981 if i+1 < len(words):
982 w2 = words[i + 1]
983 else:
984 w2 = None
985 if w1 == 'default':
986 default = 1
987 elif w1 == 'machine' and w2:
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000988 host = w2.lower()
Tim Peters88869f92001-01-14 23:36:06 +0000989 i = i + 1
990 elif w1 == 'login' and w2:
991 user = w2
992 i = i + 1
993 elif w1 == 'password' and w2:
994 passwd = w2
995 i = i + 1
996 elif w1 == 'account' and w2:
997 acct = w2
998 i = i + 1
999 elif w1 == 'macdef' and w2:
1000 macro_name = w2
1001 macro_lines = []
1002 in_macro = 1
1003 break
1004 i = i + 1
1005 if default:
1006 self.__defuser = user or self.__defuser
1007 self.__defpasswd = passwd or self.__defpasswd
1008 self.__defacct = acct or self.__defacct
1009 if host:
Raymond Hettinger54f02222002-06-01 14:18:47 +00001010 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +00001011 ouser, opasswd, oacct = \
1012 self.__hosts[host]
1013 user = user or ouser
1014 passwd = passwd or opasswd
1015 acct = acct or oacct
1016 self.__hosts[host] = user, passwd, acct
1017 fp.close()
Guido van Rossum56d1e3a1997-03-14 04:16:54 +00001018
Tim Peters88869f92001-01-14 23:36:06 +00001019 def get_hosts(self):
1020 """Return a list of hosts mentioned in the .netrc file."""
1021 return self.__hosts.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +00001022
Tim Peters88869f92001-01-14 23:36:06 +00001023 def get_account(self, host):
1024 """Returns login information for the named host.
Guido van Rossum8ca84201998-03-26 20:56:10 +00001025
Tim Peters88869f92001-01-14 23:36:06 +00001026 The return value is a triple containing userid,
1027 password, and the accounting field.
Guido van Rossum8ca84201998-03-26 20:56:10 +00001028
Tim Peters88869f92001-01-14 23:36:06 +00001029 """
Eric S. Raymondc95bf692001-02-09 10:06:47 +00001030 host = host.lower()
Tim Peters88869f92001-01-14 23:36:06 +00001031 user = passwd = acct = None
Raymond Hettinger54f02222002-06-01 14:18:47 +00001032 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +00001033 user, passwd, acct = self.__hosts[host]
1034 user = user or self.__defuser
1035 passwd = passwd or self.__defpasswd
1036 acct = acct or self.__defacct
1037 return user, passwd, acct
Guido van Rossum8ca84201998-03-26 20:56:10 +00001038
Tim Peters88869f92001-01-14 23:36:06 +00001039 def get_macros(self):
1040 """Return a list of all defined macro names."""
1041 return self.__macros.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +00001042
Tim Peters88869f92001-01-14 23:36:06 +00001043 def get_macro(self, macro):
1044 """Return a sequence of lines which define a named macro."""
1045 return self.__macros[macro]
Guido van Rossum56d1e3a1997-03-14 04:16:54 +00001046
Fred Drake475d51d1997-06-24 22:02:54 +00001047
Tim Peters88869f92001-01-14 23:36:06 +00001048
Guido van Rossum1115ab21992-11-04 15:51:30 +00001049def test():
Tim Peters88869f92001-01-14 23:36:06 +00001050 '''Test program.
Raymond Hettingerc88a6c72005-04-05 04:31:09 +00001051 Usage: ftp [-d] [-r[file]] host [-l[dir]] [-d[dir]] [-p] [file] ...
1052
1053 -d dir
1054 -l list
1055 -p password
1056 '''
1057
1058 if len(sys.argv) < 2:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001059 print(test.__doc__)
Raymond Hettingerc88a6c72005-04-05 04:31:09 +00001060 sys.exit(0)
Guido van Rossumd2560b01996-05-28 23:41:25 +00001061
Tim Peters88869f92001-01-14 23:36:06 +00001062 debugging = 0
1063 rcfile = None
1064 while sys.argv[1] == '-d':
1065 debugging = debugging+1
1066 del sys.argv[1]
1067 if sys.argv[1][:2] == '-r':
1068 # get name of alternate ~/.netrc file:
1069 rcfile = sys.argv[1][2:]
1070 del sys.argv[1]
1071 host = sys.argv[1]
1072 ftp = FTP(host)
1073 ftp.set_debuglevel(debugging)
1074 userid = passwd = acct = ''
1075 try:
1076 netrc = Netrc(rcfile)
1077 except IOError:
1078 if rcfile is not None:
1079 sys.stderr.write("Could not open account file"
1080 " -- using anonymous login.")
1081 else:
1082 try:
1083 userid, passwd, acct = netrc.get_account(host)
1084 except KeyError:
1085 # no account for host
1086 sys.stderr.write(
1087 "No account -- using anonymous login.")
1088 ftp.login(userid, passwd, acct)
1089 for file in sys.argv[2:]:
1090 if file[:2] == '-l':
1091 ftp.dir(file[2:])
1092 elif file[:2] == '-d':
1093 cmd = 'CWD'
1094 if file[2:]: cmd = cmd + ' ' + file[2:]
1095 resp = ftp.sendcmd(cmd)
1096 elif file == '-p':
1097 ftp.set_pasv(not ftp.passiveserver)
1098 else:
1099 ftp.retrbinary('RETR ' + file, \
1100 sys.stdout.write, 1024)
1101 ftp.quit()
Guido van Rossum221ec0b1995-08-04 04:39:30 +00001102
1103
1104if __name__ == '__main__':
Tim Peters88869f92001-01-14 23:36:06 +00001105 test()