在python用socket传送文件

putget_server.py

#encoding=utf-8
import socket
import common
import sys
import thread
import threading

class connection:
def __init__(self, s):
self.socket = s
self.status = "Read"
self.data = ""
self.data_len = 0

def read_command(self):
buf = ""
while True:
data = self.socket.recv(1024)
buf += data
if buf.find("/r/n/r/n") != -1:
pos = buf.find("/r/n/r/n")
self.command = buf[:pos]
self.data += buf[pos+len("/r/n/r/n"):]
self.data_len = len(self.data)
return buf

def read_n_byte(self):
self.data = self.socket.recv(1024)
self.data_len = len(self.data)

def send_file(self, filename):
fd = open(filename,"rb")
fd.seek(0, 2)
flen = fd.tell()
fd.seek(0, 0)
self.socket.send("%d/r/n/r/n"%(flen))
rest = flen
while rest >0:
buf = fd.read(1024)
rest -= len(buf)
self.socket.send(buf)

def do_socket(s):
conn = connection(s)
buf = conn.read_command()
commands = conn.command.split()
if commands[0] == "get":
conn.send_file(commands[1])
conn.socket.close()
elif commands[0] == "put":
fd = open(commands[1], "wb")
n = (long)(commands[2])
n -= conn.data_len
fd.write(conn.data)
while n>0:
conn.read_n_byte()
n -= conn.data_len
fd.write(conn.data)
fd.close()

if __name__=='__main__':
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.bind(('localhost', 9999))
sock.listen(5)
while True:
conn, addr = sock.accept()
print addr
try:
thread.start_new_thread(do_socket,(conn,))
#do_socket(conn)
except Exception,e:
print e


get.py

#encoding=utf-8
import socket
import sys

if __name__=='__main__':
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect((sys.argv[1], (int)(sys.argv[2])))
filename = sys.argv[3]
buf = "get %s/r/n/r/n"%(filename)
sock.send(buf)
data = ""
while True:
data += sock.recv(1024)
if data.find("/r/n/r/n") != -1:
break
pos = data.find("/r/n/r/n")
length = (int)(data[:pos])
data = data[pos+4:]
fd = open(filename, "wb")
fd.write(data)
rest = length - len(data)
while rest>0:
data = sock.recv(1024)
fd.write(data)
rest -= len(data)
fd.close()
sock.close()


put.py

#encoding=utf-8
import socket
import sys

if __name__=='__main__':
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
sock.connect((sys.argv[1], (int)(sys.argv[2])))
filename = sys.argv[3]
fd = open(filename, "rb")
fd.seek(0,2)
length = fd.tell()
fd.seek(0,0)
buf = "put %s %d/r/n/r/n"%(filename,length)
sock.send(buf)
rest = length
while rest>0:
buf = fd.read(1024)
rest -= len(buf)
sock.send(buf)
sock.close()


http://bbs.superlinux.cn/viewthread.php?tid=2559&extra=page%3D1&frombbs=1