Net

参考：Net

TCP_web

import socket
#  创建一个socket
# AF_INET指定使用IPv4协议，如果要用更先进的IPv6，就指定为AF_INET6。SOCK_STREAM指定使用面向流的TCP协议，
with socket.socket(socket.AF_INET,socket.SOCK_STREAM) as s:
    # 注意参数是一个tuple，包含地址和端口号。
    s.connect(('www.sina.com',80))
    # 发送数据:
    s.send(b'GET / HTTP/1.1
Host: www.sina.com.cn
Connection: close

')
    buffer=[]
    while True:
        #一次最多接收指定的字节数
        temp=s.recv(1024)
        if temp:
            buffer.append(temp)
        # recv()返回空数据，表示接收完毕
        else:
            break
buffer=b''.join(buffer)
header, html = buffer.split(b'

', 1)
print(header.decode('utf-8'))
with open('sina.html','w') as f:
    f.write(html.decode('utf-8'))

TCP_client

import socket
import os
import struct
import json
# AF_INET指定使用IPv4协议，如果要用更先进的IPv6，就指定为AF_INET6。
# SOCK_STREAM指定使用面向流的TCP协议，收发都要使用二进制数据，注意转换。
with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as s:
    # 建立连接:参数是一个tuple，包含地址和端口号。
    s.connect(('127.0.0.1', 9999))
    # 接收消息,1024 :一次最多接收指定的字节数
    print(s.recv(1024).decode('utf-8'))

    file_path='sina.html'
    # 得到文件的大小,字节
    filesize_bytes = os.path.getsize(file_path)
    filename = 'recv_'+file_path
    dirc = {
        'filename': filename,
        'filesize_bytes': filesize_bytes,
    }
    # 将字典转换成字符串
    head_info = json.dumps(dirc)
    # 将字符串的长度打包为2进制
    head_info_len = struct.pack('i', len(head_info))
    # 发送头文件
    s.send(head_info_len)
    s.send(head_info.encode('utf-8'))
    #发送文件
    with open(file_path,'rb') as f:
        s.sendall(f.read())
    # 反馈
    msg=s.recv(1024).decode('utf-8')
    print(msg)

TCP_server

import socket
import threading
import struct
import json
import sys

# 进度条
def process_bar(precent, width=50):
    use_num = int(precent*width)
    space_num = int(width-use_num)
    precent = precent*100
    print('[%s%s]%d%%'%(use_num*'#', space_num*' ',precent),file=sys.stdout,flush=True, end='
')

def tcplink(client,addr):
    client.send(b'welcome',)
    # 接受头文件数据
    head_length=struct.unpack('i', client.recv(4))[0]
    head=client.recv(head_length)
    head_dir = json.loads(head.decode('utf-8'))
    filesize = head_dir['filesize_bytes']
    filename = head_dir['filename']
    # 接受文件
    recv_size=0
    with open(filename,'w') as f:
        while True:
            temp=filesize-recv_size
            #recv() 方法在未收到消息下会阻塞当前线程
            process_bar(recv_size/filesize)
            if temp>1024:
                data=client.recv(1024)
                recv_size+=1024
                # 收发都是二进制数据流，要注意转换
                f.write(data.decode('utf-8'))
            else:
                data=client.recv(temp)
                # 收发都是二进制数据流，要注意转换
                f.write(data.decode('utf-8'))
                break
    client.send('trans finished'.encode('utf-8'))
    # 关闭连接
    client.close()
    print(addr,' closed')

# 一个Socket依赖4项：服务器地址、服务器端口、客户端地址、客户端端口来唯一确定一个Socket。
with socket.socket(socket.AF_INET,socket.SOCK_STREAM) as s:
    # 监听端口
    # # 要绑定监听的地址和端口。服务器可能有多块网卡，可以绑定到某一块网卡的IP地址上，也可以用0.0.0.0绑定到所有的网络地址，还可以用127.0.0.1绑定到本机地址。
    s.bind(('127.0.0.1', 9999))
    # 调用listen()方法开始监听端口，
    # 传入的参数指定等待连接的最大数量,实现链接多个对象
    s.listen(5)
    print('Waiting for connection...')
    while True:
        # 接受链接,accept()会等待并返回一个客户端的连接
        client,addr=s.accept()
        # 创建新线程来处理TCP连接:
        t=threading.Thread(target=tcplink,args=(client,addr))
        t.start()

UDP_client

import socket
#SOCK_DGRAM指定了这个Socket的类型是UDP。
with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
    for data in [b'Michael', b'Tracy', b'Sarah']:
        # 不需要调用connect()，直接通过sendto()给服务器发数据：
        # 发送数据,无连接，发送时指明收信方
        s.sendto(data, ('127.0.0.1', 9999))
        # 接收数据:
        print(s.recv(1024).decode('utf-8'))

UDP_server

import socket
import threading 

# 使用UDP协议时，不需要建立连接，只需要知道对方的IP地址和端口号，就可以直接发数据包。
#SOCK_DGRAM指定了这个Socket的类型是UDP。
with socket.socket(socket.AF_INET, socket.SOCK_DGRAM) as s:
    # 绑定端口,服务器绑定UDP端口和TCP端口互不冲突.
    s.bind(('127.0.0.1', 9999))
    while True:
        # 需要调用listen()方法，而是直接接收来自任何客户端的数据：
        # recvfrom()方法返回数据和客户端的地址与端口
        data,addr=s.recvfrom(1024)
        print('Received from %s:%s.' % addr)
        s.sendto(b'Hello, %s!' % data, addr)