blob: 18887a690c8964a46843d6fb165ee5e59743f77d [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.'''
Martin v. Löwis2ad25692001-07-31 08:40:21 +0000291 msg = "getaddrinfo returns an empty list"
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)
Guido van Rossumb940e112007-01-10 16:19:56 +0000298 except socket.error as msg:
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
304 if not sock:
Collin Winterce36ad82007-08-30 01:19:48 +0000305 raise socket.error(msg)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000306 sock.listen(1)
307 port = sock.getsockname()[1] # Get proper port
308 host = self.sock.getsockname()[0] # Get proper host
309 if self.af == socket.AF_INET:
310 resp = self.sendport(host, port)
311 else:
312 resp = self.sendeprt(host, port)
Giampaolo RodolĂ 5fb313b2010-04-19 22:05:54 +0000313 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
314 sock.settimeout(self.timeout)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000315 return sock
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000316
317 def makepasv(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000318 if self.af == socket.AF_INET:
319 host, port = parse227(self.sendcmd('PASV'))
320 else:
321 host, port = parse229(self.sendcmd('EPSV'), self.sock.getpeername())
322 return host, port
Guido van Rossum1115ab21992-11-04 15:51:30 +0000323
Tim Peters88869f92001-01-14 23:36:06 +0000324 def ntransfercmd(self, cmd, rest=None):
325 """Initiate a transfer over the data connection.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000326
Tim Peters88869f92001-01-14 23:36:06 +0000327 If the transfer is active, send a port command and the
328 transfer command, and accept the connection. If the server is
329 passive, send a pasv command, connect to it, and start the
330 transfer command. Either way, return the socket for the
331 connection and the expected size of the transfer. The
332 expected size may be None if it could not be determined.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000333
Tim Peters88869f92001-01-14 23:36:06 +0000334 Optional `rest' argument can be a string that is sent as the
Christian Heimes1af737c2008-01-23 08:24:23 +0000335 argument to a REST command. This is essentially a server
Tim Peters88869f92001-01-14 23:36:06 +0000336 marker used to tell the server to skip over any data up to the
337 given marker.
338 """
339 size = None
340 if self.passiveserver:
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000341 host, port = self.makepasv()
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000342 conn = socket.create_connection((host, port), self.timeout,
343 source_address=self.source_address)
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +0200344 try:
345 if rest is not None:
346 self.sendcmd("REST %s" % rest)
347 resp = self.sendcmd(cmd)
348 # Some servers apparently send a 200 reply to
349 # a LIST or STOR command, before the 150 reply
350 # (and way before the 226 reply). This seems to
351 # be in violation of the protocol (which only allows
352 # 1xx or error messages for LIST), so we just discard
353 # this response.
354 if resp[0] == '2':
355 resp = self.getresp()
356 if resp[0] != '1':
357 raise error_reply(resp)
358 except:
359 conn.close()
360 raise
Tim Peters88869f92001-01-14 23:36:06 +0000361 else:
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100362 with self.makeport() as sock:
Nadeem Vawda08f5f7a2011-07-23 14:03:00 +0200363 if rest is not None:
364 self.sendcmd("REST %s" % rest)
365 resp = self.sendcmd(cmd)
366 # See above.
367 if resp[0] == '2':
368 resp = self.getresp()
369 if resp[0] != '1':
370 raise error_reply(resp)
371 conn, sockaddr = sock.accept()
372 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
373 conn.settimeout(self.timeout)
Tim Peters88869f92001-01-14 23:36:06 +0000374 if resp[:3] == '150':
375 # this is conditional in case we received a 125
376 size = parse150(resp)
377 return conn, size
Fred Drake4de02d91997-01-10 18:26:09 +0000378
Tim Peters88869f92001-01-14 23:36:06 +0000379 def transfercmd(self, cmd, rest=None):
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000380 """Like ntransfercmd() but returns only the socket."""
Tim Peters88869f92001-01-14 23:36:06 +0000381 return self.ntransfercmd(cmd, rest)[0]
Guido van Rossumc567c601992-11-05 22:22:37 +0000382
Tim Peters88869f92001-01-14 23:36:06 +0000383 def login(self, user = '', passwd = '', acct = ''):
384 '''Login, default anonymous.'''
385 if not user: user = 'anonymous'
386 if not passwd: passwd = ''
387 if not acct: acct = ''
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000388 if user == 'anonymous' and passwd in {'', '-'}:
Tim Peterse4418602002-02-16 07:34:19 +0000389 # If there is no anonymous ftp password specified
390 # then we'll just use anonymous@
391 # We don't send any other thing because:
392 # - We want to remain anonymous
393 # - We want to stop SPAM
394 # - We don't want to let ftp sites to discriminate by the user,
395 # host or country.
Guido van Rossumc33e0772001-12-28 20:54:28 +0000396 passwd = passwd + 'anonymous@'
Tim Peters88869f92001-01-14 23:36:06 +0000397 resp = self.sendcmd('USER ' + user)
398 if resp[0] == '3': resp = self.sendcmd('PASS ' + passwd)
399 if resp[0] == '3': resp = self.sendcmd('ACCT ' + acct)
400 if resp[0] != '2':
Collin Winterce36ad82007-08-30 01:19:48 +0000401 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000402 return resp
Guido van Rossumc567c601992-11-05 22:22:37 +0000403
Tim Peters88869f92001-01-14 23:36:06 +0000404 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000405 """Retrieve data in binary mode. A new port is created for you.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000406
Christian Heimes1af737c2008-01-23 08:24:23 +0000407 Args:
408 cmd: A RETR command.
409 callback: A single parameter callable to be called on each
410 block of data read.
411 blocksize: The maximum number of bytes to read from the
412 socket at one time. [default: 8192]
413 rest: Passed to transfercmd(). [default: None]
Guido van Rossum1115ab21992-11-04 15:51:30 +0000414
Christian Heimes1af737c2008-01-23 08:24:23 +0000415 Returns:
416 The response code.
Tim Peters88869f92001-01-14 23:36:06 +0000417 """
418 self.voidcmd('TYPE I')
Benjamin Petersona8071372010-10-31 18:21:16 +0000419 with self.transfercmd(cmd, rest) as conn:
420 while 1:
421 data = conn.recv(blocksize)
422 if not data:
423 break
424 callback(data)
Tim Peters88869f92001-01-14 23:36:06 +0000425 return self.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000426
Tim Peters88869f92001-01-14 23:36:06 +0000427 def retrlines(self, cmd, callback = None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000428 """Retrieve data in line mode. A new port is created for you.
429
430 Args:
Giampaolo Rodola'd78def92011-05-06 19:49:08 +0200431 cmd: A RETR, LIST, or NLST command.
Christian Heimes1af737c2008-01-23 08:24:23 +0000432 callback: An optional single parameter callable that is called
433 for each line with the trailing CRLF stripped.
434 [default: print_line()]
435
436 Returns:
437 The response code.
438 """
Raymond Hettingere874fc32002-05-12 05:53:51 +0000439 if callback is None: callback = print_line
Tim Peters88869f92001-01-14 23:36:06 +0000440 resp = self.sendcmd('TYPE A')
Benjamin Petersona8071372010-10-31 18:21:16 +0000441 with self.transfercmd(cmd) as conn, \
442 conn.makefile('r', encoding=self.encoding) as fp:
443 while 1:
444 line = fp.readline()
445 if self.debugging > 2: print('*retr*', repr(line))
446 if not line:
447 break
448 if line[-2:] == CRLF:
449 line = line[:-2]
450 elif line[-1:] == '\n':
451 line = line[:-1]
452 callback(line)
Tim Peters88869f92001-01-14 23:36:06 +0000453 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000454
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000455 def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000456 """Store a file in binary mode. A new port is created for you.
457
458 Args:
459 cmd: A STOR command.
460 fp: A file-like object with a read(num_bytes) method.
461 blocksize: The maximum data size to read from fp and send over
462 the connection at once. [default: 8192]
463 callback: An optional single parameter callable that is called on
464 on each block of data after it is sent. [default: None]
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000465 rest: Passed to transfercmd(). [default: None]
Christian Heimes1af737c2008-01-23 08:24:23 +0000466
467 Returns:
468 The response code.
469 """
Tim Peters88869f92001-01-14 23:36:06 +0000470 self.voidcmd('TYPE I')
Benjamin Petersona8071372010-10-31 18:21:16 +0000471 with self.transfercmd(cmd, rest) as conn:
472 while 1:
473 buf = fp.read(blocksize)
474 if not buf: break
475 conn.sendall(buf)
476 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000477 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000478
Christian Heimes1af737c2008-01-23 08:24:23 +0000479 def storlines(self, cmd, fp, callback=None):
480 """Store a file in line mode. A new port is created for you.
481
482 Args:
483 cmd: A STOR command.
484 fp: A file-like object with a readline() method.
485 callback: An optional single parameter callable that is called on
486 on each line after it is sent. [default: None]
487
488 Returns:
489 The response code.
490 """
Tim Peters88869f92001-01-14 23:36:06 +0000491 self.voidcmd('TYPE A')
Benjamin Petersona8071372010-10-31 18:21:16 +0000492 with self.transfercmd(cmd) as conn:
493 while 1:
494 buf = fp.readline()
495 if not buf: break
496 if buf[-2:] != B_CRLF:
497 if buf[-1] in B_CRLF: buf = buf[:-1]
498 buf = buf + B_CRLF
499 conn.sendall(buf)
500 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000501 return self.voidresp()
Guido van Rossum0eaa74b1996-01-25 18:37:21 +0000502
Tim Peters88869f92001-01-14 23:36:06 +0000503 def acct(self, password):
504 '''Send new account name.'''
505 cmd = 'ACCT ' + password
506 return self.voidcmd(cmd)
Guido van Rossumc567c601992-11-05 22:22:37 +0000507
Tim Peters88869f92001-01-14 23:36:06 +0000508 def nlst(self, *args):
509 '''Return a list of files in a given directory (default the current).'''
510 cmd = 'NLST'
511 for arg in args:
512 cmd = cmd + (' ' + arg)
513 files = []
514 self.retrlines(cmd, files.append)
515 return files
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000516
Tim Peters88869f92001-01-14 23:36:06 +0000517 def dir(self, *args):
518 '''List a directory in long form.
519 By default list current directory to stdout.
520 Optional last argument is callback function; all
521 non-empty arguments before it are concatenated to the
522 LIST command. (This *should* only be used for a pathname.)'''
523 cmd = 'LIST'
524 func = None
525 if args[-1:] and type(args[-1]) != type(''):
526 args, func = args[:-1], args[-1]
527 for arg in args:
528 if arg:
529 cmd = cmd + (' ' + arg)
530 self.retrlines(cmd, func)
Guido van Rossumc567c601992-11-05 22:22:37 +0000531
Giampaolo Rodola'd78def92011-05-06 19:49:08 +0200532 def mlsd(self, path="", facts=[]):
533 '''List a directory in a standardized format by using MLSD
534 command (RFC-3659). If path is omitted the current directory
535 is assumed. "facts" is a list of strings representing the type
536 of information desired (e.g. ["type", "size", "perm"]).
537
538 Return a generator object yielding a tuple of two elements
539 for every file found in path.
540 First element is the file name, the second one is a dictionary
541 including a variable number of "facts" depending on the server
542 and whether "facts" argument has been provided.
543 '''
544 if facts:
545 self.sendcmd("OPTS MLST " + ";".join(facts) + ";")
546 if path:
547 cmd = "MLSD %s" % path
548 else:
549 cmd = "MLSD"
550 lines = []
551 self.retrlines(cmd, lines.append)
552 for line in lines:
553 facts_found, _, name = line.rstrip(CRLF).partition(' ')
554 entry = {}
555 for fact in facts_found[:-1].split(";"):
556 key, _, value = fact.partition("=")
557 entry[key.lower()] = value
558 yield (name, entry)
559
Tim Peters88869f92001-01-14 23:36:06 +0000560 def rename(self, fromname, toname):
561 '''Rename a file.'''
562 resp = self.sendcmd('RNFR ' + fromname)
563 if resp[0] != '3':
Collin Winterce36ad82007-08-30 01:19:48 +0000564 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000565 return self.voidcmd('RNTO ' + toname)
Guido van Rossuma61bdeb1995-10-11 17:36:31 +0000566
Tim Peters88869f92001-01-14 23:36:06 +0000567 def delete(self, filename):
568 '''Delete a file.'''
569 resp = self.sendcmd('DELE ' + filename)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000570 if resp[:3] in {'250', '200'}:
Tim Peters88869f92001-01-14 23:36:06 +0000571 return resp
Tim Peters88869f92001-01-14 23:36:06 +0000572 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000573 raise error_reply(resp)
Guido van Rossum02cf5821993-05-17 08:00:02 +0000574
Tim Peters88869f92001-01-14 23:36:06 +0000575 def cwd(self, dirname):
576 '''Change to a directory.'''
577 if dirname == '..':
578 try:
579 return self.voidcmd('CDUP')
Guido van Rossumb940e112007-01-10 16:19:56 +0000580 except error_perm as msg:
Martin v. Löwisb5255112002-03-10 15:59:58 +0000581 if msg.args[0][:3] != '500':
582 raise
Tim Peters88869f92001-01-14 23:36:06 +0000583 elif dirname == '':
584 dirname = '.' # does nothing, but could return error
585 cmd = 'CWD ' + dirname
586 return self.voidcmd(cmd)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000587
Tim Peters88869f92001-01-14 23:36:06 +0000588 def size(self, filename):
589 '''Retrieve the size of a file.'''
Christian Heimes1af737c2008-01-23 08:24:23 +0000590 # The SIZE command is defined in RFC-3659
Tim Peters88869f92001-01-14 23:36:06 +0000591 resp = self.sendcmd('SIZE ' + filename)
592 if resp[:3] == '213':
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000593 s = resp[3:].strip()
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200594 return int(s)
Guido van Rossumc567c601992-11-05 22:22:37 +0000595
Tim Peters88869f92001-01-14 23:36:06 +0000596 def mkd(self, dirname):
597 '''Make a directory, return its full pathname.'''
Giampaolo RodolĂ bbc47822010-08-23 22:10:32 +0000598 resp = self.voidcmd('MKD ' + dirname)
599 # fix around non-compliant implementations such as IIS shipped
600 # with Windows server 2003
601 if not resp.startswith('257'):
602 return ''
Tim Peters88869f92001-01-14 23:36:06 +0000603 return parse257(resp)
Guido van Rossum98245091998-02-19 21:15:44 +0000604
Tim Peters88869f92001-01-14 23:36:06 +0000605 def rmd(self, dirname):
606 '''Remove a directory.'''
607 return self.voidcmd('RMD ' + dirname)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000608
Tim Peters88869f92001-01-14 23:36:06 +0000609 def pwd(self):
610 '''Return current working directory.'''
Giampaolo RodolĂ bbc47822010-08-23 22:10:32 +0000611 resp = self.voidcmd('PWD')
612 # fix around non-compliant implementations such as IIS shipped
613 # with Windows server 2003
614 if not resp.startswith('257'):
615 return ''
Tim Peters88869f92001-01-14 23:36:06 +0000616 return parse257(resp)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000617
Tim Peters88869f92001-01-14 23:36:06 +0000618 def quit(self):
619 '''Quit, and close the connection.'''
620 resp = self.voidcmd('QUIT')
621 self.close()
622 return resp
623
624 def close(self):
625 '''Close the connection without assuming anything about it.'''
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000626 if self.file is not None:
Fred Drake9c98a422001-02-28 21:46:37 +0000627 self.file.close()
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000628 if self.sock is not None:
Fred Drake9c98a422001-02-28 21:46:37 +0000629 self.sock.close()
Giampaolo RodolĂ d6868482011-02-22 19:24:33 +0000630 self.file = self.sock = None
Guido van Rossumc567c601992-11-05 22:22:37 +0000631
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000632try:
633 import ssl
634except ImportError:
635 pass
636else:
637 class FTP_TLS(FTP):
638 '''A FTP subclass which adds TLS support to FTP as described
639 in RFC-4217.
640
641 Connect as usual to port 21 implicitly securing the FTP control
642 connection before authenticating.
643
644 Securing the data connection requires user to explicitly ask
645 for it by calling prot_p() method.
646
647 Usage example:
648 >>> from ftplib import FTP_TLS
649 >>> ftps = FTP_TLS('ftp.python.org')
Ezio Melotti4969f702011-03-15 05:59:46 +0200650 >>> ftps.login() # login anonymously previously securing control channel
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000651 '230 Guest login ok, access restrictions apply.'
652 >>> ftps.prot_p() # switch to secure data connection
653 '200 Protection level set to P'
654 >>> ftps.retrlines('LIST') # list directory content securely
655 total 9
656 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 .
657 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 ..
658 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 bin
659 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 etc
660 d-wxrwxr-x 2 ftp wheel 1024 Sep 5 13:43 incoming
661 drwxr-xr-x 2 root wheel 1024 Nov 17 1993 lib
662 drwxr-xr-x 6 1094 wheel 1024 Sep 13 19:07 pub
663 drwxr-xr-x 3 root wheel 1024 Jan 3 1994 usr
664 -rw-r--r-- 1 root root 312 Aug 1 1994 welcome.msg
665 '226 Transfer complete.'
666 >>> ftps.quit()
667 '221 Goodbye.'
668 >>>
669 '''
670 ssl_version = ssl.PROTOCOL_TLSv1
671
672 def __init__(self, host='', user='', passwd='', acct='', keyfile=None,
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000673 certfile=None, context=None,
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000674 timeout=_GLOBAL_DEFAULT_TIMEOUT, source_address=None):
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000675 if context is not None and keyfile is not None:
676 raise ValueError("context and keyfile arguments are mutually "
677 "exclusive")
678 if context is not None and certfile is not None:
679 raise ValueError("context and certfile arguments are mutually "
680 "exclusive")
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000681 self.keyfile = keyfile
682 self.certfile = certfile
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000683 self.context = context
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000684 self._prot_p = False
Giampaolo RodolĂ 396ff062011-02-28 19:19:51 +0000685 FTP.__init__(self, host, user, passwd, acct, timeout, source_address)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000686
687 def login(self, user='', passwd='', acct='', secure=True):
688 if secure and not isinstance(self.sock, ssl.SSLSocket):
689 self.auth()
690 return FTP.login(self, user, passwd, acct)
691
692 def auth(self):
693 '''Set up secure control connection by using TLS/SSL.'''
694 if isinstance(self.sock, ssl.SSLSocket):
695 raise ValueError("Already using TLS")
696 if self.ssl_version == ssl.PROTOCOL_TLSv1:
697 resp = self.voidcmd('AUTH TLS')
698 else:
699 resp = self.voidcmd('AUTH SSL')
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000700 if self.context is not None:
701 self.sock = self.context.wrap_socket(self.sock)
702 else:
703 self.sock = ssl.wrap_socket(self.sock, self.keyfile,
704 self.certfile,
705 ssl_version=self.ssl_version)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000706 self.file = self.sock.makefile(mode='r', encoding=self.encoding)
707 return resp
708
Giampaolo Rodola'096dcb12011-06-27 11:17:51 +0200709 def ccc(self):
710 '''Switch back to a clear-text control connection.'''
711 if not isinstance(self.sock, ssl.SSLSocket):
712 raise ValueError("not using TLS")
713 resp = self.voidcmd('CCC')
714 self.sock = self.sock.unwrap()
715 return resp
716
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000717 def prot_p(self):
718 '''Set up secure data connection.'''
719 # PROT defines whether or not the data channel is to be protected.
720 # Though RFC-2228 defines four possible protection levels,
721 # RFC-4217 only recommends two, Clear and Private.
722 # Clear (PROT C) means that no security is to be used on the
723 # data-channel, Private (PROT P) means that the data-channel
724 # should be protected by TLS.
725 # PBSZ command MUST still be issued, but must have a parameter of
726 # '0' to indicate that no buffering is taking place and the data
727 # connection should not be encapsulated.
728 self.voidcmd('PBSZ 0')
729 resp = self.voidcmd('PROT P')
730 self._prot_p = True
731 return resp
732
733 def prot_c(self):
734 '''Set up clear text data connection.'''
735 resp = self.voidcmd('PROT C')
736 self._prot_p = False
737 return resp
738
739 # --- Overridden FTP methods
740
741 def ntransfercmd(self, cmd, rest=None):
742 conn, size = FTP.ntransfercmd(self, cmd, rest)
743 if self._prot_p:
Giampaolo RodolĂ a67299e2010-05-26 18:06:04 +0000744 if self.context is not None:
745 conn = self.context.wrap_socket(conn)
746 else:
747 conn = ssl.wrap_socket(conn, self.keyfile, self.certfile,
748 ssl_version=self.ssl_version)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000749 return conn, size
750
751 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
752 self.voidcmd('TYPE I')
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100753 with self.transfercmd(cmd, rest) as conn:
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000754 while 1:
755 data = conn.recv(blocksize)
756 if not data:
757 break
758 callback(data)
759 # shutdown ssl layer
760 if isinstance(conn, ssl.SSLSocket):
761 conn.unwrap()
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000762 return self.voidresp()
763
764 def retrlines(self, cmd, callback = None):
765 if callback is None: callback = print_line
766 resp = self.sendcmd('TYPE A')
767 conn = self.transfercmd(cmd)
768 fp = conn.makefile('r', encoding=self.encoding)
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100769 with fp, conn:
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000770 while 1:
771 line = fp.readline()
772 if self.debugging > 2: print('*retr*', repr(line))
773 if not line:
774 break
775 if line[-2:] == CRLF:
776 line = line[:-2]
777 elif line[-1:] == '\n':
778 line = line[:-1]
779 callback(line)
780 # shutdown ssl layer
781 if isinstance(conn, ssl.SSLSocket):
782 conn.unwrap()
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000783 return self.voidresp()
784
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000785 def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000786 self.voidcmd('TYPE I')
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100787 with self.transfercmd(cmd, rest) as conn:
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000788 while 1:
789 buf = fp.read(blocksize)
790 if not buf: break
791 conn.sendall(buf)
792 if callback: callback(buf)
793 # shutdown ssl layer
794 if isinstance(conn, ssl.SSLSocket):
795 conn.unwrap()
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000796 return self.voidresp()
797
798 def storlines(self, cmd, fp, callback=None):
799 self.voidcmd('TYPE A')
Giampaolo Rodola'836e9aa2011-12-10 21:25:04 +0100800 with self.transfercmd(cmd) as conn:
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000801 while 1:
802 buf = fp.readline()
803 if not buf: break
Giampaolo RodolĂ f96482e2010-08-04 10:36:18 +0000804 if buf[-2:] != B_CRLF:
805 if buf[-1] in B_CRLF: buf = buf[:-1]
806 buf = buf + B_CRLF
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000807 conn.sendall(buf)
808 if callback: callback(buf)
809 # shutdown ssl layer
810 if isinstance(conn, ssl.SSLSocket):
811 conn.unwrap()
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000812 return self.voidresp()
813
Giampaolo Rodola'24befa82011-05-07 19:09:34 +0200814 def abort(self):
815 # overridden as we can't pass MSG_OOB flag to sendall()
816 line = b'ABOR' + B_CRLF
817 self.sock.sendall(line)
818 resp = self.getmultiline()
819 if resp[:3] not in {'426', '225', '226'}:
820 raise error_proto(resp)
821 return resp
822
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000823 __all__.append('FTP_TLS')
824 all_errors = (Error, IOError, EOFError, ssl.SSLError)
825
826
Guido van Rossumacfb82a1997-10-22 20:49:52 +0000827_150_re = None
Fred Drake4de02d91997-01-10 18:26:09 +0000828
829def parse150(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000830 '''Parse the '150' response for a RETR request.
831 Returns the expected transfer size or None; size is not guaranteed to
832 be present in the 150 message.
833 '''
834 if resp[:3] != '150':
Collin Winterce36ad82007-08-30 01:19:48 +0000835 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000836 global _150_re
837 if _150_re is None:
838 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000839 _150_re = re.compile(
840 "150 .* \((\d+) bytes\)", re.IGNORECASE | re.ASCII)
Tim Peters88869f92001-01-14 23:36:06 +0000841 m = _150_re.match(resp)
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000842 if not m:
843 return None
Florent Xicluna2bb96f52011-10-23 22:11:00 +0200844 return int(m.group(1))
Fred Drake4de02d91997-01-10 18:26:09 +0000845
846
Guido van Rossum70297d32001-08-17 17:24:29 +0000847_227_re = None
848
Guido van Rossumd2560b01996-05-28 23:41:25 +0000849def parse227(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000850 '''Parse the '227' response for a PASV request.
851 Raises error_proto if it does not contain '(h1,h2,h3,h4,p1,p2)'
852 Return ('host.addr.as.numbers', port#) tuple.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000853
Tim Peters88869f92001-01-14 23:36:06 +0000854 if resp[:3] != '227':
Collin Winterce36ad82007-08-30 01:19:48 +0000855 raise error_reply(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000856 global _227_re
857 if _227_re is None:
858 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000859 _227_re = re.compile(r'(\d+),(\d+),(\d+),(\d+),(\d+),(\d+)', re.ASCII)
Guido van Rossum70297d32001-08-17 17:24:29 +0000860 m = _227_re.search(resp)
861 if not m:
Collin Winterce36ad82007-08-30 01:19:48 +0000862 raise error_proto(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000863 numbers = m.groups()
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000864 host = '.'.join(numbers[:4])
865 port = (int(numbers[4]) << 8) + int(numbers[5])
Tim Peters88869f92001-01-14 23:36:06 +0000866 return host, port
Guido van Rossumd2560b01996-05-28 23:41:25 +0000867
868
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000869def parse229(resp, peer):
870 '''Parse the '229' response for a EPSV request.
871 Raises error_proto if it does not contain '(|||port|)'
872 Return ('host.addr.as.numbers', port#) tuple.'''
873
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000874 if resp[:3] != '229':
Collin Winterce36ad82007-08-30 01:19:48 +0000875 raise error_reply(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000876 left = resp.find('(')
Collin Winterce36ad82007-08-30 01:19:48 +0000877 if left < 0: raise error_proto(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000878 right = resp.find(')', left + 1)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000879 if right < 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000880 raise error_proto(resp) # should contain '(|||port|)'
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000881 if resp[left + 1] != resp[right - 1]:
Collin Winterce36ad82007-08-30 01:19:48 +0000882 raise error_proto(resp)
Walter Dörwalda401ae42002-06-03 10:41:45 +0000883 parts = resp[left + 1:right].split(resp[left+1])
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000884 if len(parts) != 5:
Collin Winterce36ad82007-08-30 01:19:48 +0000885 raise error_proto(resp)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000886 host = peer[0]
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000887 port = int(parts[3])
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000888 return host, port
889
890
Guido van Rossumc567c601992-11-05 22:22:37 +0000891def parse257(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000892 '''Parse the '257' response for a MKD or PWD request.
893 This is a response to a MKD or PWD request: a directory name.
894 Returns the directoryname in the 257 reply.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000895
Tim Peters88869f92001-01-14 23:36:06 +0000896 if resp[:3] != '257':
Collin Winterce36ad82007-08-30 01:19:48 +0000897 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000898 if resp[3:5] != ' "':
899 return '' # Not compliant to RFC 959, but UNIX ftpd does this
900 dirname = ''
901 i = 5
902 n = len(resp)
903 while i < n:
904 c = resp[i]
905 i = i+1
906 if c == '"':
907 if i >= n or resp[i] != '"':
908 break
909 i = i+1
910 dirname = dirname + c
911 return dirname
Guido van Rossum1115ab21992-11-04 15:51:30 +0000912
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000913
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000914def print_line(line):
Tim Peters88869f92001-01-14 23:36:06 +0000915 '''Default retrlines callback to print a line.'''
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000916 print(line)
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000917
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000918
Guido van Rossumd2560b01996-05-28 23:41:25 +0000919def ftpcp(source, sourcename, target, targetname = '', type = 'I'):
Tim Peters88869f92001-01-14 23:36:06 +0000920 '''Copy file from one FTP-instance to another.'''
921 if not targetname: targetname = sourcename
922 type = 'TYPE ' + type
923 source.voidcmd(type)
924 target.voidcmd(type)
925 sourcehost, sourceport = parse227(source.sendcmd('PASV'))
926 target.sendport(sourcehost, sourceport)
927 # RFC 959: the user must "listen" [...] BEFORE sending the
928 # transfer request.
929 # So: STOR before RETR, because here the target is a "user".
930 treply = target.sendcmd('STOR ' + targetname)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000931 if treply[:3] not in {'125', '150'}: raise error_proto # RFC 959
Tim Peters88869f92001-01-14 23:36:06 +0000932 sreply = source.sendcmd('RETR ' + sourcename)
Raymond Hettingerd5825cc2010-09-05 23:15:06 +0000933 if sreply[:3] not in {'125', '150'}: raise error_proto # RFC 959
Tim Peters88869f92001-01-14 23:36:06 +0000934 source.voidresp()
935 target.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000936
Tim Peters88869f92001-01-14 23:36:06 +0000937
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000938class Netrc:
Tim Peters88869f92001-01-14 23:36:06 +0000939 """Class to parse & provide access to 'netrc' format files.
Fred Drake475d51d1997-06-24 22:02:54 +0000940
Tim Peters88869f92001-01-14 23:36:06 +0000941 See the netrc(4) man page for information on the file format.
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000942
Tim Peters88869f92001-01-14 23:36:06 +0000943 WARNING: This class is obsolete -- use module netrc instead.
Guido van Rossumc822a451998-12-22 16:49:16 +0000944
Tim Peters88869f92001-01-14 23:36:06 +0000945 """
946 __defuser = None
947 __defpasswd = None
948 __defacct = None
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000949
Tim Peters88869f92001-01-14 23:36:06 +0000950 def __init__(self, filename=None):
Raymond Hettinger094662a2002-06-01 01:29:16 +0000951 if filename is None:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000952 if "HOME" in os.environ:
Tim Peters88869f92001-01-14 23:36:06 +0000953 filename = os.path.join(os.environ["HOME"],
954 ".netrc")
955 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000956 raise IOError("specify file to load or set $HOME")
Tim Peters88869f92001-01-14 23:36:06 +0000957 self.__hosts = {}
958 self.__macros = {}
959 fp = open(filename, "r")
960 in_macro = 0
961 while 1:
962 line = fp.readline()
963 if not line: break
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000964 if in_macro and line.strip():
Tim Peters88869f92001-01-14 23:36:06 +0000965 macro_lines.append(line)
966 continue
967 elif in_macro:
968 self.__macros[macro_name] = tuple(macro_lines)
969 in_macro = 0
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000970 words = line.split()
Tim Peters88869f92001-01-14 23:36:06 +0000971 host = user = passwd = acct = None
972 default = 0
973 i = 0
974 while i < len(words):
975 w1 = words[i]
976 if i+1 < len(words):
977 w2 = words[i + 1]
978 else:
979 w2 = None
980 if w1 == 'default':
981 default = 1
982 elif w1 == 'machine' and w2:
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000983 host = w2.lower()
Tim Peters88869f92001-01-14 23:36:06 +0000984 i = i + 1
985 elif w1 == 'login' and w2:
986 user = w2
987 i = i + 1
988 elif w1 == 'password' and w2:
989 passwd = w2
990 i = i + 1
991 elif w1 == 'account' and w2:
992 acct = w2
993 i = i + 1
994 elif w1 == 'macdef' and w2:
995 macro_name = w2
996 macro_lines = []
997 in_macro = 1
998 break
999 i = i + 1
1000 if default:
1001 self.__defuser = user or self.__defuser
1002 self.__defpasswd = passwd or self.__defpasswd
1003 self.__defacct = acct or self.__defacct
1004 if host:
Raymond Hettinger54f02222002-06-01 14:18:47 +00001005 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +00001006 ouser, opasswd, oacct = \
1007 self.__hosts[host]
1008 user = user or ouser
1009 passwd = passwd or opasswd
1010 acct = acct or oacct
1011 self.__hosts[host] = user, passwd, acct
1012 fp.close()
Guido van Rossum56d1e3a1997-03-14 04:16:54 +00001013
Tim Peters88869f92001-01-14 23:36:06 +00001014 def get_hosts(self):
1015 """Return a list of hosts mentioned in the .netrc file."""
1016 return self.__hosts.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +00001017
Tim Peters88869f92001-01-14 23:36:06 +00001018 def get_account(self, host):
1019 """Returns login information for the named host.
Guido van Rossum8ca84201998-03-26 20:56:10 +00001020
Tim Peters88869f92001-01-14 23:36:06 +00001021 The return value is a triple containing userid,
1022 password, and the accounting field.
Guido van Rossum8ca84201998-03-26 20:56:10 +00001023
Tim Peters88869f92001-01-14 23:36:06 +00001024 """
Eric S. Raymondc95bf692001-02-09 10:06:47 +00001025 host = host.lower()
Tim Peters88869f92001-01-14 23:36:06 +00001026 user = passwd = acct = None
Raymond Hettinger54f02222002-06-01 14:18:47 +00001027 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +00001028 user, passwd, acct = self.__hosts[host]
1029 user = user or self.__defuser
1030 passwd = passwd or self.__defpasswd
1031 acct = acct or self.__defacct
1032 return user, passwd, acct
Guido van Rossum8ca84201998-03-26 20:56:10 +00001033
Tim Peters88869f92001-01-14 23:36:06 +00001034 def get_macros(self):
1035 """Return a list of all defined macro names."""
1036 return self.__macros.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +00001037
Tim Peters88869f92001-01-14 23:36:06 +00001038 def get_macro(self, macro):
1039 """Return a sequence of lines which define a named macro."""
1040 return self.__macros[macro]
Guido van Rossum56d1e3a1997-03-14 04:16:54 +00001041
Fred Drake475d51d1997-06-24 22:02:54 +00001042
Tim Peters88869f92001-01-14 23:36:06 +00001043
Guido van Rossum1115ab21992-11-04 15:51:30 +00001044def test():
Tim Peters88869f92001-01-14 23:36:06 +00001045 '''Test program.
Raymond Hettingerc88a6c72005-04-05 04:31:09 +00001046 Usage: ftp [-d] [-r[file]] host [-l[dir]] [-d[dir]] [-p] [file] ...
1047
1048 -d dir
1049 -l list
1050 -p password
1051 '''
1052
1053 if len(sys.argv) < 2:
Guido van Rossumbe19ed72007-02-09 05:37:30 +00001054 print(test.__doc__)
Raymond Hettingerc88a6c72005-04-05 04:31:09 +00001055 sys.exit(0)
Guido van Rossumd2560b01996-05-28 23:41:25 +00001056
Tim Peters88869f92001-01-14 23:36:06 +00001057 debugging = 0
1058 rcfile = None
1059 while sys.argv[1] == '-d':
1060 debugging = debugging+1
1061 del sys.argv[1]
1062 if sys.argv[1][:2] == '-r':
1063 # get name of alternate ~/.netrc file:
1064 rcfile = sys.argv[1][2:]
1065 del sys.argv[1]
1066 host = sys.argv[1]
1067 ftp = FTP(host)
1068 ftp.set_debuglevel(debugging)
1069 userid = passwd = acct = ''
1070 try:
1071 netrc = Netrc(rcfile)
1072 except IOError:
1073 if rcfile is not None:
1074 sys.stderr.write("Could not open account file"
1075 " -- using anonymous login.")
1076 else:
1077 try:
1078 userid, passwd, acct = netrc.get_account(host)
1079 except KeyError:
1080 # no account for host
1081 sys.stderr.write(
1082 "No account -- using anonymous login.")
1083 ftp.login(userid, passwd, acct)
1084 for file in sys.argv[2:]:
1085 if file[:2] == '-l':
1086 ftp.dir(file[2:])
1087 elif file[:2] == '-d':
1088 cmd = 'CWD'
1089 if file[2:]: cmd = cmd + ' ' + file[2:]
1090 resp = ftp.sendcmd(cmd)
1091 elif file == '-p':
1092 ftp.set_pasv(not ftp.passiveserver)
1093 else:
1094 ftp.retrbinary('RETR ' + file, \
1095 sys.stdout.write, 1024)
1096 ftp.quit()
Guido van Rossum221ec0b1995-08-04 04:39:30 +00001097
1098
1099if __name__ == '__main__':
Tim Peters88869f92001-01-14 23:36:06 +00001100 test()