文件可以传输,但是对比传输前后的文件:socket_test.txt,末尾有一些不一致服务端代码:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
|
#!/usr/bin/python # -*- coding: utf-8 -*- import sys reload (sys) sys.setdefaultencoding( "utf-8" ) import time ''' 等待连接 等待发送文件 读取数据 写入文件并且保存 等待连接 ''' 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_传送的文件 filename = filename.strip( '\00' ) fp = open (filename, 'wb' ) # 新建文件,并且准备写入 restsize = filesize print "recving..." while 1 : if restsize > 102400 : # 如果剩余数据包大于1024,就去1024的数据包 filedata = newsock.recv( 10240 ) else : filedata = newsock.recv(restsize) fp.write(filedata) #break if not filedata: break fp.write(filedata) restsize = restsize - len (filedata) # 计算剩余数据包大小 if restsize < = 0 : break fp.close() print "recv succeeded !!File named:" , filename except Exception, e: print unicode (e).encode( 'gbk' ) print "the socket partner maybe closed" newsock.close() break sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) # 创建tcp连接 sock.bind(( '10.240.146.82' , 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' |
客户端代码:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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
51
52
|
#!/usr/bin/python # -*- coding: utf-8 -*- import sys reload (sys) sys.setdefaultencoding( "utf-8" ) ''' 输入文件名,并且上传 ''' import socket import time import struct import os f = open ( 'socket_test.txt' , 'wb' ) for i in range ( 1000000 ): f.write( 'for socket test, the line number is : ' + str (i) + '\n' ) f.close() sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM) sock.settimeout( 50 ) e = 0 try : sock.connect(( '10.240.146.82' , 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------->') # 输入文件名 filename = 'socket_test.txt' FILEINFO_SIZE = struct.calcsize( '128sI' ) # 编码格式大小 fhead = struct.pack( '128sI' , filename, os.stat(filename).st_size) # 按照规则进行打包 sock.send(fhead) # 发送文件基本信息数据 fp = open (filename, 'rb' ) fp2 = open ( 'local_test.txt' , 'wb' ) i = 0 while 1 : # 发送文件 filedata = fp.read( 10240 ) if not filedata: break sock.sendall(filedata) fp2.write(filedata) print i i = i + 1 print "sending over..." fp.close() fp2.close() |
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持服务器之家。
原文链接:https://segmentfault.com/q/1010000006727022/a-1020000006728168