pyxmpp2和gevent.monkey.patch_socket關於dns解析方法衝突的解決

最近在做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而已。

特此分享,不走彎路。
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章