security.c 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642
  1. /*
  2. * Wireless USB Host Controller
  3. * Security support: encryption enablement, etc
  4. *
  5. * Copyright (C) 2006 Intel Corporation
  6. * Inaky Perez-Gonzalez <inaky.perez-gonzalez@intel.com>
  7. *
  8. * This program is free software; you can redistribute it and/or
  9. * modify it under the terms of the GNU General Public License version
  10. * 2 as published by the Free Software Foundation.
  11. *
  12. * This program is distributed in the hope that it will be useful,
  13. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15. * GNU General Public License for more details.
  16. *
  17. * You should have received a copy of the GNU General Public License
  18. * along with this program; if not, write to the Free Software
  19. * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  20. * 02110-1301, USA.
  21. *
  22. *
  23. * FIXME: docs
  24. */
  25. #include <linux/types.h>
  26. #include <linux/usb/ch9.h>
  27. #include <linux/random.h>
  28. #include "wusbhc.h"
  29. /*
  30. * DEBUG & SECURITY WARNING!!!!
  31. *
  32. * If you enable this past 1, the debug code will weaken the
  33. * cryptographic safety of the system (on purpose, for debugging).
  34. *
  35. * Weaken means:
  36. * we print secret keys and intermediate values all the way,
  37. */
  38. #undef D_LOCAL
  39. #define D_LOCAL 2
  40. #include <linux/uwb/debug.h>
  41. static void wusbhc_set_gtk_callback(struct urb *urb);
  42. static void wusbhc_gtk_rekey_done_work(struct work_struct *work);
  43. int wusbhc_sec_create(struct wusbhc *wusbhc)
  44. {
  45. wusbhc->gtk.descr.bLength = sizeof(wusbhc->gtk.descr) + sizeof(wusbhc->gtk.data);
  46. wusbhc->gtk.descr.bDescriptorType = USB_DT_KEY;
  47. wusbhc->gtk.descr.bReserved = 0;
  48. wusbhc->gtk_index = wusb_key_index(0, WUSB_KEY_INDEX_TYPE_GTK,
  49. WUSB_KEY_INDEX_ORIGINATOR_HOST);
  50. INIT_WORK(&wusbhc->gtk_rekey_done_work, wusbhc_gtk_rekey_done_work);
  51. return 0;
  52. }
  53. /* Called when the HC is destroyed */
  54. void wusbhc_sec_destroy(struct wusbhc *wusbhc)
  55. {
  56. }
  57. /**
  58. * wusbhc_next_tkid - generate a new, currently unused, TKID
  59. * @wusbhc: the WUSB host controller
  60. * @wusb_dev: the device whose PTK the TKID is for
  61. * (or NULL for a TKID for a GTK)
  62. *
  63. * The generated TKID consist of two parts: the device's authenicated
  64. * address (or 0 or a GTK); and an incrementing number. This ensures
  65. * that TKIDs cannot be shared between devices and by the time the
  66. * incrementing number wraps around the older TKIDs will no longer be
  67. * in use (a maximum of two keys may be active at any one time).
  68. */
  69. static u32 wusbhc_next_tkid(struct wusbhc *wusbhc, struct wusb_dev *wusb_dev)
  70. {
  71. u32 *tkid;
  72. u32 addr;
  73. if (wusb_dev == NULL) {
  74. tkid = &wusbhc->gtk_tkid;
  75. addr = 0;
  76. } else {
  77. tkid = &wusb_port_by_idx(wusbhc, wusb_dev->port_idx)->ptk_tkid;
  78. addr = wusb_dev->addr & 0x7f;
  79. }
  80. *tkid = (addr << 8) | ((*tkid + 1) & 0xff);
  81. return *tkid;
  82. }
  83. static void wusbhc_generate_gtk(struct wusbhc *wusbhc)
  84. {
  85. const size_t key_size = sizeof(wusbhc->gtk.data);
  86. u32 tkid;
  87. tkid = wusbhc_next_tkid(wusbhc, NULL);
  88. wusbhc->gtk.descr.tTKID[0] = (tkid >> 0) & 0xff;
  89. wusbhc->gtk.descr.tTKID[1] = (tkid >> 8) & 0xff;
  90. wusbhc->gtk.descr.tTKID[2] = (tkid >> 16) & 0xff;
  91. get_random_bytes(wusbhc->gtk.descr.bKeyData, key_size);
  92. }
  93. /**
  94. * wusbhc_sec_start - start the security management process
  95. * @wusbhc: the WUSB host controller
  96. *
  97. * Generate and set an initial GTK on the host controller.
  98. *
  99. * Called when the HC is started.
  100. */
  101. int wusbhc_sec_start(struct wusbhc *wusbhc)
  102. {
  103. const size_t key_size = sizeof(wusbhc->gtk.data);
  104. int result;
  105. wusbhc_generate_gtk(wusbhc);
  106. result = wusbhc->set_gtk(wusbhc, wusbhc->gtk_tkid,
  107. &wusbhc->gtk.descr.bKeyData, key_size);
  108. if (result < 0)
  109. dev_err(wusbhc->dev, "cannot set GTK for the host: %d\n",
  110. result);
  111. return result;
  112. }
  113. /**
  114. * wusbhc_sec_stop - stop the security management process
  115. * @wusbhc: the WUSB host controller
  116. *
  117. * Wait for any pending GTK rekeys to stop.
  118. */
  119. void wusbhc_sec_stop(struct wusbhc *wusbhc)
  120. {
  121. cancel_work_sync(&wusbhc->gtk_rekey_done_work);
  122. }
  123. /** @returns encryption type name */
  124. const char *wusb_et_name(u8 x)
  125. {
  126. switch (x) {
  127. case USB_ENC_TYPE_UNSECURE: return "unsecure";
  128. case USB_ENC_TYPE_WIRED: return "wired";
  129. case USB_ENC_TYPE_CCM_1: return "CCM-1";
  130. case USB_ENC_TYPE_RSA_1: return "RSA-1";
  131. default: return "unknown";
  132. }
  133. }
  134. EXPORT_SYMBOL_GPL(wusb_et_name);
  135. /*
  136. * Set the device encryption method
  137. *
  138. * We tell the device which encryption method to use; we do this when
  139. * setting up the device's security.
  140. */
  141. static int wusb_dev_set_encryption(struct usb_device *usb_dev, int value)
  142. {
  143. int result;
  144. struct device *dev = &usb_dev->dev;
  145. struct wusb_dev *wusb_dev = usb_dev->wusb_dev;
  146. if (value) {
  147. value = wusb_dev->ccm1_etd.bEncryptionValue;
  148. } else {
  149. /* FIXME: should be wusb_dev->etd[UNSECURE].bEncryptionValue */
  150. value = 0;
  151. }
  152. /* Set device's */
  153. result = usb_control_msg(usb_dev, usb_sndctrlpipe(usb_dev, 0),
  154. USB_REQ_SET_ENCRYPTION,
  155. USB_DIR_OUT | USB_TYPE_STANDARD | USB_RECIP_DEVICE,
  156. value, 0, NULL, 0, 1000 /* FIXME: arbitrary */);
  157. if (result < 0)
  158. dev_err(dev, "Can't set device's WUSB encryption to "
  159. "%s (value %d): %d\n",
  160. wusb_et_name(wusb_dev->ccm1_etd.bEncryptionType),
  161. wusb_dev->ccm1_etd.bEncryptionValue, result);
  162. return result;
  163. }
  164. /*
  165. * Set the GTK to be used by a device.
  166. *
  167. * The device must be authenticated.
  168. */
  169. static int wusb_dev_set_gtk(struct wusbhc *wusbhc, struct wusb_dev *wusb_dev)
  170. {
  171. struct usb_device *usb_dev = wusb_dev->usb_dev;
  172. return usb_control_msg(
  173. usb_dev, usb_sndctrlpipe(usb_dev, 0),
  174. USB_REQ_SET_DESCRIPTOR,
  175. USB_DIR_OUT | USB_TYPE_STANDARD | USB_RECIP_DEVICE,
  176. USB_DT_KEY << 8 | wusbhc->gtk_index, 0,
  177. &wusbhc->gtk.descr, wusbhc->gtk.descr.bLength,
  178. 1000);
  179. }
  180. /* FIXME: prototype for adding security */
  181. int wusb_dev_sec_add(struct wusbhc *wusbhc,
  182. struct usb_device *usb_dev, struct wusb_dev *wusb_dev)
  183. {
  184. int result, bytes, secd_size;
  185. struct device *dev = &usb_dev->dev;
  186. struct usb_security_descriptor secd;
  187. const struct usb_encryption_descriptor *etd, *ccm1_etd = NULL;
  188. void *secd_buf;
  189. const void *itr, *top;
  190. char buf[64];
  191. d_fnstart(3, dev, "(usb_dev %p, wusb_dev %p)\n", usb_dev, wusb_dev);
  192. result = usb_get_descriptor(usb_dev, USB_DT_SECURITY,
  193. 0, &secd, sizeof(secd));
  194. if (result < sizeof(secd)) {
  195. dev_err(dev, "Can't read security descriptor or "
  196. "not enough data: %d\n", result);
  197. goto error_secd;
  198. }
  199. secd_size = le16_to_cpu(secd.wTotalLength);
  200. d_printf(5, dev, "got %d bytes of sec descriptor, total is %d\n",
  201. result, secd_size);
  202. secd_buf = kmalloc(secd_size, GFP_KERNEL);
  203. if (secd_buf == NULL) {
  204. dev_err(dev, "Can't allocate space for security descriptors\n");
  205. goto error_secd_alloc;
  206. }
  207. result = usb_get_descriptor(usb_dev, USB_DT_SECURITY,
  208. 0, secd_buf, secd_size);
  209. if (result < secd_size) {
  210. dev_err(dev, "Can't read security descriptor or "
  211. "not enough data: %d\n", result);
  212. goto error_secd_all;
  213. }
  214. d_printf(5, dev, "got %d bytes of sec descriptors\n", result);
  215. bytes = 0;
  216. itr = secd_buf + sizeof(secd);
  217. top = secd_buf + result;
  218. while (itr < top) {
  219. etd = itr;
  220. if (top - itr < sizeof(*etd)) {
  221. dev_err(dev, "BUG: bad device security descriptor; "
  222. "not enough data (%zu vs %zu bytes left)\n",
  223. top - itr, sizeof(*etd));
  224. break;
  225. }
  226. if (etd->bLength < sizeof(*etd)) {
  227. dev_err(dev, "BUG: bad device encryption descriptor; "
  228. "descriptor is too short "
  229. "(%u vs %zu needed)\n",
  230. etd->bLength, sizeof(*etd));
  231. break;
  232. }
  233. itr += etd->bLength;
  234. bytes += snprintf(buf + bytes, sizeof(buf) - bytes,
  235. "%s (0x%02x/%02x) ",
  236. wusb_et_name(etd->bEncryptionType),
  237. etd->bEncryptionValue, etd->bAuthKeyIndex);
  238. if (etd->bEncryptionType == USB_ENC_TYPE_CCM_1)
  239. ccm1_etd = etd;
  240. }
  241. /* This code only supports CCM1 as of now. */
  242. /* FIXME: user has to choose which sec mode to use?
  243. * In theory we want CCM */
  244. if (ccm1_etd == NULL) {
  245. dev_err(dev, "WUSB device doesn't support CCM1 encryption, "
  246. "can't use!\n");
  247. result = -EINVAL;
  248. goto error_no_ccm1;
  249. }
  250. wusb_dev->ccm1_etd = *ccm1_etd;
  251. dev_info(dev, "supported encryption: %s; using %s (0x%02x/%02x)\n",
  252. buf, wusb_et_name(ccm1_etd->bEncryptionType),
  253. ccm1_etd->bEncryptionValue, ccm1_etd->bAuthKeyIndex);
  254. result = 0;
  255. kfree(secd_buf);
  256. out:
  257. d_fnend(3, dev, "(usb_dev %p, wusb_dev %p) = %d\n",
  258. usb_dev, wusb_dev, result);
  259. return result;
  260. error_no_ccm1:
  261. error_secd_all:
  262. kfree(secd_buf);
  263. error_secd_alloc:
  264. error_secd:
  265. goto out;
  266. }
  267. void wusb_dev_sec_rm(struct wusb_dev *wusb_dev)
  268. {
  269. /* Nothing so far */
  270. }
  271. static void hs_printk(unsigned level, struct device *dev,
  272. struct usb_handshake *hs)
  273. {
  274. d_printf(level, dev,
  275. " bMessageNumber: %u\n"
  276. " bStatus: %u\n"
  277. " tTKID: %02x %02x %02x\n"
  278. " CDID: %02x %02x %02x %02x %02x %02x %02x %02x\n"
  279. " %02x %02x %02x %02x %02x %02x %02x %02x\n"
  280. " nonce: %02x %02x %02x %02x %02x %02x %02x %02x\n"
  281. " %02x %02x %02x %02x %02x %02x %02x %02x\n"
  282. " MIC: %02x %02x %02x %02x %02x %02x %02x %02x\n",
  283. hs->bMessageNumber, hs->bStatus,
  284. hs->tTKID[2], hs->tTKID[1], hs->tTKID[0],
  285. hs->CDID[0], hs->CDID[1], hs->CDID[2], hs->CDID[3],
  286. hs->CDID[4], hs->CDID[5], hs->CDID[6], hs->CDID[7],
  287. hs->CDID[8], hs->CDID[9], hs->CDID[10], hs->CDID[11],
  288. hs->CDID[12], hs->CDID[13], hs->CDID[14], hs->CDID[15],
  289. hs->nonce[0], hs->nonce[1], hs->nonce[2], hs->nonce[3],
  290. hs->nonce[4], hs->nonce[5], hs->nonce[6], hs->nonce[7],
  291. hs->nonce[8], hs->nonce[9], hs->nonce[10], hs->nonce[11],
  292. hs->nonce[12], hs->nonce[13], hs->nonce[14], hs->nonce[15],
  293. hs->MIC[0], hs->MIC[1], hs->MIC[2], hs->MIC[3],
  294. hs->MIC[4], hs->MIC[5], hs->MIC[6], hs->MIC[7]);
  295. }
  296. /**
  297. * Update the address of an unauthenticated WUSB device
  298. *
  299. * Once we have successfully authenticated, we take it to addr0 state
  300. * and then to a normal address.
  301. *
  302. * Before the device's address (as known by it) was usb_dev->devnum |
  303. * 0x80 (unauthenticated address). With this we update it to usb_dev->devnum.
  304. */
  305. static int wusb_dev_update_address(struct wusbhc *wusbhc,
  306. struct wusb_dev *wusb_dev)
  307. {
  308. int result = -ENOMEM;
  309. struct usb_device *usb_dev = wusb_dev->usb_dev;
  310. struct device *dev = &usb_dev->dev;
  311. u8 new_address = wusb_dev->addr & 0x7F;
  312. /* Set address 0 */
  313. result = usb_control_msg(usb_dev, usb_sndctrlpipe(usb_dev, 0),
  314. USB_REQ_SET_ADDRESS, 0,
  315. 0, 0, NULL, 0, 1000 /* FIXME: arbitrary */);
  316. if (result < 0) {
  317. dev_err(dev, "auth failed: can't set address 0: %d\n",
  318. result);
  319. goto error_addr0;
  320. }
  321. result = wusb_set_dev_addr(wusbhc, wusb_dev, 0);
  322. if (result < 0)
  323. goto error_addr0;
  324. usb_ep0_reinit(usb_dev);
  325. /* Set new (authenticated) address. */
  326. result = usb_control_msg(usb_dev, usb_sndctrlpipe(usb_dev, 0),
  327. USB_REQ_SET_ADDRESS, 0,
  328. new_address, 0, NULL, 0,
  329. 1000 /* FIXME: arbitrary */);
  330. if (result < 0) {
  331. dev_err(dev, "auth failed: can't set address %u: %d\n",
  332. new_address, result);
  333. goto error_addr;
  334. }
  335. result = wusb_set_dev_addr(wusbhc, wusb_dev, new_address);
  336. if (result < 0)
  337. goto error_addr;
  338. usb_ep0_reinit(usb_dev);
  339. usb_dev->authenticated = 1;
  340. error_addr:
  341. error_addr0:
  342. return result;
  343. }
  344. /*
  345. *
  346. *
  347. */
  348. /* FIXME: split and cleanup */
  349. int wusb_dev_4way_handshake(struct wusbhc *wusbhc, struct wusb_dev *wusb_dev,
  350. struct wusb_ckhdid *ck)
  351. {
  352. int result = -ENOMEM;
  353. struct usb_device *usb_dev = wusb_dev->usb_dev;
  354. struct device *dev = &usb_dev->dev;
  355. u32 tkid;
  356. __le32 tkid_le;
  357. struct usb_handshake *hs;
  358. struct aes_ccm_nonce ccm_n;
  359. u8 mic[8];
  360. struct wusb_keydvt_in keydvt_in;
  361. struct wusb_keydvt_out keydvt_out;
  362. hs = kzalloc(3*sizeof(hs[0]), GFP_KERNEL);
  363. if (hs == NULL) {
  364. dev_err(dev, "can't allocate handshake data\n");
  365. goto error_kzalloc;
  366. }
  367. /* We need to turn encryption before beginning the 4way
  368. * hshake (WUSB1.0[.3.2.2]) */
  369. result = wusb_dev_set_encryption(usb_dev, 1);
  370. if (result < 0)
  371. goto error_dev_set_encryption;
  372. tkid = wusbhc_next_tkid(wusbhc, wusb_dev);
  373. tkid_le = cpu_to_le32(tkid);
  374. hs[0].bMessageNumber = 1;
  375. hs[0].bStatus = 0;
  376. memcpy(hs[0].tTKID, &tkid_le, sizeof(hs[0].tTKID));
  377. hs[0].bReserved = 0;
  378. memcpy(hs[0].CDID, &wusb_dev->cdid, sizeof(hs[0].CDID));
  379. get_random_bytes(&hs[0].nonce, sizeof(hs[0].nonce));
  380. memset(hs[0].MIC, 0, sizeof(hs[0].MIC)); /* Per WUSB1.0[T7-22] */
  381. d_printf(1, dev, "I: sending hs1:\n");
  382. hs_printk(2, dev, &hs[0]);
  383. result = usb_control_msg(
  384. usb_dev, usb_sndctrlpipe(usb_dev, 0),
  385. USB_REQ_SET_HANDSHAKE,
  386. USB_DIR_OUT | USB_TYPE_STANDARD | USB_RECIP_DEVICE,
  387. 1, 0, &hs[0], sizeof(hs[0]), 1000 /* FIXME: arbitrary */);
  388. if (result < 0) {
  389. dev_err(dev, "Handshake1: request failed: %d\n", result);
  390. goto error_hs1;
  391. }
  392. /* Handshake 2, from the device -- need to verify fields */
  393. result = usb_control_msg(
  394. usb_dev, usb_rcvctrlpipe(usb_dev, 0),
  395. USB_REQ_GET_HANDSHAKE,
  396. USB_DIR_IN | USB_TYPE_STANDARD | USB_RECIP_DEVICE,
  397. 2, 0, &hs[1], sizeof(hs[1]), 1000 /* FIXME: arbitrary */);
  398. if (result < 0) {
  399. dev_err(dev, "Handshake2: request failed: %d\n", result);
  400. goto error_hs2;
  401. }
  402. d_printf(1, dev, "got HS2:\n");
  403. hs_printk(2, dev, &hs[1]);
  404. result = -EINVAL;
  405. if (hs[1].bMessageNumber != 2) {
  406. dev_err(dev, "Handshake2 failed: bad message number %u\n",
  407. hs[1].bMessageNumber);
  408. goto error_hs2;
  409. }
  410. if (hs[1].bStatus != 0) {
  411. dev_err(dev, "Handshake2 failed: bad status %u\n",
  412. hs[1].bStatus);
  413. goto error_hs2;
  414. }
  415. if (memcmp(hs[0].tTKID, hs[1].tTKID, sizeof(hs[0].tTKID))) {
  416. dev_err(dev, "Handshake2 failed: TKID mismatch "
  417. "(#1 0x%02x%02x%02x vs #2 0x%02x%02x%02x)\n",
  418. hs[0].tTKID[0], hs[0].tTKID[1], hs[0].tTKID[2],
  419. hs[1].tTKID[0], hs[1].tTKID[1], hs[1].tTKID[2]);
  420. goto error_hs2;
  421. }
  422. if (memcmp(hs[0].CDID, hs[1].CDID, sizeof(hs[0].CDID))) {
  423. dev_err(dev, "Handshake2 failed: CDID mismatch\n");
  424. goto error_hs2;
  425. }
  426. /* Setup the CCM nonce */
  427. memset(&ccm_n.sfn, 0, sizeof(ccm_n.sfn)); /* Per WUSB1.0[6.5.2] */
  428. memcpy(ccm_n.tkid, &tkid_le, sizeof(ccm_n.tkid));
  429. ccm_n.src_addr = wusbhc->uwb_rc->uwb_dev.dev_addr;
  430. ccm_n.dest_addr.data[0] = wusb_dev->addr;
  431. ccm_n.dest_addr.data[1] = 0;
  432. /* Derive the KCK and PTK from CK, the CCM, H and D nonces */
  433. memcpy(keydvt_in.hnonce, hs[0].nonce, sizeof(keydvt_in.hnonce));
  434. memcpy(keydvt_in.dnonce, hs[1].nonce, sizeof(keydvt_in.dnonce));
  435. result = wusb_key_derive(&keydvt_out, ck->data, &ccm_n, &keydvt_in);
  436. if (result < 0) {
  437. dev_err(dev, "Handshake2 failed: cannot derive keys: %d\n",
  438. result);
  439. goto error_hs2;
  440. }
  441. d_printf(2, dev, "KCK:\n");
  442. d_dump(2, dev, keydvt_out.kck, sizeof(keydvt_out.kck));
  443. d_printf(2, dev, "PTK:\n");
  444. d_dump(2, dev, keydvt_out.ptk, sizeof(keydvt_out.ptk));
  445. /* Compute MIC and verify it */
  446. result = wusb_oob_mic(mic, keydvt_out.kck, &ccm_n, &hs[1]);
  447. if (result < 0) {
  448. dev_err(dev, "Handshake2 failed: cannot compute MIC: %d\n",
  449. result);
  450. goto error_hs2;
  451. }
  452. d_printf(2, dev, "MIC:\n");
  453. d_dump(2, dev, mic, sizeof(mic));
  454. if (memcmp(hs[1].MIC, mic, sizeof(hs[1].MIC))) {
  455. dev_err(dev, "Handshake2 failed: MIC mismatch\n");
  456. goto error_hs2;
  457. }
  458. /* Send Handshake3 */
  459. hs[2].bMessageNumber = 3;
  460. hs[2].bStatus = 0;
  461. memcpy(hs[2].tTKID, &tkid_le, sizeof(hs[2].tTKID));
  462. hs[2].bReserved = 0;
  463. memcpy(hs[2].CDID, &wusb_dev->cdid, sizeof(hs[2].CDID));
  464. memcpy(hs[2].nonce, hs[0].nonce, sizeof(hs[2].nonce));
  465. result = wusb_oob_mic(hs[2].MIC, keydvt_out.kck, &ccm_n, &hs[2]);
  466. if (result < 0) {
  467. dev_err(dev, "Handshake3 failed: cannot compute MIC: %d\n",
  468. result);
  469. goto error_hs2;
  470. }
  471. d_printf(1, dev, "I: sending hs3:\n");
  472. hs_printk(2, dev, &hs[2]);
  473. result = usb_control_msg(
  474. usb_dev, usb_sndctrlpipe(usb_dev, 0),
  475. USB_REQ_SET_HANDSHAKE,
  476. USB_DIR_OUT | USB_TYPE_STANDARD | USB_RECIP_DEVICE,
  477. 3, 0, &hs[2], sizeof(hs[2]), 1000 /* FIXME: arbitrary */);
  478. if (result < 0) {
  479. dev_err(dev, "Handshake3: request failed: %d\n", result);
  480. goto error_hs3;
  481. }
  482. d_printf(1, dev, "I: turning on encryption on host for device\n");
  483. d_dump(2, dev, keydvt_out.ptk, sizeof(keydvt_out.ptk));
  484. result = wusbhc->set_ptk(wusbhc, wusb_dev->port_idx, tkid,
  485. keydvt_out.ptk, sizeof(keydvt_out.ptk));
  486. if (result < 0)
  487. goto error_wusbhc_set_ptk;
  488. d_printf(1, dev, "I: setting a GTK\n");
  489. result = wusb_dev_set_gtk(wusbhc, wusb_dev);
  490. if (result < 0) {
  491. dev_err(dev, "Set GTK for device: request failed: %d\n",
  492. result);
  493. goto error_wusbhc_set_gtk;
  494. }
  495. /* Update the device's address from unauth to auth */
  496. if (usb_dev->authenticated == 0) {
  497. d_printf(1, dev, "I: updating addres to auth from non-auth\n");
  498. result = wusb_dev_update_address(wusbhc, wusb_dev);
  499. if (result < 0)
  500. goto error_dev_update_address;
  501. }
  502. result = 0;
  503. d_printf(1, dev, "I: 4way handshke done, device authenticated\n");
  504. error_dev_update_address:
  505. error_wusbhc_set_gtk:
  506. error_wusbhc_set_ptk:
  507. error_hs3:
  508. error_hs2:
  509. error_hs1:
  510. memset(hs, 0, 3*sizeof(hs[0]));
  511. memset(&keydvt_out, 0, sizeof(keydvt_out));
  512. memset(&keydvt_in, 0, sizeof(keydvt_in));
  513. memset(&ccm_n, 0, sizeof(ccm_n));
  514. memset(mic, 0, sizeof(mic));
  515. if (result < 0) {
  516. /* error path */
  517. wusb_dev_set_encryption(usb_dev, 0);
  518. }
  519. error_dev_set_encryption:
  520. kfree(hs);
  521. error_kzalloc:
  522. return result;
  523. }
  524. /*
  525. * Once all connected and authenticated devices have received the new
  526. * GTK, switch the host to using it.
  527. */
  528. static void wusbhc_gtk_rekey_done_work(struct work_struct *work)
  529. {
  530. struct wusbhc *wusbhc = container_of(work, struct wusbhc, gtk_rekey_done_work);
  531. size_t key_size = sizeof(wusbhc->gtk.data);
  532. mutex_lock(&wusbhc->mutex);
  533. if (--wusbhc->pending_set_gtks == 0)
  534. wusbhc->set_gtk(wusbhc, wusbhc->gtk_tkid, &wusbhc->gtk.descr.bKeyData, key_size);
  535. mutex_unlock(&wusbhc->mutex);
  536. }
  537. static void wusbhc_set_gtk_callback(struct urb *urb)
  538. {
  539. struct wusbhc *wusbhc = urb->context;
  540. queue_work(wusbd, &wusbhc->gtk_rekey_done_work);
  541. }
  542. /**
  543. * wusbhc_gtk_rekey - generate and distribute a new GTK
  544. * @wusbhc: the WUSB host controller
  545. *
  546. * Generate a new GTK and distribute it to all connected and
  547. * authenticated devices. When all devices have the new GTK, the host
  548. * starts using it.
  549. *
  550. * This must be called after every device disconnect (see [WUSB]
  551. * section 6.2.11.2).
  552. */
  553. void wusbhc_gtk_rekey(struct wusbhc *wusbhc)
  554. {
  555. static const size_t key_size = sizeof(wusbhc->gtk.data);
  556. int p;
  557. wusbhc_generate_gtk(wusbhc);
  558. for (p = 0; p < wusbhc->ports_max; p++) {
  559. struct wusb_dev *wusb_dev;
  560. wusb_dev = wusbhc->port[p].wusb_dev;
  561. if (!wusb_dev || !wusb_dev->usb_dev | !wusb_dev->usb_dev->authenticated)
  562. continue;
  563. usb_fill_control_urb(wusb_dev->set_gtk_urb, wusb_dev->usb_dev,
  564. usb_sndctrlpipe(wusb_dev->usb_dev, 0),
  565. (void *)wusb_dev->set_gtk_req,
  566. &wusbhc->gtk.descr, wusbhc->gtk.descr.bLength,
  567. wusbhc_set_gtk_callback, wusbhc);
  568. if (usb_submit_urb(wusb_dev->set_gtk_urb, GFP_KERNEL) == 0)
  569. wusbhc->pending_set_gtks++;
  570. }
  571. if (wusbhc->pending_set_gtks == 0)
  572. wusbhc->set_gtk(wusbhc, wusbhc->gtk_tkid, &wusbhc->gtk.descr.bKeyData, key_size);
  573. }