test_p2ps.py 46 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965
  1. # P2P services
  2. # Copyright (c) 2014-2015, Qualcomm Atheros, Inc.
  3. #
  4. # This software may be distributed under the terms of the BSD license.
  5. # See README for more details.
  6. import logging
  7. logger = logging.getLogger()
  8. import time
  9. import random
  10. import re
  11. import hwsim_utils
  12. from wpasupplicant import WpaSupplicant
  13. from test_p2p_grpform import check_grpform_results
  14. from test_p2p_grpform import remove_group
  15. from test_p2p_persistent import go_neg_pin_authorized_persistent
  16. from utils import HwsimSkip
  17. # Dev[0] -> Advertiser
  18. # Dev[1] -> Seeker
  19. # ev0 -> Event generated at advertiser side
  20. # ev1 -> Event generated at Seeker side
  21. def p2ps_advertise(r_dev, r_role, svc_name, srv_info, rsp_info=None, cpt=None):
  22. """P2PS Advertise function"""
  23. adv_id = random.randrange(1, 0xFFFFFFFF)
  24. advid = hex(adv_id)[2:]
  25. cpt_param = (" cpt=" + cpt) if cpt is not None else ""
  26. if rsp_info is not None and srv_info is not None:
  27. if "OK" not in r_dev.global_request("P2P_SERVICE_ADD asp " + str(r_role) + " " + str(advid) + " 1 1108 " + svc_name + cpt_param + " svc_info='" + srv_info + "'" + " rsp_info=" + rsp_info + "'"):
  28. raise Exception("P2P_SERVICE_ADD with response info and service info failed")
  29. if rsp_info is None and srv_info is not None:
  30. if "OK" not in r_dev.global_request("P2P_SERVICE_ADD asp " + str(r_role) + " " + str(advid) + " 1 1108 " + svc_name + cpt_param + " svc_info='" + srv_info + "'"):
  31. raise Exception("P2P_SERVICE_ADD with service info failed")
  32. if rsp_info is None and srv_info is None:
  33. if "OK" not in r_dev.global_request("P2P_SERVICE_ADD asp " + str(r_role) + " " + str(advid) + " 1 1108 " + svc_name + cpt_param):
  34. raise Exception("P2P_SERVICE_ADD without service info and without response info failed")
  35. if rsp_info is not None and srv_info is None:
  36. if "OK" not in r_dev.global_request("P2P_SERVICE_ADD asp " + str(r_role) + " " + str(adv_id) + " 1 1108 " + svc_name + cpt_param + " svc_info='" + " rsp_info=" + rsp_info + "'"):
  37. raise Exception("P2P_SERVICE_ADD with response info failed")
  38. r_dev.p2p_listen()
  39. return advid
  40. def p2ps_exact_seek(i_dev, r_dev, svc_name, srv_info=None,
  41. single_peer_expected=True):
  42. """P2PS exact service seek request"""
  43. if srv_info is not None:
  44. ev1 = i_dev.global_request("P2P_SERV_DISC_REQ 00:00:00:00:00:00 asp 1 " + svc_name + " '" + srv_info + "'")
  45. if ev1 is None:
  46. raise Exception("Failed to add Service Discovery request for exact seek request")
  47. if "OK" not in i_dev.global_request("P2P_FIND 10 type=social seek=" + svc_name):
  48. raise Exception("Failed to initiate seek operation")
  49. timeout = time.time() + 10
  50. ev1 = i_dev.wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  51. while ev1 is not None and not single_peer_expected:
  52. if r_dev.p2p_dev_addr() in ev1 and "adv_id=" in ev1:
  53. break
  54. ev1 = i_dev.wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  55. if timeout < time.time():
  56. raise Exception("Device not found")
  57. if ev1 is None:
  58. raise Exception("P2P-DEVICE-FOUND timeout on seeker side")
  59. if r_dev.p2p_dev_addr() not in ev1:
  60. raise Exception("Unexpected peer")
  61. if srv_info is None:
  62. adv_id = ev1.split("adv_id=")[1].split(" ")[0]
  63. rcvd_svc_name = ev1.split("asp_svc=")[1].split(" ")[0]
  64. if rcvd_svc_name != svc_name:
  65. raise Exception("service name not matching")
  66. else:
  67. ev1 = i_dev.wait_global_event(["P2P-SERV-ASP-RESP"], timeout=10)
  68. if ev1 is None:
  69. raise Exception("Failed to receive Service Discovery Response")
  70. if r_dev.p2p_dev_addr() not in ev1:
  71. raise Exception("Service Discovery response from Unknown Peer")
  72. if srv_info is not None and srv_info not in ev1:
  73. raise Exception("service info not available in Service Discovery response")
  74. adv_id = ev1.split(" ")[3]
  75. rcvd_svc_name = ev1.split(" ")[6]
  76. if rcvd_svc_name != svc_name:
  77. raise Exception("service name not matching")
  78. return [adv_id, rcvd_svc_name]
  79. def p2ps_nonexact_seek(i_dev, r_dev, svc_name, srv_info=None, adv_num=None):
  80. """P2PS nonexact service seek request"""
  81. if adv_num is None:
  82. adv_num = 1
  83. if srv_info is not None:
  84. ev1 = i_dev.global_request("P2P_SERV_DISC_REQ 00:00:00:00:00:00 asp 1 " + svc_name + " '" + srv_info + "'")
  85. else:
  86. ev1 = i_dev.global_request("P2P_SERV_DISC_REQ 00:00:00:00:00:00 asp 1 " + svc_name + " '")
  87. if ev1 is None:
  88. raise Exception("Failed to add Service Discovery request for nonexact seek request")
  89. if "OK" not in i_dev.global_request("P2P_FIND 10 type=social seek="):
  90. raise Exception("Failed to initiate seek")
  91. ev1 = i_dev.wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  92. if ev1 is None:
  93. raise Exception("P2P-DEVICE-FOUND timeout on seeker side")
  94. if r_dev.p2p_dev_addr() not in ev1:
  95. raise Exception("Unexpected peer")
  96. ev_list = []
  97. for i in range (0, adv_num):
  98. ev1 = i_dev.wait_global_event(["P2P-SERV-ASP-RESP"], timeout=10)
  99. if ev1 is None:
  100. raise Exception("Failed to receive Service Discovery Response")
  101. if r_dev.p2p_dev_addr() not in ev1:
  102. raise Exception("Service Discovery response from Unknown Peer")
  103. if srv_info is not None and srv_info not in ev1:
  104. raise Exception("service info not available in Service Discovery response")
  105. adv_id = ev1.split(" ")[3]
  106. rcvd_svc_name = ev1.split(" ")[6]
  107. ev_list.append(''.join([adv_id, ' ', rcvd_svc_name]))
  108. return ev_list
  109. def p2ps_parse_event(ev, *args):
  110. ret = ()
  111. for arg in args:
  112. m = re.search("\s+" + arg + r"=(\S+)", ev)
  113. ret += (m.group(1) if m is not None else None,)
  114. return ret
  115. def p2ps_provision(seeker, advertiser, adv_id, auto_accept=True, method="1000", adv_cpt=None, seeker_cpt=None):
  116. addr0 = seeker.p2p_dev_addr()
  117. addr1 = advertiser.p2p_dev_addr()
  118. seeker.asp_provision(addr1, adv_id=str(adv_id), adv_mac=addr1, session_id=1,
  119. session_mac=addr0, method=method, cpt=seeker_cpt)
  120. if not auto_accept or method == "100":
  121. pin = None
  122. ev_pd_start = advertiser.wait_global_event(["P2PS-PROV-START"],
  123. timeout=10)
  124. if ev_pd_start is None:
  125. raise Exception("P2PS-PROV-START timeout on Advertiser side")
  126. peer = ev_pd_start.split()[1]
  127. advert_id, advert_mac, session, session_mac =\
  128. p2ps_parse_event(ev_pd_start, "adv_id", "adv_mac", "session", "mac")
  129. ev = seeker.wait_global_event(["P2P-PROV-DISC-FAILURE"], timeout=10)
  130. if ev is None:
  131. raise Exception("P2P-PROV-DISC-FAILURE timeout on seeker side")
  132. if method == "100":
  133. ev = advertiser.wait_global_event(["P2P-PROV-DISC-ENTER-PIN"],
  134. timeout=10)
  135. if ev is None:
  136. raise Exception("P2P-PROV-DISC-ENTER-PIN timeout on advertiser side")
  137. if addr0 not in ev:
  138. raise Exception("Unknown peer " + addr0)
  139. ev = seeker.wait_global_event(["P2P-PROV-DISC-SHOW-PIN"],
  140. timeout=10)
  141. if ev is None:
  142. raise Exception("P2P-PROV-DISC-SHOW-PIN timeout on seeker side")
  143. if addr1 not in ev:
  144. raise Exception("Unknown peer " + addr1)
  145. pin = ev.split()[2]
  146. elif method == "8":
  147. ev = advertiser.wait_global_event(["P2P-PROV-DISC-SHOW-PIN"],
  148. timeout=10)
  149. if ev is None:
  150. raise Exception("P2P-PROV-DISC-SHOW-PIN timeout on advertiser side")
  151. if addr0 not in ev:
  152. raise Exception("Unknown peer " + addr0)
  153. pin = ev.split()[2]
  154. advertiser.asp_provision(peer, adv_id=advert_id, adv_mac=advert_mac,
  155. session_id=int(session, 0),
  156. session_mac=session_mac, status=12,
  157. cpt=adv_cpt)
  158. ev1 = seeker.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  159. if ev1 is None:
  160. raise Exception("P2PS-PROV-DONE timeout on seeker side")
  161. ev2 = advertiser.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  162. if ev2 is None:
  163. raise Exception("P2PS-PROV-DONE timeout on advertiser side")
  164. if method == "8":
  165. ev = seeker.wait_global_event(["P2P-PROV-DISC-ENTER-PIN"],
  166. timeout=10)
  167. if ev is None:
  168. raise Exception("P2P-PROV-DISC-ENTER-PIN failed on seeker side")
  169. if addr1 not in ev:
  170. raise Exception("Unknown peer " + addr1)
  171. if pin is not None:
  172. return ev1, ev2, pin
  173. return ev1, ev2
  174. # Auto-accept is true and the method is either P2PS or advertiser is DISPLAY
  175. ev1 = seeker.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  176. if ev1 is None:
  177. raise Exception("P2PS-PROV-DONE timeout on seeker side")
  178. ev2 = advertiser.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  179. if ev2 is None:
  180. raise Exception("P2PS-PROV-DONE timeout on advertiser side")
  181. if method == "8":
  182. ev = seeker.wait_global_event(["P2P-PROV-DISC-ENTER-PIN"], timeout=10)
  183. if ev is None:
  184. raise Exception("P2P-PROV-DISC-ENTER-PIN timeout on seeker side")
  185. if addr1 not in ev:
  186. raise Exception("Unknown peer " + addr1)
  187. ev = advertiser.wait_global_event(["P2P-PROV-DISC-SHOW-PIN"],
  188. timeout=10)
  189. if ev is None:
  190. raise Exception("P2P-PROV-DISC-SHOW-PIN timeout on advertiser side")
  191. if addr0 not in ev:
  192. raise Exception("Unknown peer " + addr0)
  193. pin = ev.split()[2]
  194. return ev1, ev2, pin
  195. return ev1, ev2
  196. def p2ps_connect_pd(dev0, dev1, ev0, ev1, pin=None):
  197. conf_methods_map = {"8": "p2ps", "1": "display", "5": "keypad"}
  198. peer0 = ev0.split()[1]
  199. peer1 = ev1.split()[1]
  200. status0, conncap0, adv_id0, adv_mac0, mac0, session0, dev_passwd_id0, go0, join0, feature_cap0, persist0 =\
  201. p2ps_parse_event(ev0, "status", "conncap", "adv_id", "adv_mac", "mac", "session", "dev_passwd_id", "go", "join", "feature_cap", "persist")
  202. status1, conncap1, adv_id1, adv_mac1, mac1, session1, dev_passwd_id1, go1, join1, feature_cap1, persist1 =\
  203. p2ps_parse_event(ev1, "status", "conncap", "adv_id", "adv_mac", "mac", "session", "dev_passwd_id", "go", "join", "feature_cap", "persist")
  204. if status0 != "0" and status0 != "12":
  205. raise Exception("PD failed on " + dev0.p2p_dev_addr())
  206. if status1 != "0" and status1 != "12":
  207. raise Exception("PD failed on " + dev1.p2p_dev_addr())
  208. if status0 == "12" and status1 == "12":
  209. raise Exception("Both sides have status 12 which doesn't make sense")
  210. if adv_id0 != adv_id1 or adv_id0 is None:
  211. raise Exception("Adv. IDs don't match")
  212. if adv_mac0 != adv_mac1 or adv_mac0 is None:
  213. raise Exception("Adv. MACs don't match")
  214. if session0 != session1 or session0 is None:
  215. raise Exception("Session IDs don't match")
  216. if mac0 != mac1 or mac0 is None:
  217. raise Exception("Session MACs don't match")
  218. #TODO: Validate feature capability
  219. if bool(persist0) != bool(persist1):
  220. raise Exception("Only one peer has persistent group")
  221. if persist0 is None and not all([conncap0, conncap1, dev_passwd_id0,
  222. dev_passwd_id1]):
  223. raise Exception("Persistent group not used but conncap/dev_passwd_id are missing")
  224. if persist0 is not None and any([conncap0, conncap1, dev_passwd_id0,
  225. dev_passwd_id1]):
  226. raise Exception("Persistent group is used but conncap/dev_passwd_id are present")
  227. # Persistent Connection (todo: handle frequency)
  228. if persist0 is not None:
  229. if "OK" not in dev0.global_request("P2P_GROUP_ADD persistent=" + persist0 + " freq=2412"):
  230. raise Exception("Could not re-start persistent group")
  231. ev0 = dev0.wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  232. if ev0 is None:
  233. raise Exception("P2P-GROUP-STARTED timeout on " + dev0.p2p_dev_addr())
  234. dev0.group_form_result(ev0)
  235. if "OK" not in dev1.global_request("P2P_GROUP_ADD persistent=" + persist1 + " freq=2412"):
  236. raise Exception("Could not re-start persistent group")
  237. ev1 = dev1.wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  238. if ev1 is None:
  239. raise Exception("P2P-GROUP-STARTED timeout on " + dev1.p2p_dev_addr())
  240. dev1.group_form_result(ev1)
  241. if "GO" in ev0:
  242. ev = dev0.wait_global_event(["AP-STA-CONNECTED"], timeout=10)
  243. if ev is None:
  244. raise Exception("AP-STA-CONNECTED timeout on " + dev0.p2p_dev_addr())
  245. else:
  246. ev = dev1.wait_global_event(["AP-STA-CONNECTED"], timeout=10)
  247. if ev is None:
  248. raise Exception("AP-STA-CONNECTED timeout on " + dev1.p2p_dev_addr())
  249. else:
  250. try:
  251. method0 = conf_methods_map[dev_passwd_id0]
  252. method1 = conf_methods_map[dev_passwd_id1]
  253. except KeyError:
  254. raise Exception("Unsupported method")
  255. if method0 == "p2ps":
  256. pin = "12345670"
  257. if pin is None:
  258. raise Exception("Pin is not provided")
  259. if conncap0 == "1" and conncap1 == "1": # NEW/NEW - GON
  260. if any([join0, join1, go0, go1]):
  261. raise Exception("Unexpected join/go PD attributes")
  262. dev0.p2p_listen()
  263. if "OK" not in dev0.global_request("P2P_CONNECT " + peer0 + " " + pin + " " + method0 + " persistent auth"):
  264. raise Exception("P2P_CONNECT fails on " + dev0.p2p_dev_addr())
  265. if "OK" not in dev1.global_request("P2P_CONNECT " + peer1 + " " + pin + " " + method1 + " persistent"):
  266. raise Exception("P2P_CONNECT fails on " + dev1.p2p_dev_addr())
  267. ev = dev0.wait_global_event(["P2P-GO-NEG-SUCCESS"], timeout=10)
  268. if ev is None:
  269. raise Exception("GO Neg did not succeed on " + dev0.p2p_dev_addr())
  270. ev = dev1.wait_global_event(["P2P-GO-NEG-SUCCESS"], timeout=10)
  271. if ev is None:
  272. raise Exception("GO Neg did not succeed on " + dev1.p2p_dev_addr())
  273. ev = dev0.wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  274. if ev is None:
  275. raise Exception("P2P-GROUP-STARTED timeout on " + dev0.p2p_dev_addr())
  276. dev0.group_form_result(ev)
  277. ev = dev1.wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  278. if ev is None:
  279. raise Exception("P2P-GROUP-STARTED timeout on " + dev1.p2p_dev_addr())
  280. dev1.group_form_result(ev)
  281. else:
  282. if conncap0 == "2" and conncap1 == "4": # dev0 CLI, dev1 GO
  283. dev_cli, dev_go, go_if, join_address, go_method, cli_method = dev0, dev1, go1, join0, method1, method0
  284. elif conncap0 == "4" and conncap1 == "2": # dev0 GO, dev1 CLI
  285. dev_cli, dev_go, go_if, join_address, go_method, cli_method = dev1, dev0, go0, join1, method0, method1
  286. else:
  287. raise Exception("Bad connection capabilities")
  288. if go_if is None:
  289. raise Exception("Device " + dev_go.p2p_dev_addr() + " failed to become GO")
  290. if join_address is None:
  291. raise Exception("Device " + dev_cli.p2p_dev_addr() + " failed to become CLI")
  292. ev = dev_go.wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  293. if ev is None:
  294. raise Exception("P2P-GROUP-STARTED timeout on " + dev_go.p2p_dev_addr())
  295. dev_go.group_form_result(ev)
  296. if go_method != "p2ps":
  297. ev = dev_go.group_request("WPS_PIN any " + pin)
  298. if ev is None:
  299. raise Exception("Failed to initiate pin authorization on registrar side")
  300. if "OK" not in dev_cli.global_request("P2P_CONNECT " + join_address + " " + pin + " " + cli_method + " persistent join"):
  301. raise Exception("P2P_CONNECT failed on " + dev_cli.p2p_dev_addr())
  302. ev = dev_cli.wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  303. if ev is None:
  304. raise Exception("P2P-GROUP-STARTED timeout on " + dev_cli.p2p_dev_addr())
  305. dev_cli.group_form_result(ev)
  306. ev = dev_go.wait_global_event(["AP-STA-CONNECTED"], timeout=10)
  307. if ev is None:
  308. raise Exception("AP-STA-CONNECTED timeout on " + dev_go.p2p_dev_addr())
  309. hwsim_utils.test_connectivity_p2p(dev0, dev1)
  310. def set_no_group_iface(dev, enable):
  311. if enable:
  312. res = dev.get_driver_status()
  313. if (int(res['capa.flags'], 0) & 0x20000000):
  314. raise HwsimSkip("P2P Device used. Cannot set enable no_group_iface")
  315. dev.global_request("SET p2p_no_group_iface 1")
  316. else:
  317. dev.global_request("SET p2p_no_group_iface 0")
  318. def test_p2ps_exact_search(dev):
  319. """P2PS exact service request"""
  320. p2ps_advertise(r_dev=dev[0], r_role='1', svc_name='org.wi-fi.wfds.send.rx',
  321. srv_info='I can receive files upto size 2 GB')
  322. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  323. svc_name='org.wi-fi.wfds.send.rx')
  324. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  325. if ev0 is None:
  326. raise Exception("Unable to remove the advertisement instance")
  327. def test_p2ps_exact_search_srvinfo(dev):
  328. """P2PS exact service request with service info"""
  329. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  330. srv_info='I can receive files upto size 2 GB')
  331. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  332. svc_name='org.wi-fi.wfds.send.rx',
  333. srv_info='2 GB')
  334. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  335. if ev0 is None:
  336. raise Exception("Unable to remove the advertisement instance")
  337. def test_p2ps_nonexact_search(dev):
  338. """P2PS nonexact seek request"""
  339. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.play.rx',
  340. srv_info='I support Miracast Mode ')
  341. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  342. svc_name='org.wi-fi.wfds.play*')
  343. adv_id = ev_list[0].split()[0]
  344. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  345. if ev0 is None:
  346. raise Exception("Unable to remove the advertisement instance")
  347. def test_p2ps_nonexact_search_srvinfo(dev):
  348. """P2PS nonexact seek request with service info"""
  349. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  350. srv_info='I can receive files upto size 2 GB')
  351. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  352. svc_name='org.wi-fi.wfds.send*',
  353. srv_info='2 GB')
  354. adv_id = ev_list[0].split()[0]
  355. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  356. if ev0 is None:
  357. raise Exception("Unable to remove the advertisement instance")
  358. def test_p2ps_connect_p2ps_method_nonautoaccept(dev):
  359. """P2PS connect for non-auto-accept and P2PS config method"""
  360. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  361. srv_info='I can receive files upto size 2 GB')
  362. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  363. svc_name='org.wi-fi.wfds.send*',
  364. srv_info='2 GB')
  365. adv_id = ev_list[0].split()[0]
  366. ev1, ev0 = p2ps_provision(dev[1], dev[0], adv_id, auto_accept=False)
  367. p2ps_connect_pd(dev[0], dev[1], ev0, ev1)
  368. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  369. if ev0 is None:
  370. raise Exception("Unable to remove the advertisement instance")
  371. remove_group(dev[0], dev[1])
  372. def test_p2ps_connect_p2ps_method_autoaccept(dev):
  373. """P2PS connection with P2PS default config method and auto-accept"""
  374. p2ps_advertise(r_dev=dev[0], r_role='1', svc_name='org.wi-fi.wfds.send.rx',
  375. srv_info='I can receive files upto size 2 GB')
  376. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  377. svc_name='org.wi-fi.wfds.send.rx',
  378. srv_info='2 GB')
  379. ev1, ev0 = p2ps_provision(dev[1], dev[0], adv_id)
  380. p2ps_connect_pd(dev[0], dev[1], ev0, ev1)
  381. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  382. if ev0 is None:
  383. raise Exception("Unable to remove the advertisement instance")
  384. remove_group(dev[0], dev[1])
  385. def test_p2ps_connect_keypad_method_nonautoaccept(dev):
  386. """P2PS Connection with non-auto-accept and seeker having keypad method"""
  387. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  388. srv_info='I can receive files upto size 2 GB')
  389. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  390. svc_name='org.wi-fi.wfds.send*',
  391. srv_info='2 GB')
  392. adv_id = ev_list[0].split()[0]
  393. ev1, ev0, pin = p2ps_provision(dev[1], dev[0], adv_id, auto_accept=False, method="8")
  394. p2ps_connect_pd(dev[0], dev[1], ev0, ev1, pin)
  395. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  396. if ev0 is None:
  397. raise Exception("Unable to remove the advertisement instance")
  398. remove_group(dev[0], dev[1])
  399. def test_p2ps_connect_display_method_nonautoaccept(dev):
  400. """P2PS connection with non-auto-accept and seeker having display method"""
  401. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  402. srv_info='I can receive files upto size 2 GB')
  403. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  404. svc_name='org.wi-fi.wfds*', srv_info='2 GB')
  405. adv_id = ev_list[0].split()[0]
  406. ev1, ev0, pin = p2ps_provision(dev[1], dev[0], adv_id, auto_accept=False, method="100")
  407. p2ps_connect_pd(dev[0], dev[1], ev0, ev1, pin)
  408. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  409. if ev0 is None:
  410. raise Exception("Unable to remove the advertisement instance")
  411. remove_group(dev[0], dev[1])
  412. def test_p2ps_connect_keypad_method_autoaccept(dev):
  413. """P2PS connection with auto-accept and keypad method on seeker side"""
  414. p2ps_advertise(r_dev=dev[0], r_role='1', svc_name='org.wi-fi.wfds.send.rx',
  415. srv_info='I can receive files upto size 2 GB')
  416. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  417. svc_name='org.wi-fi.wfds.send.rx',
  418. srv_info='2 GB')
  419. ev1, ev0, pin = p2ps_provision(dev[1], dev[0], adv_id, method="8")
  420. p2ps_connect_pd(dev[0], dev[1], ev0, ev1, pin)
  421. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  422. if ev0 is None:
  423. raise Exception("Unable to remove the advertisement instance")
  424. remove_group(dev[0], dev[1])
  425. def test_p2ps_connect_display_method_autoaccept(dev):
  426. """P2PS connection with auto-accept and display method on seeker side"""
  427. p2ps_advertise(r_dev=dev[0], r_role='1', svc_name='org.wi-fi.wfds.send.rx',
  428. srv_info='I can receive files upto size 2 GB')
  429. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  430. svc_name='org.wi-fi.wfds.send.rx',
  431. srv_info='2 GB')
  432. ev1, ev0, pin = p2ps_provision(dev[1], dev[0], adv_id, method="100")
  433. p2ps_connect_pd(dev[0], dev[1], ev0, ev1, pin)
  434. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  435. if ev0 is None:
  436. raise Exception("Unable to remove the advertisement instance")
  437. remove_group(dev[0], dev[1])
  438. def test_p2ps_connect_adv_go_p2ps_method(dev):
  439. """P2PS auto-accept connection with advertisement as GO and P2PS method"""
  440. p2ps_advertise(r_dev=dev[0], r_role='4', svc_name='org.wi-fi.wfds.send.rx',
  441. srv_info='I can receive files upto size 2 GB')
  442. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  443. svc_name='org.wi-fi.wfds.send.rx',
  444. srv_info='2 GB')
  445. ev1, ev0 = p2ps_provision(dev[1], dev[0], adv_id)
  446. p2ps_connect_pd(dev[0], dev[1], ev0, ev1)
  447. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  448. if ev0 is None:
  449. raise Exception("Unable to remove the advertisement instance")
  450. remove_group(dev[0], dev[1])
  451. def test_p2ps_connect_adv_client_p2ps_method(dev):
  452. """P2PS auto-accept connection with advertisement as Client and P2PS method"""
  453. p2ps_advertise(r_dev=dev[0], r_role='2', svc_name='org.wi-fi.wfds.send.rx',
  454. srv_info='I can receive files upto size 2 GB')
  455. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  456. svc_name='org.wi-fi.wfds.send.rx',
  457. srv_info='2 GB')
  458. ev1, ev0 = p2ps_provision(dev[1], dev[0], adv_id)
  459. p2ps_connect_pd(dev[0], dev[1], ev0, ev1)
  460. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  461. if ev0 is None:
  462. raise Exception("Unable to remove the advertisement instance")
  463. remove_group(dev[0], dev[1])
  464. def p2ps_connect_adv_go_pin_method(dev, keep_group=False):
  465. p2ps_advertise(r_dev=dev[0], r_role='4', svc_name='org.wi-fi.wfds.send.rx',
  466. srv_info='I can receive files upto size 2 GB')
  467. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  468. svc_name='org.wi-fi.wfds.send.rx',
  469. srv_info='2 GB')
  470. ev1, ev0, pin = p2ps_provision(dev[1], dev[0], adv_id, method="8")
  471. p2ps_connect_pd(dev[0], dev[1], ev0, ev1, pin)
  472. if not keep_group:
  473. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  474. if ev0 is None:
  475. raise Exception("Unable to remove the advertisement instance")
  476. remove_group(dev[0], dev[1])
  477. def test_p2ps_connect_adv_go_pin_method(dev):
  478. """P2PS advertiser as GO with keypad config method on seeker side and auto-accept"""
  479. p2ps_connect_adv_go_pin_method(dev)
  480. def test_p2ps_connect_adv_client_pin_method(dev):
  481. """P2PS advertiser as client with keypad config method on seeker side and auto-accept"""
  482. dev[0].flush_scan_cache()
  483. p2ps_advertise(r_dev=dev[0], r_role='2', svc_name='org.wi-fi.wfds.send.rx',
  484. srv_info='I can receive files upto size 2 GB')
  485. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  486. svc_name='org.wi-fi.wfds.send.rx',
  487. srv_info='2 GB')
  488. ev1, ev0, pin = p2ps_provision(dev[1], dev[0], adv_id, method="8")
  489. p2ps_connect_pd(dev[0], dev[1], ev0, ev1, pin)
  490. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  491. if ev0 is None:
  492. raise Exception("Unable to remove the advertisement instance")
  493. remove_group(dev[0], dev[1])
  494. def test_p2ps_service_discovery_multiple_queries(dev):
  495. """P2P service discovery with multiple queries"""
  496. addr0 = dev[0].p2p_dev_addr()
  497. addr1 = dev[1].p2p_dev_addr()
  498. adv_id1 = p2ps_advertise(r_dev=dev[0], r_role='0',
  499. svc_name='org.wi-fi.wfds.send.tx',
  500. srv_info='I can transfer files upto size of 2 GB')
  501. adv_id2 = p2ps_advertise(r_dev=dev[0], r_role='0',
  502. svc_name='org.wi-fi.wfds.send.rx',
  503. srv_info='I can receive files upto size of 2 GB')
  504. adv_id3 = p2ps_advertise(r_dev=dev[0], r_role='1',
  505. svc_name='org.wi-fi.wfds.display.tx',
  506. srv_info='Miracast Mode')
  507. adv_id4 = p2ps_advertise(r_dev=dev[0], r_role='1',
  508. svc_name='org.wi-fi.wfds.display.rx',
  509. srv_info='Miracast Mode')
  510. dev[1].global_request("P2P_SERV_DISC_REQ " + addr0 + " asp 1 org.wi-fi.wfds.display.tx 'Miracast Mode'")
  511. dev[1].global_request("P2P_FIND 10 type=social seek=org.wi-fi.wfds.display.tx")
  512. dev[1].global_request("P2P_SERV_DISC_REQ " + addr0 + " asp 2 org.wi-fi.wfds.send* 'size of 2 GB'")
  513. dev[1].p2p_stop_find()
  514. dev[1].global_request("P2P_FIND 10 type=social seek=")
  515. ev = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  516. if ev is None:
  517. raise Exception("P2P Device Found timed out")
  518. if addr0 not in ev:
  519. raise Exception("Unexpected service discovery request source")
  520. ev_list = []
  521. for i in range(0, 3):
  522. ev = dev[1].wait_global_event(["P2P-SERV-ASP-RESP"], timeout=10)
  523. if ev is None:
  524. raise Exception("P2P Service discovery timed out")
  525. if addr0 in ev:
  526. ev_list.append(ev)
  527. if len(ev_list) == 3:
  528. break
  529. dev[1].p2p_stop_find()
  530. for test in [ ("seek=org.wi-fi.wfds.display.TX",
  531. "asp_svc=org.wi-fi.wfds.display.tx"),
  532. ("seek=foo seek=org.wi-fi.wfds.display.tx seek=bar",
  533. "asp_svc=org.wi-fi.wfds.display.tx"),
  534. ("seek=1 seek=2 seek=3 seek=org.wi-fi.wfds.display.tx seek=4 seek=5 seek=6",
  535. "asp_svc=org.wi-fi.wfds.display.tx"),
  536. ("seek=not-found", None),
  537. ("seek=org.wi-fi.wfds", "asp_svc=org.wi-fi.wfds")]:
  538. dev[2].global_request("P2P_FIND 10 type=social " + test[0])
  539. if test[1] is None:
  540. ev = dev[2].wait_global_event(["P2P-DEVICE-FOUND"], timeout=1)
  541. if ev is not None:
  542. raise Exception("Unexpected device found: " + ev)
  543. continue
  544. ev = dev[2].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  545. if ev is None:
  546. raise Exception("P2P device discovery timed out (dev2)")
  547. if test[1] not in ev:
  548. raise Exception("Expected asp_svc not reported: " + ev)
  549. dev[2].p2p_stop_find()
  550. dev[2].request("P2P_FLUSH")
  551. dev[0].p2p_stop_find()
  552. ev1 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id1))
  553. if ev1 is None:
  554. raise Exception("Unable to remove the advertisement instance")
  555. ev2 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id2))
  556. if ev2 is None:
  557. raise Exception("Unable to remove the advertisement instance")
  558. ev3 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id3))
  559. if ev3 is None:
  560. raise Exception("Unable to remove the advertisement instance")
  561. ev4 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id4))
  562. if ev4 is None:
  563. raise Exception("Unable to remove the advertisement instance")
  564. if "OK" not in dev[0].global_request("P2P_SERVICE_ADD asp 1 12345678 1 1108 org.wi-fi.wfds.foobar svc_info='Test'"):
  565. raise Exception("P2P_SERVICE_ADD failed")
  566. if "OK" not in dev[0].global_request("P2P_SERVICE_DEL asp all"):
  567. raise Exception("P2P_SERVICE_DEL asp all failed")
  568. if "OK" not in dev[0].global_request("P2P_SERVICE_ADD asp 1 12345678 1 1108 org.wi-fi.wfds.foobar svc_info='Test'"):
  569. raise Exception("P2P_SERVICE_ADD failed")
  570. if "OK" not in dev[0].global_request("P2P_SERVICE_REP asp 1 12345678 1 1108 org.wi-fi.wfds.foobar svc_info='Test'"):
  571. raise Exception("P2P_SERVICE_REP failed")
  572. if "FAIL" not in dev[0].global_request("P2P_SERVICE_REP asp 1 12345678 1 1108 org.wi-fi.wfds.Foo svc_info='Test'"):
  573. raise Exception("Invalid P2P_SERVICE_REP accepted")
  574. if "OK" not in dev[0].global_request("P2P_SERVICE_ADD asp 1 a2345678 1 1108 org.wi-fi.wfds.something svc_info='Test'"):
  575. raise Exception("P2P_SERVICE_ADD failed")
  576. if "OK" not in dev[0].global_request("P2P_SERVICE_ADD asp 1 a2345679 1 1108 org.wi-fi.wfds.Foo svc_info='Test'"):
  577. raise Exception("P2P_SERVICE_ADD failed")
  578. def get_ifnames():
  579. with open('/proc/net/dev', 'r') as f:
  580. data = f.read()
  581. ifnames = []
  582. for line in data.splitlines():
  583. ifname = line.strip().split(' ')[0]
  584. if ':' not in ifname:
  585. continue
  586. ifname = ifname.split(':')[0]
  587. ifnames.append(ifname)
  588. return ifnames
  589. def p2ps_connect_p2ps_method(dev, keep_group=False):
  590. dev[0].flush_scan_cache()
  591. dev[1].flush_scan_cache()
  592. p2ps_advertise(r_dev=dev[0], r_role='2', svc_name='org.wi-fi.wfds.send.rx',
  593. srv_info='I can receive files upto size 2 GB')
  594. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  595. svc_name='org.wi-fi.wfds.send.rx',
  596. srv_info='2 GB')
  597. ev1, ev0 = p2ps_provision(dev[1], dev[0], adv_id)
  598. ifnames = get_ifnames()
  599. p2ps_connect_pd(dev[0], dev[1], ev0, ev1)
  600. grp_ifname0 = dev[0].get_group_ifname()
  601. grp_ifname1 = dev[1].get_group_ifname()
  602. if not keep_group:
  603. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  604. if ev0 is None:
  605. raise Exception("Unable to remove the advertisement instance")
  606. ifnames = ifnames + get_ifnames()
  607. remove_group(dev[0], dev[1])
  608. ifnames = ifnames + get_ifnames()
  609. return grp_ifname0, grp_ifname1, ifnames
  610. def has_string_prefix(vals, prefix):
  611. for val in vals:
  612. if val.startswith(prefix):
  613. return True
  614. return False
  615. def test_p2ps_connect_p2ps_method_1(dev):
  616. """P2PS connection with P2PS method - no group interface"""
  617. set_no_group_iface(dev[0], 1)
  618. set_no_group_iface(dev[1], 1)
  619. (grp_ifname0, grp_ifname1, ifnames) = p2ps_connect_p2ps_method(dev)
  620. if grp_ifname0 != dev[0].ifname:
  621. raise Exception("unexpected dev0 group ifname: " + grp_ifname0)
  622. if grp_ifname1 != dev[1].ifname:
  623. raise Exception("unexpected dev1 group ifname: " + grp_ifname1)
  624. if has_string_prefix(ifnames, 'p2p-' + grp_ifname0):
  625. raise Exception("dev0 group interface unexpectedly present")
  626. if has_string_prefix(ifnames, 'p2p-' + grp_ifname1):
  627. raise Exception("dev1 group interface unexpectedly present")
  628. def test_p2ps_connect_p2ps_method_2(dev):
  629. """P2PS connection with P2PS method - group interface on dev0"""
  630. set_no_group_iface(dev[0], 0)
  631. set_no_group_iface(dev[1], 1)
  632. (grp_ifname0, grp_ifname1, ifnames) = p2ps_connect_p2ps_method(dev)
  633. if not grp_ifname0.startswith('p2p-' + dev[0].ifname + '-'):
  634. raise Exception("unexpected dev0 group ifname: " + grp_ifname0)
  635. if grp_ifname1 != dev[1].ifname:
  636. raise Exception("unexpected dev1 group ifname: " + grp_ifname1)
  637. if has_string_prefix(ifnames, 'p2p-' + grp_ifname0):
  638. raise Exception("dev0 group interface unexpectedly present")
  639. def test_p2ps_connect_p2ps_method_3(dev):
  640. """P2PS connection with P2PS method - group interface on dev1"""
  641. set_no_group_iface(dev[0], 1)
  642. set_no_group_iface(dev[1], 0)
  643. (grp_ifname0, grp_ifname1, ifnames) = p2ps_connect_p2ps_method(dev)
  644. if grp_ifname0 != dev[0].ifname:
  645. raise Exception("unexpected dev0 group ifname: " + grp_ifname0)
  646. if not grp_ifname1.startswith('p2p-' + dev[1].ifname + '-'):
  647. raise Exception("unexpected dev1 group ifname: " + grp_ifname1)
  648. if has_string_prefix(ifnames, 'p2p-' + grp_ifname0):
  649. raise Exception("dev0 group interface unexpectedly present")
  650. def test_p2ps_connect_p2ps_method_4(dev):
  651. """P2PS connection with P2PS method - group interface on both"""
  652. set_no_group_iface(dev[0], 0)
  653. set_no_group_iface(dev[1], 0)
  654. (grp_ifname0, grp_ifname1, ifnames) = p2ps_connect_p2ps_method(dev)
  655. if not grp_ifname0.startswith('p2p-' + dev[0].ifname + '-'):
  656. raise Exception("unexpected dev0 group ifname: " + res0['ifname'])
  657. if not grp_ifname1.startswith('p2p-' + dev[1].ifname + '-'):
  658. raise Exception("unexpected dev1 group ifname: " + res1['ifname'])
  659. def test_p2ps_connect_adv_go_persistent(dev):
  660. """P2PS auto-accept connection with advertisement as GO and having persistent group"""
  661. go_neg_pin_authorized_persistent(i_dev=dev[0], i_intent=15,
  662. r_dev=dev[1], r_intent=0)
  663. dev[0].remove_group()
  664. dev[1].wait_go_ending_session()
  665. p2ps_advertise(r_dev=dev[0], r_role='4', svc_name='org.wi-fi.wfds.send.rx',
  666. srv_info='I can receive files upto size 2 GB')
  667. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  668. svc_name='org.wi-fi.wfds.send.rx',
  669. srv_info='2 GB')
  670. ev1, ev0 = p2ps_provision(dev[1], dev[0], adv_id)
  671. if "persist=" not in ev0 or "persist=" not in ev1:
  672. raise Exception("Persistent group isn't used by peers")
  673. p2ps_connect_pd(dev[0], dev[1], ev0, ev1)
  674. remove_group(dev[0], dev[1])
  675. def test_p2ps_client_probe(dev):
  676. """P2PS CLI discoverability on operating channel"""
  677. cli_probe = dev[0].global_request("SET p2p_cli_probe 1")
  678. p2ps_connect_p2ps_method(dev, keep_group=True)
  679. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[2], r_dev=dev[0],
  680. svc_name='org.wi-fi.wfds.send.rx',
  681. single_peer_expected=False)
  682. dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  683. remove_group(dev[0], dev[1])
  684. def test_p2ps_go_probe(dev):
  685. """P2PS GO discoverability on operating channel"""
  686. p2ps_connect_adv_go_pin_method(dev, keep_group=True)
  687. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[2], r_dev=dev[0],
  688. svc_name='org.wi-fi.wfds.send.rx',
  689. single_peer_expected=False)
  690. dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  691. remove_group(dev[0], dev[1])
  692. def test_p2ps_wildcard_p2ps(dev):
  693. """P2PS wildcard SD Probe Request/Response"""
  694. p2ps_wildcard = "org.wi-fi.wfds"
  695. adv_id = p2ps_advertise(r_dev=dev[0], r_role='1',
  696. svc_name='org.foo.service',
  697. srv_info='I can do stuff')
  698. adv_id2 = p2ps_advertise(r_dev=dev[0], r_role='1',
  699. svc_name='org.wi-fi.wfds.send.rx',
  700. srv_info='I can receive files upto size 2 GB')
  701. if "OK" not in dev[1].global_request("P2P_FIND 10 type=social seek=org.foo.service seek=" + p2ps_wildcard):
  702. raise Exception("Failed on P2P_FIND command")
  703. ev1 = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  704. if ev1 is None:
  705. raise Exception("P2P-DEVICE-FOUND timeout on seeker side")
  706. if dev[0].p2p_dev_addr() not in ev1:
  707. raise Exception("Unexpected peer")
  708. ev2 = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  709. if ev2 is None:
  710. raise Exception("P2P-DEVICE-FOUND timeout on seeker side (2)")
  711. if dev[0].p2p_dev_addr() not in ev2:
  712. raise Exception("Unexpected peer (2)")
  713. if p2ps_wildcard not in ev1 + ev2:
  714. raise Exception("P2PS Wildcard name not found in P2P-DEVICE-FOUND event")
  715. if "org.foo.service" not in ev1 + ev2:
  716. raise Exception("Vendor specific service name not found in P2P-DEVICE-FOUND event")
  717. if "OK" not in dev[1].global_request("P2P_STOP_FIND"):
  718. raise Exception("P2P_STOP_FIND failed")
  719. dev[1].dump_monitor()
  720. res = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  721. if res is None:
  722. raise Exception("Unable to remove the advertisement instance")
  723. if "OK" not in dev[1].global_request("P2P_FIND 10 type=social seek=" + p2ps_wildcard):
  724. raise Exception("Failed on P2P_FIND command")
  725. ev1 = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  726. if ev1 is None:
  727. raise Exception("P2P-DEVICE-FOUND timeout on seeker side")
  728. if dev[0].p2p_dev_addr() not in ev1:
  729. raise Exception("Unexpected peer")
  730. if p2ps_wildcard not in ev1:
  731. raise Exception("P2PS Wildcard name not found in P2P-DEVICE-FOUND event (2)")
  732. dev[1].dump_monitor()
  733. res = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id2))
  734. if res is None:
  735. raise Exception("Unable to remove the advertisement instance 2")
  736. if "OK" not in dev[1].global_request("P2P_FIND 10 type=social seek=" + p2ps_wildcard):
  737. raise Exception("Failed on P2P_FIND command")
  738. ev1 = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=2)
  739. if ev1 is not None:
  740. raise Exception("Unexpected P2P-DEVICE-FOUND event on seeker side")
  741. dev[1].p2p_stop_find()
  742. dev[1].dump_monitor()
  743. def test_p2ps_many_services_in_probe(dev):
  744. """P2PS with large number of services in Probe Request/Response"""
  745. long1 = 'org.example.0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789.a'
  746. long2 = 'org.example.0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789.b'
  747. long3 = 'org.example.0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789.c'
  748. long4 = 'org.example.0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789.d'
  749. long5 = 'org.example.0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789.e'
  750. for name in [ long1, long2, long3, long4, long5 ]:
  751. p2ps_advertise(r_dev=dev[0], r_role='1',
  752. svc_name=name,
  753. srv_info='I can do stuff')
  754. if "OK" not in dev[1].global_request("P2P_FIND 10 type=social seek=%s seek=%s seek=%s seek=%s seek=%s" % (long1, long2, long3, long4, long5)):
  755. raise Exception("Failed on P2P_FIND command")
  756. events = ""
  757. # Note: Require only four events since all the services do not fit within
  758. # the length limit.
  759. for i in range(4):
  760. ev = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  761. if ev is None:
  762. raise Exception("Missing P2P-DEVICE-FOUND")
  763. events = events + ev
  764. dev[1].p2p_stop_find()
  765. dev[1].dump_monitor()
  766. for name in [ long2, long3, long4, long5 ]:
  767. if name not in events:
  768. raise Exception("Service missing from peer events")
  769. def p2ps_test_feature_capability_cpt(dev, adv_cpt, seeker_cpt, adv_role,
  770. result):
  771. p2ps_advertise(r_dev=dev[0], r_role=adv_role,
  772. svc_name='org.wi-fi.wfds.send.rx',
  773. srv_info='I can receive files upto size 2 GB', cpt=adv_cpt)
  774. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  775. svc_name='org.wi-fi.wfds.send.rx',
  776. srv_info='2 GB')
  777. auto_accept = adv_role != "0"
  778. ev1, ev0, pin = p2ps_provision(dev[1], dev[0], adv_id,
  779. auto_accept=auto_accept, adv_cpt=adv_cpt,
  780. seeker_cpt=seeker_cpt, method="8")
  781. status0, fcap0 = p2ps_parse_event(ev0, "status", "feature_cap")
  782. status1, fcap1 = p2ps_parse_event(ev0, "status", "feature_cap")
  783. if fcap0 is None:
  784. raise Exception("Bad feature capability on Seeker side")
  785. if fcap1 is None:
  786. raise Exception("Bad feature capability on Advertiser side")
  787. if fcap0 != fcap1:
  788. raise Exception("Incompatible feature capability values")
  789. if status0 not in ("0", "12") or status1 not in ("0", "12"):
  790. raise Exception("Unexpected PD result status")
  791. if result == "UDP" and fcap0[1] != "1":
  792. raise Exception("Unexpected CPT feature capability value (expected: UDP)")
  793. elif result == "MAC" and fcap0[1] != "2":
  794. raise Exception("Unexpected CPT feature capability value (expected: MAC)")
  795. ev = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  796. if ev is None:
  797. raise Exception("Unable to remove the advertisement instance")
  798. def test_p2ps_feature_capability_mac_autoaccept(dev):
  799. """P2PS PD Feature Capability CPT: advertiser MAC, seeker UDP:MAC, autoaccept"""
  800. p2ps_test_feature_capability_cpt(dev, adv_cpt="MAC", seeker_cpt="UDP:MAC",
  801. adv_role="4", result="MAC")
  802. def test_p2ps_feature_capability_mac_nonautoaccept(dev):
  803. """P2PS PD Feature Capability CPT: advertiser:MAC, seeker UDP:MAC, nonautoaccept"""
  804. p2ps_test_feature_capability_cpt(dev, adv_cpt="MAC", seeker_cpt="UDP:MAC",
  805. adv_role="0", result="MAC")
  806. def test_p2ps_feature_capability_mac_udp_autoaccept(dev):
  807. """P2PS PD Feature Capability CPT: advertiser MAC:UDP, seeker UDP:MAC, autoaccept"""
  808. p2ps_test_feature_capability_cpt(dev, adv_cpt="MAC:UDP",
  809. seeker_cpt="UDP:MAC", adv_role="2",
  810. result="MAC")
  811. def test_p2ps_feature_capability_mac_udp_nonautoaccept(dev):
  812. """P2PS PD Feature Capability CPT: advertiser MAC:UDP, seeker UDP:MAC, nonautoaccept"""
  813. p2ps_test_feature_capability_cpt(dev, adv_cpt="MAC:UDP",
  814. seeker_cpt="UDP:MAC", adv_role="0",
  815. result="UDP")
  816. def test_p2ps_feature_capability_udp_mac_autoaccept(dev):
  817. """P2PS PD Feature Capability CPT: advertiser UDP:MAC, seeker MAC:UDP, autoaccept"""
  818. p2ps_test_feature_capability_cpt(dev, adv_cpt="UDP:MAC",
  819. seeker_cpt="MAC:UDP", adv_role="2",
  820. result="UDP")
  821. def test_p2ps_feature_capability_udp_mac_nonautoaccept(dev):
  822. """P2PS PD Feature Capability CPT: advertiser UDP:MAC, seeker MAC:UDP, nonautoaccept"""
  823. p2ps_test_feature_capability_cpt(dev, adv_cpt="UDP:MAC",
  824. seeker_cpt="MAC:UDP", adv_role="0",
  825. result="MAC")