+++ /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);
-}