test_p2p_service.py 7.1 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143
  1. #!/usr/bin/python
  2. #
  3. # P2P service discovery test cases
  4. # Copyright (c) 2013, Jouni Malinen <j@w1.fi>
  5. #
  6. # This software may be distributed under the terms of the BSD license.
  7. # See README for more details.
  8. import logging
  9. logger = logging.getLogger()
  10. import uuid
  11. import hwsim_utils
  12. def add_bonjour_services(dev):
  13. dev.request("P2P_SERVICE_ADD bonjour 0b5f6166706f766572746370c00c000c01 074578616d706c65c027")
  14. dev.request("P2P_SERVICE_ADD bonjour 076578616d706c650b5f6166706f766572746370c00c001001 00")
  15. dev.request("P2P_SERVICE_ADD bonjour 045f697070c00c000c01 094d795072696e746572c027")
  16. dev.request("P2P_SERVICE_ADD bonjour 096d797072696e746572045f697070c00c001001 09747874766572733d311a70646c3d6170706c69636174696f6e2f706f7374736372797074")
  17. def add_upnp_services(dev):
  18. dev.request("P2P_SERVICE_ADD upnp 10 uuid:6859dede-8574-59ab-9332-123456789012::upnp:rootdevice")
  19. dev.request("P2P_SERVICE_ADD upnp 10 uuid:5566d33e-9774-09ab-4822-333456785632::upnp:rootdevice")
  20. dev.request("P2P_SERVICE_ADD upnp 10 uuid:1122de4e-8574-59ab-9322-333456789044::urn:schemas-upnp-org:service:ContentDirectory:2")
  21. dev.request("P2P_SERVICE_ADD upnp 10 uuid:5566d33e-9774-09ab-4822-333456785632::urn:schemas-upnp-org:service:ContentDirectory:2")
  22. dev.request("P2P_SERVICE_ADD upnp 10 uuid:6859dede-8574-59ab-9332-123456789012::urn:schemas-upnp-org:device:InternetGatewayDevice:1")
  23. def add_extra_services(dev):
  24. for i in range(0, 100):
  25. dev.request("P2P_SERVICE_ADD upnp 10 uuid:" + str(uuid.uuid4()) + "::upnp:rootdevice")
  26. def run_sd(dev, dst, query, exp_query=None, fragment=False):
  27. addr0 = dev[0].p2p_dev_addr()
  28. addr1 = dev[1].p2p_dev_addr()
  29. add_bonjour_services(dev[0])
  30. add_upnp_services(dev[0])
  31. if fragment:
  32. add_extra_services(dev[0])
  33. dev[0].p2p_listen()
  34. dev[1].request("P2P_FLUSH")
  35. dev[1].request("P2P_SERV_DISC_REQ " + dst + " " + query)
  36. if not dev[1].discover_peer(addr0, social=True, force_find=True):
  37. raise Exception("Peer " + addr0 + " not found")
  38. ev = dev[0].wait_event(["P2P-SERV-DISC-REQ"], timeout=10)
  39. if ev is None:
  40. raise Exception("Service discovery timed out")
  41. if addr1 not in ev:
  42. raise Exception("Unexpected service discovery request source")
  43. if exp_query is None:
  44. exp_query = query
  45. if exp_query not in ev:
  46. raise Exception("Unexpected service discovery request contents")
  47. ev = dev[1].wait_event(["P2P-SERV-DISC-RESP"], timeout=10)
  48. if ev is None:
  49. raise Exception("Service discovery timed out")
  50. if addr0 not in ev:
  51. raise Exception("Unexpected service discovery response source")
  52. return ev
  53. def test_p2p_service_discovery(dev):
  54. """P2P service discovery"""
  55. for dst in [ "00:00:00:00:00:00", dev[0].p2p_dev_addr() ]:
  56. ev = run_sd(dev, dst, "02000001")
  57. if "0b5f6166706f766572746370c00c000c01" not in ev:
  58. raise Exception("Unexpected service discovery response contents (Bonjour)")
  59. if "496e7465726e6574" not in ev:
  60. raise Exception("Unexpected service discovery response contents (UPnP)")
  61. def test_p2p_service_discovery_fragmentation(dev):
  62. """P2P service discovery with fragmentation"""
  63. for dst in [ "00:00:00:00:00:00", dev[0].p2p_dev_addr() ]:
  64. ev = run_sd(dev, dst, "02000001", fragment=True)
  65. if not "long response" in ev:
  66. if "0b5f6166706f766572746370c00c000c01" not in ev:
  67. raise Exception("Unexpected service discovery response contents (Bonjour)")
  68. if "496e7465726e6574" not in ev:
  69. raise Exception("Unexpected service discovery response contents (UPnP)")
  70. def test_p2p_service_discovery_bonjour(dev):
  71. """P2P service discovery (Bonjour)"""
  72. ev = run_sd(dev, "00:00:00:00:00:00", "02000101")
  73. if "0b5f6166706f766572746370c00c000c01" not in ev:
  74. raise Exception("Unexpected service discovery response contents (Bonjour)")
  75. if "045f697070c00c000c01" not in ev:
  76. raise Exception("Unexpected service discovery response contents (Bonjour)")
  77. if "496e7465726e6574" in ev:
  78. raise Exception("Unexpected service discovery response contents (UPnP not expected)")
  79. def test_p2p_service_discovery_bonjour2(dev):
  80. """P2P service discovery (Bonjour AFS)"""
  81. ev = run_sd(dev, "00:00:00:00:00:00", "130001010b5f6166706f766572746370c00c000c01")
  82. if "0b5f6166706f766572746370c00c000c01" not in ev:
  83. raise Exception("Unexpected service discovery response contents (Bonjour)")
  84. if "045f697070c00c000c01" in ev:
  85. raise Exception("Unexpected service discovery response contents (Bonjour mismatching)")
  86. if "496e7465726e6574" in ev:
  87. raise Exception("Unexpected service discovery response contents (UPnP not expected)")
  88. def test_p2p_service_discovery_upnp(dev):
  89. """P2P service discovery (UPnP)"""
  90. ev = run_sd(dev, "00:00:00:00:00:00", "02000201")
  91. if "0b5f6166706f766572746370c00c000c01" in ev:
  92. raise Exception("Unexpected service discovery response contents (Bonjour not expected)")
  93. if "496e7465726e6574" not in ev:
  94. raise Exception("Unexpected service discovery response contents (UPnP)")
  95. def test_p2p_service_discovery_upnp2(dev):
  96. """P2P service discovery (UPnP using request helper)"""
  97. ev = run_sd(dev, "00:00:00:00:00:00", "upnp 10 ssdp:all", "0b00020110737364703a616c6c")
  98. if "0b5f6166706f766572746370c00c000c01" in ev:
  99. raise Exception("Unexpected service discovery response contents (Bonjour not expected)")
  100. if "496e7465726e6574" not in ev:
  101. raise Exception("Unexpected service discovery response contents (UPnP)")
  102. def test_p2p_service_discovery_ws(dev):
  103. """P2P service discovery (WS-Discovery)"""
  104. ev = run_sd(dev, "00:00:00:00:00:00", "02000301")
  105. if "0b5f6166706f766572746370c00c000c01" in ev:
  106. raise Exception("Unexpected service discovery response contents (Bonjour not expected)")
  107. if "496e7465726e6574" in ev:
  108. raise Exception("Unexpected service discovery response contents (UPnP not expected)")
  109. if "0300030101" not in ev:
  110. raise Exception("Unexpected service discovery response contents (WS)")
  111. def test_p2p_service_discovery_req_cancel(dev):
  112. """Cancel a P2P service discovery request"""
  113. if "FAIL" not in dev[0].request("P2P_SERV_DISC_CANCEL_REQ ab"):
  114. raise Exception("Unexpected SD cancel success")
  115. query = dev[0].request("P2P_SERV_DISC_REQ " + dev[1].p2p_dev_addr() + " 02000001")
  116. if "OK" not in dev[0].request("P2P_SERV_DISC_CANCEL_REQ " + query):
  117. raise Exception("Unexpected SD cancel failure")
  118. query1 = dev[0].request("P2P_SERV_DISC_REQ " + dev[1].p2p_dev_addr() + " 02000001")
  119. query2 = dev[0].request("P2P_SERV_DISC_REQ " + dev[1].p2p_dev_addr() + " 02000002")
  120. query3 = dev[0].request("P2P_SERV_DISC_REQ " + dev[1].p2p_dev_addr() + " 02000003")
  121. if "OK" not in dev[0].request("P2P_SERV_DISC_CANCEL_REQ " + query2):
  122. raise Exception("Unexpected SD cancel failure")
  123. if "OK" not in dev[0].request("P2P_SERV_DISC_CANCEL_REQ " + query1):
  124. raise Exception("Unexpected SD cancel failure")
  125. if "OK" not in dev[0].request("P2P_SERV_DISC_CANCEL_REQ " + query3):
  126. raise Exception("Unexpected SD cancel failure")