blob: 490f900708f3eb77f7118dedf7eab29761272bf9 [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'fc327882012-12-17 14:20:27 +0100298 except socket.error as err:
Martin v. Löwis322c0d12001-10-07 08:53:32 +0000299 if sock:
300 sock.close()
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000301 sock = None
302 continue
303 break
Giampaolo Rodola'fc327882012-12-17 14:20:27 +0100304 if sock is None:
305 if err is not None:
306 raise err
307 else:
308 raise socket.error("getaddrinfo returns an empty list")
Collin Winterce36ad82007-08-30 01:19:48 +0000309 raise socket.error(msg)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000310 sock.listen(1)
311 port = sock.getsockname()[1] # Get proper port
312 host = self.sock.getsockname()[0] # Get proper host
313 if self.af == socket.AF_INET:
314 resp = self.sendport(host, port)
315 else:
316 resp = self.sendeprt(host, port)
Giampaolo RodolĂ 5fb313b2010-04-19 22:05:54 +0000317 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
318 sock.settimeout(self.timeout)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000319 return sock
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000320
321 def makepasv(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000322 if self.af == socket.AF_INET:
323 host, port = parse227(self.sendcmd('PASV'))
324 else:
325 host, port = parse229(self.sendcmd('EPSV'), self.sock.getpeername())
326 return host, port
Guido van Rossum1115ab21992-11-04 15:51:30 +0000327
Tim Peters88869f92001-01-14 23:36:06 +0000328 def ntransfercmd(self, cmd, rest=None):
329 """Initiate a transfer over the data connection.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000330
Tim Peters88869f92001-01-14 23:36:06 +0000331 If the transfer is active, send a port command and the
332 transfer command, and accept the connection. If the server is
333 passive, send a pasv command, connect to it, and start the
334 transfer command. Either way, return the socket for the
335 connection and the expected size of the transfer. The
336 expected size may be None if it could not be determined.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000337
Tim Peters88869f92001-01-14 23:36:06 +0000338 Optional `rest' argument can be a string that is sent as the
Christian Heimes1af737c2008-01-23 08:24:23 +0000339 argument to a REST command. This is essentially a server
Tim Peters88869f92001-01-14 23:36:06 +0000340 marker used to tell the server to skip over any data up to the
341 given marker.
342 """
343 size = None
344 if self.passiveserver:
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000345 host, port = self.makepasv()
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000346 conn = socket.create_connection((host, port), self.timeout,
347 source_address=self.source_address)
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +0200348 try:
349 if rest is not None:
350 self.sendcmd("REST %s" % rest)
351 resp = self.sendcmd(cmd)
352 # Some servers apparently send a 200 reply to
353 # a LIST or STOR command, before the 150 reply
354 # (and way before the 226 reply). This seems to
355 # be in violation of the protocol (which only allows
356 # 1xx or error messages for LIST), so we just discard
357 # this response.
358 if resp[0] == '2':
359 resp = self.getresp()
360 if resp[0] != '1':
361 raise error_reply(resp)
362 except:
363 conn.close()
364 raise
Tim Peters88869f92001-01-14 23:36:06 +0000365 else:
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100366 with self.makeport() as sock:
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +0200367 if rest is not None:
368 self.sendcmd("REST %s" % rest)
369 resp = self.sendcmd(cmd)
370 # See above.
371 if resp[0] == '2':
372 resp = self.getresp()
373 if resp[0] != '1':
374 raise error_reply(resp)
375 conn, sockaddr = sock.accept()
376 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
377 conn.settimeout(self.timeout)
Tim Peters88869f92001-01-14 23:36:06 +0000378 if resp[:3] == '150':
379 # this is conditional in case we received a 125
380 size = parse150(resp)
381 return conn, size
Fred Drake4de02d91997-01-10 18:26:09 +0000382
Tim Peters88869f92001-01-14 23:36:06 +0000383 def transfercmd(self, cmd, rest=None):
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000384 """Like ntransfercmd() but returns only the socket."""
Tim Peters88869f92001-01-14 23:36:06 +0000385 return self.ntransfercmd(cmd, rest)[0]
Guido van Rossumc567c601992-11-05 22:22:37 +0000386
Tim Peters88869f92001-01-14 23:36:06 +0000387 def login(self, user = '', passwd = '', acct = ''):
388 '''Login, default anonymous.'''
389 if not user: user = 'anonymous'
390 if not passwd: passwd = ''
391 if not acct: acct = ''
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000392 if user == 'anonymous' and passwd in {'', '-'}:
Tim Peterse4418602002-02-16 07:34:19 +0000393 # If there is no anonymous ftp password specified
394 # then we'll just use anonymous@
395 # We don't send any other thing because:
396 # - We want to remain anonymous
397 # - We want to stop SPAM
398 # - We don't want to let ftp sites to discriminate by the user,
399 # host or country.
Guido van Rossumc33e0772001-12-28 20:54:28 +0000400 passwd = passwd + 'anonymous@'
Tim Peters88869f92001-01-14 23:36:06 +0000401 resp = self.sendcmd('USER ' + user)
402 if resp[0] == '3': resp = self.sendcmd('PASS ' + passwd)
403 if resp[0] == '3': resp = self.sendcmd('ACCT ' + acct)
404 if resp[0] != '2':
Collin Winterce36ad82007-08-30 01:19:48 +0000405 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000406 return resp
Guido van Rossumc567c601992-11-05 22:22:37 +0000407
Tim Peters88869f92001-01-14 23:36:06 +0000408 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000409 """Retrieve data in binary mode. A new port is created for you.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000410
Christian Heimes1af737c2008-01-23 08:24:23 +0000411 Args:
412 cmd: A RETR command.
413 callback: A single parameter callable to be called on each
414 block of data read.
415 blocksize: The maximum number of bytes to read from the
416 socket at one time. [default: 8192]
417 rest: Passed to transfercmd(). [default: None]
Guido van Rossum1115ab21992-11-04 15:51:30 +0000418
Christian Heimes1af737c2008-01-23 08:24:23 +0000419 Returns:
420 The response code.
Tim Peters88869f92001-01-14 23:36:06 +0000421 """
422 self.voidcmd('TYPE I')
Benjamin Petersona8071372010-10-31 18:21:16 +0000423 with self.transfercmd(cmd, rest) as conn:
424 while 1:
425 data = conn.recv(blocksize)
426 if not data:
427 break
428 callback(data)
Tim Peters88869f92001-01-14 23:36:06 +0000429 return self.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000430
Tim Peters88869f92001-01-14 23:36:06 +0000431 def retrlines(self, cmd, callback = None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000432 """Retrieve data in line mode. A new port is created for you.
433
434 Args:
Giampaolo Rodola'd78def92011-05-06 19:49:08 +0200435 cmd: A RETR, LIST, or NLST command.
Christian Heimes1af737c2008-01-23 08:24:23 +0000436 callback: An optional single parameter callable that is called
437 for each line with the trailing CRLF stripped.
438 [default: print_line()]
439
440 Returns:
441 The response code.
442 """
Raymond Hettingere874fc32002-05-12 05:53:51 +0000443 if callback is None: callback = print_line
Tim Peters88869f92001-01-14 23:36:06 +0000444 resp = self.sendcmd('TYPE A')
Benjamin Petersona8071372010-10-31 18:21:16 +0000445 with self.transfercmd(cmd) as conn, \
446 conn.makefile('r', encoding=self.encoding) as fp:
447 while 1:
448 line = fp.readline()
449 if self.debugging > 2: print('*retr*', repr(line))
450 if not line:
451 break
452 if line[-2:] == CRLF:
453 line = line[:-2]
454 elif line[-1:] == '\n':
455 line = line[:-1]
456 callback(line)
Tim Peters88869f92001-01-14 23:36:06 +0000457 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000458
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000459 def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000460 """Store a file in binary mode. A new port is created for you.
461
462 Args:
463 cmd: A STOR command.
464 fp: A file-like object with a read(num_bytes) method.
465 blocksize: The maximum data size to read from fp and send over
466 the connection at once. [default: 8192]
467 callback: An optional single parameter callable that is called on
468 on each block of data after it is sent. [default: None]
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000469 rest: Passed to transfercmd(). [default: None]
Christian Heimes1af737c2008-01-23 08:24:23 +0000470
471 Returns:
472 The response code.
473 """
Tim Peters88869f92001-01-14 23:36:06 +0000474 self.voidcmd('TYPE I')
Benjamin Petersona8071372010-10-31 18:21:16 +0000475 with self.transfercmd(cmd, rest) as conn:
476 while 1:
477 buf = fp.read(blocksize)
478 if not buf: break
479 conn.sendall(buf)
480 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000481 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000482
Christian Heimes1af737c2008-01-23 08:24:23 +0000483 def storlines(self, cmd, fp, callback=None):
484 """Store a file in line mode. A new port is created for you.
485
486 Args:
487 cmd: A STOR command.
488 fp: A file-like object with a readline() method.
489 callback: An optional single parameter callable that is called on
490 on each line after it is sent. [default: None]
491
492 Returns:
493 The response code.
494 """
Tim Peters88869f92001-01-14 23:36:06 +0000495 self.voidcmd('TYPE A')
Benjamin Petersona8071372010-10-31 18:21:16 +0000496 with self.transfercmd(cmd) as conn:
497 while 1:
498 buf = fp.readline()
499 if not buf: break
500 if buf[-2:] != B_CRLF:
501 if buf[-1] in B_CRLF: buf = buf[:-1]
502 buf = buf + B_CRLF
503 conn.sendall(buf)
504 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000505 return self.voidresp()
Guido van Rossum0eaa74b1996-01-25 18:37:21 +0000506
Tim Peters88869f92001-01-14 23:36:06 +0000507 def acct(self, password):
508 '''Send new account name.'''
509 cmd = 'ACCT ' + password
510 return self.voidcmd(cmd)
Guido van Rossumc567c601992-11-05 22:22:37 +0000511
Tim Peters88869f92001-01-14 23:36:06 +0000512 def nlst(self, *args):
513 '''Return a list of files in a given directory (default the current).'''
514 cmd = 'NLST'
515 for arg in args:
516 cmd = cmd + (' ' + arg)
517 files = []
518 self.retrlines(cmd, files.append)
519 return files
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000520
Tim Peters88869f92001-01-14 23:36:06 +0000521 def dir(self, *args):
522 '''List a directory in long form.
523 By default list current directory to stdout.
524 Optional last argument is callback function; all
525 non-empty arguments before it are concatenated to the
526 LIST command. (This *should* only be used for a pathname.)'''
527 cmd = 'LIST'
528 func = None
529 if args[-1:] and type(args[-1]) != type(''):
530 args, func = args[:-1], args[-1]
531 for arg in args:
532 if arg:
533 cmd = cmd + (' ' + arg)
534 self.retrlines(cmd, func)
Guido van Rossumc567c601992-11-05 22:22:37 +0000535
Giampaolo Rodola'd78def92011-05-06 19:49:08 +0200536 def mlsd(self, path="", facts=[]):
537 '''List a directory in a standardized format by using MLSD
538 command (RFC-3659). If path is omitted the current directory
539 is assumed. "facts" is a list of strings representing the type
540 of information desired (e.g. ["type", "size", "perm"]).
541
542 Return a generator object yielding a tuple of two elements
543 for every file found in path.
544 First element is the file name, the second one is a dictionary
545 including a variable number of "facts" depending on the server
546 and whether "facts" argument has been provided.
547 '''
548 if facts:
549 self.sendcmd("OPTS MLST " + ";".join(facts) + ";")
550 if path:
551 cmd = "MLSD %s" % path
552 else:
553 cmd = "MLSD"
554 lines = []
555 self.retrlines(cmd, lines.append)
556 for line in lines:
557 facts_found, _, name = line.rstrip(CRLF).partition(' ')
558 entry = {}
559 for fact in facts_found[:-1].split(";"):
560 key, _, value = fact.partition("=")
561 entry[key.lower()] = value
562 yield (name, entry)
563
Tim Peters88869f92001-01-14 23:36:06 +0000564 def rename(self, fromname, toname):
565 '''Rename a file.'''
566 resp = self.sendcmd('RNFR ' + fromname)
567 if resp[0] != '3':
Collin Winterce36ad82007-08-30 01:19:48 +0000568 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000569 return self.voidcmd('RNTO ' + toname)
Guido van Rossuma61bdeb1995-10-11 17:36:31 +0000570
Tim Peters88869f92001-01-14 23:36:06 +0000571 def delete(self, filename):
572 '''Delete a file.'''
573 resp = self.sendcmd('DELE ' + filename)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000574 if resp[:3] in {'250', '200'}:
Tim Peters88869f92001-01-14 23:36:06 +0000575 return resp
Tim Peters88869f92001-01-14 23:36:06 +0000576 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000577 raise error_reply(resp)
Guido van Rossum02cf5821993-05-17 08:00:02 +0000578
Tim Peters88869f92001-01-14 23:36:06 +0000579 def cwd(self, dirname):
580 '''Change to a directory.'''
581 if dirname == '..':
582 try:
583 return self.voidcmd('CDUP')
Guido van Rossumb940e112007-01-10 16:19:56 +0000584 except error_perm as msg:
Martin v. Löwisb5255112002-03-10 15:59:58 +0000585 if msg.args[0][:3] != '500':
586 raise
Tim Peters88869f92001-01-14 23:36:06 +0000587 elif dirname == '':
588 dirname = '.' # does nothing, but could return error
589 cmd = 'CWD ' + dirname
590 return self.voidcmd(cmd)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000591
Tim Peters88869f92001-01-14 23:36:06 +0000592 def size(self, filename):
593 '''Retrieve the size of a file.'''
Christian Heimes1af737c2008-01-23 08:24:23 +0000594 # The SIZE command is defined in RFC-3659
Tim Peters88869f92001-01-14 23:36:06 +0000595 resp = self.sendcmd('SIZE ' + filename)
596 if resp[:3] == '213':
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000597 s = resp[3:].strip()
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200598 return int(s)
Guido van Rossumc567c601992-11-05 22:22:37 +0000599
Tim Peters88869f92001-01-14 23:36:06 +0000600 def mkd(self, dirname):
601 '''Make a directory, return its full pathname.'''
Giampaolo RodolĂ bbc47822010-08-23 22:10:32 +0000602 resp = self.voidcmd('MKD ' + dirname)
603 # fix around non-compliant implementations such as IIS shipped
604 # with Windows server 2003
605 if not resp.startswith('257'):
606 return ''
Tim Peters88869f92001-01-14 23:36:06 +0000607 return parse257(resp)
Guido van Rossum98245091998-02-19 21:15:44 +0000608
Tim Peters88869f92001-01-14 23:36:06 +0000609 def rmd(self, dirname):
610 '''Remove a directory.'''
611 return self.voidcmd('RMD ' + dirname)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000612
Tim Peters88869f92001-01-14 23:36:06 +0000613 def pwd(self):
614 '''Return current working directory.'''
Giampaolo RodolĂ bbc47822010-08-23 22:10:32 +0000615 resp = self.voidcmd('PWD')
616 # fix around non-compliant implementations such as IIS shipped
617 # with Windows server 2003
618 if not resp.startswith('257'):
619 return ''
Tim Peters88869f92001-01-14 23:36:06 +0000620 return parse257(resp)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000621
Tim Peters88869f92001-01-14 23:36:06 +0000622 def quit(self):
623 '''Quit, and close the connection.'''
624 resp = self.voidcmd('QUIT')
625 self.close()
626 return resp
627
628 def close(self):
629 '''Close the connection without assuming anything about it.'''
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000630 if self.file is not None:
Fred Drake9c98a422001-02-28 21:46:37 +0000631 self.file.close()
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000632 if self.sock is not None:
Fred Drake9c98a422001-02-28 21:46:37 +0000633 self.sock.close()
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000634 self.file = self.sock = None
Guido van Rossumc567c601992-11-05 22:22:37 +0000635
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000636try:
637 import ssl
638except ImportError:
639 pass
640else:
641 class FTP_TLS(FTP):
642 '''A FTP subclass which adds TLS support to FTP as described
643 in RFC-4217.
644
645 Connect as usual to port 21 implicitly securing the FTP control
646 connection before authenticating.
647
648 Securing the data connection requires user to explicitly ask
649 for it by calling prot_p() method.
650
651 Usage example:
652 >>> from ftplib import FTP_TLS
653 >>> ftps = FTP_TLS('ftp.python.org')
Ezio Melotti4969f702011-03-15 05:59:46 +0200654 >>> ftps.login() # login anonymously previously securing control channel
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000655 '230 Guest login ok, access restrictions apply.'
656 >>> ftps.prot_p() # switch to secure data connection
657 '200 Protection level set to P'
658 >>> ftps.retrlines('LIST') # list directory content securely
659 total 9
660 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 .
661 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 ..
662 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 bin
663 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 etc
664 d-wxrwxr-x 2 ftp wheel 1024 Sep 5 13:43 incoming
665 drwxr-xr-x 2 root wheel 1024 Nov 17 1993 lib
666 drwxr-xr-x 6 1094 wheel 1024 Sep 13 19:07 pub
667 drwxr-xr-x 3 root wheel 1024 Jan 3 1994 usr
668 -rw-r--r-- 1 root root 312 Aug 1 1994 welcome.msg
669 '226 Transfer complete.'
670 >>> ftps.quit()
671 '221 Goodbye.'
672 >>>
673 '''
674 ssl_version = ssl.PROTOCOL_TLSv1
675
676 def __init__(self, host='', user='', passwd='', acct='', keyfile=None,
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000677 certfile=None, context=None,
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000678 timeout=_GLOBAL_DEFAULT_TIMEOUT, source_address=None):
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000679 if context is not None and keyfile is not None:
680 raise ValueError("context and keyfile arguments are mutually "
681 "exclusive")
682 if context is not None and certfile is not None:
683 raise ValueError("context and certfile arguments are mutually "
684 "exclusive")
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000685 self.keyfile = keyfile
686 self.certfile = certfile
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000687 self.context = context
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000688 self._prot_p = False
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000689 FTP.__init__(self, host, user, passwd, acct, timeout, source_address)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000690
691 def login(self, user='', passwd='', acct='', secure=True):
692 if secure and not isinstance(self.sock, ssl.SSLSocket):
693 self.auth()
694 return FTP.login(self, user, passwd, acct)
695
696 def auth(self):
697 '''Set up secure control connection by using TLS/SSL.'''
698 if isinstance(self.sock, ssl.SSLSocket):
699 raise ValueError("Already using TLS")
700 if self.ssl_version == ssl.PROTOCOL_TLSv1:
701 resp = self.voidcmd('AUTH TLS')
702 else:
703 resp = self.voidcmd('AUTH SSL')
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000704 if self.context is not None:
705 self.sock = self.context.wrap_socket(self.sock)
706 else:
707 self.sock = ssl.wrap_socket(self.sock, self.keyfile,
708 self.certfile,
709 ssl_version=self.ssl_version)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000710 self.file = self.sock.makefile(mode='r', encoding=self.encoding)
711 return resp
712
Giampaolo Rodola'096dcb12011-06-27 11:17:51 +0200713 def ccc(self):
714 '''Switch back to a clear-text control connection.'''
715 if not isinstance(self.sock, ssl.SSLSocket):
716 raise ValueError("not using TLS")
717 resp = self.voidcmd('CCC')
718 self.sock = self.sock.unwrap()
719 return resp
720
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000721 def prot_p(self):
722 '''Set up secure data connection.'''
723 # PROT defines whether or not the data channel is to be protected.
724 # Though RFC-2228 defines four possible protection levels,
725 # RFC-4217 only recommends two, Clear and Private.
726 # Clear (PROT C) means that no security is to be used on the
727 # data-channel, Private (PROT P) means that the data-channel
728 # should be protected by TLS.
729 # PBSZ command MUST still be issued, but must have a parameter of
730 # '0' to indicate that no buffering is taking place and the data
731 # connection should not be encapsulated.
732 self.voidcmd('PBSZ 0')
733 resp = self.voidcmd('PROT P')
734 self._prot_p = True
735 return resp
736
737 def prot_c(self):
738 '''Set up clear text data connection.'''
739 resp = self.voidcmd('PROT C')
740 self._prot_p = False
741 return resp
742
743 # --- Overridden FTP methods
744
745 def ntransfercmd(self, cmd, rest=None):
746 conn, size = FTP.ntransfercmd(self, cmd, rest)
747 if self._prot_p:
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000748 if self.context is not None:
749 conn = self.context.wrap_socket(conn)
750 else:
751 conn = ssl.wrap_socket(conn, self.keyfile, self.certfile,
752 ssl_version=self.ssl_version)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000753 return conn, size
754
755 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
756 self.voidcmd('TYPE I')
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100757 with self.transfercmd(cmd, rest) as conn:
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000758 while 1:
759 data = conn.recv(blocksize)
760 if not data:
761 break
762 callback(data)
763 # shutdown ssl layer
764 if isinstance(conn, ssl.SSLSocket):
765 conn.unwrap()
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000766 return self.voidresp()
767
768 def retrlines(self, cmd, callback = None):
769 if callback is None: callback = print_line
770 resp = self.sendcmd('TYPE A')
771 conn = self.transfercmd(cmd)
772 fp = conn.makefile('r', encoding=self.encoding)
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100773 with fp, conn:
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000774 while 1:
775 line = fp.readline()
776 if self.debugging > 2: print('*retr*', repr(line))
777 if not line:
778 break
779 if line[-2:] == CRLF:
780 line = line[:-2]
781 elif line[-1:] == '\n':
782 line = line[:-1]
783 callback(line)
784 # shutdown ssl layer
785 if isinstance(conn, ssl.SSLSocket):
786 conn.unwrap()
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000787 return self.voidresp()
788
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000789 def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000790 self.voidcmd('TYPE I')
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100791 with self.transfercmd(cmd, rest) as conn:
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000792 while 1:
793 buf = fp.read(blocksize)
794 if not buf: break
795 conn.sendall(buf)
796 if callback: callback(buf)
797 # shutdown ssl layer
798 if isinstance(conn, ssl.SSLSocket):
799 conn.unwrap()
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000800 return self.voidresp()
801
802 def storlines(self, cmd, fp, callback=None):
803 self.voidcmd('TYPE A')
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100804 with self.transfercmd(cmd) as conn:
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000805 while 1:
806 buf = fp.readline()
807 if not buf: break
Giampaolo RodolĂ f96482e2010-08-04 10:36:18 +0000808 if buf[-2:] != B_CRLF:
809 if buf[-1] in B_CRLF: buf = buf[:-1]
810 buf = buf + B_CRLF
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000811 conn.sendall(buf)
812 if callback: callback(buf)
813 # shutdown ssl layer
814 if isinstance(conn, ssl.SSLSocket):
815 conn.unwrap()
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000816 return self.voidresp()
817
Giampaolo Rodola'24befa82011-05-07 19:09:34 +0200818 def abort(self):
819 # overridden as we can't pass MSG_OOB flag to sendall()
820 line = b'ABOR' + B_CRLF
821 self.sock.sendall(line)
822 resp = self.getmultiline()
823 if resp[:3] not in {'426', '225', '226'}:
824 raise error_proto(resp)
825 return resp
826
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000827 __all__.append('FTP_TLS')
828 all_errors = (Error, IOError, EOFError, ssl.SSLError)
829
830
Guido van Rossumacfb82a1997-10-22 20:49:52 +0000831_150_re = None
Fred Drake4de02d91997-01-10 18:26:09 +0000832
833def parse150(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000834 '''Parse the '150' response for a RETR request.
835 Returns the expected transfer size or None; size is not guaranteed to
836 be present in the 150 message.
837 '''
838 if resp[:3] != '150':
Collin Winterce36ad82007-08-30 01:19:48 +0000839 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000840 global _150_re
841 if _150_re is None:
842 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000843 _150_re = re.compile(
844 "150 .* \((\d+) bytes\)", re.IGNORECASE | re.ASCII)
Tim Peters88869f92001-01-14 23:36:06 +0000845 m = _150_re.match(resp)
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000846 if not m:
847 return None
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200848 return int(m.group(1))
Fred Drake4de02d91997-01-10 18:26:09 +0000849
850
Guido van Rossum70297d32001-08-17 17:24:29 +0000851_227_re = None
852
Guido van Rossumd2560b01996-05-28 23:41:25 +0000853def parse227(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000854 '''Parse the '227' response for a PASV request.
855 Raises error_proto if it does not contain '(h1,h2,h3,h4,p1,p2)'
856 Return ('host.addr.as.numbers', port#) tuple.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000857
Tim Peters88869f92001-01-14 23:36:06 +0000858 if resp[:3] != '227':
Collin Winterce36ad82007-08-30 01:19:48 +0000859 raise error_reply(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000860 global _227_re
861 if _227_re is None:
862 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000863 _227_re = re.compile(r'(\d+),(\d+),(\d+),(\d+),(\d+),(\d+)', re.ASCII)
Guido van Rossum70297d32001-08-17 17:24:29 +0000864 m = _227_re.search(resp)
865 if not m:
Collin Winterce36ad82007-08-30 01:19:48 +0000866 raise error_proto(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000867 numbers = m.groups()
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000868 host = '.'.join(numbers[:4])
869 port = (int(numbers[4]) << 8) + int(numbers[5])
Tim Peters88869f92001-01-14 23:36:06 +0000870 return host, port
Guido van Rossumd2560b01996-05-28 23:41:25 +0000871
872
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000873def parse229(resp, peer):
874 '''Parse the '229' response for a EPSV request.
875 Raises error_proto if it does not contain '(|||port|)'
876 Return ('host.addr.as.numbers', port#) tuple.'''
877
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000878 if resp[:3] != '229':
Collin Winterce36ad82007-08-30 01:19:48 +0000879 raise error_reply(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000880 left = resp.find('(')
Collin Winterce36ad82007-08-30 01:19:48 +0000881 if left < 0: raise error_proto(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000882 right = resp.find(')', left + 1)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000883 if right < 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000884 raise error_proto(resp) # should contain '(|||port|)'
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000885 if resp[left + 1] != resp[right - 1]:
Collin Winterce36ad82007-08-30 01:19:48 +0000886 raise error_proto(resp)
Walter Dörwalda401ae42002-06-03 10:41:45 +0000887 parts = resp[left + 1:right].split(resp[left+1])
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000888 if len(parts) != 5:
Collin Winterce36ad82007-08-30 01:19:48 +0000889 raise error_proto(resp)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000890 host = peer[0]
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000891 port = int(parts[3])
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000892 return host, port
893
894
Guido van Rossumc567c601992-11-05 22:22:37 +0000895def parse257(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000896 '''Parse the '257' response for a MKD or PWD request.
897 This is a response to a MKD or PWD request: a directory name.
898 Returns the directoryname in the 257 reply.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000899
Tim Peters88869f92001-01-14 23:36:06 +0000900 if resp[:3] != '257':
Collin Winterce36ad82007-08-30 01:19:48 +0000901 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000902 if resp[3:5] != ' "':
903 return '' # Not compliant to RFC 959, but UNIX ftpd does this
904 dirname = ''
905 i = 5
906 n = len(resp)
907 while i < n:
908 c = resp[i]
909 i = i+1
910 if c == '"':
911 if i >= n or resp[i] != '"':
912 break
913 i = i+1
914 dirname = dirname + c
915 return dirname
Guido van Rossum1115ab21992-11-04 15:51:30 +0000916
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000917
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000918def print_line(line):
Tim Peters88869f92001-01-14 23:36:06 +0000919 '''Default retrlines callback to print a line.'''
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000920 print(line)
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000921
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000922
Guido van Rossumd2560b01996-05-28 23:41:25 +0000923def ftpcp(source, sourcename, target, targetname = '', type = 'I'):
Tim Peters88869f92001-01-14 23:36:06 +0000924 '''Copy file from one FTP-instance to another.'''
925 if not targetname: targetname = sourcename
926 type = 'TYPE ' + type
927 source.voidcmd(type)
928 target.voidcmd(type)
929 sourcehost, sourceport = parse227(source.sendcmd('PASV'))
930 target.sendport(sourcehost, sourceport)
931 # RFC 959: the user must "listen" [...] BEFORE sending the
932 # transfer request.
933 # So: STOR before RETR, because here the target is a "user".
934 treply = target.sendcmd('STOR ' + targetname)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000935 if treply[:3] not in {'125', '150'}: raise error_proto # RFC 959
Tim Peters88869f92001-01-14 23:36:06 +0000936 sreply = source.sendcmd('RETR ' + sourcename)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000937 if sreply[:3] not in {'125', '150'}: raise error_proto # RFC 959
Tim Peters88869f92001-01-14 23:36:06 +0000938 source.voidresp()
939 target.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000940
Tim Peters88869f92001-01-14 23:36:06 +0000941
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000942class Netrc:
Tim Peters88869f92001-01-14 23:36:06 +0000943 """Class to parse & provide access to 'netrc' format files.
Fred Drake475d51d1997-06-24 22:02:54 +0000944
Tim Peters88869f92001-01-14 23:36:06 +0000945 See the netrc(4) man page for information on the file format.
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000946
Tim Peters88869f92001-01-14 23:36:06 +0000947 WARNING: This class is obsolete -- use module netrc instead.
Guido van Rossumc822a451998-12-22 16:49:16 +0000948
Tim Peters88869f92001-01-14 23:36:06 +0000949 """
950 __defuser = None
951 __defpasswd = None
952 __defacct = None
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000953
Tim Peters88869f92001-01-14 23:36:06 +0000954 def __init__(self, filename=None):
Raymond Hettinger094662a2002-06-01 01:29:16 +0000955 if filename is None:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000956 if "HOME" in os.environ:
Tim Peters88869f92001-01-14 23:36:06 +0000957 filename = os.path.join(os.environ["HOME"],
958 ".netrc")
959 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000960 raise IOError("specify file to load or set $HOME")
Tim Peters88869f92001-01-14 23:36:06 +0000961 self.__hosts = {}
962 self.__macros = {}
963 fp = open(filename, "r")
964 in_macro = 0
965 while 1:
966 line = fp.readline()
967 if not line: break
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000968 if in_macro and line.strip():
Tim Peters88869f92001-01-14 23:36:06 +0000969 macro_lines.append(line)
970 continue
971 elif in_macro:
972 self.__macros[macro_name] = tuple(macro_lines)
973 in_macro = 0
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000974 words = line.split()
Tim Peters88869f92001-01-14 23:36:06 +0000975 host = user = passwd = acct = None
976 default = 0
977 i = 0
978 while i < len(words):
979 w1 = words[i]
980 if i+1 < len(words):
981 w2 = words[i + 1]
982 else:
983 w2 = None
984 if w1 == 'default':
985 default = 1
986 elif w1 == 'machine' and w2:
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000987 host = w2.lower()
Tim Peters88869f92001-01-14 23:36:06 +0000988 i = i + 1
989 elif w1 == 'login' and w2:
990 user = w2
991 i = i + 1
992 elif w1 == 'password' and w2:
993 passwd = w2
994 i = i + 1
995 elif w1 == 'account' and w2:
996 acct = w2
997 i = i + 1
998 elif w1 == 'macdef' and w2:
999 macro_name = w2
1000 macro_lines = []
1001 in_macro = 1
1002 break
1003 i = i + 1
1004 if default:
1005 self.__defuser = user or self.__defuser
1006 self.__defpasswd = passwd or self.__defpasswd
1007 self.__defacct = acct or self.__defacct
1008 if host:
Raymond Hettinger54f02222002-06-01 14:18:47 +00001009 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +00001010 ouser, opasswd, oacct = \
1011 self.__hosts[host]
1012 user = user or ouser
1013 passwd = passwd or opasswd
1014 acct = acct or oacct
1015 self.__hosts[host] = user, passwd, acct
1016 fp.close()
Guido van Rossum56d1e3a1997-03-14 04:16:54 +00001017
Tim Peters88869f92001-01-14 23:36:06 +00001018 def get_hosts(self):
1019 """Return a list of hosts mentioned in the .netrc file."""
1020 return self.__hosts.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +00001021
Tim Peters88869f92001-01-14 23:36:06 +00001022 def get_account(self, host):
1023 """Returns login information for the named host.
Guido van Rossum8ca84201998-03-26 20:56:10 +00001024
Tim Peters88869f92001-01-14 23:36:06 +00001025 The return value is a triple containing userid,
1026 password, and the accounting field.
Guido van Rossum8ca84201998-03-26 20:56:10 +00001027
Tim Peters88869f92001-01-14 23:36:06 +00001028 """
Eric S. Raymondc95bf692001-02-09 10:06:47 +00001029 host = host.lower()
Tim Peters88869f92001-01-14 23:36:06 +00001030 user = passwd = acct = None
Raymond Hettinger54f02222002-06-01 14:18:47 +00001031 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +00001032 user, passwd, acct = self.__hosts[host]
1033 user = user or self.__defuser
1034 passwd = passwd or self.__defpasswd
1035 acct = acct or self.__defacct
1036 return user, passwd, acct
Guido van Rossum8ca84201998-03-26 20:56:10 +00001037
Tim Peters88869f92001-01-14 23:36:06 +00001038 def get_macros(self):
1039 """Return a list of all defined macro names."""
1040 return self.__macros.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +00001041
Tim Peters88869f92001-01-14 23:36:06 +00001042 def get_macro(self, macro):
1043 """Return a sequence of lines which define a named macro."""
1044 return self.__macros[macro]
Guido van Rossum56d1e3a1997-03-14 04:16:54 +00001045
Fred Drake475d51d1997-06-24 22:02:54 +00001046
Tim Peters88869f92001-01-14 23:36:06 +00001047
Guido van Rossum1115ab21992-11-04 15:51:30 +00001048def test():
Tim Peters88869f92001-01-14 23:36:06 +00001049 '''Test program.
Raymond Hettingerc88a6c72005-04-05 04:31:09 +00001050 Usage: ftp [-d] [-r[file]] host [-l[dir]] [-d[dir]] [-p] [file] ...
1051
1052 -d dir
1053 -l list
1054 -p password
1055 '''
1056
1057 if len(sys.argv) < 2:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001058 print(test.__doc__)
Raymond Hettingerc88a6c72005-04-05 04:31:09 +00001059 sys.exit(0)
Guido van Rossumd2560b01996-05-28 23:41:25 +00001060
Tim Peters88869f92001-01-14 23:36:06 +00001061 debugging = 0
1062 rcfile = None
1063 while sys.argv[1] == '-d':
1064 debugging = debugging+1
1065 del sys.argv[1]
1066 if sys.argv[1][:2] == '-r':
1067 # get name of alternate ~/.netrc file:
1068 rcfile = sys.argv[1][2:]
1069 del sys.argv[1]
1070 host = sys.argv[1]
1071 ftp = FTP(host)
1072 ftp.set_debuglevel(debugging)
1073 userid = passwd = acct = ''
1074 try:
1075 netrc = Netrc(rcfile)
1076 except IOError:
1077 if rcfile is not None:
1078 sys.stderr.write("Could not open account file"
1079 " -- using anonymous login.")
1080 else:
1081 try:
1082 userid, passwd, acct = netrc.get_account(host)
1083 except KeyError:
1084 # no account for host
1085 sys.stderr.write(
1086 "No account -- using anonymous login.")
1087 ftp.login(userid, passwd, acct)
1088 for file in sys.argv[2:]:
1089 if file[:2] == '-l':
1090 ftp.dir(file[2:])
1091 elif file[:2] == '-d':
1092 cmd = 'CWD'
1093 if file[2:]: cmd = cmd + ' ' + file[2:]
1094 resp = ftp.sendcmd(cmd)
1095 elif file == '-p':
1096 ftp.set_pasv(not ftp.passiveserver)
1097 else:
1098 ftp.retrbinary('RETR ' + file, \
1099 sys.stdout.write, 1024)
1100 ftp.quit()
Guido van Rossum221ec0b1995-08-04 04:39:30 +00001101
1102
1103if __name__ == '__main__':
Tim Peters88869f92001-01-14 23:36:06 +00001104 test()