socket.c 18 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852
  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. newsk = sk->sk_prot->accept(sk, flags, &err);
  282. if (!newsk)
  283. return err;
  284. lock_sock(newsk);
  285. sock_graft(newsk, newsock);
  286. newsock->state = SS_CONNECTED;
  287. release_sock(newsk);
  288. return 0;
  289. }
  290. static int pn_socket_getname(struct socket *sock, struct sockaddr *addr,
  291. int *sockaddr_len, int peer)
  292. {
  293. struct sock *sk = sock->sk;
  294. struct pn_sock *pn = pn_sk(sk);
  295. memset(addr, 0, sizeof(struct sockaddr_pn));
  296. addr->sa_family = AF_PHONET;
  297. if (!peer) /* Race with bind() here is userland's problem. */
  298. pn_sockaddr_set_object((struct sockaddr_pn *)addr,
  299. pn->sobject);
  300. *sockaddr_len = sizeof(struct sockaddr_pn);
  301. return 0;
  302. }
  303. static unsigned int pn_socket_poll(struct file *file, struct socket *sock,
  304. poll_table *wait)
  305. {
  306. struct sock *sk = sock->sk;
  307. struct pep_sock *pn = pep_sk(sk);
  308. unsigned int mask = 0;
  309. poll_wait(file, sk_sleep(sk), wait);
  310. switch (sk->sk_state) {
  311. case TCP_LISTEN:
  312. return hlist_empty(&pn->ackq) ? 0 : POLLIN;
  313. case TCP_CLOSE:
  314. return POLLERR;
  315. }
  316. if (!skb_queue_empty(&sk->sk_receive_queue))
  317. mask |= POLLIN | POLLRDNORM;
  318. if (!skb_queue_empty(&pn->ctrlreq_queue))
  319. mask |= POLLPRI;
  320. if (!mask && sk->sk_state == TCP_CLOSE_WAIT)
  321. return POLLHUP;
  322. if (sk->sk_state == TCP_ESTABLISHED &&
  323. atomic_read(&sk->sk_wmem_alloc) < sk->sk_sndbuf &&
  324. atomic_read(&pn->tx_credits))
  325. mask |= POLLOUT | POLLWRNORM | POLLWRBAND;
  326. return mask;
  327. }
  328. static int pn_socket_ioctl(struct socket *sock, unsigned int cmd,
  329. unsigned long arg)
  330. {
  331. struct sock *sk = sock->sk;
  332. struct pn_sock *pn = pn_sk(sk);
  333. if (cmd == SIOCPNGETOBJECT) {
  334. struct net_device *dev;
  335. u16 handle;
  336. u8 saddr;
  337. if (get_user(handle, (__u16 __user *)arg))
  338. return -EFAULT;
  339. lock_sock(sk);
  340. if (sk->sk_bound_dev_if)
  341. dev = dev_get_by_index(sock_net(sk),
  342. sk->sk_bound_dev_if);
  343. else
  344. dev = phonet_device_get(sock_net(sk));
  345. if (dev && (dev->flags & IFF_UP))
  346. saddr = phonet_address_get(dev, pn_addr(handle));
  347. else
  348. saddr = PN_NO_ADDR;
  349. release_sock(sk);
  350. if (dev)
  351. dev_put(dev);
  352. if (saddr == PN_NO_ADDR)
  353. return -EHOSTUNREACH;
  354. handle = pn_object(saddr, pn_port(pn->sobject));
  355. return put_user(handle, (__u16 __user *)arg);
  356. }
  357. return sk->sk_prot->ioctl(sk, cmd, arg);
  358. }
  359. static int pn_socket_listen(struct socket *sock, int backlog)
  360. {
  361. struct sock *sk = sock->sk;
  362. int err = 0;
  363. if (sock->state != SS_UNCONNECTED)
  364. return -EINVAL;
  365. if (pn_socket_autobind(sock))
  366. return -ENOBUFS;
  367. lock_sock(sk);
  368. if (sk->sk_state != TCP_CLOSE) {
  369. err = -EINVAL;
  370. goto out;
  371. }
  372. sk->sk_state = TCP_LISTEN;
  373. sk->sk_ack_backlog = 0;
  374. sk->sk_max_ack_backlog = backlog;
  375. out:
  376. release_sock(sk);
  377. return err;
  378. }
  379. static int pn_socket_sendmsg(struct kiocb *iocb, struct socket *sock,
  380. struct msghdr *m, size_t total_len)
  381. {
  382. struct sock *sk = sock->sk;
  383. if (pn_socket_autobind(sock))
  384. return -EAGAIN;
  385. return sk->sk_prot->sendmsg(iocb, sk, m, total_len);
  386. }
  387. const struct proto_ops phonet_dgram_ops = {
  388. .family = AF_PHONET,
  389. .owner = THIS_MODULE,
  390. .release = pn_socket_release,
  391. .bind = pn_socket_bind,
  392. .connect = sock_no_connect,
  393. .socketpair = sock_no_socketpair,
  394. .accept = sock_no_accept,
  395. .getname = pn_socket_getname,
  396. .poll = datagram_poll,
  397. .ioctl = pn_socket_ioctl,
  398. .listen = sock_no_listen,
  399. .shutdown = sock_no_shutdown,
  400. .setsockopt = sock_no_setsockopt,
  401. .getsockopt = sock_no_getsockopt,
  402. #ifdef CONFIG_COMPAT
  403. .compat_setsockopt = sock_no_setsockopt,
  404. .compat_getsockopt = sock_no_getsockopt,
  405. #endif
  406. .sendmsg = pn_socket_sendmsg,
  407. .recvmsg = sock_common_recvmsg,
  408. .mmap = sock_no_mmap,
  409. .sendpage = sock_no_sendpage,
  410. };
  411. const struct proto_ops phonet_stream_ops = {
  412. .family = AF_PHONET,
  413. .owner = THIS_MODULE,
  414. .release = pn_socket_release,
  415. .bind = pn_socket_bind,
  416. #ifdef CONFIG_PHONET_PIPECTRLR
  417. .connect = pn_socket_connect,
  418. #else
  419. .connect = sock_no_connect,
  420. #endif
  421. .socketpair = sock_no_socketpair,
  422. .accept = pn_socket_accept,
  423. .getname = pn_socket_getname,
  424. .poll = pn_socket_poll,
  425. .ioctl = pn_socket_ioctl,
  426. .listen = pn_socket_listen,
  427. .shutdown = sock_no_shutdown,
  428. .setsockopt = sock_common_setsockopt,
  429. .getsockopt = sock_common_getsockopt,
  430. #ifdef CONFIG_COMPAT
  431. .compat_setsockopt = compat_sock_common_setsockopt,
  432. .compat_getsockopt = compat_sock_common_getsockopt,
  433. #endif
  434. .sendmsg = pn_socket_sendmsg,
  435. .recvmsg = sock_common_recvmsg,
  436. .mmap = sock_no_mmap,
  437. .sendpage = sock_no_sendpage,
  438. };
  439. EXPORT_SYMBOL(phonet_stream_ops);
  440. /* allocate port for a socket */
  441. int pn_sock_get_port(struct sock *sk, unsigned short sport)
  442. {
  443. static int port_cur;
  444. struct net *net = sock_net(sk);
  445. struct pn_sock *pn = pn_sk(sk);
  446. struct sockaddr_pn try_sa;
  447. struct sock *tmpsk;
  448. memset(&try_sa, 0, sizeof(struct sockaddr_pn));
  449. try_sa.spn_family = AF_PHONET;
  450. WARN_ON(!mutex_is_locked(&port_mutex));
  451. if (!sport) {
  452. /* search free port */
  453. int port, pmin, pmax;
  454. phonet_get_local_port_range(&pmin, &pmax);
  455. for (port = pmin; port <= pmax; port++) {
  456. port_cur++;
  457. if (port_cur < pmin || port_cur > pmax)
  458. port_cur = pmin;
  459. pn_sockaddr_set_port(&try_sa, port_cur);
  460. tmpsk = pn_find_sock_by_sa(net, &try_sa);
  461. if (tmpsk == NULL) {
  462. sport = port_cur;
  463. goto found;
  464. } else
  465. sock_put(tmpsk);
  466. }
  467. } else {
  468. /* try to find specific port */
  469. pn_sockaddr_set_port(&try_sa, sport);
  470. tmpsk = pn_find_sock_by_sa(net, &try_sa);
  471. if (tmpsk == NULL)
  472. /* No sock there! We can use that port... */
  473. goto found;
  474. else
  475. sock_put(tmpsk);
  476. }
  477. /* the port must be in use already */
  478. return -EADDRINUSE;
  479. found:
  480. pn->sobject = pn_object(pn_addr(pn->sobject), sport);
  481. return 0;
  482. }
  483. EXPORT_SYMBOL(pn_sock_get_port);
  484. #ifdef CONFIG_PROC_FS
  485. static struct sock *pn_sock_get_idx(struct seq_file *seq, loff_t pos)
  486. {
  487. struct net *net = seq_file_net(seq);
  488. struct hlist_head *hlist = pnsocks.hlist;
  489. struct hlist_node *node;
  490. struct sock *sknode;
  491. unsigned h;
  492. for (h = 0; h < PN_HASHSIZE; h++) {
  493. sk_for_each(sknode, node, hlist) {
  494. if (!net_eq(net, sock_net(sknode)))
  495. continue;
  496. if (!pos)
  497. return sknode;
  498. pos--;
  499. }
  500. hlist++;
  501. }
  502. return NULL;
  503. }
  504. static struct sock *pn_sock_get_next(struct seq_file *seq, struct sock *sk)
  505. {
  506. struct net *net = seq_file_net(seq);
  507. do
  508. sk = sk_next(sk);
  509. while (sk && !net_eq(net, sock_net(sk)));
  510. return sk;
  511. }
  512. static void *pn_sock_seq_start(struct seq_file *seq, loff_t *pos)
  513. __acquires(pnsocks.lock)
  514. {
  515. spin_lock_bh(&pnsocks.lock);
  516. return *pos ? pn_sock_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
  517. }
  518. static void *pn_sock_seq_next(struct seq_file *seq, void *v, loff_t *pos)
  519. {
  520. struct sock *sk;
  521. if (v == SEQ_START_TOKEN)
  522. sk = pn_sock_get_idx(seq, 0);
  523. else
  524. sk = pn_sock_get_next(seq, v);
  525. (*pos)++;
  526. return sk;
  527. }
  528. static void pn_sock_seq_stop(struct seq_file *seq, void *v)
  529. __releases(pnsocks.lock)
  530. {
  531. spin_unlock_bh(&pnsocks.lock);
  532. }
  533. static int pn_sock_seq_show(struct seq_file *seq, void *v)
  534. {
  535. int len;
  536. if (v == SEQ_START_TOKEN)
  537. seq_printf(seq, "%s%n", "pt loc rem rs st tx_queue rx_queue "
  538. " uid inode ref pointer drops", &len);
  539. else {
  540. struct sock *sk = v;
  541. struct pn_sock *pn = pn_sk(sk);
  542. seq_printf(seq, "%2d %04X:%04X:%02X %02X %08X:%08X %5d %lu "
  543. "%d %p %d%n",
  544. sk->sk_protocol, pn->sobject, 0, pn->resource,
  545. sk->sk_state,
  546. sk_wmem_alloc_get(sk), sk_rmem_alloc_get(sk),
  547. sock_i_uid(sk), sock_i_ino(sk),
  548. atomic_read(&sk->sk_refcnt), sk,
  549. atomic_read(&sk->sk_drops), &len);
  550. }
  551. seq_printf(seq, "%*s\n", 127 - len, "");
  552. return 0;
  553. }
  554. static const struct seq_operations pn_sock_seq_ops = {
  555. .start = pn_sock_seq_start,
  556. .next = pn_sock_seq_next,
  557. .stop = pn_sock_seq_stop,
  558. .show = pn_sock_seq_show,
  559. };
  560. static int pn_sock_open(struct inode *inode, struct file *file)
  561. {
  562. return seq_open_net(inode, file, &pn_sock_seq_ops,
  563. sizeof(struct seq_net_private));
  564. }
  565. const struct file_operations pn_sock_seq_fops = {
  566. .owner = THIS_MODULE,
  567. .open = pn_sock_open,
  568. .read = seq_read,
  569. .llseek = seq_lseek,
  570. .release = seq_release_net,
  571. };
  572. #endif
  573. static struct {
  574. struct sock *sk[256];
  575. } pnres;
  576. /*
  577. * Find and hold socket based on resource.
  578. */
  579. struct sock *pn_find_sock_by_res(struct net *net, u8 res)
  580. {
  581. struct sock *sk;
  582. if (!net_eq(net, &init_net))
  583. return NULL;
  584. rcu_read_lock();
  585. sk = rcu_dereference(pnres.sk[res]);
  586. if (sk)
  587. sock_hold(sk);
  588. rcu_read_unlock();
  589. return sk;
  590. }
  591. static DEFINE_MUTEX(resource_mutex);
  592. int pn_sock_bind_res(struct sock *sk, u8 res)
  593. {
  594. int ret = -EADDRINUSE;
  595. if (!net_eq(sock_net(sk), &init_net))
  596. return -ENOIOCTLCMD;
  597. if (!capable(CAP_SYS_ADMIN))
  598. return -EPERM;
  599. if (pn_socket_autobind(sk->sk_socket))
  600. return -EAGAIN;
  601. mutex_lock(&resource_mutex);
  602. if (pnres.sk[res] == NULL) {
  603. sock_hold(sk);
  604. rcu_assign_pointer(pnres.sk[res], sk);
  605. ret = 0;
  606. }
  607. mutex_unlock(&resource_mutex);
  608. return ret;
  609. }
  610. int pn_sock_unbind_res(struct sock *sk, u8 res)
  611. {
  612. int ret = -ENOENT;
  613. if (!capable(CAP_SYS_ADMIN))
  614. return -EPERM;
  615. mutex_lock(&resource_mutex);
  616. if (pnres.sk[res] == sk) {
  617. rcu_assign_pointer(pnres.sk[res], NULL);
  618. ret = 0;
  619. }
  620. mutex_unlock(&resource_mutex);
  621. if (ret == 0) {
  622. synchronize_rcu();
  623. sock_put(sk);
  624. }
  625. return ret;
  626. }
  627. void pn_sock_unbind_all_res(struct sock *sk)
  628. {
  629. unsigned res, match = 0;
  630. mutex_lock(&resource_mutex);
  631. for (res = 0; res < 256; res++) {
  632. if (pnres.sk[res] == sk) {
  633. rcu_assign_pointer(pnres.sk[res], NULL);
  634. match++;
  635. }
  636. }
  637. mutex_unlock(&resource_mutex);
  638. if (match == 0)
  639. return;
  640. synchronize_rcu();
  641. while (match > 0) {
  642. sock_put(sk);
  643. match--;
  644. }
  645. }
  646. #ifdef CONFIG_PROC_FS
  647. static struct sock **pn_res_get_idx(struct seq_file *seq, loff_t pos)
  648. {
  649. struct net *net = seq_file_net(seq);
  650. unsigned i;
  651. if (!net_eq(net, &init_net))
  652. return NULL;
  653. for (i = 0; i < 256; i++) {
  654. if (pnres.sk[i] == NULL)
  655. continue;
  656. if (!pos)
  657. return pnres.sk + i;
  658. pos--;
  659. }
  660. return NULL;
  661. }
  662. static struct sock **pn_res_get_next(struct seq_file *seq, struct sock **sk)
  663. {
  664. struct net *net = seq_file_net(seq);
  665. unsigned i;
  666. BUG_ON(!net_eq(net, &init_net));
  667. for (i = (sk - pnres.sk) + 1; i < 256; i++)
  668. if (pnres.sk[i])
  669. return pnres.sk + i;
  670. return NULL;
  671. }
  672. static void *pn_res_seq_start(struct seq_file *seq, loff_t *pos)
  673. __acquires(resource_mutex)
  674. {
  675. mutex_lock(&resource_mutex);
  676. return *pos ? pn_res_get_idx(seq, *pos - 1) : SEQ_START_TOKEN;
  677. }
  678. static void *pn_res_seq_next(struct seq_file *seq, void *v, loff_t *pos)
  679. {
  680. struct sock **sk;
  681. if (v == SEQ_START_TOKEN)
  682. sk = pn_res_get_idx(seq, 0);
  683. else
  684. sk = pn_res_get_next(seq, v);
  685. (*pos)++;
  686. return sk;
  687. }
  688. static void pn_res_seq_stop(struct seq_file *seq, void *v)
  689. __releases(resource_mutex)
  690. {
  691. mutex_unlock(&resource_mutex);
  692. }
  693. static int pn_res_seq_show(struct seq_file *seq, void *v)
  694. {
  695. int len;
  696. if (v == SEQ_START_TOKEN)
  697. seq_printf(seq, "%s%n", "rs uid inode", &len);
  698. else {
  699. struct sock **psk = v;
  700. struct sock *sk = *psk;
  701. seq_printf(seq, "%02X %5d %lu%n",
  702. (int) (psk - pnres.sk), sock_i_uid(sk),
  703. sock_i_ino(sk), &len);
  704. }
  705. seq_printf(seq, "%*s\n", 63 - len, "");
  706. return 0;
  707. }
  708. static const struct seq_operations pn_res_seq_ops = {
  709. .start = pn_res_seq_start,
  710. .next = pn_res_seq_next,
  711. .stop = pn_res_seq_stop,
  712. .show = pn_res_seq_show,
  713. };
  714. static int pn_res_open(struct inode *inode, struct file *file)
  715. {
  716. return seq_open_net(inode, file, &pn_res_seq_ops,
  717. sizeof(struct seq_net_private));
  718. }
  719. const struct file_operations pn_res_seq_fops = {
  720. .owner = THIS_MODULE,
  721. .open = pn_res_open,
  722. .read = seq_read,
  723. .llseek = seq_lseek,
  724. .release = seq_release_net,
  725. };
  726. #endif