selftests: IPC message queue copy feature test
This test can be used to check wheither kernel supports IPC message queue copy and restore features (required by CRIU project). Signed-off-by: Stanislav Kinsbursky <skinsbursky@parallels.com> Cc: Serge Hallyn <serge.hallyn@canonical.com> Cc: "Eric W. Biederman" <ebiederm@xmission.com> Cc: Pavel Emelyanov <xemul@parallels.com> Cc: Al Viro <viro@zeniv.linux.org.uk> Cc: KOSAKI Motohiro <kosaki.motohiro@jp.fujitsu.com> Cc: Michael Kerrisk <mtk.manpages@gmail.com> Signed-off-by: Andrew Morton <akpm@linux-foundation.org> Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>
This commit is contained in:
parent
4a674f34ba
commit
3a665531a3
4 changed files with 275 additions and 2 deletions
|
@ -36,6 +36,7 @@ extern long do_msgsnd(int msqid, long mtype, void __user *mtext,
|
||||||
size_t msgsz, int msgflg);
|
size_t msgsz, int msgflg);
|
||||||
extern long do_msgrcv(int msqid, void __user *buf, size_t bufsz, long msgtyp,
|
extern long do_msgrcv(int msqid, void __user *buf, size_t bufsz, long msgtyp,
|
||||||
int msgflg,
|
int msgflg,
|
||||||
long (*msg_fill)(void __user *, struct msg_msg *, size_t));
|
long (*msg_fill)(void __user *, struct msg_msg *,
|
||||||
|
size_t));
|
||||||
|
|
||||||
#endif /* _LINUX_MSG_H */
|
#endif /* _LINUX_MSG_H */
|
||||||
|
|
|
@ -365,7 +365,8 @@ long compat_sys_msgrcv(int first, int second, int msgtyp, int third,
|
||||||
uptr = compat_ptr(ipck.msgp);
|
uptr = compat_ptr(ipck.msgp);
|
||||||
msgtyp = ipck.msgtyp;
|
msgtyp = ipck.msgtyp;
|
||||||
}
|
}
|
||||||
return do_msgrcv(first, uptr, second, msgtyp, third, compat_do_msg_fill);
|
return do_msgrcv(first, uptr, second, msgtyp, third,
|
||||||
|
compat_do_msg_fill);
|
||||||
}
|
}
|
||||||
#else
|
#else
|
||||||
long compat_sys_semctl(int semid, int semnum, int cmd, int arg)
|
long compat_sys_semctl(int semid, int semnum, int cmd, int arg)
|
||||||
|
|
25
tools/testing/selftests/ipc/Makefile
Normal file
25
tools/testing/selftests/ipc/Makefile
Normal file
|
@ -0,0 +1,25 @@
|
||||||
|
uname_M := $(shell uname -m 2>/dev/null || echo not)
|
||||||
|
ARCH ?= $(shell echo $(uname_M) | sed -e s/i.86/i386/)
|
||||||
|
ifeq ($(ARCH),i386)
|
||||||
|
ARCH := X86
|
||||||
|
CFLAGS := -DCONFIG_X86_32 -D__i386__
|
||||||
|
endif
|
||||||
|
ifeq ($(ARCH),x86_64)
|
||||||
|
ARCH := X86
|
||||||
|
CFLAGS := -DCONFIG_X86_64 -D__x86_64__
|
||||||
|
endif
|
||||||
|
|
||||||
|
CFLAGS += -I../../../../usr/include/
|
||||||
|
|
||||||
|
all:
|
||||||
|
ifeq ($(ARCH),X86)
|
||||||
|
gcc $(CFLAGS) msgque.c -o msgque_test
|
||||||
|
else
|
||||||
|
echo "Not an x86 target, can't build msgque selftest"
|
||||||
|
endif
|
||||||
|
|
||||||
|
run_tests: all
|
||||||
|
./msgque_test
|
||||||
|
|
||||||
|
clean:
|
||||||
|
rm -fr ./msgque_test
|
246
tools/testing/selftests/ipc/msgque.c
Normal file
246
tools/testing/selftests/ipc/msgque.c
Normal file
|
@ -0,0 +1,246 @@
|
||||||
|
#include <stdlib.h>
|
||||||
|
#include <stdio.h>
|
||||||
|
#include <string.h>
|
||||||
|
#include <errno.h>
|
||||||
|
#include <linux/msg.h>
|
||||||
|
#include <fcntl.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) {
|
||||||
|
printf("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)) {
|
||||||
|
printf("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) {
|
||||||
|
printf("Failed to create queue\n");
|
||||||
|
return -errno;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (id != msgque->msq_id) {
|
||||||
|
printf("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) {
|
||||||
|
printf("msgsnd failed (%m)\n");
|
||||||
|
ret = -errno;
|
||||||
|
goto destroy;
|
||||||
|
};
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
destroy:
|
||||||
|
if (msgctl(id, IPC_RMID, 0))
|
||||||
|
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;
|
||||||
|
printf("Failed to read IPC message: %m\n");
|
||||||
|
ret = -errno;
|
||||||
|
goto err;
|
||||||
|
}
|
||||||
|
if (ret != msgque->messages[cnt].msize) {
|
||||||
|
printf("Wrong message size: %d (expected %d)\n", ret,
|
||||||
|
msgque->messages[cnt].msize);
|
||||||
|
ret = -EINVAL;
|
||||||
|
goto err;
|
||||||
|
}
|
||||||
|
if (message.mtype != msgque->messages[cnt].mtype) {
|
||||||
|
printf("Wrong message type\n");
|
||||||
|
ret = -EINVAL;
|
||||||
|
goto err;
|
||||||
|
}
|
||||||
|
if (memcmp(message.mtext, msgque->messages[cnt].mtext, ret)) {
|
||||||
|
printf("Wrong message content\n");
|
||||||
|
ret = -EINVAL;
|
||||||
|
goto err;
|
||||||
|
}
|
||||||
|
cnt++;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (cnt != msgque->qnum) {
|
||||||
|
printf("Wrong message number\n");
|
||||||
|
ret = -EINVAL;
|
||||||
|
goto err;
|
||||||
|
}
|
||||||
|
|
||||||
|
ret = 0;
|
||||||
|
err:
|
||||||
|
if (msgctl(msgque->msq_id, IPC_RMID, 0)) {
|
||||||
|
printf("Failed to destroy queue: %d\n", -errno);
|
||||||
|
return -errno;
|
||||||
|
}
|
||||||
|
return ret;
|
||||||
|
}
|
||||||
|
|
||||||
|
int dump_queue(struct msgque_data *msgque)
|
||||||
|
{
|
||||||
|
struct msqid64_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;
|
||||||
|
printf("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) {
|
||||||
|
printf("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) {
|
||||||
|
printf("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) {
|
||||||
|
printf("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) {
|
||||||
|
printf("Second message send failed (%m)\n");
|
||||||
|
return -errno;
|
||||||
|
};
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
int main(int argc, char **argv)
|
||||||
|
{
|
||||||
|
int msg, pid, err;
|
||||||
|
struct msgque_data msgque;
|
||||||
|
|
||||||
|
msgque.key = ftok(argv[0], 822155650);
|
||||||
|
if (msgque.key == -1) {
|
||||||
|
printf("Can't make key\n");
|
||||||
|
return -errno;
|
||||||
|
}
|
||||||
|
|
||||||
|
msgque.msq_id = msgget(msgque.key, IPC_CREAT | IPC_EXCL | 0666);
|
||||||
|
if (msgque.msq_id == -1) {
|
||||||
|
printf("Can't create queue\n");
|
||||||
|
goto err_out;
|
||||||
|
}
|
||||||
|
|
||||||
|
err = fill_msgque(&msgque);
|
||||||
|
if (err) {
|
||||||
|
printf("Failed to fill queue\n");
|
||||||
|
goto err_destroy;
|
||||||
|
}
|
||||||
|
|
||||||
|
err = dump_queue(&msgque);
|
||||||
|
if (err) {
|
||||||
|
printf("Failed to dump queue\n");
|
||||||
|
goto err_destroy;
|
||||||
|
}
|
||||||
|
|
||||||
|
err = check_and_destroy_queue(&msgque);
|
||||||
|
if (err) {
|
||||||
|
printf("Failed to check and destroy queue\n");
|
||||||
|
goto err_out;
|
||||||
|
}
|
||||||
|
|
||||||
|
err = restore_queue(&msgque);
|
||||||
|
if (err) {
|
||||||
|
printf("Failed to restore queue\n");
|
||||||
|
goto err_destroy;
|
||||||
|
}
|
||||||
|
|
||||||
|
err = check_and_destroy_queue(&msgque);
|
||||||
|
if (err) {
|
||||||
|
printf("Failed to test queue\n");
|
||||||
|
goto err_out;
|
||||||
|
}
|
||||||
|
return 0;
|
||||||
|
|
||||||
|
err_destroy:
|
||||||
|
if (msgctl(msgque.msq_id, IPC_RMID, 0)) {
|
||||||
|
printf("Failed to destroy queue: %d\n", -errno);
|
||||||
|
return -errno;
|
||||||
|
}
|
||||||
|
err_out:
|
||||||
|
return err;
|
||||||
|
}
|
Loading…
Reference in a new issue