python练习-Socket实现远程cmd命令

需求:基于tcp的套接字实现远程执行命令的操作

代码示例:

 # 编辑者：闫龙
 #Client端部分
 import socket #导入骚凯特模块
 CmdObj = socket.socket(socket.AF_INET,socket.SOCK_STREAM)#实例化骚凯特对象
 #基于网络类型(AF_INET)的套接字,基于TCP协议(SOCK_STREAM)
 CmdObj.connect(("127.0.0.1",5741))#建立与服务端的通信
 print(CmdObj.recv(1024).decode("utf8"))#显示服务端的欢迎信息
 while True:
     CmdScript = input(">>>:")#写入想要执行的cmd命令
     if not CmdScript: continue #判断是否为空,避免服务端出现循环错误
     if(CmdScript == "exit"):#先判断是否退出,如果不退出则继续执行
         print("感谢使用,再见!")
         break
     CmdObj.send(CmdScript.encode("utf8"))#将cmd命令转为bytes传输给服务端
     Recv = CmdObj.recv(1024)#接收服务端传来的bytes信息
     try:
         print(Recv.decode("gbk"))
         #尝试将bytes信息使用gbk方式编码,在windows下的编码是gbk,不过如果编码错误的话还可以通过捕获异常来重新编码
     except UnicodeDecodeError as ude:#判断异常为转码异常
         print(Recv.decode("utf8"))#使用utf8编码bytes信息(这里其实就是为了返回服务端对egon进行处理的那条信息)

客户端代码

 # 编辑者：闫龙
 import socket,subprocess#导入骚凯特模块和命令行交互模块
 CmdObj = socket.socket(socket.AF_INET,socket.SOCK_STREAM)
 #实例化骚凯特对象,基于网络类型(AF_INET)的TCP流传输(SOCK_STREAM)
 CmdObj.bind(("127.0.0.1",5741))#建立服务端的绑定信息服务IP是127.0.0.1,端口号是5741
 CmdObj.listen(5)#建立监听,后备链接最大值为5
 while True:
     Clients,ClientIP = CmdObj.accept()#等待客户端的连接.........
     #并获得连接句柄Clients,和连接地址信息ClientIP
     Clients.send(("您当前的登陆IP地址为:%s"%(ClientIP[0])).encode("utf8"))
     #发送给客户端欢迎信息
     while True:
         try:
             CmdScript = Clients.recv(1024)#接收客户端传来的bytes数据
             if not CmdScript : continue#如果为空跳出本次循环,避免服务端和粗线死循环情况
             if(CmdScript.decode("utf8") == "egon"):#这里是个趣味性代码,认真学习者可忽略
                 Clients.send("SomeBody不是一个有效的dos命令!".encode("utf8"))
                 continue
             Res = subprocess.Popen(CmdScript.decode("utf8"),shell=True,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
             #实例化subprocess对象,获得由客户端发来的命令执行后的结果
             ResData = Res.stdout.read()#读取命令执行成功的结果
             if not ResData:#如果读取命令执行成功的结果失败
                 ResData = Res.stderr.read()#则读取命令执行失败的结果
             Clients.send(ResData)#将结果返回给客户端,这里由于是在windows下所以默认的bytes编码格式是gbk
         except Exception as  e:
             print(e)
             break
     Clients.close()
 CmdObj.close()

服务器端代码

以上,仅供参考,还是希望大家可以写出属于自己的完美代码