day 28 黏包 ssh模块 subprocess模块

套接字的信仰

　　一切皆文件

昨日作业：

import socket

sock=socket.socket() # TCP协议
IP_PORT=("127.0.0.1",8899)
sock.bind(IP_PORT)
sock.listen(5)

while 1:
    conn,addr=sock.accept()
    while 1:
        try:
            data=conn.recv(1024).decode("utf8")
            user,pwd=data.strip().split("|")  #strip 是为了去掉接受到文件中的空格
                         #split是以|为分割线，分隔输入进来的字符串分割后形成列表

            # 文件操作
            flag=False
            with open("account","r") as f:

                for line in f:
                    username,password=line.strip().split(":")
                    if username==user and password==pwd:
                        flag=True
                        break
            if flag:
                conn.send(b"success")
            else:
                conn.send(b"fail")
        except Exception as e:
            break                                                    

服务端

import socket

sock=socket.socket() # TCP
sock.connect(("127.0.0.1",8899))

while 1:
    user=input("用户名>>>>")
    pwd=input("密码>>>>")
    val=("%s|%s"%(user,pwd)).encode("utf8")
    sock.send(val)
    response=sock.recv(1024)
    print(response.decode("utf8"))
    if response.decode("utf8")=="success":
        break
    else:
        print("用户名或者密码错误！")
        continue

客户端

一、模拟ssh

　　调用执行命令的程序：

import subprocess

res=subprocess.Popen("dir",
                     shell=True,
                     stderr=subprocess.PIPE,
                     stdout=subprocess.PIPE)

print(res.stdout.read().decode("gbk"))

subprocess

同时执行多条命令之后，得到的结果很可能只有一部分，在执行其他命令的时候又接收到之前执行的另外一部分结果，这种显现就是黏包。

二、黏包

import struct

res=struct.pack("i","52526") #i 是一个整形模式

print(res) #b'Q\xe4\x06\x00' 是一个看不懂四个字节

print(len(res))

obj=struct.unpack("i",res)

print（obj） #（52526，）

print(obj[0]) #取元组的第一项

#功能，把一组数据转化为一个4位数字是东西
# 构造报头数据

hashlib 的经典用法:做数据移植性校验，就是本地上用一次，对面用一次，利用不可逆性，发生修改的话队形的hashlib将会不一样
import hashlib

md5=hashlib.md5()
md5.update(b"hello")
md5.update(b"yuan")

print(md5.hexdigest())
print(len(md5.hexdigest()))

# # helloyuan:   d843cc930aa76f7799bba1780f578439      两次加密的效果是一样的
# #              d843cc930aa76f7799bba1780f578439

#############################################

md5=hashlib.md5()

with open("ssh_client.py","rb") as f:
    for line in f:   #文件较大的话，一次打开太占内存
        md5.update(line) #此方法和md5.update(f.read())的结果是一样的 和上面的效果是一样

print(md5.hexdigest()) # f.read()

课上练习题：

 # by luffycity.com

 import socket
 import subprocess

 server = socket.socket()

 server.bind(('127.0.0.1',8008))

 server.listen(5)

 while True:
     print("server is working.....")
     conn,addr = server.accept()
     # 字节类型
     while True:
         # 针对window系统
         try:
             cmd = conn.recv(1024).decode("utf8") # 阻塞

             if cmd == b'exit':
                 break

             res=subprocess.Popen(cmd,
                              shell=True,
                              stderr=subprocess.PIPE,
                              stdout=subprocess.PIPE,
                              )
             # print("stdout",res.stdout.read())
             # print("stderr",res.stderr.read().decode("gbk"))
             out=res.stdout.read()
             err=res.stderr.read()

             print("out响应长度",len(out))
             print("err响应长度",len(err))
             if err:
                  import struct
                  header_pack = struct.pack("i", len(err))
                  conn.send(header_pack)
                  conn.send(err)
             else:
                  #构建报头
                  import struct
                  header_pack=struct.pack("i",len(out))
                  print("header_pack",header_pack)
                  # # 发送报头
                  conn.send(str(len(out)).encode("utf8"))
                  # 发送数据
                  conn.send(out)

         except Exception as e:
             break

     conn.close()

 import socket
 import struct
 sk = socket.socket()

 sk.connect(('127.0.0.1',8008))

 while 1:
     cmd = input("请输入命令：")
     sk.send(cmd.encode('utf-8')) # 字节
     if cmd=="":
         continue
     if cmd == 'exit':
         break

     # header_pack=sk.recv(4)
     # data_length=struct.unpack("i",header_pack)[0]
     # print("data_length",data_length)
     '''
     b'xxx/xxx/xxx/bbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbbb'
     '''
     data_length=int(sk.recv(1024).decode("utf8"))
     print("data_length",data_length)

     recv_data_length=0
     recv_data=b""

     while recv_data_length<data_length:
         data=sk.recv(1024)
         recv_data_length+=len(data)
         recv_data+=data

     print(recv_data.decode("gbk"))

 sk.close()