代码均已测试:希望高人能指点一二 服务端: #coding=gbk ''' 等待连接 等待发送文件 读取数据 写入文件并且保存 等待连接 ''' import socket import threading import time import struct def function(newsock, address): FILEINFO_SIZE = struct.calcsize('128sI') '''定义文件信息(包含文件名和文件大小)大小。128s代表128个char[](文件名),I代表一个integer or long(文件大小)''' while 1: try: fhead = newsock.recv(FILEINFO_SIZE) filename, filesize = struct.unpack('128sI', fhead) '''把接收到的数据库进行解包,按照打包规则128sI''' print "address is: ",address print filename, len(filename),type(filename) print filesize filename = 'new_'+filename.strip('\00')#命名新文件new_传送的文件 fp = open(filename,'wb')#新建文件,并且准备写入 restsize = filesize print "recving..." while 1: if restsize > 1024:#如果剩余数据包大于1024,就去1024的数据包 filedata = newsock.recv(1024) else: filedata = newsock.recv(restsize) fp.write(filedata) break if not filedata: break fp.write(filedata) restsize = testsize - len(filedata)#计算剩余数据包大小 if restsize <= 0: break fp.close() print "recv succeeded !!File named:",filename except: print "he socket partner maybe closed" newsock.close() break sock=socket.socket(socket.AF_INET, socket.SOCK_STREAM)#创建tcp连接 sock.bind(('127.0.0.1',8887))#定于端口和ip sock.listen(5)#监听 while True: newsock, address = sock.accept() print "accept another connection" tmpThread = threading.Thread(target=function,args=(newsock,address)) #如果接收到文件,创建线程 tmpThread.start()#执行线程 print 'end'
|
客户端:
#coding=gbk
'''
输入文件名,并且上传
'''
import socket
import time
import struct
import os
sock = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
sock.settimeout(1)
e=0
try:
sock.connect(('127.0.0.1',8887))
print 'connect...'
except socket.timeout,e:
print 'timeout',e
except socket.error,e:
print 'error',e
except e:
print 'any',e
if not e:
while (1):
filename = raw_input('input your filename------->')#输入文件名
FILEINFO_SIZE = struct.calcsize('128sI')#编码格式大小
fhead = struct.pack('128sI',filename,os.stat(filename).st_size)#按照规则进行打包
sock.send(fhead)#发送文件基本信息数据
fp = open(filename,'rb')
while 1: #发送文件
filedata = fp.read(1024)
if not filedata:
break
sock.send(filedata)
print "sending over..."
fp.close() |