UDP编程

模拟qq聊天

# -*- coding:utf-8 -*-
import socket
from threading import Thread


def read_msg(socket, msg):
    print(msg)
    while True:
        receive_msg, IP = socket.recvfrom(1024)
        print("from %s:%s" % (IP, receive_msg.decode("gbk")))


def send_msg(socket, ip):
    while True:
        msg = input(">>")
        socket.sendto(msg.encode("gbk"), ip)


def main():
    # 创建套接字对象
    service_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # 绑定地址
    address = ('192.168.133.70', 2000)
    service_socket.bind(address)
    # 接收消息，获取客户端IP
    receive_msg, receive_IP = service_socket.recvfrom(1024)
    # 创建线程，读取消息
    read_t1 = Thread(target=read_msg, args=(service_socket, receive_msg))
    read_t1.start()
    # 创建线程，发送消息
    send_t2 = Thread(target=send_msg, args=(service_socket, receive_IP))
    send_t2.start()

if __name__ == "__main__":
    main()

广播

# -*- coding:utf-8 -*-
import socket


def main():
    # 创建套接字
    broadcast_socket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
    # 设置允许广播
    broadcast_socket.setsockopt(socket.SOL_SOCKET, socket.SO_BROADCAST, 1)
    # 设置地址
    address = ("192.168.133.110", 2425)
    # 写入要广播的内容
    msg = input(">>")
    # 广播出去
    broadcast = ("1:99:zl:hrz:32:%s"% msg).encode("gbk")   # 把消息整个转换成gbk
    # 广播给飞秋
    broadcast_socket.sendto(broadcast, address)
    # 固定IP广播
    # broadcast_socket.sendto(msg.encode("gbk"), address)
    # 接收消息
    msg, ip = broadcast_socket.recvfrom(1024)
    print("from %s: %s" % (ip, msg.decode("gbk")))

if __name__ == "__main__":
    main()