feat(routing): Support session based routing
[ric-plt/lib/rmr.git] / test / test_support.c
1 // : vi ts=4 sw=4 noet :
2 /*
3 ==================================================================================
4             Copyright (c) 2019 Nokia
5             Copyright (c) 2018-2019 AT&T Intellectual Property.
6
7    Licensed under the Apache License, Version 2.0 (the "License");
8    you may not use this file except in compliance with the License.
9    You may obtain a copy of the License at
10
11            http://www.apache.org/licenses/LICENSE-2.0
12
13    Unless required by applicable law or agreed to in writing, software
14    distributed under the License is distributed on an "AS IS" BASIS,
15    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16    See the License for the specific language governing permissions and
17    limitations under the License.
18 ==================================================================================
19 */
20
21 /*
22         Mnemonic:       test_tools.c
23         Abstract:       Functions for test applications to make their life a bit easier.
24                                 This file is probably compiled to a .o, and then included on
25                                 the cc command for the test.
26         Author:         E. Scott Daniels
27         Date:           6 January 2019
28 */
29
30 #ifndef _test_support_c
31 #define _test_support_c
32
33 #include <signal.h>
34 #include <string.h>
35 #include <stdio.h>
36 #include <stdlib.h>
37 #include <fcntl.h>
38 #include <unistd.h>
39
40 #ifndef BAD
41 #define BAD 1                   // these are exit codes unless user overrides
42 #define GOOD 0
43 #endif
44
45 /*
46         Snag the optional positional parameter at pp, return defval if not there.
47 */
48 static char* snag_pp( int pp, int argc, char** argv, char* defval ) {
49
50         if( pp < argc ) {
51                 return argv[pp];
52         }
53
54         return defval;
55 }
56
57 /*
58         Signal handler -- inside of the tests we will exit cleanly for hup/temp/intr
59         signals so that the coverage stuff will generate the needed data files. If
60         we inter/term the process they don't drive.
61 */
62
63 void sig_clean_exit( int sign ) {
64         fprintf( stderr, "signal trapped for clean exit: %d\n", sign );
65         exit( 0 );
66 }
67
68 /*
69         Setup all of the signal handling for signals that we want to force a clean exit:
70         term, intr, hup, quit, usr1/2 alarm, etc.  All others we'll let default.
71 */
72 static void set_signals( void ) {
73         struct sigaction sa;
74         int     sig_list[] = { SIGINT, SIGQUIT, SIGILL, SIGALRM, SIGTERM, SIGUSR1 , SIGUSR2 };
75         int i;
76         int nele;               // number of elements in the list
77
78         nele = (int) ( sizeof( sig_list )/sizeof( int ) );              // convert raw size to the number of elements
79         for( i = 0; i < nele; i ++ ) {
80                 memset( &sa, 0, sizeof( sa ) );
81                 sa.sa_handler = sig_clean_exit;
82                 sigaction( sig_list[i], &sa, NULL );
83         }
84 }
85
86
87 static int fail_if_nil( void* p, char* what ) {
88         if( !p ) {
89                 fprintf( stderr, "<FAIL> %s: pointer was nil\n", what );
90         }
91         return p ? GOOD : BAD;
92 }
93
94 static int fail_not_nil( void* p, char* what ) {
95         if( p ) {
96                 fprintf( stderr, "<FAIL> %s: pointer was not nil\n", what );
97         }
98         return !p ? GOOD : BAD;
99 }
100
101 static int fail_if_false( int bv, char* what ) {
102         if( !bv ) {
103                 fprintf( stderr, "<FAIL> %s: expected true, boolean test was false (%d)\n", what, bv );
104         }
105
106         return bv ? GOOD : BAD;
107 }
108
109 static int fail_if_true( int bv, char* what ) {
110         if( bv ) {
111                 fprintf( stderr, "<FAIL> %s: expected false, boolean test was true (%d)\n", what, bv );
112         }
113         return bv ? BAD : GOOD;
114 }
115
116 /*
117         Same as fail_if_true(), but reads easier in the test code.
118 */
119 static int fail_if( int bv, char* what ) {
120
121         if( bv ) {
122                 fprintf( stderr, "<FAIL> %s: expected false, boolean test was true (%d)\n", what, bv );
123         }
124         return bv ? BAD : GOOD;
125 }
126
127 static int fail_not_equal( int a, int b, char* what ) {
128         if( a != b ) {
129                 fprintf( stderr, "<FAIL> %s: values were not equal a=%d b=%d\n", what, a, b );
130         }
131         return a == b ? GOOD : BAD;                     // user may override good/bad so do NOT return a==b directly!
132 }
133
134 static int fail_if_equal( int a, int b, char* what ) {
135         if( a == b ) {
136                 fprintf( stderr, "<FAIL> %s values were equal a=%d b=%d\n", what, a, b );
137         }
138         return a != b ? GOOD : BAD;                     // user may override good/bad so do NOT return a==b directly!
139 }
140
141
142 #ifndef NO_DUMMY_RMR
143 /*
144         Dummy message allocator for testing without sr_static functions
145 */
146 static rmr_mbuf_t* test_mk_msg( int len, int tr_len ) {
147         rmr_mbuf_t*     new_msg;
148         uta_mhdr_t* hdr;
149         size_t  alen;
150
151         alen = sizeof( *hdr ) + tr_len + len;
152
153         new_msg = (rmr_mbuf_t *) malloc( sizeof *new_msg );
154         new_msg->tp_buf = (void *) malloc( alen );
155         memset( new_msg->tp_buf, 0, alen );
156
157         hdr = (uta_mhdr_t*) new_msg->tp_buf;
158         SET_HDR_LEN( hdr );
159         SET_HDR_TR_LEN( hdr, tr_len );
160
161         new_msg->header = new_msg->tp_buf;
162         new_msg->payload =  new_msg->header + PAYLOAD_OFFSET( hdr );
163         new_msg->alloc_len = alen;
164         new_msg->len = 0;
165
166         return new_msg;
167 }
168 #endif
169
170 #endif