python学习 day29之基于TCP协议通信的套接字程序

服务端

"""
服务端应该满足的特性：
    1、一直对外提供服务
    2、并发地提供服务
"""

import socket
import subprocess

# 1、买手机
phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # SOCK_STREAM=》TCP协议

# 2、插手机卡
phone.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)  # 重置端口号，以免端口发生冲突

phone.bind(("127.0.0.1", 8080))  # 本地回环

# 3、开机
phone.listen(5)
print('starting %s:%s' % ("127.0.0.1", 8080))

# 4、等电话链接=>链接循环
while True:
    conn, client_addr = phone.accept()
    print(client_addr)
    # 5、收/发消息=>通信循环
    while True:
        try:
            cmd = conn.recv(1024)  # 最大接收的字节个数
            if len(cmd) == 0:  # 针对linux系统，客户端断开连接时
                break

			# 将客户端发来的命令输入给CMD，并返回成功信息与失败信息
            obj = subprocess.Popen(cmd.decode('utf-8'),
                                   shell=True,
                                   stdout=subprocess.PIPE,
                                   stderr=subprocess.PIPE
                                   )

            res=obj.stdout.read()+obj.stderr.read()
            print(res.decode('gbk')
            conn.send(res)
        except Exception:  # 针对windows系统,客户端断开连接时
            break

    # 6、关闭
    conn.close()  # 挂电话
phone.close()  # 关机

客户端

import socket

# 1、买手机
phone = socket.socket(socket.AF_INET, socket.SOCK_STREAM)  # SOCK_STREAM=》TCP协议

# 2、拨电话
phone.connect(("127.0.0.1", 8080))

# 3、发/收消息=>通信循环
while True:
    cmd = input("[root@localhost]# ").strip()
    phone.send(cmd.encode('utf-8'))
    data = phone.recv(1024)
    print(data.decode('gbk'))

# 4、关闭
phone.close()