270 lines
6.5 KiB
C
270 lines
6.5 KiB
C
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
/*
|
|
* Copyright (c) 2014-2020 Fujitsu Ltd.
|
|
* Author: Xiaoguang Wang <wangxg.fnst@cn.fujitsu.com>
|
|
* Author: Yang Xu <xuyang2018.jy@cn.fujitsu.com>
|
|
*
|
|
* Basic test for msgrcv(2) using MSG_EXCEPT, MSG_NOERROR, MSG_COPY and
|
|
* different msg_typ(zero,positive,negative).
|
|
*
|
|
* * With MSG_EXCEPT flag any message type but the one passed to the function
|
|
* is received.
|
|
*
|
|
* * With MSG_NOERROR and buffer size less than message size only part of the
|
|
* buffer is received.
|
|
*
|
|
* * With MSG_COPY and IPC_NOWAIT flag read the msg but don't destroy it in
|
|
* msg queue.
|
|
*
|
|
* * With msgtyp is 0, then the first message in the queue is read.
|
|
*
|
|
* * With msgtyp is greater than 0, then the first message in the queue of type
|
|
* msgtyp is read.
|
|
*
|
|
* * With msgtyp is less than 0, then the first message in the queue with the
|
|
* lowest type less than or equal to absolute value of msgtyp is received.
|
|
*/
|
|
|
|
#define _GNU_SOURCE
|
|
#include <sys/wait.h>
|
|
#include "tst_test.h"
|
|
#include "tst_safe_sysv_ipc.h"
|
|
#include "libnewipc.h"
|
|
#include "lapi/msg.h"
|
|
|
|
#define MSGTYPE1 1
|
|
#define MSGTYPE2 2
|
|
#define MSG1 "messagetype1"
|
|
#define MSG2 "messagetype2"
|
|
|
|
static key_t msgkey;
|
|
static int queue_id = -1, msg_copy_sup;
|
|
static struct buf {
|
|
long type;
|
|
char mtext[MSGSIZE];
|
|
} rcv_buf, snd_buf[2] = {
|
|
{MSGTYPE1, MSG1},
|
|
{MSGTYPE2, MSG2}
|
|
};
|
|
|
|
static void prepare_queue(void)
|
|
{
|
|
queue_id = SAFE_MSGGET(msgkey, IPC_CREAT | IPC_EXCL | MSG_RW);
|
|
SAFE_MSGSND(queue_id, &snd_buf[0], MSGSIZE, 0);
|
|
SAFE_MSGSND(queue_id, &snd_buf[1], MSGSIZE, 0);
|
|
memset(&rcv_buf, 0, sizeof(rcv_buf));
|
|
}
|
|
|
|
static void test_msg_except(void)
|
|
{
|
|
prepare_queue();
|
|
|
|
TEST(msgrcv(queue_id, &rcv_buf, MSGSIZE, MSGTYPE2, MSG_EXCEPT));
|
|
if (TST_RET == -1) {
|
|
tst_res(TFAIL | TTERRNO, "msgrcv(MSG_EXCEPT) failed");
|
|
goto exit;
|
|
}
|
|
|
|
tst_res(TPASS, "msgrcv(MSG_EXCEPT) succeeded");
|
|
|
|
if (strcmp(rcv_buf.mtext, MSG1) == 0 && rcv_buf.type == MSGTYPE1)
|
|
tst_res(TPASS, "MSG_EXCEPT excepted MSGTYPE2 and got MSGTYPE1");
|
|
else
|
|
tst_res(TFAIL, "MSG_EXCEPT didn't get MSGTYPE1 message");
|
|
|
|
exit:
|
|
SAFE_MSGCTL(queue_id, IPC_RMID, NULL);
|
|
queue_id = -1;
|
|
}
|
|
|
|
static void test_msg_noerror(void)
|
|
{
|
|
int msg_len;
|
|
|
|
queue_id = SAFE_MSGGET(msgkey, IPC_CREAT | IPC_EXCL | MSG_RW);
|
|
SAFE_MSGSND(queue_id, &snd_buf[0], MSGSIZE, 0);
|
|
msg_len = sizeof(MSG1) / 2;
|
|
memset(&rcv_buf, 0, sizeof(rcv_buf));
|
|
|
|
TEST(msgrcv(queue_id, &rcv_buf, msg_len, MSGTYPE1, MSG_NOERROR));
|
|
if (TST_RET == -1) {
|
|
tst_res(TFAIL | TTERRNO, "msgrcv(MSG_NOERROR) failed");
|
|
goto exit;
|
|
}
|
|
|
|
tst_res(TPASS, "msgrcv(MSG_NOERROR) succeeded");
|
|
|
|
if (strncmp(rcv_buf.mtext, MSG1, msg_len) == 0 && rcv_buf.type == MSGTYPE1)
|
|
tst_res(TPASS, "MSG_NOERROR truncated message correctly");
|
|
else
|
|
tst_res(TFAIL, "MSG_NOERROR truncated message incorrectly");
|
|
|
|
exit:
|
|
SAFE_MSGCTL(queue_id, IPC_RMID, NULL);
|
|
queue_id = -1;
|
|
}
|
|
|
|
static void test_msg_copy(void)
|
|
{
|
|
struct msqid_ds buf = {0};
|
|
|
|
if (!msg_copy_sup) {
|
|
tst_res(TCONF, "MSG_COPY not supported");
|
|
return;
|
|
}
|
|
|
|
prepare_queue();
|
|
|
|
/*
|
|
* If MSG_COPY flag was specified, then mtype is interpreted as number
|
|
* of the message to copy.
|
|
*/
|
|
TEST(msgrcv(queue_id, &rcv_buf, MSGSIZE, 0, MSG_COPY | IPC_NOWAIT));
|
|
if (TST_RET == -1) {
|
|
if (TST_ERR == ENOSYS) {
|
|
tst_res(TCONF,
|
|
"MSG_COPY needs CONFIG_CHECKPORINT_RESTORE");
|
|
msg_copy_sup = 0;
|
|
} else {
|
|
tst_res(TFAIL | TTERRNO, "msgrcv(0, MSG_COPY) failed");
|
|
}
|
|
goto exit;
|
|
}
|
|
|
|
tst_res(TPASS, "msgrcv(0, MSG_COPY) succeeded");
|
|
|
|
if (strcmp(rcv_buf.mtext, MSG1) == 0 && rcv_buf.type == MSGTYPE1)
|
|
tst_res(TPASS, "MSG_COPY got MSGTYPE1 data correctly");
|
|
else
|
|
tst_res(TFAIL, "MSG_COPY got MSGTYPE1 data incorrectly");
|
|
|
|
memset(&rcv_buf, 0, sizeof(rcv_buf));
|
|
TEST(msgrcv(queue_id, &rcv_buf, MSGSIZE, 1, MSG_COPY | IPC_NOWAIT));
|
|
if (TST_RET == -1) {
|
|
tst_res(TFAIL | TTERRNO, "msgrcv(1, MSG_COPY) failed");
|
|
goto exit;
|
|
}
|
|
|
|
tst_res(TPASS, "msgrcv(1, MSG_COPY) succeeded");
|
|
|
|
if (strcmp(rcv_buf.mtext, MSG2) == 0 && rcv_buf.type == MSGTYPE2)
|
|
tst_res(TPASS, "MSG_COPY got MSGTYPE2 data correctly");
|
|
else
|
|
tst_res(TFAIL, "MSG_COPY got MSGTYPE2 data incorrectly");
|
|
|
|
SAFE_MSGCTL(queue_id, IPC_STAT, &buf);
|
|
if (buf.msg_qnum == 2) {
|
|
tst_res(TPASS, "Two messages still in queue");
|
|
} else {
|
|
tst_res(TFAIL, "Expected 2 msgs in queue got %d",
|
|
(int)buf.msg_qnum);
|
|
}
|
|
|
|
exit:
|
|
SAFE_MSGCTL(queue_id, IPC_RMID, NULL);
|
|
queue_id = -1;
|
|
}
|
|
|
|
static void test_zero_msgtyp(void)
|
|
{
|
|
prepare_queue();
|
|
|
|
TEST(msgrcv(queue_id, &rcv_buf, MSGSIZE, 0, 0));
|
|
if (TST_RET == -1) {
|
|
tst_res(TFAIL | TTERRNO, "msgrcv(zero_msgtyp) failed");
|
|
goto exit;
|
|
}
|
|
|
|
tst_res(TPASS, "msgrcv(zero_msgtyp) succeeded");
|
|
|
|
if (strcmp(rcv_buf.mtext, MSG1) == 0 && rcv_buf.type == MSGTYPE1)
|
|
tst_res(TPASS, "zero_msgtyp got the first message");
|
|
else
|
|
tst_res(TFAIL, "zero_msgtyp didn't get the first message");
|
|
|
|
exit:
|
|
SAFE_MSGCTL(queue_id, IPC_RMID, NULL);
|
|
queue_id = -1;
|
|
}
|
|
|
|
static void test_positive_msgtyp(void)
|
|
{
|
|
prepare_queue();
|
|
|
|
TEST(msgrcv(queue_id, &rcv_buf, MSGSIZE, MSGTYPE2, 0));
|
|
if (TST_RET == -1) {
|
|
tst_res(TFAIL | TTERRNO, "msgrcv(positive_msgtyp) failed");
|
|
goto exit;
|
|
}
|
|
|
|
tst_res(TPASS, "msgrcv(positive_msgtyp) succeeded");
|
|
|
|
if (strcmp(rcv_buf.mtext, MSG2) == 0 && rcv_buf.type == MSGTYPE2) {
|
|
tst_res(TPASS,
|
|
"msgtyp got the first message in the queue of type msgtyp");
|
|
} else {
|
|
tst_res(TFAIL,
|
|
"msgtyp didn't get the first message in the queue of type msgtyp");
|
|
}
|
|
|
|
exit:
|
|
SAFE_MSGCTL(queue_id, IPC_RMID, NULL);
|
|
queue_id = -1;
|
|
}
|
|
|
|
static void test_negative_msgtyp(void)
|
|
{
|
|
prepare_queue();
|
|
|
|
TEST(msgrcv(queue_id, &rcv_buf, MSGSIZE, -MSGTYPE2, 0));
|
|
if (TST_RET == -1) {
|
|
tst_res(TFAIL | TTERRNO, "msgrcv(negative_msgtyp) failed");
|
|
goto exit;
|
|
}
|
|
|
|
tst_res(TPASS, "msgrcv(negative_msgtyp) succeeded");
|
|
|
|
if (strcmp(rcv_buf.mtext, MSG1) == 0 && rcv_buf.type == MSGTYPE1) {
|
|
tst_res(TPASS,
|
|
"-msgtyp got the first message in the queue with the lowest type");
|
|
} else {
|
|
tst_res(TFAIL,
|
|
"-msgtyp didn't get the first message in the queue with the lowest type");
|
|
}
|
|
|
|
exit:
|
|
SAFE_MSGCTL(queue_id, IPC_RMID, NULL);
|
|
queue_id = -1;
|
|
}
|
|
|
|
static void cleanup(void)
|
|
{
|
|
if (queue_id != -1)
|
|
SAFE_MSGCTL(queue_id, IPC_RMID, NULL);
|
|
}
|
|
|
|
static void setup(void)
|
|
{
|
|
msgkey = GETIPCKEY();
|
|
|
|
if (tst_kvercmp(3, 8, 0) >= 0)
|
|
msg_copy_sup = 1;
|
|
}
|
|
|
|
static void (*testfunc[])(void) = {test_msg_except, test_msg_noerror,
|
|
test_msg_copy, test_zero_msgtyp,
|
|
test_positive_msgtyp, test_negative_msgtyp};
|
|
|
|
static void verify_msgcrv(unsigned int n)
|
|
{
|
|
(*testfunc[n])();
|
|
}
|
|
|
|
static struct tst_test test = {
|
|
.needs_tmpdir = 1,
|
|
.setup = setup,
|
|
.cleanup = cleanup,
|
|
.test = verify_msgcrv,
|
|
.tcnt = ARRAY_SIZE(testfunc),
|
|
};
|