python Socket网络编程
- Socket 介绍 及 两种传输协议(TCP,UDP)
- Socket 参数介绍
- 基本 Socket实例
- 通过Socket实现简单SSH
- 通过Socket实现文件加密传输
- SocketServer
1,Socket 介绍
Socket
简单来说 socket网络编程 就是将数据传输的一些协议,传输方式等底层东西打包起来,使得我们可以直接调用进行数据传输
套接字(socket)是一个抽象层,应用程序可以通过它发送或接收数据,可对其进行像对文件一样的打开、读写和关闭等
操作。套接字允许应用程序将I/O插入到网络中,并与网络中的其他应用程序进行通信。网络套接字是IP地址与端口的组合。
套接字Socket=(IP地址:端口号 )
例如,如果IP地址是210.37.145.1,而端口号是23,那么得到套接字就是(210.37.145.1:23)
Socket Families(地址簇)
socket.AF_UNIX unix本机进程间通信
socket.AF_INET IPV4
socket.AF_INET6 IPV6
这些常量表示用于socket()的第一个参数的地址(和协议)系列。 如果未定义AF_UNIX常量,则不支持此协议。
根据系统的不同,可能会有更多常量。
Socket Types
socket.SOCK_STREAM #for tcp
socket.SOCK_DGRAM #for udp
socket.SOCK_RAW #原始套接字,普通的套接字无法处理ICMP、IGMP等网络报文,而SOCK_RAW可以;
其次,SOCK_RAW也可以处理特殊的IPv4报文;
此外,利用原始套接字,可以通过IP_HDRINCL套接字选项由用户构造IP头。
socket.SOCK_RDM 是一种可靠的UDP形式,即保证交付数据报但不保证顺序。
SOCK_RAM 用来提供对原始协议的低级访问,在需要执行某些特殊操作时使用,如发送ICMP报文。
通常仅限于高级用户或管理员运行的程序使用。
这些常量表示套接字类型,用于socket()的第二个参数。 根据系统的不同,可能会有更多常量。
只有SOCK_STREAM和SOCK_DGRAM似乎通常很有用。
两类传输协议:TCP UDP
传输控制协议(TCP,Transmission Control Protocol)是为了在不可靠的互联网络上提供可靠的端到端字节流而专门设计的一个传输协议。
TCP是因特网中的传输层协议,使用三次握手协议建立连接。
TCP三次握手的过程如下:
1,客户端发送SYN(SEQ=x)报文给服务器端,进入SYN_SEND状态。
2,服务器端收到SYN报文,回应一个SYN (SEQ=y)ACK(ACK=x+1)报文,进入SYN_RECV状态。
3,客户端收到服务器端的SYN报文,回应一个ACK(ACK=y+1)报文,进入Established状态。
三次握手完成,TCP客户端和服务器端成功地建立连接,可以开始传输数据了。
TCP连接终止
建立一个连接需要三次握手,而终止一个连接要经过四次握手,这是由TCP的半关闭(half-close)造成的
TCP 四次断开过程如下:
1,某个应用进程首先调用close,称该端执行“主动关闭”(active close)该端的TCP于是发送一个FIN分节,表示数据发送完毕。
2,接收到这个FIN的对端执行 “被动关闭”(passive close),这个FIN由TCP确认。
3,一段时间后,接收到这个文件结束符的应用进程将调用close关闭它的套接字。这导致它的TCP也发送一个FIN。
4,接收这个最终FIN的原发送端TCP(即执行主动关闭的那一端)确认这个FIN。
UDP 协议
Internet 协议集支持一个无连接的传输协议,该协议称为用户数据报协议(UDP,User Datagram Protocol)。
UDP 为应用程序提供了一种无需建立连接就可以发送封装的 IP 数据报的方法。
每个数据报都是一个独立的信息,包括完整的源地址或目的地址,
它在网络上以任何可能的路径传往目的地,因此能否到达目的地,到达目的地的时间以及内容的正确性都是不能被保证的。
2,Socket 参数介绍
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)
发送文件
3,基本 Socket实例
Socket网络编程可以分为以下步骤
1.建立连接
2.打开Socket关联的输出流
3.数据流读取信息
4.关闭所有数据流和Socket
socket_client 客户端
# 单次数据交换
# import socket
#
# client = socket.socket() #声明 socket 类型,同时生成 socket 连接对象
# client.connect(('localhost',6969))
#
# # client.send(b"hello world!") # 只能使用 bytes 字节类型的数据
# client.send("我要下 a 片".encode("utf-8")) # 这样就不用使用 bytes 字节类型数据
# data = client.recv(1024)
# print("recv: ",data.decode())
# client.close()
# 可以一直连线进行数据交换
import socket
client = socket.socket() #声明socket类型,同时生成socket连接对象
client.connect(('localhost',6969))
# client.connect(('localhost',9999))
while True:
msg = input(">>:").strip()
if len(msg) == 0:continue
client.send(msg.encode("utf-8"))
data = client.recv(10240)
print("recv:",data.decode())
client.close()
socket_server 服务端
# import socket
#
# server = socket.socket()
# server.bind(('localhost',6969)) #绑定要监听的端口
# server.listen() #监听
#
# print("我要开始等电话了")
# conn,addr = server.accept() #等电话打进来
# #conn 就是客户端连过来而在服务端为其生成的一个连接实例
# print(conn,addr)
#
# print("电话来了")
# data = conn.recv(1024)
# # print("recv: ",data)
# print("recv: ",data.decode())
#
# conn.send(data.upper())
#
# server.close()
import socket
server = socket.socket()
server.bind(('localhost',6969)) #绑定要监听的端口
server.listen(5) #监听
print("我要开始等电话了")
while True:
conn,addr = server.accept() # 等电话打进来
#conn 就是客户端连过来而在服务端为其生成的一个连接实例
print(conn,addr)
print("电话来了")
count = 0;
while True:
data = conn.recv(1024)
print("recv: ",data)
if not data:
print("client has lost...")
break
conn.send(data.upper())
count+=1
if count > 10:
break
server.close()
socket黏包-- 数据连在一起
1,import time (一般不用)
time.sleep(0.5)
2,增加一个交互,进行分开
#client.send(b"ready to recv file")
client_ack = conn.recv(1023) #wait client to confirm
4,通过Socket实现简单SSH
极简版的ssh,就是客户端连接上服务器后,让服务器执行命令,并返回结果给客户端。
socket_server_client
import socket
client = socket.socket()
# client.connect('192.168.16.200',9999)
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) #每次收到的长度有可能小于1024,所以用len 判断实际长度
# print(data.decode())
received_data += data
else:
print("cmd res receive done...",received_size)
print(received_data.decode())
client.close()
socket_server_ssh
import socket,os
server = socket.socket()
server.bind(('localhost',9999))
server.listen()
while True:
conn,addr = server.accept()
print("new conn: ",addr)
while True:
print("等待新指令")
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()
5,通过Socket实现文件加密传输
ftp_client_md5
import socket
import 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("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 < file_total_size:
if file_total_size - received_size > 1024: # 要收不止一次
size = 1024
else: # 最后一次了,剩多少收多少
size = file_total_size - received_size
print("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()
ftp_server
import socket,os,time,hashlib
server = socket.socket()
server.bind(('localhost',9999))
server.listen()
while True:
conn,addr = server.accept()
print("new conn: ",addr)
while True:
print("等待新指令")
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()) #send md5
print("send done")
server.close()
6,SocketServer
对于实际应用来说,多并发才是常态,而SocketServer 中的 类就可以满足这样的需求
socket_server
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)
self.request.send(self.data.upper())
except ConnectionResetError as e:
print("error: ",e)
break
if __name__ == "__main__":
HOST, PORT = "localhost", 9999
# server = socketserver.TCPServer((HOST, PORT), MyTCPHandler) # 一对一
server = socketserver.ThreadingTCPServer((HOST, PORT), MyTCPHandler)#多并发
server.serve_forever()
``