pep.c 24 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076
  1. /*
  2. * File: pep.c
  3. *
  4. * Phonet pipe protocol end point socket
  5. *
  6. * Copyright (C) 2008 Nokia Corporation.
  7. *
  8. * Author: Rémi Denis-Courmont <remi.denis-courmont@nokia.com>
  9. *
  10. * This program is free software; you can redistribute it and/or
  11. * modify it under the terms of the GNU General Public License
  12. * version 2 as published by the Free Software Foundation.
  13. *
  14. * This program is distributed in the hope that it will be useful, but
  15. * WITHOUT ANY WARRANTY; without even the implied warranty of
  16. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
  17. * General Public License for more details.
  18. *
  19. * You should have received a copy of the GNU General Public License
  20. * along with this program; if not, write to the Free Software
  21. * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
  22. * 02110-1301 USA
  23. */
  24. #include <linux/kernel.h>
  25. #include <linux/socket.h>
  26. #include <net/sock.h>
  27. #include <net/tcp_states.h>
  28. #include <asm/ioctls.h>
  29. #include <linux/phonet.h>
  30. #include <net/phonet/phonet.h>
  31. #include <net/phonet/pep.h>
  32. #include <net/phonet/gprs.h>
  33. /* sk_state values:
  34. * TCP_CLOSE sock not in use yet
  35. * TCP_CLOSE_WAIT disconnected pipe
  36. * TCP_LISTEN listening pipe endpoint
  37. * TCP_SYN_RECV connected pipe in disabled state
  38. * TCP_ESTABLISHED connected pipe in enabled state
  39. *
  40. * pep_sock locking:
  41. * - sk_state, ackq, hlist: sock lock needed
  42. * - listener: read only
  43. * - pipe_handle: read only
  44. */
  45. #define CREDITS_MAX 10
  46. #define CREDITS_THR 7
  47. static const struct sockaddr_pn pipe_srv = {
  48. .spn_family = AF_PHONET,
  49. .spn_resource = 0xD9, /* pipe service */
  50. };
  51. #define pep_sb_size(s) (((s) + 5) & ~3) /* 2-bytes head, 32-bits aligned */
  52. /* Get the next TLV sub-block. */
  53. static unsigned char *pep_get_sb(struct sk_buff *skb, u8 *ptype, u8 *plen,
  54. void *buf)
  55. {
  56. void *data = NULL;
  57. struct {
  58. u8 sb_type;
  59. u8 sb_len;
  60. } *ph, h;
  61. int buflen = *plen;
  62. ph = skb_header_pointer(skb, 0, 2, &h);
  63. if (ph == NULL || ph->sb_len < 2 || !pskb_may_pull(skb, ph->sb_len))
  64. return NULL;
  65. ph->sb_len -= 2;
  66. *ptype = ph->sb_type;
  67. *plen = ph->sb_len;
  68. if (buflen > ph->sb_len)
  69. buflen = ph->sb_len;
  70. data = skb_header_pointer(skb, 2, buflen, buf);
  71. __skb_pull(skb, 2 + ph->sb_len);
  72. return data;
  73. }
  74. static int pep_reply(struct sock *sk, struct sk_buff *oskb,
  75. u8 code, const void *data, int len, gfp_t priority)
  76. {
  77. const struct pnpipehdr *oph = pnp_hdr(oskb);
  78. struct pnpipehdr *ph;
  79. struct sk_buff *skb;
  80. skb = alloc_skb(MAX_PNPIPE_HEADER + len, priority);
  81. if (!skb)
  82. return -ENOMEM;
  83. skb_set_owner_w(skb, sk);
  84. skb_reserve(skb, MAX_PNPIPE_HEADER);
  85. __skb_put(skb, len);
  86. skb_copy_to_linear_data(skb, data, len);
  87. __skb_push(skb, sizeof(*ph));
  88. skb_reset_transport_header(skb);
  89. ph = pnp_hdr(skb);
  90. ph->utid = oph->utid;
  91. ph->message_id = oph->message_id + 1; /* REQ -> RESP */
  92. ph->pipe_handle = oph->pipe_handle;
  93. ph->error_code = code;
  94. return pn_skb_send(sk, skb, &pipe_srv);
  95. }
  96. #define PAD 0x00
  97. static int pep_accept_conn(struct sock *sk, struct sk_buff *skb)
  98. {
  99. static const u8 data[20] = {
  100. PAD, PAD, PAD, 2 /* sub-blocks */,
  101. PN_PIPE_SB_REQUIRED_FC_TX, pep_sb_size(5), 3, PAD,
  102. PN_MULTI_CREDIT_FLOW_CONTROL,
  103. PN_ONE_CREDIT_FLOW_CONTROL,
  104. PN_LEGACY_FLOW_CONTROL,
  105. PAD,
  106. PN_PIPE_SB_PREFERRED_FC_RX, pep_sb_size(5), 3, PAD,
  107. PN_MULTI_CREDIT_FLOW_CONTROL,
  108. PN_ONE_CREDIT_FLOW_CONTROL,
  109. PN_LEGACY_FLOW_CONTROL,
  110. PAD,
  111. };
  112. might_sleep();
  113. return pep_reply(sk, skb, PN_PIPE_NO_ERROR, data, sizeof(data),
  114. GFP_KERNEL);
  115. }
  116. static int pep_reject_conn(struct sock *sk, struct sk_buff *skb, u8 code)
  117. {
  118. static const u8 data[4] = { PAD, PAD, PAD, 0 /* sub-blocks */ };
  119. WARN_ON(code == PN_PIPE_NO_ERROR);
  120. return pep_reply(sk, skb, code, data, sizeof(data), GFP_ATOMIC);
  121. }
  122. /* Control requests are not sent by the pipe service and have a specific
  123. * message format. */
  124. static int pep_ctrlreq_error(struct sock *sk, struct sk_buff *oskb, u8 code,
  125. gfp_t priority)
  126. {
  127. const struct pnpipehdr *oph = pnp_hdr(oskb);
  128. struct sk_buff *skb;
  129. struct pnpipehdr *ph;
  130. struct sockaddr_pn dst;
  131. skb = alloc_skb(MAX_PNPIPE_HEADER + 4, priority);
  132. if (!skb)
  133. return -ENOMEM;
  134. skb_set_owner_w(skb, sk);
  135. skb_reserve(skb, MAX_PHONET_HEADER);
  136. ph = (struct pnpipehdr *)skb_put(skb, sizeof(*ph) + 4);
  137. ph->utid = oph->utid;
  138. ph->message_id = PNS_PEP_CTRL_RESP;
  139. ph->pipe_handle = oph->pipe_handle;
  140. ph->data[0] = oph->data[1]; /* CTRL id */
  141. ph->data[1] = oph->data[0]; /* PEP type */
  142. ph->data[2] = code; /* error code, at an usual offset */
  143. ph->data[3] = PAD;
  144. ph->data[4] = PAD;
  145. pn_skb_get_src_sockaddr(oskb, &dst);
  146. return pn_skb_send(sk, skb, &dst);
  147. }
  148. static int pipe_snd_status(struct sock *sk, u8 type, u8 status, gfp_t priority)
  149. {
  150. struct pep_sock *pn = pep_sk(sk);
  151. struct pnpipehdr *ph;
  152. struct sk_buff *skb;
  153. skb = alloc_skb(MAX_PNPIPE_HEADER + 4, priority);
  154. if (!skb)
  155. return -ENOMEM;
  156. skb_set_owner_w(skb, sk);
  157. skb_reserve(skb, MAX_PNPIPE_HEADER + 4);
  158. __skb_push(skb, sizeof(*ph) + 4);
  159. skb_reset_transport_header(skb);
  160. ph = pnp_hdr(skb);
  161. ph->utid = 0;
  162. ph->message_id = PNS_PEP_STATUS_IND;
  163. ph->pipe_handle = pn->pipe_handle;
  164. ph->pep_type = PN_PEP_TYPE_COMMON;
  165. ph->data[1] = type;
  166. ph->data[2] = PAD;
  167. ph->data[3] = PAD;
  168. ph->data[4] = status;
  169. return pn_skb_send(sk, skb, &pipe_srv);
  170. }
  171. /* Send our RX flow control information to the sender.
  172. * Socket must be locked. */
  173. static void pipe_grant_credits(struct sock *sk)
  174. {
  175. struct pep_sock *pn = pep_sk(sk);
  176. BUG_ON(sk->sk_state != TCP_ESTABLISHED);
  177. switch (pn->rx_fc) {
  178. case PN_LEGACY_FLOW_CONTROL: /* TODO */
  179. break;
  180. case PN_ONE_CREDIT_FLOW_CONTROL:
  181. pipe_snd_status(sk, PN_PEP_IND_FLOW_CONTROL,
  182. PEP_IND_READY, GFP_ATOMIC);
  183. pn->rx_credits = 1;
  184. break;
  185. case PN_MULTI_CREDIT_FLOW_CONTROL:
  186. if ((pn->rx_credits + CREDITS_THR) > CREDITS_MAX)
  187. break;
  188. if (pipe_snd_status(sk, PN_PEP_IND_ID_MCFC_GRANT_CREDITS,
  189. CREDITS_MAX - pn->rx_credits,
  190. GFP_ATOMIC) == 0)
  191. pn->rx_credits = CREDITS_MAX;
  192. break;
  193. }
  194. }
  195. static int pipe_rcv_status(struct sock *sk, struct sk_buff *skb)
  196. {
  197. struct pep_sock *pn = pep_sk(sk);
  198. struct pnpipehdr *hdr = pnp_hdr(skb);
  199. if (!pskb_may_pull(skb, sizeof(*hdr) + 4))
  200. return -EINVAL;
  201. if (hdr->data[0] != PN_PEP_TYPE_COMMON) {
  202. LIMIT_NETDEBUG(KERN_DEBUG"Phonet unknown PEP type: %u\n",
  203. (unsigned)hdr->data[0]);
  204. return -EOPNOTSUPP;
  205. }
  206. switch (hdr->data[1]) {
  207. case PN_PEP_IND_FLOW_CONTROL:
  208. switch (pn->tx_fc) {
  209. case PN_LEGACY_FLOW_CONTROL:
  210. switch (hdr->data[4]) {
  211. case PEP_IND_BUSY:
  212. pn->tx_credits = 0;
  213. break;
  214. case PEP_IND_READY:
  215. pn->tx_credits = 1;
  216. break;
  217. }
  218. break;
  219. case PN_ONE_CREDIT_FLOW_CONTROL:
  220. if (hdr->data[4] == PEP_IND_READY)
  221. pn->tx_credits = 1;
  222. break;
  223. }
  224. break;
  225. case PN_PEP_IND_ID_MCFC_GRANT_CREDITS:
  226. if (pn->tx_fc != PN_MULTI_CREDIT_FLOW_CONTROL)
  227. break;
  228. if (pn->tx_credits + hdr->data[4] > 0xff)
  229. pn->tx_credits = 0xff;
  230. else
  231. pn->tx_credits += hdr->data[4];
  232. break;
  233. default:
  234. LIMIT_NETDEBUG(KERN_DEBUG"Phonet unknown PEP indication: %u\n",
  235. (unsigned)hdr->data[1]);
  236. return -EOPNOTSUPP;
  237. }
  238. if (pn->tx_credits)
  239. sk->sk_write_space(sk);
  240. return 0;
  241. }
  242. static int pipe_rcv_created(struct sock *sk, struct sk_buff *skb)
  243. {
  244. struct pep_sock *pn = pep_sk(sk);
  245. struct pnpipehdr *hdr = pnp_hdr(skb);
  246. u8 n_sb = hdr->data[0];
  247. pn->rx_fc = pn->tx_fc = PN_LEGACY_FLOW_CONTROL;
  248. __skb_pull(skb, sizeof(*hdr));
  249. while (n_sb > 0) {
  250. u8 type, buf[2], len = sizeof(buf);
  251. u8 *data = pep_get_sb(skb, &type, &len, buf);
  252. if (data == NULL)
  253. return -EINVAL;
  254. switch (type) {
  255. case PN_PIPE_SB_NEGOTIATED_FC:
  256. if (len < 2 || (data[0] | data[1]) > 3)
  257. break;
  258. pn->tx_fc = data[0] & 3;
  259. pn->rx_fc = data[1] & 3;
  260. break;
  261. }
  262. n_sb--;
  263. }
  264. return 0;
  265. }
  266. /* Queue an skb to a connected sock.
  267. * Socket lock must be held. */
  268. static int pipe_do_rcv(struct sock *sk, struct sk_buff *skb)
  269. {
  270. struct pep_sock *pn = pep_sk(sk);
  271. struct pnpipehdr *hdr = pnp_hdr(skb);
  272. struct sk_buff_head *queue;
  273. int err = 0;
  274. BUG_ON(sk->sk_state == TCP_CLOSE_WAIT);
  275. switch (hdr->message_id) {
  276. case PNS_PEP_CONNECT_REQ:
  277. pep_reject_conn(sk, skb, PN_PIPE_ERR_PEP_IN_USE);
  278. break;
  279. case PNS_PEP_DISCONNECT_REQ:
  280. pep_reply(sk, skb, PN_PIPE_NO_ERROR, NULL, 0, GFP_ATOMIC);
  281. sk->sk_state = TCP_CLOSE_WAIT;
  282. if (!sock_flag(sk, SOCK_DEAD))
  283. sk->sk_state_change(sk);
  284. break;
  285. case PNS_PEP_ENABLE_REQ:
  286. /* Wait for PNS_PIPE_(ENABLED|REDIRECTED)_IND */
  287. pep_reply(sk, skb, PN_PIPE_NO_ERROR, NULL, 0, GFP_ATOMIC);
  288. break;
  289. case PNS_PEP_RESET_REQ:
  290. switch (hdr->state_after_reset) {
  291. case PN_PIPE_DISABLE:
  292. pn->init_enable = 0;
  293. break;
  294. case PN_PIPE_ENABLE:
  295. pn->init_enable = 1;
  296. break;
  297. default: /* not allowed to send an error here!? */
  298. err = -EINVAL;
  299. goto out;
  300. }
  301. /* fall through */
  302. case PNS_PEP_DISABLE_REQ:
  303. pn->tx_credits = 0;
  304. pep_reply(sk, skb, PN_PIPE_NO_ERROR, NULL, 0, GFP_ATOMIC);
  305. break;
  306. case PNS_PEP_CTRL_REQ:
  307. if (skb_queue_len(&pn->ctrlreq_queue) >= PNPIPE_CTRLREQ_MAX)
  308. break;
  309. __skb_pull(skb, 4);
  310. queue = &pn->ctrlreq_queue;
  311. goto queue;
  312. case PNS_PIPE_DATA:
  313. __skb_pull(skb, 3); /* Pipe data header */
  314. if (!pn_flow_safe(pn->rx_fc)) {
  315. err = sock_queue_rcv_skb(sk, skb);
  316. if (!err)
  317. return 0;
  318. break;
  319. }
  320. if (pn->rx_credits == 0) {
  321. err = -ENOBUFS;
  322. break;
  323. }
  324. pn->rx_credits--;
  325. queue = &sk->sk_receive_queue;
  326. goto queue;
  327. case PNS_PEP_STATUS_IND:
  328. pipe_rcv_status(sk, skb);
  329. break;
  330. case PNS_PIPE_REDIRECTED_IND:
  331. err = pipe_rcv_created(sk, skb);
  332. break;
  333. case PNS_PIPE_CREATED_IND:
  334. err = pipe_rcv_created(sk, skb);
  335. if (err)
  336. break;
  337. /* fall through */
  338. case PNS_PIPE_RESET_IND:
  339. if (!pn->init_enable)
  340. break;
  341. /* fall through */
  342. case PNS_PIPE_ENABLED_IND:
  343. if (!pn_flow_safe(pn->tx_fc)) {
  344. pn->tx_credits = 1;
  345. sk->sk_write_space(sk);
  346. }
  347. if (sk->sk_state == TCP_ESTABLISHED)
  348. break; /* Nothing to do */
  349. sk->sk_state = TCP_ESTABLISHED;
  350. pipe_grant_credits(sk);
  351. break;
  352. case PNS_PIPE_DISABLED_IND:
  353. sk->sk_state = TCP_SYN_RECV;
  354. pn->rx_credits = 0;
  355. break;
  356. default:
  357. LIMIT_NETDEBUG(KERN_DEBUG"Phonet unknown PEP message: %u\n",
  358. hdr->message_id);
  359. err = -EINVAL;
  360. }
  361. out:
  362. kfree_skb(skb);
  363. return err;
  364. queue:
  365. skb->dev = NULL;
  366. skb_set_owner_r(skb, sk);
  367. err = skb->len;
  368. skb_queue_tail(queue, skb);
  369. if (!sock_flag(sk, SOCK_DEAD))
  370. sk->sk_data_ready(sk, err);
  371. return 0;
  372. }
  373. /* Destroy connected sock. */
  374. static void pipe_destruct(struct sock *sk)
  375. {
  376. struct pep_sock *pn = pep_sk(sk);
  377. skb_queue_purge(&sk->sk_receive_queue);
  378. skb_queue_purge(&pn->ctrlreq_queue);
  379. }
  380. static int pep_connreq_rcv(struct sock *sk, struct sk_buff *skb)
  381. {
  382. struct sock *newsk;
  383. struct pep_sock *newpn, *pn = pep_sk(sk);
  384. struct pnpipehdr *hdr;
  385. struct sockaddr_pn dst;
  386. u16 peer_type;
  387. u8 pipe_handle, enabled, n_sb;
  388. if (!pskb_pull(skb, sizeof(*hdr) + 4))
  389. return -EINVAL;
  390. hdr = pnp_hdr(skb);
  391. pipe_handle = hdr->pipe_handle;
  392. switch (hdr->state_after_connect) {
  393. case PN_PIPE_DISABLE:
  394. enabled = 0;
  395. break;
  396. case PN_PIPE_ENABLE:
  397. enabled = 1;
  398. break;
  399. default:
  400. pep_reject_conn(sk, skb, PN_PIPE_ERR_INVALID_PARAM);
  401. return -EINVAL;
  402. }
  403. peer_type = hdr->other_pep_type << 8;
  404. if (unlikely(sk->sk_state != TCP_LISTEN) || sk_acceptq_is_full(sk)) {
  405. pep_reject_conn(sk, skb, PN_PIPE_ERR_PEP_IN_USE);
  406. return -ENOBUFS;
  407. }
  408. /* Parse sub-blocks (options) */
  409. n_sb = hdr->data[4];
  410. while (n_sb > 0) {
  411. u8 type, buf[1], len = sizeof(buf);
  412. const u8 *data = pep_get_sb(skb, &type, &len, buf);
  413. if (data == NULL)
  414. return -EINVAL;
  415. switch (type) {
  416. case PN_PIPE_SB_CONNECT_REQ_PEP_SUB_TYPE:
  417. if (len < 1)
  418. return -EINVAL;
  419. peer_type = (peer_type & 0xff00) | data[0];
  420. break;
  421. }
  422. n_sb--;
  423. }
  424. skb = skb_clone(skb, GFP_ATOMIC);
  425. if (!skb)
  426. return -ENOMEM;
  427. /* Create a new to-be-accepted sock */
  428. newsk = sk_alloc(sock_net(sk), PF_PHONET, GFP_ATOMIC, sk->sk_prot);
  429. if (!newsk) {
  430. kfree_skb(skb);
  431. return -ENOMEM;
  432. }
  433. sock_init_data(NULL, newsk);
  434. newsk->sk_state = TCP_SYN_RECV;
  435. newsk->sk_backlog_rcv = pipe_do_rcv;
  436. newsk->sk_protocol = sk->sk_protocol;
  437. newsk->sk_destruct = pipe_destruct;
  438. newpn = pep_sk(newsk);
  439. pn_skb_get_dst_sockaddr(skb, &dst);
  440. newpn->pn_sk.sobject = pn_sockaddr_get_object(&dst);
  441. newpn->pn_sk.resource = pn->pn_sk.resource;
  442. skb_queue_head_init(&newpn->ctrlreq_queue);
  443. newpn->pipe_handle = pipe_handle;
  444. newpn->peer_type = peer_type;
  445. newpn->rx_credits = newpn->tx_credits = 0;
  446. newpn->rx_fc = newpn->tx_fc = PN_LEGACY_FLOW_CONTROL;
  447. newpn->init_enable = enabled;
  448. BUG_ON(!skb_queue_empty(&newsk->sk_receive_queue));
  449. skb_queue_head(&newsk->sk_receive_queue, skb);
  450. if (!sock_flag(sk, SOCK_DEAD))
  451. sk->sk_data_ready(sk, 0);
  452. sk_acceptq_added(sk);
  453. sk_add_node(newsk, &pn->ackq);
  454. return 0;
  455. }
  456. /* Listening sock must be locked */
  457. static struct sock *pep_find_pipe(const struct hlist_head *hlist,
  458. const struct sockaddr_pn *dst,
  459. u8 pipe_handle)
  460. {
  461. struct hlist_node *node;
  462. struct sock *sknode;
  463. u16 dobj = pn_sockaddr_get_object(dst);
  464. sk_for_each(sknode, node, hlist) {
  465. struct pep_sock *pnnode = pep_sk(sknode);
  466. /* Ports match, but addresses might not: */
  467. if (pnnode->pn_sk.sobject != dobj)
  468. continue;
  469. if (pnnode->pipe_handle != pipe_handle)
  470. continue;
  471. if (sknode->sk_state == TCP_CLOSE_WAIT)
  472. continue;
  473. sock_hold(sknode);
  474. return sknode;
  475. }
  476. return NULL;
  477. }
  478. /*
  479. * Deliver an skb to a listening sock.
  480. * Socket lock must be held.
  481. * We then queue the skb to the right connected sock (if any).
  482. */
  483. static int pep_do_rcv(struct sock *sk, struct sk_buff *skb)
  484. {
  485. struct pep_sock *pn = pep_sk(sk);
  486. struct sock *sknode;
  487. struct pnpipehdr *hdr = pnp_hdr(skb);
  488. struct sockaddr_pn dst;
  489. int err = NET_RX_SUCCESS;
  490. u8 pipe_handle;
  491. if (!pskb_may_pull(skb, sizeof(*hdr)))
  492. goto drop;
  493. hdr = pnp_hdr(skb);
  494. pipe_handle = hdr->pipe_handle;
  495. if (pipe_handle == PN_PIPE_INVALID_HANDLE)
  496. goto drop;
  497. pn_skb_get_dst_sockaddr(skb, &dst);
  498. /* Look for an existing pipe handle */
  499. sknode = pep_find_pipe(&pn->hlist, &dst, pipe_handle);
  500. if (sknode)
  501. return sk_receive_skb(sknode, skb, 1);
  502. /* Look for a pipe handle pending accept */
  503. sknode = pep_find_pipe(&pn->ackq, &dst, pipe_handle);
  504. if (sknode) {
  505. sock_put(sknode);
  506. if (net_ratelimit())
  507. printk(KERN_WARNING"Phonet unconnected PEP ignored");
  508. err = NET_RX_DROP;
  509. goto drop;
  510. }
  511. switch (hdr->message_id) {
  512. case PNS_PEP_CONNECT_REQ:
  513. err = pep_connreq_rcv(sk, skb);
  514. break;
  515. case PNS_PEP_DISCONNECT_REQ:
  516. pep_reply(sk, skb, PN_PIPE_NO_ERROR, NULL, 0, GFP_ATOMIC);
  517. break;
  518. case PNS_PEP_CTRL_REQ:
  519. pep_ctrlreq_error(sk, skb, PN_PIPE_INVALID_HANDLE, GFP_ATOMIC);
  520. break;
  521. case PNS_PEP_RESET_REQ:
  522. case PNS_PEP_ENABLE_REQ:
  523. case PNS_PEP_DISABLE_REQ:
  524. /* invalid handle is not even allowed here! */
  525. default:
  526. err = NET_RX_DROP;
  527. }
  528. drop:
  529. kfree_skb(skb);
  530. return err;
  531. }
  532. /* associated socket ceases to exist */
  533. static void pep_sock_close(struct sock *sk, long timeout)
  534. {
  535. struct pep_sock *pn = pep_sk(sk);
  536. int ifindex = 0;
  537. sk_common_release(sk);
  538. lock_sock(sk);
  539. if (sk->sk_state == TCP_LISTEN) {
  540. /* Destroy the listen queue */
  541. struct sock *sknode;
  542. struct hlist_node *p, *n;
  543. sk_for_each_safe(sknode, p, n, &pn->ackq)
  544. sk_del_node_init(sknode);
  545. sk->sk_state = TCP_CLOSE;
  546. }
  547. ifindex = pn->ifindex;
  548. pn->ifindex = 0;
  549. release_sock(sk);
  550. if (ifindex)
  551. gprs_detach(sk);
  552. }
  553. static int pep_wait_connreq(struct sock *sk, int noblock)
  554. {
  555. struct task_struct *tsk = current;
  556. struct pep_sock *pn = pep_sk(sk);
  557. long timeo = sock_rcvtimeo(sk, noblock);
  558. for (;;) {
  559. DEFINE_WAIT(wait);
  560. if (sk->sk_state != TCP_LISTEN)
  561. return -EINVAL;
  562. if (!hlist_empty(&pn->ackq))
  563. break;
  564. if (!timeo)
  565. return -EWOULDBLOCK;
  566. if (signal_pending(tsk))
  567. return sock_intr_errno(timeo);
  568. prepare_to_wait_exclusive(&sk->sk_socket->wait, &wait,
  569. TASK_INTERRUPTIBLE);
  570. release_sock(sk);
  571. timeo = schedule_timeout(timeo);
  572. lock_sock(sk);
  573. finish_wait(&sk->sk_socket->wait, &wait);
  574. }
  575. return 0;
  576. }
  577. static struct sock *pep_sock_accept(struct sock *sk, int flags, int *errp)
  578. {
  579. struct pep_sock *pn = pep_sk(sk);
  580. struct sock *newsk = NULL;
  581. struct sk_buff *oskb;
  582. int err;
  583. lock_sock(sk);
  584. err = pep_wait_connreq(sk, flags & O_NONBLOCK);
  585. if (err)
  586. goto out;
  587. newsk = __sk_head(&pn->ackq);
  588. oskb = skb_dequeue(&newsk->sk_receive_queue);
  589. err = pep_accept_conn(newsk, oskb);
  590. if (err) {
  591. skb_queue_head(&newsk->sk_receive_queue, oskb);
  592. newsk = NULL;
  593. goto out;
  594. }
  595. sock_hold(sk);
  596. pep_sk(newsk)->listener = sk;
  597. sock_hold(newsk);
  598. sk_del_node_init(newsk);
  599. sk_acceptq_removed(sk);
  600. sk_add_node(newsk, &pn->hlist);
  601. __sock_put(newsk);
  602. out:
  603. release_sock(sk);
  604. *errp = err;
  605. return newsk;
  606. }
  607. static int pep_ioctl(struct sock *sk, int cmd, unsigned long arg)
  608. {
  609. struct pep_sock *pn = pep_sk(sk);
  610. int answ;
  611. switch (cmd) {
  612. case SIOCINQ:
  613. if (sk->sk_state == TCP_LISTEN)
  614. return -EINVAL;
  615. lock_sock(sk);
  616. if (sock_flag(sk, SOCK_URGINLINE)
  617. && !skb_queue_empty(&pn->ctrlreq_queue))
  618. answ = skb_peek(&pn->ctrlreq_queue)->len;
  619. else if (!skb_queue_empty(&sk->sk_receive_queue))
  620. answ = skb_peek(&sk->sk_receive_queue)->len;
  621. else
  622. answ = 0;
  623. release_sock(sk);
  624. return put_user(answ, (int __user *)arg);
  625. }
  626. return -ENOIOCTLCMD;
  627. }
  628. static int pep_init(struct sock *sk)
  629. {
  630. struct pep_sock *pn = pep_sk(sk);
  631. INIT_HLIST_HEAD(&pn->ackq);
  632. INIT_HLIST_HEAD(&pn->hlist);
  633. skb_queue_head_init(&pn->ctrlreq_queue);
  634. pn->pipe_handle = PN_PIPE_INVALID_HANDLE;
  635. return 0;
  636. }
  637. static int pep_setsockopt(struct sock *sk, int level, int optname,
  638. char __user *optval, int optlen)
  639. {
  640. struct pep_sock *pn = pep_sk(sk);
  641. int val = 0, err = 0;
  642. if (level != SOL_PNPIPE)
  643. return -ENOPROTOOPT;
  644. if (optlen >= sizeof(int)) {
  645. if (get_user(val, (int __user *) optval))
  646. return -EFAULT;
  647. }
  648. lock_sock(sk);
  649. switch (optname) {
  650. case PNPIPE_ENCAP:
  651. if (val && val != PNPIPE_ENCAP_IP) {
  652. err = -EINVAL;
  653. break;
  654. }
  655. if (!pn->ifindex == !val)
  656. break; /* Nothing to do! */
  657. if (!capable(CAP_NET_ADMIN)) {
  658. err = -EPERM;
  659. break;
  660. }
  661. if (val) {
  662. release_sock(sk);
  663. err = gprs_attach(sk);
  664. if (err > 0) {
  665. pn->ifindex = err;
  666. err = 0;
  667. }
  668. } else {
  669. pn->ifindex = 0;
  670. release_sock(sk);
  671. gprs_detach(sk);
  672. err = 0;
  673. }
  674. goto out_norel;
  675. default:
  676. err = -ENOPROTOOPT;
  677. }
  678. release_sock(sk);
  679. out_norel:
  680. return err;
  681. }
  682. static int pep_getsockopt(struct sock *sk, int level, int optname,
  683. char __user *optval, int __user *optlen)
  684. {
  685. struct pep_sock *pn = pep_sk(sk);
  686. int len, val;
  687. if (level != SOL_PNPIPE)
  688. return -ENOPROTOOPT;
  689. if (get_user(len, optlen))
  690. return -EFAULT;
  691. switch (optname) {
  692. case PNPIPE_ENCAP:
  693. val = pn->ifindex ? PNPIPE_ENCAP_IP : PNPIPE_ENCAP_NONE;
  694. break;
  695. case PNPIPE_IFINDEX:
  696. val = pn->ifindex;
  697. break;
  698. default:
  699. return -ENOPROTOOPT;
  700. }
  701. len = min_t(unsigned int, sizeof(int), len);
  702. if (put_user(len, optlen))
  703. return -EFAULT;
  704. if (put_user(val, (int __user *) optval))
  705. return -EFAULT;
  706. return 0;
  707. }
  708. static int pipe_skb_send(struct sock *sk, struct sk_buff *skb)
  709. {
  710. struct pep_sock *pn = pep_sk(sk);
  711. struct pnpipehdr *ph;
  712. skb_push(skb, 3);
  713. skb_reset_transport_header(skb);
  714. ph = pnp_hdr(skb);
  715. ph->utid = 0;
  716. ph->message_id = PNS_PIPE_DATA;
  717. ph->pipe_handle = pn->pipe_handle;
  718. if (pn_flow_safe(pn->tx_fc) && pn->tx_credits)
  719. pn->tx_credits--;
  720. return pn_skb_send(sk, skb, &pipe_srv);
  721. }
  722. static int pep_sendmsg(struct kiocb *iocb, struct sock *sk,
  723. struct msghdr *msg, size_t len)
  724. {
  725. struct pep_sock *pn = pep_sk(sk);
  726. struct sk_buff *skb = NULL;
  727. long timeo;
  728. int flags = msg->msg_flags;
  729. int err, done;
  730. if (msg->msg_flags & MSG_OOB || !(msg->msg_flags & MSG_EOR))
  731. return -EOPNOTSUPP;
  732. lock_sock(sk);
  733. timeo = sock_sndtimeo(sk, flags & MSG_DONTWAIT);
  734. if ((1 << sk->sk_state) & (TCPF_LISTEN|TCPF_CLOSE)) {
  735. err = -ENOTCONN;
  736. goto out;
  737. }
  738. if (sk->sk_state != TCP_ESTABLISHED) {
  739. /* Wait until the pipe gets to enabled state */
  740. disabled:
  741. err = sk_stream_wait_connect(sk, &timeo);
  742. if (err)
  743. goto out;
  744. if (sk->sk_state == TCP_CLOSE_WAIT) {
  745. err = -ECONNRESET;
  746. goto out;
  747. }
  748. }
  749. BUG_ON(sk->sk_state != TCP_ESTABLISHED);
  750. /* Wait until flow control allows TX */
  751. done = pn->tx_credits > 0;
  752. while (!done) {
  753. DEFINE_WAIT(wait);
  754. if (!timeo) {
  755. err = -EAGAIN;
  756. goto out;
  757. }
  758. if (signal_pending(current)) {
  759. err = sock_intr_errno(timeo);
  760. goto out;
  761. }
  762. prepare_to_wait(&sk->sk_socket->wait, &wait,
  763. TASK_INTERRUPTIBLE);
  764. done = sk_wait_event(sk, &timeo, pn->tx_credits > 0);
  765. finish_wait(&sk->sk_socket->wait, &wait);
  766. if (sk->sk_state != TCP_ESTABLISHED)
  767. goto disabled;
  768. }
  769. if (!skb) {
  770. skb = sock_alloc_send_skb(sk, MAX_PNPIPE_HEADER + len,
  771. flags & MSG_DONTWAIT, &err);
  772. if (skb == NULL)
  773. goto out;
  774. skb_reserve(skb, MAX_PHONET_HEADER + 3);
  775. if (sk->sk_state != TCP_ESTABLISHED || !pn->tx_credits)
  776. goto disabled; /* sock_alloc_send_skb might sleep */
  777. }
  778. err = memcpy_fromiovec(skb_put(skb, len), msg->msg_iov, len);
  779. if (err < 0)
  780. goto out;
  781. err = pipe_skb_send(sk, skb);
  782. if (err >= 0)
  783. err = len; /* success! */
  784. skb = NULL;
  785. out:
  786. release_sock(sk);
  787. kfree_skb(skb);
  788. return err;
  789. }
  790. int pep_writeable(struct sock *sk)
  791. {
  792. struct pep_sock *pn = pep_sk(sk);
  793. return (sk->sk_state == TCP_ESTABLISHED) ? pn->tx_credits : 0;
  794. }
  795. int pep_write(struct sock *sk, struct sk_buff *skb)
  796. {
  797. struct sk_buff *rskb, *fs;
  798. int flen = 0;
  799. rskb = alloc_skb(MAX_PNPIPE_HEADER, GFP_ATOMIC);
  800. if (!rskb) {
  801. kfree_skb(skb);
  802. return -ENOMEM;
  803. }
  804. skb_shinfo(rskb)->frag_list = skb;
  805. rskb->len += skb->len;
  806. rskb->data_len += rskb->len;
  807. rskb->truesize += rskb->len;
  808. /* Avoid nested fragments */
  809. for (fs = skb_shinfo(skb)->frag_list; fs; fs = fs->next)
  810. flen += fs->len;
  811. skb->next = skb_shinfo(skb)->frag_list;
  812. skb_shinfo(skb)->frag_list = NULL;
  813. skb->len -= flen;
  814. skb->data_len -= flen;
  815. skb->truesize -= flen;
  816. skb_reserve(rskb, MAX_PHONET_HEADER + 3);
  817. return pipe_skb_send(sk, rskb);
  818. }
  819. struct sk_buff *pep_read(struct sock *sk)
  820. {
  821. struct sk_buff *skb = skb_dequeue(&sk->sk_receive_queue);
  822. if (sk->sk_state == TCP_ESTABLISHED)
  823. pipe_grant_credits(sk);
  824. return skb;
  825. }
  826. static int pep_recvmsg(struct kiocb *iocb, struct sock *sk,
  827. struct msghdr *msg, size_t len, int noblock,
  828. int flags, int *addr_len)
  829. {
  830. struct sk_buff *skb;
  831. int err;
  832. if (unlikely(1 << sk->sk_state & (TCPF_LISTEN | TCPF_CLOSE)))
  833. return -ENOTCONN;
  834. if ((flags & MSG_OOB) || sock_flag(sk, SOCK_URGINLINE)) {
  835. /* Dequeue and acknowledge control request */
  836. struct pep_sock *pn = pep_sk(sk);
  837. skb = skb_dequeue(&pn->ctrlreq_queue);
  838. if (skb) {
  839. pep_ctrlreq_error(sk, skb, PN_PIPE_NO_ERROR,
  840. GFP_KERNEL);
  841. msg->msg_flags |= MSG_OOB;
  842. goto copy;
  843. }
  844. if (flags & MSG_OOB)
  845. return -EINVAL;
  846. }
  847. skb = skb_recv_datagram(sk, flags, noblock, &err);
  848. lock_sock(sk);
  849. if (skb == NULL) {
  850. if (err == -ENOTCONN && sk->sk_state == TCP_CLOSE_WAIT)
  851. err = -ECONNRESET;
  852. release_sock(sk);
  853. return err;
  854. }
  855. if (sk->sk_state == TCP_ESTABLISHED)
  856. pipe_grant_credits(sk);
  857. release_sock(sk);
  858. copy:
  859. msg->msg_flags |= MSG_EOR;
  860. if (skb->len > len)
  861. msg->msg_flags |= MSG_TRUNC;
  862. else
  863. len = skb->len;
  864. err = skb_copy_datagram_iovec(skb, 0, msg->msg_iov, len);
  865. if (!err)
  866. err = (flags & MSG_TRUNC) ? skb->len : len;
  867. skb_free_datagram(sk, skb);
  868. return err;
  869. }
  870. static void pep_sock_unhash(struct sock *sk)
  871. {
  872. struct pep_sock *pn = pep_sk(sk);
  873. struct sock *skparent = NULL;
  874. lock_sock(sk);
  875. if ((1 << sk->sk_state) & ~(TCPF_CLOSE|TCPF_LISTEN)) {
  876. skparent = pn->listener;
  877. sk_del_node_init(sk);
  878. release_sock(sk);
  879. sk = skparent;
  880. pn = pep_sk(skparent);
  881. lock_sock(sk);
  882. }
  883. /* Unhash a listening sock only when it is closed
  884. * and all of its active connected pipes are closed. */
  885. if (hlist_empty(&pn->hlist))
  886. pn_sock_unhash(&pn->pn_sk.sk);
  887. release_sock(sk);
  888. if (skparent)
  889. sock_put(skparent);
  890. }
  891. static struct proto pep_proto = {
  892. .close = pep_sock_close,
  893. .accept = pep_sock_accept,
  894. .ioctl = pep_ioctl,
  895. .init = pep_init,
  896. .setsockopt = pep_setsockopt,
  897. .getsockopt = pep_getsockopt,
  898. .sendmsg = pep_sendmsg,
  899. .recvmsg = pep_recvmsg,
  900. .backlog_rcv = pep_do_rcv,
  901. .hash = pn_sock_hash,
  902. .unhash = pep_sock_unhash,
  903. .get_port = pn_sock_get_port,
  904. .obj_size = sizeof(struct pep_sock),
  905. .owner = THIS_MODULE,
  906. .name = "PNPIPE",
  907. };
  908. static struct phonet_protocol pep_pn_proto = {
  909. .ops = &phonet_stream_ops,
  910. .prot = &pep_proto,
  911. .sock_type = SOCK_SEQPACKET,
  912. };
  913. static int __init pep_register(void)
  914. {
  915. return phonet_proto_register(PN_PROTO_PIPE, &pep_pn_proto);
  916. }
  917. static void __exit pep_unregister(void)
  918. {
  919. phonet_proto_unregister(PN_PROTO_PIPE, &pep_pn_proto);
  920. }
  921. module_init(pep_register);
  922. module_exit(pep_unregister);
  923. MODULE_AUTHOR("Remi Denis-Courmont, Nokia");
  924. MODULE_DESCRIPTION("Phonet pipe protocol");
  925. MODULE_LICENSE("GPL");
  926. MODULE_ALIAS_NET_PF_PROTO(PF_PHONET, PN_PROTO_PIPE);