socket 远程执行命令事例

场景 ：基于socket，进行远端连接以及运行命令，并把远端命令的执行结果返回。

服务端：

import socket as sk,subprocess as ss
sk_obj = sk.socket(sk.AF_INET,sk.SOCK_STREAM)  #tcp协议的sock连接
sk_obj.bind(('127.0.0.1',8000))
sk_obj.listen(5)
while True:
    conn,ipaddr  = sk_obj.accept()  #阻塞链接
    print('当前连接IP:%s '%ipaddr[0])
    while True:
        try:
            from_recv = conn.recv(8096)  #接收对端发来的消息
            if len(from_recv) == 0 :continue
            print("来自客户端ip:%s 消息:%s"%(ipaddr[0],from_recv))
            res = ss.Popen(from_recv.decode('utf-8'),shell=True,stdout=ss.PIPE,stderr=ss.PIPE)
            msg = res.stdout.read()
            if len(msg) == 0:
                msg = res.stderr.read()
            conn.send(msg)
        except Exception :
            break
    conn.close()
sk_obj.close()

客户端:

import socket as sk
sk_obj = sk.socket(sk.AF_INET,sk.SOCK_STREAM)
sk_obj.connect(('127.0.0.1',8000))

while True:
    msg = input(">>>").strip()
    if len(msg) == 0 :continue
    sk_obj.send(msg.encode('utf-8'))
    data = sk_obj.recv(8096) #1024
    print("服务端发来消息：%s" % data.decode('gbk'))  #这里用gbk是因为windows默认系统编码格式为GBK
sk_obj.close()

 client运行结果:

 1 >>>ddddd
 2 服务端发来消息：'ddddd' 不是内部或外部命令，也不是可运行的程序
 3 或批处理文件。
 4 
 5 >>>dir .
 6 服务端发来消息： 驱动器 J 中的卷没有标签。
 7  卷的序列号是 AC23-3670
 8 
 9  J:pdataday36job 的目录
10 
11 2017/05/04  02:03    <DIR>          .
12 2017/05/04  02:03    <DIR>          ..
13 2017/05/04  02:03               333 job_client.py
14 2017/05/04  02:03               807 job_server.py
15                2 个文件          1,140 字节
16                2 个目录  5,046,833,152 可用字节

 server运行结果：

J/day36/job/job_server.py
当前连接IP:127.0.0.1 
来自客户端ip:127.0.0.1 消息:b'ddddd'
来自客户端ip:127.0.0.1 消息:b'dir .'
当前连接IP:127.0.0.1 

 解决粘包：

#coding:utf-8
import socket as sk,subprocess as ss
import struct
import json
sk_obj = sk.socket(sk.AF_INET,sk.SOCK_STREAM)
sk_obj.setsockopt(sk.SOL_SOCKET,sk.SO_REUSEADDR,1)  #定制重连接
sk_obj.bind(('192.168.16.132',8000))
sk_obj.listen(5)
while True:
    conn,ipaddr = sk_obj.accept()
    print("客户机ip:%s" %ipaddr[0])
    while True:
        try:
            from_recv = conn.recv(1024)
            print(from_recv.encode('utf-8'))
            if len(from_recv) == 0:break
            res = ss.Popen(from_recv.decode('utf-8'),
                           shell=True,
                           stdout=ss.PIPE,
                           stderr=ss.PIPE
                           )

            msg_stdout = res.stdout.read()
            msg_stderr = res.stderr.read()

            # 定制包头大小
            data_size = len(msg_stdout) + len(msg_stderr)
            header_dict = {'data_size': data_size}
            header_json = json.dumps(header_dict)
            header_json_bytes = header_json.encode('utf-8')
            head_json_bytes_len = struct.pack('i', len(header_json_bytes))

            # parit1 发送包头长度
            conn.send(head_json_bytes_len)
            # parit2 发送包头
            conn.send(header_json_bytes)
            # 真实消息发送
            conn.send(msg_stdout)
            conn.send(msg_stderr)
        except Exception:
            break
    conn.close()
sk_obj.close()

server端

 1 import socket as sk ,subprocess as ss
 2 import struct
 3 import json
 4 sk_obj = sk.socket(sk.AF_INET,sk.SOCK_STREAM)
 5 sk_obj.connect(('192.168.16.132',8000))#linux
 6 # sk_obj.connect(('127.0.0.1',8000))
 7 while True:
 8     msg_input = input(">>>:").strip()
 9     if not msg_input: continue
10     sk_obj.send(msg_input.encode('utf-8'))
11     #sk_obj.send(bytes(msg_input, encoding='utf-8'))
12     header_struct = sk_obj.recv(4)
13     header_bytes_len = struct.unpack('i',header_struct)[0]
14     header_bytes = sk_obj.recv(header_bytes_len)
15     header_json = json.loads(header_bytes.decode('utf-8'))
16     print(header_json)
17     header_size = header_json['data_size']
18     recvsize  = 0
19     recv_data = b''
20     while recvsize < header_size:
21         data = sk_obj.recv(1024)
22         recvsize += len(data)
23         recv_data += data
24 
25     # data = sk_obj.recv(1024)
26     print("来自客户端消息%s"%recv_data.decode('gbk'))
27 sk_obj.close()

服务端运行结果：

[root@centnfs ~]# python Ftpserver.py 
客户机ip:192.168.16.132
dir
ls /tmp

++++++++++++++++++++++++++++++++++

客户端运行结果：

>>>:dir
{'data_size': 280}
来自客户端消息ab.sh         db        Ftpserver.py      mon_rpcd.sh    Pictures
ach.sh         Desktop    initial-setup-ks.cfg  Music        Public
anaconda-ks.cfg  Documents  install_ngx.sh      nfs.sh    Templates
a.txt         Downloads  mon_nfs.sh          nginx.conf    test.txt
checkmem.sh     d.txt        mon_nginx.sh      onkeyinstall    Videos

>>>:ls /tmp
{'data_size': 191}
来自客户端消息systemd-private-e7b8cd4ded9b474c9a3d50fbae61134e-cups.service-FDqOvd
systemd-private-e8a7a17af09640a085e380f1a076a6fb-cups.service-vc3Hrz
VMwareTools-8.8.2-590212.tar.gz
vmware-tools-distrib