sock.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877
  1. /*
  2. * Copyright (C) 2011 Intel Corporation. All rights reserved.
  3. *
  4. * This program is free software; you can redistribute it and/or modify
  5. * it under the terms of the GNU General Public License as published by
  6. * the Free Software Foundation; either version 2 of the License, or
  7. * (at your option) any later version.
  8. *
  9. * This program is distributed in the hope that it will be useful,
  10. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  12. * GNU General Public License for more details.
  13. *
  14. * You should have received a copy of the GNU General Public License
  15. * along with this program; if not, write to the
  16. * Free Software Foundation, Inc.,
  17. * 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
  18. */
  19. #define pr_fmt(fmt) "llcp: %s: " fmt, __func__
  20. #include <linux/init.h>
  21. #include <linux/kernel.h>
  22. #include <linux/module.h>
  23. #include <linux/nfc.h>
  24. #include "../nfc.h"
  25. #include "llcp.h"
  26. static int sock_wait_state(struct sock *sk, int state, unsigned long timeo)
  27. {
  28. DECLARE_WAITQUEUE(wait, current);
  29. int err = 0;
  30. pr_debug("sk %p", sk);
  31. add_wait_queue(sk_sleep(sk), &wait);
  32. set_current_state(TASK_INTERRUPTIBLE);
  33. while (sk->sk_state != state) {
  34. if (!timeo) {
  35. err = -EINPROGRESS;
  36. break;
  37. }
  38. if (signal_pending(current)) {
  39. err = sock_intr_errno(timeo);
  40. break;
  41. }
  42. release_sock(sk);
  43. timeo = schedule_timeout(timeo);
  44. lock_sock(sk);
  45. set_current_state(TASK_INTERRUPTIBLE);
  46. err = sock_error(sk);
  47. if (err)
  48. break;
  49. }
  50. __set_current_state(TASK_RUNNING);
  51. remove_wait_queue(sk_sleep(sk), &wait);
  52. return err;
  53. }
  54. static struct proto llcp_sock_proto = {
  55. .name = "NFC_LLCP",
  56. .owner = THIS_MODULE,
  57. .obj_size = sizeof(struct nfc_llcp_sock),
  58. };
  59. static int llcp_sock_bind(struct socket *sock, struct sockaddr *addr, int alen)
  60. {
  61. struct sock *sk = sock->sk;
  62. struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
  63. struct nfc_llcp_local *local;
  64. struct nfc_dev *dev;
  65. struct sockaddr_nfc_llcp llcp_addr;
  66. int len, ret = 0;
  67. if (!addr || addr->sa_family != AF_NFC)
  68. return -EINVAL;
  69. pr_debug("sk %p addr %p family %d\n", sk, addr, addr->sa_family);
  70. memset(&llcp_addr, 0, sizeof(llcp_addr));
  71. len = min_t(unsigned int, sizeof(llcp_addr), alen);
  72. memcpy(&llcp_addr, addr, len);
  73. /* This is going to be a listening socket, dsap must be 0 */
  74. if (llcp_addr.dsap != 0)
  75. return -EINVAL;
  76. lock_sock(sk);
  77. if (sk->sk_state != LLCP_CLOSED) {
  78. ret = -EBADFD;
  79. goto error;
  80. }
  81. dev = nfc_get_device(llcp_addr.dev_idx);
  82. if (dev == NULL) {
  83. ret = -ENODEV;
  84. goto error;
  85. }
  86. local = nfc_llcp_find_local(dev);
  87. if (local == NULL) {
  88. ret = -ENODEV;
  89. goto put_dev;
  90. }
  91. llcp_sock->dev = dev;
  92. llcp_sock->local = nfc_llcp_local_get(local);
  93. llcp_sock->nfc_protocol = llcp_addr.nfc_protocol;
  94. llcp_sock->service_name_len = min_t(unsigned int,
  95. llcp_addr.service_name_len,
  96. NFC_LLCP_MAX_SERVICE_NAME);
  97. llcp_sock->service_name = kmemdup(llcp_addr.service_name,
  98. llcp_sock->service_name_len,
  99. GFP_KERNEL);
  100. llcp_sock->ssap = nfc_llcp_get_sdp_ssap(local, llcp_sock);
  101. if (llcp_sock->ssap == LLCP_SAP_MAX) {
  102. ret = -EADDRINUSE;
  103. goto put_dev;
  104. }
  105. llcp_sock->reserved_ssap = llcp_sock->ssap;
  106. nfc_llcp_sock_link(&local->sockets, sk);
  107. pr_debug("Socket bound to SAP %d\n", llcp_sock->ssap);
  108. sk->sk_state = LLCP_BOUND;
  109. put_dev:
  110. nfc_put_device(dev);
  111. error:
  112. release_sock(sk);
  113. return ret;
  114. }
  115. static int llcp_raw_sock_bind(struct socket *sock, struct sockaddr *addr,
  116. int alen)
  117. {
  118. struct sock *sk = sock->sk;
  119. struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
  120. struct nfc_llcp_local *local;
  121. struct nfc_dev *dev;
  122. struct sockaddr_nfc_llcp llcp_addr;
  123. int len, ret = 0;
  124. if (!addr || addr->sa_family != AF_NFC)
  125. return -EINVAL;
  126. pr_debug("sk %p addr %p family %d\n", sk, addr, addr->sa_family);
  127. memset(&llcp_addr, 0, sizeof(llcp_addr));
  128. len = min_t(unsigned int, sizeof(llcp_addr), alen);
  129. memcpy(&llcp_addr, addr, len);
  130. lock_sock(sk);
  131. if (sk->sk_state != LLCP_CLOSED) {
  132. ret = -EBADFD;
  133. goto error;
  134. }
  135. dev = nfc_get_device(llcp_addr.dev_idx);
  136. if (dev == NULL) {
  137. ret = -ENODEV;
  138. goto error;
  139. }
  140. local = nfc_llcp_find_local(dev);
  141. if (local == NULL) {
  142. ret = -ENODEV;
  143. goto put_dev;
  144. }
  145. llcp_sock->dev = dev;
  146. llcp_sock->local = nfc_llcp_local_get(local);
  147. llcp_sock->nfc_protocol = llcp_addr.nfc_protocol;
  148. nfc_llcp_sock_link(&local->raw_sockets, sk);
  149. sk->sk_state = LLCP_BOUND;
  150. put_dev:
  151. nfc_put_device(dev);
  152. error:
  153. release_sock(sk);
  154. return ret;
  155. }
  156. static int llcp_sock_listen(struct socket *sock, int backlog)
  157. {
  158. struct sock *sk = sock->sk;
  159. int ret = 0;
  160. pr_debug("sk %p backlog %d\n", sk, backlog);
  161. lock_sock(sk);
  162. if ((sock->type != SOCK_SEQPACKET && sock->type != SOCK_STREAM) ||
  163. sk->sk_state != LLCP_BOUND) {
  164. ret = -EBADFD;
  165. goto error;
  166. }
  167. sk->sk_max_ack_backlog = backlog;
  168. sk->sk_ack_backlog = 0;
  169. pr_debug("Socket listening\n");
  170. sk->sk_state = LLCP_LISTEN;
  171. error:
  172. release_sock(sk);
  173. return ret;
  174. }
  175. void nfc_llcp_accept_unlink(struct sock *sk)
  176. {
  177. struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
  178. pr_debug("state %d\n", sk->sk_state);
  179. list_del_init(&llcp_sock->accept_queue);
  180. sk_acceptq_removed(llcp_sock->parent);
  181. llcp_sock->parent = NULL;
  182. sock_put(sk);
  183. }
  184. void nfc_llcp_accept_enqueue(struct sock *parent, struct sock *sk)
  185. {
  186. struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
  187. struct nfc_llcp_sock *llcp_sock_parent = nfc_llcp_sock(parent);
  188. /* Lock will be free from unlink */
  189. sock_hold(sk);
  190. list_add_tail(&llcp_sock->accept_queue,
  191. &llcp_sock_parent->accept_queue);
  192. llcp_sock->parent = parent;
  193. sk_acceptq_added(parent);
  194. }
  195. struct sock *nfc_llcp_accept_dequeue(struct sock *parent,
  196. struct socket *newsock)
  197. {
  198. struct nfc_llcp_sock *lsk, *n, *llcp_parent;
  199. struct sock *sk;
  200. llcp_parent = nfc_llcp_sock(parent);
  201. list_for_each_entry_safe(lsk, n, &llcp_parent->accept_queue,
  202. accept_queue) {
  203. sk = &lsk->sk;
  204. lock_sock(sk);
  205. if (sk->sk_state == LLCP_CLOSED) {
  206. release_sock(sk);
  207. nfc_llcp_accept_unlink(sk);
  208. continue;
  209. }
  210. if (sk->sk_state == LLCP_CONNECTED || !newsock) {
  211. list_del_init(&lsk->accept_queue);
  212. sock_put(sk);
  213. if (newsock)
  214. sock_graft(sk, newsock);
  215. release_sock(sk);
  216. pr_debug("Returning sk state %d\n", sk->sk_state);
  217. sk_acceptq_removed(parent);
  218. return sk;
  219. }
  220. release_sock(sk);
  221. }
  222. return NULL;
  223. }
  224. static int llcp_sock_accept(struct socket *sock, struct socket *newsock,
  225. int flags)
  226. {
  227. DECLARE_WAITQUEUE(wait, current);
  228. struct sock *sk = sock->sk, *new_sk;
  229. long timeo;
  230. int ret = 0;
  231. pr_debug("parent %p\n", sk);
  232. lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
  233. if (sk->sk_state != LLCP_LISTEN) {
  234. ret = -EBADFD;
  235. goto error;
  236. }
  237. timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
  238. /* Wait for an incoming connection. */
  239. add_wait_queue_exclusive(sk_sleep(sk), &wait);
  240. while (!(new_sk = nfc_llcp_accept_dequeue(sk, newsock))) {
  241. set_current_state(TASK_INTERRUPTIBLE);
  242. if (!timeo) {
  243. ret = -EAGAIN;
  244. break;
  245. }
  246. if (signal_pending(current)) {
  247. ret = sock_intr_errno(timeo);
  248. break;
  249. }
  250. release_sock(sk);
  251. timeo = schedule_timeout(timeo);
  252. lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
  253. }
  254. __set_current_state(TASK_RUNNING);
  255. remove_wait_queue(sk_sleep(sk), &wait);
  256. if (ret)
  257. goto error;
  258. newsock->state = SS_CONNECTED;
  259. pr_debug("new socket %p\n", new_sk);
  260. error:
  261. release_sock(sk);
  262. return ret;
  263. }
  264. static int llcp_sock_getname(struct socket *sock, struct sockaddr *uaddr,
  265. int *len, int peer)
  266. {
  267. struct sock *sk = sock->sk;
  268. struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
  269. DECLARE_SOCKADDR(struct sockaddr_nfc_llcp *, llcp_addr, uaddr);
  270. if (llcp_sock == NULL || llcp_sock->dev == NULL)
  271. return -EBADFD;
  272. pr_debug("%p %d %d %d\n", sk, llcp_sock->target_idx,
  273. llcp_sock->dsap, llcp_sock->ssap);
  274. uaddr->sa_family = AF_NFC;
  275. *len = sizeof(struct sockaddr_nfc_llcp);
  276. llcp_addr->dev_idx = llcp_sock->dev->idx;
  277. llcp_addr->target_idx = llcp_sock->target_idx;
  278. llcp_addr->dsap = llcp_sock->dsap;
  279. llcp_addr->ssap = llcp_sock->ssap;
  280. llcp_addr->service_name_len = llcp_sock->service_name_len;
  281. memcpy(llcp_addr->service_name, llcp_sock->service_name,
  282. llcp_addr->service_name_len);
  283. return 0;
  284. }
  285. static inline unsigned int llcp_accept_poll(struct sock *parent)
  286. {
  287. struct nfc_llcp_sock *llcp_sock, *n, *parent_sock;
  288. struct sock *sk;
  289. parent_sock = nfc_llcp_sock(parent);
  290. list_for_each_entry_safe(llcp_sock, n, &parent_sock->accept_queue,
  291. accept_queue) {
  292. sk = &llcp_sock->sk;
  293. if (sk->sk_state == LLCP_CONNECTED)
  294. return POLLIN | POLLRDNORM;
  295. }
  296. return 0;
  297. }
  298. static unsigned int llcp_sock_poll(struct file *file, struct socket *sock,
  299. poll_table *wait)
  300. {
  301. struct sock *sk = sock->sk;
  302. unsigned int mask = 0;
  303. pr_debug("%p\n", sk);
  304. sock_poll_wait(file, sk_sleep(sk), wait);
  305. if (sk->sk_state == LLCP_LISTEN)
  306. return llcp_accept_poll(sk);
  307. if (sk->sk_err || !skb_queue_empty(&sk->sk_error_queue))
  308. mask |= POLLERR;
  309. if (!skb_queue_empty(&sk->sk_receive_queue))
  310. mask |= POLLIN | POLLRDNORM;
  311. if (sk->sk_state == LLCP_CLOSED)
  312. mask |= POLLHUP;
  313. if (sk->sk_shutdown & RCV_SHUTDOWN)
  314. mask |= POLLRDHUP | POLLIN | POLLRDNORM;
  315. if (sk->sk_shutdown == SHUTDOWN_MASK)
  316. mask |= POLLHUP;
  317. if (sock_writeable(sk))
  318. mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
  319. else
  320. set_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
  321. pr_debug("mask 0x%x\n", mask);
  322. return mask;
  323. }
  324. static int llcp_sock_release(struct socket *sock)
  325. {
  326. struct sock *sk = sock->sk;
  327. struct nfc_llcp_local *local;
  328. struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
  329. int err = 0;
  330. if (!sk)
  331. return 0;
  332. pr_debug("%p\n", sk);
  333. local = llcp_sock->local;
  334. if (local == NULL) {
  335. err = -ENODEV;
  336. goto out;
  337. }
  338. lock_sock(sk);
  339. /* Send a DISC */
  340. if (sk->sk_state == LLCP_CONNECTED)
  341. nfc_llcp_disconnect(llcp_sock);
  342. if (sk->sk_state == LLCP_LISTEN) {
  343. struct nfc_llcp_sock *lsk, *n;
  344. struct sock *accept_sk;
  345. list_for_each_entry_safe(lsk, n, &llcp_sock->accept_queue,
  346. accept_queue) {
  347. accept_sk = &lsk->sk;
  348. lock_sock(accept_sk);
  349. nfc_llcp_disconnect(lsk);
  350. nfc_llcp_accept_unlink(accept_sk);
  351. release_sock(accept_sk);
  352. }
  353. }
  354. if (llcp_sock->reserved_ssap < LLCP_SAP_MAX)
  355. nfc_llcp_put_ssap(llcp_sock->local, llcp_sock->ssap);
  356. release_sock(sk);
  357. if (sock->type == SOCK_RAW)
  358. nfc_llcp_sock_unlink(&local->raw_sockets, sk);
  359. else
  360. nfc_llcp_sock_unlink(&local->sockets, sk);
  361. out:
  362. sock_orphan(sk);
  363. sock_put(sk);
  364. return err;
  365. }
  366. static int llcp_sock_connect(struct socket *sock, struct sockaddr *_addr,
  367. int len, int flags)
  368. {
  369. struct sock *sk = sock->sk;
  370. struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
  371. struct sockaddr_nfc_llcp *addr = (struct sockaddr_nfc_llcp *)_addr;
  372. struct nfc_dev *dev;
  373. struct nfc_llcp_local *local;
  374. int ret = 0;
  375. pr_debug("sock %p sk %p flags 0x%x\n", sock, sk, flags);
  376. if (!addr || len < sizeof(struct sockaddr_nfc) ||
  377. addr->sa_family != AF_NFC)
  378. return -EINVAL;
  379. if (addr->service_name_len == 0 && addr->dsap == 0)
  380. return -EINVAL;
  381. pr_debug("addr dev_idx=%u target_idx=%u protocol=%u\n", addr->dev_idx,
  382. addr->target_idx, addr->nfc_protocol);
  383. lock_sock(sk);
  384. if (sk->sk_state == LLCP_CONNECTED) {
  385. ret = -EISCONN;
  386. goto error;
  387. }
  388. dev = nfc_get_device(addr->dev_idx);
  389. if (dev == NULL) {
  390. ret = -ENODEV;
  391. goto error;
  392. }
  393. local = nfc_llcp_find_local(dev);
  394. if (local == NULL) {
  395. ret = -ENODEV;
  396. goto put_dev;
  397. }
  398. device_lock(&dev->dev);
  399. if (dev->dep_link_up == false) {
  400. ret = -ENOLINK;
  401. device_unlock(&dev->dev);
  402. goto put_dev;
  403. }
  404. device_unlock(&dev->dev);
  405. if (local->rf_mode == NFC_RF_INITIATOR &&
  406. addr->target_idx != local->target_idx) {
  407. ret = -ENOLINK;
  408. goto put_dev;
  409. }
  410. llcp_sock->dev = dev;
  411. llcp_sock->local = nfc_llcp_local_get(local);
  412. llcp_sock->miu = llcp_sock->local->remote_miu;
  413. llcp_sock->ssap = nfc_llcp_get_local_ssap(local);
  414. if (llcp_sock->ssap == LLCP_SAP_MAX) {
  415. ret = -ENOMEM;
  416. goto put_dev;
  417. }
  418. llcp_sock->reserved_ssap = llcp_sock->ssap;
  419. if (addr->service_name_len == 0)
  420. llcp_sock->dsap = addr->dsap;
  421. else
  422. llcp_sock->dsap = LLCP_SAP_SDP;
  423. llcp_sock->nfc_protocol = addr->nfc_protocol;
  424. llcp_sock->service_name_len = min_t(unsigned int,
  425. addr->service_name_len,
  426. NFC_LLCP_MAX_SERVICE_NAME);
  427. llcp_sock->service_name = kmemdup(addr->service_name,
  428. llcp_sock->service_name_len,
  429. GFP_KERNEL);
  430. nfc_llcp_sock_link(&local->connecting_sockets, sk);
  431. ret = nfc_llcp_send_connect(llcp_sock);
  432. if (ret)
  433. goto sock_unlink;
  434. ret = sock_wait_state(sk, LLCP_CONNECTED,
  435. sock_sndtimeo(sk, flags & O_NONBLOCK));
  436. if (ret)
  437. goto sock_unlink;
  438. release_sock(sk);
  439. return 0;
  440. sock_unlink:
  441. nfc_llcp_put_ssap(local, llcp_sock->ssap);
  442. nfc_llcp_sock_unlink(&local->connecting_sockets, sk);
  443. put_dev:
  444. nfc_put_device(dev);
  445. error:
  446. release_sock(sk);
  447. return ret;
  448. }
  449. static int llcp_sock_sendmsg(struct kiocb *iocb, struct socket *sock,
  450. struct msghdr *msg, size_t len)
  451. {
  452. struct sock *sk = sock->sk;
  453. struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
  454. int ret;
  455. pr_debug("sock %p sk %p", sock, sk);
  456. ret = sock_error(sk);
  457. if (ret)
  458. return ret;
  459. if (msg->msg_flags & MSG_OOB)
  460. return -EOPNOTSUPP;
  461. lock_sock(sk);
  462. if (sk->sk_type == SOCK_DGRAM) {
  463. struct sockaddr_nfc_llcp *addr =
  464. (struct sockaddr_nfc_llcp *)msg->msg_name;
  465. if (msg->msg_namelen < sizeof(*addr)) {
  466. release_sock(sk);
  467. return -EINVAL;
  468. }
  469. release_sock(sk);
  470. return nfc_llcp_send_ui_frame(llcp_sock, addr->dsap, addr->ssap,
  471. msg, len);
  472. }
  473. if (sk->sk_state != LLCP_CONNECTED) {
  474. release_sock(sk);
  475. return -ENOTCONN;
  476. }
  477. release_sock(sk);
  478. return nfc_llcp_send_i_frame(llcp_sock, msg, len);
  479. }
  480. static int llcp_sock_recvmsg(struct kiocb *iocb, struct socket *sock,
  481. struct msghdr *msg, size_t len, int flags)
  482. {
  483. int noblock = flags & MSG_DONTWAIT;
  484. struct sock *sk = sock->sk;
  485. unsigned int copied, rlen;
  486. struct sk_buff *skb, *cskb;
  487. int err = 0;
  488. pr_debug("%p %zu\n", sk, len);
  489. msg->msg_namelen = 0;
  490. lock_sock(sk);
  491. if (sk->sk_state == LLCP_CLOSED &&
  492. skb_queue_empty(&sk->sk_receive_queue)) {
  493. release_sock(sk);
  494. return 0;
  495. }
  496. release_sock(sk);
  497. if (flags & (MSG_OOB))
  498. return -EOPNOTSUPP;
  499. skb = skb_recv_datagram(sk, flags, noblock, &err);
  500. if (!skb) {
  501. pr_err("Recv datagram failed state %d %d %d",
  502. sk->sk_state, err, sock_error(sk));
  503. if (sk->sk_shutdown & RCV_SHUTDOWN)
  504. return 0;
  505. return err;
  506. }
  507. rlen = skb->len; /* real length of skb */
  508. copied = min_t(unsigned int, rlen, len);
  509. cskb = skb;
  510. if (skb_copy_datagram_iovec(cskb, 0, msg->msg_iov, copied)) {
  511. if (!(flags & MSG_PEEK))
  512. skb_queue_head(&sk->sk_receive_queue, skb);
  513. return -EFAULT;
  514. }
  515. sock_recv_timestamp(msg, sk, skb);
  516. if (sk->sk_type == SOCK_DGRAM && msg->msg_name) {
  517. struct nfc_llcp_ui_cb *ui_cb = nfc_llcp_ui_skb_cb(skb);
  518. struct sockaddr_nfc_llcp *sockaddr =
  519. (struct sockaddr_nfc_llcp *) msg->msg_name;
  520. msg->msg_namelen = sizeof(struct sockaddr_nfc_llcp);
  521. pr_debug("Datagram socket %d %d\n", ui_cb->dsap, ui_cb->ssap);
  522. memset(sockaddr, 0, sizeof(*sockaddr));
  523. sockaddr->sa_family = AF_NFC;
  524. sockaddr->nfc_protocol = NFC_PROTO_NFC_DEP;
  525. sockaddr->dsap = ui_cb->dsap;
  526. sockaddr->ssap = ui_cb->ssap;
  527. }
  528. /* Mark read part of skb as used */
  529. if (!(flags & MSG_PEEK)) {
  530. /* SOCK_STREAM: re-queue skb if it contains unreceived data */
  531. if (sk->sk_type == SOCK_STREAM ||
  532. sk->sk_type == SOCK_DGRAM ||
  533. sk->sk_type == SOCK_RAW) {
  534. skb_pull(skb, copied);
  535. if (skb->len) {
  536. skb_queue_head(&sk->sk_receive_queue, skb);
  537. goto done;
  538. }
  539. }
  540. kfree_skb(skb);
  541. }
  542. /* XXX Queue backlogged skbs */
  543. done:
  544. /* SOCK_SEQPACKET: return real length if MSG_TRUNC is set */
  545. if (sk->sk_type == SOCK_SEQPACKET && (flags & MSG_TRUNC))
  546. copied = rlen;
  547. return copied;
  548. }
  549. static const struct proto_ops llcp_sock_ops = {
  550. .family = PF_NFC,
  551. .owner = THIS_MODULE,
  552. .bind = llcp_sock_bind,
  553. .connect = llcp_sock_connect,
  554. .release = llcp_sock_release,
  555. .socketpair = sock_no_socketpair,
  556. .accept = llcp_sock_accept,
  557. .getname = llcp_sock_getname,
  558. .poll = llcp_sock_poll,
  559. .ioctl = sock_no_ioctl,
  560. .listen = llcp_sock_listen,
  561. .shutdown = sock_no_shutdown,
  562. .setsockopt = sock_no_setsockopt,
  563. .getsockopt = sock_no_getsockopt,
  564. .sendmsg = llcp_sock_sendmsg,
  565. .recvmsg = llcp_sock_recvmsg,
  566. .mmap = sock_no_mmap,
  567. };
  568. static const struct proto_ops llcp_rawsock_ops = {
  569. .family = PF_NFC,
  570. .owner = THIS_MODULE,
  571. .bind = llcp_raw_sock_bind,
  572. .connect = sock_no_connect,
  573. .release = llcp_sock_release,
  574. .socketpair = sock_no_socketpair,
  575. .accept = sock_no_accept,
  576. .getname = llcp_sock_getname,
  577. .poll = llcp_sock_poll,
  578. .ioctl = sock_no_ioctl,
  579. .listen = sock_no_listen,
  580. .shutdown = sock_no_shutdown,
  581. .setsockopt = sock_no_setsockopt,
  582. .getsockopt = sock_no_getsockopt,
  583. .sendmsg = sock_no_sendmsg,
  584. .recvmsg = llcp_sock_recvmsg,
  585. .mmap = sock_no_mmap,
  586. };
  587. static void llcp_sock_destruct(struct sock *sk)
  588. {
  589. struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
  590. pr_debug("%p\n", sk);
  591. if (sk->sk_state == LLCP_CONNECTED)
  592. nfc_put_device(llcp_sock->dev);
  593. skb_queue_purge(&sk->sk_receive_queue);
  594. nfc_llcp_sock_free(llcp_sock);
  595. if (!sock_flag(sk, SOCK_DEAD)) {
  596. pr_err("Freeing alive NFC LLCP socket %p\n", sk);
  597. return;
  598. }
  599. }
  600. struct sock *nfc_llcp_sock_alloc(struct socket *sock, int type, gfp_t gfp)
  601. {
  602. struct sock *sk;
  603. struct nfc_llcp_sock *llcp_sock;
  604. sk = sk_alloc(&init_net, PF_NFC, gfp, &llcp_sock_proto);
  605. if (!sk)
  606. return NULL;
  607. llcp_sock = nfc_llcp_sock(sk);
  608. sock_init_data(sock, sk);
  609. sk->sk_state = LLCP_CLOSED;
  610. sk->sk_protocol = NFC_SOCKPROTO_LLCP;
  611. sk->sk_type = type;
  612. sk->sk_destruct = llcp_sock_destruct;
  613. llcp_sock->ssap = 0;
  614. llcp_sock->dsap = LLCP_SAP_SDP;
  615. llcp_sock->rw = LLCP_DEFAULT_RW;
  616. llcp_sock->miu = LLCP_DEFAULT_MIU;
  617. llcp_sock->send_n = llcp_sock->send_ack_n = 0;
  618. llcp_sock->recv_n = llcp_sock->recv_ack_n = 0;
  619. llcp_sock->remote_ready = 1;
  620. llcp_sock->reserved_ssap = LLCP_SAP_MAX;
  621. skb_queue_head_init(&llcp_sock->tx_queue);
  622. skb_queue_head_init(&llcp_sock->tx_pending_queue);
  623. INIT_LIST_HEAD(&llcp_sock->accept_queue);
  624. if (sock != NULL)
  625. sock->state = SS_UNCONNECTED;
  626. return sk;
  627. }
  628. void nfc_llcp_sock_free(struct nfc_llcp_sock *sock)
  629. {
  630. kfree(sock->service_name);
  631. skb_queue_purge(&sock->tx_queue);
  632. skb_queue_purge(&sock->tx_pending_queue);
  633. list_del_init(&sock->accept_queue);
  634. sock->parent = NULL;
  635. nfc_llcp_local_put(sock->local);
  636. }
  637. static int llcp_sock_create(struct net *net, struct socket *sock,
  638. const struct nfc_protocol *nfc_proto)
  639. {
  640. struct sock *sk;
  641. pr_debug("%p\n", sock);
  642. if (sock->type != SOCK_STREAM &&
  643. sock->type != SOCK_DGRAM &&
  644. sock->type != SOCK_RAW)
  645. return -ESOCKTNOSUPPORT;
  646. if (sock->type == SOCK_RAW)
  647. sock->ops = &llcp_rawsock_ops;
  648. else
  649. sock->ops = &llcp_sock_ops;
  650. sk = nfc_llcp_sock_alloc(sock, sock->type, GFP_ATOMIC);
  651. if (sk == NULL)
  652. return -ENOMEM;
  653. return 0;
  654. }
  655. static const struct nfc_protocol llcp_nfc_proto = {
  656. .id = NFC_SOCKPROTO_LLCP,
  657. .proto = &llcp_sock_proto,
  658. .owner = THIS_MODULE,
  659. .create = llcp_sock_create
  660. };
  661. int __init nfc_llcp_sock_init(void)
  662. {
  663. return nfc_proto_register(&llcp_nfc_proto);
  664. }
  665. void nfc_llcp_sock_exit(void)
  666. {
  667. nfc_proto_unregister(&llcp_nfc_proto);
  668. }