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)