ftp.py 3.54 KB
Newer Older
Guido van Rossum's avatar
Guido van Rossum committed
1 2 3 4 5 6
# A simple FTP client.
#
# The information to write this program was gathered from RFC 959,
# but this is not a complete implementation!  Yet it shows how a simple
# FTP client can be built, and you are welcome to extend it to suit
# it to your needs...
7 8 9 10 11 12 13 14 15 16 17 18
#
# How it works (assuming you've read the RFC):
#
# User commands are passed uninterpreted to the server.  However, the
# user never needs to send a PORT command.  Rather, the client opens a
# port right away and sends the appropriate PORT command to the server.
# When a response code 150 is received, this port is used to receive
# the data (which is written to stdout in this version), and when the
# data is exhausted, a new port is opened and a corresponding PORT
# command sent.  In order to avoid errors when reusing ports quickly
# (and because there is no s.getsockname() method in Python yet) we
# cycle through a number of ports in the 50000 range.
Guido van Rossum's avatar
Guido van Rossum committed
19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50


import sys, posix, string
from socket import *


BUFSIZE = 1024

# Default port numbers used by the FTP protocol.
#
FTP_PORT = 21
FTP_DATA_PORT = FTP_PORT - 1

# Change the data port to something not needing root permissions.
#
FTP_DATA_PORT = FTP_DATA_PORT + 50000


# Main program (called at the end of this file).
#
def main():
	hostname = sys.argv[1]
	control(hostname)


# Control process (user interface and user protocol interpreter).
#
def control(hostname):
	#
	# Create control connection
	#
	s = socket(AF_INET, SOCK_STREAM)
51
	s.connect((hostname, FTP_PORT))
Guido van Rossum's avatar
Guido van Rossum committed
52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74
	f = s.makefile('r') # Reading the replies is easier from a file...
	#
	# Control loop
	#
	r = None
	while 1:
		code = getreply(f)
		if code in ('221', 'EOF'): break
		if code == '150':
			getdata(r)
			code = getreply(f)
			r = None
		if not r:
			r = newdataport(s, f)
		cmd = getcommand()
		if not cmd: break
		s.send(cmd + '\r\n')


# Create a new data port and send a PORT command to the server for it.
# (Cycle through a number of ports to avoid problems with reusing
# a port within a short time.)
#
75
nextport = 0
Guido van Rossum's avatar
Guido van Rossum committed
76 77
#
def newdataport(s, f):
78 79 80
	global nextport
	port = nextport + FTP_DATA_PORT
	nextport = (nextport+1) % 16
Guido van Rossum's avatar
Guido van Rossum committed
81
	r = socket(AF_INET, SOCK_STREAM)
82
	r.bind((gethostbyname(gethostname()), port))
83
	r.listen(1)
Guido van Rossum's avatar
Guido van Rossum committed
84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116
	sendportcmd(s, f, port)
	return r


# Send an appropriate port command.
#
def sendportcmd(s, f, port):
	hostname = gethostname()
	hostaddr = gethostbyname(hostname)
	hbytes = string.splitfields(hostaddr, '.')
	pbytes = [`port/256`, `port%256`]
	bytes = hbytes + pbytes
	cmd = 'PORT ' + string.joinfields(bytes, ',')
	s.send(cmd + '\r\n')
	code = getreply(f)


# Process an ftp reply and return the 3-digit reply code (as a string).
# The reply should be a line of text starting with a 3-digit number.
# If the 4th char is '-', it is a multi-line reply and is
# terminate by a line starting with the same 3-digit number.
# Any text while receiving the reply is echoed to the file.
#
def getreply(f):
	line = f.readline()
	if not line: return 'EOF'
	print line,
	code = line[:3]
	if line[3:4] == '-':
		while 1:
			line = f.readline()
			if not line: break # Really an error
			print line,
117
			if line[:3] == code and line[3:4] != '-': break
Guido van Rossum's avatar
Guido van Rossum committed
118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136
	return code


# Get the data from the data connection.
#
def getdata(r):
	print '(accepting data connection)'
	conn, host = r.accept()
	print '(data connection accepted)'
	while 1:
		data = conn.recv(BUFSIZE)
		if not data: break
		sys.stdout.write(data)
	print '(end of data connection)'

# Get a command from the user.
#
def getcommand():
	try:
137 138 139
		while 1:
			line = raw_input('ftp.py> ')
			if line: return line
Guido van Rossum's avatar
Guido van Rossum committed
140 141 142 143 144 145 146
	except EOFError:
		return ''


# Call the main program.
#
main()