day29 socketsever ftp功能简单讲解

时间：2023-03-09 02:05:11
day29 socketsever ftp功能简单讲解

今日所学

一.ftp上传简单实例

二.socketsever的固定用法

三.验证合法性连接

1.ftp上传实例

这个题目是我们现在网络编程比较基础一点的题目

day29 socketsever ftp功能简单讲解

下面我们只写简单上传的代码

上传服务端的代码

import socket

import struct

import json

sever=socket.socket()

ip_port=('127.0.0.1',8008)

sever.bind(ip_port)

sever.listen()

conn,addr=sever.accept()

#先接受4个字节,为文件的信息的长度

struct_len_info=conn.recv(4)

# 拿到字典的长度,(拿到的结果为元组,从元组中把数据拿出来)

data_len=struct.unpack('i',struct_len_info)[0]

# 然后根据拿到的长度,接收字典的字节类型

data_dict_bytes=conn.recv(data_len)

# 将字节转换为json字符串类型

data_json=data_dict_bytes.decode('utf-8')

# 将json字符串转换为字典

data_dict=json.loads(data_json)

sum=0

#算出文件的路径

file_path='相对路径'+'\\'+data_dict['file_name']

with open(file_path,mode='wb') as f:

    while sum<data_dict['file_size']:

        from_client_msg=conn.recv(1024)

        sum+=len(from_client_msg)

        f.write(from_client_msg)　　

上传客户端代码

import socket

import os

import struct

import json

client=socket.socket()

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

#读取文件的大小(读出来的内容是文件的字节类型)

file_size=os.path.getsize(r'相对路径')

# 在发送真实的文件之前,先将文件的信息包装成字典发送给服务端(文件名字,文件大小,文件路径)

file_info={

    'file_name':'XXX.XXX',

    'file_size' : file_size,

  'file_LuJing': '文件的路径'

}

#将字典序列化成json字符串

file_info_str=json.dumps(file_info)

#将json字符串编码成字节

file_info_byte=file_info_str.encode('utf-8')

#求出字典的字节长度

info_len=len(file_info_byte)

# 打包

info_struct=struct.pack('i',info_len)

# 将打包好的信息和长度发送给服务端

client.send(info_struct+file_info_byte)

with open('你要发送的文件.xxx',mode='rb') as f:

    #如果你累加的长度小于文件的长度,让他继续执行

    while sum<file_size:

        #每次读取的大小最大为1024

        every_read=f.read(1024)

        sum+=len(every_read)

        #将读取到的数据发送给服务端

        client.send(every_read)　　

注 : 上传和下载只是角色互换而已,上传为客户端向服务端中保存数据,而下载是客户端从服务端的内容中下载相应的文件.

2.socketsever的固定用法

话不多说,上代码

先上传服务端的代码

import socketsever

class 类名(socketsever.BaseRequestHandler):
# 注:这里的handle方法名是固定的,是不允许变的

    def handle(self):

        while 1:

            # self.request相当于conn

            from_client_msg=self.request.recv(1024)

            print(from_client_msg.decode('utf-8'))

            msg=input('服务端说:)

            self.request.send(msg.encode('utf-8'))

if __name__='__main__':

    ip_port=('127.0.0.1',8008)

    sever=socketsever.ThreadingTCPSever(ip_port,类名)

    sever.serve_forever()
以下是客户端的代码
import socket

client=socket.socket()

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

while 1:

    msg=input('客户端说>>>')

    if msg=='byebye':

        break

    client.send(msg.encode('utf-8'))

    from_sever_msg=client.recv(1024)

    print(from_sever_msg.decode('utf-8'))

client.close()

3. 验证合法性连接的服务端
服务端:
from socket import *

import hmac,os

#秘钥

secret_key=b'Jedan has a big key!'

def conn_auth(conn):

    print('开始验证新链接的合法性')

    msg=os.urandom(32)#生成一个32字节的随机字符串

    conn.sendall(msg)

    h=hmac.new(secret_key,msg)

    digest=h.digest()

    respone=conn.recv(len(digest))

    return hmac.compare_digest(respone,digest)

def data_handler(conn,bufsize=1024):

    if not conn_auth(conn):

        print('该链接不合法,关闭')

        conn.close()

        return

    print('链接合法,开始通信')

    while True:

        data=conn.recv(bufsize)

        if not data:break

        conn.sendall(data.upper())

def server_handler(ip_port,bufsize,backlog=5):

    '''

    只处理链接

    :param ip_port:

    :return:

    '''

    tcp_socket_server=socket(AF_INET,SOCK_STREAM)

    tcp_socket_server.bind(ip_port)

    tcp_socket_server.listen(backlog)

    while True:

        conn,addr=tcp_socket_server.accept()

        print('新连接[%s:%s]' %(addr[0],addr[1]))

        data_handler(conn,bufsize)

if __name__ == '__main__':

    ip_port=('127.0.0.1',9999)

    # 自己定义接收大小

    bufsize=1024

    server_handler(ip_port,bufsize)

　客户端:

from socket import *

import hmac,os

secret_key=b'Jedan has a big key!'

def conn_auth(conn):

    '''

    验证客户端到服务器的链接

    :param conn:

    :return:

    '''

    msg=conn.recv(32)

    h=hmac.new(secret_key,msg)

    digest=h.digest()

    conn.sendall(digest)

def client_handler(ip_port,bufsize=1024):

    tcp_socket_client=socket(AF_INET,SOCK_STREAM)

    tcp_socket_client.connect(ip_port)

    conn_auth(tcp_socket_client)

    while True:

        data=input('>>: ').strip()

        if not data:continue

        if data == 'quit':break

        tcp_socket_client.sendall(data.encode('utf-8'))

        respone=tcp_socket_client.recv(bufsize)

        print(respone.decode('utf-8'))

    tcp_socket_client.close()

if __name__ == '__main__':

    ip_port=('127.0.0.1',9999)

    bufsize=1024

    client_handler(ip_port,bufsize)

相关文章