create a new case to test MSG_EXCEPT, MSG_NOERROR flag for msgrcv(2)

Signed-off-by: Xiaoguang Wang <[email protected]>
---
 runtest/ltplite                                 |   1 +
 runtest/stress.part3                            |   1 +
 runtest/syscalls                                |   1 +
 runtest/syscalls-ipc                            |   1 +
 testcases/kernel/syscalls/ipc/msgrcv/msgrcv07.c | 227 ++++++++++++++++++++++++
 5 files changed, 231 insertions(+)
 create mode 100644 testcases/kernel/syscalls/ipc/msgrcv/msgrcv07.c

diff --git a/runtest/ltplite b/runtest/ltplite
index c90bc48..d22008a 100644
--- a/runtest/ltplite
+++ b/runtest/ltplite
@@ -488,6 +488,7 @@ msgrcv03 msgrcv03
 msgrcv04 msgrcv04
 msgrcv05 msgrcv05
 msgrcv06 msgrcv06
+msgrcv07 msgrcv07
 
 msgsnd01 msgsnd01
 msgsnd02 msgsnd02
diff --git a/runtest/stress.part3 b/runtest/stress.part3
index eac28d0..b833c68 100644
--- a/runtest/stress.part3
+++ b/runtest/stress.part3
@@ -411,6 +411,7 @@ msgrcv03 msgrcv03
 msgrcv04 msgrcv04
 msgrcv05 msgrcv05
 msgrcv06 msgrcv06
+msgrcv07 msgrcv07
 
 msgsnd01 msgsnd01
 msgsnd02 msgsnd02
diff --git a/runtest/syscalls b/runtest/syscalls
index c5bbe8f..1eb23d2 100644
--- a/runtest/syscalls
+++ b/runtest/syscalls
@@ -639,6 +639,7 @@ msgrcv03 msgrcv03
 msgrcv04 msgrcv04
 msgrcv05 msgrcv05
 msgrcv06 msgrcv06
+msgrcv07 msgrcv07
 
 msgsnd01 msgsnd01
 msgsnd02 msgsnd02
diff --git a/runtest/syscalls-ipc b/runtest/syscalls-ipc
index f57a96e..ea1447c 100644
--- a/runtest/syscalls-ipc
+++ b/runtest/syscalls-ipc
@@ -21,6 +21,7 @@ msgrcv03 msgrcv03
 msgrcv04 msgrcv04
 msgrcv05 msgrcv05
 msgrcv06 msgrcv06
+msgrcv07 msgrcv07
 
 msgsnd01 msgsnd01
 msgsnd02 msgsnd02
diff --git a/testcases/kernel/syscalls/ipc/msgrcv/msgrcv07.c 
b/testcases/kernel/syscalls/ipc/msgrcv/msgrcv07.c
new file mode 100644
index 0000000..576f2ff
--- /dev/null
+++ b/testcases/kernel/syscalls/ipc/msgrcv/msgrcv07.c
@@ -0,0 +1,227 @@
+/*
+ * Copyright (c) 2013 Fujitsu Ltd.
+ * Author: Xiaoguang Wang <[email protected]>
+ *
+ * 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 "usctest.h"
+#include "ipcmsg.h"
+
+
+#define MSGTYPE1       1
+#define MSGTYPE2       2
+#define MSG1   "message type1"
+#define MSG2   "message type2"
+
+static void wait4child(pid_t child, int index);
+static void creat_msg_queue(void);
+
+static void setup_msg_except(void);
+static void test_msg_except(int index);
+static void cleanup_msg_except(void);
+
+static void setup_msg_noerror(void);
+static void test_msg_noerror(int index);
+static void cleanup_msg_noerror(void);
+
+static int msgq_id;
+MSGBUF rcv_buf, snd_buf1, snd_buf2;
+static int tst_result;
+
+static struct test_case_t {
+       int msgflg;
+       void (*setup)(void);
+       void (*testfunc)(int index);
+       void (*cleanup)(void);
+       char *des;
+} test_cases[] = {
+       {MSG_EXCEPT, setup_msg_except, test_msg_except, cleanup_msg_except,
+        "MSG_EXCEPT"},
+       {MSG_NOERROR, setup_msg_noerror, test_msg_noerror,
+        cleanup_msg_noerror, "MSG_NOERROR"},
+};
+
+char *TCID = "msgrcv07";
+int TST_TOTAL = ARRAY_SIZE(test_cases);
+
+int main(int ac, char **av)
+{
+       int lc;
+       char *msg;
+       int i;
+
+       msg = parse_opts(ac, av, NULL, NULL);
+       if (msg != NULL)
+               tst_brkm(TBROK, NULL, "OPTION PARSING ERROR - %s", msg);
+
+       setup();
+
+       for (lc = 0; TEST_LOOPING(lc); lc++) {
+               tst_count = 0;
+
+               for (i = 0; i < TST_TOTAL; i++) {
+                       if (test_cases[i].setup)
+                               test_cases[i].setup();
+                       if (test_cases[i].testfunc)
+                               test_cases[i].testfunc(i);
+                       if (test_cases[i].cleanup)
+                               test_cases[i].cleanup();
+               }
+       }
+
+       cleanup();
+       tst_exit();
+}
+
+void setup(void)
+{
+       tst_sig(FORK, DEF_HANDLER, cleanup);
+
+       tst_tmpdir();
+
+       TEST_PAUSE;
+
+       snd_buf1.mtype = MSGTYPE1;
+       strcpy(snd_buf1.mtext, MSG1);
+       snd_buf2.mtype = MSGTYPE2;
+       strcpy(snd_buf2.mtext, MSG2);
+}
+
+void creat_msg_queue(void)
+{
+       msgkey = getipckey();
+
+       /* create a message queue with read/write permission */
+       msgq_id = msgget(msgkey, IPC_CREAT | IPC_EXCL | MSG_RW);
+       if (msgq_id == -1)
+               tst_brkm(TBROK | TERRNO, cleanup, "Can't create message queue");
+}
+
+void setup_msg_except(void)
+{
+       creat_msg_queue();
+
+       /* put the MSG1 on the queue */
+       if (msgsnd(msgq_id, &snd_buf1, MSGSIZE, 0) == -1)
+               tst_brkm(TBROK | TERRNO, cleanup, "Can't enqueue message");
+
+       /* put the MSG2 on the queue */
+       if (msgsnd(msgq_id, &snd_buf2, MSGSIZE, 0) == -1)
+               tst_brkm(TBROK | TERRNO, cleanup, "Can't enqueue message");
+}
+
+void test_msg_except(int index)
+{
+       pid_t child_pid;
+
+       fflush(stdout);
+       child_pid = FORK_OR_VFORK();
+       if (child_pid == -1) {
+               tst_brkm(TBROK, cleanup, "fork failed");
+       } else if (child_pid > 0) {
+               wait4child(child_pid, index);
+       } else {
+               TEST(msgrcv(msgq_id, &rcv_buf, MSGSIZE, MSGTYPE2,
+                    test_cases[index].msgflg));
+               if (TEST_RETURN == -1) {
+                       fprintf(stderr, "msgrcv failed\n");
+                       exit(TBROK);
+               }
+               /* check the received message */
+               if (strcmp(rcv_buf.mtext, MSG1) == 0)
+                       tst_result = TPASS;
+               else
+                       tst_result = TFAIL;
+               exit(tst_result);
+       }
+}
+
+static void cleanup_msg_except(void)
+{
+       rm_queue(msgq_id);
+}
+
+static void setup_msg_noerror(void)
+{
+       creat_msg_queue();
+
+       /* put the MSG1 on the queue */
+       if (msgsnd(msgq_id, &snd_buf1, MSGSIZE, 0) == -1)
+               tst_brkm(TBROK | TERRNO, cleanup, "Can't enqueue message");
+}
+
+static void test_msg_noerror(int index)
+{
+       pid_t child_pid;
+       int msg_len;
+
+       fflush(stdout);
+       child_pid = FORK_OR_VFORK();
+       if (child_pid == -1) {
+               tst_brkm(TBROK, cleanup, "fork failed");
+       } else if (child_pid > 0) {
+               wait4child(child_pid, index);
+       } else {
+               msg_len = sizeof(MSG1) / 2;
+               TEST(msgrcv(msgq_id, &rcv_buf, msg_len, MSGTYPE1,
+                    test_cases[index].msgflg));
+               if (TEST_RETURN == -1)
+                       tst_result = TFAIL;
+               else
+                       tst_result = TPASS;
+               exit(tst_result);
+       }
+}
+
+static void cleanup_msg_noerror(void)
+{
+       rm_queue(msgq_id);
+}
+
+static void wait4child(pid_t child, int index)
+{
+       int status;
+       int ret;
+
+       if (waitpid(child, &status, 0) == -1)
+               tst_resm(TBROK | TERRNO, "waitpid");
+       if (WIFEXITED(status))
+               ret = WEXITSTATUS(status);
+       else
+               ret = status;
+
+       if (ret == 0) {
+               tst_resm(TPASS, "test %s success", test_cases[index].des);
+       } else if (ret == 1) {
+               tst_resm(TFAIL, "test %s failed, status: %d",
+                        test_cases[index].des, status);
+       } else {
+               tst_brkm(TBROK | TERRNO, cleanup, "msgrcv failed unexpectedly");
+       }
+}
+
+void cleanup(void)
+{
+       TEST_CLEANUP;
+
+       tst_rmdir();
+}
-- 
1.8.2.1


------------------------------------------------------------------------------
Rapidly troubleshoot problems before they affect your business. Most IT 
organizations don't have a clear picture of how application performance 
affects their revenue. With AppDynamics, you get 100% visibility into your 
Java,.NET, & PHP application. Start your 15-day FREE TRIAL of AppDynamics Pro!
http://pubads.g.doubleclick.net/gampad/clk?id=84349831&iu=/4140/ostg.clktrk
_______________________________________________
Ltp-list mailing list
[email protected]
https://lists.sourceforge.net/lists/listinfo/ltp-list

Reply via email to