Socket Families(地址簇)
socket.AF_UNIX unix本机进程间通信
socket.AF_INET IPV4
socket.AF_INET6 IPV6
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报文。SOCK_RAM通常仅限于高级用户或管理员运行的程序使用。
简单的socket例子
import socket
server = socket.socket()
server.bind(('localhost', 6969)) #绑定要监听的端口
server.listen() #开始监听
print("我要等电话了")
conn,addr = server.accept() ##接受并建立与客户端的连接,程序在此处开始阻塞,只到有客户端连接进来...
print("conn\t", conn,'addr:',addr)
#conn就是客户端连过来而在服务端为其生成的一个连接实例
print('电话来了')
data = conn.recv(1024)
print('recv: ', data)
conn.send(data.upper())
server.close()
import socket
client = socket.socket() #声明socket类型(默认IPv4),同时生成socket连接对象
client.connect(("localhost", 6969))
client.send(b"Hello World")
data = client.recv(1024)
print("recv: ",data)
client.close()
上面的代码的有一个问题, 就是接收了一次客户端的data就退出了,但实际场景中,一个连接建立起来后,可能要进行多次往返的通信。
通过socket实现减半的SSH
import socket,os
server = socket.socket()
server.bind(('localhost', 6969))
server.listen()
while True: #持续可建立多个连接
conn,addr = server.accept() #事件驱动
while True: #跟客户端持续交互
data = conn.recv(1024)
if not data:
print("客户端已断开")
break
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')) #先发结果大小给客户端(注意是cmd_res.encode(),因为一个中文的byte长度是3)
conn.send(cmd_res.encode('utf-8'))
server.close()
import socket
client = socket.socket()
client.connect(('localhost', 6969))
while True: #跟服务端持续交互
cmd = input(">>: ").strip()
if not cmd: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判断
received_data += data
print("reveived_size = ", received_size)
else:
print("cmd res received!")
print(received_data.decode())
client.close()