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 11 #include <linux/spinlock.h> 12 #include <linux/init.h> 13 #include <linux/security.h> 14 #include <linux/slab.h> 15 #include <linux/ipc.h> 16 #include <linux/msg.h> 17 #include <linux/ipc_namespace.h> 18 #include <linux/utsname.h> 19 #include <linux/proc_fs.h> 20 #include <asm/uaccess.h> 21 22 #include "util.h" 23 24 DEFINE_SPINLOCK(mq_lock); 25 26 /* 27 * The next 2 defines are here bc this is the only file 28 * compiled when either CONFIG_SYSVIPC and CONFIG_POSIX_MQUEUE 29 * and not CONFIG_IPC_NS. 30 */ 31 struct ipc_namespace init_ipc_ns = { 32 .count = ATOMIC_INIT(1), 33 .user_ns = &init_user_ns, 34 .proc_inum = PROC_IPC_INIT_INO, 35 }; 36 37 atomic_t nr_ipc_ns = ATOMIC_INIT(1); 38 39 struct msg_msgseg { 40 struct msg_msgseg* next; 41 /* the next part of the message follows immediately */ 42 }; 43 44 #define DATALEN_MSG (int)(PAGE_SIZE-sizeof(struct msg_msg)) 45 #define DATALEN_SEG (int)(PAGE_SIZE-sizeof(struct msg_msgseg)) 46 47 48 static struct msg_msg *alloc_msg(int len) 49 { 50 struct msg_msg *msg; 51 struct msg_msgseg **pseg; 52 int alen; 53 54 alen = min(len, DATALEN_MSG); 55 msg = kmalloc(sizeof(*msg) + alen, GFP_KERNEL); 56 if (msg == NULL) 57 return NULL; 58 59 msg->next = NULL; 60 msg->security = NULL; 61 62 len -= alen; 63 pseg = &msg->next; 64 while (len > 0) { 65 struct msg_msgseg *seg; 66 alen = min(len, DATALEN_SEG); 67 seg = kmalloc(sizeof(*seg) + alen, GFP_KERNEL); 68 if (seg == NULL) 69 goto out_err; 70 *pseg = seg; 71 seg->next = NULL; 72 pseg = &seg->next; 73 len -= alen; 74 } 75 76 return msg; 77 78 out_err: 79 free_msg(msg); 80 return NULL; 81 } 82 83 struct msg_msg *load_msg(const void __user *src, int len) 84 { 85 struct msg_msg *msg; 86 struct msg_msgseg *seg; 87 int err; 88 int alen; 89 90 msg = alloc_msg(len); 91 if (msg == NULL) 92 return ERR_PTR(-ENOMEM); 93 94 alen = min(len, DATALEN_MSG); 95 if (copy_from_user(msg + 1, src, alen)) { 96 err = -EFAULT; 97 goto out_err; 98 } 99 100 for (seg = msg->next; seg != NULL; seg = seg->next) { 101 len -= alen; 102 src = (char __user *)src + alen; 103 alen = min(len, DATALEN_SEG); 104 if (copy_from_user(seg + 1, src, alen)) { 105 err = -EFAULT; 106 goto out_err; 107 } 108 } 109 110 err = security_msg_msg_alloc(msg); 111 if (err) 112 goto out_err; 113 114 return msg; 115 116 out_err: 117 free_msg(msg); 118 return ERR_PTR(err); 119 } 120 #ifdef CONFIG_CHECKPOINT_RESTORE 121 struct msg_msg *copy_msg(struct msg_msg *src, struct msg_msg *dst) 122 { 123 struct msg_msgseg *dst_pseg, *src_pseg; 124 int len = src->m_ts; 125 int alen; 126 127 BUG_ON(dst == NULL); 128 if (src->m_ts > dst->m_ts) 129 return ERR_PTR(-EINVAL); 130 131 alen = min(len, DATALEN_MSG); 132 memcpy(dst + 1, src + 1, alen); 133 134 for (dst_pseg = dst->next, src_pseg = src->next; 135 src_pseg != NULL; 136 dst_pseg = dst_pseg->next, src_pseg = src_pseg->next) { 137 138 len -= alen; 139 alen = min(len, DATALEN_SEG); 140 memcpy(dst_pseg + 1, src_pseg + 1, alen); 141 } 142 143 dst->m_type = src->m_type; 144 dst->m_ts = src->m_ts; 145 146 return dst; 147 } 148 #else 149 struct msg_msg *copy_msg(struct msg_msg *src, struct msg_msg *dst) 150 { 151 return ERR_PTR(-ENOSYS); 152 } 153 #endif 154 int store_msg(void __user *dest, struct msg_msg *msg, int len) 155 { 156 int alen; 157 struct msg_msgseg *seg; 158 159 alen = min(len, DATALEN_MSG); 160 if (copy_to_user(dest, msg + 1, alen)) 161 return -1; 162 163 for (seg = msg->next; seg != NULL; seg = seg->next) { 164 len -= alen; 165 dest = (char __user *)dest + alen; 166 alen = min(len, DATALEN_SEG); 167 if (copy_to_user(dest, seg + 1, alen)) 168 return -1; 169 } 170 return 0; 171 } 172 173 void free_msg(struct msg_msg *msg) 174 { 175 struct msg_msgseg *seg; 176 177 security_msg_msg_free(msg); 178 179 seg = msg->next; 180 kfree(msg); 181 while (seg != NULL) { 182 struct msg_msgseg *tmp = seg->next; 183 kfree(seg); 184 seg = tmp; 185 } 186 } 187