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