blob: a0c35784cba3b438fe7c29407bb2b60e987d0a76 [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
Guido van Rossum1115ab21992-11-04 15:51:30 +000041
Guido van Rossumb6775db1994-08-01 11:34:53 +000042# Import SOCKS module if it exists, else standard socket module socket
43try:
Tim Peters88869f92001-01-14 23:36:06 +000044 import SOCKS; socket = SOCKS; del SOCKS # import SOCKS as socket
45 from socket import getfqdn; socket.getfqdn = getfqdn; del getfqdn
Guido van Rossumb6775db1994-08-01 11:34:53 +000046except ImportError:
Tim Peters88869f92001-01-14 23:36:06 +000047 import socket
Georg Brandlf78e02b2008-06-10 17:40:04 +000048from socket import _GLOBAL_DEFAULT_TIMEOUT
Guido van Rossumb6775db1994-08-01 11:34:53 +000049
Skip Montanaroeccd02a2001-01-20 23:34:12 +000050__all__ = ["FTP","Netrc"]
Guido van Rossum1115ab21992-11-04 15:51:30 +000051
Guido van Rossumd3166071993-05-24 14:16:22 +000052# Magic number from <socket.h>
Tim Peters88869f92001-01-14 23:36:06 +000053MSG_OOB = 0x1 # Process data out of band
Guido van Rossumd3166071993-05-24 14:16:22 +000054
55
Guido van Rossumc567c601992-11-05 22:22:37 +000056# The standard FTP server control port
Guido van Rossum1115ab21992-11-04 15:51:30 +000057FTP_PORT = 21
Guido van Rossum1115ab21992-11-04 15:51:30 +000058
59
Guido van Rossum21974791992-11-06 13:34:17 +000060# Exception raised when an error or invalid response is received
Fred Drake227b1202000-08-17 05:06:49 +000061class Error(Exception): pass
Tim Peters88869f92001-01-14 23:36:06 +000062class error_reply(Error): pass # unexpected [123]xx reply
63class error_temp(Error): pass # 4xx errors
64class error_perm(Error): pass # 5xx errors
65class error_proto(Error): pass # response does not begin with [1-5]
Guido van Rossum1115ab21992-11-04 15:51:30 +000066
67
Guido van Rossum21974791992-11-06 13:34:17 +000068# All exceptions (hopefully) that may be raised here and that aren't
69# (always) programming errors on our side
Christian Heimes33fe8092008-04-13 13:53:33 +000070all_errors = (Error, IOError, EOFError)
Guido van Rossum21974791992-11-06 13:34:17 +000071
72
Guido van Rossum1115ab21992-11-04 15:51:30 +000073# Line terminators (we always output CRLF, but accept any of CRLF, CR, LF)
74CRLF = '\r\n'
Benjamin Petersonbe17a112008-09-27 21:49:47 +000075B_CRLF = b'\r\n'
Guido van Rossum1115ab21992-11-04 15:51:30 +000076
Guido van Rossum1115ab21992-11-04 15:51:30 +000077# The class itself
78class FTP:
79
Tim Peters88869f92001-01-14 23:36:06 +000080 '''An FTP client class.
Guido van Rossumd2560b01996-05-28 23:41:25 +000081
Guido van Rossumd8faa362007-04-27 19:54:29 +000082 To create a connection, call the class using these arguments:
83 host, user, passwd, acct, timeout
84
85 The first four arguments are all strings, and have default value ''.
86 timeout must be numeric and defaults to None if not passed,
87 meaning that no timeout will be set on any ftp socket(s)
88 If a timeout is passed, then this is now the default timeout for all ftp
89 socket operations for this instance.
90
Tim Peters88869f92001-01-14 23:36:06 +000091 Then use self.connect() with optional host and port argument.
Guido van Rossumd2560b01996-05-28 23:41:25 +000092
Tim Peters88869f92001-01-14 23:36:06 +000093 To download a file, use ftp.retrlines('RETR ' + filename),
94 or ftp.retrbinary() with slightly different arguments.
95 To upload a file, use ftp.storlines() or ftp.storbinary(),
96 which have an open file as argument (see their definitions
97 below for details).
98 The download/upload functions first issue appropriate TYPE
99 and PORT or PASV commands.
Guido van Rossum52b89762007-07-17 20:45:57 +0000100 '''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000101
Fred Drake9c98a422001-02-28 21:46:37 +0000102 debugging = 0
103 host = ''
104 port = FTP_PORT
105 sock = None
106 file = None
107 welcome = None
108 passiveserver = 1
Brett Cannon6733d702007-10-08 19:48:15 +0000109 encoding = "latin1"
Fred Drake9c98a422001-02-28 21:46:37 +0000110
Tim Peters88869f92001-01-14 23:36:06 +0000111 # Initialization method (called by class instantiation).
112 # Initialize host to localhost, port to standard ftp port
113 # Optional arguments are host (for connect()),
114 # and user, passwd, acct (for login())
Georg Brandlf78e02b2008-06-10 17:40:04 +0000115 def __init__(self, host='', user='', passwd='', acct='',
116 timeout=_GLOBAL_DEFAULT_TIMEOUT):
Guido van Rossumd8faa362007-04-27 19:54:29 +0000117 self.timeout = timeout
Tim Peters88869f92001-01-14 23:36:06 +0000118 if host:
Fred Drake9c98a422001-02-28 21:46:37 +0000119 self.connect(host)
Guido van Rossumd8faa362007-04-27 19:54:29 +0000120 if user:
121 self.login(user, passwd, acct)
Guido van Rossum52fc1f61993-06-17 12:38:10 +0000122
Georg Brandlf78e02b2008-06-10 17:40:04 +0000123 def connect(self, host='', port=0, timeout=-999):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000124 '''Connect to host. Arguments are:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000125 - host: hostname to connect to (string, default previous host)
126 - port: port to connect to (integer, default previous port)
127 '''
128 if host != '':
129 self.host = host
130 if port > 0:
131 self.port = port
Georg Brandlf78e02b2008-06-10 17:40:04 +0000132 if timeout != -999:
Guido van Rossumd8faa362007-04-27 19:54:29 +0000133 self.timeout = timeout
134 self.sock = socket.create_connection((self.host, self.port), self.timeout)
135 self.af = self.sock.family
Guido van Rossum52b89762007-07-17 20:45:57 +0000136 self.file = self.sock.makefile('r', encoding=self.encoding)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000137 self.welcome = self.getresp()
138 return self.welcome
Guido van Rossum1115ab21992-11-04 15:51:30 +0000139
Tim Peters88869f92001-01-14 23:36:06 +0000140 def getwelcome(self):
141 '''Get the welcome message from the server.
142 (this is read and squirreled away by connect())'''
143 if self.debugging:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000144 print('*welcome*', self.sanitize(self.welcome))
Tim Peters88869f92001-01-14 23:36:06 +0000145 return self.welcome
Guido van Rossum1115ab21992-11-04 15:51:30 +0000146
Tim Peters88869f92001-01-14 23:36:06 +0000147 def set_debuglevel(self, level):
148 '''Set the debugging level.
149 The required argument level means:
150 0: no debugging output (default)
151 1: print commands and responses but not body text etc.
152 2: also print raw lines read and sent before stripping CR/LF'''
153 self.debugging = level
154 debug = set_debuglevel
Guido van Rossum1115ab21992-11-04 15:51:30 +0000155
Tim Peters88869f92001-01-14 23:36:06 +0000156 def set_pasv(self, val):
157 '''Use passive or active mode for data transfers.
158 With a false argument, use the normal PORT mode,
159 With a true argument, use the PASV command.'''
160 self.passiveserver = val
Guido van Rossumd2560b01996-05-28 23:41:25 +0000161
Tim Peters88869f92001-01-14 23:36:06 +0000162 # Internal: "sanitize" a string for printing
163 def sanitize(self, s):
164 if s[:5] == 'pass ' or s[:5] == 'PASS ':
165 i = len(s)
166 while i > 5 and s[i-1] in '\r\n':
167 i = i-1
168 s = s[:5] + '*'*(i-5) + s[i:]
Walter Dörwald70a6b492004-02-12 17:35:32 +0000169 return repr(s)
Guido van Rossumebaf1041995-05-05 15:54:14 +0000170
Tim Peters88869f92001-01-14 23:36:06 +0000171 # Internal: send one line to the server, appending CRLF
172 def putline(self, line):
173 line = line + CRLF
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000174 if self.debugging > 1: print('*put*', self.sanitize(line))
Guido van Rossum52b89762007-07-17 20:45:57 +0000175 self.sock.sendall(line.encode(self.encoding))
Guido van Rossum1115ab21992-11-04 15:51:30 +0000176
Tim Peters88869f92001-01-14 23:36:06 +0000177 # Internal: send one command to the server (through putline())
178 def putcmd(self, line):
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000179 if self.debugging: print('*cmd*', self.sanitize(line))
Tim Peters88869f92001-01-14 23:36:06 +0000180 self.putline(line)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000181
Tim Peters88869f92001-01-14 23:36:06 +0000182 # Internal: return one line from the server, stripping CRLF.
183 # Raise EOFError if the connection is closed
184 def getline(self):
185 line = self.file.readline()
186 if self.debugging > 1:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000187 print('*get*', self.sanitize(line))
Tim Peters88869f92001-01-14 23:36:06 +0000188 if not line: raise EOFError
189 if line[-2:] == CRLF: line = line[:-2]
190 elif line[-1:] in CRLF: line = line[:-1]
191 return line
Guido van Rossum1115ab21992-11-04 15:51:30 +0000192
Tim Peters88869f92001-01-14 23:36:06 +0000193 # Internal: get a response from the server, which may possibly
194 # consist of multiple lines. Return a single string with no
195 # trailing CRLF. If the response consists of multiple lines,
196 # these are separated by '\n' characters in the string
197 def getmultiline(self):
198 line = self.getline()
199 if line[3:4] == '-':
200 code = line[:3]
201 while 1:
202 nextline = self.getline()
203 line = line + ('\n' + nextline)
204 if nextline[:3] == code and \
205 nextline[3:4] != '-':
206 break
207 return line
Guido van Rossum1115ab21992-11-04 15:51:30 +0000208
Tim Peters88869f92001-01-14 23:36:06 +0000209 # Internal: get a response from the server.
210 # Raise various errors if the response indicates an error
211 def getresp(self):
212 resp = self.getmultiline()
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000213 if self.debugging: print('*resp*', self.sanitize(resp))
Tim Peters88869f92001-01-14 23:36:06 +0000214 self.lastresp = resp[:3]
215 c = resp[:1]
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000216 if c in ('1', '2', '3'):
217 return resp
Tim Peters88869f92001-01-14 23:36:06 +0000218 if c == '4':
Collin Winterce36ad82007-08-30 01:19:48 +0000219 raise error_temp(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000220 if c == '5':
Collin Winterce36ad82007-08-30 01:19:48 +0000221 raise error_perm(resp)
222 raise error_proto(resp)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000223
Tim Peters88869f92001-01-14 23:36:06 +0000224 def voidresp(self):
225 """Expect a response beginning with '2'."""
226 resp = self.getresp()
Benjamin Petersond23f8222009-04-05 19:13:16 +0000227 if resp[:1] != '2':
Collin Winterce36ad82007-08-30 01:19:48 +0000228 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000229 return resp
Guido van Rossumc567c601992-11-05 22:22:37 +0000230
Tim Peters88869f92001-01-14 23:36:06 +0000231 def abort(self):
232 '''Abort a file transfer. Uses out-of-band data.
233 This does not follow the procedure from the RFC to send Telnet
234 IP and Synch; that doesn't seem to work with the servers I've
235 tried. Instead, just send the ABOR command as OOB data.'''
236 line = 'ABOR' + CRLF
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000237 if self.debugging > 1: print('*put urgent*', self.sanitize(line))
Martin v. Löwise12454f2002-02-16 23:06:19 +0000238 self.sock.sendall(line, MSG_OOB)
Tim Peters88869f92001-01-14 23:36:06 +0000239 resp = self.getmultiline()
Giampaolo Rodolàceb513a2010-04-18 13:36:35 +0000240 if resp[:3] not in ('426', '225', '226'):
Collin Winterce36ad82007-08-30 01:19:48 +0000241 raise error_proto(resp)
Guido van Rossumd3166071993-05-24 14:16:22 +0000242
Tim Peters88869f92001-01-14 23:36:06 +0000243 def sendcmd(self, cmd):
244 '''Send a command and return the response.'''
245 self.putcmd(cmd)
246 return self.getresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000247
Tim Peters88869f92001-01-14 23:36:06 +0000248 def voidcmd(self, cmd):
249 """Send a command and expect a response beginning with '2'."""
250 self.putcmd(cmd)
251 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000252
Tim Peters88869f92001-01-14 23:36:06 +0000253 def sendport(self, host, port):
254 '''Send a PORT command with the current host and the given
255 port number.
256 '''
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000257 hbytes = host.split('.')
Benjamin Peterson3a53fbb2008-09-27 22:04:16 +0000258 pbytes = [repr(port//256), repr(port%256)]
Tim Peters88869f92001-01-14 23:36:06 +0000259 bytes = hbytes + pbytes
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000260 cmd = 'PORT ' + ','.join(bytes)
Tim Peters88869f92001-01-14 23:36:06 +0000261 return self.voidcmd(cmd)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000262
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000263 def sendeprt(self, host, port):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000264 '''Send a EPRT command with the current host and the given port number.'''
265 af = 0
266 if self.af == socket.AF_INET:
267 af = 1
268 if self.af == socket.AF_INET6:
269 af = 2
270 if af == 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000271 raise error_proto('unsupported address family')
Walter Dörwald70a6b492004-02-12 17:35:32 +0000272 fields = ['', repr(af), host, repr(port), '']
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000273 cmd = 'EPRT ' + '|'.join(fields)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000274 return self.voidcmd(cmd)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000275
Tim Peters88869f92001-01-14 23:36:06 +0000276 def makeport(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000277 '''Create a new socket and send a PORT command for it.'''
Martin v. Löwis2ad25692001-07-31 08:40:21 +0000278 msg = "getaddrinfo returns an empty list"
Martin v. Löwis322c0d12001-10-07 08:53:32 +0000279 sock = None
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000280 for res in socket.getaddrinfo(None, 0, self.af, socket.SOCK_STREAM, 0, socket.AI_PASSIVE):
281 af, socktype, proto, canonname, sa = res
282 try:
283 sock = socket.socket(af, socktype, proto)
284 sock.bind(sa)
Guido van Rossumb940e112007-01-10 16:19:56 +0000285 except socket.error as msg:
Martin v. Löwis322c0d12001-10-07 08:53:32 +0000286 if sock:
287 sock.close()
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000288 sock = None
289 continue
290 break
291 if not sock:
Collin Winterce36ad82007-08-30 01:19:48 +0000292 raise socket.error(msg)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000293 sock.listen(1)
294 port = sock.getsockname()[1] # Get proper port
295 host = self.sock.getsockname()[0] # Get proper host
296 if self.af == socket.AF_INET:
297 resp = self.sendport(host, port)
298 else:
299 resp = self.sendeprt(host, port)
Giampaolo Rodolà5fb313b2010-04-19 22:05:54 +0000300 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
301 sock.settimeout(self.timeout)
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000302 return sock
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000303
304 def makepasv(self):
Martin v. Löwis4eb59402001-07-26 13:37:33 +0000305 if self.af == socket.AF_INET:
306 host, port = parse227(self.sendcmd('PASV'))
307 else:
308 host, port = parse229(self.sendcmd('EPSV'), self.sock.getpeername())
309 return host, port
Guido van Rossum1115ab21992-11-04 15:51:30 +0000310
Tim Peters88869f92001-01-14 23:36:06 +0000311 def ntransfercmd(self, cmd, rest=None):
312 """Initiate a transfer over the data connection.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000313
Tim Peters88869f92001-01-14 23:36:06 +0000314 If the transfer is active, send a port command and the
315 transfer command, and accept the connection. If the server is
316 passive, send a pasv command, connect to it, and start the
317 transfer command. Either way, return the socket for the
318 connection and the expected size of the transfer. The
319 expected size may be None if it could not be determined.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000320
Tim Peters88869f92001-01-14 23:36:06 +0000321 Optional `rest' argument can be a string that is sent as the
Christian Heimes1af737c2008-01-23 08:24:23 +0000322 argument to a REST command. This is essentially a server
Tim Peters88869f92001-01-14 23:36:06 +0000323 marker used to tell the server to skip over any data up to the
324 given marker.
325 """
326 size = None
327 if self.passiveserver:
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000328 host, port = self.makepasv()
Guido van Rossumcd16bf62007-06-13 18:07:49 +0000329 conn = socket.create_connection((host, port), self.timeout)
Tim Peters88869f92001-01-14 23:36:06 +0000330 if rest is not None:
331 self.sendcmd("REST %s" % rest)
332 resp = self.sendcmd(cmd)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000333 # Some servers apparently send a 200 reply to
334 # a LIST or STOR command, before the 150 reply
335 # (and way before the 226 reply). This seems to
336 # be in violation of the protocol (which only allows
337 # 1xx or error messages for LIST), so we just discard
338 # this response.
339 if resp[0] == '2':
Thomas Wouters9fe394c2007-02-05 01:24:16 +0000340 resp = self.getresp()
Tim Peters88869f92001-01-14 23:36:06 +0000341 if resp[0] != '1':
Collin Winterce36ad82007-08-30 01:19:48 +0000342 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000343 else:
344 sock = self.makeport()
345 if rest is not None:
346 self.sendcmd("REST %s" % rest)
347 resp = self.sendcmd(cmd)
Thomas Wouters89f507f2006-12-13 04:49:30 +0000348 # See above.
349 if resp[0] == '2':
Thomas Wouters9fe394c2007-02-05 01:24:16 +0000350 resp = self.getresp()
Tim Peters88869f92001-01-14 23:36:06 +0000351 if resp[0] != '1':
Collin Winterce36ad82007-08-30 01:19:48 +0000352 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000353 conn, sockaddr = sock.accept()
Giampaolo Rodolà5fb313b2010-04-19 22:05:54 +0000354 if self.timeout is not _GLOBAL_DEFAULT_TIMEOUT:
355 conn.settimeout(self.timeout)
Tim Peters88869f92001-01-14 23:36:06 +0000356 if resp[:3] == '150':
357 # this is conditional in case we received a 125
358 size = parse150(resp)
359 return conn, size
Fred Drake4de02d91997-01-10 18:26:09 +0000360
Tim Peters88869f92001-01-14 23:36:06 +0000361 def transfercmd(self, cmd, rest=None):
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000362 """Like ntransfercmd() but returns only the socket."""
Tim Peters88869f92001-01-14 23:36:06 +0000363 return self.ntransfercmd(cmd, rest)[0]
Guido van Rossumc567c601992-11-05 22:22:37 +0000364
Tim Peters88869f92001-01-14 23:36:06 +0000365 def login(self, user = '', passwd = '', acct = ''):
366 '''Login, default anonymous.'''
367 if not user: user = 'anonymous'
368 if not passwd: passwd = ''
369 if not acct: acct = ''
370 if user == 'anonymous' and passwd in ('', '-'):
Tim Peterse4418602002-02-16 07:34:19 +0000371 # If there is no anonymous ftp password specified
372 # then we'll just use anonymous@
373 # We don't send any other thing because:
374 # - We want to remain anonymous
375 # - We want to stop SPAM
376 # - We don't want to let ftp sites to discriminate by the user,
377 # host or country.
Guido van Rossumc33e0772001-12-28 20:54:28 +0000378 passwd = passwd + 'anonymous@'
Tim Peters88869f92001-01-14 23:36:06 +0000379 resp = self.sendcmd('USER ' + user)
380 if resp[0] == '3': resp = self.sendcmd('PASS ' + passwd)
381 if resp[0] == '3': resp = self.sendcmd('ACCT ' + acct)
382 if resp[0] != '2':
Collin Winterce36ad82007-08-30 01:19:48 +0000383 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000384 return resp
Guido van Rossumc567c601992-11-05 22:22:37 +0000385
Tim Peters88869f92001-01-14 23:36:06 +0000386 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000387 """Retrieve data in binary mode. A new port is created for you.
Barry Warsaw100d81e2000-09-01 06:09:23 +0000388
Christian Heimes1af737c2008-01-23 08:24:23 +0000389 Args:
390 cmd: A RETR command.
391 callback: A single parameter callable to be called on each
392 block of data read.
393 blocksize: The maximum number of bytes to read from the
394 socket at one time. [default: 8192]
395 rest: Passed to transfercmd(). [default: None]
Guido van Rossum1115ab21992-11-04 15:51:30 +0000396
Christian Heimes1af737c2008-01-23 08:24:23 +0000397 Returns:
398 The response code.
Tim Peters88869f92001-01-14 23:36:06 +0000399 """
400 self.voidcmd('TYPE I')
401 conn = self.transfercmd(cmd, rest)
402 while 1:
403 data = conn.recv(blocksize)
404 if not data:
405 break
406 callback(data)
407 conn.close()
408 return self.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000409
Tim Peters88869f92001-01-14 23:36:06 +0000410 def retrlines(self, cmd, callback = None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000411 """Retrieve data in line mode. A new port is created for you.
412
413 Args:
414 cmd: A RETR, LIST, NLST, or MLSD command.
415 callback: An optional single parameter callable that is called
416 for each line with the trailing CRLF stripped.
417 [default: print_line()]
418
419 Returns:
420 The response code.
421 """
Raymond Hettingere874fc32002-05-12 05:53:51 +0000422 if callback is None: callback = print_line
Tim Peters88869f92001-01-14 23:36:06 +0000423 resp = self.sendcmd('TYPE A')
424 conn = self.transfercmd(cmd)
Guido van Rossum52b89762007-07-17 20:45:57 +0000425 fp = conn.makefile('r', encoding=self.encoding)
Tim Peters88869f92001-01-14 23:36:06 +0000426 while 1:
427 line = fp.readline()
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000428 if self.debugging > 2: print('*retr*', repr(line))
Tim Peters88869f92001-01-14 23:36:06 +0000429 if not line:
430 break
431 if line[-2:] == CRLF:
432 line = line[:-2]
433 elif line[-1:] == '\n':
434 line = line[:-1]
435 callback(line)
436 fp.close()
437 conn.close()
438 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000439
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000440 def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
Christian Heimes1af737c2008-01-23 08:24:23 +0000441 """Store a file in binary mode. A new port is created for you.
442
443 Args:
444 cmd: A STOR command.
445 fp: A file-like object with a read(num_bytes) method.
446 blocksize: The maximum data size to read from fp and send over
447 the connection at once. [default: 8192]
448 callback: An optional single parameter callable that is called on
449 on each block of data after it is sent. [default: None]
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000450 rest: Passed to transfercmd(). [default: None]
Christian Heimes1af737c2008-01-23 08:24:23 +0000451
452 Returns:
453 The response code.
454 """
Tim Peters88869f92001-01-14 23:36:06 +0000455 self.voidcmd('TYPE I')
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000456 conn = self.transfercmd(cmd, rest)
Tim Peters88869f92001-01-14 23:36:06 +0000457 while 1:
458 buf = fp.read(blocksize)
459 if not buf: break
Martin v. Löwise12454f2002-02-16 23:06:19 +0000460 conn.sendall(buf)
Christian Heimes1af737c2008-01-23 08:24:23 +0000461 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000462 conn.close()
463 return self.voidresp()
Guido van Rossumc567c601992-11-05 22:22:37 +0000464
Christian Heimes1af737c2008-01-23 08:24:23 +0000465 def storlines(self, cmd, fp, callback=None):
466 """Store a file in line mode. A new port is created for you.
467
468 Args:
469 cmd: A STOR command.
470 fp: A file-like object with a readline() method.
471 callback: An optional single parameter callable that is called on
472 on each line after it is sent. [default: None]
473
474 Returns:
475 The response code.
476 """
Tim Peters88869f92001-01-14 23:36:06 +0000477 self.voidcmd('TYPE A')
478 conn = self.transfercmd(cmd)
479 while 1:
480 buf = fp.readline()
481 if not buf: break
Benjamin Petersonbe17a112008-09-27 21:49:47 +0000482 if buf[-2:] != B_CRLF:
483 if buf[-1] in B_CRLF: buf = buf[:-1]
484 buf = buf + B_CRLF
Martin v. Löwise12454f2002-02-16 23:06:19 +0000485 conn.sendall(buf)
Christian Heimes1af737c2008-01-23 08:24:23 +0000486 if callback: callback(buf)
Tim Peters88869f92001-01-14 23:36:06 +0000487 conn.close()
488 return self.voidresp()
Guido van Rossum0eaa74b1996-01-25 18:37:21 +0000489
Tim Peters88869f92001-01-14 23:36:06 +0000490 def acct(self, password):
491 '''Send new account name.'''
492 cmd = 'ACCT ' + password
493 return self.voidcmd(cmd)
Guido van Rossumc567c601992-11-05 22:22:37 +0000494
Tim Peters88869f92001-01-14 23:36:06 +0000495 def nlst(self, *args):
496 '''Return a list of files in a given directory (default the current).'''
497 cmd = 'NLST'
498 for arg in args:
499 cmd = cmd + (' ' + arg)
500 files = []
501 self.retrlines(cmd, files.append)
502 return files
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000503
Tim Peters88869f92001-01-14 23:36:06 +0000504 def dir(self, *args):
505 '''List a directory in long form.
506 By default list current directory to stdout.
507 Optional last argument is callback function; all
508 non-empty arguments before it are concatenated to the
509 LIST command. (This *should* only be used for a pathname.)'''
510 cmd = 'LIST'
511 func = None
512 if args[-1:] and type(args[-1]) != type(''):
513 args, func = args[:-1], args[-1]
514 for arg in args:
515 if arg:
516 cmd = cmd + (' ' + arg)
517 self.retrlines(cmd, func)
Guido van Rossumc567c601992-11-05 22:22:37 +0000518
Tim Peters88869f92001-01-14 23:36:06 +0000519 def rename(self, fromname, toname):
520 '''Rename a file.'''
521 resp = self.sendcmd('RNFR ' + fromname)
522 if resp[0] != '3':
Collin Winterce36ad82007-08-30 01:19:48 +0000523 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000524 return self.voidcmd('RNTO ' + toname)
Guido van Rossuma61bdeb1995-10-11 17:36:31 +0000525
Tim Peters88869f92001-01-14 23:36:06 +0000526 def delete(self, filename):
527 '''Delete a file.'''
528 resp = self.sendcmd('DELE ' + filename)
529 if resp[:3] in ('250', '200'):
530 return resp
Tim Peters88869f92001-01-14 23:36:06 +0000531 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000532 raise error_reply(resp)
Guido van Rossum02cf5821993-05-17 08:00:02 +0000533
Tim Peters88869f92001-01-14 23:36:06 +0000534 def cwd(self, dirname):
535 '''Change to a directory.'''
536 if dirname == '..':
537 try:
538 return self.voidcmd('CDUP')
Guido van Rossumb940e112007-01-10 16:19:56 +0000539 except error_perm as msg:
Martin v. Löwisb5255112002-03-10 15:59:58 +0000540 if msg.args[0][:3] != '500':
541 raise
Tim Peters88869f92001-01-14 23:36:06 +0000542 elif dirname == '':
543 dirname = '.' # does nothing, but could return error
544 cmd = 'CWD ' + dirname
545 return self.voidcmd(cmd)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000546
Tim Peters88869f92001-01-14 23:36:06 +0000547 def size(self, filename):
548 '''Retrieve the size of a file.'''
Christian Heimes1af737c2008-01-23 08:24:23 +0000549 # The SIZE command is defined in RFC-3659
Tim Peters88869f92001-01-14 23:36:06 +0000550 resp = self.sendcmd('SIZE ' + filename)
551 if resp[:3] == '213':
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000552 s = resp[3:].strip()
553 try:
554 return int(s)
Guido van Rossum1f74cb32001-10-17 17:21:47 +0000555 except (OverflowError, ValueError):
Guido van Rossume2a383d2007-01-15 16:59:06 +0000556 return int(s)
Guido van Rossumc567c601992-11-05 22:22:37 +0000557
Tim Peters88869f92001-01-14 23:36:06 +0000558 def mkd(self, dirname):
559 '''Make a directory, return its full pathname.'''
560 resp = self.sendcmd('MKD ' + dirname)
561 return parse257(resp)
Guido van Rossum98245091998-02-19 21:15:44 +0000562
Tim Peters88869f92001-01-14 23:36:06 +0000563 def rmd(self, dirname):
564 '''Remove a directory.'''
565 return self.voidcmd('RMD ' + dirname)
Guido van Rossum1115ab21992-11-04 15:51:30 +0000566
Tim Peters88869f92001-01-14 23:36:06 +0000567 def pwd(self):
568 '''Return current working directory.'''
569 resp = self.sendcmd('PWD')
570 return parse257(resp)
Guido van Rossum17ed1ae1993-06-01 13:21:04 +0000571
Tim Peters88869f92001-01-14 23:36:06 +0000572 def quit(self):
573 '''Quit, and close the connection.'''
574 resp = self.voidcmd('QUIT')
575 self.close()
576 return resp
577
578 def close(self):
579 '''Close the connection without assuming anything about it.'''
Fred Drake9c98a422001-02-28 21:46:37 +0000580 if self.file:
581 self.file.close()
582 self.sock.close()
583 self.file = self.sock = None
Guido van Rossumc567c601992-11-05 22:22:37 +0000584
585
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000586try:
587 import ssl
588except ImportError:
589 pass
590else:
591 class FTP_TLS(FTP):
592 '''A FTP subclass which adds TLS support to FTP as described
593 in RFC-4217.
594
595 Connect as usual to port 21 implicitly securing the FTP control
596 connection before authenticating.
597
598 Securing the data connection requires user to explicitly ask
599 for it by calling prot_p() method.
600
601 Usage example:
602 >>> from ftplib import FTP_TLS
603 >>> ftps = FTP_TLS('ftp.python.org')
604 >>> ftps.login() # login anonimously previously securing control channel
605 '230 Guest login ok, access restrictions apply.'
606 >>> ftps.prot_p() # switch to secure data connection
607 '200 Protection level set to P'
608 >>> ftps.retrlines('LIST') # list directory content securely
609 total 9
610 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 .
611 drwxr-xr-x 8 root wheel 1024 Jan 3 1994 ..
612 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 bin
613 drwxr-xr-x 2 root wheel 1024 Jan 3 1994 etc
614 d-wxrwxr-x 2 ftp wheel 1024 Sep 5 13:43 incoming
615 drwxr-xr-x 2 root wheel 1024 Nov 17 1993 lib
616 drwxr-xr-x 6 1094 wheel 1024 Sep 13 19:07 pub
617 drwxr-xr-x 3 root wheel 1024 Jan 3 1994 usr
618 -rw-r--r-- 1 root root 312 Aug 1 1994 welcome.msg
619 '226 Transfer complete.'
620 >>> ftps.quit()
621 '221 Goodbye.'
622 >>>
623 '''
624 ssl_version = ssl.PROTOCOL_TLSv1
625
626 def __init__(self, host='', user='', passwd='', acct='', keyfile=None,
627 certfile=None, timeout=_GLOBAL_DEFAULT_TIMEOUT):
628 self.keyfile = keyfile
629 self.certfile = certfile
630 self._prot_p = False
631 FTP.__init__(self, host, user, passwd, acct, timeout)
632
633 def login(self, user='', passwd='', acct='', secure=True):
634 if secure and not isinstance(self.sock, ssl.SSLSocket):
635 self.auth()
636 return FTP.login(self, user, passwd, acct)
637
638 def auth(self):
639 '''Set up secure control connection by using TLS/SSL.'''
640 if isinstance(self.sock, ssl.SSLSocket):
641 raise ValueError("Already using TLS")
642 if self.ssl_version == ssl.PROTOCOL_TLSv1:
643 resp = self.voidcmd('AUTH TLS')
644 else:
645 resp = self.voidcmd('AUTH SSL')
646 self.sock = ssl.wrap_socket(self.sock, self.keyfile, self.certfile,
647 ssl_version=self.ssl_version)
648 self.file = self.sock.makefile(mode='r', encoding=self.encoding)
649 return resp
650
651 def prot_p(self):
652 '''Set up secure data connection.'''
653 # PROT defines whether or not the data channel is to be protected.
654 # Though RFC-2228 defines four possible protection levels,
655 # RFC-4217 only recommends two, Clear and Private.
656 # Clear (PROT C) means that no security is to be used on the
657 # data-channel, Private (PROT P) means that the data-channel
658 # should be protected by TLS.
659 # PBSZ command MUST still be issued, but must have a parameter of
660 # '0' to indicate that no buffering is taking place and the data
661 # connection should not be encapsulated.
662 self.voidcmd('PBSZ 0')
663 resp = self.voidcmd('PROT P')
664 self._prot_p = True
665 return resp
666
667 def prot_c(self):
668 '''Set up clear text data connection.'''
669 resp = self.voidcmd('PROT C')
670 self._prot_p = False
671 return resp
672
673 # --- Overridden FTP methods
674
675 def ntransfercmd(self, cmd, rest=None):
676 conn, size = FTP.ntransfercmd(self, cmd, rest)
677 if self._prot_p:
678 conn = ssl.wrap_socket(conn, self.keyfile, self.certfile,
679 ssl_version=self.ssl_version)
680 return conn, size
681
682 def retrbinary(self, cmd, callback, blocksize=8192, rest=None):
683 self.voidcmd('TYPE I')
684 conn = self.transfercmd(cmd, rest)
685 try:
686 while 1:
687 data = conn.recv(blocksize)
688 if not data:
689 break
690 callback(data)
691 # shutdown ssl layer
692 if isinstance(conn, ssl.SSLSocket):
693 conn.unwrap()
694 finally:
695 conn.close()
696 return self.voidresp()
697
698 def retrlines(self, cmd, callback = None):
699 if callback is None: callback = print_line
700 resp = self.sendcmd('TYPE A')
701 conn = self.transfercmd(cmd)
702 fp = conn.makefile('r', encoding=self.encoding)
703 try:
704 while 1:
705 line = fp.readline()
706 if self.debugging > 2: print('*retr*', repr(line))
707 if not line:
708 break
709 if line[-2:] == CRLF:
710 line = line[:-2]
711 elif line[-1:] == '\n':
712 line = line[:-1]
713 callback(line)
714 # shutdown ssl layer
715 if isinstance(conn, ssl.SSLSocket):
716 conn.unwrap()
717 finally:
718 fp.close()
719 conn.close()
720 return self.voidresp()
721
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000722 def storbinary(self, cmd, fp, blocksize=8192, callback=None, rest=None):
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000723 self.voidcmd('TYPE I')
Antoine Pitrou648bcd72009-11-27 13:23:26 +0000724 conn = self.transfercmd(cmd, rest)
Antoine Pitrouf988cd02009-11-17 20:21:14 +0000725 try:
726 while 1:
727 buf = fp.read(blocksize)
728 if not buf: break
729 conn.sendall(buf)
730 if callback: callback(buf)
731 # shutdown ssl layer
732 if isinstance(conn, ssl.SSLSocket):
733 conn.unwrap()
734 finally:
735 conn.close()
736 return self.voidresp()
737
738 def storlines(self, cmd, fp, callback=None):
739 self.voidcmd('TYPE A')
740 conn = self.transfercmd(cmd)
741 try:
742 while 1:
743 buf = fp.readline()
744 if not buf: break
745 if buf[-2:] != B_CRLF:
746 if buf[-1] in B_CRLF: buf = buf[:-1]
747 buf = buf + B_CRLF
748 conn.sendall(buf)
749 if callback: callback(buf)
750 # shutdown ssl layer
751 if isinstance(conn, ssl.SSLSocket):
752 conn.unwrap()
753 finally:
754 conn.close()
755 return self.voidresp()
756
757 __all__.append('FTP_TLS')
758 all_errors = (Error, IOError, EOFError, ssl.SSLError)
759
760
Guido van Rossumacfb82a1997-10-22 20:49:52 +0000761_150_re = None
Fred Drake4de02d91997-01-10 18:26:09 +0000762
763def parse150(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000764 '''Parse the '150' response for a RETR request.
765 Returns the expected transfer size or None; size is not guaranteed to
766 be present in the 150 message.
767 '''
768 if resp[:3] != '150':
Collin Winterce36ad82007-08-30 01:19:48 +0000769 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000770 global _150_re
771 if _150_re is None:
772 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000773 _150_re = re.compile(
774 "150 .* \((\d+) bytes\)", re.IGNORECASE | re.ASCII)
Tim Peters88869f92001-01-14 23:36:06 +0000775 m = _150_re.match(resp)
Guido van Rossumb6aca6a2001-10-16 19:45:52 +0000776 if not m:
777 return None
778 s = m.group(1)
779 try:
780 return int(s)
Guido van Rossum1f74cb32001-10-17 17:21:47 +0000781 except (OverflowError, ValueError):
Guido van Rossume2a383d2007-01-15 16:59:06 +0000782 return int(s)
Fred Drake4de02d91997-01-10 18:26:09 +0000783
784
Guido van Rossum70297d32001-08-17 17:24:29 +0000785_227_re = None
786
Guido van Rossumd2560b01996-05-28 23:41:25 +0000787def parse227(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000788 '''Parse the '227' response for a PASV request.
789 Raises error_proto if it does not contain '(h1,h2,h3,h4,p1,p2)'
790 Return ('host.addr.as.numbers', port#) tuple.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000791
Tim Peters88869f92001-01-14 23:36:06 +0000792 if resp[:3] != '227':
Collin Winterce36ad82007-08-30 01:19:48 +0000793 raise error_reply(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000794 global _227_re
795 if _227_re is None:
796 import re
Antoine Pitroufd036452008-08-19 17:56:33 +0000797 _227_re = re.compile(r'(\d+),(\d+),(\d+),(\d+),(\d+),(\d+)', re.ASCII)
Guido van Rossum70297d32001-08-17 17:24:29 +0000798 m = _227_re.search(resp)
799 if not m:
Collin Winterce36ad82007-08-30 01:19:48 +0000800 raise error_proto(resp)
Guido van Rossum70297d32001-08-17 17:24:29 +0000801 numbers = m.groups()
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000802 host = '.'.join(numbers[:4])
803 port = (int(numbers[4]) << 8) + int(numbers[5])
Tim Peters88869f92001-01-14 23:36:06 +0000804 return host, port
Guido van Rossumd2560b01996-05-28 23:41:25 +0000805
806
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000807def parse229(resp, peer):
808 '''Parse the '229' response for a EPSV request.
809 Raises error_proto if it does not contain '(|||port|)'
810 Return ('host.addr.as.numbers', port#) tuple.'''
811
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000812 if resp[:3] != '229':
Collin Winterce36ad82007-08-30 01:19:48 +0000813 raise error_reply(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000814 left = resp.find('(')
Collin Winterce36ad82007-08-30 01:19:48 +0000815 if left < 0: raise error_proto(resp)
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000816 right = resp.find(')', left + 1)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000817 if right < 0:
Collin Winterce36ad82007-08-30 01:19:48 +0000818 raise error_proto(resp) # should contain '(|||port|)'
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000819 if resp[left + 1] != resp[right - 1]:
Collin Winterce36ad82007-08-30 01:19:48 +0000820 raise error_proto(resp)
Walter Dörwalda401ae42002-06-03 10:41:45 +0000821 parts = resp[left + 1:right].split(resp[left+1])
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000822 if len(parts) != 5:
Collin Winterce36ad82007-08-30 01:19:48 +0000823 raise error_proto(resp)
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000824 host = peer[0]
Neal Norwitz7ce734c2002-05-31 14:13:04 +0000825 port = int(parts[3])
Martin v. Löwisa43c2f82001-07-24 20:34:08 +0000826 return host, port
827
828
Guido van Rossumc567c601992-11-05 22:22:37 +0000829def parse257(resp):
Tim Peters88869f92001-01-14 23:36:06 +0000830 '''Parse the '257' response for a MKD or PWD request.
831 This is a response to a MKD or PWD request: a directory name.
832 Returns the directoryname in the 257 reply.'''
Guido van Rossumd2560b01996-05-28 23:41:25 +0000833
Tim Peters88869f92001-01-14 23:36:06 +0000834 if resp[:3] != '257':
Collin Winterce36ad82007-08-30 01:19:48 +0000835 raise error_reply(resp)
Tim Peters88869f92001-01-14 23:36:06 +0000836 if resp[3:5] != ' "':
837 return '' # Not compliant to RFC 959, but UNIX ftpd does this
838 dirname = ''
839 i = 5
840 n = len(resp)
841 while i < n:
842 c = resp[i]
843 i = i+1
844 if c == '"':
845 if i >= n or resp[i] != '"':
846 break
847 i = i+1
848 dirname = dirname + c
849 return dirname
Guido van Rossum1115ab21992-11-04 15:51:30 +0000850
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000851
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000852def print_line(line):
Tim Peters88869f92001-01-14 23:36:06 +0000853 '''Default retrlines callback to print a line.'''
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000854 print(line)
Guido van Rossumae3b3a31993-11-30 13:43:54 +0000855
Guido van Rossum2f3941d1997-10-07 14:49:56 +0000856
Guido van Rossumd2560b01996-05-28 23:41:25 +0000857def ftpcp(source, sourcename, target, targetname = '', type = 'I'):
Tim Peters88869f92001-01-14 23:36:06 +0000858 '''Copy file from one FTP-instance to another.'''
859 if not targetname: targetname = sourcename
860 type = 'TYPE ' + type
861 source.voidcmd(type)
862 target.voidcmd(type)
863 sourcehost, sourceport = parse227(source.sendcmd('PASV'))
864 target.sendport(sourcehost, sourceport)
865 # RFC 959: the user must "listen" [...] BEFORE sending the
866 # transfer request.
867 # So: STOR before RETR, because here the target is a "user".
868 treply = target.sendcmd('STOR ' + targetname)
869 if treply[:3] not in ('125', '150'): raise error_proto # RFC 959
870 sreply = source.sendcmd('RETR ' + sourcename)
871 if sreply[:3] not in ('125', '150'): raise error_proto # RFC 959
872 source.voidresp()
873 target.voidresp()
Guido van Rossum1115ab21992-11-04 15:51:30 +0000874
Tim Peters88869f92001-01-14 23:36:06 +0000875
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000876class Netrc:
Tim Peters88869f92001-01-14 23:36:06 +0000877 """Class to parse & provide access to 'netrc' format files.
Fred Drake475d51d1997-06-24 22:02:54 +0000878
Tim Peters88869f92001-01-14 23:36:06 +0000879 See the netrc(4) man page for information on the file format.
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000880
Tim Peters88869f92001-01-14 23:36:06 +0000881 WARNING: This class is obsolete -- use module netrc instead.
Guido van Rossumc822a451998-12-22 16:49:16 +0000882
Tim Peters88869f92001-01-14 23:36:06 +0000883 """
884 __defuser = None
885 __defpasswd = None
886 __defacct = None
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000887
Tim Peters88869f92001-01-14 23:36:06 +0000888 def __init__(self, filename=None):
Raymond Hettinger094662a2002-06-01 01:29:16 +0000889 if filename is None:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000890 if "HOME" in os.environ:
Tim Peters88869f92001-01-14 23:36:06 +0000891 filename = os.path.join(os.environ["HOME"],
892 ".netrc")
893 else:
Collin Winterce36ad82007-08-30 01:19:48 +0000894 raise IOError("specify file to load or set $HOME")
Tim Peters88869f92001-01-14 23:36:06 +0000895 self.__hosts = {}
896 self.__macros = {}
897 fp = open(filename, "r")
898 in_macro = 0
899 while 1:
900 line = fp.readline()
901 if not line: break
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000902 if in_macro and line.strip():
Tim Peters88869f92001-01-14 23:36:06 +0000903 macro_lines.append(line)
904 continue
905 elif in_macro:
906 self.__macros[macro_name] = tuple(macro_lines)
907 in_macro = 0
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000908 words = line.split()
Tim Peters88869f92001-01-14 23:36:06 +0000909 host = user = passwd = acct = None
910 default = 0
911 i = 0
912 while i < len(words):
913 w1 = words[i]
914 if i+1 < len(words):
915 w2 = words[i + 1]
916 else:
917 w2 = None
918 if w1 == 'default':
919 default = 1
920 elif w1 == 'machine' and w2:
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000921 host = w2.lower()
Tim Peters88869f92001-01-14 23:36:06 +0000922 i = i + 1
923 elif w1 == 'login' and w2:
924 user = w2
925 i = i + 1
926 elif w1 == 'password' and w2:
927 passwd = w2
928 i = i + 1
929 elif w1 == 'account' and w2:
930 acct = w2
931 i = i + 1
932 elif w1 == 'macdef' and w2:
933 macro_name = w2
934 macro_lines = []
935 in_macro = 1
936 break
937 i = i + 1
938 if default:
939 self.__defuser = user or self.__defuser
940 self.__defpasswd = passwd or self.__defpasswd
941 self.__defacct = acct or self.__defacct
942 if host:
Raymond Hettinger54f02222002-06-01 14:18:47 +0000943 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +0000944 ouser, opasswd, oacct = \
945 self.__hosts[host]
946 user = user or ouser
947 passwd = passwd or opasswd
948 acct = acct or oacct
949 self.__hosts[host] = user, passwd, acct
950 fp.close()
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000951
Tim Peters88869f92001-01-14 23:36:06 +0000952 def get_hosts(self):
953 """Return a list of hosts mentioned in the .netrc file."""
954 return self.__hosts.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +0000955
Tim Peters88869f92001-01-14 23:36:06 +0000956 def get_account(self, host):
957 """Returns login information for the named host.
Guido van Rossum8ca84201998-03-26 20:56:10 +0000958
Tim Peters88869f92001-01-14 23:36:06 +0000959 The return value is a triple containing userid,
960 password, and the accounting field.
Guido van Rossum8ca84201998-03-26 20:56:10 +0000961
Tim Peters88869f92001-01-14 23:36:06 +0000962 """
Eric S. Raymondc95bf692001-02-09 10:06:47 +0000963 host = host.lower()
Tim Peters88869f92001-01-14 23:36:06 +0000964 user = passwd = acct = None
Raymond Hettinger54f02222002-06-01 14:18:47 +0000965 if host in self.__hosts:
Tim Peters88869f92001-01-14 23:36:06 +0000966 user, passwd, acct = self.__hosts[host]
967 user = user or self.__defuser
968 passwd = passwd or self.__defpasswd
969 acct = acct or self.__defacct
970 return user, passwd, acct
Guido van Rossum8ca84201998-03-26 20:56:10 +0000971
Tim Peters88869f92001-01-14 23:36:06 +0000972 def get_macros(self):
973 """Return a list of all defined macro names."""
974 return self.__macros.keys()
Guido van Rossum8ca84201998-03-26 20:56:10 +0000975
Tim Peters88869f92001-01-14 23:36:06 +0000976 def get_macro(self, macro):
977 """Return a sequence of lines which define a named macro."""
978 return self.__macros[macro]
Guido van Rossum56d1e3a1997-03-14 04:16:54 +0000979
Fred Drake475d51d1997-06-24 22:02:54 +0000980
Tim Peters88869f92001-01-14 23:36:06 +0000981
Guido van Rossum1115ab21992-11-04 15:51:30 +0000982def test():
Tim Peters88869f92001-01-14 23:36:06 +0000983 '''Test program.
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000984 Usage: ftp [-d] [-r[file]] host [-l[dir]] [-d[dir]] [-p] [file] ...
985
986 -d dir
987 -l list
988 -p password
989 '''
990
991 if len(sys.argv) < 2:
Guido van Rossumbe19ed72007-02-09 05:37:30 +0000992 print(test.__doc__)
Raymond Hettingerc88a6c72005-04-05 04:31:09 +0000993 sys.exit(0)
Guido van Rossumd2560b01996-05-28 23:41:25 +0000994
Tim Peters88869f92001-01-14 23:36:06 +0000995 debugging = 0
996 rcfile = None
997 while sys.argv[1] == '-d':
998 debugging = debugging+1
999 del sys.argv[1]
1000 if sys.argv[1][:2] == '-r':
1001 # get name of alternate ~/.netrc file:
1002 rcfile = sys.argv[1][2:]
1003 del sys.argv[1]
1004 host = sys.argv[1]
1005 ftp = FTP(host)
1006 ftp.set_debuglevel(debugging)
1007 userid = passwd = acct = ''
1008 try:
1009 netrc = Netrc(rcfile)
1010 except IOError:
1011 if rcfile is not None:
1012 sys.stderr.write("Could not open account file"
1013 " -- using anonymous login.")
1014 else:
1015 try:
1016 userid, passwd, acct = netrc.get_account(host)
1017 except KeyError:
1018 # no account for host
1019 sys.stderr.write(
1020 "No account -- using anonymous login.")
1021 ftp.login(userid, passwd, acct)
1022 for file in sys.argv[2:]:
1023 if file[:2] == '-l':
1024 ftp.dir(file[2:])
1025 elif file[:2] == '-d':
1026 cmd = 'CWD'
1027 if file[2:]: cmd = cmd + ' ' + file[2:]
1028 resp = ftp.sendcmd(cmd)
1029 elif file == '-p':
1030 ftp.set_pasv(not ftp.passiveserver)
1031 else:
1032 ftp.retrbinary('RETR ' + file, \
1033 sys.stdout.write, 1024)
1034 ftp.quit()
Guido van Rossum221ec0b1995-08-04 04:39:30 +00001035
1036
1037if __name__ == '__main__':
Tim Peters88869f92001-01-14 23:36:06 +00001038 test()