test_p2ps.py 47 KB

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