Python3 Socket编程案例：UDP与TCP

1. 按

在Python中socket.SOCK_DGRAM用于UDP通信，socket.SOCK_STREAM用于TCP通信

2. UDP案例源码

客户端可以向服务器端主动发送信息

2.1. 客户端

import socket

BUF_SIZE = 1024  # 设置缓冲区
server_addr = ('127.0.0.1', 8888)  # IP和端口构成表示地址
client = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)  # 生成新的套接字对象

while True:
    data = input('Please Input data > ')
    client.sendto(data.encode(), server_addr)  # 向服务器发送数据
    data, addr = client.recvfrom(BUF_SIZE)  # 从服务器接收数据
    print("Data : ", data)
client.close()

2.2. 服务器端

import socket

BUF_SIZE = 1024  # 设置缓冲区大小
server_addr = ('127.0.0.1', 8888)  # IP和端口构成表示地址
server = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)  # 生成新的套接字对象
server.bind(server_addr)  # 套接字绑定IP和端口
while True:
    print("waitting for data")
    data, client_addr = server.recvfrom(BUF_SIZE)  # 从客户端接收数据
    print('Connected by', client_addr, ' Receive Data : ', data)
    server.sendto(data, client_addr)  # 发送数据给客户端

server.close()

2. TCP案例源码

客户端可以向服务器端主动发送信息，服务器端也可以主动向客户端发送信息

2.1. 客户端

import socket
import threading


def recvlink(client):  # 接受服务器返回的数据的函数
    while True:
        msg = client.recv(1024)
        print('Ubuntu say: ' + msg.decode('utf-8'))


def main():
    # 创建ipv4的socket对象，使用TCP协议(SOCK_STREAM)
    client = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 设置服务器ip地址，注意应该是服务器的公网ip
    host = '127.0.0.1'
    # 设置要发送到的服务器端口,需要在云服务器管理界面打开对应端口的防火墙
    port = 1234

    # 建立TCP协议连接,这时候服务器就会监听到到连接请求，并开始等待接受client发送的数据
    client.connect((host, port))

    # 建立连接后，服务器端会返回连接成功消息
    start_msg = client.recv(1024)
    print(start_msg.decode('utf-8'))

    # 开启一个线程用来接受服务器发来的消息
    t = threading.Thread(target=recvlink, args=(client,))
    t.start()

    # 发送消息
    while True:
        # 输入要发送的信息
        sendmsg = input()
        # 向服务器发送消息
        client.send(sendmsg.encode('utf-8'))
        if sendmsg == 'quit':
            break
    # 结束时关闭客户端
    client.close()


if __name__ == '__main__':
    main()

2.2. 服务器端

import socket
import threading


def recv_msg(clientsocket):  # 接受客户端消息函数
    while True:
        # 接受客户端消息,设置一次最多接受1024字节的数据
        recv_msg = clientsocket.recv(1024)
        # 把接收到的东西解码
        msg = recv_msg.decode('utf-8')
        # 如果用户输入了quit，就退出此对话
        if msg == 'quit':
            exit(0)
        print('Win10 say: ' + msg)


def main():
    # 创建服务器端socket对象 ipv4 + TCP协议，和客户端一样
    socket_server = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

    # 注意注意注意，我们要绑定监听的地址和端口。服务器可能有多块网卡，可以绑定到某一块网卡的IP地址上，也可以用0.0.0.0绑定到所有的网络地址
    # 还可以用127.0.0.1绑定到本机地址。127.0.0.1是一个特殊的IP地址，表示本机地址，如果绑定到这个地址，客户端必须同时在本机运行才能连接，也就是说，外部的计算机无法连接进来。
    # 这个程序中host使用'0.0.0.0'或服务器内网ip地址都可以，我这里就使用了内网ip地址
    # host='0.0.0.0'
    host = '127.0.0.1'
    # 设置被监听的端口号,小于1024的端口号不能使用，因为他们是Internet标准服务的端口号
    port = 1234

    # 绑定地址
    socket_server.bind((host, port))
    # 设置最大监听数，也就是最多可以同时响应几个客户端请求，一般配合多线程使用
    socket_server.listen(5)
    # 等待客户端连接,一旦有了连接就立刻向下执行，否则等待
    # accept()函数会返回一个元组，第一个元素是客户端socket对象，第二个元素是客户端地址（ip地址+端口号）
    clientsocket, addr = socket_server.accept()

    # 有了客户端连接后之后才能执行以下代码，我们先向客户端发送连接成功消息
    clientsocket.send('你现在已经连接上了服务器啦，我们来聊天吧！'.encode('utf-8'))

    # 和客户端一样开启一个线程接受客户端的信息
    t = threading.Thread(target=recv_msg, args=(clientsocket,))
    t.start()

    # 发送消息
    while True:
        reply = input()
        clientsocket.send(reply.encode('utf-8'))

    clientsocket.close()


if __name__ == '__main__':
    main()