Quelle  msgque.c   Sprache: C

// 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) {
   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();
}