python 3.x 学习笔记14 (socket_ssh and socket_文件传输)

ssh服务端

import socket,os

server = socket.socket()
server.bind(('localhost',6666))

server.listen()
conn, addr = server.accept()
while True:
    print('等待新数据！')
    cmd = conn.recv(1024)

    if len(cmd) == 0:
        break

    data = os.popen(cmd.decode()).read()
    cmd_size = len(data.encode('utf-8'))
    print('检测到数据大小',cmd_size)

    if len(data) == 0:
        data = '没有此命令！！'

    conn.send(str(len(data.encode('utf-8'))).encode())
    order = conn.recv(1024)

    conn.send(data.encode('utf-8'))
    print('传送完成！')

server.close()

ssh客户端

import socket

client = socket.socket()
client.connect(('localhost',6666))
while True:
    cmd = input('>>>:')

    if cmd == 0:
        continue

    client.send(cmd.encode('utf-8'))
    cmd_size = client.recv(1024)
    print('服务端检测到数据大小：', cmd_size)

    if len(cmd_size) !=0:
        client.send('可以发送数据了'.encode(encoding='utf-8'))

    receive_size = 0
    receive_cmd = b''

    while receive_size < int(cmd_size.decode()):
        data = client.recv(1024)
        receive_size += len(data)
        receive_cmd += data
    else:
        print('接收到数据大小：',receive_size)
        print('数据接收完成！')
        print(receive_cmd.decode())

socket文件传输并校验

服务端

import socket,os,hashlib

server = socket.socket()
server.bind(('localhost',6666))
server.listen()
conn, addr = server.accept()
while True:
    print('等待新数据！')
    cmd = conn.recv(1024)
    if len(cmd) == 0:
        break
    filename = cmd.decode().split()[1]
    print(filename)
    if os.path.isfile(filename):          #判断文件是否存在
        f = open(filename,'rb')           #打开文件
        m = hashlib.md5()
        file_size = os.stat(filename).st_size   #查询文件大小
        print(file_size)
        conn.send(str(file_size).encode())         #给客户端发送文件大小
        conn.recv(1024)                              #等待并接受客户端的确认，这一步可以解决粘包问题
        for line in f:                              #边读边循环发送文件
            m.update(line)                           #得出每一句的md5值
            conn.send(line)                           #发送给客户端
        print('file md5:',m.hexdigest())
        f.close()
        conn.recv(1024)
        conn.send(m.hexdigest().encode())

    print('传送完成！')

server.close()

socket文件传输并校验

客户端

import socket,hashlib

client = socket.socket()
client.connect(('localhost',6666))
while True:
    cmd = input('>>>:').strip()
    if len(cmd) == 0:
        continue
    if cmd.startswith('get'):          #传输文件名要以get开头如：get var
        client.send(cmd.encode('utf-8'))
        file_size = client.recv(1024)
        print('服务端检测到文件大小：', int(file_size))
        client.send('可以发送数据了'.encode(encoding='utf-8'))
        file_total_size = int(file_size)
        receive_size = 0
        filename = cmd.split()[1]         #以空格分割然后取出文件名
        f = open('new_'+filename ,'wb')
        m = hashlib.md5()
        while receive_size < file_total_size:
            data = client.recv(1024)
            m.update(data)
            receive_size += len(data)
            f.write(data)
            print(file_total_size, receive_size)
        else:
            new_file_md5 = m.hexdigest()
            print('数据接收完成！')
            f.close()
            client.send('可以发送md5值了'.encode())
        serve_md5 = client.recv(1024)
        print('新文件md5:',new_file_md5)
        print('原文件md5:',serve_md5)

client.close()