pyxmpp2和gevent.monkey.patch_socket关于dns解析方法冲突的解决

最新推荐文章于 2024-08-22 10:08:25 发布

原创最新推荐文章于 2024-08-22 10:08:25 发布 · 497 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#socket #gevent #pyxmpp2 #xmpp #tcp

Python 同时被 3 个专栏收录

7 篇文章

订阅专栏

gevent

1 篇文章

订阅专栏

pyxmpp2

1 篇文章

订阅专栏

本文介绍了一个关于使用Gevent和PyXMPP2进行XMPP项目开发的问题。在使用多进程和Gevent协程的情况下，发现当对socket进行补丁处理(patch_socket())后，对于特定的SRV域名解析出现问题，导致消息发送失败。文中详细解释了问题原因，并提供了解决方案。

最近在做xmpp相关项目。使用gevent和pyxmpp2。
多进程+gevent协程，效果还是不错。
但是打上patch_socket()就会出现问题，如果SRV域名的A记录已存在的话，将会直接解析起A记录。

举个例子：
SRV:_xmpp-client._TCP.gmail.com SRV 20 0 0 talk.l.gmail.com
A:gmail.com A 79.18.125.19

当patch_socket()后，就会直接解析gmail.com A记录到 79.18.125.19，而不是解析talk.l.gmail.com.
gmail.com并没有5222端口提供xmpp服务，因此消息会发送失败。

阅读pyxmpp2源码和gevent文档发现：
pyxmpp2.transport._connect()中解析地址会先flag=socket.AI_NUMERICHOST 强制使用IP地址方式，失败之后才会尝试SRV解析。

而gevent.socket文档明确说明getaddrinfo会忽略flag参数：
Differs in the following ways:

raises DNSError (a subclass of gaierror) with libevent-dns error codes instead of standard socket error codes
flags argument is ignored
for IPv6, flow info and scope id are always 0
因此，patch_socket()且gmail.com存在A记录的时候，gevent.socket将会解析成功，并且返回使用。

解决方法：
import gevent
from gevent import monkey
monkey.patch_socket( dns = False, aggressive = True )

patch时加dns=False，不使用gevent.socket dns相关方法，使pyxmpp2.transport中connect函数flag=socket.AI_NUMERICHOST生效，方法包括：
__dns__ = ['getaddrinfo',
'gethostbyname',
'gethostbyname_ex',
'gethostbyaddr',
'getnameinfo',
'getfqdn']

一般来说并不会影响patch_socket性能,毕竟只是dns解析还是使用python.socket而已。

特此分享，不走弯路。