X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=test%2Ftest_nng_em.c;h=88f5966fc0ce633d393e0e5d65e01161930bc678;hb=0cb4b57403307f128b3597a52fa77cbc8524f078;hp=c2101f5490d95f611f4bc5a4cde7c3944526f3f3;hpb=412d53dfa2f9b5b56a448797d0dfec3b0f11f666;p=ric-plt%2Flib%2Frmr.git

diff --git a/test/test_nng_em.c b/test/test_nng_em.c
index c2101f5..88f5966 100644
--- a/test/test_nng_em.c
+++ b/test/test_nng_em.c
@@ -50,12 +50,37 @@ 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;
+static int em_gen_long_hostname = 0;		// if set the emulated hostname generates a longer name (>40 char)
 
 
+// ----------- gethostname emulation ---------------------------------------
+#define gethostname  em_gethostname
+static int em_gethostname( char* buf, size_t len ) {
+	if( len < 1 ) {
+		errno = EINVAL;
+		return 1;
+	}
+
+	if( em_gen_long_hostname ) {
+		snprintf( buf, len, "hostname-which-is-long-a860430b890219-dfw82" );
+	} else {
+		snprintf( buf, len, "em-hostname" );
+	}
+
+	return 0;
+}
+
+static int em_set_long_hostname( int v ) {
+	em_gen_long_hostname = !!v;
+}
+
 // ----------- epoll emulation ---------------------------------------------
 
-// CAUTION: sys/epoll.h must be included before this define and function will properly compile.
+// CAUTION: sys/epoll.h must be included before these define and function will properly compile.
 #define epoll_wait em_wait
+#define epoll_ctl  em_ep_ctl
+#define epoll_create  em_ep_create
+
 /*
 	Every other call returns 1 ready; alternate calls return 0 ready.
 	Mostly for testing the timeout receive call. First call should return
@@ -69,6 +94,14 @@ static int em_wait( int fd, void* events, int n, int to ) {
 	return ready;
 }
 
+int em_ep_ctl( int epfd, int op, int fd, struct epoll_event *event ) {
+	return 0;
+}
+
+int em_ep_create( int size ) {
+	return 0;
+}
+
 
 
 /*
@@ -92,7 +125,7 @@ struct em_msg {
 	we don't add a payload even if setting a v1 type.
 */
 #define ALT_MSG_VER 1	// alternate every so often
-#define MSG_VER 2		// default version to insert
+#define MSG_VER 3		// default version to insert
 struct em_msg {
 	int32_t	mtype;						// message type  ("long" network integer)
 	int32_t	plen;						// payload length
@@ -110,6 +143,9 @@ struct em_msg {
 	int32_t len2;                       // length of data 1 (d1)
 	int32_t len3;                       // length of data 2 (d2)
 	int32_t	sub_id;						// subscription id (-1 invalid)
+
+										// V3 stuff
+	unsigned char srcip[64];				// sender ID for return to sender needs
 };
 
 
@@ -195,8 +231,8 @@ static int em_nng_recvmsg( nng_socket s, nng_msg ** m, int i ) {
 		d1_size = 4;
 	}
 
-	b = (void *) malloc( 2048 );
 	if( m != NULL ) {
+		b = (void *) malloc( 2048 );
 		memset( b, 0, 2048 );
 
 		*m = (nng_msg *) b;
@@ -231,10 +267,14 @@ static int em_nng_recvmsg( nng_socket s, nng_msg ** m, int i ) {
 		}
 		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
+		snprintf( msg->src, 64, "localhost:4562" );		// set src id (unrealistic) so that rts() can be tested
+		snprintf( msg->srcip, 64, "89.2.19.19:4562" );		// set src ip for rts testing
+
+		//fprintf( stderr, ">>> simulated received message: %s %s p=%p len0=%d\n", msg->src, msg->srcip, msg, (int) ntohl( msg->len0 ) );
+	} else {
+		fprintf( stderr, "<WARN> em: simulated receive no msg pointer provided\n" );
 	}
 
-	//fprintf( stderr, ">>> simulated received message: %s\n", msg->xid );
 	return return_value;
 }
 
@@ -452,6 +492,7 @@ static int em_nn_recv (int s, void *m, size_t len, int flags ) {
 	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
+	int d1_size = 0;
 
 	if( em_timeout > 0 ) {
 		counter++;
@@ -460,6 +501,10 @@ static int em_nn_recv (int s, void *m, size_t len, int flags ) {
 		}
 	}
 
+	if( em_mtc_msgs ) {
+		d1_size = 4;
+	}
+
 	b = (void *) malloc( 2048 );
 	if( m != NULL ) {						// blindly we assume this is 2k or bigger
 		memset( m, 0, 2048 );
@@ -474,14 +519,17 @@ static int em_nn_recv (int s, void *m, size_t len, int flags ) {
 		msg->plen = htonl( 220 );
 		msg->len0 = htonl( sizeof( struct em_msg ) );
 		msg->len1 = htonl( trace_size );
+		msg->len2 = htonl( d1_size );
+		msg->len3 = htonl( 0 );
 		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
+		snprintf( msg->src, 64, "localhost:4562" );		// set src id (unrealistic) so that rts() can be tested
+		snprintf( msg->srcip, 64, "89.2.19.19:4562" );		// set src ip for rts testing
 		//fprintf( stderr, "<EM>   returning message len=%d\n\n", ntohl( msg->plen ) );
 	} else {
 		fprintf( stderr, "<EM>   message was nil\n\n" );
 	}
 
-	//fprintf( stderr, ">>> simulated received message: %s len=%d\n", msg->xid, msg->plen );
+	//fprintf( stderr, ">>> simulated received message: %s %s len=%d p=%p\n", msg->src, msg->srcip, ntohl( msg->plen ), m );
 	return 2048;
 }