Python 使用Socket提供对TCP / UDP网络通信的支持。
TCP
client
TCP是带连接的可靠通信协议。创建TCP连接时,主动发起连接的称为客户端,被动响应连接称为服务器。
示例,请求博客园首页:
import socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(('www.cnblogs.com',80))
s.send(b'GET / HTTP/1.1\r\nHost: www.cnblogs.com\r\nConnection: close\r\n\r\n')
# receive
buf = list()
while True:
d = s.recv(1024)
if d:
buf.append(d)
else:
break
print(buf)
s.close()
说明:
- socket.socket()
建立socket连接。
socket.AF_INET代表使用IPv4协议,socket.AF_INET6使用更新的IPv6协议。
SOCK_STREAM指定使用面向流的TCP协议。
- s.connect()
进行socket连接,参数为包含url和port的元组。
- s.send(str)
通过socket发送数据。
- s.recv(max)
接收返回的数据,max指定最大字节数。
server
服务端程序监听接口,拦截请求然后进行响应。
先编写一个client:
import socket
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.connect(('127.0.0.1', 8081))
print(s.recv(1024).decode('utf-8'))
s.send('Finley')
s.close()
server代码:
import socket
import threading
def respond(sock, addr):
buf = list()
while True:
d = sock.recv(1024)
if d:
buf.append(d)
else:
break
data = 'Hello, ' + buf
sock.send(data)
if __name__ == '__main__':
s = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
s.bind(('127.0.0.1', 8081))
s.listen(5)
while True:
sock, addr = s.accept()
s.send('Hello from server')
t = threading.Thread(target=respond, args=(sock, addr))
t.start()
s.close()
很多地方与client一致,主要多了监听端口所需的API:
- bind()
接受一个包含URL和port的元组作为参数,绑定该端口
- listen(max)
执行监听, 阻塞式调用max为最大连接数
- accept()
建立连接
UDP
UDP是无连接的用户数据报协议,通常用来发送要求速度快不要求可靠到达的数据,如检测用户是否在线。
UDP的socket为SOCK_DGRAM,udp无需连接,服务器无需绑定接口进行监听。
client:
import socket
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
data = 'hello'
s.sendto(data, ('127.0.0.1', 9999))
print(s.recv(1024).decode('utf-8'))
s.close()
使用sendto(data, addr_tuple)发送数据报
server:
import socket
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.bind(('127.0.0.1', 9999))
print('bind udp on 9999')
data, addr = s.recvfrom(1024)
print('Received from %s:%s.' % addr)
s.sendto(b'Hello, %s!' % data, addr)
s.close()
使用recvfrom(max)接收发送来的数据。
实例:发送端可以不停的发送新文件,接收端可以不停的接收新文件。
例如:发送端输入:e:\visio.rar,接收端会默认保存为 e:\new_visio.rar,支持多并发,具体实现如下;
接收端:
方法一:
#-*- coding: UTF-8 -*-
import socket,time,SocketServer,struct,os,thread
host='192.168.50.74'
port=12307
s=socket.socket(socket.AF_INET,socket.SOCK_STREAM) #定义socket类型
s.bind((host,port)) #绑定需要监听的Ip和端口号,tuple格式
s.listen(1)
def conn_thread(connection,address):
while True:
try:
connection.settimeout(600)
fileinfo_size=struct.calcsize('128sl')
buf = connection.recv(fileinfo_size)
if buf: #如果不加这个if,第一个文件传输完成后会自动走到下一句
filename,filesize =struct.unpack('128sl',buf)
filename_f = filename.strip('\00')
filenewname = os.path.join('e:\\',('new_'+ filename_f))
print 'file new name is %s, filesize is %s' %(filenewname,filesize)
recvd_size = 0 #定义接收了的文件大小
file = open(filenewname,'wb')
print 'stat receiving...'
while not recvd_size == filesize:
if filesize - recvd_size > 1024:
rdata = connection.recv(1024)
recvd_size += len(rdata)
else:
rdata = connection.recv(filesize - recvd_size)
recvd_size = filesize
file.write(rdata)
file.close()
print 'receive done'
#connection.close()
except socket.timeout:
connection.close()
while True:
connection,address=s.accept()
print('Connected by ',address)
#thread = threading.Thread(target=conn_thread,args=(connection,address)) #使用threading也可以
#thread.start()
thread.start_new_thread(conn_thread,(connection,address))
s.close()
方法二:
#-*- coding: UTF-8 -*-
import socket,time,SocketServer,struct,os
host='192.168.50.74'
port=12307
ADDR=(host,port)
class MyRequestHandler(SocketServer.BaseRequestHandler):
def handle(self):
print('connected from:', self.client_address)
while True:
fileinfo_size=struct.calcsize('128sl') #定义文件信息。128s表示文件名为128bytes长,l表示一个int或log文件类型,在此为文件大小
self.buf = self.request.recv(fileinfo_size)
if self.buf: #如果不加这个if,第一个文件传输完成后会自动走到下一句
self.filename,self.filesize =struct.unpack('128sl',self.buf) #根据128sl解包文件信息,与client端的打包规则相同
print 'filesize is: ',self.filesize,'filename size is: ',len(self.filename) #文件名长度为128,大于文件名实际长度
self.filenewname = os.path.join('e:\\',('new_'+ self.filename).strip('\00')) #使用strip()删除打包时附加的多余空字符
print self.filenewname,type(self.filenewname)
recvd_size = 0 #定义接收了的文件大小
file = open(self.filenewname,'wb')
print 'stat receiving...'
while not recvd_size == self.filesize:
if self.filesize - recvd_size > 1024:
rdata = self.request.recv(1024)
recvd_size += len(rdata)
else:
rdata = self.request.recv(self.filesize - recvd_size)
recvd_size = self.filesize
file.write(rdata)
file.close()
print 'receive done'
#self.request.close()
tcpServ = SocketServer.ThreadingTCPServer(ADDR, MyRequestHandler)
print('waiting for connection...' )
tcpServ.serve_forever()
发送端:
#-*- coding: UTF-8 -*-
import socket,os,struct
s = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
s.connect(('192.168.50.74',12307))
while True:
filepath = raw_input('Please Enter chars:\r\n')
if os.path.isfile(filepath):
fileinfo_size=struct.calcsize('128sl') #定义打包规则
#定义文件头信息,包含文件名和文件大小
fhead = struct.pack('128sl',os.path.basename(filepath),os.stat(filepath).st_size)
s.send(fhead)
print 'client filepath: ',filepath
# with open(filepath,'rb') as fo: 这样发送文件有问题,发送完成后还会发一些东西过去
fo = open(filepath,'rb')
while True:
filedata = fo.read(1024)
if not filedata:
break
s.send(filedata)
fo.close()
print 'send over...'
#s.close()