sock.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874
  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. nfc_llcp_accept_unlink(sk);
  212. if (newsock)
  213. sock_graft(sk, newsock);
  214. release_sock(sk);
  215. pr_debug("Returning sk state %d\n", sk->sk_state);
  216. sk_acceptq_removed(parent);
  217. return sk;
  218. }
  219. release_sock(sk);
  220. }
  221. return NULL;
  222. }
  223. static int llcp_sock_accept(struct socket *sock, struct socket *newsock,
  224. int flags)
  225. {
  226. DECLARE_WAITQUEUE(wait, current);
  227. struct sock *sk = sock->sk, *new_sk;
  228. long timeo;
  229. int ret = 0;
  230. pr_debug("parent %p\n", sk);
  231. lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
  232. if (sk->sk_state != LLCP_LISTEN) {
  233. ret = -EBADFD;
  234. goto error;
  235. }
  236. timeo = sock_rcvtimeo(sk, flags & O_NONBLOCK);
  237. /* Wait for an incoming connection. */
  238. add_wait_queue_exclusive(sk_sleep(sk), &wait);
  239. while (!(new_sk = nfc_llcp_accept_dequeue(sk, newsock))) {
  240. set_current_state(TASK_INTERRUPTIBLE);
  241. if (!timeo) {
  242. ret = -EAGAIN;
  243. break;
  244. }
  245. if (signal_pending(current)) {
  246. ret = sock_intr_errno(timeo);
  247. break;
  248. }
  249. release_sock(sk);
  250. timeo = schedule_timeout(timeo);
  251. lock_sock_nested(sk, SINGLE_DEPTH_NESTING);
  252. }
  253. __set_current_state(TASK_RUNNING);
  254. remove_wait_queue(sk_sleep(sk), &wait);
  255. if (ret)
  256. goto error;
  257. newsock->state = SS_CONNECTED;
  258. pr_debug("new socket %p\n", new_sk);
  259. error:
  260. release_sock(sk);
  261. return ret;
  262. }
  263. static int llcp_sock_getname(struct socket *sock, struct sockaddr *uaddr,
  264. int *len, int peer)
  265. {
  266. struct sock *sk = sock->sk;
  267. struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
  268. DECLARE_SOCKADDR(struct sockaddr_nfc_llcp *, llcp_addr, uaddr);
  269. if (llcp_sock == NULL || llcp_sock->dev == NULL)
  270. return -EBADFD;
  271. pr_debug("%p %d %d %d\n", sk, llcp_sock->target_idx,
  272. llcp_sock->dsap, llcp_sock->ssap);
  273. uaddr->sa_family = AF_NFC;
  274. *len = sizeof(struct sockaddr_nfc_llcp);
  275. llcp_addr->dev_idx = llcp_sock->dev->idx;
  276. llcp_addr->target_idx = llcp_sock->target_idx;
  277. llcp_addr->dsap = llcp_sock->dsap;
  278. llcp_addr->ssap = llcp_sock->ssap;
  279. llcp_addr->service_name_len = llcp_sock->service_name_len;
  280. memcpy(llcp_addr->service_name, llcp_sock->service_name,
  281. llcp_addr->service_name_len);
  282. return 0;
  283. }
  284. static inline unsigned int llcp_accept_poll(struct sock *parent)
  285. {
  286. struct nfc_llcp_sock *llcp_sock, *n, *parent_sock;
  287. struct sock *sk;
  288. parent_sock = nfc_llcp_sock(parent);
  289. list_for_each_entry_safe(llcp_sock, n, &parent_sock->accept_queue,
  290. accept_queue) {
  291. sk = &llcp_sock->sk;
  292. if (sk->sk_state == LLCP_CONNECTED)
  293. return POLLIN | POLLRDNORM;
  294. }
  295. return 0;
  296. }
  297. static unsigned int llcp_sock_poll(struct file *file, struct socket *sock,
  298. poll_table *wait)
  299. {
  300. struct sock *sk = sock->sk;
  301. unsigned int mask = 0;
  302. pr_debug("%p\n", sk);
  303. sock_poll_wait(file, sk_sleep(sk), wait);
  304. if (sk->sk_state == LLCP_LISTEN)
  305. return llcp_accept_poll(sk);
  306. if (sk->sk_err || !skb_queue_empty(&sk->sk_error_queue))
  307. mask |= POLLERR;
  308. if (!skb_queue_empty(&sk->sk_receive_queue))
  309. mask |= POLLIN | POLLRDNORM;
  310. if (sk->sk_state == LLCP_CLOSED)
  311. mask |= POLLHUP;
  312. if (sk->sk_shutdown & RCV_SHUTDOWN)
  313. mask |= POLLRDHUP | POLLIN | POLLRDNORM;
  314. if (sk->sk_shutdown == SHUTDOWN_MASK)
  315. mask |= POLLHUP;
  316. if (sock_writeable(sk))
  317. mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
  318. else
  319. set_bit(SOCK_ASYNC_NOSPACE, &sk->sk_socket->flags);
  320. pr_debug("mask 0x%x\n", mask);
  321. return mask;
  322. }
  323. static int llcp_sock_release(struct socket *sock)
  324. {
  325. struct sock *sk = sock->sk;
  326. struct nfc_llcp_local *local;
  327. struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
  328. int err = 0;
  329. if (!sk)
  330. return 0;
  331. pr_debug("%p\n", sk);
  332. local = llcp_sock->local;
  333. if (local == NULL) {
  334. err = -ENODEV;
  335. goto out;
  336. }
  337. lock_sock(sk);
  338. /* Send a DISC */
  339. if (sk->sk_state == LLCP_CONNECTED)
  340. nfc_llcp_disconnect(llcp_sock);
  341. if (sk->sk_state == LLCP_LISTEN) {
  342. struct nfc_llcp_sock *lsk, *n;
  343. struct sock *accept_sk;
  344. list_for_each_entry_safe(lsk, n, &llcp_sock->accept_queue,
  345. accept_queue) {
  346. accept_sk = &lsk->sk;
  347. lock_sock(accept_sk);
  348. nfc_llcp_disconnect(lsk);
  349. nfc_llcp_accept_unlink(accept_sk);
  350. release_sock(accept_sk);
  351. sock_orphan(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. lock_sock(sk);
  490. if (sk->sk_state == LLCP_CLOSED &&
  491. skb_queue_empty(&sk->sk_receive_queue)) {
  492. release_sock(sk);
  493. return 0;
  494. }
  495. release_sock(sk);
  496. if (flags & (MSG_OOB))
  497. return -EOPNOTSUPP;
  498. skb = skb_recv_datagram(sk, flags, noblock, &err);
  499. if (!skb) {
  500. pr_err("Recv datagram failed state %d %d %d",
  501. sk->sk_state, err, sock_error(sk));
  502. if (sk->sk_shutdown & RCV_SHUTDOWN)
  503. return 0;
  504. return err;
  505. }
  506. rlen = skb->len; /* real length of skb */
  507. copied = min_t(unsigned int, rlen, len);
  508. cskb = skb;
  509. if (skb_copy_datagram_iovec(cskb, 0, msg->msg_iov, copied)) {
  510. if (!(flags & MSG_PEEK))
  511. skb_queue_head(&sk->sk_receive_queue, skb);
  512. return -EFAULT;
  513. }
  514. sock_recv_timestamp(msg, sk, skb);
  515. if (sk->sk_type == SOCK_DGRAM && msg->msg_name) {
  516. struct nfc_llcp_ui_cb *ui_cb = nfc_llcp_ui_skb_cb(skb);
  517. struct sockaddr_nfc_llcp *sockaddr =
  518. (struct sockaddr_nfc_llcp *) msg->msg_name;
  519. msg->msg_namelen = sizeof(struct sockaddr_nfc_llcp);
  520. pr_debug("Datagram socket %d %d\n", ui_cb->dsap, ui_cb->ssap);
  521. sockaddr->sa_family = AF_NFC;
  522. sockaddr->nfc_protocol = NFC_PROTO_NFC_DEP;
  523. sockaddr->dsap = ui_cb->dsap;
  524. sockaddr->ssap = ui_cb->ssap;
  525. }
  526. /* Mark read part of skb as used */
  527. if (!(flags & MSG_PEEK)) {
  528. /* SOCK_STREAM: re-queue skb if it contains unreceived data */
  529. if (sk->sk_type == SOCK_STREAM ||
  530. sk->sk_type == SOCK_DGRAM ||
  531. sk->sk_type == SOCK_RAW) {
  532. skb_pull(skb, copied);
  533. if (skb->len) {
  534. skb_queue_head(&sk->sk_receive_queue, skb);
  535. goto done;
  536. }
  537. }
  538. kfree_skb(skb);
  539. }
  540. /* XXX Queue backlogged skbs */
  541. done:
  542. /* SOCK_SEQPACKET: return real length if MSG_TRUNC is set */
  543. if (sk->sk_type == SOCK_SEQPACKET && (flags & MSG_TRUNC))
  544. copied = rlen;
  545. return copied;
  546. }
  547. static const struct proto_ops llcp_sock_ops = {
  548. .family = PF_NFC,
  549. .owner = THIS_MODULE,
  550. .bind = llcp_sock_bind,
  551. .connect = llcp_sock_connect,
  552. .release = llcp_sock_release,
  553. .socketpair = sock_no_socketpair,
  554. .accept = llcp_sock_accept,
  555. .getname = llcp_sock_getname,
  556. .poll = llcp_sock_poll,
  557. .ioctl = sock_no_ioctl,
  558. .listen = llcp_sock_listen,
  559. .shutdown = sock_no_shutdown,
  560. .setsockopt = sock_no_setsockopt,
  561. .getsockopt = sock_no_getsockopt,
  562. .sendmsg = llcp_sock_sendmsg,
  563. .recvmsg = llcp_sock_recvmsg,
  564. .mmap = sock_no_mmap,
  565. };
  566. static const struct proto_ops llcp_rawsock_ops = {
  567. .family = PF_NFC,
  568. .owner = THIS_MODULE,
  569. .bind = llcp_raw_sock_bind,
  570. .connect = sock_no_connect,
  571. .release = llcp_sock_release,
  572. .socketpair = sock_no_socketpair,
  573. .accept = sock_no_accept,
  574. .getname = llcp_sock_getname,
  575. .poll = llcp_sock_poll,
  576. .ioctl = sock_no_ioctl,
  577. .listen = sock_no_listen,
  578. .shutdown = sock_no_shutdown,
  579. .setsockopt = sock_no_setsockopt,
  580. .getsockopt = sock_no_getsockopt,
  581. .sendmsg = sock_no_sendmsg,
  582. .recvmsg = llcp_sock_recvmsg,
  583. .mmap = sock_no_mmap,
  584. };
  585. static void llcp_sock_destruct(struct sock *sk)
  586. {
  587. struct nfc_llcp_sock *llcp_sock = nfc_llcp_sock(sk);
  588. pr_debug("%p\n", sk);
  589. if (sk->sk_state == LLCP_CONNECTED)
  590. nfc_put_device(llcp_sock->dev);
  591. skb_queue_purge(&sk->sk_receive_queue);
  592. nfc_llcp_sock_free(llcp_sock);
  593. if (!sock_flag(sk, SOCK_DEAD)) {
  594. pr_err("Freeing alive NFC LLCP socket %p\n", sk);
  595. return;
  596. }
  597. }
  598. struct sock *nfc_llcp_sock_alloc(struct socket *sock, int type, gfp_t gfp)
  599. {
  600. struct sock *sk;
  601. struct nfc_llcp_sock *llcp_sock;
  602. sk = sk_alloc(&init_net, PF_NFC, gfp, &llcp_sock_proto);
  603. if (!sk)
  604. return NULL;
  605. llcp_sock = nfc_llcp_sock(sk);
  606. sock_init_data(sock, sk);
  607. sk->sk_state = LLCP_CLOSED;
  608. sk->sk_protocol = NFC_SOCKPROTO_LLCP;
  609. sk->sk_type = type;
  610. sk->sk_destruct = llcp_sock_destruct;
  611. llcp_sock->ssap = 0;
  612. llcp_sock->dsap = LLCP_SAP_SDP;
  613. llcp_sock->rw = LLCP_DEFAULT_RW;
  614. llcp_sock->miu = LLCP_DEFAULT_MIU;
  615. llcp_sock->send_n = llcp_sock->send_ack_n = 0;
  616. llcp_sock->recv_n = llcp_sock->recv_ack_n = 0;
  617. llcp_sock->remote_ready = 1;
  618. llcp_sock->reserved_ssap = LLCP_SAP_MAX;
  619. skb_queue_head_init(&llcp_sock->tx_queue);
  620. skb_queue_head_init(&llcp_sock->tx_pending_queue);
  621. INIT_LIST_HEAD(&llcp_sock->accept_queue);
  622. if (sock != NULL)
  623. sock->state = SS_UNCONNECTED;
  624. return sk;
  625. }
  626. void nfc_llcp_sock_free(struct nfc_llcp_sock *sock)
  627. {
  628. kfree(sock->service_name);
  629. skb_queue_purge(&sock->tx_queue);
  630. skb_queue_purge(&sock->tx_pending_queue);
  631. list_del_init(&sock->accept_queue);
  632. sock->parent = NULL;
  633. nfc_llcp_local_put(sock->local);
  634. }
  635. static int llcp_sock_create(struct net *net, struct socket *sock,
  636. const struct nfc_protocol *nfc_proto)
  637. {
  638. struct sock *sk;
  639. pr_debug("%p\n", sock);
  640. if (sock->type != SOCK_STREAM &&
  641. sock->type != SOCK_DGRAM &&
  642. sock->type != SOCK_RAW)
  643. return -ESOCKTNOSUPPORT;
  644. if (sock->type == SOCK_RAW)
  645. sock->ops = &llcp_rawsock_ops;
  646. else
  647. sock->ops = &llcp_sock_ops;
  648. sk = nfc_llcp_sock_alloc(sock, sock->type, GFP_ATOMIC);
  649. if (sk == NULL)
  650. return -ENOMEM;
  651. return 0;
  652. }
  653. static const struct nfc_protocol llcp_nfc_proto = {
  654. .id = NFC_SOCKPROTO_LLCP,
  655. .proto = &llcp_sock_proto,
  656. .owner = THIS_MODULE,
  657. .create = llcp_sock_create
  658. };
  659. int __init nfc_llcp_sock_init(void)
  660. {
  661. return nfc_proto_register(&llcp_nfc_proto);
  662. }
  663. void nfc_llcp_sock_exit(void)
  664. {
  665. nfc_proto_unregister(&llcp_nfc_proto);
  666. }