1 #include <stdlib.h> 2 #include <stdio.h> 3 #include <string.h> 4 #include <errno.h> 5 #include <linux/msg.h> 6 #include <fcntl.h> 7 8 #include "../kselftest.h" 9 10 #define MAX_MSG_SIZE 32 11 12 struct msg1 { 13 int msize; 14 long mtype; 15 char mtext[MAX_MSG_SIZE]; 16 }; 17 18 #define TEST_STRING "Test sysv5 msg" 19 #define MSG_TYPE 1 20 21 #define ANOTHER_TEST_STRING "Yet another test sysv5 msg" 22 #define ANOTHER_MSG_TYPE 26538 23 24 struct msgque_data { 25 key_t key; 26 int msq_id; 27 int qbytes; 28 int qnum; 29 int mode; 30 struct msg1 *messages; 31 }; 32 33 int restore_queue(struct msgque_data *msgque) 34 { 35 int fd, ret, id, i; 36 char buf[32]; 37 38 fd = open("/proc/sys/kernel/msg_next_id", O_WRONLY); 39 if (fd == -1) { 40 printf("Failed to open /proc/sys/kernel/msg_next_id\n"); 41 return -errno; 42 } 43 sprintf(buf, "%d", msgque->msq_id); 44 45 ret = write(fd, buf, strlen(buf)); 46 if (ret != strlen(buf)) { 47 printf("Failed to write to /proc/sys/kernel/msg_next_id\n"); 48 return -errno; 49 } 50 51 id = msgget(msgque->key, msgque->mode | IPC_CREAT | IPC_EXCL); 52 if (id == -1) { 53 printf("Failed to create queue\n"); 54 return -errno; 55 } 56 57 if (id != msgque->msq_id) { 58 printf("Restored queue has wrong id (%d instead of %d)\n", 59 id, msgque->msq_id); 60 ret = -EFAULT; 61 goto destroy; 62 } 63 64 for (i = 0; i < msgque->qnum; i++) { 65 if (msgsnd(msgque->msq_id, &msgque->messages[i].mtype, 66 msgque->messages[i].msize, IPC_NOWAIT) != 0) { 67 printf("msgsnd failed (%m)\n"); 68 ret = -errno; 69 goto destroy; 70 }; 71 } 72 return 0; 73 74 destroy: 75 if (msgctl(id, IPC_RMID, 0)) 76 printf("Failed to destroy queue: %d\n", -errno); 77 return ret; 78 } 79 80 int check_and_destroy_queue(struct msgque_data *msgque) 81 { 82 struct msg1 message; 83 int cnt = 0, ret; 84 85 while (1) { 86 ret = msgrcv(msgque->msq_id, &message.mtype, MAX_MSG_SIZE, 87 0, IPC_NOWAIT); 88 if (ret < 0) { 89 if (errno == ENOMSG) 90 break; 91 printf("Failed to read IPC message: %m\n"); 92 ret = -errno; 93 goto err; 94 } 95 if (ret != msgque->messages[cnt].msize) { 96 printf("Wrong message size: %d (expected %d)\n", ret, 97 msgque->messages[cnt].msize); 98 ret = -EINVAL; 99 goto err; 100 } 101 if (message.mtype != msgque->messages[cnt].mtype) { 102 printf("Wrong message type\n"); 103 ret = -EINVAL; 104 goto err; 105 } 106 if (memcmp(message.mtext, msgque->messages[cnt].mtext, ret)) { 107 printf("Wrong message content\n"); 108 ret = -EINVAL; 109 goto err; 110 } 111 cnt++; 112 } 113 114 if (cnt != msgque->qnum) { 115 printf("Wrong message number\n"); 116 ret = -EINVAL; 117 goto err; 118 } 119 120 ret = 0; 121 err: 122 if (msgctl(msgque->msq_id, IPC_RMID, 0)) { 123 printf("Failed to destroy queue: %d\n", -errno); 124 return -errno; 125 } 126 return ret; 127 } 128 129 int dump_queue(struct msgque_data *msgque) 130 { 131 struct msqid64_ds ds; 132 int kern_id; 133 int i, ret; 134 135 for (kern_id = 0; kern_id < 256; kern_id++) { 136 ret = msgctl(kern_id, MSG_STAT, &ds); 137 if (ret < 0) { 138 if (errno == -EINVAL) 139 continue; 140 printf("Failed to get stats for IPC queue with id %d\n", 141 kern_id); 142 return -errno; 143 } 144 145 if (ret == msgque->msq_id) 146 break; 147 } 148 149 msgque->messages = malloc(sizeof(struct msg1) * ds.msg_qnum); 150 if (msgque->messages == NULL) { 151 printf("Failed to get stats for IPC queue\n"); 152 return -ENOMEM; 153 } 154 155 msgque->qnum = ds.msg_qnum; 156 msgque->mode = ds.msg_perm.mode; 157 msgque->qbytes = ds.msg_qbytes; 158 159 for (i = 0; i < msgque->qnum; i++) { 160 ret = msgrcv(msgque->msq_id, &msgque->messages[i].mtype, 161 MAX_MSG_SIZE, i, IPC_NOWAIT | MSG_COPY); 162 if (ret < 0) { 163 printf("Failed to copy IPC message: %m (%d)\n", errno); 164 return -errno; 165 } 166 msgque->messages[i].msize = ret; 167 } 168 return 0; 169 } 170 171 int fill_msgque(struct msgque_data *msgque) 172 { 173 struct msg1 msgbuf; 174 175 msgbuf.mtype = MSG_TYPE; 176 memcpy(msgbuf.mtext, TEST_STRING, sizeof(TEST_STRING)); 177 if (msgsnd(msgque->msq_id, &msgbuf.mtype, sizeof(TEST_STRING), 178 IPC_NOWAIT) != 0) { 179 printf("First message send failed (%m)\n"); 180 return -errno; 181 }; 182 183 msgbuf.mtype = ANOTHER_MSG_TYPE; 184 memcpy(msgbuf.mtext, ANOTHER_TEST_STRING, sizeof(ANOTHER_TEST_STRING)); 185 if (msgsnd(msgque->msq_id, &msgbuf.mtype, sizeof(ANOTHER_TEST_STRING), 186 IPC_NOWAIT) != 0) { 187 printf("Second message send failed (%m)\n"); 188 return -errno; 189 }; 190 return 0; 191 } 192 193 int main(int argc, char **argv) 194 { 195 int msg, pid, err; 196 struct msgque_data msgque; 197 198 if (getuid() != 0) { 199 printf("Please run the test as root - Exiting.\n"); 200 return ksft_exit_fail(); 201 } 202 203 msgque.key = ftok(argv[0], 822155650); 204 if (msgque.key == -1) { 205 printf("Can't make key: %d\n", -errno); 206 return ksft_exit_fail(); 207 } 208 209 msgque.msq_id = msgget(msgque.key, IPC_CREAT | IPC_EXCL | 0666); 210 if (msgque.msq_id == -1) { 211 err = -errno; 212 printf("Can't create queue: %d\n", err); 213 goto err_out; 214 } 215 216 err = fill_msgque(&msgque); 217 if (err) { 218 printf("Failed to fill queue: %d\n", err); 219 goto err_destroy; 220 } 221 222 err = dump_queue(&msgque); 223 if (err) { 224 printf("Failed to dump queue: %d\n", err); 225 goto err_destroy; 226 } 227 228 err = check_and_destroy_queue(&msgque); 229 if (err) { 230 printf("Failed to check and destroy queue: %d\n", err); 231 goto err_out; 232 } 233 234 err = restore_queue(&msgque); 235 if (err) { 236 printf("Failed to restore queue: %d\n", err); 237 goto err_destroy; 238 } 239 240 err = check_and_destroy_queue(&msgque); 241 if (err) { 242 printf("Failed to test queue: %d\n", err); 243 goto err_out; 244 } 245 return ksft_exit_pass(); 246 247 err_destroy: 248 if (msgctl(msgque.msq_id, IPC_RMID, 0)) { 249 printf("Failed to destroy queue: %d\n", -errno); 250 return ksft_exit_fail(); 251 } 252 err_out: 253 return ksft_exit_fail(); 254 } 255
