--- /dev/null
+/*
+ *
+ * Copyright 2019 AT&T Intellectual Property
+ * Copyright 2019 Nokia
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ *
+ */
+
+int epoll_to = 1;// global default epoll timout -- 1ms
+char* eparm; // generic env pointer
+
+struct rmr_context {
+ void* mrc;
+ int rcv_fd; // pollable fd
+ struct epoll_event epe; // event definition for event to listen to
+ struct epoll_event events[10]; // wait on 10 possible events
+ int ep_fd;
+ int max_payload_size; // ++SCOTT
+ rmr_mbuf_t* sbuf; // send buffer
+ rmr_mbuf_t* rbuf; // received buffer
+};
+
+
+struct rmr_context * rmr_init_wrapper(char* lport){
+
+ struct rmr_context *rmr_c = malloc(sizeof (struct rmr_context));
+ fprintf( stderr, "[INFO] glistening for replies on %s\n", lport );
+
+
+ rmr_c->mrc = rmr_init(lport, RMR_MAX_RCV_BYTES, RMRFL_NONE ); // setup RMr and get a context (rmr_c.mrc)
+ rmr_c->ep_fd=-1;
+
+ //polling related initializations
+ rmr_c->rcv_fd = rmr_get_rcvfd( rmr_c->mrc );// get the fd to poll for messages received
+ if( rmr_c->rcv_fd < 0 ) {
+ fprintf( stderr, "[FAIL] unable to set up polling fd\n" );
+ exit( 1 );
+ }
+
+ if( (rmr_c->ep_fd = epoll_create1( 0 )) < 0 ) {
+ fprintf( stderr, "[FAIL] unable to create epoll fd: %d\n", errno );
+ exit( 1 );
+ }
+ rmr_c->epe.events = EPOLLIN;
+ rmr_c->epe.data.fd = rmr_c->rcv_fd;
+
+ if( epoll_ctl( rmr_c->ep_fd, EPOLL_CTL_ADD, rmr_c->rcv_fd, &rmr_c->epe ) != 0 ) {
+ fprintf( stderr, "[FAIL] epoll_ctl status not 0 : %s\n", strerror( errno ) );
+ exit( 1 );
+ }
+ //end of polling related initializations
+
+ //++SCOTT------ next lines until !!SCOTT
+ rmr_c->max_payload_size = 240; //default
+ if( (eparm = getenv( "DEMO_MAX_PAYLOAD_BYTES" )) != NULL ) {
+ rmr_c->max_payload_size = atoi(eparm); // override with env
+ }
+ //!!SCOTT
+
+ //~~SCOTT next line
+ rmr_c->sbuf = rmr_alloc_msg( rmr_c->mrc, rmr_c->max_payload_size ); // allocate largest payload to send
+ rmr_c->rbuf = NULL;
+
+ return rmr_c;
+}
+
+void rmr_send_wrapper (struct rmr_context *rmr_c, int mtype, char* message) {
+ //--SCOTT int max_payload_size = 240; //default
+ //--SCOTT if( (eparm = getenv( "DEMO_MAX_PAYLOAD_BYTES" )) != NULL ) {
+ //--SCOTT max_payload_size = atoi(eparm);
+ //--SCOTT }
+
+ //~~SCOTT next line
+ snprintf( rmr_c->sbuf->payload, rmr_c->max_payload_size, "%s", message); // simple send message -- replace with real content
+
+ rmr_c->sbuf->mtype = mtype; // fill in message meta data
+ rmr_c->sbuf->len = strlen( rmr_c->sbuf->payload ) + 1; // actual length of payload (count the nil end of string)
+ rmr_c->sbuf->state = 0;
+
+ //retry send for a few times before giving up
+ long natter = 0; // natter on for errors only once in a while
+ if( (rmr_c->sbuf = rmr_send_msg( rmr_c->mrc, rmr_c->sbuf )) != NULL ) {// unlikely, but might get a null pointer back if NNG really is buggered
+ if( rmr_c->sbuf->state != RMR_OK ) {
+ if( errno == EAGAIN ) {
+ while( rmr_c->sbuf->state != RMR_OK && errno == EAGAIN ) {// NNG likes to refuse sends, just keep trying on eagain
+ rmr_send_msg( rmr_c->mrc, rmr_c->sbuf );
+ }
+ } else {// most likely connection refused, don't natter on
+ if( time( NULL ) > natter ) {
+ fprintf( stderr, "[WARN] send failed, pausing (%s)\n", strerror( errno ) );
+ natter = time( NULL ) + 2;
+ }
+ sleep( 1 );
+ }
+ }
+ } else {
+ if( time( NULL ) > natter ) {
+ fprintf( stderr, "[WARN] send failed, pausing (%s)\n", strerror( errno ) );
+ natter = time( NULL ) + 2;
+ }
+ sleep( 1 );
+ }
+
+}
+
+int rmr_poll_for_message(struct rmr_context *rmr_c){
+ int nready;
+ int i;
+ int return_type =0;
+ //start polling the channel to read the acknowledgment
+ //~~SCOTT next line
+
+ nready = epoll_wait( rmr_c->ep_fd, rmr_c->events, 10, epoll_to ); // wait up to epoll_to ms for a response
+ for( i = 0; i < nready && i < 10; i++ ) { // loop through to find what is ready
+ if( rmr_c->events[i].data.fd == rmr_c->rcv_fd ) { // RMr has something
+ errno = 0;
+ rmr_c->rbuf = rmr_rcv_msg( rmr_c->mrc, rmr_c->rbuf ); // something ready; this should not block
+ if( rmr_c->rbuf ) {
+ // fprintf( stderr, "<TEST> acknowledgment received:%s\n",rmr_c->rbuf->payload);
+ return_type = 1; //the message has been acknowledged
+ }
+ }
+ }
+ return return_type;
+}
+
+void rmr_close_wrapper (struct rmr_context *rmr_c){
+
+ rmr_close( rmr_c->mrc );
+ free(rmr_c);
+}
Code Review / sim / e2-interface.git / blobdiff