一、Socket Families 地址簇
- socket.AF_UNIX
- socket.AF_INET
socket.AF_INET6
二、Socker Types
- socket.SOCK_STREAM #tcp
- socket.SOCK_DGRAM #udp
- socket.SOCK_RAM
- socket.SOCK_RDM
socket.SOCK_SEQPACKET
三、socket的收发实例
- client
# Author:Li Dongfei
import socket
client = socket.socket() #声明socket类型,同时生成socket连接对象
client.connect(('localhost',6969))
client.send(b'hello python')
data = client.recv(1024)
print("recv:",data)
client.close()
- server
# Author:Li Dongfei
import socket
server = socket.socket()
server.bind(('localhost',6969)) #绑定要监听的端口
server.listen() #监听
print("开始接收")
conn,addr = server.accept() #等待接收数据
print(conn,addr) #conn是客户端连过来而在服务器端为其生成的一个连接实例
print("数据来了")
data = conn.recv(1024)
print("recv:",data)
conn.send(data.upper())
server.close()
发送中文示例
client.send('你好'.encode("utf-8"))
四、实现连续通信
- client
# Author:Li Dongfei
import socket
client = socket.socket() #声明socket类型,同时生成socket连接对象
client.connect(('localhost',6969))
while True:
msg = input(">>: ")
client.send(msg.encode("utf-8"))
data = client.recv(1024)
print("recv:",data.decode())
client.close()
- server
# Author:Li Dongfei
import socket
server = socket.socket()
server.bind(('localhost',6969)) #绑定要监听的端口
server.listen() #监听
print("开始接收")
conn, addr = server.accept() # 等待接收数据
print("数据来了")
while True:
data = conn.recv(1024)
print("recv:",data.decode())
conn.send(data.upper())
server.close()
五、实现客户端断开服务器不断
- server
# Author:Li Dongfei
import socket
server = socket.socket()
server.bind(('localhost',6969)) #绑定要监听的端口
server.listen(5) #监听,最大允许5个连接
while True:
print("开始接收..")
conn,addr = server.accept() # 等待接收数据
print("数据来了")
while True:
data = conn.recv(1024)
print("recv:",data.decode())
if not data:
print("client has lost...")
break
conn.send(data.upper())
server.close()
- client
# Author:Li Dongfei
import socket
client = socket.socket() #声明socket类型,同时生成socket连接对象
client.connect(('localhost',6969))
while True:
msg = input(">>: ")
if len(msg) == 0:continue
client.send(msg.encode("utf-8"))
data = client.recv(1024)
print("recv:",data.decode())
client.close()
六、ssh服务器和ssh客户端
- client
# Author:Li Dongfei
import socket
client = socket.socket()
client.connect(('localhost',9999))
while True:
cmd = input(">>: ").strip()
if len(cmd) == 0: continue
client.send(cmd.encode("utf-8"))
cmd_res_size = client.recv(1024) #接收命令结果的长度
received_size = 0
#received_data = b''
while received_size < int(cmd_res_size.decode()):
data = client.recv(1024)
received_size += len(data)
print(data.decode("utf-8"))
else:
print("received_size:",received_size)
client.close()
- server
# Author:Li Dongfei
import socket,os
server = socket.socket()
server.bind(('localhost',9999))
server.listen(5)
while True:
conn,addr = server.accept()
print("new conn:",addr)
while True:
data = conn.recv(1024)
if not data:
print("客户端已断开")
break
print("执行指令: ",data)
cmd_res = os.popen(data.decode()).read()
if len(cmd_res) == 0:
cmd_res = "cmd has no output..."
conn.send(str(len(cmd_res.encode())).encode("utf-8")) #先发数据大小给客户端
conn.send(cmd_res.encode("utf-8"))
server.close()
七、最终版本
- sock_server_ssh.py
# Author:Li Dongfei
import socket,os
server = socket.socket()
server.bind(('localhost',9999))
server.listen()
while True:
conn, addr = server.accept()
print("new conn:", addr)
while True:
data = conn.recv(1024)
if not data:
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 has no output."
conn.send(str(len(cmd_res.encode())).encode("utf-8"))
conn.send(cmd_res.encode("utf-8"))
print("send done")
server.close()
- sock_client_ssh
# Author:Li Dongfei
import socket
client = socket.socket()
client.connect(('localhost',9999))
while True:
cmd = input(">>:").strip()
if len(cmd) == 0: continue
client.send(cmd.encode("utf-8"))
cmd_res_size = client.recv(1024)
print("命令结果大小",cmd_res_size)
received_size = 0
#received_data = b''
while received_size < int(cmd_res_size.decode()):
data = client.recv(1024)
received_size += len(data)
print(data.decode())
else:
print("cmd res receive done.",received_size)
#cmd_res = client.recv(1024)
#print(cmd_res.decode())
client.close()
八、socket粘包
conn.send(str(len(cmd_res.encode())).encode("utf-8"))
conn.send(cmd_res.encode("utf-8"))
类似以上代码会将两条send的数据放到一个包中发出
解决方法
- server
# Author:Li Dongfei
import socket,os
server = socket.socket()
server.bind(('localhost',9999))
server.listen()
while True:
conn, addr = server.accept()
print("new conn:", addr)
while True:
data = conn.recv(1024)
if not data:
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 has no output."
conn.send(str(len(cmd_res.encode())).encode("utf-8"))
client_ack = conn.recv(1024)
print("ack from client",client_ack.decode())
conn.send(cmd_res.encode("utf-8"))
print("send done")
server.close()
- client
# Author:Li Dongfei
import socket
client = socket.socket()
client.connect(('localhost',9999))
while True:
cmd = input(">>:").strip()
if len(cmd) == 0: continue
client.send(cmd.encode("utf-8"))
cmd_res_size = client.recv(1024)
print("命令结果大小",cmd_res_size)
client.send("准备好接受了".encode("utf-8"))
received_size = 0
#received_data = b''
while received_size < int(cmd_res_size.decode()):
data = client.recv(1024)
received_size += len(data)
print(data.decode())
else:
print("cmd res receive done.",received_size)
#cmd_res = client.recv(1024)
#print(cmd_res.decode())
client.close()
九、ftp server
- 读取文件名
- 检测文件是否存在
- 打开文件
- 检测文件大小
- 发送文件大小给客户端
- 等客户端确认
- 开始边读边发数据
- 发送md5
- server
# Author:Li Dongfei
import socket,os,hashlib
server = socket.socket()
server.bind(('localhost',9999))
server.listen()
while True:
conn, addr = server.accept()
print("new conn:", addr)
while True:
data = conn.recv(1024)
if not data:
print("客户端已断开")
break
cmd,filename = data.decode().split()
print(filename)
if os.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 in f:
m.update(line)
conn.send(line)
print("file md5:",m.hexdigest())
f.close()
conn.send(m.hexdigest().encode())
print("send done.")
server.close()
- client
# Author:Li Dongfei
import socket,hashlib
client = socket.socket()
client.connect(('localhost',9999))
while True:
cmd = input(">>:").strip()
if len(cmd) == 0: continue
if cmd.startswith("get"):
client.send(cmd.encode())
server_response = client.recv(1024)
print("server 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,"wb")
m = hashlib.md5()
while received_size < file_total_size:
if file_total_size - received_size > 1024: #要收不止一次
size = 1024
else: #最后一次剩余的数
size = file_total_size - received_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.",file_total_size,received_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()
十、socketserver
- 基本写法
# Author:Li Dongfei
import socketserver
class MyTCPHandler(socketserver.BaseRequestHandler):
def handle(self):
while True:
try:
self.data = self.request.recv(1024).strip()
print("{} wrote:".format(self.client_address[0]))
print(self.data)
if not self.data: #表示客户端断开
print(self.client_address,"断开")
break
self.request.send(self.data.upper())
except ConnectionResetError as e:
print("err",e)
break
if __name__ == "__main__":
HOST, PORT = "localhost", 9999
server = socketserver.TCPServer((HOST, PORT), MyTCPHandler)
server.serve_forever()
- client
# Author:Li Dongfei
import socket
client = socket.socket() #声明socket类型,同时生成socket连接对象
client.connect(('localhost',9999))
while True:
msg = input(">>: ")
client.send(msg.encode("utf-8"))
data = client.recv(1024)
print("recv:",data.decode())
十一、socket多线程并发
# Author:Li Dongfei
import socketserver
class MyTCPHandler(socketserver.BaseRequestHandler):
def handle(self):
while True:
try:
self.data = self.request.recv(1024).strip()
print("{} wrote:".format(self.client_address[0]))
print(self.data)
if not self.data: #表示客户端断开
print(self.client_address,"断开")
break
self.request.send(self.data.upper())
except ConnectionResetError as e:
print("err",e)
break
if __name__ == "__main__":
HOST, PORT = "localhost", 9999
server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler) #多线程
server.serve_forever()