def _host_cb(arg, status, timeouts, hostent):
callback = _ffi.from_handle(arg)
if status != _lib.ARES_SUCCESS:
result = None
else:
result = ares_host_result(hostent)
status = None
callback(result, status)
_global_set.discard(arg)
def _nameinfo_cb(arg, status, timeouts, node, service):
callback = _ffi.from_handle(arg)
if status != _lib.ARES_SUCCESS:
result = None
else:
result = ares_nameinfo_result(node, service)
status = None
callback(result, status)
_global_set.discard(arg)
def _query_cb(arg, status, timeouts, abuf, alen):
result = None
callback, query_type = _ffi.from_handle(arg)
if status == _lib.ARES_SUCCESS:
if query_type == _lib.T_A:
addrttls = _ffi.new("struct ares_addrttl[]", PYCARES_ADDRTTL_SIZE)
naddrttls = _ffi.new("int*", PYCARES_ADDRTTL_SIZE)
parse_status = _lib.ares_parse_a_reply(abuf, alen, _ffi.NULL, addrttls, naddrttls)
if parse_status != ARES_SUCCESS:
result = None
status = parse_status
else:
result = [ares_query_simple_result(addrttls[i]) for i in range(naddrttls[0])]
status = None
elif query_type == _lib.T_AAAA:
addrttls = _ffi.new("struct ares_addr6ttl[]", PYCARES_ADDRTTL_SIZE)
naddrttls = _ffi.new("int*", PYCARES_ADDRTTL_SIZE)
parse_status = _lib.ares_parse_aaaa_reply(abuf, alen, _ffi.NULL, addrttls, naddrttls)
if parse_status != ARES_SUCCESS:
result = None
status = parse_status
else:
result = [ares_query_simple_result(addrttls[i]) for i in range(naddrttls[0])]
status = None
elif query_type == _lib.T_CNAME:
host = _ffi.new("struct hostent **")
parse_status = _lib.ares_parse_a_reply(abuf, alen, host, _ffi.NULL, _ffi.NULL)
if parse_status != ARES_SUCCESS:
result = None
status = parse_status
else:
result = ares_query_cname_result(host[0])
_lib.ares_free_hostent(host[0])
status = None
elif query_type == _lib.T_MX:
mx_reply = _ffi.new("struct ares_mx_reply **")
parse_status = _lib.ares_parse_mx_reply(abuf, alen, mx_reply);
if parse_status != ARES_SUCCESS:
result = None
status = parse_status
else:
result = []
mx_reply_ptr = _ffi.new("struct ares_mx_reply **")
mx_reply_ptr[0] = mx_reply[0]
while True:
if mx_reply_ptr[0] == _ffi.NULL:
break
result.append(ares_query_mx_result(mx_reply[0]))
mx_reply_ptr[0] = mx_reply_ptr[0].next
_lib.ares_free_data(mx_reply)
status = None
elif query_type == _lib.T_NAPTR:
naptr_reply = _ffi.new("struct ares_naptr_reply **")
parse_status = _lib.ares_parse_naptr_reply(abuf, alen, naptr_reply);
if parse_status != ARES_SUCCESS:
result = None
status = parse_status
else:
result = []
naptr_reply_ptr = _ffi.new("struct ares_naptr_reply **")
naptr_reply_ptr[0] = naptr_reply[0]
while True:
if naptr_reply_ptr[0] == _ffi.NULL:
break
result.append(ares_query_naptr_result(naptr_reply[0]))
naptr_reply_ptr[0] = naptr_reply_ptr[0].next
_lib.ares_free_data(naptr_reply)
status = None
elif query_type == _lib.T_NS:
hostent = _ffi.new("struct hostent **")
parse_status = _lib.ares_parse_ns_reply(abuf, alen, hostent);
if parse_status != ARES_SUCCESS:
result = None
status = parse_status
else:
result = []
host = hostent[0]
for i in range(100):
if host.h_aliases[i] == _ffi.NULL:
break
result.append(ares_query_ns_result(host.h_aliases[i]))
_lib.ares_free_hostent(host)
status = None
elif query_type == _lib.T_PTR:
hostent = _ffi.new("struct hostent **")
parse_status = _lib.ares_parse_ptr_reply(abuf, alen, _ffi.NULL, 0, socket.AF_UNSPEC, hostent);
if parse_status != ARES_SUCCESS:
result = None
status = parse_status
else:
result = ares_query_ptr_result(hostent[0])
_lib.ares_free_hostent(hostent[0])
status = None
elif query_type == _lib.T_SOA:
soa_reply = _ffi.new("struct ares_soa_reply **")
parse_status = _lib.ares_parse_soa_reply(abuf, alen, soa_reply);
if parse_status != ARES_SUCCESS:
result = None
status = parse_status
else:
result = ares_query_soa_result(soa_reply[0])
_lib.ares_free_data(soa_reply[0])
status = None
elif query_type == _lib.T_SRV:
srv_reply = _ffi.new("struct ares_srv_reply **")
parse_status = _lib.ares_parse_srv_reply(abuf, alen, srv_reply);
if parse_status != ARES_SUCCESS:
result = None
status = parse_status
else:
result = []
srv_reply_ptr = _ffi.new("struct ares_srv_reply **")
srv_reply_ptr[0] = srv_reply[0]
while True:
if srv_reply_ptr[0] == _ffi.NULL:
break
result.append(ares_query_srv_result(srv_reply_ptr[0]))
srv_reply_ptr[0] = srv_reply_ptr[0].next
_lib.ares_free_data(srv_reply[0])
status = None
elif query_type == _lib.T_TXT:
txt_reply = _ffi.new("struct ares_txt_ext **")
parse_status = _lib.ares_parse_txt_reply_ext(abuf, alen, txt_reply);
if parse_status != ARES_SUCCESS:
result = None
status = parse_status
else:
result = []
txt_reply_ptr = _ffi.new("struct ares_txt_ext **")
txt_reply_ptr[0] = txt_reply[0]
tmp_obj = None
while True:
if txt_reply_ptr[0] == _ffi.NULL:
if tmp_obj is not None:
result.append(tmp_obj)
break
if txt_reply_ptr[0].record_start == 1:
if tmp_obj is not None:
result.append(tmp_obj)
tmp_obj = ares_query_txt_result(txt_reply_ptr[0])
else:
new_chunk = ares_query_txt_result(txt_reply_ptr[0])
tmp_obj.text += new_chunk.text
txt_reply_ptr[0] = txt_reply_ptr[0].next
_lib.ares_free_data(txt_reply[0])
status = None
else:
raise ValueError("invalid query type specified")
callback(result, status)
_global_set.discard(arg)
def _sock_state_cb(data, socket_fd, readable, writable):
sock_state_cb = _ffi.from_handle(data)
sock_state_cb(socket_fd, readable, writable)
_global_set.discard(data)
