summaryrefslogtreecommitdiffstats
path: root/src/usr/testcore/kernel/msgtest.H
blob: 6f3415e6cced3816b80445ac59f124408f38bbac (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
/* IBM_PROLOG_BEGIN_TAG                                                   */
/* This is an automatically generated prolog.                             */
/*                                                                        */
/* $Source: src/usr/testcore/kernel/msgtest.H $                           */
/*                                                                        */
/* IBM CONFIDENTIAL                                                       */
/*                                                                        */
/* COPYRIGHT International Business Machines Corp. 2012                   */
/*                                                                        */
/* p1                                                                     */
/*                                                                        */
/* Object Code Only (OCO) source materials                                */
/* Licensed Internal Code Source Materials                                */
/* IBM HostBoot Licensed Internal Code                                    */
/*                                                                        */
/* The source code for this program is not published or otherwise         */
/* divested of its trade secrets, irrespective of what has been           */
/* deposited with the U.S. Copyright Office.                              */
/*                                                                        */
/* Origin: 30                                                             */
/*                                                                        */
/* IBM_PROLOG_END_TAG                                                     */

/** @file msgtest.H
 *  @brief Testcases for the messaging subsystem of the kernel.
 */

#include <cxxtest/TestSuite.H>
#include <sys/task.h>
#include <sys/msg.h>

class MessagingTest : public CxxTest::TestSuite
{
    public:

        /** Test sync-messaging interfaces */
        void testSync()
        {
            msg_q_t client = msg_q_create(),
                    server = msg_q_create();

            msg_t* sync_msg = NULL;
            int rc = 0;

            tid_t server_task = task_create(MessagingTest::processMsg,
                                            server);

            // Send a message with the normal sync messaging interface.
            sync_msg = msg_allocate();
            sync_msg->type = ECHO;
            sync_msg->data[0] = 1;
            rc = msg_sendrecv(server, sync_msg);
            if (rc)
            {
                TS_FAIL("Failure to send message (msg_sendrecv).");
            }
            else if (sync_msg->data[0] != (2 << 1))
            {
                TS_FAIL("Message response doesn't match expected value for "
                        "sync message.  %d", sync_msg->data[0]);
            }
            msg_free(sync_msg);

            // Send a message with the non-blocking sync message interface.
            barrier_t barrier;
            barrier_init(&barrier, 2);
            sync_msg = msg_allocate();
            sync_msg->type = ECHO;
            sync_msg->data[0] = 2;
            sync_msg->data[1] = reinterpret_cast<uint64_t>(&barrier);
            rc = msg_sendrecv_noblk(server, sync_msg, client);
            barrier_wait(&barrier);
            sync_msg = msg_wait(client);
            if (rc)
            {
                TS_FAIL("Failure to send message (msg_sendrecv_noblk)");
            }
            else if (sync_msg->data[0] != (2 << 2))
            {
                TS_FAIL("Message response doesn't match expected value for "
                        "no-blk sync message.  %d", sync_msg->data[0]);
            }
            msg_free(sync_msg);
            barrier_destroy(&barrier);

            // Shutdown the child thread.
            msg_t* shutdown_msg = msg_allocate();
            shutdown_msg->type = SHUTDOWN;
            msg_send(server, shutdown_msg);

            task_wait_tid(server_task, NULL, NULL);

        };

    private:

        enum msg_types { SHUTDOWN, ECHO };

        static void* processMsg(void* _msgQ)
        {
            msg_q_t msgQ = static_cast<msg_q_t>(_msgQ);

            while(msg_t* msg = msg_wait(msgQ))
            {
                switch (msg->type)
                {
                    case SHUTDOWN: // Shutdown.
                        msg_free(msg);
                        return NULL;

                    case ECHO:
                        if (msg->data[1])
                        {
                            barrier_wait(
                                reinterpret_cast<barrier_t*>(msg->data[1]));
                        }
                        msg->data[0] = (2 << msg->data[0]);
                        msg_respond(msgQ, msg);
                        break;
                }
            }
            return NULL;
        }
};
OpenPOWER on IntegriCloud