python模块——socket (实现简单的C/S架构端通信操作CMD）

提示：扫一扫查出行
【扫一扫了解最新限行尾号】
复制提示

# 服务端代码
#!/usr/bin/env python
# -*- coding:utf-8 -*-
__author__ = "loki"
import socket
import subprocess
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)
ip_port = ('127.0.0.1', 8011)
server.bind(ip_port)
server.listen(5)# 链接循环
while 1:
print('Waiting...')
conn, addr = server.accept()
print('-->conn: ', conn)
print('-->addr: ', addr)
print('Got it...') while 1:
try:
cmd = conn.recv(1024)
res = subprocess.Popen(cmd.decode('utf-8'),
shell=True,
stdout=subprocess.PIPE,
stderr=subprocess.PIPE)
conn.send(res.stdout.read())
conn.send(res.stderr.read())
except Exception:
break
conn.close()
# server.close()

客户端
#!/usr/bin/env python
# -*- coding:utf-8 -*-
__author__ = "loki"
import socketclient = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
ip_port = ('127.0.0.1', 8011)
client.connect(ip_port)
# 通信循环
while 1:
# 发消息
cmd = input('>>: ').strip()
if not cmd:
continue
client.send(bytes(cmd, encoding='utf-8'))
# 收消息
data = client.recv(8196)
print(data.decode('gbk'))
# client.close()

以上代码是基于TCP 流的方式进行C/S控制，注意编码Linux适用于utf-8，Windows适用于gbk

已知BUG：

1.在接受的信息大于8196（或者你自己定义的值的时候）可能会出现粘包问题

2.执行可执行程序或者需要交互的cmd命令，client会宕住，需要重新开客户端

3.如果是作为纯py脚本使用还需要进行sys.platform的系统判断，如果是win采用编码gbk,linux采用编码utf-8

4.以上代码仅供学习参考使用

# 解决粘包问题的TCP版本 —— server

#!/usr/bin/env python
# -*- coding:utf-8 -*-
__author__ = "loki"
import socket
import subprocess
import struct
user_input = input('Please input server_ip: ').strip()ip_port = ('%s' % user_input, 9991)
buff_size = 1024stick_pack_server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
stick_pack_server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)stick_pack_server.bind(ip_port)
stick_pack_server.listen(5)while 1:
print('Waiting...')
msg, address = stick_pack_server.accept()
print("msg-->: ", msg)
print("addr-->: ", address)
while 1:
try:
cmd = msg.recv(buff_size)
if not msg:
break
res = subprocess.Popen(cmd.decode("utf-8"), shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE,
stdin=subprocess.PIPE)
stderr = res.stderr.read()
stdout = res.stdout.read() data_size = len(stderr) + len(stdout) # send header
msg.send(struct.pack("i", data_size)) # send real data
msg.send(stderr)
msg.send(stdout)
except Exception as e:
print('---->', e)
break
msg.close()
# phone.close()

# 解决粘包问题的TCP版本 —— client

#!/usr/bin/env python
# -*- coding:utf-8 -*-
__author__ = "loki"
import socket
import structuser_input = input("Please input client_ip: ").strip()ip_port = ('%s' % user_input, 9991)
buff_size = 1024stick_pack_client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
stick_pack_client.connect(ip_port)while 1:
# send message
cmd = input('>>: ').strip()
if not cmd:
continue
stick_pack_client.send(cmd.encode("utf-8")) # receive header
baotou = stick_pack_client.recv(4)
data_size = struct.unpack("i", baotou)[0] # receive data
receive_size = 0
receive_data = b''
while receive_size < data_size:
data = stick_pack_client.recv(1024)
receive_size += len(data)
receive_data += data print(receive_data.decode("gbk"))# stick_pack_client.close()