socket 又称为 “套接字”,应用程序通常通过“套接字”向网络发出请求或应答网络请求。使主机间或者一台服务器上的进程可以通信。
实例说明:
客户端:
1 import socket 2 3 sk = socket.socket() #创建套接字 4 5 addr = ('127.0.0.1',8888) 6 sk.bind(addr) # 以元组的形式绑定地址到套接字(ip,port) 7 8 sk.listen(3) # 打开tcp监听,表示在结束连接前队列中支持挂起的最大连接数 9 conn,address=sk.accept() # 被动接受客户端的连接10 11 inp = input(">>>")12 conn.send(bytes(inp,'utf8')) # 发送tcp数据,python3支持字节数据的传输13 conn.close() # 关闭单个(本次)套接字的连接。sk.close()关闭所有套接字的连接
服务端:
1 import socket 2 3 sk = socket.socket() # 创建套接字 4 addr = ('127.0.0.1',8888) 5 sk.connect(addr) # 以元组的形式主动连接server端 6 7 data=sk.recv(1024) # 接收TCP数据,1024是指定的最大数据量 8 print(str(data,'utf8')) 9 10 sk.close() # 关闭套接字
socket对象的内置方法:
函数 | 描述 |
---|---|
服务器端套接字 | |
s.bind() | 绑定地址(host,port)到套接字, 在AF_INET下,以元组(host,port)的形式表示地址。 |
s.listen() | 开始TCP监听。backlog指定在拒绝连接之前,操作系统可以挂起的最大连接数量。该值至少为1,大部分应用程序设为5就可以了。 |
s.accept() | 被动接受TCP客户端连接,(阻塞式)等待连接的到来 |
客户端套接字 | |
s.connect() | 主动初始化TCP服务器连接,。一般address的格式为元组(hostname,port),如果连接出错,返回socket.error错误。 |
s.connect_ex() | connect()函数的扩展版本,出错时返回出错码,而不是抛出异常 |
公共用途的套接字函数 | |
s.recv() | 接收TCP数据,数据以字符串形式返回,bufsize指定要接收的最大数据量。flag提供有关消息的其他信息,通常可以忽略。 |
s.send() | 发送TCP数据,将string中的数据发送到连接的套接字。返回值是要发送的字节数量,该数量可能小于string的字节大小。 |
s.sendall() | 完整发送TCP数据,完整发送TCP数据。将string中的数据发送到连接的套接字,但在返回之前会尝试发送所有数据。成功返回None,失败则抛出异常。 |
s.recvfrom() | 接收UDP数据,与recv()类似,但返回值是(data,address)。其中data是包含接收数据的字符串,address是发送数据的套接字地址。 |
s.sendto() | 发送UDP数据,将数据发送到套接字,address是形式为(ipaddr,port)的元组,指定远程地址。返回值是发送的字节数。 |
s.close() | 关闭套接字 |
s.getpeername() | 返回连接套接字的远程地址。返回值通常是元组(ipaddr,port)。 |
s.getsockname() | 返回套接字自己的地址。通常是一个元组(ipaddr,port) |
s.setsockopt(level,optname,value) | 设置给定套接字选项的值。 |
s.getsockopt(level,optname[.buflen]) | 返回套接字选项的值。 |
s.settimeout(timeout) | 设置套接字操作的超时期,timeout是一个浮点数,单位是秒。值为None表示没有超时期。一般,超时期应该在刚创建套接字时设置,因为它们可能用于连接的操作(如connect()) |
s.gettimeout() | 返回当前超时期的值,单位是秒,如果没有设置超时期,则返回None。 |
s.fileno() | 返回套接字的文件描述符。 |
s.setblocking(flag) | 如果flag为0,则将套接字设为非阻塞模式,否则将套接字设为阻塞模式(默认值)。非阻塞模式下,如果调用recv()没有发现任何数据,或send()调用无法立即发送数据,那么将引起socket.error异常。 |
s.makefile() | 创建一个与该套接字相关连的文件 |
循环聊天:
1 # Author : Adair 2 # -*- coding:utf-8 -*- 3 4 5 import socket 6 7 sk = socket.socket() #创建套接字 8 9 addr = ('127.0.0.1',8888)10 sk.bind(addr) # 以元组的形式绑定地址到套接字(ip,port)11 12 sk.listen(3) # 打开tcp监听,表示在结束连接前队列中支持挂起的最大连接数13 14 15 while True:16 conn, address = sk.accept() # 被动接受客户端的连接17 data = conn.recv(1024)18 if not data:19 continue20 print(str(data,"utf8"))21 22 inp = input(">>>")23 if inp == "exit": break24 conn.send(bytes(inp,'utf8')) # 发送tcp数据,python3支持字节数据的传输25 26 conn.close() # 关闭单个(本次)套接字的连接。sk.close()关闭所有套接字的连接
1 # Author : Adair 2 # -*- coding:utf-8 -*- 3 import socket 4 5 sk = socket.socket() # 创建套接字 6 addr = ('127.0.0.1',8888) 7 sk.connect(addr) # 以元组的形式主动连接server端 8 9 while True:10 inp = input(">>>")11 if inp == "exit":break12 sk.send(bytes(inp,"UTF8"))13 14 data=sk.recv(1024) # 接收TCP数据,1024是指定的最大数据量15 if not data: print("[ x ] The other party has been taken offline!");break16 print(str(data,'utf8'))17 18 sk.close() # 关闭套接字
执行远程命令:
1 # Author : Adair 2 # -*- coding:utf-8 -*- 3 4 5 import socket,os 6 7 sk = socket.socket() #创建套接字 8 9 addr = ('127.0.0.1',8888)10 sk.bind(addr) # 以元组的形式绑定地址到套接字(ip,port)11 12 sk.listen(3) # 打开tcp监听,表示在结束连接前队列中支持挂起的最大连接数13 14 while True:15 conn, address = sk.accept() # 被动接受客户端的连接16 print("[ X ] %s has been connected. Port number:%s" %(address[0],address[1]))17 while True:18 data = conn.recv(1024)19 if not data:20 break21 CmdReturn=os.popen(str(data,'utf8')).read()22 CmdLen=bytes(str(len(CmdReturn)),'utf8') # int字符型不能直接转换成bytes类型23 print("[ X ] The orders executed are:%s,Length is:%s"%(data,CmdLen))24 conn.sendall(CmdLen) # 只能发送bytes类型的数据25 conn.sendall(bytes(CmdReturn,'utf8'))26 27 conn.close() # 关闭单个(本次)套接字的连接。sk.close()关闭所有套接字的连接
1 # Author : Adair 2 # -*- coding:utf-8 -*- 3 import socket 4 5 sk = socket.socket() # 创建套接字 6 addr = ('127.0.0.1',8888) 7 sk.connect(addr) # 以元组的形式主动连接server端 8 9 while True:10 inp = input(">>>")11 if inp == "exit":break12 sk.send(bytes(inp,"utf8"))13 14 data_len=int(str(sk.recv(1024),'utf8'))# 接收的是data的字符长度15 if data_len == 0:print("[ X ] Command execution failed!")16 data=bytes() # 定义一个空的bytes,用于接收合并TCP数据17 while len(data) < int(data_len):18 data_recv=sk.recv(1024) # 接收TCP数据,1024是指定的最大数据量19 data+=data_recv20 if not data: print("[ x ] The other party has been taken offline!");break21 print(str(data,'utf8'))22 23 sk.close() # 关闭套接字
上传/下载:
1 # Author : Adair 2 # -*- coding:utf-8 -*- 3 4 import socket,os 5 6 address = ("127.0.0.1",1111) 7 8 sk = socket.socket() 9 sk.bind(address)10 sk.listen(3)11 12 conn,addr=sk.accept()13 14 TargetPath = str(conn.recv(1024),'utf8')15 FileSize = int(str(conn.recv(1024),'utf8'))16 print(type(FileSize))17 FileData= conn.recv(FileSize)18 19 20 21 22 TargetFilePath,TargetFileName=os.path.split(TargetPath)23 if not os.path.exists(TargetFilePath):24 os.makedirs(TargetFilePath)25 print("The target directory has been created!")26 27 with open(TargetPath,'wb') as file:28 file.write(FileData)
1 # Author : Adair 2 # -*- coding:utf-8 -*- 3 4 import socket,os 5 6 MenuList=['Upload','Download','Exit'] # 菜单 7 for i,element in enumerate(MenuList,start=1): # 菜单内容输出 8 print("\t%s、%s"%(i,element)) 9 10 def switch(para): # 菜单项调用功能函数执行11 return {12 '1':Upload,13 '2':Download,14 '3':exit15 }.get(para,'The selected parameter does not exist')16 17 class SocketLib(): # socket 接口18 def __init__(self,TargetPath,FileData,FileSize):19 20 self.TargetPath=TargetPath21 self.FileData=FileData22 self.FileSize=FileSize23 24 self.address = ('127.0.0.1', 1111)25 self.sk = socket.socket()26 self.sk.connect(self.address)27 28 def SocketSend(self):29 self.sk.send(bytes(self.TargetPath,'utf8'))30 self.sk.send(str(self.FileSize).encode())31 self.sk.send(bytes(self.FileData))32 33 34 35 36 def Upload():37 SourcePath = input("Please enter the source path:")38 TargetPath = input("Please enter the target path:")39 OpenFile=open(SourcePath,'rb')40 FileData=OpenFile.read()41 FileSize=os.path.getsize(SourcePath)42 43 SkLib=SocketLib(TargetPath,FileData,FileSize)44 SkLib.SocketSend()45 def Download():46 print("download!!!")47 48 49 ActionChoice=input("Please select the action you want to operate:")50 switch(ActionChoice)()51 # sk.send(bytes("test","utf8"))
socketserver是socket的升级版,是socket的封装,他支持多线程的并发。socketserver中包含了两种类,一种为服务类,一种是服务请求类,前一种是创建连接的工程(绑定、监听、运行),后者主要处理事物逻辑(用户发过来的数据)
server:
1 # Author : Adair 2 # -*- coding:utf-8 -*- 3 4 import socketserver 5 6 class MyTCPServer(socketserver.BaseRequestHandler): # 创建一个继承类 7 8 def handle(self): # 重写父类的handle,此方法中处理业务逻辑 9 print("******")10 while True:11 conn = self.request # self.request 为客户端的连接句柄12 print(self.client_address) # self.client_address 为客户端的ip、port13 while True:14 ClientData=conn.recv(1024)15 print(str(ClientData,"UTF8"))16 InP=input(">>>")17 conn.sendall(str(InP).encode())18 conn.close()19 20 21 if __name__ == "__main__":22 server = socketserver.ThreadingTCPServer(("127.0.0.1",8090),MyTCPServer) # 选择合适的服务类型(ThreadingTCPServer)创建连接23 server.serve_forever() # 处理无限的请求
client:
1 # Author : Adair 2 # -*- coding:utf-8 -*- 3 4 import socket 5 6 addr = ("127.0.0.1",8090) 7 sk = socket.socket() 8 sk.connect(addr) 9 10 while True:11 msg = input(">>>")12 sk.send(str(msg).encode())13 data=sk.recv(1024)14 print("data:",str(data,"utf8"))15 sk.close()