python运维开发之第八天（socket）

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author  : Willpower-chen
# @blog: http://www.cnblogs.com/willpower-chen/

import socket,os,time    #导入socket和os模块

#实例化，AF_INET地址簇ipv4协议，SOCK_STREAM这是TCP/IP协议
server = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
server.bind((‘0.0.0.0‘,9999))    #绑定地址及服务端口
server.listen(5)                        #限制连接数
while True:                        #当一个client端断开时，接收下一个客户端
    # 每个客户端发送过来的数据，生成相应的实例conn，client端连接地址addr
    conn,addr = server.accept()
    print(‘新的连接：‘,addr)
    while True:
        print(‘等待新的指令‘)
        data = conn.recv(1024)    #接收client端发来的数据
        if not data:
            print(‘客户端已断开‘)
            break        #当数据为空时，跳出循环等待下一个连接
        print(‘执行指令‘,data.decode())    #打印要执行的命令
        cmd_res = os.popen(data.decode()).read()#返回client发过来的cmd命令，在server端的执行结果
        #由于每次接收数据的大小为1024，为了确定client需要接收几次，所以需要先把总数据大小发给client
        print(‘开始发送‘,len(cmd_res))
        if len(cmd_res) == 0 :     #如果cmd执行结果为空时，server端不会发送数据
            cmd_res = ‘cmd_res is not info‘    #为了防止为空时不发数据，自定义一个数据
        print(‘执行命令结果大小：‘,len(cmd_res))#打印命令执行结果数据大小
        conn.send(str(len(cmd_res.encode(‘utf-8‘))).encode(‘utf-8‘))    #给client发送数据总大小信息
        #time.sleep(0.5)
        client_ack = conn.recv(1024)    #等待client端确认ack，防止粘包
        print(‘ack from client‘,client_ack.decode(‘utf-8‘))    #打印ack确认信息
        conn.send(cmd_res.encode(‘utf-8‘))    #给client发送命令执行结果数据
        print(‘发送完毕‘)
server.close()

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Author  : Willpower-chen
# @blog: http://www.cnblogs.com/willpower-chen/
import socket

client = socket.socket() #默认地址簇是ipv4，传输协议是TCP/IP
client.connect((‘localhost‘,9999))#与server建立连接
while True:
    cmd = input(‘>>:‘).strip()  # 用户输入cmd命令
    if len(cmd) == 0: continue #如果用户输入为空，循环输入
    client.send(cmd.encode(‘utf-8‘))#发送cmd命令，encode是把字符串转为byte(python3.0传输只能是二进制)
    cmd_res_size = client.recv(1024)#确认接收命令结果的长度
    print(‘命令结果大小‘,cmd_res_size.decode())    #打印命令结果的长度
    client.send(‘准备好接收了，loser可以发了‘.encode(‘utf-8‘))#ack确认，防止粘包，注意这里不能用b‘‘因为加中文了
    receive_size = 0    #已经接收的数据，默认先设置为0
    receive_data = ‘‘.encode()
    while receive_size < int(cmd_res_size.decode()): #已接收数据小于总数据时，循环接收数据
        data = client.recv(1024)
        receive_size += len(data) #每次收到的有可能小于1024，所以用len（data）计算
        receive_data += data
    else:
        print(‘cmd res receive done...‘,receive_size)
        print(receive_data.decode(‘utf-8‘))
client.close()