1.socket基本语法:
函数 | 描述 |
---|---|
服务器端套接字 | |
s.bind() | 绑定地址(host,port)到套接字, 在AF_INET下,以元组(host,port)的形式表示地址。 |
s.listen() | 开始TCP监听。backlog指定在拒绝连接之前,操作系统可以挂起的最大连接数量。该值至少为1,大部分应用程序设为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数据。将string中的数据发送到连接的套接字,但在返回之前会尝试发送所有数据。成功返回None,失败则抛出异常。 |
s.recvfrom() | 接收UDP数据,与recv()类似,但返回值是(data,address)。其中data是包含接收数据的字符串,address是发送数据的套接字地址。 |
s.sendto() | 发送UDP数据,将数据发送到套接字,address是形式为(ipaddr,port)的元组,指定远程地址。返回值是发送的字节数。 |
s.close() | 关闭套接字 |
s.getpeername() | 返回连接套接字的远程地址。返回值通常是元组(ipaddr,port)。 |
s.getsockname() | 返回套接字自己的地址。通常是一个元组(ipaddr,port) |
s.setsockopt(level,optname,value) | 设置给定套接字选项的值。 |
s.getsockopt(level,optname[.buflen]) | 返回套接字选项的值。 |
s.settimeout(timeout) | 设置套接字操作的超时期,timeout是一个浮点数,单位是秒。值为None表示没有超时期。一般,超时期应该在刚创建套接字时设置,因为它们可能用于连接的操作(如connect()) |
s.gettimeout() | 返回当前超时期的值,单位是秒,如果没有设置超时期,则返回None。 |
s.fileno() | 返回套接字的文件描述符。 |
s.setblocking(flag) | 如果flag为0,则将套接字设为非阻塞模式,否则将套接字设为阻塞模式(默认值)。非阻塞模式下,如果调用recv()没有发现任何数据,或send()调用无法立即发送数据,那么将引起socket.error异常。 |
s.makefile() | 创建一个与该套接字相关连的文件 |
2.socket编程
2.1实现ssh链接输出结果
server端
- 导入socket模块
- 创建socket对象socket()
- 绑定地址bind()
- 等待客户链接listen()
- 建立客户端连接accept()
- 获取数据recv()
- 发送数据send()
- 执行获取的数据之后执行指令,将执行的发送过去。
client端
- 导入socket模块
- 创建socket对象socket()
- 链接到服务端connect()
- 发送数据send()
- 获取数据recv()
执行结果如下:
但是这样会出现一个问题,执行指令的结果超过我们设置的接受size,这样会导致数据一次传送不完。
2.2socket数据发送size判断
- 发送ready,判断数据是否开始发送,以及传输数据的大小;
- 增加了数据判断,当发送start之后才真正的开始发送数据,然后判断是否接受完成;
import socket import subprocess sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0) ip_port = ('127.0.0.1', 9999) sock.bind(ip_port) sock.listen(5) while True: conn, address = sock.accept() while True: try: recv_data = conn.recv(1024) if str(recv_data, encoding='utf-8') == 'exit': break p = subprocess.Popen(str(recv_data, encoding='utf-8'), shell=True, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE) res = p.stdout.read() if len(res) == 0: res = p.stderr.read() send_data = bytes(str(res, encoding='gbk'), encoding='utf-8') conn.send(bytes('Ready|%s' % len(send_data), encoding='utf-8')) send_flag = str(conn.recv(1024), encoding='utf-8') if send_flag == 'Start': conn.send(send_data) except Exception as e: break conn.close()
import socket client = socket.socket(socket.AF_INET, socket.SOCK_STREAM, 0) ip_port = ('127.0.0.1', 9999) client.connect(ip_port) while True: s = input('Client:').strip() if len(s) == 0: break client.send(bytes(s, encoding='utf-8')) if s == 'exit': break ready_data = str(client.recv((1024)), encoding='utf-8') if ready_data.startswith('Ready'): msg_size = int(ready_data.split('|')[1]) client.send(bytes('Start', encoding='utf-8')) recv_size = 0 recv_msg = b'' while recv_size < msg_size: recv_data = client.recv(1024) recv_msg += recv_data recv_size += len(recv_data) print('数据总长:%s,已接收数据长度:%s' % (msg_size, recv_size)) print('Server:', str(recv_msg, encoding='utf-8'))
执行结果如下:
这样就能保证数据的传输完整。