test_p2ps.py 46 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916
  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 hwsim_utils
  11. from wpasupplicant import WpaSupplicant
  12. from test_p2p_grpform import check_grpform_results
  13. from test_p2p_grpform import remove_group
  14. from test_p2p_persistent import go_neg_pin_authorized_persistent
  15. # Dev[0] -> Advertiser
  16. # Dev[1] -> Seeker
  17. # ev0 -> Event generated at advertiser side
  18. # ev1 -> Event generated at Seeker side
  19. def p2ps_advertise(r_dev, r_role, svc_name, srv_info, rsp_info=None):
  20. """P2PS Advertise function"""
  21. adv_id = random.randrange(1, 0xFFFFFFFF)
  22. advid = hex(adv_id)[2:]
  23. if rsp_info is not None and srv_info is not None:
  24. 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 + "'"):
  25. raise Exception("P2P_SERVICE_ADD with response info and service info failed")
  26. if rsp_info is 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 + " svc_info='" + srv_info + "'"):
  28. raise Exception("P2P_SERVICE_ADD with service info failed")
  29. if rsp_info is None and srv_info is None:
  30. if "OK" not in r_dev.global_request("P2P_SERVICE_ADD asp " + str(r_role) + " " + str(advid) + " 1 1108 " + svc_name + " "):
  31. raise Exception("P2P_SERVICE_ADD without service info and without response info failed")
  32. if rsp_info is not None and srv_info is None:
  33. 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 + "'"):
  34. raise Exception("P2P_SERVICE_ADD with response info failed")
  35. r_dev.p2p_listen()
  36. return advid
  37. def p2ps_exact_seek(i_dev, r_dev, svc_name, srv_info=None):
  38. """P2PS exact service seek request"""
  39. if srv_info is not None:
  40. ev1 = i_dev.global_request("P2P_SERV_DISC_REQ 00:00:00:00:00:00 asp 1 " + svc_name + " '" + srv_info + "'")
  41. if ev1 is None:
  42. raise Exception("Failed to add Service Discovery request for exact seek request")
  43. if "OK" not in i_dev.global_request("P2P_FIND 10 type=social seek=" + svc_name):
  44. raise Exception("Failed to initiate seek operation")
  45. ev1 = i_dev.wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  46. if ev1 is None:
  47. raise Exception("P2P-DEVICE-FOUND timeout on seeker side")
  48. if r_dev.p2p_dev_addr() not in ev1:
  49. raise Exception("Unexpected peer")
  50. if srv_info is None:
  51. adv_id = ev1.split("adv_id=")[1].split(" ")[0]
  52. rcvd_svc_name = ev1.split("asp_svc=")[1].split(" ")[0]
  53. if rcvd_svc_name != svc_name:
  54. raise Exception("service name not matching")
  55. else:
  56. ev1 = i_dev.wait_global_event(["P2P-SERV-ASP-RESP"], timeout=10)
  57. if ev1 is None:
  58. raise Exception("Failed to receive Service Discovery Response")
  59. if r_dev.p2p_dev_addr() not in ev1:
  60. raise Exception("Service Discovery response from Unknown Peer")
  61. if srv_info is not None and srv_info not in ev1:
  62. raise Exception("service info not available in Service Discovery response")
  63. adv_id = ev1.split(" ")[3]
  64. rcvd_svc_name = ev1.split(" ")[6]
  65. if rcvd_svc_name != svc_name:
  66. raise Exception("service name not matching")
  67. return [adv_id, rcvd_svc_name]
  68. def p2ps_nonexact_seek(i_dev, r_dev, svc_name, srv_info=None, adv_num=None):
  69. """P2PS nonexact service seek request"""
  70. if adv_num is None:
  71. adv_num = 1
  72. if srv_info is not None:
  73. ev1 = i_dev.global_request("P2P_SERV_DISC_REQ 00:00:00:00:00:00 asp 1 " + svc_name + " '" + srv_info + "'")
  74. else:
  75. ev1 = i_dev.global_request("P2P_SERV_DISC_REQ 00:00:00:00:00:00 asp 1 " + svc_name + " '")
  76. if ev1 is None:
  77. raise Exception("Failed to add Service Discovery request for nonexact seek request")
  78. if "OK" not in i_dev.global_request("P2P_FIND 10 type=social seek="):
  79. raise Exception("Failed to initiate seek")
  80. ev1 = i_dev.wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  81. if ev1 is None:
  82. raise Exception("P2P-DEVICE-FOUND timeout on seeker side")
  83. if r_dev.p2p_dev_addr() not in ev1:
  84. raise Exception("Unexpected peer")
  85. ev_list = []
  86. for i in range (0, adv_num):
  87. ev1 = i_dev.wait_global_event(["P2P-SERV-ASP-RESP"], timeout=10)
  88. if ev1 is None:
  89. raise Exception("Failed to receive Service Discovery Response")
  90. if r_dev.p2p_dev_addr() not in ev1:
  91. raise Exception("Service Discovery response from Unknown Peer")
  92. if srv_info is not None and srv_info not in ev1:
  93. raise Exception("service info not available in Service Discovery response")
  94. adv_id = ev1.split(" ")[3]
  95. rcvd_svc_name = ev1.split(" ")[6]
  96. ev_list.append(''.join([adv_id, ' ', rcvd_svc_name]))
  97. return ev_list
  98. def p2p_connect_p2ps_method(i_dev, r_dev, autoaccept):
  99. """P2PS connect function with p2ps method"""
  100. if autoaccept == False:
  101. if "OK" not in i_dev.global_request("P2P_CONNECT " + r_dev.p2p_dev_addr() + " 12345670 p2ps persistent auth"):
  102. raise Exception("P2P_CONNECT fails on seeker side")
  103. ev0 = r_dev.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  104. if ev0 is None:
  105. raise Exception("P2PS-PROV-DONE timeout on Advertiser side")
  106. if "OK" not in r_dev.global_request("P2P_CONNECT " + i_dev.p2p_dev_addr() + " 12345670 p2ps persistent"):
  107. raise Exception("P2P_CONNECT fails on Advertiser side")
  108. else:
  109. if "OK" not in r_dev.global_request("P2P_CONNECT " + i_dev.p2p_dev_addr() + " 12345670 p2ps persistent auth"):
  110. raise Exception("P2P_CONNECT fails on Advertiser side")
  111. ev1 = i_dev.wait_global_event(["P2PS-PROV-DONE"], timeout=5)
  112. if ev1 is None:
  113. raise Exception("Failed to receive deferred acceptance at seeker")
  114. if "OK" not in i_dev.global_request("P2P_CONNECT " + r_dev.p2p_dev_addr() + " 12345670 p2ps persistent"):
  115. raise Exception("P2P_CONNECT fails on seeker side")
  116. ev0 = r_dev.wait_global_event(["P2P-GO-NEG-SUCCESS"], timeout=10)
  117. if ev0 is None:
  118. raise Exception("GO Neg did not succeed")
  119. ev0 = r_dev.wait_global_event(["P2P-GROUP-STARTED"], timeout=15)
  120. if ev0 is None:
  121. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  122. ev1 = i_dev.wait_global_event(["P2P-GROUP-STARTED"], timeout=5)
  123. if ev1 is None:
  124. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  125. def p2ps_provision_keypad_method(i_dev, r_dev, autoaccept,
  126. initiator_or_responder):
  127. """P2PS keypad method provisioning function"""
  128. if autoaccept == False and initiator_or_responder == 'initiator':
  129. ev = i_dev.wait_global_event(["P2P-PROV-DISC-FAILURE"], timeout=10)
  130. if ev is None:
  131. raise Exception("Provisioning deferred on seeker side")
  132. ev1 = i_dev.wait_global_event(["P2P-PROV-DISC-ENTER-PIN"], timeout=10)
  133. if ev1 is None:
  134. raise Exception("P2P-PROV-DISC-ENTER-PIN timeout on seeker side")
  135. if r_dev.p2p_dev_addr() not in ev1:
  136. raise Exception("Unknown peer ")
  137. ev = r_dev.wait_global_event(["P2PS-PROV-START"], timeout=10)
  138. if ev is None:
  139. raise Exception("P2PS-PROV-START timeout on Advertiser side")
  140. if autoaccept == False and initiator_or_responder == 'responder':
  141. ev0 = r_dev.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  142. if ev0 is None:
  143. raise Exception("P2PS-PROV-DONE timeout on seeker side")
  144. ev0 = r_dev.wait_global_event(["P2P-PROV-DISC-ENTER-PIN"], timeout=5)
  145. if ev0 is None:
  146. raise Exception("P2P-PROV-DISC-ENTER-PIN timeout on advertiser side")
  147. if autoaccept == True and initiator_or_responder == 'initiator':
  148. ev1 = i_dev.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  149. if ev1 is None:
  150. raise Exception("P2PS-PROV-DONE timeout on seeker side")
  151. ev1 = i_dev.wait_global_event(["P2P-PROV-DISC-ENTER-PIN"], timeout=10)
  152. if ev1 is None:
  153. raise Exception("P2P-PROV-DISC-ENTER-PIN failed on seeker side")
  154. if r_dev.p2p_dev_addr() not in ev1:
  155. raise Exception("Unknown peer ")
  156. return ev1
  157. def p2ps_provision_display_method(i_dev, r_dev, autoaccept,
  158. initiator_or_responder):
  159. """P2PS display method provisioning function"""
  160. if initiator_or_responder == 'initiator':
  161. ev0 = r_dev.wait_global_event(["P2PS-PROV-START"], timeout=10)
  162. if ev0 is None:
  163. raise Exception("P2PS-PROV-START timeout on Advertiser side")
  164. if autoaccept == False:
  165. ev = i_dev.wait_global_event(["P2P-PROV-DISC-FAILURE"], timeout=10)
  166. if ev is None:
  167. raise Exception("Provisioning deferred on seeker side")
  168. ev1 = i_dev.wait_global_event(["P2P-PROV-DISC-SHOW-PIN"], timeout=10)
  169. if ev1 is None:
  170. raise Exception("P2P-PROV-DISC-SHOW-PIN timeout on Seeker side")
  171. if r_dev.p2p_dev_addr() not in ev1:
  172. raise Exception("Unknown peer ")
  173. pin = ev1.split(" ")[2]
  174. else:
  175. ev0 = r_dev.wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  176. if ev0 is None:
  177. raise Exception("P2PS-PROV-DONE timeout on advertiser")
  178. ev0 = r_dev.wait_global_event(["P2P-PROV-DISC-SHOW-PIN"], timeout=5)
  179. if ev0 is None:
  180. raise Exception("PIN Display on advertiser side")
  181. pin = ev0.split(" ")[2]
  182. return pin
  183. def p2ps_connect_pin(pin, i_dev, r_dev, initiator_method):
  184. """P2PS function to perform connection using PIN method"""
  185. if initiator_method=="display":
  186. if "OK" not in i_dev.global_request("P2P_CONNECT " + r_dev.p2p_dev_addr() + " " + pin + " display persistent "):
  187. raise Exception("P2P_CONNECT fails on seeker side")
  188. ev0 = r_dev.wait_global_event(["P2P-GO-NEG-REQUEST"], timeout=10)
  189. if ev0 is None:
  190. raise Exception("Failed to receive P2P_GO-NEG-REQUEST on responder side")
  191. if "OK" not in r_dev.global_request("P2P_CONNECT " + i_dev.p2p_dev_addr() + " " + pin + " keypad persistent "):
  192. raise Exception("P2P_CONNECT fails on Advertiser side")
  193. else:
  194. if "OK" not in i_dev.global_request("P2P_CONNECT " + r_dev.p2p_dev_addr() + " " + pin + " keypad persistent "):
  195. raise Exception("P2P_CONNECT fails on seeker side")
  196. ev0 = r_dev.wait_global_event(["P2P-GO-NEG-REQUEST"], timeout=10)
  197. if ev0 is None:
  198. raise Exception("Failed to receive P2P_GO-NEG-REQUEST on responder side")
  199. if "OK" not in r_dev.global_request("P2P_CONNECT " + i_dev.p2p_dev_addr() + " " + pin + " display persistent "):
  200. raise Exception("P2P_CONNECT fails on Advertiser side")
  201. ev0 = r_dev.wait_global_event(["P2P-GO-NEG-SUCCESS"], timeout=10)
  202. if ev0 is None:
  203. raise Exception("GO Neg did not succeed on advertiser side")
  204. peer_mac = ev0.split("peer_dev=")[1].split(" ")[0]
  205. ev1 = i_dev.wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  206. if ev1 is None:
  207. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  208. ev_grpfrm = r_dev.wait_global_event(["P2P-GROUP-FORMATION-SUCCESS"],
  209. timeout=10)
  210. if ev_grpfrm is None:
  211. raise Exception("Group Formation failed on advertiser side")
  212. ev = r_dev.wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  213. if ev is None:
  214. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  215. Role = ev.split(" ")[2]
  216. if Role == "GO":
  217. ev_grpfrm = r_dev.wait_global_event(["AP-STA-CONNECTED"], timeout=10)
  218. if ev_grpfrm is None:
  219. raise Exception("AP-STA-CONNECTED timeout on advertiser side")
  220. if i_dev.p2p_dev_addr() not in ev_grpfrm:
  221. raise Exception("Group formed with unknown Peer")
  222. else:
  223. ev1 = i_dev.wait_global_event(["AP-STA-CONNECTED"], timeout=5)
  224. if ev1 is None:
  225. raise Exception("AP-STA-CONNECTED timeout on Seeker side")
  226. if r_dev.p2p_dev_addr() not in ev1:
  227. raise Exception("Group formed with unknown Peer")
  228. def test_p2ps_exact_search(dev):
  229. """P2PS exact service request"""
  230. addr0 = dev[0].p2p_dev_addr()
  231. addr1 = dev[1].p2p_dev_addr()
  232. p2ps_advertise(r_dev=dev[0], r_role='1', svc_name='org.wi-fi.wfds.send.rx',
  233. srv_info='I can receive files upto size 2 GB')
  234. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  235. svc_name='org.wi-fi.wfds.send.rx')
  236. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  237. if ev0 is None:
  238. raise Exception("Unable to remove the advertisement instance")
  239. def test_p2ps_exact_search_srvinfo(dev):
  240. """P2PS exact service request with service info"""
  241. addr0 = dev[0].p2p_dev_addr()
  242. addr1 = dev[1].p2p_dev_addr()
  243. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  244. srv_info='I can receive files upto size 2 GB')
  245. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  246. svc_name='org.wi-fi.wfds.send.rx',
  247. srv_info='2 GB')
  248. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  249. if ev0 is None:
  250. raise Exception("Unable to remove the advertisement instance")
  251. def test_p2ps_nonexact_search(dev):
  252. """P2PS nonexact seek request"""
  253. addr0 = dev[0].p2p_dev_addr()
  254. addr1 = dev[1].p2p_dev_addr()
  255. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.play.rx',
  256. srv_info='I support Miracast Mode ')
  257. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  258. svc_name='org.wi-fi.wfds.play*')
  259. adv_id = ev_list[0].split(" ")[0]
  260. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  261. if ev0 is None:
  262. raise Exception("Unable to remove the advertisement instance")
  263. def test_p2ps_nonexact_search_srvinfo(dev):
  264. """P2PS nonexact seek request with service info"""
  265. addr0 = dev[0].p2p_dev_addr()
  266. addr1 = dev[1].p2p_dev_addr()
  267. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  268. srv_info='I can receive files upto size 2 GB')
  269. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  270. svc_name='org.wi-fi.wfds.send*',
  271. srv_info='2 GB')
  272. adv_id = ev_list[0].split(" ")[0]
  273. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  274. if ev0 is None:
  275. raise Exception("Unable to remove the advertisement instance")
  276. def test_p2ps_connect_p2ps_method_nonautoaccept(dev):
  277. """P2PS connect for non-auto-accept and P2PS config method"""
  278. addr0 = dev[0].p2p_dev_addr()
  279. addr1 = dev[1].p2p_dev_addr()
  280. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  281. srv_info='I can receive files upto size 2 GB')
  282. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  283. svc_name='org.wi-fi.wfds.send*',
  284. srv_info='2 GB')
  285. adv_id = ev_list[0].split(" ")[0]
  286. 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"):
  287. raise Exception("Failed to request provisioning on seeker")
  288. ev0 = dev[0].wait_global_event(["P2PS-PROV-START"], timeout=10)
  289. if ev0 is None:
  290. raise Exception("P2PS-PROV-START timeout on advertiser side")
  291. ev1 = dev[1].wait_global_event(["P2P-PROV-DISC-FAILURE"], timeout=15)
  292. if ev1 is None:
  293. raise Exception("Provisioning deferred timeout on seeker side")
  294. dev[1].p2p_listen()
  295. 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"):
  296. raise Exception("Failed to send deferred acceptance from advertizer")
  297. ev1 = dev[1].wait_global_event(["P2PS-PROV-DONE"], timeout=15)
  298. if ev1 is None:
  299. raise Exception("Failed to receive deferred acceptance at seeker")
  300. p2p_connect_p2ps_method(i_dev=dev[1], r_dev=dev[0], autoaccept=False)
  301. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  302. if ev0 is None:
  303. raise Exception("Unable to remove the advertisement instance")
  304. remove_group(dev[0], dev[1])
  305. def test_p2ps_connect_p2ps_method_autoaccept(dev):
  306. """P2PS connection with P2PS default config method and auto-accept"""
  307. addr0 = dev[0].p2p_dev_addr()
  308. addr1 = dev[1].p2p_dev_addr()
  309. p2ps_advertise(r_dev=dev[0], r_role='1', svc_name='org.wi-fi.wfds.send.rx',
  310. srv_info='I can receive files upto size 2 GB')
  311. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  312. svc_name='org.wi-fi.wfds.send.rx',
  313. srv_info='2 GB')
  314. 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"):
  315. raise Exception("P2P_ASP_PROVISION failed on seeker side")
  316. ev0 = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  317. if ev0 is None:
  318. raise Exception("P2PS-PROV-DONE timeout on advertiser side")
  319. p2p_connect_p2ps_method(i_dev=dev[1], r_dev=dev[0], autoaccept=True)
  320. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  321. if ev0 is None:
  322. raise Exception("Unable to remove the advertisement instance")
  323. remove_group(dev[0], dev[1])
  324. def test_p2ps_connect_keypad_method_nonautoaccept(dev):
  325. """P2PS Connection with non-auto-accept and seeker having keypad method"""
  326. addr0 = dev[0].p2p_dev_addr()
  327. addr1 = dev[1].p2p_dev_addr()
  328. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  329. srv_info='I can receive files upto size 2 GB')
  330. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  331. svc_name='org.wi-fi.wfds.send*',
  332. srv_info='2 GB')
  333. adv_id = ev_list[0].split(" ")[0]
  334. 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
  335. raise Exception("Failed to request provisioning on seeker")
  336. p2ps_provision_keypad_method(i_dev=dev[1], r_dev=dev[0], autoaccept=False,
  337. initiator_or_responder='initiator')
  338. dev[1].p2p_listen()
  339. 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"):
  340. raise Exception("Failed to send deferred acceptance from advertizer")
  341. pin = p2ps_provision_display_method(i_dev=dev[1], r_dev=dev[0],
  342. autoaccept=False,
  343. initiator_or_responder='responder')
  344. p2ps_connect_pin(pin, i_dev=dev[0], r_dev=dev[1],
  345. initiator_method="display")
  346. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  347. if ev0 is None:
  348. raise Exception("Unable to remove the advertisement instance")
  349. remove_group(dev[0], dev[1])
  350. def test_p2ps_connect_display_method_nonautoaccept(dev):
  351. """P2PS connection with non-auto-accept and seeker having display method"""
  352. addr0 = dev[0].p2p_dev_addr()
  353. addr1 = dev[1].p2p_dev_addr()
  354. p2ps_advertise(r_dev=dev[0], r_role='0', svc_name='org.wi-fi.wfds.send.rx',
  355. srv_info='I can receive files upto size 2 GB')
  356. ev_list = p2ps_nonexact_seek(i_dev=dev[1], r_dev=dev[0],
  357. svc_name='org.wi-fi.wfds*', srv_info='2 GB')
  358. adv_id = ev_list[0].split(" ")[0]
  359. 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
  360. raise Exception("Failed to request provisioning on seeker")
  361. pin = p2ps_provision_display_method(i_dev=dev[1], r_dev=dev[0],
  362. autoaccept=False,
  363. initiator_or_responder='initiator')
  364. dev[1].p2p_listen()
  365. 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"):
  366. raise Exception("Failed to send deferred acceptance from advertiser")
  367. p2ps_provision_keypad_method(i_dev=dev[1], r_dev=dev[0], autoaccept=False,
  368. initiator_or_responder='responder')
  369. p2ps_connect_pin(pin, i_dev=dev[0], r_dev=dev[1], initiator_method="keypad")
  370. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  371. if ev0 is None:
  372. raise Exception("Unable to remove the advertisement instance")
  373. remove_group(dev[0], dev[1])
  374. def test_p2ps_connect_keypad_method_autoaccept(dev):
  375. """P2PS connection with auto-accept and keypad method on seeker side"""
  376. addr0 = dev[0].p2p_dev_addr()
  377. addr1 = dev[1].p2p_dev_addr()
  378. p2ps_advertise(r_dev=dev[0], r_role='1', svc_name='org.wi-fi.wfds.send.rx',
  379. srv_info='I can receive files upto size 2 GB')
  380. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  381. svc_name='org.wi-fi.wfds.send.rx',
  382. srv_info='2 GB')
  383. 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
  384. raise Exception("Failed to request provisioning on seeker")
  385. p2ps_provision_keypad_method(i_dev=dev[1], r_dev=dev[0], autoaccept=True,
  386. initiator_or_responder='initiator')
  387. pin = p2ps_provision_display_method(i_dev=dev[1], r_dev=dev[0],
  388. autoaccept=True,
  389. initiator_or_responder='responder')
  390. p2ps_connect_pin(pin, i_dev=dev[1], r_dev=dev[0], initiator_method="Keypad")
  391. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  392. if ev0 is None:
  393. raise Exception("Unable to remove the advertisement instance")
  394. remove_group(dev[0], dev[1])
  395. def test_p2ps_connect_display_method_autoaccept(dev):
  396. """P2PS connection with auto-accept and display method on seeker side"""
  397. addr0 = dev[0].p2p_dev_addr()
  398. addr1 = dev[1].p2p_dev_addr()
  399. p2ps_advertise(r_dev=dev[0], r_role='1', svc_name='org.wi-fi.wfds.send.rx',
  400. srv_info='I can receive files upto size 2 GB')
  401. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  402. svc_name='org.wi-fi.wfds.send.rx',
  403. srv_info='2 GB')
  404. 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
  405. raise Exception("Failed to request provisioning on seeker")
  406. pin = p2ps_provision_display_method(i_dev=dev[1], r_dev=dev[0],
  407. autoaccept=True,
  408. initiator_or_responder='initiator')
  409. dev[1].p2p_listen()
  410. p2ps_connect_pin(pin, i_dev=dev[1], r_dev=dev[0], initiator_method="display")
  411. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  412. if ev0 is None:
  413. raise Exception("Unable to remove the advertisement instance")
  414. remove_group(dev[0], dev[1])
  415. def test_p2ps_connect_adv_go_p2ps_method(dev):
  416. """P2PS auto-accept connection with advertisement as GO and P2PS method"""
  417. addr0 = dev[0].p2p_dev_addr()
  418. addr1 = dev[1].p2p_dev_addr()
  419. p2ps_advertise(r_dev=dev[0], r_role='4', svc_name='org.wi-fi.wfds.send.rx',
  420. srv_info='I can receive files upto size 2 GB')
  421. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  422. svc_name='org.wi-fi.wfds.send.rx',
  423. srv_info='2 GB')
  424. 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"):
  425. raise Exception("Failed to request provisioning on seeker")
  426. ev0 = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  427. if ev0 is None:
  428. raise Exception("Timed out while waiting for prov done on advertizer")
  429. if "go=" not in ev0:
  430. raise Exception("Advertiser failed to become GO")
  431. adv_conncap = ev0.split("conncap=")[1].split(" ")[0]
  432. if adv_conncap == "4":
  433. logger.info("Advertiser is GO")
  434. ev0 = dev[0].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  435. if ev0 is None:
  436. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  437. ev1 = dev[1].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  438. if ev1 is None:
  439. raise Exception("P2PS-PROV-DONE timeout on seeker side")
  440. seeker_conncap = ev1.split("conncap=")[1].split(" ")[0]
  441. if "join=" in ev1:
  442. if "OK" not in dev[1].global_request("P2P_CONNECT " + addr0 + " 12345670 p2ps persistent join"):
  443. raise Exception("P2P_CONNECT failed on seeker side")
  444. ev1 = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=15)
  445. if ev1 is None:
  446. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  447. ev0 = dev[0].wait_global_event(["AP-STA-CONNECTED"], timeout=5)
  448. if ev0 is None:
  449. raise Exception("AP-STA-CONNECTED timeout on advertiser side")
  450. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  451. if ev0 is None:
  452. raise Exception("Unable to remove the advertisement instance")
  453. remove_group(dev[0], dev[1])
  454. def test_p2ps_connect_adv_client_p2ps_method(dev):
  455. """P2PS auto-accept connection with advertisement as Client and P2PS method"""
  456. addr0 = dev[0].p2p_dev_addr()
  457. addr1 = dev[1].p2p_dev_addr()
  458. p2ps_advertise(r_dev=dev[0], r_role='2', svc_name='org.wi-fi.wfds.send.rx',
  459. srv_info='I can receive files upto size 2 GB')
  460. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  461. svc_name='org.wi-fi.wfds.send.rx',
  462. srv_info='2 GB')
  463. 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"):
  464. raise Exception("Failed to request provisioning on seeker")
  465. ev0 = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  466. if ev0 is None:
  467. raise Exception("P2PS-PROV-DONE timeout on advertiser side")
  468. if "join=" not in ev0:
  469. raise Exception("Advertiser failed to become Client")
  470. adv_conncap = ev0.split("conncap=")[1].split(" ")[0]
  471. if adv_conncap == "2":
  472. logger.info("Advertiser is Client")
  473. ev1 = dev[1].wait_global_event(["P2PS-PROV-DONE"], timeout=5)
  474. if ev1 is None:
  475. raise Exception("Provisioning failed on seeker side")
  476. seeker_conncap = ev1.split("conncap=")[1].split(" ")[0]
  477. ev1 = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  478. if ev1 is None:
  479. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  480. if "join=" in ev0:
  481. if "OK" not in dev[0].global_request("P2P_CONNECT " + addr1 + " 12345670 p2ps persistent join"):
  482. raise Exception("P2P_CONNECT failed on seeker side")
  483. ev0 = dev[0].wait_global_event(["P2P-GROUP-FORMATION-SUCCESS"], timeout=15)
  484. if ev0 is None:
  485. raise Exception("P2P Group Formation failed on advertiser side")
  486. ev0 = dev[0].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  487. if ev0 is None:
  488. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  489. ev1 = dev[1].wait_global_event(["AP-STA-CONNECTED"], timeout=5)
  490. if ev1 is None:
  491. raise Exception("Group formation failed")
  492. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  493. if ev0 is None:
  494. raise Exception("Unable to remove the advertisement instance")
  495. remove_group(dev[0], dev[1])
  496. def test_p2ps_connect_adv_go_pin_method(dev):
  497. """P2PS advertiser as GO with keypad config method on seeker side and auto-accept"""
  498. addr0 = dev[0].p2p_dev_addr()
  499. addr1 = dev[1].p2p_dev_addr()
  500. p2ps_advertise(r_dev=dev[0], r_role='4', svc_name='org.wi-fi.wfds.send.rx',
  501. srv_info='I can receive files upto size 2 GB')
  502. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  503. svc_name='org.wi-fi.wfds.send.rx',
  504. srv_info='2 GB')
  505. 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
  506. raise Exception("Failed to request provisioning on seeker")
  507. seek_prov_ev = p2ps_provision_keypad_method(i_dev=dev[1], r_dev=dev[0],
  508. autoaccept=True,
  509. initiator_or_responder='initiator')
  510. ev0 = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  511. if ev0 is None:
  512. raise Exception("P2PS-PROV-DONE timeout on advertier side")
  513. adv_conncap = ev0.split("conncap=")[1].split(" ")[0]
  514. if adv_conncap == "4":
  515. logger.info("Advertiser is GO")
  516. ev0 = dev[0].wait_global_event(["P2P-PROV-DISC-SHOW-PIN"], timeout=5)
  517. if ev0 is None:
  518. raise Exception("PIN Display on advertiser side")
  519. pin = ev0.split(" ")[2]
  520. ev0 = dev[0].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  521. if ev0 is None:
  522. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  523. ev0 = dev[0].group_request("WPS_PIN any " + pin)
  524. if ev0 is None:
  525. raise Exception("Failed to initiate Pin authorization on registrar side")
  526. if "join=" in seek_prov_ev:
  527. if "OK" not in dev[1].global_request("P2P_CONNECT " + addr0 + " " + pin + " keypad persistent join"):
  528. raise Exception("P2P_CONNECT failed on seeker side")
  529. ev1 = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  530. if ev1 is None:
  531. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  532. ev0 = dev[0].wait_global_event(["AP-STA-CONNECTED"], timeout=10)
  533. if ev0 is None:
  534. raise Exception("AP-STA-CONNECTED timeout on advertiser side")
  535. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  536. if ev0 is None:
  537. raise Exception("Unable to remove the advertisement instance")
  538. remove_group(dev[0], dev[1])
  539. def test_p2ps_connect_adv_client_pin_method(dev):
  540. """P2PS advertiser as client with keypad config method on seeker side and auto-accept"""
  541. addr0 = dev[0].p2p_dev_addr()
  542. addr1 = dev[1].p2p_dev_addr()
  543. dev[0].flush_scan_cache()
  544. p2ps_advertise(r_dev=dev[0], r_role='2', svc_name='org.wi-fi.wfds.send.rx',
  545. srv_info='I can receive files upto size 2 GB')
  546. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  547. svc_name='org.wi-fi.wfds.send.rx',
  548. srv_info='2 GB')
  549. 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
  550. raise Exception("Failed to request provisioning on seeker")
  551. seek_prov_ev = p2ps_provision_keypad_method(i_dev=dev[1], r_dev=dev[0],
  552. autoaccept=True,
  553. initiator_or_responder='initiator')
  554. adv_prov = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  555. if adv_prov is None:
  556. raise Exception("Prov failed on advertiser")
  557. adv_conncap = adv_prov.split("conncap=")[1].split(" ")[0]
  558. if adv_conncap == "2":
  559. logger.info("Advertiser is Client")
  560. adv_pin_show_event = dev[0].wait_global_event(["P2P-PROV-DISC-SHOW-PIN"],
  561. timeout=5)
  562. if adv_pin_show_event is None:
  563. raise Exception("PIN Display on advertiser side")
  564. pin = adv_pin_show_event.split(" ")[2]
  565. ev1 = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  566. if ev1 is None:
  567. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  568. ev1 = dev[1].group_request("WPS_PIN any " + pin)
  569. if ev1 is None:
  570. raise Exception("Failed to initiate Pin authorization on registrar side")
  571. if "join=" in adv_prov:
  572. if "OK" not in dev[0].global_request("P2P_CONNECT " + addr1 + " " + pin + " display persistent join"):
  573. raise Exception("P2P_CONNECT failed on advertiser side")
  574. ev0 = dev[0].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  575. if ev0 is None:
  576. raise Exception("Group formation failed to start on seeker side")
  577. ev1 = dev[1].wait_global_event(["AP-STA-CONNECTED"], timeout=10)
  578. if ev1 is None:
  579. raise Exception("AP-STA-CONNECTED timeout on advertiser side")
  580. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  581. if ev0 is None:
  582. raise Exception("Unable to remove the advertisement instance")
  583. remove_group(dev[0], dev[1])
  584. def test_p2ps_service_discovery_multiple_queries(dev):
  585. """P2P service discovery with multiple queries"""
  586. addr0 = dev[0].p2p_dev_addr()
  587. addr1 = dev[1].p2p_dev_addr()
  588. adv_id1 = p2ps_advertise(r_dev=dev[0], r_role='0',
  589. svc_name='org.wi-fi.wfds.send.tx',
  590. srv_info='I can transfer files upto size of 2 GB')
  591. adv_id2 = p2ps_advertise(r_dev=dev[0], r_role='0',
  592. svc_name='org.wi-fi.wfds.send.rx',
  593. srv_info='I can receive files upto size of 2 GB')
  594. adv_id3 = p2ps_advertise(r_dev=dev[0], r_role='1',
  595. svc_name='org.wi-fi.wfds.display.tx',
  596. srv_info='Miracast Mode')
  597. adv_id4 = p2ps_advertise(r_dev=dev[0], r_role='1',
  598. svc_name='org.wi-fi.wfds.display.rx',
  599. srv_info='Miracast Mode')
  600. dev[1].global_request("P2P_SERV_DISC_REQ " + addr0 + " asp 1 org.wi-fi.wfds.display.tx 'Miracast Mode'")
  601. dev[1].global_request("P2P_FIND 10 type=social seek=org.wi-fi.wfds.display.tx")
  602. dev[1].global_request("P2P_SERV_DISC_REQ " + addr0 + " asp 2 org.wi-fi.wfds.send* 'size of 2 GB'")
  603. dev[1].p2p_stop_find()
  604. dev[1].global_request("P2P_FIND 10 type=social seek=")
  605. ev = dev[1].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  606. if ev is None:
  607. raise Exception("P2P Device Found timed out")
  608. if addr0 not in ev:
  609. raise Exception("Unexpected service discovery request source")
  610. ev_list = []
  611. for i in range(0, 3):
  612. ev = dev[1].wait_global_event(["P2P-SERV-ASP-RESP"], timeout=10)
  613. if ev is None:
  614. raise Exception("P2P Service discovery timed out")
  615. if addr0 in ev:
  616. ev_list.append(ev)
  617. if len(ev_list) == 3:
  618. break
  619. dev[1].p2p_stop_find()
  620. for test in [ ("seek=org.wi-fi.wfds.display.TX",
  621. "asp_svc=org.wi-fi.wfds.display.tx"),
  622. ("seek=foo seek=org.wi-fi.wfds.display.tx seek=bar",
  623. "asp_svc=org.wi-fi.wfds.display.tx"),
  624. ("seek=1 seek=2 seek=3 seek=org.wi-fi.wfds.display.tx seek=4 seek=5 seek=6",
  625. "asp_svc=org.wi-fi.wfds.display.tx"),
  626. ("seek=not-found", None),
  627. ("seek=org.wi-fi.wfds", "asp_svc=org.wi-fi.wfds")]:
  628. dev[2].global_request("P2P_FIND 10 type=social " + test[0])
  629. if test[1] is None:
  630. ev = dev[2].wait_global_event(["P2P-DEVICE-FOUND"], timeout=1)
  631. if ev is not None:
  632. raise Exception("Unexpected device found: " + ev)
  633. continue
  634. ev = dev[2].wait_global_event(["P2P-DEVICE-FOUND"], timeout=10)
  635. if ev is None:
  636. raise Exception("P2P device discovery timed out (dev2)")
  637. if test[1] not in ev:
  638. raise Exception("Expected asp_svc not reported: " + ev)
  639. dev[2].p2p_stop_find()
  640. dev[2].request("P2P_FLUSH")
  641. dev[0].p2p_stop_find()
  642. ev1 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id1))
  643. if ev1 is None:
  644. raise Exception("Unable to remove the advertisement instance")
  645. ev2 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id2))
  646. if ev2 is None:
  647. raise Exception("Unable to remove the advertisement instance")
  648. ev3 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id3))
  649. if ev3 is None:
  650. raise Exception("Unable to remove the advertisement instance")
  651. ev4 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id4))
  652. if ev4 is None:
  653. raise Exception("Unable to remove the advertisement instance")
  654. if "OK" not in dev[0].global_request("P2P_SERVICE_ADD asp 1 12345678 1 1108 org.wi-fi.wfds.foobar svc_info='Test'"):
  655. raise Exception("P2P_SERVICE_ADD failed")
  656. if "OK" not in dev[0].global_request("P2P_SERVICE_DEL asp all"):
  657. raise Exception("P2P_SERVICE_DEL asp all failed")
  658. if "OK" not in dev[0].global_request("P2P_SERVICE_ADD asp 1 12345678 1 1108 org.wi-fi.wfds.foobar svc_info='Test'"):
  659. raise Exception("P2P_SERVICE_ADD failed")
  660. if "OK" not in dev[0].global_request("P2P_SERVICE_REP asp 1 12345678 1 1108 org.wi-fi.wfds.foobar svc_info='Test'"):
  661. raise Exception("P2P_SERVICE_REP failed")
  662. if "FAIL" not in dev[0].global_request("P2P_SERVICE_REP asp 1 12345678 1 1108 org.wi-fi.wfds.Foo svc_info='Test'"):
  663. raise Exception("Invalid P2P_SERVICE_REP accepted")
  664. if "OK" not in dev[0].global_request("P2P_SERVICE_ADD asp 1 a2345678 1 1108 org.wi-fi.wfds.something svc_info='Test'"):
  665. raise Exception("P2P_SERVICE_ADD failed")
  666. if "OK" not in dev[0].global_request("P2P_SERVICE_ADD asp 1 a2345679 1 1108 org.wi-fi.wfds.Foo svc_info='Test'"):
  667. raise Exception("P2P_SERVICE_ADD failed")
  668. def get_ifnames():
  669. with open('/proc/net/dev', 'r') as f:
  670. data = f.read()
  671. ifnames = []
  672. for line in data.splitlines():
  673. ifname = line.strip().split(' ')[0]
  674. if ':' not in ifname:
  675. continue
  676. ifname = ifname.split(':')[0]
  677. ifnames.append(ifname)
  678. return ifnames
  679. def p2ps_connect_p2ps_method(dev):
  680. addr0 = dev[0].p2p_dev_addr()
  681. addr1 = dev[1].p2p_dev_addr()
  682. dev[0].flush_scan_cache()
  683. dev[1].flush_scan_cache()
  684. p2ps_advertise(r_dev=dev[0], r_role='2', svc_name='org.wi-fi.wfds.send.rx',
  685. srv_info='I can receive files upto size 2 GB')
  686. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  687. svc_name='org.wi-fi.wfds.send.rx',
  688. srv_info='2 GB')
  689. 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"):
  690. raise Exception("Failed to request provisioning on seeker")
  691. ev0 = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  692. if ev0 is None:
  693. raise Exception("P2PS-PROV-DONE timeout on advertiser side")
  694. if "join=" not in ev0:
  695. raise Exception("join parameter missing from P2PS-PROV-DONE")
  696. ev1 = dev[1].wait_global_event(["P2PS-PROV-DONE"], timeout=5)
  697. if ev1 is None:
  698. raise Exception("Provisioning failed on seeker side")
  699. ev1 = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=15)
  700. if ev1 is None:
  701. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  702. res1 = dev[1].group_form_result(ev1)
  703. ifnames1 = get_ifnames()
  704. if "OK" not in dev[0].global_request("P2P_CONNECT " + addr1 + " 12345670 p2ps persistent join"):
  705. raise Exception("P2P_CONNECT failed on seeker side")
  706. ev0 = dev[0].wait_global_event(["P2P-GROUP-STARTED"], timeout=15)
  707. if ev0 is None:
  708. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  709. res0 = dev[0].group_form_result(ev0)
  710. ev1 = dev[1].wait_global_event(["AP-STA-CONNECTED"], timeout=5)
  711. if ev1 is None:
  712. raise Exception("Group formation failed")
  713. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  714. if ev0 is None:
  715. raise Exception("Unable to remove the advertisement instance")
  716. ifnames2 = get_ifnames()
  717. remove_group(dev[0], dev[1])
  718. ifnames3 = get_ifnames()
  719. return (res0, res1, ifnames1 + ifnames2 + ifnames3)
  720. def has_string_prefix(vals, prefix):
  721. for val in vals:
  722. if val.startswith(prefix):
  723. return True
  724. return False
  725. def test_p2ps_connect_p2ps_method_1(dev):
  726. """P2PS connection with P2PS method - no group interface"""
  727. (res0, res1, ifnames) = p2ps_connect_p2ps_method(dev)
  728. if res0['ifname'] != dev[0].ifname:
  729. raise Exception("unexpected dev0 group ifname: " + res0['ifname'])
  730. if res1['ifname'] != dev[1].ifname:
  731. raise Exception("unexpected dev1 group ifname: " + res1['ifname'])
  732. if has_string_prefix(ifnames, 'p2p-' + res0['ifname']):
  733. raise Exception("dev0 group interface unexpectedly present")
  734. if has_string_prefix(ifnames, 'p2p-' + res1['ifname']):
  735. raise Exception("dev1 group interface unexpectedly present")
  736. def test_p2ps_connect_p2ps_method_2(dev):
  737. """P2PS connection with P2PS method - group interface on dev0"""
  738. dev[0].request("SET p2p_no_group_iface 0")
  739. (res0, res1, ifnames) = p2ps_connect_p2ps_method(dev)
  740. if not res0['ifname'].startswith('p2p-' + dev[0].ifname + '-'):
  741. raise Exception("unexpected dev0 group ifname: " + res0['ifname'])
  742. if res1['ifname'] != dev[1].ifname:
  743. raise Exception("unexpected dev1 group ifname: " + res1['ifname'])
  744. if has_string_prefix(ifnames, 'p2p-' + res0['ifname']):
  745. raise Exception("dev0 group interface unexpectedly present")
  746. def test_p2ps_connect_p2ps_method_3(dev):
  747. """P2PS connection with P2PS method - group interface on dev1"""
  748. dev[1].request("SET p2p_no_group_iface 0")
  749. (res0, res1, ifnames) = p2ps_connect_p2ps_method(dev)
  750. if res0['ifname'] != dev[0].ifname:
  751. raise Exception("unexpected dev0 group ifname: " + res0['ifname'])
  752. if not res1['ifname'].startswith('p2p-' + dev[1].ifname + '-'):
  753. raise Exception("unexpected dev1 group ifname: " + res1['ifname'])
  754. if has_string_prefix(ifnames, 'p2p-' + res0['ifname']):
  755. raise Exception("dev0 group interface unexpectedly present")
  756. def test_p2ps_connect_p2ps_method_4(dev):
  757. """P2PS connection with P2PS method - group interface on both"""
  758. dev[0].request("SET p2p_no_group_iface 0")
  759. dev[1].request("SET p2p_no_group_iface 0")
  760. (res0, res1, ifnames) = p2ps_connect_p2ps_method(dev)
  761. if not res0['ifname'].startswith('p2p-' + dev[0].ifname + '-'):
  762. raise Exception("unexpected dev0 group ifname: " + res0['ifname'])
  763. if not res1['ifname'].startswith('p2p-' + dev[1].ifname + '-'):
  764. raise Exception("unexpected dev1 group ifname: " + res1['ifname'])
  765. def test_p2ps_connect_adv_go_persistent(dev):
  766. """P2PS auto-accept connection with advertisement as GO and having persistent group"""
  767. addr0 = dev[0].p2p_dev_addr()
  768. addr1 = dev[1].p2p_dev_addr()
  769. go_neg_pin_authorized_persistent(i_dev=dev[0], i_intent=15,
  770. r_dev=dev[1], r_intent=0)
  771. dev[0].remove_group()
  772. dev[1].wait_go_ending_session()
  773. p2ps_advertise(r_dev=dev[0], r_role='4', svc_name='org.wi-fi.wfds.send.rx',
  774. srv_info='I can receive files upto size 2 GB')
  775. [adv_id, rcvd_svc_name] = p2ps_exact_seek(i_dev=dev[1], r_dev=dev[0],
  776. svc_name='org.wi-fi.wfds.send.rx',
  777. srv_info='2 GB')
  778. 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"):
  779. raise Exception("Failed to request provisioning on seeker")
  780. ev0 = dev[0].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  781. if ev0 is None:
  782. raise Exception("Timed out while waiting for prov done on advertizer")
  783. if "persist=" not in ev0:
  784. raise Exception("Advertiser did not indicate persistent group")
  785. id0 = ev0.split("persist=")[1].split(" ")[0]
  786. if "OK" not in dev[0].global_request("P2P_GROUP_ADD persistent=" + id0 + " freq=2412"):
  787. raise Exception("Could not re-start persistent group")
  788. ev0 = dev[0].wait_global_event(["P2P-GROUP-STARTED"], timeout=10)
  789. if ev0 is None:
  790. raise Exception("P2P-GROUP-STARTED timeout on advertiser side")
  791. ev1 = dev[1].wait_global_event(["P2PS-PROV-DONE"], timeout=10)
  792. if ev1 is None:
  793. raise Exception("P2PS-PROV-DONE timeout on seeker side")
  794. if "persist=" not in ev1:
  795. raise Exception("Seeker did not indicate persistent group")
  796. id1 = ev1.split("persist=")[1].split(" ")[0]
  797. if "OK" not in dev[1].global_request("P2P_GROUP_ADD persistent=" + id1 + " freq=2412"):
  798. raise Exception("Could not re-start persistent group")
  799. ev1 = dev[1].wait_global_event(["P2P-GROUP-STARTED"], timeout=15)
  800. if ev1 is None:
  801. raise Exception("P2P-GROUP-STARTED timeout on seeker side")
  802. ev0 = dev[0].wait_global_event(["AP-STA-CONNECTED"], timeout=15)
  803. if ev0 is None:
  804. raise Exception("AP-STA-CONNECTED timeout on advertiser side")
  805. ev0 = dev[0].global_request("P2P_SERVICE_DEL asp " + str(adv_id))
  806. if ev0 is None:
  807. raise Exception("Unable to remove the advertisement instance")
  808. remove_group(dev[0], dev[1])