mirror of
https://github.com/checkpoint-restore/criu
synced 2025-08-22 09:58:09 +00:00
93dd984ca0
Acked-by: Mike Rapoport <rppt@linux.ibm.com> Signed-off-by: Adrian Reber <areber@redhat.com>
137 lines
2.9 KiB
C
137 lines
2.9 KiB
C
#include <sched.h>
|
|
|
|
#include <stdio.h>
|
|
#include <string.h>
|
|
#include <stdlib.h>
|
|
#include <unistd.h>
|
|
#include <sys/types.h>
|
|
#include <sys/wait.h>
|
|
#include <sys/sem.h>
|
|
#include <sys/ipc.h>
|
|
#include <sys/msg.h>
|
|
#include <signal.h>
|
|
#include <errno.h>
|
|
|
|
#include "zdtmtst.h"
|
|
|
|
const char *test_doc = "Tests sysv5 msg queues supporting by checkpointing";
|
|
const char *test_author = "Stanislav Kinsbursky <skinsbursky@openvz.org>";
|
|
|
|
struct msg1 {
|
|
long mtype;
|
|
char mtext[30];
|
|
};
|
|
#define TEST_STRING "Test sysv5 msg"
|
|
#define MSG_TYPE 1
|
|
|
|
#define ANOTHER_TEST_STRING "Yet another test sysv5 msg"
|
|
#define ANOTHER_MSG_TYPE 26538
|
|
|
|
int main(int argc, char **argv)
|
|
{
|
|
key_t key;
|
|
int msg, pid;
|
|
struct msg1 msgbuf;
|
|
int chret;
|
|
|
|
test_init(argc, argv);
|
|
|
|
key = ftok(argv[0], 822155650);
|
|
if (key == -1) {
|
|
pr_perror("Can't make key");
|
|
exit(1);
|
|
}
|
|
|
|
pid = test_fork();
|
|
if (pid < 0) {
|
|
pr_perror("Can't fork");
|
|
exit(1);
|
|
}
|
|
|
|
msg = msgget(key, IPC_CREAT | IPC_EXCL | 0666);
|
|
if (msg == -1) {
|
|
msg = msgget(key, 0666);
|
|
if (msg == -1) {
|
|
pr_perror("Can't get queue");
|
|
goto err_kill;
|
|
}
|
|
}
|
|
|
|
if (pid == 0) {
|
|
test_waitsig();
|
|
|
|
if (msgrcv(msg, &msgbuf, sizeof(TEST_STRING), MSG_TYPE, IPC_NOWAIT) == -1) {
|
|
fail("Child: msgrcv failed");
|
|
return -errno;
|
|
}
|
|
|
|
if (strncmp(TEST_STRING, msgbuf.mtext, sizeof(TEST_STRING))) {
|
|
fail("Child: the source and received strings aren't equal");
|
|
return -errno;
|
|
}
|
|
test_msg("Child: received %s\n", msgbuf.mtext);
|
|
|
|
msgbuf.mtype = ANOTHER_MSG_TYPE;
|
|
memcpy(msgbuf.mtext, ANOTHER_TEST_STRING, sizeof(ANOTHER_TEST_STRING));
|
|
if (msgsnd(msg, &msgbuf, sizeof(ANOTHER_TEST_STRING), IPC_NOWAIT) != 0) {
|
|
fail("Child: msgsnd failed");
|
|
return -errno;
|
|
};
|
|
pass();
|
|
return 0;
|
|
} else {
|
|
msgbuf.mtype = MSG_TYPE;
|
|
memcpy(msgbuf.mtext, TEST_STRING, sizeof(TEST_STRING));
|
|
if (msgsnd(msg, &msgbuf, sizeof(TEST_STRING), IPC_NOWAIT) != 0) {
|
|
fail("Parent: msgsnd failed");
|
|
goto err_kill;
|
|
};
|
|
|
|
msgbuf.mtype = ANOTHER_MSG_TYPE;
|
|
memcpy(msgbuf.mtext, ANOTHER_TEST_STRING, sizeof(ANOTHER_TEST_STRING));
|
|
if (msgsnd(msg, &msgbuf, sizeof(ANOTHER_TEST_STRING), IPC_NOWAIT) != 0) {
|
|
fail("child: msgsnd (2) failed");
|
|
return -errno;
|
|
};
|
|
|
|
test_daemon();
|
|
test_waitsig();
|
|
|
|
kill(pid, SIGTERM);
|
|
|
|
wait(&chret);
|
|
chret = WEXITSTATUS(chret);
|
|
if (chret) {
|
|
fail("Parent: child exited with non-zero code %d (%s)", chret, strerror(chret));
|
|
goto out;
|
|
}
|
|
|
|
if (msgrcv(msg, &msgbuf, sizeof(ANOTHER_TEST_STRING), ANOTHER_MSG_TYPE, IPC_NOWAIT) == -1) {
|
|
fail("Parent: msgrcv failed");
|
|
goto err;
|
|
}
|
|
|
|
if (strncmp(ANOTHER_TEST_STRING, msgbuf.mtext, sizeof(ANOTHER_TEST_STRING))) {
|
|
fail("Parent: the source and received strings aren't equal");
|
|
goto err;
|
|
}
|
|
test_msg("Parent: received %s\n", msgbuf.mtext);
|
|
|
|
pass();
|
|
}
|
|
|
|
out:
|
|
if (msgctl(msg, IPC_RMID, 0)) {
|
|
fail("Failed to destroy message queue");
|
|
return -errno;
|
|
}
|
|
return chret;
|
|
|
|
err_kill:
|
|
kill(pid, SIGKILL);
|
|
wait(NULL);
|
|
err:
|
|
chret = -errno;
|
|
goto out;
|
|
}
|