173 lines
4.0 KiB
C
173 lines
4.0 KiB
C
/*
|
|
* Copyright (c) 2014 Fujitsu Ltd.
|
|
* Author: Xiaoguang Wang <wangxg.fnst@cn.fujitsu.com>
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify it
|
|
* under the terms of version 2 of the GNU General Public License as
|
|
* published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it would be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
|
|
*
|
|
* You should have received a copy of the GNU General Public License along
|
|
* with this program; if not, write the Free Software Foundation, Inc.,
|
|
* 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*/
|
|
|
|
/*
|
|
* Description:
|
|
* Basic test for msgrcv(2) using MSG_EXCEPT, MSG_NOERROR
|
|
*/
|
|
|
|
#define _GNU_SOURCE
|
|
#include <sys/wait.h>
|
|
#include "test.h"
|
|
#include "ipcmsg.h"
|
|
|
|
|
|
#define MSGTYPE1 1
|
|
#define MSGTYPE2 2
|
|
#define MSG1 "message type1"
|
|
#define MSG2 "message type2"
|
|
|
|
static void wait4child(pid_t child, char *tst_flag);
|
|
|
|
static void test_msg_except(void);
|
|
static void test_msg_noerror(void);
|
|
|
|
static void (*testfunc[])(void) = { test_msg_except, test_msg_noerror };
|
|
|
|
char *TCID = "msgrcv07";
|
|
int TST_TOTAL = ARRAY_SIZE(testfunc);
|
|
|
|
int main(int ac, char **av)
|
|
{
|
|
int lc;
|
|
int i;
|
|
|
|
tst_parse_opts(ac, av, NULL, NULL);
|
|
|
|
setup();
|
|
|
|
for (lc = 0; TEST_LOOPING(lc); lc++) {
|
|
tst_count = 0;
|
|
|
|
for (i = 0; i < TST_TOTAL; i++)
|
|
(*testfunc[i])();
|
|
}
|
|
|
|
cleanup();
|
|
tst_exit();
|
|
}
|
|
|
|
void setup(void)
|
|
{
|
|
tst_sig(FORK, DEF_HANDLER, cleanup);
|
|
|
|
TEST_PAUSE;
|
|
}
|
|
|
|
static void test_msg_except(void)
|
|
{
|
|
pid_t child_pid;
|
|
int msgq_id;
|
|
MSGBUF snd_buf1 = {.mtype = MSGTYPE1, .mtext = MSG1};
|
|
MSGBUF snd_buf2 = {.mtype = MSGTYPE2, .mtext = MSG2};
|
|
MSGBUF rcv_buf;
|
|
|
|
msgq_id = msgget(IPC_PRIVATE, MSG_RW);
|
|
if (msgq_id == -1)
|
|
tst_brkm(TBROK | TERRNO, cleanup, "Can't create message queue");
|
|
|
|
if (msgsnd(msgq_id, &snd_buf1, MSGSIZE, 0) == -1)
|
|
tst_brkm(TBROK | TERRNO, cleanup, "Can't enqueue message");
|
|
|
|
if (msgsnd(msgq_id, &snd_buf2, MSGSIZE, 0) == -1)
|
|
tst_brkm(TBROK | TERRNO, cleanup, "Can't enqueue message");
|
|
|
|
child_pid = tst_fork();
|
|
if (child_pid == -1) {
|
|
tst_brkm(TBROK, cleanup, "fork failed");
|
|
} else if (child_pid > 0) {
|
|
wait4child(child_pid, "MSG_EXCEPT");
|
|
} else {
|
|
memset(&rcv_buf, 0, sizeof(rcv_buf));
|
|
TEST(msgrcv(msgq_id, &rcv_buf, MSGSIZE, MSGTYPE2, MSG_EXCEPT));
|
|
if (TEST_RETURN == -1) {
|
|
fprintf(stderr, "msgrcv(MSG_EXCEPT) failed\n");
|
|
exit(TBROK);
|
|
}
|
|
/* check the received message */
|
|
if (strcmp(rcv_buf.mtext, MSG1) == 0 &&
|
|
rcv_buf.mtype == MSGTYPE1)
|
|
exit(TPASS);
|
|
else
|
|
exit(TFAIL);
|
|
}
|
|
|
|
rm_queue(msgq_id);
|
|
}
|
|
|
|
|
|
static void test_msg_noerror(void)
|
|
{
|
|
pid_t child_pid;
|
|
int msg_len, msgq_id;
|
|
MSGBUF snd_buf1 = {.mtype = MSGTYPE1, .mtext = MSG1};
|
|
MSGBUF rcv_buf;
|
|
|
|
msgq_id = msgget(IPC_PRIVATE, MSG_RW);
|
|
if (msgq_id == -1)
|
|
tst_brkm(TBROK | TERRNO, cleanup, "Can't create message queue");
|
|
|
|
if (msgsnd(msgq_id, &snd_buf1, MSGSIZE, 0) == -1)
|
|
tst_brkm(TBROK | TERRNO, cleanup, "Can't enqueue message");
|
|
|
|
child_pid = tst_fork();
|
|
if (child_pid == -1) {
|
|
tst_brkm(TBROK, cleanup, "fork failed");
|
|
} else if (child_pid > 0) {
|
|
wait4child(child_pid, "MSG_NOERROR");
|
|
} else {
|
|
msg_len = sizeof(MSG1) / 2;
|
|
memset(&rcv_buf, 0, sizeof(rcv_buf));
|
|
|
|
TEST(msgrcv(msgq_id, &rcv_buf, msg_len, MSGTYPE1, MSG_NOERROR));
|
|
if (TEST_RETURN == -1)
|
|
exit(TFAIL);
|
|
|
|
if (strncmp(rcv_buf.mtext, MSG1, msg_len) == 0 &&
|
|
rcv_buf.mtype == MSGTYPE1)
|
|
exit(TPASS);
|
|
exit(TFAIL);
|
|
}
|
|
|
|
rm_queue(msgq_id);
|
|
}
|
|
|
|
static void wait4child(pid_t child, char *tst_flag)
|
|
{
|
|
int status;
|
|
int ret;
|
|
|
|
if (waitpid(child, &status, 0) == -1)
|
|
tst_resm(TBROK | TERRNO, "waitpid");
|
|
if (WIFEXITED(status)) {
|
|
ret = WEXITSTATUS(status);
|
|
if (ret == 0)
|
|
tst_resm(TPASS, "test %s success", tst_flag);
|
|
else if (ret == 1)
|
|
tst_resm(TFAIL, "test %s failed", tst_flag);
|
|
else
|
|
tst_brkm(TBROK, cleanup, "msgrcv failed unexpectedly");
|
|
} else {
|
|
tst_brkm(TBROK, cleanup, "child process terminated "
|
|
"abnormally. status: %d", status);
|
|
}
|
|
}
|
|
|
|
void cleanup(void)
|
|
{
|
|
}
|