基于TCP协议的远程终端控制并发socketserver实现以及粘包问题处理

# 客户端

# -*- coding: utf-8 -*-

import socketserver
import struct
import json
import subprocess


class MyTcpHandler(socketserver.BaseRequestHandler):
    def handle(self):
        while True:
            try:
                cmd = self.request.recv(1024)
                if not cmd: break
                print(cmd.decode('utf-8'))
                obj = subprocess.Popen(cmd.decode('utf-8'),
                                       shell=True,
                                       stdout=subprocess.PIPE,
                                       stderr=subprocess.PIPE
                                       )
                stdout = obj.stdout.read()
                stderr = obj.stderr.read()
                # 制作报头
                header_dic = {
                    'file_name': cmd.decode('utf-8'),
                    'data_size': len(stdout) + len(stderr)
                }
                header_json = json.dumps(header_dic)
                header_bytes = header_json.encode('utf-8')
                # 发送报头字典长度
                self.request.send(struct.pack('i', len(header_bytes)))
                # 发送报头
                self.request.send(header_bytes)
                # 发送正文
                self.request.send(stdout + stderr)
            except ConnectionResetError:
                break


if __name__ == '__main__':
    server = socketserver.ThreadingTCPServer(('127.0.0.1', 8080), MyTcpHandler)
    server.serve_forever()

  

 

 

# 服务端

# -*- coding: utf-8 -*-

from socket import *
import struct
import json

client = socket(AF_INET, SOCK_STREAM)
client.connect(('127.0.0.1', 8080))

while True:
    cmd = input('>>>:').strip()
    if not cmd:
        continue
    client.send(cmd.encode('utf-8'))
    # 先接收报头长度并解析
    dic_size = client.recv(4)
    header_size = struct.unpack('i', dic_size)[0]
    # 获取报头字典
    header_json = client.recv(header_size).decode('utf-8')
    header_dic = json.loads(header_json)
    file_size = header_dic['data_size']
    get_size = 0
    info = b''
    while get_size < file_size:
        data = client.recv(1024)
        get_size += len(data)
        info += data
    print(info.decode('gbk'))

  

转载于:https://www.cnblogs.com/qianzhengkai/p/10821256.html

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值