套接字
套接字是计算机网络数据结构,它体现了通信节点概念,在任何类型的通信开始之前,网络应用程序必须创建套接字,可以将它比作电话插孔,没有他将无法通信。
套接字最早是加利福尼亚大学的伯克利版本UNIX(UNIX BSD)套接字称为bokeli套接字或BSD套接字。套接字最初是为同一主机上的应用程序创建的,使主机上运行的一个程序(又名一个进程)和另一个运行程序进行通讯。这就是所谓的进程通讯,有两种的套接字类型:基于文件和面向网络的
UNIX套接字是第一个套接字家族,并拥有家族名字 AF_UNIX,他代表地址家族:UNIX
因为两个进程运行在同一台计算机上,所以这些套接字基于文件,这就意味着文件系统支持他们的底层基础结构,这就能够说得通了,因为文件是一个运行在同一主机上的多个进程之间的常量。
第二种类型的套接字是面向网络的,它有自己的家族名称AF_INET,或者家族:因特网 。另一个家族AF_INET6用于第6版因特网协议(IPV6)寻址
面向连接套接字与无连接套接字
1.面向连接套接字
面向连接套接字意味着通讯之前必须先建立一个连接,例如使用电话系统给朋友打电话。这种类型的通讯我们称为虚拟电路或流套接字。
面对连接的通讯提供序列化的,可靠的和不重复的数据交付,而没有记录边界,这基本意味着每条信息可以拆分成多个片段,并且每一段信息片段都能确保能够到达目的地,然后将它按顺序组合在一起,最后完整消息传递给正在等待的应用。
实现这种类型的主要协议是传输控制协议TCP,为了创建TCP套接字,必须使用SOCK_STREAM作为套字节类型。TCP套接字的名字SOCK_STREAM基于流套接字的其中一种表示。因为这些套接字(AF_INET)的网络版使用因特网协议IP来搜寻网络中的主机,所以整个系统结合这两种协议(TCP和IP来进行)
2.无连接的套接字
无连接的套接字被称为数据报类型的套接字,通讯不用建立连接,传输无法保证重复性,循序性,可靠性,可以整体发送。它成本低廉性能更好。
实现这种连接类型的主要协议是用户数据报协议UDP。为了创建UDP套接字,必需使用SOCK_DGRAM作为套接字类型,因为这些套接字也是用因特网协议来寻找网络主机,所以有个更普通的名字UDP/IP
另外还有
socket.SOCK_RAW #原始套接字,普通的套接字无法处理ICMP、IGMP等网络报文,而SOCK_RAW可以;其次,SOCK_RAW也可以处理特殊的IPv4报文;此外,利用原始套接字,可以通过IP_HDRINCL套接字选项由用户构造IP头。
socket.SOCK_RDM #是一种可靠的UDP形式,即保证交付数据报但不保证顺序。SOCK_RAM用来提供对原始协议的低级访问,在需要执行某些特殊操作时使用,如发送ICMP报文。SOCK_RAM通常仅限于高级用户或管理员运行的程序使用。
socket.getaddrinfo(host, port, family=0, type=0, proto=0, flags=0) #获取要连接的对端主机地址
sk.bind(address)
s.bind(address) 将套接字绑定到地址。address地址的格式取决于地址族。在AF_INET下,以元组(host,port)的形式表示地址。
sk.listen(backlog)
开始监听传入连接。backlog指定在拒绝连接之前,可以挂起的最大连接数量。
backlog等于5,表示内核已经接到了连接请求,但服务器还没有调用accept进行处理的连接个数最大为5
这个值不能无限大,因为要在内核中维护连接队列
sk.setblocking(bool)
是否阻塞(默认True),如果设置False,那么accept和recv时一旦无数据,则报错。
sk.accept()
接受连接并返回(conn,address),其中conn是新的套接字对象,可以用来接收和发送数据。address是连接客户端的地址。
接收TCP 客户的连接(阻塞式)等待连接的到来
sk.connect(address)
连接到address处的套接字。一般,address的格式为元组(hostname,port),如果连接出错,返回socket.error错误。
sk.connect_ex(address)
同上,只不过会有返回值,连接成功时返回 0 ,连接失败时候返回编码,例如:10061
sk.close()
关闭套接字
sk.recv(bufsize[,flag])
接受套接字的数据。数据以字符串形式返回,bufsize指定最多可以接收的数量。flag提供有关消息的其他信息,通常可以忽略。
sk.recvfrom(bufsize[.flag])
与recv()类似,但返回值是(data,address)。其中data是包含接收数据的字符串,address是发送数据的套接字地址。
sk.send(string[,flag])
将string中的数据发送到连接的套接字。返回值是要发送的字节数量,该数量可能小于string的字节大小。即:可能未将指定内容全部发送。
sk.sendall(string[,flag])
将string中的数据发送到连接的套接字,但在返回之前会尝试发送所有数据。成功返回None,失败则抛出异常。
内部通过递归调用send,将所有内容发送出去。
sk.sendto(string[,flag],address)
将数据发送到套接字,address是形式为(ipaddr,port)的元组,指定远程地址。返回值是发送的字节数。该函数主要用于UDP协议。
sk.settimeout(timeout)
设置套接字操作的超时期,timeout是一个浮点数,单位是秒。值为None表示没有超时期。一般,超时期应该在刚创建套接字时设置,因为它们可能用于连接的操作(如 client 连接最多等待5s )
sk.getpeername()
返回连接套接字的远程地址。返回值通常是元组(ipaddr,port)。
sk.getsockname()
返回套接字自己的地址。通常是一个元组(ipaddr,port)
sk.fileno()
套接字的文件描述符
socket.sendfile(file, offset=0, count=None)
客户端
#Author:Zhiyu Su
#客户端
importsocket
client= socket.socket() #声明socket类型,同时生成socket链接对象
client.connect(('localhost',6969))whileTrue:
msg= input('>>:').strip()if len(msg) == 0:continueclient.send(msg.encode('utf-8'))
data= client.recv(1024)print('recv:',data.decode())
client.close()
View Code
服务器端
#Author:Zhiyu Su
#服务器端
importosimportsocket
server=socket.socket()
server.bind(('localhost',6969)) #绑定要监听端口
server.listen(5)#监听最多允许等待多少链接
whileTrue:print('我要开始等电话了')
conn, addr= server.accept() #等电话打进来#conn 就是客户端连过来而在服务器端为期生成的一个链接
print('电话来了')whileTrue:
data= conn.recv(1024)print('recv :',data)if notdata:print('client has lost...')breakres=os.popen(data).read()
conn.send(data.upper())
server.close()
View Code
远程执行命令行命令(以及粘包的处理)
客户端
#Author:Zhiyu Su
importsocket
client=socket.socket()
client.connect(('localhost',9999))whileTrue:
cmd= input('>>>:').strip()if len(cmd)==0:continueclient.send(cmd.encode('utf-8'))
cmd_res_size= client.recv(1024) #接受命令结果的长度
print('命令行结果大小',cmd_res_size)
client.send('准备好接受了 loser开始发'.encode('utf-8'))#收到数据完毕之后相应,防止粘包
received_size=0
received_data= b''
while received_size
data= client.recv(1024)
received_size+= len(data) #每次收到的有可能小于1024
received_data+=dataelse:print('cmd res received done',received_size)print(received_data.decode())
client.close()
View Code
服务器端
#Author:Zhiyu Su
importos,timeimportsocket
server=socket.socket()
server.bind(('localhost',9999))
server.listen(5)whileTrue:print('等待新指令')
conn,addr=server.accept()print('new conn:',addr)whileTrue:
data= conn.recv(1024)if notdata:print('客户端断开')break
print('执行指令',data)
cmd_res= os.popen(data.decode()).read()#接受字符串,执行结果也是字符串
print('before send',len(cmd_res))if len(cmd_res) ==0:
cmd_res= 'cmd hsa no output....'conn.send(str(len(cmd_res.encode())).encode('utf-8')) #先发大小给客户端如果encode()服务器端len的长度不一样
#time.sleep(0.5)
client_ack = conn.recv(1024)#wait client to confirm recv如果收不到数据就会堵塞所以只能等上一个发送完毕,验证客户端响应防止粘包
print('ack from client:',client_ack.decode())
conn.send(cmd_res.encode('utf-8'))print('send done')
server.slose()
View Code
文件传输(MD5加密)
客户端
__author__ = "Alex Li"
importsocketimporthashlib
client=socket.socket()
client.connect(('localhost', 9999))whileTrue:
cmd= input(">>:").strip()if len(cmd) == 0: continue
if cmd.startswith("get"):
client.send(cmd.encode())
server_response= client.recv(1024)print("servr response:", server_response)
client.send(b"ready to recv file")
file_total_size=int(server_response.decode())
received_size=0
filename= cmd.split()[1]
f= open(filename + ".new", "wb")
m=hashlib.md5()while received_size 1024: #要收不止一次
size = 1024
else: #最后一次了,剩多少收多少
size = file_total_size -received_sizeprint("last receive:", size)
data=client.recv(size)
received_size+=len(data)
m.update(data)
f.write(data)#print(file_total_size,received_size)
else:
new_file_md5=m.hexdigest()print("file recv done", received_size, file_total_size)
f.close()
server_file_md5= client.recv(1024)print("server file md5:", server_file_md5)print("client file md5:", new_file_md5)
client.close()
View Code
服务器端
importhashlibimportsocket ,os,time
server=socket.socket()
server.bind(('0.0.0.0',9999) )
server.listen()whileTrue:
conn, addr=server.accept()print("new conn:",addr)whileTrue:print("等待新指令")
data= conn.recv(1024)if notdata:print("客户端已断开")breakcmd,filename=data.decode().split()print(filename)ifos.path.isfile(filename):
f= open(filename,"rb")
m=hashlib.md5()
file_size=os.stat(filename).st_size
conn.send( str(file_size).encode() )#send file size
conn.recv(1024) #wait for ack
for line inf:
m.update(line)
conn.send(line)print("file md5", m.hexdigest())
f.close()
conn.send(m.hexdigest().encode())#send md5
print("send done")
server.close()
View Code
socketserver(多线程)
客户端
#Author:Zhiyu Su
importsocket
client=socket.socket()
client.connect(('localhost',9999))whileTrue:
msg= input('>>:').strip()if len(msg) == 0:continueclient.send(msg.encode('utf-8'))
data= client.recv(10240)print('recv:',data.decode())
client.close()
View Code
服务器端
importsocketserverclass MyTCPHandler(socketserver.BaseRequestHandler): #自己重写的父类需要继承socketserver和BaseRequestHandler
"""The request handler class for our server.
It is instantiated once per connection to the server, and must
override the handle() method to implement communication to the
client."""
def handle(self): #重写handle 和客户端的交互都在handle里写的
whileTrue:try:#self.request is the TCP socket connected to the client
self.data = self.request.recv(1024).strip() #接受数据 ,每一请求过来都会实例MyTCPHandler
print("{} wrote:".format(self.client_address[0])) #打印客户端的地址
print(self.data)#if not self.data: #代表客户端断开
#print(self.client_address,'断开了')
#break
#just send back the same data, but upper-cased
self.request.sendall(self.data.upper()) #传回数据
exceptConnectionResetError as e:print('err',e)break
if __name__ == "__main__":
HOST, PORT= "localhost", 9999 #定义端口
#Create the server, binding to localhost on port 9999
server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler) #多线程把ip地址端口和实例化的类传给TCPServer ,然后开始监听
#ForkingTCPServer多进程(在win上不能使用os里没fork)
#server = socketserver.TCPServer((HOST, PORT), MyTCPHandler)
#Activate the server; this will keep running until you
#interrupt the program with Ctrl-C
server.serve_forever()
View Code