msg.c 19 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715
  1. /*
  2. * net/tipc/msg.c: TIPC message header routines
  3. *
  4. * Copyright (c) 2000-2006, 2014-2015, Ericsson AB
  5. * Copyright (c) 2005, 2010-2011, Wind River Systems
  6. * All rights reserved.
  7. *
  8. * Redistribution and use in source and binary forms, with or without
  9. * modification, are permitted provided that the following conditions are met:
  10. *
  11. * 1. Redistributions of source code must retain the above copyright
  12. * notice, this list of conditions and the following disclaimer.
  13. * 2. Redistributions in binary form must reproduce the above copyright
  14. * notice, this list of conditions and the following disclaimer in the
  15. * documentation and/or other materials provided with the distribution.
  16. * 3. Neither the names of the copyright holders nor the names of its
  17. * contributors may be used to endorse or promote products derived from
  18. * this software without specific prior written permission.
  19. *
  20. * Alternatively, this software may be distributed under the terms of the
  21. * GNU General Public License ("GPL") version 2 as published by the Free
  22. * Software Foundation.
  23. *
  24. * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  25. * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  26. * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  27. * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
  28. * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  29. * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  30. * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  31. * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  32. * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  33. * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  34. * POSSIBILITY OF SUCH DAMAGE.
  35. */
  36. #include <net/sock.h>
  37. #include "core.h"
  38. #include "msg.h"
  39. #include "addr.h"
  40. #include "name_table.h"
  41. #define MAX_FORWARD_SIZE 1024
  42. #define BUF_HEADROOM (LL_MAX_HEADER + 48)
  43. #define BUF_TAILROOM 16
  44. static unsigned int align(unsigned int i)
  45. {
  46. return (i + 3) & ~3u;
  47. }
  48. /**
  49. * tipc_buf_acquire - creates a TIPC message buffer
  50. * @size: message size (including TIPC header)
  51. *
  52. * Returns a new buffer with data pointers set to the specified size.
  53. *
  54. * NOTE: Headroom is reserved to allow prepending of a data link header.
  55. * There may also be unrequested tailroom present at the buffer's end.
  56. */
  57. struct sk_buff *tipc_buf_acquire(u32 size, gfp_t gfp)
  58. {
  59. struct sk_buff *skb;
  60. unsigned int buf_size = (BUF_HEADROOM + size + 3) & ~3u;
  61. skb = alloc_skb_fclone(buf_size, gfp);
  62. if (skb) {
  63. skb_reserve(skb, BUF_HEADROOM);
  64. skb_put(skb, size);
  65. skb->next = NULL;
  66. }
  67. return skb;
  68. }
  69. void tipc_msg_init(u32 own_node, struct tipc_msg *m, u32 user, u32 type,
  70. u32 hsize, u32 dnode)
  71. {
  72. memset(m, 0, hsize);
  73. msg_set_version(m);
  74. msg_set_user(m, user);
  75. msg_set_hdr_sz(m, hsize);
  76. msg_set_size(m, hsize);
  77. msg_set_prevnode(m, own_node);
  78. msg_set_type(m, type);
  79. if (hsize > SHORT_H_SIZE) {
  80. msg_set_orignode(m, own_node);
  81. msg_set_destnode(m, dnode);
  82. }
  83. }
  84. struct sk_buff *tipc_msg_create(uint user, uint type,
  85. uint hdr_sz, uint data_sz, u32 dnode,
  86. u32 onode, u32 dport, u32 oport, int errcode)
  87. {
  88. struct tipc_msg *msg;
  89. struct sk_buff *buf;
  90. buf = tipc_buf_acquire(hdr_sz + data_sz, GFP_ATOMIC);
  91. if (unlikely(!buf))
  92. return NULL;
  93. msg = buf_msg(buf);
  94. tipc_msg_init(onode, msg, user, type, hdr_sz, dnode);
  95. msg_set_size(msg, hdr_sz + data_sz);
  96. msg_set_origport(msg, oport);
  97. msg_set_destport(msg, dport);
  98. msg_set_errcode(msg, errcode);
  99. if (hdr_sz > SHORT_H_SIZE) {
  100. msg_set_orignode(msg, onode);
  101. msg_set_destnode(msg, dnode);
  102. }
  103. return buf;
  104. }
  105. /* tipc_buf_append(): Append a buffer to the fragment list of another buffer
  106. * @*headbuf: in: NULL for first frag, otherwise value returned from prev call
  107. * out: set when successful non-complete reassembly, otherwise NULL
  108. * @*buf: in: the buffer to append. Always defined
  109. * out: head buf after successful complete reassembly, otherwise NULL
  110. * Returns 1 when reassembly complete, otherwise 0
  111. */
  112. int tipc_buf_append(struct sk_buff **headbuf, struct sk_buff **buf)
  113. {
  114. struct sk_buff *head = *headbuf;
  115. struct sk_buff *frag = *buf;
  116. struct sk_buff *tail = NULL;
  117. struct tipc_msg *msg;
  118. u32 fragid;
  119. int delta;
  120. bool headstolen;
  121. if (!frag)
  122. goto err;
  123. msg = buf_msg(frag);
  124. fragid = msg_type(msg);
  125. frag->next = NULL;
  126. skb_pull(frag, msg_hdr_sz(msg));
  127. if (fragid == FIRST_FRAGMENT) {
  128. if (unlikely(head))
  129. goto err;
  130. if (unlikely(skb_unclone(frag, GFP_ATOMIC)))
  131. goto err;
  132. head = *headbuf = frag;
  133. *buf = NULL;
  134. TIPC_SKB_CB(head)->tail = NULL;
  135. if (skb_is_nonlinear(head)) {
  136. skb_walk_frags(head, tail) {
  137. TIPC_SKB_CB(head)->tail = tail;
  138. }
  139. } else {
  140. skb_frag_list_init(head);
  141. }
  142. return 0;
  143. }
  144. if (!head)
  145. goto err;
  146. if (skb_try_coalesce(head, frag, &headstolen, &delta)) {
  147. kfree_skb_partial(frag, headstolen);
  148. } else {
  149. tail = TIPC_SKB_CB(head)->tail;
  150. if (!skb_has_frag_list(head))
  151. skb_shinfo(head)->frag_list = frag;
  152. else
  153. tail->next = frag;
  154. head->truesize += frag->truesize;
  155. head->data_len += frag->len;
  156. head->len += frag->len;
  157. TIPC_SKB_CB(head)->tail = frag;
  158. }
  159. if (fragid == LAST_FRAGMENT) {
  160. TIPC_SKB_CB(head)->validated = false;
  161. if (unlikely(!tipc_msg_validate(&head)))
  162. goto err;
  163. *buf = head;
  164. TIPC_SKB_CB(head)->tail = NULL;
  165. *headbuf = NULL;
  166. return 1;
  167. }
  168. *buf = NULL;
  169. return 0;
  170. err:
  171. kfree_skb(*buf);
  172. kfree_skb(*headbuf);
  173. *buf = *headbuf = NULL;
  174. return 0;
  175. }
  176. /* tipc_msg_validate - validate basic format of received message
  177. *
  178. * This routine ensures a TIPC message has an acceptable header, and at least
  179. * as much data as the header indicates it should. The routine also ensures
  180. * that the entire message header is stored in the main fragment of the message
  181. * buffer, to simplify future access to message header fields.
  182. *
  183. * Note: Having extra info present in the message header or data areas is OK.
  184. * TIPC will ignore the excess, under the assumption that it is optional info
  185. * introduced by a later release of the protocol.
  186. */
  187. bool tipc_msg_validate(struct sk_buff **_skb)
  188. {
  189. struct sk_buff *skb = *_skb;
  190. struct tipc_msg *hdr;
  191. int msz, hsz;
  192. /* Ensure that flow control ratio condition is satisfied */
  193. if (unlikely(skb->truesize / buf_roundup_len(skb) >= 4)) {
  194. skb = skb_copy_expand(skb, BUF_HEADROOM, 0, GFP_ATOMIC);
  195. if (!skb)
  196. return false;
  197. kfree_skb(*_skb);
  198. *_skb = skb;
  199. }
  200. if (unlikely(TIPC_SKB_CB(skb)->validated))
  201. return true;
  202. if (unlikely(!pskb_may_pull(skb, MIN_H_SIZE)))
  203. return false;
  204. hsz = msg_hdr_sz(buf_msg(skb));
  205. if (unlikely(hsz < MIN_H_SIZE) || (hsz > MAX_H_SIZE))
  206. return false;
  207. if (unlikely(!pskb_may_pull(skb, hsz)))
  208. return false;
  209. hdr = buf_msg(skb);
  210. if (unlikely(msg_version(hdr) != TIPC_VERSION))
  211. return false;
  212. msz = msg_size(hdr);
  213. if (unlikely(msz < hsz))
  214. return false;
  215. if (unlikely((msz - hsz) > TIPC_MAX_USER_MSG_SIZE))
  216. return false;
  217. if (unlikely(skb->len < msz))
  218. return false;
  219. TIPC_SKB_CB(skb)->validated = true;
  220. return true;
  221. }
  222. /**
  223. * tipc_msg_build - create buffer chain containing specified header and data
  224. * @mhdr: Message header, to be prepended to data
  225. * @m: User message
  226. * @dsz: Total length of user data
  227. * @pktmax: Max packet size that can be used
  228. * @list: Buffer or chain of buffers to be returned to caller
  229. *
  230. * Note that the recursive call we are making here is safe, since it can
  231. * logically go only one further level down.
  232. *
  233. * Returns message data size or errno: -ENOMEM, -EFAULT
  234. */
  235. int tipc_msg_build(struct tipc_msg *mhdr, struct msghdr *m, int offset,
  236. int dsz, int pktmax, struct sk_buff_head *list)
  237. {
  238. int mhsz = msg_hdr_sz(mhdr);
  239. struct tipc_msg pkthdr;
  240. int msz = mhsz + dsz;
  241. int pktrem = pktmax;
  242. struct sk_buff *skb;
  243. int drem = dsz;
  244. int pktno = 1;
  245. char *pktpos;
  246. int pktsz;
  247. int rc;
  248. msg_set_size(mhdr, msz);
  249. /* No fragmentation needed? */
  250. if (likely(msz <= pktmax)) {
  251. skb = tipc_buf_acquire(msz, GFP_KERNEL);
  252. /* Fall back to smaller MTU if node local message */
  253. if (unlikely(!skb)) {
  254. if (pktmax != MAX_MSG_SIZE)
  255. return -ENOMEM;
  256. rc = tipc_msg_build(mhdr, m, offset, dsz, FB_MTU, list);
  257. if (rc != dsz)
  258. return rc;
  259. if (tipc_msg_assemble(list))
  260. return dsz;
  261. return -ENOMEM;
  262. }
  263. skb_orphan(skb);
  264. __skb_queue_tail(list, skb);
  265. skb_copy_to_linear_data(skb, mhdr, mhsz);
  266. pktpos = skb->data + mhsz;
  267. if (copy_from_iter_full(pktpos, dsz, &m->msg_iter))
  268. return dsz;
  269. rc = -EFAULT;
  270. goto error;
  271. }
  272. /* Prepare reusable fragment header */
  273. tipc_msg_init(msg_prevnode(mhdr), &pkthdr, MSG_FRAGMENTER,
  274. FIRST_FRAGMENT, INT_H_SIZE, msg_destnode(mhdr));
  275. msg_set_size(&pkthdr, pktmax);
  276. msg_set_fragm_no(&pkthdr, pktno);
  277. msg_set_importance(&pkthdr, msg_importance(mhdr));
  278. /* Prepare first fragment */
  279. skb = tipc_buf_acquire(pktmax, GFP_KERNEL);
  280. if (!skb)
  281. return -ENOMEM;
  282. skb_orphan(skb);
  283. __skb_queue_tail(list, skb);
  284. pktpos = skb->data;
  285. skb_copy_to_linear_data(skb, &pkthdr, INT_H_SIZE);
  286. pktpos += INT_H_SIZE;
  287. pktrem -= INT_H_SIZE;
  288. skb_copy_to_linear_data_offset(skb, INT_H_SIZE, mhdr, mhsz);
  289. pktpos += mhsz;
  290. pktrem -= mhsz;
  291. do {
  292. if (drem < pktrem)
  293. pktrem = drem;
  294. if (!copy_from_iter_full(pktpos, pktrem, &m->msg_iter)) {
  295. rc = -EFAULT;
  296. goto error;
  297. }
  298. drem -= pktrem;
  299. if (!drem)
  300. break;
  301. /* Prepare new fragment: */
  302. if (drem < (pktmax - INT_H_SIZE))
  303. pktsz = drem + INT_H_SIZE;
  304. else
  305. pktsz = pktmax;
  306. skb = tipc_buf_acquire(pktsz, GFP_KERNEL);
  307. if (!skb) {
  308. rc = -ENOMEM;
  309. goto error;
  310. }
  311. skb_orphan(skb);
  312. __skb_queue_tail(list, skb);
  313. msg_set_type(&pkthdr, FRAGMENT);
  314. msg_set_size(&pkthdr, pktsz);
  315. msg_set_fragm_no(&pkthdr, ++pktno);
  316. skb_copy_to_linear_data(skb, &pkthdr, INT_H_SIZE);
  317. pktpos = skb->data + INT_H_SIZE;
  318. pktrem = pktsz - INT_H_SIZE;
  319. } while (1);
  320. msg_set_type(buf_msg(skb), LAST_FRAGMENT);
  321. return dsz;
  322. error:
  323. __skb_queue_purge(list);
  324. __skb_queue_head_init(list);
  325. return rc;
  326. }
  327. /**
  328. * tipc_msg_bundle(): Append contents of a buffer to tail of an existing one
  329. * @skb: the buffer to append to ("bundle")
  330. * @msg: message to be appended
  331. * @mtu: max allowable size for the bundle buffer
  332. * Consumes buffer if successful
  333. * Returns true if bundling could be performed, otherwise false
  334. */
  335. bool tipc_msg_bundle(struct sk_buff *skb, struct tipc_msg *msg, u32 mtu)
  336. {
  337. struct tipc_msg *bmsg;
  338. unsigned int bsz;
  339. unsigned int msz = msg_size(msg);
  340. u32 start, pad;
  341. u32 max = mtu - INT_H_SIZE;
  342. if (likely(msg_user(msg) == MSG_FRAGMENTER))
  343. return false;
  344. if (!skb)
  345. return false;
  346. bmsg = buf_msg(skb);
  347. bsz = msg_size(bmsg);
  348. start = align(bsz);
  349. pad = start - bsz;
  350. if (unlikely(msg_user(msg) == TUNNEL_PROTOCOL))
  351. return false;
  352. if (unlikely(msg_user(msg) == BCAST_PROTOCOL))
  353. return false;
  354. if (unlikely(msg_user(bmsg) != MSG_BUNDLER))
  355. return false;
  356. if (unlikely(skb_tailroom(skb) < (pad + msz)))
  357. return false;
  358. if (unlikely(max < (start + msz)))
  359. return false;
  360. if ((msg_importance(msg) < TIPC_SYSTEM_IMPORTANCE) &&
  361. (msg_importance(bmsg) == TIPC_SYSTEM_IMPORTANCE))
  362. return false;
  363. skb_put(skb, pad + msz);
  364. skb_copy_to_linear_data_offset(skb, start, msg, msz);
  365. msg_set_size(bmsg, start + msz);
  366. msg_set_msgcnt(bmsg, msg_msgcnt(bmsg) + 1);
  367. return true;
  368. }
  369. /**
  370. * tipc_msg_extract(): extract bundled inner packet from buffer
  371. * @skb: buffer to be extracted from.
  372. * @iskb: extracted inner buffer, to be returned
  373. * @pos: position in outer message of msg to be extracted.
  374. * Returns position of next msg
  375. * Consumes outer buffer when last packet extracted
  376. * Returns true when when there is an extracted buffer, otherwise false
  377. */
  378. bool tipc_msg_extract(struct sk_buff *skb, struct sk_buff **iskb, int *pos)
  379. {
  380. struct tipc_msg *hdr, *ihdr;
  381. int imsz;
  382. *iskb = NULL;
  383. if (unlikely(skb_linearize(skb)))
  384. goto none;
  385. hdr = buf_msg(skb);
  386. if (unlikely(*pos > (msg_data_sz(hdr) - MIN_H_SIZE)))
  387. goto none;
  388. ihdr = (struct tipc_msg *)(msg_data(hdr) + *pos);
  389. imsz = msg_size(ihdr);
  390. if ((*pos + imsz) > msg_data_sz(hdr))
  391. goto none;
  392. *iskb = tipc_buf_acquire(imsz, GFP_ATOMIC);
  393. if (!*iskb)
  394. goto none;
  395. skb_copy_to_linear_data(*iskb, ihdr, imsz);
  396. if (unlikely(!tipc_msg_validate(iskb)))
  397. goto none;
  398. *pos += align(imsz);
  399. return true;
  400. none:
  401. kfree_skb(skb);
  402. kfree_skb(*iskb);
  403. *iskb = NULL;
  404. return false;
  405. }
  406. /**
  407. * tipc_msg_make_bundle(): Create bundle buf and append message to its tail
  408. * @list: the buffer chain, where head is the buffer to replace/append
  409. * @skb: buffer to be created, appended to and returned in case of success
  410. * @msg: message to be appended
  411. * @mtu: max allowable size for the bundle buffer, inclusive header
  412. * @dnode: destination node for message. (Not always present in header)
  413. * Returns true if success, otherwise false
  414. */
  415. bool tipc_msg_make_bundle(struct sk_buff **skb, struct tipc_msg *msg,
  416. u32 mtu, u32 dnode)
  417. {
  418. struct sk_buff *_skb;
  419. struct tipc_msg *bmsg;
  420. u32 msz = msg_size(msg);
  421. u32 max = mtu - INT_H_SIZE;
  422. if (msg_user(msg) == MSG_FRAGMENTER)
  423. return false;
  424. if (msg_user(msg) == TUNNEL_PROTOCOL)
  425. return false;
  426. if (msg_user(msg) == BCAST_PROTOCOL)
  427. return false;
  428. if (msz > (max / 2))
  429. return false;
  430. _skb = tipc_buf_acquire(max, GFP_ATOMIC);
  431. if (!_skb)
  432. return false;
  433. skb_trim(_skb, INT_H_SIZE);
  434. bmsg = buf_msg(_skb);
  435. tipc_msg_init(msg_prevnode(msg), bmsg, MSG_BUNDLER, 0,
  436. INT_H_SIZE, dnode);
  437. msg_set_importance(bmsg, msg_importance(msg));
  438. msg_set_seqno(bmsg, msg_seqno(msg));
  439. msg_set_ack(bmsg, msg_ack(msg));
  440. msg_set_bcast_ack(bmsg, msg_bcast_ack(msg));
  441. tipc_msg_bundle(_skb, msg, mtu);
  442. *skb = _skb;
  443. return true;
  444. }
  445. /**
  446. * tipc_msg_reverse(): swap source and destination addresses and add error code
  447. * @own_node: originating node id for reversed message
  448. * @skb: buffer containing message to be reversed; may be replaced.
  449. * @err: error code to be set in message, if any
  450. * Consumes buffer at failure
  451. * Returns true if success, otherwise false
  452. */
  453. bool tipc_msg_reverse(u32 own_node, struct sk_buff **skb, int err)
  454. {
  455. struct sk_buff *_skb = *skb;
  456. struct tipc_msg *hdr;
  457. struct tipc_msg ohdr;
  458. int dlen;
  459. if (skb_linearize(_skb))
  460. goto exit;
  461. hdr = buf_msg(_skb);
  462. dlen = min_t(uint, msg_data_sz(hdr), MAX_FORWARD_SIZE);
  463. if (msg_dest_droppable(hdr))
  464. goto exit;
  465. if (msg_errcode(hdr))
  466. goto exit;
  467. /* Take a copy of original header before altering message */
  468. memcpy(&ohdr, hdr, msg_hdr_sz(hdr));
  469. /* Never return SHORT header; expand by replacing buffer if necessary */
  470. if (msg_short(hdr)) {
  471. *skb = tipc_buf_acquire(BASIC_H_SIZE + dlen, GFP_ATOMIC);
  472. if (!*skb)
  473. goto exit;
  474. memcpy((*skb)->data + BASIC_H_SIZE, msg_data(hdr), dlen);
  475. kfree_skb(_skb);
  476. _skb = *skb;
  477. hdr = buf_msg(_skb);
  478. memcpy(hdr, &ohdr, BASIC_H_SIZE);
  479. msg_set_hdr_sz(hdr, BASIC_H_SIZE);
  480. }
  481. /* Now reverse the concerned fields */
  482. msg_set_errcode(hdr, err);
  483. msg_set_non_seq(hdr, 0);
  484. msg_set_origport(hdr, msg_destport(&ohdr));
  485. msg_set_destport(hdr, msg_origport(&ohdr));
  486. msg_set_destnode(hdr, msg_prevnode(&ohdr));
  487. msg_set_prevnode(hdr, own_node);
  488. msg_set_orignode(hdr, own_node);
  489. msg_set_size(hdr, msg_hdr_sz(hdr) + dlen);
  490. skb_trim(_skb, msg_size(hdr));
  491. skb_orphan(_skb);
  492. return true;
  493. exit:
  494. kfree_skb(_skb);
  495. *skb = NULL;
  496. return false;
  497. }
  498. /**
  499. * tipc_msg_lookup_dest(): try to find new destination for named message
  500. * @skb: the buffer containing the message.
  501. * @err: error code to be used by caller if lookup fails
  502. * Does not consume buffer
  503. * Returns true if a destination is found, false otherwise
  504. */
  505. bool tipc_msg_lookup_dest(struct net *net, struct sk_buff *skb, int *err)
  506. {
  507. struct tipc_msg *msg = buf_msg(skb);
  508. u32 dport, dnode;
  509. u32 onode = tipc_own_addr(net);
  510. if (!msg_isdata(msg))
  511. return false;
  512. if (!msg_named(msg))
  513. return false;
  514. if (msg_errcode(msg))
  515. return false;
  516. *err = TIPC_ERR_NO_NAME;
  517. if (skb_linearize(skb))
  518. return false;
  519. msg = buf_msg(skb);
  520. if (msg_reroute_cnt(msg))
  521. return false;
  522. dnode = tipc_scope2node(net, msg_lookup_scope(msg));
  523. dport = tipc_nametbl_translate(net, msg_nametype(msg),
  524. msg_nameinst(msg), &dnode);
  525. if (!dport)
  526. return false;
  527. msg_incr_reroute_cnt(msg);
  528. if (dnode != onode)
  529. msg_set_prevnode(msg, onode);
  530. msg_set_destnode(msg, dnode);
  531. msg_set_destport(msg, dport);
  532. *err = TIPC_OK;
  533. if (!skb_cloned(skb))
  534. return true;
  535. return true;
  536. }
  537. /* tipc_msg_assemble() - assemble chain of fragments into one message
  538. */
  539. bool tipc_msg_assemble(struct sk_buff_head *list)
  540. {
  541. struct sk_buff *skb, *tmp = NULL;
  542. if (skb_queue_len(list) == 1)
  543. return true;
  544. while ((skb = __skb_dequeue(list))) {
  545. skb->next = NULL;
  546. if (tipc_buf_append(&tmp, &skb)) {
  547. __skb_queue_tail(list, skb);
  548. return true;
  549. }
  550. if (!tmp)
  551. break;
  552. }
  553. __skb_queue_purge(list);
  554. __skb_queue_head_init(list);
  555. pr_warn("Failed do assemble buffer\n");
  556. return false;
  557. }
  558. /* tipc_msg_reassemble() - clone a buffer chain of fragments and
  559. * reassemble the clones into one message
  560. */
  561. bool tipc_msg_reassemble(struct sk_buff_head *list, struct sk_buff_head *rcvq)
  562. {
  563. struct sk_buff *skb, *_skb;
  564. struct sk_buff *frag = NULL;
  565. struct sk_buff *head = NULL;
  566. int hdr_len;
  567. /* Copy header if single buffer */
  568. if (skb_queue_len(list) == 1) {
  569. skb = skb_peek(list);
  570. hdr_len = skb_headroom(skb) + msg_hdr_sz(buf_msg(skb));
  571. _skb = __pskb_copy(skb, hdr_len, GFP_ATOMIC);
  572. if (!_skb)
  573. return false;
  574. __skb_queue_tail(rcvq, _skb);
  575. return true;
  576. }
  577. /* Clone all fragments and reassemble */
  578. skb_queue_walk(list, skb) {
  579. frag = skb_clone(skb, GFP_ATOMIC);
  580. if (!frag)
  581. goto error;
  582. frag->next = NULL;
  583. if (tipc_buf_append(&head, &frag))
  584. break;
  585. if (!head)
  586. goto error;
  587. }
  588. __skb_queue_tail(rcvq, frag);
  589. return true;
  590. error:
  591. pr_warn("Failed do clone local mcast rcv buffer\n");
  592. kfree_skb(head);
  593. return false;
  594. }
  595. bool tipc_msg_pskb_copy(u32 dst, struct sk_buff_head *msg,
  596. struct sk_buff_head *cpy)
  597. {
  598. struct sk_buff *skb, *_skb;
  599. skb_queue_walk(msg, skb) {
  600. _skb = pskb_copy(skb, GFP_ATOMIC);
  601. if (!_skb) {
  602. __skb_queue_purge(cpy);
  603. return false;
  604. }
  605. msg_set_destnode(buf_msg(_skb), dst);
  606. __skb_queue_tail(cpy, _skb);
  607. }
  608. return true;
  609. }
  610. /* tipc_skb_queue_sorted(); sort pkt into list according to sequence number
  611. * @list: list to be appended to
  612. * @seqno: sequence number of buffer to add
  613. * @skb: buffer to add
  614. */
  615. void __tipc_skb_queue_sorted(struct sk_buff_head *list, u16 seqno,
  616. struct sk_buff *skb)
  617. {
  618. struct sk_buff *_skb, *tmp;
  619. if (skb_queue_empty(list) || less(seqno, buf_seqno(skb_peek(list)))) {
  620. __skb_queue_head(list, skb);
  621. return;
  622. }
  623. if (more(seqno, buf_seqno(skb_peek_tail(list)))) {
  624. __skb_queue_tail(list, skb);
  625. return;
  626. }
  627. skb_queue_walk_safe(list, _skb, tmp) {
  628. if (more(seqno, buf_seqno(_skb)))
  629. continue;
  630. if (seqno == buf_seqno(_skb))
  631. break;
  632. __skb_queue_before(list, _skb, skb);
  633. return;
  634. }
  635. kfree_skb(skb);
  636. }
  637. void tipc_skb_reject(struct net *net, int err, struct sk_buff *skb,
  638. struct sk_buff_head *xmitq)
  639. {
  640. if (tipc_msg_reverse(tipc_own_addr(net), &skb, err))
  641. __skb_queue_tail(xmitq, skb);
  642. }