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

# 服务端代码

#!/usr/bin/env python#-*- coding:utf-8 -*-

__author__ = "loki"

importsocketimportsubprocess

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())exceptException:breakconn.close()#server.close()

客户端

#!/usr/bin/env python#-*- coding:utf-8 -*-

__author__ = "loki"

importsocket

client=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 notcmd:continueclient.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"

importsocketimportsubprocessimportstruct

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 notmsg:breakres= 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)exceptException as e:print('---->', e)breakmsg.close()#phone.close()

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

#!/usr/bin/env python#-*- coding:utf-8 -*-

__author__ = "loki"

importsocketimportstruct

user_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 notcmd:continuestick_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= stick_pack_client.recv(1024)

receive_size+=len(data)

receive_data+=dataprint(receive_data.decode("gbk"))#stick_pack_client.close()

Wrap-up

1.注意tcp客户端需要使用 xx.connect()

2.注意tcp服务端发送消息是使用的accept，以及发送消息使用的是conn