msg.c 3.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111
  1. /*
  2. * net/tipc/msg.c: TIPC message header routines
  3. *
  4. * Copyright (c) 2000-2006, 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 "core.h"
  37. #include "msg.h"
  38. u32 tipc_msg_tot_importance(struct tipc_msg *m)
  39. {
  40. if (likely(msg_isdata(m))) {
  41. if (likely(msg_orignode(m) == tipc_own_addr))
  42. return msg_importance(m);
  43. return msg_importance(m) + 4;
  44. }
  45. if ((msg_user(m) == MSG_FRAGMENTER) &&
  46. (msg_type(m) == FIRST_FRAGMENT))
  47. return msg_importance(msg_get_wrapped(m));
  48. return msg_importance(m);
  49. }
  50. void tipc_msg_init(struct tipc_msg *m, u32 user, u32 type,
  51. u32 hsize, u32 destnode)
  52. {
  53. memset(m, 0, hsize);
  54. msg_set_version(m);
  55. msg_set_user(m, user);
  56. msg_set_hdr_sz(m, hsize);
  57. msg_set_size(m, hsize);
  58. msg_set_prevnode(m, tipc_own_addr);
  59. msg_set_type(m, type);
  60. msg_set_orignode(m, tipc_own_addr);
  61. msg_set_destnode(m, destnode);
  62. }
  63. /**
  64. * tipc_msg_build - create message using specified header and data
  65. *
  66. * Note: Caller must not hold any locks in case copy_from_user() is interrupted!
  67. *
  68. * Returns message data size or errno
  69. */
  70. int tipc_msg_build(struct tipc_msg *hdr, struct iovec const *msg_sect,
  71. u32 num_sect, unsigned int total_len,
  72. int max_size, int usrmem, struct sk_buff **buf)
  73. {
  74. int dsz, sz, hsz, pos, res, cnt;
  75. dsz = total_len;
  76. pos = hsz = msg_hdr_sz(hdr);
  77. sz = hsz + dsz;
  78. msg_set_size(hdr, sz);
  79. if (unlikely(sz > max_size)) {
  80. *buf = NULL;
  81. return dsz;
  82. }
  83. *buf = tipc_buf_acquire(sz);
  84. if (!(*buf))
  85. return -ENOMEM;
  86. skb_copy_to_linear_data(*buf, hdr, hsz);
  87. for (res = 1, cnt = 0; res && (cnt < num_sect); cnt++) {
  88. if (likely(usrmem))
  89. res = !copy_from_user((*buf)->data + pos,
  90. msg_sect[cnt].iov_base,
  91. msg_sect[cnt].iov_len);
  92. else
  93. skb_copy_to_linear_data_offset(*buf, pos,
  94. msg_sect[cnt].iov_base,
  95. msg_sect[cnt].iov_len);
  96. pos += msg_sect[cnt].iov_len;
  97. }
  98. if (likely(res))
  99. return dsz;
  100. kfree_skb(*buf);
  101. *buf = NULL;
  102. return -EFAULT;
  103. }