参考:https://www.runoob.com/python/python-socket.html

Socket又称"套接字",应用程序通常通过"套接字"向网络发出请求或者应答网络请求,使主机间或者一台计算机上的进程间可以通讯。

socket函数

python中使用socket()函数创建套接字,如下所示

import socket
# 参数1:family, 套接字家族可以使用AF_UNIX or AF_INET
# 参数2:type, 套接字类型可以根据是面向连接的还是非连接分为SOCK_STREAM或SOCK_DGRAM
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

案例用到的方法简介

服务器端套接字:

s.bind()

绑定地址(host,port)到套接字, 在AF_INET下,以元组(host,port)的形式表示地址。

s.listen() 

开始TCP监听。backlog指定在拒绝连接之前,操作系统可以挂起的最大连接数量。大部分应用程序设为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数据。成功返回None,失败则抛出异常。

s.close()

关闭套接字

s.getpeername()

返回连接套接字的远程地址。返回值通常是元组(ipaddr,port)。

s.getsockname()

返回套接字自己的地址。通常是一个元组(ipaddr,port)

s.settimeout(timeout)

设置套接字操作的超时期,timeout是一个浮点数,单位是秒。值为None表示没有超时期。一般,超时期应该在刚创建套接字时设置,因为它们可能用于连接的操作(如connect())

s.gettimeout()

返回当前超时期的值,单位是秒,如果没有设置超时期,则返回None。

 

连接执行案例

服务器端代码

import os
import socket
import datetime
import threading
import urllib.parse


local_ip = '0.0.0.0'        # 配置socket server绑定的本地IP
local_port = 4321           # 配置socket server绑定的本地端口

server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind((local_ip, local_port))
server.listen(5)

# 只响应白名单中的计算机发来的任务
admin_filter = ['允许的ip', '127.0.0.1']


print('White list:' + str(admin_filter))
print('Server bind on %s:%s' % (local_ip, str(local_port)))
print('-----------------服务启动成功-----------------')


def exe_pro(msg):
    # 执行命令
    os.system(msg)


while True:
    # 建立与客户端连接
    conn, addr = server.accept()
    msg = urllib.parse.unquote(conn.recv(1024).decode('utf-8'))
    
    peer_name = conn.getpeername()      # 返回连接套接字的远程地址。返回值通常是元组(ipaddr,port)。
    sock_name = conn.getsockname()      # 返回套接字自己的地址。通常是一个元组(ipaddr,port)

    # peer_name是个tuple,peer_name[0]是ip,peer_name[1]是端口号
    now_dt = str(datetime.datetime.now())
    print('%s, visitor: %s:%s' % (now_dt, peer_name[0], peer_name[1]))
    # 所谓的白名单,管理员权限验证
    if peer_name[0] in admin_filter:
        print(msg)
        if 'quit' == msg:
            conn.close()
            exit(0)
        t = threading.Thread(target=exe_pro, args=(msg,))
        t.start()
    
    # 发送TCP数据,将string中的数据发送到连接的套接字。
    # conn.send('server: I received '.encode('utf-8'))

用户端代码

def send_command(server, instr):
    try:
        client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
        client.connect((server, 4321))
        client.settimeout(5)
        client.sendall(parse.quote(instr).encode('utf-8'))
        # 接收TCP数据,数据以字符串形式返回,bufsize指定要接收的最大数据量
        # print(client.recv(1024))
    except Exception as e:
        print(e)
    finally:
        client.close()


# server是服务器地址,instr是要执行的命令