socket

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()