msgutil.c 2.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148
  1. /*
  2. * linux/ipc/msgutil.c
  3. * Copyright (C) 1999, 2004 Manfred Spraul
  4. *
  5. * This file is released under GNU General Public Licence version 2 or
  6. * (at your option) any later version.
  7. *
  8. * See the file COPYING for more details.
  9. */
  10. #include <linux/spinlock.h>
  11. #include <linux/init.h>
  12. #include <linux/security.h>
  13. #include <linux/slab.h>
  14. #include <linux/ipc.h>
  15. #include <linux/ipc_namespace.h>
  16. #include <asm/uaccess.h>
  17. #include "util.h"
  18. DEFINE_SPINLOCK(mq_lock);
  19. /*
  20. * The next 2 defines are here bc this is the only file
  21. * compiled when either CONFIG_SYSVIPC and CONFIG_POSIX_MQUEUE
  22. * and not CONFIG_IPC_NS.
  23. */
  24. struct ipc_namespace init_ipc_ns = {
  25. .count = ATOMIC_INIT(1),
  26. #ifdef CONFIG_POSIX_MQUEUE
  27. .mq_queues_max = DFLT_QUEUESMAX,
  28. .mq_msg_max = DFLT_MSGMAX,
  29. .mq_msgsize_max = DFLT_MSGSIZEMAX,
  30. #endif
  31. .user_ns = &init_user_ns,
  32. };
  33. atomic_t nr_ipc_ns = ATOMIC_INIT(1);
  34. struct msg_msgseg {
  35. struct msg_msgseg* next;
  36. /* the next part of the message follows immediately */
  37. };
  38. #define DATALEN_MSG (PAGE_SIZE-sizeof(struct msg_msg))
  39. #define DATALEN_SEG (PAGE_SIZE-sizeof(struct msg_msgseg))
  40. struct msg_msg *load_msg(const void __user *src, int len)
  41. {
  42. struct msg_msg *msg;
  43. struct msg_msgseg **pseg;
  44. int err;
  45. int alen;
  46. alen = len;
  47. if (alen > DATALEN_MSG)
  48. alen = DATALEN_MSG;
  49. msg = kmalloc(sizeof(*msg) + alen, GFP_KERNEL);
  50. if (msg == NULL)
  51. return ERR_PTR(-ENOMEM);
  52. msg->next = NULL;
  53. msg->security = NULL;
  54. if (copy_from_user(msg + 1, src, alen)) {
  55. err = -EFAULT;
  56. goto out_err;
  57. }
  58. len -= alen;
  59. src = ((char __user *)src) + alen;
  60. pseg = &msg->next;
  61. while (len > 0) {
  62. struct msg_msgseg *seg;
  63. alen = len;
  64. if (alen > DATALEN_SEG)
  65. alen = DATALEN_SEG;
  66. seg = kmalloc(sizeof(*seg) + alen,
  67. GFP_KERNEL);
  68. if (seg == NULL) {
  69. err = -ENOMEM;
  70. goto out_err;
  71. }
  72. *pseg = seg;
  73. seg->next = NULL;
  74. if (copy_from_user(seg + 1, src, alen)) {
  75. err = -EFAULT;
  76. goto out_err;
  77. }
  78. pseg = &seg->next;
  79. len -= alen;
  80. src = ((char __user *)src) + alen;
  81. }
  82. err = security_msg_msg_alloc(msg);
  83. if (err)
  84. goto out_err;
  85. return msg;
  86. out_err:
  87. free_msg(msg);
  88. return ERR_PTR(err);
  89. }
  90. int store_msg(void __user *dest, struct msg_msg *msg, int len)
  91. {
  92. int alen;
  93. struct msg_msgseg *seg;
  94. alen = len;
  95. if (alen > DATALEN_MSG)
  96. alen = DATALEN_MSG;
  97. if (copy_to_user(dest, msg + 1, alen))
  98. return -1;
  99. len -= alen;
  100. dest = ((char __user *)dest) + alen;
  101. seg = msg->next;
  102. while (len > 0) {
  103. alen = len;
  104. if (alen > DATALEN_SEG)
  105. alen = DATALEN_SEG;
  106. if (copy_to_user(dest, seg + 1, alen))
  107. return -1;
  108. len -= alen;
  109. dest = ((char __user *)dest) + alen;
  110. seg = seg->next;
  111. }
  112. return 0;
  113. }
  114. void free_msg(struct msg_msg *msg)
  115. {
  116. struct msg_msgseg *seg;
  117. security_msg_msg_free(msg);
  118. seg = msg->next;
  119. kfree(msg);
  120. while (seg != NULL) {
  121. struct msg_msgseg *tmp = seg->next;
  122. kfree(seg);
  123. seg = tmp;
  124. }
  125. }