socket.c 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850
  1. /*
  2. * File: socket.c
  3. *
  4. * Phonet sockets
  5. *
  6. * Copyright (C) 2008 Nokia Corporation.
  7. *
  8. * Contact: Remi Denis-Courmont <remi.denis-courmont@nokia.com>
  9. * Original author: Sakari Ailus <sakari.ailus@nokia.com>
  10. *
  11. * This program is free software; you can redistribute it and/or
  12. * modify it under the terms of the GNU General Public License
  13. * version 2 as published by the Free Software Foundation.
  14. *
  15. * This program is distributed in the hope that it will be useful, but
  16. * WITHOUT ANY WARRANTY; without even the implied warranty of
  17. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  18. * General Public License for more details.
  19. *
  20. * You should have received a copy of the GNU General Public License
  21. * along with this program; if not, write to the Free Software
  22. * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
  23. * 02110-1301 USA
  24. */
  25. #include <linux/gfp.h>
  26. #include <linux/kernel.h>
  27. #include <linux/net.h>
  28. #include <linux/poll.h>
  29. #include <net/sock.h>
  30. #include <net/tcp_states.h>
  31. #include <linux/phonet.h>
  32. #include <net/phonet/phonet.h>
  33. #include <net/phonet/pep.h>
  34. #include <net/phonet/pn_dev.h>
  35. static int pn_socket_release(struct socket *sock)
  36. {
  37. struct sock *sk = sock->sk;
  38. if (sk) {
  39. sock->sk = NULL;
  40. sk->sk_prot->close(sk, 0);
  41. }
  42. return 0;
  43. }
  44. #define PN_HASHSIZE 16
  45. #define PN_HASHMASK (PN_HASHSIZE-1)
  46. static struct {
  47. struct hlist_head hlist[PN_HASHSIZE];
  48. spinlock_t lock;
  49. } pnsocks;
  50. void __init pn_sock_init(void)
  51. {
  52. unsigned i;
  53. for (i = 0; i < PN_HASHSIZE; i++)
  54. INIT_HLIST_HEAD(pnsocks.hlist + i);
  55. spin_lock_init(&pnsocks.lock);
  56. }
  57. static struct hlist_head *pn_hash_list(u16 obj)
  58. {
  59. return pnsocks.hlist + (obj & PN_HASHMASK);
  60. }
  61. /*
  62. * Find address based on socket address, match only certain fields.
  63. * Also grab sock if it was found. Remember to sock_put it later.
  64. */
  65. struct sock *pn_find_sock_by_sa(struct net *net, const struct sockaddr_pn *spn)
  66. {
  67. struct hlist_node *node;
  68. struct sock *sknode;
  69. struct sock *rval = NULL;
  70. u16 obj = pn_sockaddr_get_object(spn);
  71. u8 res = spn->spn_resource;
  72. struct hlist_head *hlist = pn_hash_list(obj);
  73. spin_lock_bh(&pnsocks.lock);
  74. sk_for_each(sknode, node, hlist) {
  75. struct pn_sock *pn = pn_sk(sknode);
  76. BUG_ON(!pn->sobject); /* unbound socket */
  77. if (!net_eq(sock_net(sknode), net))
  78. continue;
  79. if (pn_port(obj)) {
  80. /* Look up socket by port */
  81. if (pn_port(pn->sobject) != pn_port(obj))
  82. continue;
  83. } else {
  84. /* If port is zero, look up by resource */
  85. if (pn->resource != res)
  86. continue;
  87. }
  88. if (pn_addr(pn->sobject) &&
  89. pn_addr(pn->sobject) != pn_addr(obj))
  90. continue;
  91. rval = sknode;
  92. sock_hold(sknode);
  93. break;
  94. }
  95. spin_unlock_bh(&pnsocks.lock);
  96. return rval;
  97. }
  98. /* Deliver a broadcast packet (only in bottom-half) */
  99. void pn_deliver_sock_broadcast(struct net *net, struct sk_buff *skb)
  100. {
  101. struct hlist_head *hlist = pnsocks.hlist;
  102. unsigned h;
  103. spin_lock(&pnsocks.lock);
  104. for (h = 0; h < PN_HASHSIZE; h++) {
  105. struct hlist_node *node;
  106. struct sock *sknode;
  107. sk_for_each(sknode, node, hlist) {
  108. struct sk_buff *clone;
  109. if (!net_eq(sock_net(sknode), net))
  110. continue;
  111. if (!sock_flag(sknode, SOCK_BROADCAST))
  112. continue;
  113. clone = skb_clone(skb, GFP_ATOMIC);
  114. if (clone) {
  115. sock_hold(sknode);
  116. sk_receive_skb(sknode, clone, 0);
  117. }
  118. }
  119. hlist++;
  120. }
  121. spin_unlock(&pnsocks.lock);
  122. }
  123. void pn_sock_hash(struct sock *sk)
  124. {
  125. struct hlist_head *hlist = pn_hash_list(pn_sk(sk)->sobject);
  126. spin_lock_bh(&pnsocks.lock);
  127. sk_add_node(sk, hlist);
  128. spin_unlock_bh(&pnsocks.lock);
  129. }
  130. EXPORT_SYMBOL(pn_sock_hash);
  131. void pn_sock_unhash(struct sock *sk)
  132. {
  133. spin_lock_bh(&pnsocks.lock);
  134. sk_del_node_init(sk);
  135. spin_unlock_bh(&pnsocks.lock);
  136. pn_sock_unbind_all_res(sk);
  137. }
  138. EXPORT_SYMBOL(pn_sock_unhash);
  139. static DEFINE_MUTEX(port_mutex);
  140. static int pn_socket_bind(struct socket *sock, struct sockaddr *addr, int len)
  141. {
  142. struct sock *sk = sock->sk;
  143. struct pn_sock *pn = pn_sk(sk);
  144. struct sockaddr_pn *spn = (struct sockaddr_pn *)addr;
  145. int err;
  146. u16 handle;
  147. u8 saddr;
  148. if (sk->sk_prot->bind)
  149. return sk->sk_prot->bind(sk, addr, len);
  150. if (len < sizeof(struct sockaddr_pn))
  151. return -EINVAL;
  152. if (spn->spn_family != AF_PHONET)
  153. return -EAFNOSUPPORT;
  154. handle = pn_sockaddr_get_object((struct sockaddr_pn *)addr);
  155. saddr = pn_addr(handle);
  156. if (saddr && phonet_address_lookup(sock_net(sk), saddr))
  157. return -EADDRNOTAVAIL;
  158. lock_sock(sk);
  159. if (sk->sk_state != TCP_CLOSE || pn_port(pn->sobject)) {
  160. err = -EINVAL; /* attempt to rebind */
  161. goto out;
  162. }
  163. WARN_ON(sk_hashed(sk));
  164. mutex_lock(&port_mutex);
  165. err = sk->sk_prot->get_port(sk, pn_port(handle));
  166. if (err)
  167. goto out_port;
  168. /* get_port() sets the port, bind() sets the address if applicable */
  169. pn->sobject = pn_object(saddr, pn_port(pn->sobject));
  170. pn->resource = spn->spn_resource;
  171. /* Enable RX on the socket */
  172. sk->sk_prot->hash(sk);
  173. out_port:
  174. mutex_unlock(&port_mutex);
  175. out:
  176. release_sock(sk);
  177. return err;
  178. }
  179. static int pn_socket_autobind(struct socket *sock)
  180. {
  181. struct sockaddr_pn sa;
  182. int err;
  183. memset(&sa, 0, sizeof(sa));
  184. sa.spn_family = AF_PHONET;
  185. err = pn_socket_bind(sock, (struct sockaddr *)&sa,
  186. sizeof(struct sockaddr_pn));
  187. if (err != -EINVAL)
  188. return err;
  189. BUG_ON(!pn_port(pn_sk(sock->sk)->sobject));
  190. return 0; /* socket was already bound */
  191. }
  192. #ifdef CONFIG_PHONET_PIPECTRLR
  193. static int pn_socket_connect(struct socket *sock, struct sockaddr *addr,
  194. int len, int flags)
  195. {
  196. struct sock *sk = sock->sk;
  197. struct sockaddr_pn *spn = (struct sockaddr_pn *)addr;
  198. long timeo;
  199. int err;
  200. if (len < sizeof(struct sockaddr_pn))
  201. return -EINVAL;
  202. if (spn->spn_family != AF_PHONET)
  203. return -EAFNOSUPPORT;
  204. lock_sock(sk);
  205. switch (sock->state) {
  206. case SS_UNCONNECTED:
  207. sk->sk_state = TCP_CLOSE;
  208. break;
  209. case SS_CONNECTING:
  210. switch (sk->sk_state) {
  211. case TCP_SYN_RECV:
  212. sock->state = SS_CONNECTED;
  213. err = -EISCONN;
  214. goto out;
  215. case TCP_CLOSE:
  216. err = -EALREADY;
  217. if (flags & O_NONBLOCK)
  218. goto out;
  219. goto wait_connect;
  220. }
  221. break;
  222. case SS_CONNECTED:
  223. switch (sk->sk_state) {
  224. case TCP_SYN_RECV:
  225. err = -EISCONN;
  226. goto out;
  227. case TCP_CLOSE:
  228. sock->state = SS_UNCONNECTED;
  229. break;
  230. }
  231. break;
  232. case SS_DISCONNECTING:
  233. case SS_FREE:
  234. break;
  235. }
  236. sk->sk_state = TCP_CLOSE;
  237. sk_stream_kill_queues(sk);
  238. sock->state = SS_CONNECTING;
  239. err = sk->sk_prot->connect(sk, addr, len);
  240. if (err < 0) {
  241. sock->state = SS_UNCONNECTED;
  242. sk->sk_state = TCP_CLOSE;
  243. goto out;
  244. }
  245. err = -EINPROGRESS;
  246. wait_connect:
  247. if (sk->sk_state != TCP_SYN_RECV && (flags & O_NONBLOCK))
  248. goto out;
  249. timeo = sock_sndtimeo(sk, flags & O_NONBLOCK);
  250. release_sock(sk);
  251. err = -ERESTARTSYS;
  252. timeo = wait_event_interruptible_timeout(*sk_sleep(sk),
  253. sk->sk_state != TCP_CLOSE,
  254. timeo);
  255. lock_sock(sk);
  256. if (timeo < 0)
  257. goto out; /* -ERESTARTSYS */
  258. err = -ETIMEDOUT;
  259. if (timeo == 0 && sk->sk_state != TCP_SYN_RECV)
  260. goto out;
  261. if (sk->sk_state != TCP_SYN_RECV) {
  262. sock->state = SS_UNCONNECTED;
  263. err = sock_error(sk);
  264. if (!err)
  265. err = -ECONNREFUSED;
  266. goto out;
  267. }
  268. sock->state = SS_CONNECTED;
  269. err = 0;
  270. out:
  271. release_sock(sk);
  272. return err;
  273. }
  274. #endif
  275. static int pn_socket_accept(struct socket *sock, struct socket *newsock,
  276. int flags)
  277. {
  278. struct sock *sk = sock->sk;
  279. struct sock *newsk;
  280. int err;
  281. if (unlikely(sk->sk_state != TCP_LISTEN))
  282. return -EINVAL;
  283. newsk = sk->sk_prot->accept(sk, flags, &err);
  284. if (!newsk)
  285. return err;
  286. lock_sock(newsk);
  287. sock_graft(newsk, newsock);
  288. newsock->state = SS_CONNECTED;
  289. release_sock(newsk);
  290. return 0;
  291. }
  292. static int pn_socket_getname(struct socket *sock, struct sockaddr *addr,
  293. int *sockaddr_len, int peer)
  294. {
  295. struct sock *sk = sock->sk;
  296. struct pn_sock *pn = pn_sk(sk);
  297. memset(addr, 0, sizeof(struct sockaddr_pn));
  298. addr->sa_family = AF_PHONET;
  299. if (!peer) /* Race with bind() here is userland's problem. */
  300. pn_sockaddr_set_object((struct sockaddr_pn *)addr,
  301. pn->sobject);
  302. *sockaddr_len = sizeof(struct sockaddr_pn);
  303. return 0;
  304. }
  305. static unsigned int pn_socket_poll(struct file *file, struct socket *sock,
  306. poll_table *wait)
  307. {
  308. struct sock *sk = sock->sk;
  309. struct pep_sock *pn = pep_sk(sk);
  310. unsigned int mask = 0;
  311. poll_wait(file, sk_sleep(sk), wait);
  312. if (sk->sk_state == TCP_CLOSE)
  313. return POLLERR;
  314. if (!skb_queue_empty(&sk->sk_receive_queue))
  315. mask |= POLLIN | POLLRDNORM;
  316. if (!skb_queue_empty(&pn->ctrlreq_queue))
  317. mask |= POLLPRI;
  318. if (!mask && sk->sk_state == TCP_CLOSE_WAIT)
  319. return POLLHUP;
  320. if (sk->sk_state == TCP_ESTABLISHED &&
  321. atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf &&
  322. atomic_read(&pn->tx_credits))
  323. mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
  324. return mask;
  325. }
  326. static int pn_socket_ioctl(struct socket *sock, unsigned int cmd,
  327. unsigned long arg)
  328. {
  329. struct sock *sk = sock->sk;
  330. struct pn_sock *pn = pn_sk(sk);
  331. if (cmd == SIOCPNGETOBJECT) {
  332. struct net_device *dev;
  333. u16 handle;
  334. u8 saddr;
  335. if (get_user(handle, (__u16 __user *)arg))
  336. return -EFAULT;
  337. lock_sock(sk);
  338. if (sk->sk_bound_dev_if)
  339. dev = dev_get_by_index(sock_net(sk),
  340. sk->sk_bound_dev_if);
  341. else
  342. dev = phonet_device_get(sock_net(sk));
  343. if (dev && (dev->flags & IFF_UP))
  344. saddr = phonet_address_get(dev, pn_addr(handle));
  345. else
  346. saddr = PN_NO_ADDR;
  347. release_sock(sk);
  348. if (dev)
  349. dev_put(dev);
  350. if (saddr == PN_NO_ADDR)
  351. return -EHOSTUNREACH;
  352. handle = pn_object(saddr, pn_port(pn->sobject));
  353. return put_user(handle, (__u16 __user *)arg);
  354. }
  355. return sk->sk_prot->ioctl(sk, cmd, arg);
  356. }
  357. static int pn_socket_listen(struct socket *sock, int backlog)
  358. {
  359. struct sock *sk = sock->sk;
  360. int err = 0;
  361. if (pn_socket_autobind(sock))
  362. return -ENOBUFS;
  363. lock_sock(sk);
  364. if (sock->state != SS_UNCONNECTED) {
  365. err = -EINVAL;
  366. goto out;
  367. }
  368. if (sk->sk_state != TCP_LISTEN) {
  369. sk->sk_state = TCP_LISTEN;
  370. sk->sk_ack_backlog = 0;
  371. }
  372. sk->sk_max_ack_backlog = backlog;
  373. out:
  374. release_sock(sk);
  375. return err;
  376. }
  377. static int pn_socket_sendmsg(struct kiocb *iocb, struct socket *sock,
  378. struct msghdr *m, size_t total_len)
  379. {
  380. struct sock *sk = sock->sk;
  381. if (pn_socket_autobind(sock))
  382. return -EAGAIN;
  383. return sk->sk_prot->sendmsg(iocb, sk, m, total_len);
  384. }
  385. const struct proto_ops phonet_dgram_ops = {
  386. .family = AF_PHONET,
  387. .owner = THIS_MODULE,
  388. .release = pn_socket_release,
  389. .bind = pn_socket_bind,
  390. .connect = sock_no_connect,
  391. .socketpair = sock_no_socketpair,
  392. .accept = sock_no_accept,
  393. .getname = pn_socket_getname,
  394. .poll = datagram_poll,
  395. .ioctl = pn_socket_ioctl,
  396. .listen = sock_no_listen,
  397. .shutdown = sock_no_shutdown,
  398. .setsockopt = sock_no_setsockopt,
  399. .getsockopt = sock_no_getsockopt,
  400. #ifdef CONFIG_COMPAT
  401. .compat_setsockopt = sock_no_setsockopt,
  402. .compat_getsockopt = sock_no_getsockopt,
  403. #endif
  404. .sendmsg = pn_socket_sendmsg,
  405. .recvmsg = sock_common_recvmsg,
  406. .mmap = sock_no_mmap,
  407. .sendpage = sock_no_sendpage,
  408. };
  409. const struct proto_ops phonet_stream_ops = {
  410. .family = AF_PHONET,
  411. .owner = THIS_MODULE,
  412. .release = pn_socket_release,
  413. .bind = pn_socket_bind,
  414. #ifdef CONFIG_PHONET_PIPECTRLR
  415. .connect = pn_socket_connect,
  416. #else
  417. .connect = sock_no_connect,
  418. #endif
  419. .socketpair = sock_no_socketpair,
  420. .accept = pn_socket_accept,
  421. .getname = pn_socket_getname,
  422. .poll = pn_socket_poll,
  423. .ioctl = pn_socket_ioctl,
  424. .listen = pn_socket_listen,
  425. .shutdown = sock_no_shutdown,
  426. .setsockopt = sock_common_setsockopt,
  427. .getsockopt = sock_common_getsockopt,
  428. #ifdef CONFIG_COMPAT
  429. .compat_setsockopt = compat_sock_common_setsockopt,
  430. .compat_getsockopt = compat_sock_common_getsockopt,
  431. #endif
  432. .sendmsg = pn_socket_sendmsg,
  433. .recvmsg = sock_common_recvmsg,
  434. .mmap = sock_no_mmap,
  435. .sendpage = sock_no_sendpage,
  436. };
  437. EXPORT_SYMBOL(phonet_stream_ops);
  438. /* allocate port for a socket */
  439. int pn_sock_get_port(struct sock *sk, unsigned short sport)
  440. {
  441. static int port_cur;
  442. struct net *net = sock_net(sk);
  443. struct pn_sock *pn = pn_sk(sk);
  444. struct sockaddr_pn try_sa;
  445. struct sock *tmpsk;
  446. memset(&try_sa, 0, sizeof(struct sockaddr_pn));
  447. try_sa.spn_family = AF_PHONET;
  448. WARN_ON(!mutex_is_locked(&port_mutex));
  449. if (!sport) {
  450. /* search free port */
  451. int port, pmin, pmax;
  452. phonet_get_local_port_range(&pmin, &pmax);
  453. for (port = pmin; port <= pmax; port++) {
  454. port_cur++;
  455. if (port_cur < pmin || port_cur > pmax)
  456. port_cur = pmin;
  457. pn_sockaddr_set_port(&try_sa, port_cur);
  458. tmpsk = pn_find_sock_by_sa(net, &try_sa);
  459. if (tmpsk == NULL) {
  460. sport = port_cur;
  461. goto found;
  462. } else
  463. sock_put(tmpsk);
  464. }
  465. } else {
  466. /* try to find specific port */
  467. pn_sockaddr_set_port(&try_sa, sport);
  468. tmpsk = pn_find_sock_by_sa(net, &try_sa);
  469. if (tmpsk == NULL)
  470. /* No sock there! We can use that port... */
  471. goto found;
  472. else
  473. sock_put(tmpsk);
  474. }
  475. /* the port must be in use already */
  476. return -EADDRINUSE;
  477. found:
  478. pn->sobject = pn_object(pn_addr(pn->sobject), sport);
  479. return 0;
  480. }
  481. EXPORT_SYMBOL(pn_sock_get_port);
  482. #ifdef CONFIG_PROC_FS
  483. static struct sock *pn_sock_get_idx(struct seq_file *seq, loff_t pos)
  484. {
  485. struct net *net = seq_file_net(seq);
  486. struct hlist_head *hlist = pnsocks.hlist;
  487. struct hlist_node *node;
  488. struct sock *sknode;
  489. unsigned h;
  490. for (h = 0; h < PN_HASHSIZE; h++) {
  491. sk_for_each(sknode, node, hlist) {
  492. if (!net_eq(net, sock_net(sknode)))
  493. continue;
  494. if (!pos)
  495. return sknode;
  496. pos--;
  497. }
  498. hlist++;
  499. }
  500. return NULL;
  501. }
  502. static struct sock *pn_sock_get_next(struct seq_file *seq, struct sock *sk)
  503. {
  504. struct net *net = seq_file_net(seq);
  505. do
  506. sk = sk_next(sk);
  507. while (sk && !net_eq(net, sock_net(sk)));
  508. return sk;
  509. }
  510. static void *pn_sock_seq_start(struct seq_file *seq, loff_t *pos)
  511. __acquires(pnsocks.lock)
  512. {
  513. spin_lock_bh(&pnsocks.lock);
  514. return *pos ? pn_sock_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
  515. }
  516. static void *pn_sock_seq_next(struct seq_file *seq, void *v, loff_t *pos)
  517. {
  518. struct sock *sk;
  519. if (v == SEQ_START_TOKEN)
  520. sk = pn_sock_get_idx(seq, 0);
  521. else
  522. sk = pn_sock_get_next(seq, v);
  523. (*pos)++;
  524. return sk;
  525. }
  526. static void pn_sock_seq_stop(struct seq_file *seq, void *v)
  527. __releases(pnsocks.lock)
  528. {
  529. spin_unlock_bh(&pnsocks.lock);
  530. }
  531. static int pn_sock_seq_show(struct seq_file *seq, void *v)
  532. {
  533. int len;
  534. if (v == SEQ_START_TOKEN)
  535. seq_printf(seq, "%s%n", "pt loc rem rs st tx_queue rx_queue "
  536. " uid inode ref pointer drops", &len);
  537. else {
  538. struct sock *sk = v;
  539. struct pn_sock *pn = pn_sk(sk);
  540. seq_printf(seq, "%2d %04X:%04X:%02X %02X %08X:%08X %5d %lu "
  541. "%d %p %d%n",
  542. sk->sk_protocol, pn->sobject, pn->dobject,
  543. pn->resource, sk->sk_state,
  544. sk_wmem_alloc_get(sk), sk_rmem_alloc_get(sk),
  545. sock_i_uid(sk), sock_i_ino(sk),
  546. atomic_read(&sk->sk_refcnt), sk,
  547. atomic_read(&sk->sk_drops), &len);
  548. }
  549. seq_printf(seq, "%*s\n", 127 - len, "");
  550. return 0;
  551. }
  552. static const struct seq_operations pn_sock_seq_ops = {
  553. .start = pn_sock_seq_start,
  554. .next = pn_sock_seq_next,
  555. .stop = pn_sock_seq_stop,
  556. .show = pn_sock_seq_show,
  557. };
  558. static int pn_sock_open(struct inode *inode, struct file *file)
  559. {
  560. return seq_open_net(inode, file, &pn_sock_seq_ops,
  561. sizeof(struct seq_net_private));
  562. }
  563. const struct file_operations pn_sock_seq_fops = {
  564. .owner = THIS_MODULE,
  565. .open = pn_sock_open,
  566. .read = seq_read,
  567. .llseek = seq_lseek,
  568. .release = seq_release_net,
  569. };
  570. #endif
  571. static struct {
  572. struct sock *sk[256];
  573. } pnres;
  574. /*
  575. * Find and hold socket based on resource.
  576. */
  577. struct sock *pn_find_sock_by_res(struct net *net, u8 res)
  578. {
  579. struct sock *sk;
  580. if (!net_eq(net, &init_net))
  581. return NULL;
  582. rcu_read_lock();
  583. sk = rcu_dereference(pnres.sk[res]);
  584. if (sk)
  585. sock_hold(sk);
  586. rcu_read_unlock();
  587. return sk;
  588. }
  589. static DEFINE_MUTEX(resource_mutex);
  590. int pn_sock_bind_res(struct sock *sk, u8 res)
  591. {
  592. int ret = -EADDRINUSE;
  593. if (!net_eq(sock_net(sk), &init_net))
  594. return -ENOIOCTLCMD;
  595. if (!capable(CAP_SYS_ADMIN))
  596. return -EPERM;
  597. if (pn_socket_autobind(sk->sk_socket))
  598. return -EAGAIN;
  599. mutex_lock(&resource_mutex);
  600. if (pnres.sk[res] == NULL) {
  601. sock_hold(sk);
  602. rcu_assign_pointer(pnres.sk[res], sk);
  603. ret = 0;
  604. }
  605. mutex_unlock(&resource_mutex);
  606. return ret;
  607. }
  608. int pn_sock_unbind_res(struct sock *sk, u8 res)
  609. {
  610. int ret = -ENOENT;
  611. if (!capable(CAP_SYS_ADMIN))
  612. return -EPERM;
  613. mutex_lock(&resource_mutex);
  614. if (pnres.sk[res] == sk) {
  615. rcu_assign_pointer(pnres.sk[res], NULL);
  616. ret = 0;
  617. }
  618. mutex_unlock(&resource_mutex);
  619. if (ret == 0) {
  620. synchronize_rcu();
  621. sock_put(sk);
  622. }
  623. return ret;
  624. }
  625. void pn_sock_unbind_all_res(struct sock *sk)
  626. {
  627. unsigned res, match = 0;
  628. mutex_lock(&resource_mutex);
  629. for (res = 0; res < 256; res++) {
  630. if (pnres.sk[res] == sk) {
  631. rcu_assign_pointer(pnres.sk[res], NULL);
  632. match++;
  633. }
  634. }
  635. mutex_unlock(&resource_mutex);
  636. if (match == 0)
  637. return;
  638. synchronize_rcu();
  639. while (match > 0) {
  640. sock_put(sk);
  641. match--;
  642. }
  643. }
  644. #ifdef CONFIG_PROC_FS
  645. static struct sock **pn_res_get_idx(struct seq_file *seq, loff_t pos)
  646. {
  647. struct net *net = seq_file_net(seq);
  648. unsigned i;
  649. if (!net_eq(net, &init_net))
  650. return NULL;
  651. for (i = 0; i < 256; i++) {
  652. if (pnres.sk[i] == NULL)
  653. continue;
  654. if (!pos)
  655. return pnres.sk + i;
  656. pos--;
  657. }
  658. return NULL;
  659. }
  660. static struct sock **pn_res_get_next(struct seq_file *seq, struct sock **sk)
  661. {
  662. struct net *net = seq_file_net(seq);
  663. unsigned i;
  664. BUG_ON(!net_eq(net, &init_net));
  665. for (i = (sk - pnres.sk) + 1; i < 256; i++)
  666. if (pnres.sk[i])
  667. return pnres.sk + i;
  668. return NULL;
  669. }
  670. static void *pn_res_seq_start(struct seq_file *seq, loff_t *pos)
  671. __acquires(resource_mutex)
  672. {
  673. mutex_lock(&resource_mutex);
  674. return *pos ? pn_res_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
  675. }
  676. static void *pn_res_seq_next(struct seq_file *seq, void *v, loff_t *pos)
  677. {
  678. struct sock **sk;
  679. if (v == SEQ_START_TOKEN)
  680. sk = pn_res_get_idx(seq, 0);
  681. else
  682. sk = pn_res_get_next(seq, v);
  683. (*pos)++;
  684. return sk;
  685. }
  686. static void pn_res_seq_stop(struct seq_file *seq, void *v)
  687. __releases(resource_mutex)
  688. {
  689. mutex_unlock(&resource_mutex);
  690. }
  691. static int pn_res_seq_show(struct seq_file *seq, void *v)
  692. {
  693. int len;
  694. if (v == SEQ_START_TOKEN)
  695. seq_printf(seq, "%s%n", "rs uid inode", &len);
  696. else {
  697. struct sock **psk = v;
  698. struct sock *sk = *psk;
  699. seq_printf(seq, "%02X %5d %lu%n",
  700. (int) (psk - pnres.sk), sock_i_uid(sk),
  701. sock_i_ino(sk), &len);
  702. }
  703. seq_printf(seq, "%*s\n", 63 - len, "");
  704. return 0;
  705. }
  706. static const struct seq_operations pn_res_seq_ops = {
  707. .start = pn_res_seq_start,
  708. .next = pn_res_seq_next,
  709. .stop = pn_res_seq_stop,
  710. .show = pn_res_seq_show,
  711. };
  712. static int pn_res_open(struct inode *inode, struct file *file)
  713. {
  714. return seq_open_net(inode, file, &pn_res_seq_ops,
  715. sizeof(struct seq_net_private));
  716. }
  717. const struct file_operations pn_res_seq_fops = {
  718. .owner = THIS_MODULE,
  719. .open = pn_res_open,
  720. .read = seq_read,
  721. .llseek = seq_lseek,
  722. .release = seq_release_net,
  723. };
  724. #endif