socket浅析（二）： 构建通过socket通信的服务端和客户端

socket服务端代码：

 

import socket
import os,subprocess


server = socket.socket() #获得socket实例
server.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1)

ip_port = ('127.0.0.1', 10000) # 要绑定的IP及端口，元组数据形式

server.bind(ip_port) # 绑定ip port
server.listen()  # 开始监听

while True: # 第一层loop

    print("server is waiting...")
    conn,addr = server.accept() # 接受并建立与客户端的连接,程序在此处开始阻塞,只到有客户端连接进来...
    print("Connected by :",addr ) # 打印出当前是谁链接了客户端

    while True:
        data = conn.recv(1024)
        if not data:
            print("客户端断开了...")
            break # 这里断开就会再次回到第一次外层的loop

        print("收到要执行的命令:",data)
        #res = os.popen(data.decode()).read() #py3 里socket发送的只有bytes,os.popen又只能接受str,所以要decode一下
        res = subprocess.Popen(data,shell=True,stdout=subprocess.PIPE).stdout.read() # 跟上面那条命令的效果是一样的

        if len(res) == 0:
            res = "cmd exec success,has not output!".encode("utf-8")

        conn.send(str(len(res)).encode("utf-8")) # 发送数据之前,先告诉客户端要发多少数据给它
        client_final_ack = conn.recv(1024) # 等待客户端响应
        if client_final_ack.decode() == int(len(res)): # 如果收到客户端响应的标志位和发送的标志位相等，即代表客户端已经准备好接收数据了
            
            print(type(res))
            conn.sendall(res) #发送端也有最大数据量限制,所以这里用sendall,相当于重复循环调用conn.send,直至数据发送完毕

server.close()

 

 

socket客户端代码：

 

import socket
import sys

client = socket.socket() # 实例化socket
ip_port = ('127.0.0.1', 10000) # 要绑定的IP及端口，元组数据形式


client.connect(ip_port)

while True:

    msg = input(">>:").strip() # 获取客户端的输入信息并去除空格

    if len(msg) == 0:continue # 如果输入为空则继续循环等待输入

    client.send(msg.encode("utf-8"))  # 转码并发送给服务端

    res_return_size  = client.recv(1024) #接收这条命令执行结果的大小
    print("getting cmd result , ", res_return_size)

    total_rece_size = int(res_return_size)
    print("total size:",res_return_size)

    client.send(res_return_size.encode("utf-8"))
    received_size = 0 #已接收到的数据
    cmd_res = b''
    f = open("test_copy.html","wb")#把接收到的结果存下来,一会看看收到的数据 对不对
    while received_size != total_rece_size: #代表还没收完
        data = client.recv(1024)
        received_size += len(data) #为什么不是直接1024,还判断len干嘛,注意,实际收到的data有可能比1024少
        cmd_res += data
    else:
        print("数据收完了",received_size)
        #print(cmd_res.decode())
        f.write(cmd_res) #把接收到的结果存下来,一会看看收到的数据 对不对
        #print(data.decode()) #命令执行结果

client.close()

 

程序缺限：

这是一个简单的socket通信，里面存在一些bug

1.在客户端输入回车，会挂死。

2.服务端返回的数据大于1024，客户端显示不全。

3.单进程，如果多个客户端连接，要排队，前一个断开，后一个客户端才能通信。

先贴一个多线程的导图：