第33天 初识socket编程

# 服务端代码

import  socket
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind(('127.0.0.1', 65535))
server.listen(5)  # 此处定义半连接数为5，也就是当前没有连接成功的数量必须小于等于5
while True:
    conn, addr = server.accept()
    while True:
        try:
            res = conn.recv(1024)
            print('服务端发送的数据', res)
        except Exception as e:
            print(e)
            break
    conn.close()


# 客户端
import  socket
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

client.connect(('127.0.0.1', 65535))
while True:
    msg = input('>>>:')
    client.send(msg.encode('utf-8'))端代码

import socket
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind(('127.0.0.1', 65535))
server.listen(5)
conn, addr = server.accept()
data = conn.recv(1024)
conn.send(data.upper() + b'sb')

import socket
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect(('127.0.0.1', 65535))
msg = input('>>')
client.send(msg.encode('utf-8'))
data = client.recv(1024)
print('服务器回复的数据', data)

import socket
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect(('127.0.0.1', 65535))
while True:
    msg = input('>>')
    client.send(msg.encode('utf-8'))
    data = client.recv(1024)
    print('服务器回复的数据', data)

import socket
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind(('127.0.0.1', 65535))
server.listen(5)
conn, addr = server.accept()
while True:
    data = conn.recv(1024)
    conn.send(data.upper() + b'sb')

import socket
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind(('127.0.0.1', 65535))
server.listen(5)
conn, addr = server.accept()
while True:
    # 在此处捕捉异常，一旦捕捉到异常则退出循环关闭当前连接
    try:
        data = conn.recv(1024)
        conn.send(data.upper() + b'sb')
    except Exception as e:
        print(e)
        break
conn.close()
server.close()

import socket
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect(('127.0.0.1', 65535))
while True:
    msg = input('>>')
    if not msg:
        continue
    client.send(msg.encode('utf-8'))
    data = client.recv(1024)
    print('服务器回复的数据', data)

import socket
server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind(('127.0.0.1', 65535))
server.listen(5)
while True:
    conn, addr = server.accept()
    while True:
        try:
            data = conn.recv(1024)
            conn.send(data.upper() + b'sb')
        except Exception as e:
            print(e)
            break
    conn.close()
server.close()

import socket
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect(('127.0.0.1', 65535))
while True:
    msg = input('>>')
    if not msg:
        continue
    client.send(msg.encode('utf-8'))
    data = client.recv(5)   # 此时接收的字节数改变了
    print(data)

import socket
import subprocess
import struct

server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind(('127.0.0.1', 65535))
server.listen(5)
while True:
    conn, addr = server.accept()
    while True:
        try:
            cmd = conn.recv(1024)
            print(cmd)
            obj = subprocess.Popen(
                cmd.decode('utf-8'),
                shell=True,
                stdout=subprocess.PIPE,
                stderr=subprocess.PIPE
            )
            print(obj)
            # 注意此处的stdout和stderr不能位置不能相反，必须先读out后读err
            stdout = obj.stdout.read()
            print(stdout)
            stderr = obj.stderr.read()
            print(stderr)
            # 计算数据的长度,返回的是一个字节，在客户端先读取四个字节获得数据长度
            head = struct.pack('i', len(stderr + stdout))
            print(head)
            # 拼接数据，并进行发送
            data = head + stderr + stdout
            print(len(stdout) + len(stderr))
            conn.send(data)
        except Exception as e:
            print(e)
            break
    conn.close()

import socket
import struct
client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect(('127.0.0.1', 65535))
while True:
    msg = input('>>')
    if not msg:
        continue
    client.send(msg.encode('utf-8'))
    head = client.recv(4)    # 获得报头
    total_len = struct.unpack('i', head)[0]  # 将报头中前四个字节解压得到数据的总长度
    current_len = 0    # 当前读取的报文长度
    data = b''
    finally_data = b''
    while current_len < total_len:
        print(current_len, total_len)
        data = client.recv(1024)
        current_len += len(data)
        finally_data += data
    print(finally_data.decode('gbk'))

"""客户端传过来一个名字，复制这个文件到客户端上面，并且在自定义一个文件头传递过去"""

import socket
import hashlib
import struct
import json

server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
server.bind(('127.0.0.1', 8080))
server.listen(5)
while True:
    conn, addr = server.accept()
    while True:
        try:
            file_name = conn.recv(1024)  # 获得一个字节形式的文件名
            # print(file_name)
            hash_value = hashlib.md5()   # 当前文件的hash值
            file_data_bytes = b''  # 存储文件的数据
            with open(file_name.decode('utf-8'), 'rb') as f:
                for line in f:
                    file_data_bytes += line
                    hash_value.update(line)

            # 自定义文件头
            head_dict = {
                'name': file_name.decode('utf-8'),
                'md5': hash_value.hexdigest(),
                'file_data_len': len(file_data_bytes)
            }
            print(head_dict)
            # 用json转换当前文件头为字节形式
            head_bytes = json.dumps(head_dict).encode('utf-8')

            # 计算当前文件头的长度，并且转换成固定长度(i为4)的字节形式
            head_len_bytes = struct.pack('i', len(head_bytes))

            # print(head_len_bytes,)

            # 发送数据
            conn.send(head_len_bytes + head_bytes + file_data_bytes)

        except Exception as e:
            print(e)
            break
    conn.close()

import json
import socket
import struct
import os

client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
client.connect(('127.0.0.1', 8080))

temp = '请'
while True:
    file_name = input(temp + '输入你要下载的文件名称(Q退出)>>>').strip()
    file_name = 'H:\python_study\day34\自定义\客户端.py'
    if file_name.upper() == 'Q':
        break
    if not file_name:
        print('输入不能为空')
        temp = '请重新'
        continue
    client.send(file_name.encode('utf-8'))
    # print(client.recv(100))
    # 读取文件头的内容
    head_len = struct.unpack('i', client.recv(4))[0]  # 获取文件头的长度
    print(head_len)
    head_data_bytes = b''
    while head_len > 0:
        if head_len - 10 > 0:
            head_data_bytes += client.recv(10)
        else:
            head_data_bytes += client.recv(head_len)
        head_len -= 10
    print(head_data_bytes.decode('utf-8'))
    head_data_dict = json.loads(head_data_bytes.decode('utf-8'))
    print(head_data_dict)
    # 读取真实的文件数据并且进行保存
    file_data_len = head_data_dict['file_data_len']
    file_data_bytes = b''
    while file_data_len > 0:
        file_data_bytes += client.recv(10)
        file_data_len -= 10

    file_path = os.path.normpath(os.path.join(
        __file__,
        os.pardir,
        '下载',
        os.path.basename(head_data_dict['name'])
    ))
    with open(file_path, 'wb') as f:
        f.write(file_data_bytes)
client.close()

import socket

server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
server.bind(('127.0.0.1', 8080))   # 和tcp一样需要去绑定ip和端口，但是它不必像tcp一样去监听和接收连接
while True:
    data, addr = server.recvfrom(1024)  # 直接recvfrom就可以了
    server.sendto(data.upper(), addr)

import socket

client = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
while True:
    msg = input('>>>')
    client.sendto(msg.encode('utf-8'), ('127.0.0.1', 8080))  # 客户端需要发送数据的时候需要添加一个ip和端口
    data, addr = client.recvfrom(1024)
    print('服务器发送过来的数据>', data.decode('utf-8'))

import socket

server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
server.bind(('127.0.0.1', 8080))   # 和tcp一样需要去绑定ip和端口，但是它不必像tcp一样去监听和接收连接

client_set = set({})
while True:
    data, addr = server.recvfrom(1024)  # 直接recvfrom就可以了
    client_set.add(addr)
    for client in client_set:
        server.sendto(data, client)

import socket

client = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
while True:
    msg = input('>>>')
    client.sendto(msg.encode('utf-8'), ('127.0.0.1', 8080))  # 客户端需要发送数据的时候需要添加一个ip和端口
    data, addr = client.recvfrom(1024)
    print('服务器发送过来的数据>', data.decode('utf-8'))

import socket

client = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
while True:
    msg = input('>>')
    client.sendto(msg.encode('utf-8'), ('127.0.0.1', 8080))

import socket

client = socket.socket(socket.AF_INET,socket.SOCK_DGRAM)
while True:
    msg = input('>>')
    client.sendto(msg.encode('utf-8'), ('127.0.0.1', 8080))