X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=test%2Ftest_nng_em.c;h=c2101f5490d95f611f4bc5a4cde7c3944526f3f3;hb=refs%2Fchanges%2F44%2F244%2F4;hp=685a43c3c4b3877ef5d1d865adee13c689763494;hpb=8790bf0c4f4f08fd05853afa67e211112b344a42;p=ric-plt%2Flib%2Frmr.git diff --git a/test/test_nng_em.c b/test/test_nng_em.c index 685a43c..c2101f5 100644 --- a/test/test_nng_em.c +++ b/test/test_nng_em.c @@ -29,13 +29,28 @@ Author: E. Scott Daniels */ + +#include "rmr.h" // we use some of rmr defs in building dummy messages, so we need these +#include "rmr_agnostic.h" + // ---------------------- emulated nng functions --------------------------- #ifndef _em_nn #define _em_nn +#include + static int em_send_failures = 0; // test programme can set this to emulate eagain send failures +static int em_timeout = -1; // set by set socket option +static int em_mtc_msgs = 0; // set to generate 'received' messages with mt-call header data +static int return_value = 0; // functions should return this value +static int rcv_count = 0; // receive counter for transaction id to allow test to rest +static int rcv_delay = 0; // forced delay before call to rcvmsg starts to work + +static int gates_ok = 0; +static pthread_mutex_t rcv_gate; + // ----------- epoll emulation --------------------------------------------- @@ -56,29 +71,6 @@ static int em_wait( int fd, void* events, int n, int to ) { -//-------------------------------------------------------------------------- -#ifdef EMULATE_NNG -struct nn_msghdr { - int boo; -}; - -static int return_value = 0; - -/* - Test app can call this to have all emulated functions return failure instead - of success. -*/ -static void en_set_retur( int rv ) { - return_value = rv; -} - - - -static int em_nng_foo() { - fprintf( stderr, "emulated functions in play" ); -} - - /* Simulated v1 message for receive to return. This needs to match the RMr header so that we can fill in length, type and xaction id things. @@ -117,37 +109,128 @@ struct em_msg { int32_t len1; // length of the tracing data int32_t len2; // length of data 1 (d1) int32_t len3; // length of data 2 (d2) + int32_t sub_id; // subscription id (-1 invalid) +}; + + + +// -- emulation control functions ------------------------------------------------------ + +/* + Test app can call this to have all emulated functions return failure instead + of success. +*/ +static void en_set_return( int rv ) { + return_value = rv; +} + + + +static int em_nng_foo() { + fprintf( stderr, "emulated functions in play" ); +} + + +/* + Turns on/off the generation of multi-threaded call messages +*/ +static int em_set_mtc_msgs( int state ) { + em_mtc_msgs = state; +} + +/* + Returns the size of the header we inserted +*/ +static int em_hdr_size() { + if( em_mtc_msgs ) { + return (int) sizeof( struct em_msg ) + 4; + } + + return (int) sizeof( struct em_msg ); +} + +static void em_set_rcvcount( int v ) { + rcv_count = v; +} + +static void em_set_rcvdelay( int v ) { + rcv_delay = v; +} + +static void em_start() { + if( ! gates_ok ) { + pthread_mutex_init( &rcv_gate, NULL ); + gates_ok = 1; + } +} +//-------------------------------------------------------------------------- +#ifdef EMULATE_NNG +struct nn_msghdr { + int boo; }; + /* Receive message must allocate a new buffer and return the pointer into *m. Every 9 messages or so we'll simulate an old version message + + If em_mtc_msgs is set, then we add a non-zero d1 field with + the call-id set to 2, and alternate the call flag */ static int em_nng_recvmsg( nng_socket s, nng_msg ** m, int i ) { + static int call_flag = 0; + void* b; struct em_msg* msg; - static int count = 0; // we'll simulate a message going in by dropping an rmr-ish msg with transaction id only int trace_size = 0; + int d1_size = 0; + unsigned char* d1; - //sleep( 1 ); + if( rcv_delay > 0 ) { + sleep( rcv_delay ); + } + + if( em_mtc_msgs ) { + d1_size = 4; + } b = (void *) malloc( 2048 ); if( m != NULL ) { memset( b, 0, 2048 ); + *m = (nng_msg *) b; msg = (struct em_msg *) b; - if( count % 10 == 9 ) { - //msg->rmr_ver = htonl( MSG_VER ); - msg->rmr_ver = ALT_MSG_VER; // emulate the bug in RMr v1 + if( ! em_mtc_msgs && (rcv_count % 10) == 9 ) { + msg->rmr_ver = ALT_MSG_VER; // allow emulation the bug in RMr v1 } else { msg->rmr_ver = htonl( MSG_VER ); } + msg->mtype = htonl( 1 ); - msg->plen = htonl( 129 ); + msg->plen = htonl( 220 ); msg->len0 = htonl( sizeof( struct em_msg ) ); msg->len1 = htonl( trace_size ); - snprintf( msg->xid, 32, "%015d", count++ ); // simple transaction id so we can test receive specific and ring stuff + msg->len2 = htonl( d1_size ); + msg->len3 = htonl( 0 ); + + pthread_mutex_lock( &rcv_gate ); // hold lock to update counter/flag + if( em_mtc_msgs ) { + d1 = DATA1_ADDR( msg ); + d1[0] = 2; // simulated msgs always on chute 2 + if( call_flag ) { + rcv_count++; + msg->flags |= HFL_CALL_MSG; + } + if( rcv_delay > 0 ) { + fprintf( stderr, " count=%d flag=%d %02x \n", rcv_count, call_flag, msg->flags ); + } + call_flag = !call_flag; + } else { + rcv_count++; + } + pthread_mutex_unlock( &rcv_gate ); + snprintf( msg->xid, 32, "%015d", rcv_count ); // simple transaction id so we can test receive specific and ring stuff snprintf( msg->src, 16, "localhost:4562" ); // set src id (unrealistic) so that rts() can be tested } @@ -197,6 +280,7 @@ static int em_nng_sub0_open(nng_socket * s ) { return return_value; } static int em_nng_recv(nng_socket s, void * v, size_t * t, int i ) { + return return_value; } static int em_nng_send( nng_socket s, void* m, int l, int f ) { @@ -337,16 +421,16 @@ static int em_nn_close (int s ) { return 1; } -static int em_nn_setsockopt (int s, int level, int option, const void *optval, size_t optvallen ) { - return 1; -} +//static int em_nn_setsockopt (int s, int level, int option, const void *optval, size_t optvallen ) { + //return 1; +//} static int em_nn_getsockopt (int s, int level, int option, void *optval, size_t *optvallen ) { return 1; } static int em_nn_bind (int s, const char *addr ) { -fprintf( stderr, ">>> ===== emulated bind called ====\n" ); + // fprintf( stderr, ">>> ===== emulated bind called ====\n" ); return 1; } @@ -362,8 +446,43 @@ static int em_nn_send (int s, const void *buf, size_t len, int flags ) { return 1; } -static int em_nn_recv (int s, void *buf, size_t len, int flags ) { - return 1; +static int em_nn_recv (int s, void *m, size_t len, int flags ) { + void* b; + struct em_msg* msg; + static int count = 0; // we'll simulate a message going in by dropping an rmr-ish msg with transaction id only + int trace_size = 0; + static int counter = 0; // if timeout value is set; we return timeout (eagain) every 3 calls + + if( em_timeout > 0 ) { + counter++; + if( counter % 3 == 0 ) { + return EAGAIN; + } + } + + b = (void *) malloc( 2048 ); + if( m != NULL ) { // blindly we assume this is 2k or bigger + memset( m, 0, 2048 ); + msg = (struct em_msg *) m; + if( count % 10 == 9 ) { + //msg->rmr_ver = htonl( MSG_VER ); + msg->rmr_ver = ALT_MSG_VER; // emulate the bug in RMr v1 + } else { + msg->rmr_ver = htonl( MSG_VER ); + } + msg->mtype = htonl( 1 ); + msg->plen = htonl( 220 ); + msg->len0 = htonl( sizeof( struct em_msg ) ); + msg->len1 = htonl( trace_size ); + snprintf( msg->xid, 32, "%015d", count++ ); // simple transaction id so we can test receive specific and ring stuff + snprintf( msg->src, 16, "localhost:4562" ); // set src id (unrealistic) so that rts() can be tested + //fprintf( stderr, " returning message len=%d\n\n", ntohl( msg->plen ) ); + } else { + fprintf( stderr, " message was nil\n\n" ); + } + + //fprintf( stderr, ">>> simulated received message: %s len=%d\n", msg->xid, msg->plen ); + return 2048; } static int em_sendmsg (int s, const struct em_nn_msghdr *msghdr, int flags ) { @@ -374,10 +493,24 @@ static int em_nn_recvmsg (int s, struct nn_msghdr *msghdr, int flags ) { return 1; } +static void em_nn_freemsg( void* ptr ) { + return; +} + +/* + Hacky implementation of set sock opt. We assume value is a pointer to int and ignore size. +*/ +static int em_setsockopt( int sock, int foo, int action, int* value, int size ) { + if( action == NN_RCVTIMEO ) { + em_timeout = *value; + } +} + + // nanomsg #define nn_socket em_nn_socket #define nn_close em_nn_close -#define nn_setsockopt em_nn_setsockopt +//#define nn_setsockopt em_nn_setsockopt #define nn_getsockopt em_nn_getsockopt #define nn_bind em_nn_bind #define nn_connect em_nn_connect @@ -386,6 +519,8 @@ static int em_nn_recvmsg (int s, struct nn_msghdr *msghdr, int flags ) { #define nn_recv em_nn_recv #define nn_sendmsg em_nn_sendmsg #define nn_recvmsg em_nn_recvmsg +#define nn_setsockopt em_setsockopt +#define nn_freemsg em_nn_freemsg #endif