test_p2ps.py 57 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174
  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):
  22. """P2PS Advertise function"""
  23. adv_id = random.randrange(1, 0xFFFFFFFF)
  24. advid = hex(adv_id)[2:]
  25. if rsp_info is not None and srv_info is not None:
  26. if "OK" not in r_dev.global_request("P2P_SERVICE_ADD asp " + str(r_role) + " " + str(advid) + " 1 1108 " + svc_name + " svc_info='" + srv_info + "'" + " rsp_info=" + rsp_info + "'"):
  27. raise Exception("P2P_SERVICE_ADD with response info and service info failed")
  28. if rsp_info is None and srv_info is not None:
  29. if "OK" not in r_dev.global_request("P2P_SERVICE_ADD asp " + str(r_role) + " " + str(advid) + " 1 1108 " + svc_name + " svc_info='" + srv_info + "'"):
  30. raise Exception("P2P_SERVICE_ADD with service info failed")
  31. if rsp_info is None and srv_info is None:
  32. if "OK" not in r_dev.global_request("P2P_SERVICE_ADD asp " + str(r_role) + " " + str(advid) + " 1 1108 " + svc_name + " "):
  33. raise Exception("P2P_SERVICE_ADD without service info and without response info failed")
  34. if rsp_info is not None and srv_info is None:
  35. if "OK" not in r_dev.global_request("P2P_SERVICE_ADD asp " + str(r_role) + " " + str(adv_id) + " 1 1108 " + svc_name + " svc_info='" + " rsp_info=" + rsp_info + "'"):
  36. raise Exception("P2P_SERVICE_ADD with response info failed")
  37. r_dev.p2p_listen()
  38. return advid
  39. def p2ps_exact_seek(i_dev, r_dev, svc_name, srv_info=None,
  40. single_peer_expected=True):
  41. """P2PS exact service seek request"""
  42. if srv_info is not None:
  43. ev1 = i_dev.global_request("P2P_SERV_DISC_REQ 00:00:00:00:00:00 asp 1 " + svc_name + " '" + srv_info + "'")
  44. if ev1 is None:
  45. raise Exception("Failed to add Service Discovery request for exact seek request")
  46. if "OK" not in i_dev.global_request("P2P_FIND 10 type=social seek=" + svc_name):
  47. raise Exception("Failed to initiate seek operation")
  48. timeout = time.time() + 10
  49. ev1 = i_dev.wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  50. while ev1 is not None and not single_peer_expected:
  51. if r_dev.p2p_dev_addr() in ev1 and "adv_id=" in ev1:
  52. break
  53. ev1 = i_dev.wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  54. if timeout < time.time():
  55. raise Exception("Device not found")
  56. if ev1 is None:
  57. raise Exception("P2P-DEVICE-FOUND timeout on seeker side")
  58. if r_dev.p2p_dev_addr() not in ev1:
  59. raise Exception("Unexpected peer")
  60. if srv_info is None:
  61. adv_id = ev1.split("adv_id=")[1].split(" ")[0]
  62. rcvd_svc_name = ev1.split("asp_svc=")[1].split(" ")[0]
  63. if rcvd_svc_name != svc_name:
  64. raise Exception("service name not matching")
  65. else:
  66. ev1 = i_dev.wait_global_event(["P2P-SERV-ASP-RESP"], timeout=10)
  67. if ev1 is None:
  68. raise Exception("Failed to receive Service Discovery Response")
  69. if r_dev.p2p_dev_addr() not in ev1:
  70. raise Exception("Service Discovery response from Unknown Peer")
  71. if srv_info is not None and srv_info not in ev1:
  72. raise Exception("service info not available in Service Discovery response")
  73. adv_id = ev1.split(" ")[3]
  74. rcvd_svc_name = ev1.split(" ")[6]
  75. if rcvd_svc_name != svc_name:
  76. raise Exception("service name not matching")
  77. return [adv_id, rcvd_svc_name]
  78. def p2ps_nonexact_seek(i_dev, r_dev, svc_name, srv_info=None, adv_num=None):
  79. """P2PS nonexact service seek request"""
  80. if adv_num is None:
  81. adv_num = 1
  82. if srv_info is not None:
  83. ev1 = i_dev.global_request("P2P_SERV_DISC_REQ 00:00:00:00:00:00 asp 1 " + svc_name + " '" + srv_info + "'")
  84. else:
  85. ev1 = i_dev.global_request("P2P_SERV_DISC_REQ 00:00:00:00:00:00 asp 1 " + svc_name + " '")
  86. if ev1 is None:
  87. raise Exception("Failed to add Service Discovery request for nonexact seek request")
  88. if "OK" not in i_dev.global_request("P2P_FIND 10 type=social seek="):
  89. raise Exception("Failed to initiate seek")
  90. ev1 = i_dev.wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  91. if ev1 is None:
  92. raise Exception("P2P-DEVICE-FOUND timeout on seeker side")
  93. if r_dev.p2p_dev_addr() not in ev1:
  94. raise Exception("Unexpected peer")
  95. ev_list = []
  96. for i in range (0, adv_num):
  97. ev1 = i_dev.wait_global_event(["P2P-SERV-ASP-RESP"], timeout=10)
  98. if ev1 is None:
  99. raise Exception("Failed to receive Service Discovery Response")
  100. if r_dev.p2p_dev_addr() not in ev1:
  101. raise Exception("Service Discovery response from Unknown Peer")
  102. if srv_info is not None and srv_info not in ev1:
  103. raise Exception("service info not available in Service Discovery response")
  104. adv_id = ev1.split(" ")[3]
  105. rcvd_svc_name = ev1.split(" ")[6]
  106. ev_list.append(''.join([adv_id, ' ', rcvd_svc_name]))
  107. return ev_list
  108. def p2ps_parse_event(ev, *args):
  109. ret = ()
  110. for arg in args:
  111. m = re.search("\s+" + arg + r"=(\S+)", ev)
  112. ret += (m.group(1) if m is not None else None,)
  113. return ret
  114. def p2ps_provision(seeker, advertiser, adv_id, auto_accept=True, method="1000"):
  115. addr0 = seeker.p2p_dev_addr()
  116. addr1 = advertiser.p2p_dev_addr()
  117. seeker.asp_provision(addr1, adv_id=str(adv_id), adv_mac=addr1, session_id=1,
  118. session_mac=addr0, method=method)
  119. if not auto_accept or method == "100":
  120. pin = None
  121. ev_pd_start = advertiser.wait_global_event(["P2PS-PROV-START"],
  122. timeout=10)
  123. if ev_pd_start is None:
  124. raise Exception("P2PS-PROV-START timeout on Advertiser side")
  125. peer = ev_pd_start.split()[1]
  126. advert_id, advert_mac, session, session_mac =\
  127. p2ps_parse_event(ev_pd_start, "adv_id", "adv_mac", "session", "mac")
  128. ev = seeker.wait_global_event(["P2P-PROV-DISC-FAILURE"], timeout=10)
  129. if ev is None:
  130. raise Exception("P2P-PROV-DISC-FAILURE timeout on seeker side")
  131. if method == "100":
  132. ev = advertiser.wait_global_event(["P2P-PROV-DISC-ENTER-PIN"],
  133. timeout=10)
  134. if ev is None:
  135. raise Exception("P2P-PROV-DISC-ENTER-PIN timeout on advertiser side")
  136. if addr0 not in ev:
  137. raise Exception("Unknown peer " + addr0)
  138. ev = seeker.wait_global_event(["P2P-PROV-DISC-SHOW-PIN"],
  139. timeout=10)
  140. if ev is None:
  141. raise Exception("P2P-PROV-DISC-SHOW-PIN timeout on seeker side")
  142. if addr1 not in ev:
  143. raise Exception("Unknown peer " + addr1)
  144. pin = ev.split()[2]
  145. elif method == "8":
  146. ev = advertiser.wait_global_event(["P2P-PROV-DISC-SHOW-PIN"],
  147. timeout=10)
  148. if ev is None:
  149. raise Exception("P2P-PROV-DISC-SHOW-PIN timeout on advertiser side")
  150. if addr0 not in ev:
  151. raise Exception("Unknown peer " + addr0)
  152. pin = ev.split()[2]
  153. advertiser.asp_provision(peer, adv_id=advert_id, adv_mac=advert_mac,
  154. session_id=int(session, 0),
  155. session_mac=session_mac, status=12)
  156. ev1 = seeker.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  157. if ev1 is None:
  158. raise Exception("P2PS-PROV-DONE timeout on seeker side")
  159. ev2 = advertiser.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  160. if ev2 is None:
  161. raise Exception("P2PS-PROV-DONE timeout on advertiser side")
  162. if method == "8":
  163. ev = seeker.wait_global_event(["P2P-PROV-DISC-ENTER-PIN"],
  164. timeout=10)
  165. if ev is None:
  166. raise Exception("P2P-PROV-DISC-ENTER-PIN failed on seeker side")
  167. if addr1 not in ev:
  168. raise Exception("Unknown peer " + addr1)
  169. if pin is not None:
  170. return ev1, ev2, pin
  171. return ev1, ev2
  172. # Auto-accept is true and the method is either P2PS or advertiser is DISPLAY
  173. ev1 = seeker.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  174. if ev1 is None:
  175. raise Exception("P2PS-PROV-DONE timeout on seeker side")
  176. ev2 = advertiser.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  177. if ev2 is None:
  178. raise Exception("P2PS-PROV-DONE timeout on advertiser side")
  179. if method == "8":
  180. ev = seeker.wait_global_event(["P2P-PROV-DISC-ENTER-PIN"], timeout=10)
  181. if ev is None:
  182. raise Exception("P2P-PROV-DISC-ENTER-PIN timeout on seeker side")
  183. if addr1 not in ev:
  184. raise Exception("Unknown peer " + addr1)
  185. ev = advertiser.wait_global_event(["P2P-PROV-DISC-SHOW-PIN"],
  186. timeout=10)
  187. if ev is None:
  188. raise Exception("P2P-PROV-DISC-SHOW-PIN timeout on advertiser side")
  189. if addr0 not in ev:
  190. raise Exception("Unknown peer " + addr0)
  191. pin = ev.split()[2]
  192. return ev1, ev2, pin
  193. return ev1, ev2
  194. def p2p_connect_p2ps_method(i_dev, r_dev, autoaccept):
  195. """P2PS connect function with p2ps method"""
  196. if autoaccept == False:
  197. if "OK" not in i_dev.global_request("P2P_CONNECT " + r_dev.p2p_dev_addr() + " 12345670 p2ps persistent auth"):
  198. raise Exception("P2P_CONNECT fails on seeker side")
  199. ev0 = r_dev.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  200. if ev0 is None:
  201. raise Exception("P2PS-PROV-DONE timeout on Advertiser side")
  202. if "OK" not in r_dev.global_request("P2P_CONNECT " + i_dev.p2p_dev_addr() + " 12345670 p2ps persistent"):
  203. raise Exception("P2P_CONNECT fails on Advertiser side")
  204. else:
  205. if "OK" not in r_dev.global_request("P2P_CONNECT " + i_dev.p2p_dev_addr() + " 12345670 p2ps persistent auth"):
  206. raise Exception("P2P_CONNECT fails on Advertiser side")
  207. ev1 = i_dev.wait_global_event(["P2PS-PROV-DONE"], timeout=5)
  208. if ev1 is None:
  209. raise Exception("Failed to receive deferred acceptance at seeker")
  210. if "OK" not in i_dev.global_request("P2P_CONNECT " + r_dev.p2p_dev_addr() + " 12345670 p2ps persistent"):
  211. raise Exception("P2P_CONNECT fails on seeker side")
  212. ev0 = r_dev.wait_global_event(["P2P-GO-NEG-SUCCESS"], timeout=10)
  213. if ev0 is None:
  214. raise Exception("GO Neg did not succeed")
  215. ev0 = r_dev.wait_global_event(["P2P-GROUP-STARTED"], timeout=15)
  216. if ev0 is None:
  217. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  218. r_dev.group_form_result(ev0)
  219. ev1 = i_dev.wait_global_event(["P2P-GROUP-STARTED"], timeout=5)
  220. if ev1 is None:
  221. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  222. i_dev.group_form_result(ev1)
  223. def p2ps_provision_keypad_method(i_dev, r_dev, autoaccept,
  224. initiator_or_responder):
  225. """P2PS keypad method provisioning function"""
  226. if autoaccept == False and initiator_or_responder == 'initiator':
  227. ev = i_dev.wait_global_event(["P2P-PROV-DISC-FAILURE"], timeout=10)
  228. if ev is None:
  229. raise Exception("Provisioning deferred on seeker side")
  230. ev1 = i_dev.wait_global_event(["P2P-PROV-DISC-ENTER-PIN"], timeout=10)
  231. if ev1 is None:
  232. raise Exception("P2P-PROV-DISC-ENTER-PIN timeout on seeker side")
  233. if r_dev.p2p_dev_addr() not in ev1:
  234. raise Exception("Unknown peer ")
  235. ev = r_dev.wait_global_event(["P2PS-PROV-START"], timeout=10)
  236. if ev is None:
  237. raise Exception("P2PS-PROV-START timeout on Advertiser side")
  238. if autoaccept == False and initiator_or_responder == 'responder':
  239. ev0 = r_dev.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  240. if ev0 is None:
  241. raise Exception("P2PS-PROV-DONE timeout on seeker side")
  242. ev0 = r_dev.wait_global_event(["P2P-PROV-DISC-ENTER-PIN"], timeout=5)
  243. if ev0 is None:
  244. raise Exception("P2P-PROV-DISC-ENTER-PIN timeout on advertiser side")
  245. if autoaccept == True and initiator_or_responder == 'initiator':
  246. ev1 = i_dev.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  247. if ev1 is None:
  248. raise Exception("P2PS-PROV-DONE timeout on seeker side")
  249. ev2 = i_dev.wait_global_event(["P2P-PROV-DISC-ENTER-PIN"], timeout=10)
  250. if ev2 is None:
  251. raise Exception("P2P-PROV-DISC-ENTER-PIN failed on seeker side")
  252. if r_dev.p2p_dev_addr() not in ev2:
  253. raise Exception("Unknown peer ")
  254. return ev1
  255. def p2ps_provision_display_method(i_dev, r_dev, autoaccept,
  256. initiator_or_responder):
  257. """P2PS display method provisioning function"""
  258. if initiator_or_responder == 'initiator':
  259. ev0 = r_dev.wait_global_event(["P2PS-PROV-START"], timeout=10)
  260. if ev0 is None:
  261. raise Exception("P2PS-PROV-START timeout on Advertiser side")
  262. if autoaccept == False:
  263. ev = i_dev.wait_global_event(["P2P-PROV-DISC-FAILURE"], timeout=10)
  264. if ev is None:
  265. raise Exception("Provisioning deferred on seeker side")
  266. ev1 = i_dev.wait_global_event(["P2P-PROV-DISC-SHOW-PIN"], timeout=10)
  267. if ev1 is None:
  268. raise Exception("P2P-PROV-DISC-SHOW-PIN timeout on Seeker side")
  269. if r_dev.p2p_dev_addr() not in ev1:
  270. raise Exception("Unknown peer ")
  271. pin = ev1.split(" ")[2]
  272. else:
  273. ev0 = r_dev.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  274. if ev0 is None:
  275. raise Exception("P2PS-PROV-DONE timeout on advertiser")
  276. ev0 = r_dev.wait_global_event(["P2P-PROV-DISC-SHOW-PIN"], timeout=5)
  277. if ev0 is None:
  278. raise Exception("PIN Display on advertiser side")
  279. pin = ev0.split(" ")[2]
  280. return pin
  281. def p2ps_connect_pin(pin, i_dev, r_dev, initiator_method):
  282. """P2PS function to perform connection using PIN method"""
  283. if initiator_method=="display":
  284. if "OK" not in i_dev.global_request("P2P_CONNECT " + r_dev.p2p_dev_addr() + " " + pin + " display persistent "):
  285. raise Exception("P2P_CONNECT fails on seeker side")
  286. ev0 = r_dev.wait_global_event(["P2P-GO-NEG-REQUEST"], timeout=10)
  287. if ev0 is None:
  288. raise Exception("Failed to receive P2P_GO-NEG-REQUEST on responder side")
  289. if "OK" not in r_dev.global_request("P2P_CONNECT " + i_dev.p2p_dev_addr() + " " + pin + " keypad persistent "):
  290. raise Exception("P2P_CONNECT fails on Advertiser side")
  291. else:
  292. if "OK" not in i_dev.global_request("P2P_CONNECT " + r_dev.p2p_dev_addr() + " " + pin + " keypad persistent "):
  293. raise Exception("P2P_CONNECT fails on seeker side")
  294. ev0 = r_dev.wait_global_event(["P2P-GO-NEG-REQUEST"], timeout=10)
  295. if ev0 is None:
  296. raise Exception("Failed to receive P2P_GO-NEG-REQUEST on responder side")
  297. if "OK" not in r_dev.global_request("P2P_CONNECT " + i_dev.p2p_dev_addr() + " " + pin + " display persistent "):
  298. raise Exception("P2P_CONNECT fails on Advertiser side")
  299. ev0 = r_dev.wait_global_event(["P2P-GO-NEG-SUCCESS"], timeout=10)
  300. if ev0 is None:
  301. raise Exception("GO Neg did not succeed on advertiser side")
  302. peer_mac = ev0.split("peer_dev=")[1].split(" ")[0]
  303. ev1 = i_dev.wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  304. if ev1 is None:
  305. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  306. i_dev.group_form_result(ev1)
  307. ev_grpfrm = r_dev.wait_global_event(["P2P-GROUP-FORMATION-SUCCESS"],
  308. timeout=10)
  309. if ev_grpfrm is None:
  310. raise Exception("Group Formation failed on advertiser side")
  311. ev = r_dev.wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  312. if ev is None:
  313. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  314. res = r_dev.group_form_result(ev)
  315. if res['role'] == "GO":
  316. ev_grpfrm = r_dev.wait_global_event(["AP-STA-CONNECTED"], timeout=10)
  317. if ev_grpfrm is None:
  318. raise Exception("AP-STA-CONNECTED timeout on advertiser side")
  319. if i_dev.p2p_dev_addr() not in ev_grpfrm:
  320. raise Exception("Group formed with unknown Peer")
  321. else:
  322. ev1 = i_dev.wait_global_event(["AP-STA-CONNECTED"], timeout=5)
  323. if ev1 is None:
  324. raise Exception("AP-STA-CONNECTED timeout on Seeker side")
  325. if r_dev.p2p_dev_addr() not in ev1:
  326. raise Exception("Group formed with unknown Peer")
  327. def set_no_group_iface(dev, enable):
  328. if enable:
  329. res = dev.get_driver_status()
  330. if (int(res['capa.flags'], 0) & 0x20000000):
  331. raise HwsimSkip("P2P Device used. Cannot set enable no_group_iface")
  332. dev.global_request("SET p2p_no_group_iface 1")
  333. else:
  334. dev.global_request("SET p2p_no_group_iface 0")
  335. def test_p2ps_exact_search(dev):
  336. """P2PS exact service request"""
  337. p2ps_advertise(r_dev=dev[0], r_role='1', svc_name='org.wi-fi.wfds.send.rx',
  338. srv_info='I can receive files upto size 2 GB')
  339. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  340. svc_name='org.wi-fi.wfds.send.rx')
  341. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  342. if ev0 is None:
  343. raise Exception("Unable to remove the advertisement instance")
  344. def test_p2ps_exact_search_srvinfo(dev):
  345. """P2PS exact service request with service info"""
  346. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  347. srv_info='I can receive files upto size 2 GB')
  348. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  349. svc_name='org.wi-fi.wfds.send.rx',
  350. srv_info='2 GB')
  351. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  352. if ev0 is None:
  353. raise Exception("Unable to remove the advertisement instance")
  354. def test_p2ps_nonexact_search(dev):
  355. """P2PS nonexact seek request"""
  356. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.play.rx',
  357. srv_info='I support Miracast Mode ')
  358. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  359. svc_name='org.wi-fi.wfds.play*')
  360. adv_id = ev_list[0].split()[0]
  361. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  362. if ev0 is None:
  363. raise Exception("Unable to remove the advertisement instance")
  364. def test_p2ps_nonexact_search_srvinfo(dev):
  365. """P2PS nonexact seek request with service info"""
  366. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  367. srv_info='I can receive files upto size 2 GB')
  368. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  369. svc_name='org.wi-fi.wfds.send*',
  370. srv_info='2 GB')
  371. adv_id = ev_list[0].split()[0]
  372. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  373. if ev0 is None:
  374. raise Exception("Unable to remove the advertisement instance")
  375. def test_p2ps_connect_p2ps_method_nonautoaccept(dev):
  376. """P2PS connect for non-auto-accept and P2PS config method"""
  377. addr0 = dev[0].p2p_dev_addr()
  378. addr1 = dev[1].p2p_dev_addr()
  379. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  380. srv_info='I can receive files upto size 2 GB')
  381. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  382. svc_name='org.wi-fi.wfds.send*',
  383. srv_info='2 GB')
  384. adv_id = ev_list[0].split(" ")[0]
  385. if "OK" not in dev[1].global_request("P2P_ASP_PROVISION " + addr0 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " info='' method=1000"):
  386. raise Exception("Failed to request provisioning on seeker")
  387. ev0 = dev[0].wait_global_event(["P2PS-PROV-START"], timeout=10)
  388. if ev0 is None:
  389. raise Exception("P2PS-PROV-START timeout on advertiser side")
  390. ev1 = dev[1].wait_global_event(["P2P-PROV-DISC-FAILURE"], timeout=15)
  391. if ev1 is None:
  392. raise Exception("Provisioning deferred timeout on seeker side")
  393. dev[1].p2p_listen()
  394. if "OK" not in dev[0].global_request("P2P_ASP_PROVISION_RESP " + addr1 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " status=12"):
  395. raise Exception("Failed to send deferred acceptance from advertizer")
  396. ev1 = dev[1].wait_global_event(["P2PS-PROV-DONE"], timeout=15)
  397. if ev1 is None:
  398. raise Exception("Failed to receive deferred acceptance at seeker")
  399. p2p_connect_p2ps_method(i_dev=dev[1], r_dev=dev[0], autoaccept=False)
  400. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  401. if ev0 is None:
  402. raise Exception("Unable to remove the advertisement instance")
  403. remove_group(dev[0], dev[1])
  404. def test_p2ps_connect_p2ps_method_autoaccept(dev):
  405. """P2PS connection with P2PS default config method and auto-accept"""
  406. addr0 = dev[0].p2p_dev_addr()
  407. addr1 = dev[1].p2p_dev_addr()
  408. p2ps_advertise(r_dev=dev[0], r_role='1', svc_name='org.wi-fi.wfds.send.rx',
  409. srv_info='I can receive files upto size 2 GB')
  410. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  411. svc_name='org.wi-fi.wfds.send.rx',
  412. srv_info='2 GB')
  413. if "OK" not in dev[1].global_request("P2P_ASP_PROVISION " + addr0 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " info='' method=1000"):
  414. raise Exception("P2P_ASP_PROVISION failed on seeker side")
  415. ev0 = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  416. if ev0 is None:
  417. raise Exception("P2PS-PROV-DONE timeout on advertiser side")
  418. p2p_connect_p2ps_method(i_dev=dev[1], r_dev=dev[0], autoaccept=True)
  419. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  420. if ev0 is None:
  421. raise Exception("Unable to remove the advertisement instance")
  422. remove_group(dev[0], dev[1])
  423. def test_p2ps_connect_keypad_method_nonautoaccept(dev):
  424. """P2PS Connection with non-auto-accept and seeker having keypad method"""
  425. addr0 = dev[0].p2p_dev_addr()
  426. addr1 = dev[1].p2p_dev_addr()
  427. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  428. srv_info='I can receive files upto size 2 GB')
  429. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  430. svc_name='org.wi-fi.wfds.send*',
  431. srv_info='2 GB')
  432. adv_id = ev_list[0].split(" ")[0]
  433. if "OK" not in dev[1].global_request("P2P_ASP_PROVISION " + addr0 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " info='' method=8"): # keypad method on seeker side
  434. raise Exception("Failed to request provisioning on seeker")
  435. p2ps_provision_keypad_method(i_dev=dev[1], r_dev=dev[0], autoaccept=False,
  436. initiator_or_responder='initiator')
  437. dev[1].p2p_listen()
  438. if "OK" not in dev[0].global_request("P2P_ASP_PROVISION_RESP " + addr1 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " status=12"):
  439. raise Exception("Failed to send deferred acceptance from advertizer")
  440. pin = p2ps_provision_display_method(i_dev=dev[1], r_dev=dev[0],
  441. autoaccept=False,
  442. initiator_or_responder='responder')
  443. p2ps_connect_pin(pin, i_dev=dev[0], r_dev=dev[1],
  444. initiator_method="display")
  445. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  446. if ev0 is None:
  447. raise Exception("Unable to remove the advertisement instance")
  448. remove_group(dev[0], dev[1])
  449. def test_p2ps_connect_display_method_nonautoaccept(dev):
  450. """P2PS connection with non-auto-accept and seeker having display method"""
  451. addr0 = dev[0].p2p_dev_addr()
  452. addr1 = dev[1].p2p_dev_addr()
  453. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  454. srv_info='I can receive files upto size 2 GB')
  455. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  456. svc_name='org.wi-fi.wfds*', srv_info='2 GB')
  457. adv_id = ev_list[0].split(" ")[0]
  458. if "OK" not in dev[1].global_request("P2P_ASP_PROVISION " + addr0 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " info='' method=100"): # keypad method on seeker side
  459. raise Exception("Failed to request provisioning on seeker")
  460. pin = p2ps_provision_display_method(i_dev=dev[1], r_dev=dev[0],
  461. autoaccept=False,
  462. initiator_or_responder='initiator')
  463. dev[1].p2p_listen()
  464. if "OK" not in dev[0].global_request("P2P_ASP_PROVISION_RESP " + addr1 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " status=12"):
  465. raise Exception("Failed to send deferred acceptance from advertiser")
  466. p2ps_provision_keypad_method(i_dev=dev[1], r_dev=dev[0], autoaccept=False,
  467. initiator_or_responder='responder')
  468. p2ps_connect_pin(pin, i_dev=dev[0], r_dev=dev[1], initiator_method="keypad")
  469. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  470. if ev0 is None:
  471. raise Exception("Unable to remove the advertisement instance")
  472. remove_group(dev[0], dev[1])
  473. def test_p2ps_connect_keypad_method_autoaccept(dev):
  474. """P2PS connection with auto-accept and keypad method on seeker side"""
  475. addr0 = dev[0].p2p_dev_addr()
  476. addr1 = dev[1].p2p_dev_addr()
  477. p2ps_advertise(r_dev=dev[0], r_role='1', svc_name='org.wi-fi.wfds.send.rx',
  478. srv_info='I can receive files upto size 2 GB')
  479. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  480. svc_name='org.wi-fi.wfds.send.rx',
  481. srv_info='2 GB')
  482. if "OK" not in dev[1].global_request("P2P_ASP_PROVISION " + addr0 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " info='' method=8"): # keypad method on seeker side
  483. raise Exception("Failed to request provisioning on seeker")
  484. p2ps_provision_keypad_method(i_dev=dev[1], r_dev=dev[0], autoaccept=True,
  485. initiator_or_responder='initiator')
  486. pin = p2ps_provision_display_method(i_dev=dev[1], r_dev=dev[0],
  487. autoaccept=True,
  488. initiator_or_responder='responder')
  489. p2ps_connect_pin(pin, i_dev=dev[1], r_dev=dev[0], initiator_method="Keypad")
  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_connect_display_method_autoaccept(dev):
  495. """P2PS connection with auto-accept and display method on seeker side"""
  496. addr0 = dev[0].p2p_dev_addr()
  497. addr1 = dev[1].p2p_dev_addr()
  498. p2ps_advertise(r_dev=dev[0], r_role='1', svc_name='org.wi-fi.wfds.send.rx',
  499. srv_info='I can receive files upto size 2 GB')
  500. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  501. svc_name='org.wi-fi.wfds.send.rx',
  502. srv_info='2 GB')
  503. if "OK" not in dev[1].global_request("P2P_ASP_PROVISION " + addr0 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " info='' method=100"): # display method on seeker side
  504. raise Exception("Failed to request provisioning on seeker")
  505. pin = p2ps_provision_display_method(i_dev=dev[1], r_dev=dev[0],
  506. autoaccept=True,
  507. initiator_or_responder='initiator')
  508. dev[1].p2p_listen()
  509. p2ps_connect_pin(pin, i_dev=dev[1], r_dev=dev[0], initiator_method="display")
  510. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  511. if ev0 is None:
  512. raise Exception("Unable to remove the advertisement instance")
  513. remove_group(dev[0], dev[1])
  514. def test_p2ps_connect_adv_go_p2ps_method(dev):
  515. """P2PS auto-accept connection with advertisement as GO and P2PS method"""
  516. addr0 = dev[0].p2p_dev_addr()
  517. addr1 = dev[1].p2p_dev_addr()
  518. p2ps_advertise(r_dev=dev[0], r_role='4', svc_name='org.wi-fi.wfds.send.rx',
  519. srv_info='I can receive files upto size 2 GB')
  520. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  521. svc_name='org.wi-fi.wfds.send.rx',
  522. srv_info='2 GB')
  523. if "OK" not in dev[1].global_request("P2P_ASP_PROVISION " + addr0 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " info='' method=1000"):
  524. raise Exception("Failed to request provisioning on seeker")
  525. ev0 = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  526. if ev0 is None:
  527. raise Exception("Timed out while waiting for prov done on advertizer")
  528. if "go=" not in ev0:
  529. raise Exception("Advertiser failed to become GO")
  530. adv_conncap = ev0.split("conncap=")[1].split(" ")[0]
  531. if adv_conncap == "4":
  532. logger.info("Advertiser is GO")
  533. ev0 = dev[0].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  534. if ev0 is None:
  535. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  536. dev[0].group_form_result(ev0)
  537. ev1 = dev[1].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  538. if ev1 is None:
  539. raise Exception("P2PS-PROV-DONE timeout on seeker side")
  540. seeker_conncap = ev1.split("conncap=")[1].split(" ")[0]
  541. if "join=" in ev1:
  542. if "OK" not in dev[1].global_request("P2P_CONNECT " + addr0 + " 12345670 p2ps persistent join"):
  543. raise Exception("P2P_CONNECT failed on seeker side")
  544. ev1 = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=15)
  545. if ev1 is None:
  546. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  547. dev[1].group_form_result(ev1)
  548. ev0 = dev[0].wait_global_event(["AP-STA-CONNECTED"], timeout=5)
  549. if ev0 is None:
  550. raise Exception("AP-STA-CONNECTED timeout on advertiser side")
  551. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  552. if ev0 is None:
  553. raise Exception("Unable to remove the advertisement instance")
  554. remove_group(dev[0], dev[1])
  555. def test_p2ps_connect_adv_client_p2ps_method(dev):
  556. """P2PS auto-accept connection with advertisement as Client and P2PS method"""
  557. addr0 = dev[0].p2p_dev_addr()
  558. addr1 = dev[1].p2p_dev_addr()
  559. p2ps_advertise(r_dev=dev[0], r_role='2', svc_name='org.wi-fi.wfds.send.rx',
  560. srv_info='I can receive files upto size 2 GB')
  561. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  562. svc_name='org.wi-fi.wfds.send.rx',
  563. srv_info='2 GB')
  564. if "OK" not in dev[1].global_request("P2P_ASP_PROVISION " + addr0 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " info='' method=1000"):
  565. raise Exception("Failed to request provisioning on seeker")
  566. ev0 = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  567. if ev0 is None:
  568. raise Exception("P2PS-PROV-DONE timeout on advertiser side")
  569. if "join=" not in ev0:
  570. raise Exception("Advertiser failed to become Client")
  571. adv_conncap = ev0.split("conncap=")[1].split(" ")[0]
  572. if adv_conncap == "2":
  573. logger.info("Advertiser is Client")
  574. ev1 = dev[1].wait_global_event(["P2PS-PROV-DONE"], timeout=5)
  575. if ev1 is None:
  576. raise Exception("Provisioning failed on seeker side")
  577. seeker_conncap = ev1.split("conncap=")[1].split(" ")[0]
  578. ev1 = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  579. if ev1 is None:
  580. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  581. dev[1].group_form_result(ev1)
  582. if "join=" in ev0:
  583. if "OK" not in dev[0].global_request("P2P_CONNECT " + addr1 + " 12345670 p2ps persistent join"):
  584. raise Exception("P2P_CONNECT failed on seeker side")
  585. ev0 = dev[0].wait_global_event(["P2P-GROUP-FORMATION-SUCCESS"], timeout=15)
  586. if ev0 is None:
  587. raise Exception("P2P Group Formation failed on advertiser side")
  588. ev0 = dev[0].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  589. if ev0 is None:
  590. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  591. dev[0].group_form_result(ev0)
  592. ev1 = dev[1].wait_global_event(["AP-STA-CONNECTED"], timeout=5)
  593. if ev1 is None:
  594. raise Exception("Group formation failed")
  595. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  596. if ev0 is None:
  597. raise Exception("Unable to remove the advertisement instance")
  598. remove_group(dev[0], dev[1])
  599. def p2ps_connect_adv_go_pin_method(dev, keep_group=False):
  600. addr0 = dev[0].p2p_dev_addr()
  601. addr1 = dev[1].p2p_dev_addr()
  602. p2ps_advertise(r_dev=dev[0], r_role='4', svc_name='org.wi-fi.wfds.send.rx',
  603. srv_info='I can receive files upto size 2 GB')
  604. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  605. svc_name='org.wi-fi.wfds.send.rx',
  606. srv_info='2 GB')
  607. if "OK" not in dev[1].global_request("P2P_ASP_PROVISION " + addr0 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " info='' method=8"): # keypad method on seeker side
  608. raise Exception("Failed to request provisioning on seeker")
  609. seek_prov_ev = p2ps_provision_keypad_method(i_dev=dev[1], r_dev=dev[0],
  610. autoaccept=True,
  611. initiator_or_responder='initiator')
  612. ev0 = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  613. if ev0 is None:
  614. raise Exception("P2PS-PROV-DONE timeout on advertier side")
  615. adv_conncap = ev0.split("conncap=")[1].split(" ")[0]
  616. if adv_conncap == "4":
  617. logger.info("Advertiser is GO")
  618. ev0 = dev[0].wait_global_event(["P2P-PROV-DISC-SHOW-PIN"], timeout=5)
  619. if ev0 is None:
  620. raise Exception("PIN Display on advertiser side")
  621. pin = ev0.split(" ")[2]
  622. ev0 = dev[0].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  623. if ev0 is None:
  624. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  625. dev[0].group_form_result(ev0)
  626. ev0 = dev[0].group_request("WPS_PIN any " + pin)
  627. if ev0 is None:
  628. raise Exception("Failed to initiate Pin authorization on registrar side")
  629. if "join=" in seek_prov_ev:
  630. if "OK" not in dev[1].global_request("P2P_CONNECT " + addr0 + " " + pin + " keypad persistent join"):
  631. raise Exception("P2P_CONNECT failed on seeker side")
  632. ev1 = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  633. if ev1 is None:
  634. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  635. dev[1].group_form_result(ev1)
  636. ev0 = dev[0].wait_global_event(["AP-STA-CONNECTED"], timeout=10)
  637. if ev0 is None:
  638. raise Exception("AP-STA-CONNECTED timeout on advertiser side")
  639. if not keep_group:
  640. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  641. if ev0 is None:
  642. raise Exception("Unable to remove the advertisement instance")
  643. remove_group(dev[0], dev[1])
  644. def test_p2ps_connect_adv_go_pin_method(dev):
  645. """P2PS advertiser as GO with keypad config method on seeker side and auto-accept"""
  646. p2ps_connect_adv_go_pin_method(dev)
  647. def test_p2ps_connect_adv_client_pin_method(dev):
  648. """P2PS advertiser as client with keypad config method on seeker side and auto-accept"""
  649. addr0 = dev[0].p2p_dev_addr()
  650. addr1 = dev[1].p2p_dev_addr()
  651. dev[0].flush_scan_cache()
  652. p2ps_advertise(r_dev=dev[0], r_role='2', svc_name='org.wi-fi.wfds.send.rx',
  653. srv_info='I can receive files upto size 2 GB')
  654. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  655. svc_name='org.wi-fi.wfds.send.rx',
  656. srv_info='2 GB')
  657. if "OK" not in dev[1].global_request("P2P_ASP_PROVISION " + addr0 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " info='' method=8"): # keypad method on seeker side
  658. raise Exception("Failed to request provisioning on seeker")
  659. seek_prov_ev = p2ps_provision_keypad_method(i_dev=dev[1], r_dev=dev[0],
  660. autoaccept=True,
  661. initiator_or_responder='initiator')
  662. adv_prov = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  663. if adv_prov is None:
  664. raise Exception("Prov failed on advertiser")
  665. adv_conncap = adv_prov.split("conncap=")[1].split(" ")[0]
  666. if adv_conncap == "2":
  667. logger.info("Advertiser is Client")
  668. adv_pin_show_event = dev[0].wait_global_event(["P2P-PROV-DISC-SHOW-PIN"],
  669. timeout=5)
  670. if adv_pin_show_event is None:
  671. raise Exception("PIN Display on advertiser side")
  672. pin = adv_pin_show_event.split(" ")[2]
  673. ev1 = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  674. if ev1 is None:
  675. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  676. dev[1].group_form_result(ev1)
  677. ev1 = dev[1].group_request("WPS_PIN any " + pin)
  678. if ev1 is None:
  679. raise Exception("Failed to initiate Pin authorization on registrar side")
  680. if "join=" in adv_prov:
  681. if "OK" not in dev[0].global_request("P2P_CONNECT " + addr1 + " " + pin + " display persistent join"):
  682. raise Exception("P2P_CONNECT failed on advertiser side")
  683. ev0 = dev[0].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  684. if ev0 is None:
  685. raise Exception("Group formation failed to start on seeker side")
  686. dev[0].group_form_result(ev0)
  687. ev1 = dev[1].wait_global_event(["AP-STA-CONNECTED"], timeout=10)
  688. if ev1 is None:
  689. raise Exception("AP-STA-CONNECTED timeout on advertiser side")
  690. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  691. if ev0 is None:
  692. raise Exception("Unable to remove the advertisement instance")
  693. remove_group(dev[0], dev[1])
  694. def test_p2ps_service_discovery_multiple_queries(dev):
  695. """P2P service discovery with multiple queries"""
  696. addr0 = dev[0].p2p_dev_addr()
  697. addr1 = dev[1].p2p_dev_addr()
  698. adv_id1 = p2ps_advertise(r_dev=dev[0], r_role='0',
  699. svc_name='org.wi-fi.wfds.send.tx',
  700. srv_info='I can transfer files upto size of 2 GB')
  701. adv_id2 = p2ps_advertise(r_dev=dev[0], r_role='0',
  702. svc_name='org.wi-fi.wfds.send.rx',
  703. srv_info='I can receive files upto size of 2 GB')
  704. adv_id3 = p2ps_advertise(r_dev=dev[0], r_role='1',
  705. svc_name='org.wi-fi.wfds.display.tx',
  706. srv_info='Miracast Mode')
  707. adv_id4 = p2ps_advertise(r_dev=dev[0], r_role='1',
  708. svc_name='org.wi-fi.wfds.display.rx',
  709. srv_info='Miracast Mode')
  710. dev[1].global_request("P2P_SERV_DISC_REQ " + addr0 + " asp 1 org.wi-fi.wfds.display.tx 'Miracast Mode'")
  711. dev[1].global_request("P2P_FIND 10 type=social seek=org.wi-fi.wfds.display.tx")
  712. dev[1].global_request("P2P_SERV_DISC_REQ " + addr0 + " asp 2 org.wi-fi.wfds.send* 'size of 2 GB'")
  713. dev[1].p2p_stop_find()
  714. dev[1].global_request("P2P_FIND 10 type=social seek=")
  715. ev = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  716. if ev is None:
  717. raise Exception("P2P Device Found timed out")
  718. if addr0 not in ev:
  719. raise Exception("Unexpected service discovery request source")
  720. ev_list = []
  721. for i in range(0, 3):
  722. ev = dev[1].wait_global_event(["P2P-SERV-ASP-RESP"], timeout=10)
  723. if ev is None:
  724. raise Exception("P2P Service discovery timed out")
  725. if addr0 in ev:
  726. ev_list.append(ev)
  727. if len(ev_list) == 3:
  728. break
  729. dev[1].p2p_stop_find()
  730. for test in [ ("seek=org.wi-fi.wfds.display.TX",
  731. "asp_svc=org.wi-fi.wfds.display.tx"),
  732. ("seek=foo seek=org.wi-fi.wfds.display.tx seek=bar",
  733. "asp_svc=org.wi-fi.wfds.display.tx"),
  734. ("seek=1 seek=2 seek=3 seek=org.wi-fi.wfds.display.tx seek=4 seek=5 seek=6",
  735. "asp_svc=org.wi-fi.wfds.display.tx"),
  736. ("seek=not-found", None),
  737. ("seek=org.wi-fi.wfds", "asp_svc=org.wi-fi.wfds")]:
  738. dev[2].global_request("P2P_FIND 10 type=social " + test[0])
  739. if test[1] is None:
  740. ev = dev[2].wait_global_event(["P2P-DEVICE-FOUND"], timeout=1)
  741. if ev is not None:
  742. raise Exception("Unexpected device found: " + ev)
  743. continue
  744. ev = dev[2].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  745. if ev is None:
  746. raise Exception("P2P device discovery timed out (dev2)")
  747. if test[1] not in ev:
  748. raise Exception("Expected asp_svc not reported: " + ev)
  749. dev[2].p2p_stop_find()
  750. dev[2].request("P2P_FLUSH")
  751. dev[0].p2p_stop_find()
  752. ev1 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id1))
  753. if ev1 is None:
  754. raise Exception("Unable to remove the advertisement instance")
  755. ev2 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id2))
  756. if ev2 is None:
  757. raise Exception("Unable to remove the advertisement instance")
  758. ev3 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id3))
  759. if ev3 is None:
  760. raise Exception("Unable to remove the advertisement instance")
  761. ev4 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id4))
  762. if ev4 is None:
  763. raise Exception("Unable to remove the advertisement instance")
  764. if "OK" not in dev[0].global_request("P2P_SERVICE_ADD asp 1 12345678 1 1108 org.wi-fi.wfds.foobar svc_info='Test'"):
  765. raise Exception("P2P_SERVICE_ADD failed")
  766. if "OK" not in dev[0].global_request("P2P_SERVICE_DEL asp all"):
  767. raise Exception("P2P_SERVICE_DEL asp all failed")
  768. if "OK" not in dev[0].global_request("P2P_SERVICE_ADD asp 1 12345678 1 1108 org.wi-fi.wfds.foobar svc_info='Test'"):
  769. raise Exception("P2P_SERVICE_ADD failed")
  770. if "OK" not in dev[0].global_request("P2P_SERVICE_REP asp 1 12345678 1 1108 org.wi-fi.wfds.foobar svc_info='Test'"):
  771. raise Exception("P2P_SERVICE_REP failed")
  772. if "FAIL" not in dev[0].global_request("P2P_SERVICE_REP asp 1 12345678 1 1108 org.wi-fi.wfds.Foo svc_info='Test'"):
  773. raise Exception("Invalid P2P_SERVICE_REP accepted")
  774. if "OK" not in dev[0].global_request("P2P_SERVICE_ADD asp 1 a2345678 1 1108 org.wi-fi.wfds.something svc_info='Test'"):
  775. raise Exception("P2P_SERVICE_ADD failed")
  776. if "OK" not in dev[0].global_request("P2P_SERVICE_ADD asp 1 a2345679 1 1108 org.wi-fi.wfds.Foo svc_info='Test'"):
  777. raise Exception("P2P_SERVICE_ADD failed")
  778. def get_ifnames():
  779. with open('/proc/net/dev', 'r') as f:
  780. data = f.read()
  781. ifnames = []
  782. for line in data.splitlines():
  783. ifname = line.strip().split(' ')[0]
  784. if ':' not in ifname:
  785. continue
  786. ifname = ifname.split(':')[0]
  787. ifnames.append(ifname)
  788. return ifnames
  789. def p2ps_connect_p2ps_method(dev, keep_group=False):
  790. addr0 = dev[0].p2p_dev_addr()
  791. addr1 = dev[1].p2p_dev_addr()
  792. dev[0].flush_scan_cache()
  793. dev[1].flush_scan_cache()
  794. p2ps_advertise(r_dev=dev[0], r_role='2', svc_name='org.wi-fi.wfds.send.rx',
  795. srv_info='I can receive files upto size 2 GB')
  796. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  797. svc_name='org.wi-fi.wfds.send.rx',
  798. srv_info='2 GB')
  799. if "OK" not in dev[1].global_request("P2P_ASP_PROVISION " + addr0 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " info='test-session-info-data' method=1000"):
  800. raise Exception("Failed to request provisioning on seeker")
  801. ev0 = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  802. if ev0 is None:
  803. raise Exception("P2PS-PROV-DONE timeout on advertiser side")
  804. if "join=" not in ev0:
  805. raise Exception("join parameter missing from P2PS-PROV-DONE")
  806. ev1 = dev[1].wait_global_event(["P2PS-PROV-DONE"], timeout=5)
  807. if ev1 is None:
  808. raise Exception("Provisioning failed on seeker side")
  809. ev1 = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=15)
  810. if ev1 is None:
  811. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  812. res1 = dev[1].group_form_result(ev1)
  813. ifnames = get_ifnames()
  814. if "OK" not in dev[0].global_request("P2P_CONNECT " + addr1 + " 12345670 p2ps persistent join"):
  815. raise Exception("P2P_CONNECT failed on seeker side")
  816. ev0 = dev[0].wait_global_event(["P2P-GROUP-STARTED"], timeout=15)
  817. if ev0 is None:
  818. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  819. res0 = dev[0].group_form_result(ev0)
  820. ev1 = dev[1].wait_global_event(["AP-STA-CONNECTED"], timeout=5)
  821. if ev1 is None:
  822. raise Exception("Group formation failed")
  823. if not keep_group:
  824. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  825. if ev0 is None:
  826. raise Exception("Unable to remove the advertisement instance")
  827. ifnames = ifnames + get_ifnames()
  828. remove_group(dev[0], dev[1])
  829. ifnames = ifnames + get_ifnames()
  830. return (res0, res1, ifnames)
  831. def has_string_prefix(vals, prefix):
  832. for val in vals:
  833. if val.startswith(prefix):
  834. return True
  835. return False
  836. def test_p2ps_connect_p2ps_method_1(dev):
  837. """P2PS connection with P2PS method - no group interface"""
  838. set_no_group_iface(dev[0], 1)
  839. set_no_group_iface(dev[1], 1)
  840. (res0, res1, ifnames) = p2ps_connect_p2ps_method(dev)
  841. if res0['ifname'] != dev[0].ifname:
  842. raise Exception("unexpected dev0 group ifname: " + res0['ifname'])
  843. if res1['ifname'] != dev[1].ifname:
  844. raise Exception("unexpected dev1 group ifname: " + res1['ifname'])
  845. if has_string_prefix(ifnames, 'p2p-' + res0['ifname']):
  846. raise Exception("dev0 group interface unexpectedly present")
  847. if has_string_prefix(ifnames, 'p2p-' + res1['ifname']):
  848. raise Exception("dev1 group interface unexpectedly present")
  849. def test_p2ps_connect_p2ps_method_2(dev):
  850. """P2PS connection with P2PS method - group interface on dev0"""
  851. set_no_group_iface(dev[0], 0)
  852. set_no_group_iface(dev[1], 1)
  853. (res0, res1, ifnames) = p2ps_connect_p2ps_method(dev)
  854. if not res0['ifname'].startswith('p2p-' + dev[0].ifname + '-'):
  855. raise Exception("unexpected dev0 group ifname: " + res0['ifname'])
  856. if res1['ifname'] != dev[1].ifname:
  857. raise Exception("unexpected dev1 group ifname: " + res1['ifname'])
  858. if has_string_prefix(ifnames, 'p2p-' + res0['ifname']):
  859. raise Exception("dev0 group interface unexpectedly present")
  860. def test_p2ps_connect_p2ps_method_3(dev):
  861. """P2PS connection with P2PS method - group interface on dev1"""
  862. set_no_group_iface(dev[0], 1)
  863. set_no_group_iface(dev[1], 0)
  864. (res0, res1, ifnames) = p2ps_connect_p2ps_method(dev)
  865. if res0['ifname'] != dev[0].ifname:
  866. raise Exception("unexpected dev0 group ifname: " + res0['ifname'])
  867. if not res1['ifname'].startswith('p2p-' + dev[1].ifname + '-'):
  868. raise Exception("unexpected dev1 group ifname: " + res1['ifname'])
  869. if has_string_prefix(ifnames, 'p2p-' + res0['ifname']):
  870. raise Exception("dev0 group interface unexpectedly present")
  871. def test_p2ps_connect_p2ps_method_4(dev):
  872. """P2PS connection with P2PS method - group interface on both"""
  873. set_no_group_iface(dev[0], 0)
  874. set_no_group_iface(dev[1], 0)
  875. (res0, res1, ifnames) = p2ps_connect_p2ps_method(dev)
  876. if not res0['ifname'].startswith('p2p-' + dev[0].ifname + '-'):
  877. raise Exception("unexpected dev0 group ifname: " + res0['ifname'])
  878. if not res1['ifname'].startswith('p2p-' + dev[1].ifname + '-'):
  879. raise Exception("unexpected dev1 group ifname: " + res1['ifname'])
  880. def test_p2ps_connect_adv_go_persistent(dev):
  881. """P2PS auto-accept connection with advertisement as GO and having persistent group"""
  882. addr0 = dev[0].p2p_dev_addr()
  883. addr1 = dev[1].p2p_dev_addr()
  884. go_neg_pin_authorized_persistent(i_dev=dev[0], i_intent=15,
  885. r_dev=dev[1], r_intent=0)
  886. dev[0].remove_group()
  887. dev[1].wait_go_ending_session()
  888. p2ps_advertise(r_dev=dev[0], r_role='4', svc_name='org.wi-fi.wfds.send.rx',
  889. srv_info='I can receive files upto size 2 GB')
  890. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  891. svc_name='org.wi-fi.wfds.send.rx',
  892. srv_info='2 GB')
  893. if "OK" not in dev[1].global_request("P2P_ASP_PROVISION " + addr0 + " adv_id=" + str(adv_id) + " adv_mac=" + addr0 + " session=1 session_mac=" + addr1 + " info='' method=1000"):
  894. raise Exception("Failed to request provisioning on seeker")
  895. ev0 = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  896. if ev0 is None:
  897. raise Exception("Timed out while waiting for prov done on advertizer")
  898. if "persist=" not in ev0:
  899. raise Exception("Advertiser did not indicate persistent group")
  900. id0 = ev0.split("persist=")[1].split(" ")[0]
  901. if "OK" not in dev[0].global_request("P2P_GROUP_ADD persistent=" + id0 + " freq=2412"):
  902. raise Exception("Could not re-start persistent group")
  903. ev0 = dev[0].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  904. if ev0 is None:
  905. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  906. dev[0].group_form_result(ev0)
  907. ev1 = dev[1].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  908. if ev1 is None:
  909. raise Exception("P2PS-PROV-DONE timeout on seeker side")
  910. if "persist=" not in ev1:
  911. raise Exception("Seeker did not indicate persistent group")
  912. id1 = ev1.split("persist=")[1].split(" ")[0]
  913. if "OK" not in dev[1].global_request("P2P_GROUP_ADD persistent=" + id1 + " freq=2412"):
  914. raise Exception("Could not re-start persistent group")
  915. ev1 = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=15)
  916. if ev1 is None:
  917. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  918. dev[1].group_form_result(ev1)
  919. ev0 = dev[0].wait_global_event(["AP-STA-CONNECTED"], timeout=15)
  920. if ev0 is None:
  921. raise Exception("AP-STA-CONNECTED timeout on advertiser side")
  922. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  923. if ev0 is None:
  924. raise Exception("Unable to remove the advertisement instance")
  925. remove_group(dev[0], dev[1])
  926. def test_p2ps_client_probe(dev):
  927. """P2PS CLI discoverability on operating channel"""
  928. cli_probe = dev[0].global_request("SET p2p_cli_probe 1")
  929. p2ps_connect_p2ps_method(dev, keep_group=True)
  930. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[2], r_dev=dev[0],
  931. svc_name='org.wi-fi.wfds.send.rx',
  932. single_peer_expected=False)
  933. dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  934. remove_group(dev[0], dev[1])
  935. def test_p2ps_go_probe(dev):
  936. """P2PS GO discoverability on operating channel"""
  937. p2ps_connect_adv_go_pin_method(dev, keep_group=True)
  938. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[2], r_dev=dev[0],
  939. svc_name='org.wi-fi.wfds.send.rx',
  940. single_peer_expected=False)
  941. dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  942. remove_group(dev[0], dev[1])
  943. def test_p2ps_wildcard_p2ps(dev):
  944. """P2PS wildcard SD Probe Request/Response"""
  945. p2ps_wildcard = "org.wi-fi.wfds"
  946. adv_id = p2ps_advertise(r_dev=dev[0], r_role='1',
  947. svc_name='org.foo.service',
  948. srv_info='I can do stuff')
  949. adv_id2 = p2ps_advertise(r_dev=dev[0], r_role='1',
  950. svc_name='org.wi-fi.wfds.send.rx',
  951. srv_info='I can receive files upto size 2 GB')
  952. if "OK" not in dev[1].global_request("P2P_FIND 10 type=social seek=org.foo.service seek=" + p2ps_wildcard):
  953. raise Exception("Failed on P2P_FIND command")
  954. ev1 = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  955. if ev1 is None:
  956. raise Exception("P2P-DEVICE-FOUND timeout on seeker side")
  957. if dev[0].p2p_dev_addr() not in ev1:
  958. raise Exception("Unexpected peer")
  959. ev2 = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  960. if ev2 is None:
  961. raise Exception("P2P-DEVICE-FOUND timeout on seeker side (2)")
  962. if dev[0].p2p_dev_addr() not in ev2:
  963. raise Exception("Unexpected peer (2)")
  964. if p2ps_wildcard not in ev1 + ev2:
  965. raise Exception("P2PS Wildcard name not found in P2P-DEVICE-FOUND event")
  966. if "org.foo.service" not in ev1 + ev2:
  967. raise Exception("Vendor specific service name not found in P2P-DEVICE-FOUND event")
  968. if "OK" not in dev[1].global_request("P2P_STOP_FIND"):
  969. raise Exception("P2P_STOP_FIND failed")
  970. dev[1].dump_monitor()
  971. res = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  972. if res is None:
  973. raise Exception("Unable to remove the advertisement instance")
  974. if "OK" not in dev[1].global_request("P2P_FIND 10 type=social seek=" + p2ps_wildcard):
  975. raise Exception("Failed on P2P_FIND command")
  976. ev1 = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  977. if ev1 is None:
  978. raise Exception("P2P-DEVICE-FOUND timeout on seeker side")
  979. if dev[0].p2p_dev_addr() not in ev1:
  980. raise Exception("Unexpected peer")
  981. if p2ps_wildcard not in ev1:
  982. raise Exception("P2PS Wildcard name not found in P2P-DEVICE-FOUND event (2)")
  983. dev[1].dump_monitor()
  984. res = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id2))
  985. if res is None:
  986. raise Exception("Unable to remove the advertisement instance 2")
  987. if "OK" not in dev[1].global_request("P2P_FIND 10 type=social seek=" + p2ps_wildcard):
  988. raise Exception("Failed on P2P_FIND command")
  989. ev1 = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=2)
  990. if ev1 is not None:
  991. raise Exception("Unexpected P2P-DEVICE-FOUND event on seeker side")
  992. dev[1].p2p_stop_find()
  993. dev[1].dump_monitor()
  994. def test_p2ps_many_services_in_probe(dev):
  995. """P2PS with large number of services in Probe Request/Response"""
  996. long1 = 'org.example.0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789.a'
  997. long2 = 'org.example.0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789.b'
  998. long3 = 'org.example.0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789.c'
  999. long4 = 'org.example.0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789.d'
  1000. long5 = 'org.example.0123456789012345678901234567890123456789012345678901234567890123456789012345678901234567890123456789.e'
  1001. for name in [ long1, long2, long3, long4, long5 ]:
  1002. p2ps_advertise(r_dev=dev[0], r_role='1',
  1003. svc_name=name,
  1004. srv_info='I can do stuff')
  1005. 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)):
  1006. raise Exception("Failed on P2P_FIND command")
  1007. events = ""
  1008. # Note: Require only four events since all the services do not fit within
  1009. # the length limit.
  1010. for i in range(4):
  1011. ev = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  1012. if ev is None:
  1013. raise Exception("Missing P2P-DEVICE-FOUND")
  1014. events = events + ev
  1015. dev[1].p2p_stop_find()
  1016. dev[1].dump_monitor()
  1017. for name in [ long2, long3, long4, long5 ]:
  1018. if name not in events:
  1019. raise Exception("Service missing from peer events")