def get_dns_query_for_esp(esp_host): dns = dpkt.dns.DNS( b'\x00\x00\x01\x00\x00\x01\x00\x00\x00\x00\x00\x00\x00\x00\x01\x00\x01' ) dns.qd[0].name = esp_host + u'.local' console_log('Created query for esp host: {} '.format(dns.__repr__())) return dns.pack()
def get_dns_query_for_esp(esp_host): # type:(str) -> dpkt.dns.Msg dns = dpkt.dns.DNS() arr = dpkt.dns.DNS.RR() arr.cls = dpkt.dns.DNS_IN arr.name = esp_host + u'.local' dns.qd.append(arr) console_log('Created query for esp host: {} '.format(dns.__repr__())) return dns.pack()
def get_dns_answer_to_mdns(tester_host): # type:(str) -> dpkt.dns.Msg dns = dpkt.dns.DNS() dns.op = dpkt.dns.DNS_QR | dpkt.dns.DNS_AA dns.rcode = dpkt.dns.DNS_RCODE_NOERR arr = dpkt.dns.DNS.RR() arr.cls = dpkt.dns.DNS_IN arr.type = dpkt.dns.DNS_A arr.name = tester_host arr.ip = socket.inet_aton('127.0.0.1') dns.an.append(arr) console_log('Created answer to mdns query: {} '.format(dns.__repr__())) return dns.pack()
def get_mdns_service_query(service): # type:(str) -> dpkt.dns.Msg dns = dpkt.dns.DNS() dns.op = dpkt.dns.DNS_QR | dpkt.dns.DNS_AA dns.rcode = dpkt.dns.DNS_RCODE_NOERR arr = dpkt.dns.DNS.RR() arr.cls = dpkt.dns.DNS_IN arr.type = dpkt.dns.DNS_SRV arr.name = service arr.target = socket.inet_aton('127.0.0.1') arr.srvname = service dns.qd.append(arr) console_log('Created mdns service query: {} '.format(dns.__repr__())) return dns.pack()
def mdns_server(esp_host): global esp_answered UDP_IP = '0.0.0.0' UDP_PORT = 5353 MCAST_GRP = '224.0.0.251' TESTER_NAME = u'tinytester.local' TESTER_NAME_LWIP = u'tinytester-lwip.local' QUERY_TIMEOUT = 0.2 sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1) sock.setblocking(False) sock.bind((UDP_IP,UDP_PORT)) mreq = struct.pack('4sl', socket.inet_aton(MCAST_GRP), socket.INADDR_ANY) sock.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mreq) last_query_timepoint = time.time() while not stop_mdns_server.is_set(): try: current_time = time.time() if current_time - last_query_timepoint > QUERY_TIMEOUT: last_query_timepoint = current_time if not esp_answered.is_set(): sock.sendto(get_dns_query_for_esp(esp_host), (MCAST_GRP,UDP_PORT)) if not esp_delegated_answered.is_set(): sock.sendto(get_dns_query_for_esp(esp_host + '-delegated'), (MCAST_GRP,UDP_PORT)) timeout = max(0, QUERY_TIMEOUT - (current_time - last_query_timepoint)) read_socks, _, _ = select.select([sock], [], [], timeout) if not read_socks: continue data, addr = sock.recvfrom(1024) dns = dpkt.dns.DNS(data) if len(dns.qd) > 0 and dns.qd[0].type == dpkt.dns.DNS_A: if dns.qd[0].name == TESTER_NAME: console_log('Received query: {} '.format(dns.__repr__())) sock.sendto(get_dns_answer_to_mdns(TESTER_NAME), (MCAST_GRP,UDP_PORT)) elif dns.qd[0].name == TESTER_NAME_LWIP: console_log('Received query: {} '.format(dns.__repr__())) sock.sendto(get_dns_answer_to_mdns_lwip(TESTER_NAME_LWIP, dns.id), addr) if len(dns.an) > 0 and dns.an[0].type == dpkt.dns.DNS_A: console_log('Received answer from {}'.format(dns.an[0].name)) if dns.an[0].name == esp_host + u'.local': console_log('Received answer to esp32-mdns query: {}'.format(dns.__repr__())) esp_answered.set() if dns.an[0].name == esp_host + u'-delegated.local': console_log('Received answer to esp32-mdns-delegate query: {}'.format(dns.__repr__())) esp_delegated_answered.set() except socket.timeout: break except dpkt.UnpackError: continue
def get_dns_answer_to_service_query( mdns_service): # type:(str) -> dpkt.dns.Msg dns = dpkt.dns.DNS() dns.op = dpkt.dns.DNS_QR | dpkt.dns.DNS_AA dns.rcode = dpkt.dns.DNS_RCODE_NOERR arr = dpkt.dns.DNS.RR() arr.name = mdns_service arr.cls = dpkt.dns.DNS_IN arr.type = dpkt.dns.DNS_SRV arr.priority = 0 arr.weight = 0 arr.port = 100 arr.srvname = mdns_service arr.ip = socket.inet_aton('127.0.0.1') dns.an.append(arr) console_log('Created answer to mdns query: {} '.format(dns.__repr__())) return dns.pack()
def test_examples_protocol_mdns(env, extra_data): global stop_mdns_server """ steps: | 1. obtain IP address + init mdns example 2. get the dut host name (and IP address) 3. check the mdns name is accessible 4. check DUT output if mdns advertized host is resolved """ dut1 = env.get_dut('mdns-test', 'examples/protocols/mdns', dut_class=ttfw_idf.ESP32DUT, app_config_name='eth_kit') # check and log bin size binary_file = os.path.join(dut1.app.binary_path, 'mdns_test.bin') bin_size = os.path.getsize(binary_file) ttfw_idf.log_performance('mdns-test_bin_size', '{}KB'.format(bin_size // 1024)) # 1. start mdns application dut1.start_app() # 2. get the dut host name (and IP address) specific_host = dut1.expect(re.compile(r'mdns hostname set to: \[([^\]]+)\]'), timeout=30)[0] mdns_responder = Thread(target=mdns_server, args=(str(specific_host),)) try: ip_address = dut1.expect(re.compile(r' eth ip: ([^,]+),'), timeout=30)[0] console_log('Connected to AP with IP: {}'.format(ip_address)) except DUT.ExpectTimeout: raise ValueError('ENV_TEST_FAILURE: Cannot connect to AP') try: # 3. check the mdns name is accessible mdns_responder.start() if not esp_answered.wait(timeout=30): raise ValueError('Test has failed: did not receive mdns answer within timeout') if not esp_delegated_answered.wait(timeout=30): raise ValueError('Test has failed: did not receive mdns answer for delegated host within timeout') # 4. check DUT output if mdns advertized host is resolved dut1.expect(re.compile(r'mdns-test: Query A: tinytester.local resolved to: 127.0.0.1'), timeout=30) dut1.expect(re.compile(r'mdns-test: gethostbyname: tinytester-lwip.local resolved to: 127.0.0.1'), timeout=30) dut1.expect(re.compile(r'mdns-test: getaddrinfo: tinytester-lwip.local resolved to: 127.0.0.1'), timeout=30) # 5. check the DUT answers to `dig` command dig_output = subprocess.check_output(['dig', '+short', '-p', '5353', '@224.0.0.251', '{}.local'.format(specific_host)]) console_log('Resolving {} using "dig" succeeded with:\n{}'.format(specific_host, dig_output)) if not ip_address.encode('utf-8') in dig_output: raise ValueError('Test has failed: Incorrectly resolved DUT hostname using dig' "Output should've contained DUT's IP address:{}".format(ip_address)) finally: stop_mdns_server.set() mdns_responder.join()
def get_test_result(self): _res = True console_log('Test Results:') for tc in JunitReport.JUNIT_TEST_SUITE.test_cases: if tc.failures: if tc.name in self.known_failure_cases: console_log(' Known Failure: ' + tc.name, color='orange') else: console_log(' Test Fail: ' + tc.name, color='red') _res = False else: console_log(' Test Succeed: ' + tc.name, color='green') return _res
def get_test_result(self): _res = True console_log("Test Results:") for tc in JunitReport.JUNIT_TEST_SUITE.test_cases: if tc.failures: if self.is_known_issue(tc.name, self.known_failure_cases): console_log(" Known Failure: " + tc.name, color="orange") else: console_log(" Test Fail: " + tc.name, color="red") _res = False else: console_log(" Test Succeed: " + tc.name, color="green") return _res
def mdns_listener(esp_host): # type:(str) -> None print('mdns_listener thread started') UDP_IP = '0.0.0.0' sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM) sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEADDR, 1) sock.setsockopt(socket.SOL_SOCKET, socket.SO_REUSEPORT, 1) sock.setblocking(False) sock.bind((UDP_IP, UDP_PORT)) mreq = struct.pack('4sl', socket.inet_aton(MCAST_GRP), socket.INADDR_ANY) sock.setsockopt(socket.IPPROTO_IP, socket.IP_ADD_MEMBERSHIP, mreq) last_query_timepoint = time.time() QUERY_TIMEOUT = 0.2 while not stop_mdns_listener.is_set(): try: start_mdns_listener.set() current_time = time.time() if current_time - last_query_timepoint > QUERY_TIMEOUT: last_query_timepoint = current_time timeout = max( 0, QUERY_TIMEOUT - (current_time - last_query_timepoint)) read_socks, _, _ = select.select([sock], [], [], timeout) if not read_socks: continue data, _ = sock.recvfrom(1024) dns = dpkt.dns.DNS(data) if len(dns.qd) > 0: if dns.qd[0].name == HOST_NAME: console_log('Received query: {} '.format(dns.__repr__())) sock.sendto(get_dns_answer_to_mdns(HOST_NAME), (MCAST_GRP, UDP_PORT)) if dns.qd[0].name == HOST_NAME: console_log('Received query: {} '.format(dns.__repr__())) sock.sendto(get_dns_answer_to_mdns(HOST_NAME), (MCAST_GRP, UDP_PORT)) if dns.qd[0].name == MDNS_HOST_SERVICE: print(dns.qd[0].name) console_log('Received query: {} '.format(dns.__repr__())) sock.sendto( get_dns_answer_to_service_query(MDNS_HOST_SERVICE), (MCAST_GRP, UDP_PORT)) if len(dns.an) == 1: if dns.an[0].name == SERVICE_NAME: console_log('Received answer to service query: {}'.format( dns.__repr__())) esp_service_answered.set() if len(dns.an) > 1: if dns.an[1].name == SUB_SERVICE_NAME: console_log( 'Received answer for sub service query: {}'.format( dns.__repr__())) esp_sub_service_answered.set() if len(dns.an) > 0 and dns.an[0].type == dpkt.dns.DNS_A: if dns.an[0].name == esp_host + u'.local': console_log( 'Received answer to esp32-mdns query: {}'.format( dns.__repr__())) esp_host_answered.set() if dns.an[0].name == esp_host + u'-delegated.local': console_log( 'Received answer to esp32-mdns-delegate query: {}'. format(dns.__repr__())) esp_delegated_host_answered.set() except socket.timeout: break except dpkt.UnpackError: continue