python3套接字编程之socket和socketserver(TCP和UDP通信)

socket和socketserver是python3中socket通信模块,关于其使用做如下总结。

目录

1.socket

1.1模块引入

1.2套接字获取

1.3套接字接口

1.3.1 服务端

1.3.2 客户端套接字函数

1.3.3 公共套接字函数

1.3.4 面向锁的套接字方法

1.3.5 面向文件的套接字的函数

2.socketserver

3.TCP

3.1 socket类型TCP

3.2 socketserver类型TCP

4.UDP

3.1 socket类型UDP

3.2 socketserver类型UDP

5.额外补充:strace分析Python中subprocess.Popen实现

5.1错误命令

5.2正确命令


1.socket

1.1模块引入

import socket

1.2套接字获取

接口:socket.socket(socket_family, socket_type, protocal=0)
参数:
    socket_family:AF_UNIX 或 AF_INET
    socket_type:SOCK_STREAM 或 SOCK_DGRAM
    protocol: 一般不填,默认值为 0
(1)tcp套接字
tcpSock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
(2)udp套接字
udpSock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

1.3套接字接口

1.3.1 服务端

s.bind():绑定(主机,端口号)到套接字
s.listen():TCP监听
s.accept():接受TCP客户的连接

1.3.2 客户端套接字函数

s.connect():初始化TCP服务器连接
s.connect()
s.connect_ex():函数的扩展版本,出错时返回出错码,而不是抛出异常


1.3.3 公共套接字函数

s.recv():接收TCP数据
s.send():发送TCP数据(send在待发送数据量大于己端缓存区剩余空间时,数据丢失,不会发完)
s.sendall():发送完整的TCP数据(本质就是循环调用send,sendall在待发送数据量大于己端缓存区剩余空间时,数据不丢失,循环调用send直到发完)
s.recvfrom():接收UDP数据
s.sendto():发送UDP数据
s.getpeername():连接到当前套接字的远端的地址
s.getsockname():当前套接字的地址
s.getsockopt():返回指定套接字的参数
s.setsockopt():设置指定套接字的参数
s.close():关闭套接字

1.3.4 面向锁的套接字方法

s.setblocking():设置套接字的阻塞与非阻塞模式
s.settimeout():设置阻塞套接字操作的超时时间
s.gettimeout():得到阻塞套接字操作的超时时间

1.3.5 面向文件的套接字的函数

s.fileno():套接字的文件描述符
s.makefile():创建一个与该套接字相关的文件

2.socketserver

socketserver是socket的升级版本,可以并发处理多个客户端的连接,其包含两个大类server类和request类,server类解决连接问题,request类解决通信问题
引用如下:
import socketserver

3.TCP

3.1 socket类型TCP

server.py

# -*- coding: UTF-8 -*-

import socket
from socket import SOL_SOCKET, SO_REUSEADDR
import subprocess
import struct
import json

PORT = 18284

#简单TCP通信
def main():
    tcpSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    print(tcpSocket)
    tcpSocket.setsockopt(SOL_SOCKET, SO_REUSEADDR, 1)
    tcpSocket.bind(('127.0.0.1', PORT))
    tcpSocket.listen(5) 
    print('start....')
    while True:
        conn, client_addr = tcpSocket.accept()
        print('new client connected ', conn, client_addr)
        while True: 
            try:
                print('recv data ...')
                data = conn.recv(1024) 
                if len(data) == 0: 
                    break 
                print('recv data is ', data)
                conn.send(data.upper())
            except ConnectionResetError:
                break

    conn.close()
    phone.close()

#仿写ssh服务程序
def main1():
    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server.bind(('127.0.0.1', PORT))
    server.listen(5)
    print('start...')
    while True:
        conn, client_addr = server.accept()
        while True:
            print('from client:', client_addr)
            cmd = conn.recv(1024)
            if len(cmd) == 0: 
                break
            print('cmd:', cmd)
            obj = subprocess.Popen(cmd.decode('utf8'), # 输入的cmd命令
                    shell=True, # 通过shell运行
                    stderr=subprocess.PIPE, # 把错误输出放入管道,以便打印
                    stdout=subprocess.PIPE) # 把正确输出放入管道,以便打印
 
            stdout = obj.stdout.read() # 打印正确输出
            stderr = obj.stderr.read() # 打印错误输出
 
            conn.send(stdout)
            conn.send(stderr)
        conn.close()
        server.close()

#自定义数据包
def main2():
    server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    server.bind(('127.0.0.1', PORT))
    server.listen(5)
    print('start...')
    while True:
        conn, client_addr = server.accept()
        print(conn, client_addr)
        while True:
            cmd = conn.recv(1024)
            obj = subprocess.Popen(cmd.decode('utf8'),
                                    shell=True,
                                    stderr=subprocess.PIPE,
                                    stdout=subprocess.PIPE)
            stderr = obj.stderr.read()
            stdout = obj.stdout.read()
            print("stderr:", stderr)
            print("stdout:", stdout)
            data_dict = {
                'body_size': len(stdout) + len(stderr),
                'body': stderr.decode('utf-8') + stdout.decode('utf-8')
            }
            data_json = json.dumps(data_dict)
            data_bytes = data_json.encode('utf8')
            conn.send(struct.pack('i', len(data_bytes)))
            conn.send(data_bytes)
            conn.close()
            break
    server.close()


if __name__ == '__main__':
    main2()

client.py

# -*- coding: UTF-8 -*-

import socket
import json
import struct

PORT = 18284

def main():
    cs = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    cs.connect(('127.0.0.1', PORT))
    while True: 
        msg = input('input data >>').strip() 
        if len(msg) == 0: 
            continue
        cs.send(msg.encode('utf-8'))
        data = cs.recv(1024)
        print(data)
    phone.close()


def main2():
    client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    client.connect(('127.0.0.1', PORT))
    while True:
        cmd = input('enter cmd >> ')
        if len(cmd) == 0: 
            continue
        #encode:字符串转字节数组
        client.send(cmd.encode('utf8'))
        data_len = struct.unpack('i', client.recv(4))[0]
        print("data_len: ", data_len)
        data_bytes = client.recv(data_len)
        print("data_bytes: ", data_bytes)
        #decode:字节数组转转字符串
        data_json = data_bytes.decode('utf8')
        print("data_json: ", data_json)
        data_dict = json.loads(data_json)
        print("data_dict: ", data_dict['body'])
        break
    client.close()
    
if __name__ == '__main__':
    main2()

运行效果如下:

python3套接字编程之socket和socketserver(TCP和UDP通信)_第1张图片

python3套接字编程之socket和socketserver(TCP和UDP通信)_第2张图片

python3套接字编程之socket和socketserver(TCP和UDP通信)_第3张图片

3.2 socketserver类型TCP

server.py

# -*- coding: UTF-8 -*-

import socketserver

PORT = 18286

class MyHandler(socketserver.BaseRequestHandler):
    def handle(self):
        while True:
            print(self.client_address)
            print(self.request)
            try:
                data = self.request.recv(1024)
                if len(data) == 0: break
                self.request.send(data.upper())
            except ConnectionResetError:
                break
            
if __name__ == '__main__':
    s = socketserver.ThreadingTCPServer(('127.0.0.1', PORT), 
                                        MyHandler, 
                                        bind_and_activate=True)
    s.serve_forever()

client.py

# -*- coding: UTF-8 -*-

import socket

PORT = 18286

def main():
    client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
    client.connect(('127.0.0.1', PORT)) 
    while True:
        msg=input('input msg >> ').strip()
        if len(msg) == 0: continue
        client.send(msg.encode('utf-8'))
        data = client.recv(1024)
        print(data.decode('utf-8'))
    client.close()

if __name__ == '__main__':
    main()

4.UDP

关于UDP,有如下需要注意:
(1)无连接的,先启动哪一端都不会报错,并且可以同时多个客户端去跟服务端通信
(2)数据报协议,发空的时候也会自带报头,因此客户端输入空,服务端也能收到,一般用于传输小数据
(4)无粘包问题,但是不能替代TCP套接字,因为UPD协议有一个缺陷:如果数据发送的途中,数据丢失,则数据就丢失了,而TCP协议则不会有这种缺陷,因此UPD套接字多用于无关紧要的数据发送,例如IM聊天工具

3.1 socket类型UDP

server.py

# -*- coding: UTF-8 -*-

import socket

PORT = 18285

def main():
    server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    server.bind(('127.0.0.1', PORT))
    while True:
        data, client_addr = server.recvfrom(1024)
        print('recvfrom:', data, client_addr)
        server.sendto(data.upper(), client_addr)
    server.close()

if __name__ == '__main__':
    main()

client.py

# -*- coding: UTF-8 -*-

import socket

PORT = 18285

def main():
    client = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    while True:
        msg = input('input >> ').strip()
        client.sendto(msg.encode('utf-8'), ('127.0.0.1', PORT))
        data, server_addr = client.recvfrom(1024)
        print(data, data.decode('utf-8'))
    client.close()

if __name__ == '__main__':
    main()


3.2 socketserver类型UDP

基于udp的socketserver自己定义的类,其中
self.request是一个元组(第一个元素是客户端发来的数据,第二部分是服务端的udp套接字对象)
self.client_address即客户端地址

server.py

# -*- coding: UTF-8 -*-

import socketserver

PORT = 18287

class MyHandler(socketserver.BaseRequestHandler):
    def handle(self):
        print(self.client_address)
        print(self.request)
        data = self.request[0]
        print('client msg:', data)
        self.request[1].sendto(data.upper(), self.client_address)
        
if __name__ == '__main__':
    s = socketserver.ThreadingUDPServer(('127.0.0.1', PORT), MyHandler)
    s.serve_forever()

client.py

# -*- coding: UTF-8 -*-

import socket

PORT = 18287

def main():
    client = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    while True:
        msg=input('input >> ').strip()
        client.sendto(msg.encode('utf-8'), ('127.0.0.1', PORT))
        data, server_addr = client.recvfrom(1024)
        print(data)
    client.close()

if __name__ == '__main__':
    main()

运行效果如下:

python3套接字编程之socket和socketserver(TCP和UDP通信)_第4张图片

 

python3套接字编程之socket和socketserver(TCP和UDP通信)_第5张图片

5.额外补充:strace分析Python中subprocess.Popen实现

5.1错误命令

服务端:strace  -ff -o ./output  python3 server.py 

客户端:python3 client.py 

运行结果:
python3套接字编程之socket和socketserver(TCP和UDP通信)_第6张图片

产生了两个文件,说明subprocess.Popen是通过多进程实现的。

主进程:

python3套接字编程之socket和socketserver(TCP和UDP通信)_第7张图片

python3套接字编程之socket和socketserver(TCP和UDP通信)_第8张图片

子进程:

python3套接字编程之socket和socketserver(TCP和UDP通信)_第9张图片

python3套接字编程之socket和socketserver(TCP和UDP通信)_第10张图片

python3套接字编程之socket和socketserver(TCP和UDP通信)_第11张图片

分析:主进程pipe2 [5, 6],pipe2 [7, 8],创建两个匿名管道,用来接收子进程正确输出和错误输出,主进程用5,7读取正确和错误信息,子进程6,8来写正确和错误信息。调用clone产生子进程, dup2(6, 1) dup2(8, 2),使用1,2来代替6,8。子进程调用execve来覆盖子进程进程空间并在其中执行命令。

5.2正确命令

服务端:strace  -ff -o ./output  python3 server.py 

 客户端:python3 client.py 

 运行结果:

python3套接字编程之socket和socketserver(TCP和UDP通信)_第12张图片

通过输出可以看到执行一次命令,会产生两个子进程。

主进程:

python3套接字编程之socket和socketserver(TCP和UDP通信)_第13张图片

python3套接字编程之socket和socketserver(TCP和UDP通信)_第14张图片

  子进程:

python3套接字编程之socket和socketserver(TCP和UDP通信)_第15张图片

python3套接字编程之socket和socketserver(TCP和UDP通信)_第16张图片

python3套接字编程之socket和socketserver(TCP和UDP通信)_第17张图片

 孙子进程: 

python3套接字编程之socket和socketserver(TCP和UDP通信)_第18张图片

python3套接字编程之socket和socketserver(TCP和UDP通信)_第19张图片 主进程14912调用clone产生子进程14986,子进程中通过execve执行sh程序来替换子进程空间, sh进程执行时又调用clone产生子进程14987,然后孙子进程14987调用execve来真正执行ls -i命令。 主进程 -> sh子进程 -> ls -i孙子进程

你可能感兴趣的:(socket,socketserver)