X-Git-Url: https://gerrit.o-ran-sc.org/r/gitweb?a=blobdiff_plain;f=test%2Ftest_transport_em.c;h=2ca3f08fe7f614dc8dcd24a8a70902b3966d0740;hb=77526eb4f4ef7b2be5fda0c8d719d3c46c75c5c7;hp=92a6b939d193525dd13ee2537d17d6bfce2556b1;hpb=43b7981cee870dcf523b910a3af92ccc53556b2a;p=ric-plt%2Flib%2Frmr.git

diff --git a/test/test_transport_em.c b/test/test_transport_em.c
index 92a6b93..2ca3f08 100644
--- a/test/test_transport_em.c
+++ b/test/test_transport_em.c
@@ -36,14 +36,18 @@ int		tpem_last_len = 0;
 
 int tpem_addr_dup = 0;			// getsockname duplicates last addr if true
 int tpem_conn_state = 0;		// states returned by emulated functions allowing failures to be driven
-int tpem_sock_state = 0;
+int tpem_sock_state = 0;		// if sock state 0, then socket call returns good fd
 int tpem_listen_state = 0;
 int tpem_bind_state = 0;
+int tpem_accept_fd = 5;			// file desc returned by accept
+int tpem_sel_ef = -1;			// select sets this fd's error if >= 0
+int tpem_sel_block = 0;			// set if select call inidcates would block
+int	tpem_send_err = 0;			// set to cause send to return error
 
 // ------------ emulation control -------------------------------------------
 
 /*
-	All test prog to set various things
+	Allow test prog to set various things
 */
 static void tpem_set_conn_state( int s ) {
 	tpem_conn_state = s;
@@ -61,6 +65,22 @@ static void tpem_set_bind_state( int s ) {
 	tpem_bind_state = s;
 }
 
+static void tpem_set_accept_fd( int s ) {
+	tpem_accept_fd = s;
+}
+
+static void tpem_set_selef_fd( int s ) {
+	tpem_sel_ef = s;
+}
+
+static void tpem_set_sel_blk( int s ) {
+	tpem_sel_block = s;
+}
+
+static void tpem_set_send_err( int s ) {
+	tpem_send_err = s;
+}
+
 // ---- emulated functions ---------------------------------------------------
 
 static int tpem_bind( int socket, struct sockaddr* addr, socklen_t alen ) {
@@ -105,7 +125,7 @@ static int tpem_socket( int domain, int type, int protocol ) {
 
 	if( tpem_sock_state == 0 ) {
 		if( ++fd > 10 ) {
-			fd = 1;
+			fd = 3;				// ensure we don't stomp on std* descriptors
 		}
 
 		return fd;
@@ -114,6 +134,43 @@ static int tpem_socket( int domain, int type, int protocol ) {
 	return -1;
 }
 
+static int tpem_accept( int socket, struct sockaddr *restrict address, socklen_t *restrict address_len) {
+	return tpem_accept_fd;
+}
+
+/*
+	Emulate a select. If tpem_sel_ef is set, then the error fd set for the fd is set to true.
+	If sel_woudl_block is set, then the select returns blocking
+*/
+static int tpem_select( int fd_count, fd_set* rf, fd_set* wf, fd_set* ef, void* time ) {
+	fprintf( stderr, "<SYSTEM> select returns %d (1==no-block)\n", tpem_sel_block ? -1 : 1  );
+
+	if( tpem_sel_block ) {
+		return -1;
+	}
+
+	if( tpem_sel_ef >= 0 ) {
+		FD_SET( tpem_sel_ef, ef );
+	} else {
+		FD_ZERO( ef );
+	}
+
+	return 1;
+}
+
+/*
+	If tpem_send_err is set, we return less than count;
+*/
+static int tpem_send( int fd, void* buf, int count, int flags ) {
+	errno = tpem_send_err;
+
+	fprintf( stderr, "<SYSTEM> send on fd=%d for %d bytes ret=%d\n", fd, count, tpem_send_err ? -1 : count );
+	return tpem_send_err ? -1 : count;
+}
+
+
+// ---------------------------------------------------------------------------------------
+
 /*
 	redefine all system calls to reference functions here. There are two defs
 	SI functions should use the capitalised verision so that sliding ff under
@@ -129,19 +186,22 @@ static int tpem_socket( int domain, int type, int protocol ) {
 #define socket tpem_socket
 #define LISTEN tpem_listen
 #define listen tpem_listen
+#define accept tpem_accept
+#define ACCEPT tpem_accept
+#define SEND	tpem_send
+#define SELECT	tpem_select
+#define select	tpem_select
 
 /*
 	these are defined in SI so that we can use the system stack or FFstack
 	they must exist and reference system calls if not defined above.
 */
-#define ACCEPT		accept
 #define CLOSE		close
 #define SHUTDOWN	shutdown
 #define	GETSOCKOPT	getscokopt
 #define SETSOCKOPT	setsockopt
 #define READ		read
 #define WRITE		write
-#define SEND		send
 #define SENDTO		sendto
 #define RECV		recv
 #define RECVFROM	recvfrom