Socket 通信之 UDP 通信

前段时间,我们在这篇文章中谈到了多进程和进程之间的通信方式,主要谈到了本地进程之间使用队列(Queue)进程通信,如果我们要通信的进程不在同一台主机上,我们就无法使用队列进行通信了,这时就需要使用 Socket(套接字)。
Socket 是应用层和传输层之间的一层抽象协议,可以用来进行进程间通信,一般有 UDP 和 TCP 两种通信方式,前者速度稍快,稳定性不好,无法丢包重传。后者速度稍慢一点,但稳定性很好,可以丢包重传。
本文首先介绍使用 Socket 进行 UDP 通信。

UDP 通信流程

使用 Socket 进行 UDP 通信的流程如下:

  • 创建一个 socket 对象
  • 准备好目标主机 IP 和目标端口
  • 绑定本机的端口(不必须)
  • 发送/接收数据
  • 关闭套接字

下面依次进行讲解。

创建套接字对象

要进行 Socket 通信,我们需要使用 socket 模块,首先需要创建一个 Socket 对象。下面是两种创建方式:

# 使用 UDP 进行通信
udpSocket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)

# 使用 TCP 进行通信
tcpSocket = socket.socket(socket.AF_INET, socket.SOCK_STREAM)

目标主机的 IP 和端口

如果我们需要向别的主机发送数据,我们需要改主机的 IP 地址和相应的端口号。在使用 Socket 进行通信时,需要将两个信息写在一元组中,元组的第一项为目标主机 IP 地址,第二项为接受数据的端口号:

targetAddr = ("192.168.0.1", 8080)

其中,IP 地址使用字符串类型,端口号使用数字类型。

绑定本机的端口

如果不绑定端口,每次使用 Socket 时都会由操作系统动态分配一个端口,我们也可以绑定为某个固定的端口。这样做的好处是:如果我们想要接受其他主机的信息,其他主机可以直接向这个端口发送数据,如果使用动态端口的话,发送方并不知道目标端口是什么,因此无法向接收方发送数据。
绑定端口需要使用 Socket 对象的 bind 方法:

selfBind = ( [IP], PORT )
socketObj.bind( selfBind )

bind 方法接受一个元组作为参数,元组的第一项为绑定的 IP 地址,第二项为绑定的端口号。我们可以把第一项指定为本机上的任意一个 IP 地址,也可以设置为一个空字符串 "",表示本机上任意合法的 IP 地址。

发送/接受数据

使用 UDP 套接字协议时,发送数据使用 Socket 对象的 sendto 方法,接受数据使用 Socket 对象的 recvfrom 方法。这两个方法的使用方式如下:

# 发送数据
socketObj.sendto( data, targetAddr )

# 接受数据
socketObj.recvfrom( maxSize )

sendto 方法接受两个参数:发送的数据和目标主机的 IP 和端口元组,在 Python3 中,发送的数据应该转为 byte 类型发送,Python2 中可以直接发送字符串。
recvfrom 接受一个参数:本次接受的最大数据尺寸。该方法是阻塞的,只有在接收到数据后才能进行后续的操作。

关闭套接字

就像使用文件那样,在使用完套接字后,需要关闭它,调用 close 方法即可。

socketObj.close()

单工通信的例子

上面我们介绍了 Socket 的使用方式,下面我们来做一个单工通信的例子(一方负责发送信息,一方负责接收信息)。
我们这里来创建两个文件:用以发送信息的 send.py 和用以接收信息的 recv.py。该实例在虚拟机中模拟(注意将虚拟机设置为桥接模式)。
创建 send.py:

# 导包
from socket import *

# 创建 socket 对象
udpSocket = socket(AF_INET, SOCK_DGRAM)
# 目标主机的 IP 和端口号
targetAddr = ("192.168.2.142",8080)
# 本地主机绑定端口
udpSocket.bind(("",3000))

# 主函数
def main():
    while True:
        # 接受发送的信息
        msg = input("Enter message:")
        # 发送信息 使用 utf-8 编码
        udpSocket.sendto(msg.encode("utf-8"),targetAddr)

if __name__ == '__main__':
    main()

创建 recv.py:

# 导包
from socket import *

# 创建 socket 对象
udpSocket = socket(AF_INET, SOCK_DGRAM)

# 本地主机绑定端口
udpSocket.bind(("",8080))

# 主函数
def main():
    while True:
        # 接收信息
        msg,addrInfo = udpSocket.recvfrom(1024)
        # 收到信息后使用 utf-8 解码
        print("收到来自 %s 的信息:%s"%(addrInfo[0],msg.decode("utf-8")))

if __name__ == '__main__':
    main()

运行结果如下:

socket单工通信.gif

双工通信例子

上面实现了一个单工通信的例子:一方负责发,一方负责接收。下面我们继续实现一个双工通信的例子,使双方都能够收发消息。
由于接收和发送消息时是使用 while 循环不断轮询的,因此要实现同时发送和接受,我们需要进行多任务处理。
新建一个 msg.py:

from threading import Thread
from socket import *

# 发送数据
class SendMsg(Thread):
    def __init__(self):
        Thread.__init__(self)
        self.__selfPort = 3000
        self.__targetPort = 3001
        self.__socket = socket(AF_INET, SOCK_DGRAM)
        self.getInfo()

    # 获取目标主机的 IP
    def getInfo(self):
        # 获取目标主机的 IP
        self.__targetIp = input("请输入目标主机IP地址:")
        # 绑定本机端口
        self.__socket.bind(("",self.__selfPort))
        self.__targetAddr = (self.__targetIp, self.__targetPort)

    # 发送数据
    def run(self):
        while True:
            msg = input("\rEnter Message:")
            self.__socket.sendto(msg.encode("utf-8"),self.__targetAddr)

# 接受数据
class RecvMsg(Thread):
    def __init__(self):
        Thread.__init__(self)
        self.__selfPort = 3001
        self.__socket = socket(AF_INET, SOCK_DGRAM)
        self.__socket.bind(("",self.__selfPort))

    # 发送数据
    def run(self):
        while True:
            msg,addrInfo  = self.__socket.recvfrom(1024)
            print("\r%s:%s"%(addrInfo[0],msg.decode("utf-8")))
            print("\rEnter Message:")

if __name__ == '__main__':
    send = SendMsg()
    recv = RecvMsg()
    send.start()
    recv.start()

这里我们使用 3000 端口发送数据,3001 端口接收数据,运行程序时只需填写目标主机的 IP 地址,就可以进行通信。
运行效果:

socket双工通信.gif

广播

我们还可以进行局域网内的广播,只需对 Socket 加上一条设置:

socketObj.setsockopt(SOL_SOCKET, SO_BROADCAST, 1)

同时,发送广播需要一个广播地址,以及目标主机接受广播的端口:

broadAddr = ( "192.168.0.255", 8080 )

上面的设置只能给 0 网段的主机发送广播,要想给局域网中所有的主机发送广播,可以这样设置:

broadAddr = ( "<broadcast>", 8080 )

下面我们新建一个 send.py 用来发送广播:

from threading import Thread
from socket import *
from time import sleep

# 发送数据
class SendMsg(Thread):
    def __init__(self):
        Thread.__init__(self)
        self.__dest = ("<broadcast>",3001)
        self.__socket = socket(AF_INET, SOCK_DGRAM)
        self.__socket.setsockopt(SOL_SOCKET,SO_BROADCAST,1)

    def sendInfo(self):
        self.__socket.sendto("hello~".encode("utf-8"),self.__dest)

    # 发送数据
    def run(self):
        while True:
            self.sendInfo()
            sleep(1)

if __name__ == '__main__':
    send = SendMsg()
    send.start()

新建一个 recv.py 用来接收广播:

from threading import Thread
from socket import *

# 接受数据
class RecvMsg(Thread):
    def __init__(self):
        Thread.__init__(self)
        self.__selfPort = 3001
        self.__socket = socket(AF_INET, SOCK_DGRAM)
        self.__socket.bind(("",self.__selfPort))

    # 发送数据
    def run(self):
        while True:
            msg,addrInfo  = self.__socket.recvfrom(1024)
            print("\r%s:%s"%(addrInfo[0],msg.decode("utf-8")))
            
if __name__ == '__main__':
    recv = RecvMsg()
    recv.start()

运行效果如图:

udp广播.gif

完。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,277评论 6 503
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,689评论 3 393
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,624评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,356评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,402评论 6 392
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,292评论 1 301
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,135评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,992评论 0 275
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,429评论 1 314
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,636评论 3 334
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,785评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,492评论 5 345
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,092评论 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,723评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,858评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,891评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,713评论 2 354

推荐阅读更多精彩内容