day8-------socket网络编程

简单的socket  一个server同时只能处理一个链接   代码如下：

 server端代码
 #author = ruixin li

 import socket
 server = socket.socket()
 server.bind(("0.0.0.0",5000))
 server.listen(5)
 print("开始监听")
 while True:
     conn,addr = server.accept()
     print("一个新连接")
     print(conn,addr)
     while True:
         try:
             data = conn.recv(1024)
             print("收到的信息：",data)
             conn.send(data.upper())
         except ConnectionResetError as e:
             print("客户端主动断开连接")
             break
 server.close()

 客户端程序

 import socket

 cli = socket.socket()
 cli.connect(("localhost",5000))
 while True:
     inp = input("请输入指令：".strip())
     if len(inp) == 0:
         continue
     else:
         cli.send(inp.encode())
         data = cli.recv(1024)
         print(data)

python socket 实现简单的 ssh  命令

 #author  ruixin li

 import socket
 import subprocess
 ips = input("请输入绑定的IP：")
 ports = int(input("请输入绑定的端口："))
 server = socket.socket()
 server.bind((ips,ports))
 server.listen(5)
 print("开始监听")
 while True:
     print("跳到外部循环")
     conn,addr = server.accept()
     print("一个新连接")
     print(conn,addr)
     while True:
         try:
             date = conn.recv(1024)
             if len(date) == 0:
                 print("收到0信息，服务器主动断开连接")
                 break
             else:
                 print("收到的信息：",date)
                 msg = subprocess.Popen(date,stdout=subprocess.PIPE,stderr=subprocess.PIPE,shell=True)
                 data = msg.stdout.read()
                 print(data)
                 dataerr = msg.stderr.read()
                 print("err:",dataerr)
                 if len(data) == 0:
                     conn.send(dataerr)
                 else:
                     conn.send(data)
         except ConnectionResetError as e:
             print("客户端主动断开连接")
             break
 server.close()

客户端

 import socket
 ips = input("请输入要连接的IP：")
 ports = int(input("请输入要连接的端口："))

 cli = socket.socket()
 cli.connect((ips,ports))
 while True:
     inp = input("请输入指令：".strip())
     if len(inp) == 0:
         continue
     else:
         cli.send(inp.encode())
         data = cli.recv(1024)
         print(data)

socket 实现简单的FTP上传下载功能

 #客户端代码
 #author ： ruixin li

 import socket
 import hashlib
 import os
 #ips = input("请输入要连接的IP：")
 #ports = int(input("请输入要连接的端口："))

 print("说明：下载服务器文件到当前路径")
 cli = socket.socket()
 cli.connect(("localhost",5000))
 while True:
     inp = input("请输入指令： 按Q退出： ".strip())
     if len(inp) == 0:
         continue
     elif inp == "q":
         exit()
     elif inp.split()[0] == "get":
         inp_file_name = inp.split()[1]
         cli.send(inp.encode())
         tmp = cli.recv(1024).decode()
         print("文件大小",tmp)
         print(type(tmp))
         remote_file_size = int(tmp)
         print("服务端返回文件总大小：",remote_file_size)
         local_size = 0
         cli.send("收到大小，请发文件内容".encode())
         file_name = inp.split()[1]
         f = open(file_name.split("/")[-1] + ".new","wb")
         while local_size < remote_file_size:
             #f = open(file_name + ".new","wb")
             data = cli.recv(1024)
             f.write(data)
             local_size += len(data)
            # f.close()
             print("文件总大小：",remote_file_size,"接收到:",local_size)
         else:
             print("文件下载完成，文件总大小：",remote_file_size,"接收到:",local_size)
             f.close()
             cli.send("请发送MD5".encode())
             remote_md5 = cli.recv(1024).decode()
             print("接收到服务端的MD5:",remote_md5)
             f = open(file_name.split("/")[-1] + ".new", "rb")
             m2 = hashlib.md5()
             for i in f:
                 m2.update(i)
             print("本地文件MD5",m2.hexdigest())
             f.close()
             if m2.hexdigest() == remote_md5 :
                 print("文件校验成功")
             else:
                 print("文件下载不完整")
     elif inp.split()[0] == "put":
         inp_file_name = inp.split()[1]
         cli.send(inp.encode())

         if os.path.isfile(inp_file_name):

             put_file_size = str(os.stat(inp_file_name).st_size).encode()
             cli.send(put_file_size)
             tmp = cli.recv(1024)
             print("等待确认",tmp.decode())
             f = open(inp_file_name,"rb")
             for line in f:
                 cli.send(line)
             f.close()
             print("文件发送完毕")

 cli.close()

 #服务端代码

 import socket
 import subprocess
 import os
 import hashlib
 #ips = input("请输入绑定的IP：")
 #ports = int(input("请输入绑定的端口："))
 server = socket.socket()
 server.bind(("localhost",5000))
 server.listen(5)
 print("开始监听")
 while True:
     print("跳到外部循环")
     conn,addr = server.accept()
     print("一个新连接")
     print(conn,addr)
     while True:
             date = conn.recv(1024)
             if len(date) == 0:
                 print("收到0信息，服务器主动断开连接")
                 break
             else:
                 print("收到的信息：",date)
                 args1,file_name = date.split()
                 print("第一个参数",args1.decode(),"第二个参数",file_name.decode())
                 if args1.decode() == "get":
                     print("走下载")
                     if os.path.isfile(file_name.decode()):
                         file_total_size = str(os.stat(file_name.decode()).st_size).encode()
                         print(file_total_size)
                         print(type(file_total_size))
                         conn.send(file_total_size)
                         conn.recv(1024)
                         print("等待客户端ack应答")
                         f = open(file_name.decode(),"rb")
                         m = hashlib.md5()
                         for line in f:
                             conn.send(line)
                             m.update(line)
                         hash_md5 = (m.hexdigest())
                         print("文件MD5：",hash_md5)
                         f.close()
                         print("文件发送完毕")
                         tmp = conn.recv(1024)
                         print(tmp.decode())
                         conn.send(hash_md5.encode())
                     else:
                         print("文件不存在")
                         continue
                 elif args1.decode() == "put":
                     print("开始上传文件")
                     # if  os.path.isfile(file_name.decode().split("/")[-1]):
                     #     print("文件已存在")
                     #     conn.send("文件已存在".encode())

                     remote_file_size = int(conn.recv(1024).decode())
                     print("收到客户端文件大小：",remote_file_size)
                     put_file_size = 0
                     conn.send("可以发送数据了".encode())
                     f = open(file_name.decode().split("/")[-1],"wb")
                     while put_file_size < remote_file_size:
                         recv_data = conn.recv(1024)
                         f.write(recv_data)
                         put_file_size += len(recv_data)
                     f.close()
                     print("文件总大小：",remote_file_size,"接收到的大小",put_file_size)