Something went wrong. Try again.
Linux kernel mirror (for testing) git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git
Something went wrong. Try again.
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257// SPDX-License-Identifier: GPL-2.0#define _GNU_SOURCE#include <stdlib.h>#include <stdio.h>#include <string.h>#include <errno.h>#include <sys/msg.h>#include <fcntl.h>
#include "kselftest.h"
#define MAX_MSG_SIZE 32
struct msg1 { int msize; long mtype; char mtext[MAX_MSG_SIZE];};
#define TEST_STRING "Test sysv5 msg"#define MSG_TYPE 1
#define ANOTHER_TEST_STRING "Yet another test sysv5 msg"#define ANOTHER_MSG_TYPE 26538
struct msgque_data { key_t key; int msq_id; int qbytes; int qnum; int mode; struct msg1 *messages;};
int restore_queue(struct msgque_data *msgque){ int fd, ret, id, i; char buf[32];
fd = open("/proc/sys/kernel/msg_next_id", O_WRONLY); if (fd == -1) { ksft_test_result_fail("Failed to open /proc/sys/kernel/msg_next_id\n"); return -errno; } sprintf(buf, "%d", msgque->msq_id);
ret = write(fd, buf, strlen(buf)); if (ret != strlen(buf)) { ksft_test_result_fail("Failed to write to /proc/sys/kernel/msg_next_id\n"); return -errno; }
id = msgget(msgque->key, msgque->mode | IPC_CREAT | IPC_EXCL); if (id == -1) { ksft_test_result_fail("Failed to create queue\n"); return -errno; }
if (id != msgque->msq_id) { ksft_test_result_fail("Restored queue has wrong id (%d instead of %d)\n" , id, msgque->msq_id); ret = -EFAULT; goto destroy; }
for (i = 0; i < msgque->qnum; i++) { if (msgsnd(msgque->msq_id, &msgque->messages[i].mtype, msgque->messages[i].msize, IPC_NOWAIT) != 0) { ksft_test_result_fail("msgsnd failed (%m)\n"); ret = -errno; goto destroy; } } return 0;
destroy: if (msgctl(id, IPC_RMID, NULL)) printf("Failed to destroy queue: %d\n", -errno); return ret;}
int check_and_destroy_queue(struct msgque_data *msgque){ struct msg1 message; int cnt = 0, ret;
while (1) { ret = msgrcv(msgque->msq_id, &message.mtype, MAX_MSG_SIZE, 0, IPC_NOWAIT); if (ret < 0) { if (errno == ENOMSG) break; ksft_test_result_fail("Failed to read IPC message: %m\n"); ret = -errno; goto err; } if (ret != msgque->messages[cnt].msize) { ksft_test_result_fail("Wrong message size: %d (expected %d)\n", ret, msgque->messages[cnt].msize); ret = -EINVAL; goto err; } if (message.mtype != msgque->messages[cnt].mtype) { ksft_test_result_fail("Wrong message type\n"); ret = -EINVAL; goto err; } if (memcmp(message.mtext, msgque->messages[cnt].mtext, ret)) { ksft_test_result_fail("Wrong message content\n"); ret = -EINVAL; goto err; } cnt++; }
if (cnt != msgque->qnum) { ksft_test_result_fail("Wrong message number\n"); ret = -EINVAL; goto err; }
ret = 0;err: if (msgctl(msgque->msq_id, IPC_RMID, NULL)) { printf("Failed to destroy queue: %d\n", -errno); return -errno; } return ret;}
int dump_queue(struct msgque_data *msgque){ struct msqid_ds ds; int kern_id; int i, ret;
for (kern_id = 0; kern_id < 256; kern_id++) { ret = msgctl(kern_id, MSG_STAT, &ds); if (ret < 0) { if (errno == EINVAL) continue; ksft_test_result_fail("Failed to get stats for IPC queue with id %d\n", kern_id); return -errno; }
if (ret == msgque->msq_id) break; }
msgque->messages = malloc(sizeof(struct msg1) * ds.msg_qnum); if (msgque->messages == NULL) { ksft_test_result_fail("Failed to get stats for IPC queue\n"); return -ENOMEM; }
msgque->qnum = ds.msg_qnum; msgque->mode = ds.msg_perm.mode; msgque->qbytes = ds.msg_qbytes;
for (i = 0; i < msgque->qnum; i++) { ret = msgrcv(msgque->msq_id, &msgque->messages[i].mtype, MAX_MSG_SIZE, i, IPC_NOWAIT | MSG_COPY); if (ret < 0) { if (errno == ENOSYS) ksft_exit_skip("MSG_COPY not supported\n");
ksft_test_result_fail("Failed to copy IPC message: %m (%d)\n", errno); return -errno; } msgque->messages[i].msize = ret; } return 0;}
int fill_msgque(struct msgque_data *msgque){ struct msg1 msgbuf;
msgbuf.mtype = MSG_TYPE; memcpy(msgbuf.mtext, TEST_STRING, sizeof(TEST_STRING)); if (msgsnd(msgque->msq_id, &msgbuf.mtype, sizeof(TEST_STRING), IPC_NOWAIT) != 0) { ksft_test_result_fail("First message send failed (%m)\n"); return -errno; }
msgbuf.mtype = ANOTHER_MSG_TYPE; memcpy(msgbuf.mtext, ANOTHER_TEST_STRING, sizeof(ANOTHER_TEST_STRING)); if (msgsnd(msgque->msq_id, &msgbuf.mtype, sizeof(ANOTHER_TEST_STRING), IPC_NOWAIT) != 0) { ksft_test_result_fail("Second message send failed (%m)\n"); return -errno; } return 0;}
int main(int argc, char **argv){ int err; struct msgque_data msgque;
if (getuid() != 0) ksft_exit_skip("Please run the test as root - Exiting.\n");
msgque.key = ftok(argv[0], 822155650); if (msgque.key == -1) { ksft_test_result_fail("Can't make key: %d\n", -errno); ksft_exit_fail(); }
msgque.msq_id = msgget(msgque.key, IPC_CREAT | IPC_EXCL | 0666); if (msgque.msq_id == -1) { err = -errno; ksft_test_result_fail("Can't create queue: %d\n", err); goto err_out; }
err = fill_msgque(&msgque); if (err) { ksft_test_result_fail("Failed to fill queue: %d\n", err); goto err_destroy; }
err = dump_queue(&msgque); if (err) { ksft_test_result_fail("Failed to dump queue: %d\n", err); goto err_destroy; }
err = check_and_destroy_queue(&msgque); if (err) { ksft_test_result_fail("Failed to check and destroy queue: %d\n", err); goto err_out; }
err = restore_queue(&msgque); if (err) { ksft_test_result_fail("Failed to restore queue: %d\n", err); goto err_destroy; }
err = check_and_destroy_queue(&msgque); if (err) { ksft_test_result_fail("Failed to test queue: %d\n", err); goto err_out; } ksft_exit_pass();
err_destroy: if (msgctl(msgque.msq_id, IPC_RMID, NULL)) { printf("Failed to destroy queue: %d\n", -errno); ksft_exit_fail(); }err_out: ksft_exit_fail();}