28fb69c9f04403537359529d4e9c867dd68ab955
[ric-plt/lib/rmr.git] / src / rmr / si / src / sr_si_static.c
1 // vim: ts=4 sw=4 noet :
2 /*
3 ==================================================================================
4         Copyright (c) 2019-2020 Nokia
5         Copyright (c) 2018-2020 AT&T Intellectual Property.
6
7    Licensed under the Apache License, Version 2.0 (the "License");
8    you may not use this file except in compliance with the License.
9    You may obtain a copy of the License at
10
11            http://www.apache.org/licenses/LICENSE-2.0
12
13    Unless required by applicable law or agreed to in writing, software
14    distributed under the License is distributed on an "AS IS" BASIS,
15    WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16    See the License for the specific language governing permissions and
17    limitations under the License.
18 ==================================================================================
19 */
20
21 /*
22         Mnemonic:       sr_si_static.c
23         Abstract:       These are static send/receive primatives which (sadly)
24                                 differ based on the underlying protocol (nng vs SI95).
25                                 Split from rmr_nng.c  for easier wormhole support.
26
27         Author:         E. Scott Daniels
28         Date:           13 February 2019
29 */
30
31 #ifndef _sr_si_static_c
32 #define _sr_si_static_c
33
34 static void dump_n( char *p, char* label, int n ) {
35         int i;
36         int j;
37         int t = 0;
38         int     rows;
39
40
41         if( label ) {
42                 fprintf( stderr, "[DUMP] %s p=%p %d bytes\n", label, p, n );
43         }
44
45         rows = (n/16) + ((n % 16) ? 1 : 0);
46
47         for( j = 0; j < rows; j++ ) {
48                 fprintf( stderr, "[DUMP] %04x: ", j * 16 );
49
50                 for( i = 0; t < n && i < 16; i++, t++ ) {
51                         fprintf( stderr, "%02x ", (unsigned char) *p );
52                         p++;
53                 }
54                 fprintf( stderr, "\n" );
55         }
56 }
57
58 /*
59         backwards compatability.
60 */
61 static void dump_40( char *p, char* label ) {
62         dump_n( p, label, 40 );
63 }
64
65 /*
66         Translates the nng state passed in to one of ours that is suitable to put
67         into the message, and sets errno to something that might be useful.
68         If we don't have a specific RMr state, then we return the default (e.g.
69         receive failed).
70
71         The addition of the connection shut error code to the switch requires
72         that the NNG version at commit e618abf8f3db2a94269a (or after) be
73         used for compiling RMR.
74 */
75 static inline int xlate_si_state( int state, int def_state ) {
76
77         switch( state ) {
78                 case SI_ERR_NONE:
79                         errno = 0;
80                         state = RMR_OK;
81                         break;
82
83                 default:
84                         state = def_state;
85                         errno = EAGAIN;
86                         break;
87
88                 case SI_ERR_QUEUED:
89                 case SI_ERR_TPORT:
90                 case SI_ERR_UPORT:
91                 case SI_ERR_FORK:
92                 case SI_ERR_HANDLE:
93                 case SI_ERR_SESSID:
94                 case SI_ERR_TP:
95                 case SI_ERR_SHUTD:
96                 case SI_ERR_NOFDS:
97                 case SI_ERR_SIGUSR1:
98                 case SI_ERR_SIGUSR2:
99                 case SI_ERR_DISC:
100                 case SI_ERR_TIMEOUT:
101                 case SI_ERR_ORDREL:
102                 case SI_ERR_SIGALRM:
103                 case SI_ERR_NOMEM:
104                 case SI_ERR_ADDR:
105                         errno  = ENOTSUP;
106                         state = def_state;
107                         break;
108         }
109
110         return state;
111 }
112
113 /*
114         Given a message size and a buffer (assumed to be TP_SZFIELD_LEN or larger)
115         this will put in the size such that it is compatable with old versions
116         of RMR (that expect the message size to not be in network byte order)
117         and with new versions that do. See extract function in mt_call_si_static.c
118         for details on what ends up in the buffer.
119 */
120 static inline void insert_mlen( uint32_t len, char* buf ) {
121         uint32_t* blen;                                                 // pointer into buffer where we'll add the len
122
123         blen = (uint32_t *) buf;                                // old systems expect an unconverted integer
124         *blen = len;
125
126         blen++;
127         *blen = htonl( len );                                   // new systems want a converted integer
128
129         memset( &buf[TP_SZFIELD_LEN], 0, 4 );   // clear to prevent future conversion issues
130         buf[TP_SZFIELD_LEN-1] = TP_SZ_MARKER;   // marker to flag this is generated by a new message
131 }
132
133 /*
134         Alloc a new nano zero copy buffer and put into msg. If msg is nil, then we will alloc
135         a new message struct as well. Size is the size of the zc buffer to allocate (not
136         including our header). If size is 0, then the buffer allocated is the size previously
137         allocated (if msg is !nil) or the default size given at initialisation).
138
139         The trlo (trace data lengh override) is used for trace length if >0. If <= 0, then
140         the context value is used.
141
142         NOTE:  while accurate, the nng doc implies that both the msg buffer and data buffer
143                 are zero copy, however ONLY the message is zero copy. We now allocate and use
144                 nng messages.
145 */
146 static rmr_mbuf_t* alloc_zcmsg( uta_ctx_t* ctx, rmr_mbuf_t* msg, int size, int state, int trlo ) {
147         size_t          mlen = -1;                      // size of the transport buffer that we'll allocate
148         uta_mhdr_t*     hdr;                    // convenience pointer
149         int                     tr_len;                 // trace data len (default or override)
150         int*            alen;                   // convenience pointer to set allocated len
151
152         tr_len = trlo > 0 ? trlo : ctx->trace_data_len;
153
154         mlen = sizeof( uta_mhdr_t ) + tr_len + ctx->d1_len + ctx->d2_len;       // start with header and trace/data lengths
155         mlen += (size > 0 ? size  : ctx->max_plen);                                                     // add user requested size or size set during init
156         mlen = sizeof( char ) * (mlen + TP_HDR_LEN);                                            // finally add the transport header len
157
158         if( msg == NULL && (msg = (rmr_mbuf_t *) uta_ring_extract( ctx->zcb_mring )) == NULL ) {
159                 msg = (rmr_mbuf_t *) malloc( sizeof *msg );
160                 if( msg == NULL ) {
161                         rmr_vlog( RMR_VL_CRIT, "rmr_alloc_zc: cannot get memory for message\n" );
162                         return NULL;                                                            // we used to exit -- that seems wrong
163                 }
164                 memset( msg, 0, sizeof( *msg ) );       // tp_buffer will be allocated below
165         } else {                                                                // user message or message from the ring
166                 if( mlen > msg->alloc_len ) {           // current allocation is too small
167                         msg->alloc_len = 0;                             // force tp_buffer realloc below
168                         if( msg->tp_buf ) {
169                                 free( msg->tp_buf );
170                                 msg->tp_buf = NULL;
171                         }
172                 } else {
173                         mlen = msg->alloc_len;                                                  // msg given, allocate the same size as before
174                 }
175         }
176
177         msg->rts_fd = -1;                                       // must force to be invalid; not a received message that can be returned
178
179         if( !msg->alloc_len && (msg->tp_buf = (void *) malloc( mlen )) == NULL ) {
180                 rmr_vlog( RMR_VL_CRIT, "rmr_alloc_zc: cannot get memory for zero copy buffer: %d bytes\n", (int) mlen );
181                 abort( );                                                                                       // toss out a core file for this
182         }
183
184         if( DEBUG ) {
185                 // for speed we don't do this in production; for testing valgrind will complain about uninitialised use if not set
186                 memset( msg->tp_buf, 0, mlen );
187                 memcpy( msg->tp_buf, "@@!!@@!!@@!!@@!!@@!!@@!!@@!!@@!!==", 34 );                // do NOT use a $ in this string!
188         }
189
190         insert_mlen( (uint32_t) mlen, msg->tp_buf );                    // this will likely be overwriten on send to shirnk
191
192         msg->header = ((char *) msg->tp_buf) + TP_HDR_LEN;
193         memset( msg->header, 0, sizeof( uta_mhdr_t ) );                         // ensure no junk in the header area
194         if( (hdr = (uta_mhdr_t *) msg->header) != NULL ) {
195                 hdr->rmr_ver = htonl( RMR_MSG_VER );                                    // set current version
196                 hdr->sub_id = htonl( UNSET_SUBID );
197                 SET_HDR_LEN( hdr );                                                                             // ensure these are converted to net byte order
198                 SET_HDR_TR_LEN( hdr, ctx->trace_data_len );
199                 SET_HDR_D1_LEN( hdr, ctx->d1_len );
200                 //SET_HDR_D2_LEN( hdr, ctx->d2_len );                           // future
201         }
202         msg->len = 0;                                                                                   // length of data in the payload
203         msg->cookie = 0x4942;
204         msg->alloc_len = mlen;                                                                  // length of allocated transport buffer (caller size + rmr header)
205         msg->sub_id = UNSET_SUBID;
206         msg->mtype = UNSET_MSGTYPE;
207         msg->payload = PAYLOAD_ADDR( hdr );                                             // point to payload (past all header junk)
208         msg->xaction = ((uta_mhdr_t *)msg->header)->xid;                // point at transaction id in header area
209         msg->state = state;                                                                             // fill in caller's state (likely the state of the last operation)
210         msg->flags = MFL_ZEROCOPY;                                                              // this is a zerocopy sendable message
211         msg->ring = ctx->zcb_mring;                                                             // original msg_free() api doesn't get context so must dup on eaach :(
212         strncpy( (char *) ((uta_mhdr_t *)msg->header)->src, ctx->my_name, RMR_MAX_SRC );
213         strncpy( (char *) ((uta_mhdr_t *)msg->header)->srcip, ctx->my_ip, RMR_MAX_SRC );
214
215         if( DEBUG > 1 ) rmr_vlog( RMR_VL_DEBUG, "alloc_zcmsg mlen=%ld size=%d mpl=%d flags=%02x\n", (long) mlen, size, ctx->max_plen, msg->flags );
216
217         return msg;
218 }
219
220 /*
221         Allocates only the mbuf and does NOT allocate an underlying transport buffer since
222         transport receive should allocate that on its own.
223 */
224 static rmr_mbuf_t* alloc_mbuf( uta_ctx_t* ctx, int state ) {
225         size_t  mlen;
226         uta_mhdr_t* hdr;                        // convenience pointer
227         rmr_mbuf_t* msg;
228
229         if( (msg = (rmr_mbuf_t *) uta_ring_extract( ctx->zcb_mring )) != NULL ) {
230                 if( msg->tp_buf ) {
231                         free( msg->tp_buf );            // caller doesn't want it -- future put this on an accumulation ring
232                 }
233         } else {
234                 if( (msg = (rmr_mbuf_t *) malloc( sizeof *msg )) == NULL ) {
235                         rmr_vlog( RMR_VL_CRIT, "rmr_alloc_mbuf: cannot get memory for message\n" );
236                         return NULL;                                                    // this used to exit, but that seems wrong
237                 }
238         }
239
240         memset( msg, 0, sizeof( *msg ) );
241
242         msg->cookie = 0x4942;
243         msg->sub_id = UNSET_SUBID;
244         msg->mtype = UNSET_MSGTYPE;
245         msg->tp_buf = NULL;
246         msg->header = NULL;
247         msg->len = -1;                                                                                  // no payload; invalid len
248         msg->alloc_len = 0;
249         msg->payload = NULL;
250         msg->xaction = NULL;
251         msg->state = state;
252         msg->flags = 0;
253         msg->ring = ctx->zcb_mring;                                                     // original msg_free() api doesn't get context so must dup on eaach :(
254
255         return msg;
256 }
257
258 /*
259         This accepts a message with the assumption that only the tp_buf pointer is valid. It
260         sets all of the various header/payload/xaction pointers in the mbuf to the proper
261         spot in the transport layer buffer.  The len in the header is assumed to be the
262         allocated len (a receive buffer that nng created);
263
264         The alen parm is the assumed allocated length; assumed because it's a value likely
265         to have come from si receive and the actual alloc len might be larger, but we
266         can only assume this is the total usable space. Because we are managing a transport
267         header in the first n bytes of the real msg, we must adjust this length down by the
268         size of the tp header (for testing 50 bytes, but this should change to a struct if
269         we adopt this interface).
270
271         This function returns the message with an error state set if it detects that the
272         received message might have been truncated.  Check is done here as the calculation
273         is somewhat based on header version.
274 */
275 static void ref_tpbuf( rmr_mbuf_t* msg, size_t alen )  {
276         uta_mhdr_t* hdr = NULL;                 // current header
277         uta_v1mhdr_t* v1hdr;                    // version 1 header
278         int ver;
279         int     hlen;                                           // header len to use for a truncation check
280
281         msg->header = ((char *) msg->tp_buf) + TP_HDR_LEN;
282
283         v1hdr = (uta_v1mhdr_t *) msg->header;                                   // v1 will always allow us to suss out the version
284
285         if( v1hdr->rmr_ver == 1 ) {                     // bug in verion 1 didn't encode the version in network byte order
286                 ver = 1;
287                 v1hdr->rmr_ver = htonl( 1 );            // save it correctly in case we clone the message
288         } else {
289                 ver = ntohl( v1hdr->rmr_ver );
290         }
291
292         switch( ver ) {
293                 case 1:
294                         msg->len = ntohl( v1hdr->plen );                                                // length sender says is in the payload (received length could be larger)
295                         msg->alloc_len = alen;                                                                  // length of whole tp buffer (including header, trace and data bits)
296                         msg->payload = msg->header + sizeof( uta_v1mhdr_t );    // point past header to payload (single buffer allocation above)
297
298                         msg->xaction = &v1hdr->xid[0];                                                  // point at transaction id in header area
299                         msg->flags |= MFL_ZEROCOPY;                                                             // this is a zerocopy sendable message
300                         msg->mtype = ntohl( v1hdr->mtype );                                             // capture and convert from network order to local order
301                         msg->sub_id = UNSET_SUBID;                                                              // type 1 messages didn't have this
302                         msg->state = RMR_OK;
303                         hlen = sizeof( uta_v1mhdr_t );
304                         break;
305
306                 default:                                                                                                        // current version always lands here
307                         hdr = (uta_mhdr_t *) msg->header;
308                         msg->len = ntohl( hdr->plen );                                                  // length sender says is in the payload (received length could be larger)
309                         msg->alloc_len = alen;                                                                  // length of whole tp buffer (including header, trace and data bits)
310
311                         msg->payload = PAYLOAD_ADDR( hdr );                                             // at user payload
312                         msg->xaction = &hdr->xid[0];                                                    // point at transaction id in header area
313                         msg->flags |= MFL_ZEROCOPY;                                                             // this is a zerocopy sendable message
314                         msg->mtype = ntohl( hdr->mtype );                                               // capture and convert from network order to local order
315                         msg->sub_id = ntohl( hdr->sub_id );
316                         hlen = RMR_HDR_LEN( hdr );                                                              // len to use for truncated check later
317                         break;
318         }
319
320         if( msg->len > (msg->alloc_len - hlen ) ) {
321                 msg->state = RMR_ERR_TRUNC;
322                 msg->len = msg->alloc_len -  hlen;                                                      // adjust len down so user app doesn't overrun
323         } else {
324                 msg->state = RMR_OK;
325         }
326 }
327
328 /*
329         This will clone a message into a new zero copy buffer and return the cloned message.
330 */
331 static inline rmr_mbuf_t* clone_msg( rmr_mbuf_t* old_msg  ) {
332         rmr_mbuf_t* nm;                 // new message buffer
333         size_t  mlen;
334         int state;
335         uta_mhdr_t* hdr;
336         uta_v1mhdr_t* v1hdr;
337
338         nm = (rmr_mbuf_t *) malloc( sizeof *nm );
339         if( nm == NULL ) {
340                 rmr_vlog( RMR_VL_CRIT, "rmr_clone: cannot get memory for message buffer\n" );
341                 exit( 1 );
342         }
343         memset( nm, 0, sizeof( *nm ) );
344
345         mlen = old_msg->alloc_len;                                                                              // length allocated before
346         if( (nm->tp_buf = (void *) malloc( sizeof( char ) * (mlen + TP_HDR_LEN) )) == NULL ) {
347                 rmr_vlog( RMR_VL_CRIT, "rmr_si_clone: cannot get memory for zero copy buffer: %d\n", (int) mlen );
348                 abort();
349         }
350
351         nm->header = ((char *) nm->tp_buf) + TP_HDR_LEN;
352         v1hdr = (uta_v1mhdr_t *) old_msg->header;                               // v1 will work to dig header out of any version
353         switch( ntohl( v1hdr->rmr_ver ) ) {
354                 case 1:
355                         hdr = nm->header;
356                         memcpy( hdr, old_msg->header, sizeof( *v1hdr ) );               // copy complete header
357                         nm->payload = (void *) v1hdr + sizeof( *v1hdr );
358                         break;
359
360                 default:                                                                                        // current message always caught  here
361                         hdr = nm->header;
362                         memcpy( hdr, old_msg->header, RMR_HDR_LEN( old_msg->header ) + RMR_TR_LEN( old_msg->header ) + RMR_D1_LEN( old_msg->header ) + RMR_D2_LEN( old_msg->header ));  // copy complete header, trace and other data
363                         nm->payload = PAYLOAD_ADDR( hdr );                              // at user payload
364                         break;
365         }
366
367         // --- these are all version agnostic -----------------------------------
368         nm->mtype = old_msg->mtype;
369         nm->sub_id = old_msg->sub_id;
370         nm->len = old_msg->len;                                                                 // length of data in the payload
371         nm->alloc_len = mlen;                                                                   // length of allocated payload
372
373         nm->xaction = &hdr->xid[0];                                                             // reference xaction
374         nm->state = old_msg->state;                                                             // fill in caller's state (likely the state of the last operation)
375         nm->flags = old_msg->flags | MFL_ZEROCOPY;                              // this is a zerocopy sendable message
376         memcpy( nm->payload, old_msg->payload, old_msg->len );
377
378         return nm;
379 }
380
381 /*
382         This will clone a message with a change to the trace area in the header such that
383         it will be tr_len passed in. The trace area in the cloned message will be uninitialised.
384         The orignal message will be left unchanged, and a pointer to the new message is returned.
385         It is not possible to realloc buffers and change the data sizes.
386 */
387 static inline rmr_mbuf_t* realloc_msg( rmr_mbuf_t* old_msg, int tr_len  ) {
388         rmr_mbuf_t* nm;                 // new message buffer
389         size_t  mlen;
390         int state;
391         uta_mhdr_t* hdr;
392         uta_v1mhdr_t* v1hdr;
393         int     tr_old_len;                     // tr size in new buffer
394         int*    alen;                   // convenience pointer to set toal xmit len FIX ME!
395         int             tpb_len;                // total transmit buffer len (user space, rmr header and tp header)
396
397         nm = (rmr_mbuf_t *) malloc( sizeof *nm );
398         if( nm == NULL ) {
399                 rmr_vlog( RMR_VL_CRIT, "rmr_clone: cannot get memory for message buffer\n" );
400                 exit( 1 );
401         }
402         memset( nm, 0, sizeof( *nm ) );
403
404         hdr = old_msg->header;
405         tr_old_len = RMR_TR_LEN( hdr );                         // bytes in old header for trace
406
407         mlen = old_msg->alloc_len + (tr_len - tr_old_len);                                                      // new length with trace adjustment
408         if( DEBUG ) rmr_vlog( RMR_VL_DEBUG, "tr_realloc old size=%d new size=%d new tr_len=%d\n", (int) old_msg->alloc_len, (int) mlen, (int) tr_len );
409
410         tpb_len = mlen + TP_HDR_LEN;
411         if( (nm->tp_buf = (void *) malloc( tpb_len)) == NULL ) {
412                 rmr_vlog( RMR_VL_CRIT, "rmr_clone: cannot get memory for zero copy buffer: %d\n", ENOMEM );
413                 exit( 1 );
414         }
415         if( DEBUG ) {
416                 memset( nm->tp_buf, 0, tpb_len );
417                 memcpy( nm->tp_buf, "@@!!@@!!@@!!@@!!@@!!@@!!@@!!@@!!==", 34 );         // DEBUGGING do NOT use $ in this string!!
418         }
419
420         insert_mlen( (uint32_t) tpb_len, nm->tp_buf );                  // this len will likely be reset on send to shrink
421
422         nm->header = ((char *) nm->tp_buf) + TP_HDR_LEN;
423
424         v1hdr = (uta_v1mhdr_t *) old_msg->header;                               // v1 will work to dig header out of any version
425         switch( ntohl( v1hdr->rmr_ver ) ) {
426                 case 1:
427                         memcpy( v1hdr, old_msg->header, sizeof( *v1hdr ) );             // copy complete header
428                         nm->payload = (void *) v1hdr + sizeof( *v1hdr );
429                         break;
430
431                 default:                                                                                        // current message version always caught  here
432                         hdr = nm->header;
433                         memcpy( hdr, old_msg->header, sizeof( uta_mhdr_t ) );           // ONLY copy the header portion; trace and data offsets might have changed
434                         SET_HDR_TR_LEN( hdr, tr_len );                                                          // must adjust trace len in new message before copy
435
436                         if( RMR_D1_LEN( hdr )  ) {
437                                 memcpy( DATA1_ADDR( hdr ), DATA1_ADDR( old_msg->header), RMR_D1_LEN( hdr ) );           // copy data1 and data2 if necessary
438                         }
439                         if( RMR_D2_LEN( hdr )  ) {
440                                 memcpy( DATA2_ADDR( hdr ), DATA2_ADDR( old_msg->header), RMR_D2_LEN( hdr ) );
441                         }
442
443                         nm->payload = PAYLOAD_ADDR( hdr );                                                                      // directly at the payload
444                         break;
445         }
446
447         // --- these are all version agnostic -----------------------------------
448         nm->mtype = old_msg->mtype;
449         nm->sub_id = old_msg->sub_id;
450         nm->len = old_msg->len;                                                                 // length of data in the payload
451         nm->alloc_len = mlen;                                                                   // length of allocated payload
452
453         nm->xaction = &hdr->xid[0];                                                             // reference xaction
454         nm->state = old_msg->state;                                                             // fill in caller's state (likely the state of the last operation)
455         nm->flags = old_msg->flags | MFL_ZEROCOPY;                              // this is a zerocopy sendable message
456         memcpy( nm->payload, old_msg->payload, old_msg->len );
457
458         return nm;
459 }
460
461 /*
462         Realloc the message such that the payload is at least payload_len bytes.
463         The clone and copy options affect what portion of the original payload is copied to
464         the reallocated message, and whether or not the original payload is lost after the
465         reallocation process has finished.
466
467                 copy == true
468                 The entire payload from the original message will be coppied to the reallocated
469                 payload.
470
471                 copy == false
472                 Only the header (preserving return to sender information, message type, etc)
473                 is preserved after reallocation; the payload used lengrh is set to 0 and the
474                 payload is NOT initialised/cleared.
475
476                 clone == true
477                 The orignal message is preserved and a completely new message buffer and payload
478                 are allocated (even if the size given is the same). A pointer to the new message
479                 buffer is returned and it is the user application's responsibility to manage the
480                 old buffer (e.g. free when not needed).
481
482                 clone == false
483                 The old payload will be lost after reallocation. The message buffer pointer which
484                 is returned will likely reference the same structure (don't depend on that).
485
486
487         CAUTION:
488         If the message is not a message which was received, the mtype, sub-id, length values in the
489         RMR header in the allocated transport buffer will NOT be accurate and will cause the resulting
490         mbuffer information for mtype and subid to be reset even when copy is true. To avoid silently
491         resetting information in the mbuffer, this funciton will reset the mbuf values from the current
492         settings and NOT from the copied RMR header in transport buffer.
493 */
494 static inline rmr_mbuf_t* realloc_payload( rmr_mbuf_t* old_msg, int payload_len, int copy, int clone ) {
495         rmr_mbuf_t* nm = NULL;  // new message buffer when cloning
496         size_t  mlen;
497         uta_mhdr_t* omhdr;              // old message header
498         int             tr_old_len;             // tr size in new buffer
499         int             old_psize = 0;  // size of payload in the message passed in (alloc size - tp header and rmr header lengths)
500         int             hdr_len = 0;    // length of RMR and transport headers in old msg
501         void*   old_tp_buf;             // pointer to the old tp buffer
502         int             free_tp = 1;    // free the transport buffer (old) when done (when not cloning)
503         int             old_mt;                 // msg type and sub-id from the message passed in
504         int             old_sid;
505         int             old_len;
506         int             old_rfd;                // rts file descriptor from old message
507
508         if( old_msg == NULL || payload_len <= 0 ) {
509                 errno = EINVAL;
510                 return NULL;
511         }
512
513         old_mt = old_msg->mtype;                        // preserve mbuf info
514         old_sid = old_msg->sub_id;
515         old_len = old_msg->len;
516         old_rfd = old_msg->rts_fd;
517
518         old_psize = old_msg->alloc_len - (RMR_HDR_LEN( old_msg->header ) + TP_HDR_LEN);         // user payload size in orig message
519
520         if( !clone  && payload_len <= old_psize ) {                                                                             // not cloning and old is large enough; nothing to do
521                 if( DEBUG ) rmr_vlog( RMR_VL_DEBUG, "rmr_realloc_payload: old msg payload larger than requested: cur=%d need=%d\n", old_psize, payload_len );
522                 return old_msg;
523         }
524
525         hdr_len = RMR_HDR_LEN( old_msg->header ) + TP_HDR_LEN;                          // with SI we manage the transport header; must include in len
526         old_tp_buf = old_msg->tp_buf;
527
528         if( clone ) {
529                 if( DEBUG ) rmr_vlog( RMR_VL_DEBUG, "rmr_realloc_payload: cloning message\n" );
530                 free_tp = 0;
531
532                 nm = (rmr_mbuf_t *) malloc( sizeof( *nm ) );
533                 if( nm == NULL ) {
534                         rmr_vlog( RMR_VL_CRIT, "rmr_realloc_payload: cannot get memory for message buffer. bytes requested: %d\n", (int) sizeof(*nm) );
535                         return NULL;
536                 }
537                 memset( nm, 0, sizeof( *nm ) );
538                 nm->rts_fd = old_rfd;                           // this is managed only in the mbuf; dup now
539         } else {
540                 nm = old_msg;
541         }
542
543         omhdr = old_msg->header;
544         mlen = hdr_len + (payload_len > old_psize ? payload_len : old_psize);           // must have larger in case copy is true
545
546         if( DEBUG ) rmr_vlog( RMR_VL_DEBUG, "reallocate for payload increase. new message size: %d\n", (int) mlen );
547         if( (nm->tp_buf = (char *) malloc( sizeof( char ) * mlen )) == NULL ) {
548                 rmr_vlog( RMR_VL_CRIT, "rmr_realloc_payload: cannot get memory for zero copy buffer. bytes requested: %d\n", (int) mlen );
549                 free( nm );
550                 return NULL;
551         }
552
553         nm->header = ((char *) nm->tp_buf) + TP_HDR_LEN;                        // point at the new header and copy from old
554         SET_HDR_LEN( nm->header );
555
556         if( copy ) {                                                                                                                            // if we need to copy the old payload too
557                 memcpy( nm->header, omhdr, sizeof( char ) * (old_psize + RMR_HDR_LEN( omhdr )) );
558                 if( DEBUG ) rmr_vlog( RMR_VL_DEBUG, "rmr_realloc_payload: copy payload into new message: %d bytes\n", old_psize );
559         } else {                                                                                                                                        // just need to copy header
560                 memcpy( nm->header, omhdr, sizeof( char ) * RMR_HDR_LEN( omhdr ) );
561                 if( DEBUG ) rmr_vlog( RMR_VL_DEBUG, "rmr_realloc_payload: copy only header into new message: %d bytes\n", RMR_HDR_LEN( nm->header ) );
562         }
563
564         ref_tpbuf( nm, mlen );                  // set payload and other pointers in the message to the new tp buffer
565
566         if( !copy ) {
567                 nm->mtype = -1;                                         // didn't copy payload, so mtype, sub-id, and rts fd are invalid
568                 nm->sub_id = -1;
569                 nm->len = 0;                                            // and len is 0
570         } else {
571                 nm->len = old_len;                                      // we must force these to avoid losing info if msg wasn't a received message
572                 nm->mtype = old_mt;
573                 nm->sub_id = old_sid;
574         }
575
576         if( free_tp ) {
577                 free( old_tp_buf );                             // we did not clone, so free b/c no references
578         }
579
580         return nm;
581 }
582
583 /*
584         For SI95 based transport all receives are driven through the threaded
585         ring and thus this function should NOT be called. If it is we will panic
586         and abort straight away.
587 */
588 static rmr_mbuf_t* rcv_msg( uta_ctx_t* ctx, rmr_mbuf_t* old_msg ) {
589
590 fprintf( stderr, "\n\n>>> rcv_msg: bad things just happened!\n\n>>>>>> abort!  rcv_msg called and it shouldn't be\n" );
591 exit( 1 );
592
593         return NULL;
594 }
595
596 /*
597         This does the hard work of actually sending the message to the given socket. On success,
598         a new message struct is returned. On error, the original msg is returned with the state
599         set to a reasonable value. If the message being sent as MFL_NOALLOC set, then a new
600         buffer will not be allocated and returned (mostly for call() interal processing since
601         the return message from call() is a received buffer, not a new one).
602
603         Called by rmr_send_msg() and rmr_rts_msg(), etc. and thus we assume that all pointer
604         validation has been done prior.
605
606         When msg->state is not ok, this function must set tp_state in the message as some API
607         fucntions return the message directly and do not propigate errno into the message.
608 */
609 static rmr_mbuf_t* send_msg( uta_ctx_t* ctx, rmr_mbuf_t* msg, int nn_sock, int retries ) {
610         int state;
611         uta_mhdr_t*     hdr;
612         int spin_retries = 1000;                                // if eagain/timeout we'll spin, at max, this many times before giving up the CPU
613         int     tr_len;                                                         // trace len in sending message so we alloc new message with same trace sizes
614         int tot_len;                                                    // total send length (hdr + user data + tp header)
615
616         // future: ensure that application did not overrun the XID buffer; last byte must be 0
617
618         hdr = (uta_mhdr_t *) msg->header;
619         hdr->mtype = htonl( msg->mtype );                                                               // stash type/len/sub_id in network byte order for transport
620         hdr->sub_id = htonl( msg->sub_id );
621         hdr->plen = htonl( msg->len );
622         tr_len = RMR_TR_LEN( hdr );                                                                             // snarf trace len before sending as hdr is invalid after send
623
624         if( msg->flags & MFL_ADDSRC ) {                                                                 // buffer was allocated as a receive buffer; must add our source
625                 strncpy( (char *) ((uta_mhdr_t *)msg->header)->src, ctx->my_name, RMR_MAX_SRC );                                        // must overlay the source to be ours
626                 strncpy( (char *) ((uta_mhdr_t *)msg->header)->srcip, ctx->my_ip, RMR_MAX_SRC );
627         }
628
629         if( retries == 0 ) {
630                 spin_retries = 100;
631                 retries++;
632         }
633
634         errno = 0;
635         msg->state = RMR_OK;
636         do {
637                 tot_len = msg->len + PAYLOAD_OFFSET( hdr ) + TP_HDR_LEN;                        // we only send what was used + header lengths
638                 if( tot_len > msg->alloc_len ) {
639                         tot_len = msg->alloc_len;                                                                       // likely bad length from user :(
640                 }
641                 insert_mlen( tot_len, msg->tp_buf );    // shrink to fit
642
643                 if( DEBUG > 1 ) rmr_vlog( RMR_VL_DEBUG, "send_msg: ending %d (%x) bytes  usr_len=%d alloc=%d retries=%d\n", tot_len, tot_len, msg->len, msg->alloc_len, retries );
644                 if( DEBUG > 2 ) dump_40( msg->tp_buf, "sending" );
645
646                 if( (state = SIsendt( ctx->si_ctx, nn_sock, msg->tp_buf, tot_len )) != SI_OK ) {
647                         if( DEBUG > 1 ) rmr_vlog( RMR_VL_DEBUG, "send_msg:  error!! sent state=%d\n", state );
648                         msg->state = state;
649                         if( retries > 0 && state == SI_ERR_BLOCKED ) {
650                                 if( --spin_retries <= 0 ) {                             // don't give up the processor if we don't have to
651                                         retries--;
652                                         if( retries > 0 ) {                                     // only if we'll loop through again
653                                                 usleep( 1 );                                    // sigh, give up the cpu and hope it's just 1 miscrosec
654                                         }
655                                         spin_retries = 1000;
656                                 }
657                         } else {
658                                 state = 0;                      // don't loop
659                         }
660                 } else {
661                         if( DEBUG > 2 ) rmr_vlog( RMR_VL_DEBUG, "sent OK state=%d\n", state );
662                         state = 0;
663                         msg->state = RMR_OK;
664                         hdr = NULL;
665                 }
666         } while( state && retries > 0 );
667
668         if( msg->state == RMR_OK ) {                                                                    // successful send
669                 if( !(msg->flags & MFL_NOALLOC) ) {                                                     // allocate another sendable zc buffer unless told otherwise
670                         return alloc_zcmsg( ctx, msg, 0, RMR_OK, tr_len );              // preallocate a zero-copy buffer and return msg
671                 } else {
672                         rmr_free_msg( msg );                                            // not wanting a meessage back, trash this one
673                         return NULL;
674                 }
675         } else {                                                                                        // send failed -- return original message
676                 if( msg->state == 98 ) {                // FIX ME: this is just broken, but needs SI changes to work correctly for us
677                         errno = EAGAIN;
678                         msg->state = RMR_ERR_RETRY;                                     // errno will have nano reason
679                 } else {
680                         msg->state = RMR_ERR_SENDFAILED;
681                 }
682
683                 if( DEBUG ) rmr_vlog( RMR_VL_DEBUG, "send failed: %d %s\n", (int) msg->state, strerror( msg->state ) );
684         }
685
686         return msg;
687 }
688
689 /*
690         send message with maximum timeout.
691         Accept a message and send it to an endpoint based on message type.
692         If NNG reports that the send attempt timed out, or should be retried,
693         RMr will retry for approximately max_to microseconds; rounded to the next
694         higher value of 10.
695
696         Allocates a new message buffer for the next send. If a message type has
697         more than one group of endpoints defined, then the message will be sent
698         in round robin fashion to one endpoint in each group.
699
700         An endpoint will be looked up in the route table using the message type and
701         the subscription id. If the subscription id is "UNSET_SUBID", then only the
702         message type is used.  If the initial lookup, with a subid, fails, then a
703         second lookup using just the mtype is tried.
704
705         When msg->state is not OK, this function must set tp_state in the message as
706         some API fucntions return the message directly and do not propigate errno into
707         the message.
708
709         CAUTION: this is a non-blocking send.  If the message cannot be sent, then
710                 it will return with an error and errno set to eagain. If the send is
711                 a limited fanout, then the returned status is the status of the last
712                 send attempt.
713
714 */
715 static  rmr_mbuf_t* mtosend_msg( void* vctx, rmr_mbuf_t* msg, int max_to ) {
716         endpoint_t*     ep;                                     // end point that we're attempting to send to
717         rtable_ent_t*   rte;                    // the route table entry which matches the message key
718         int     nn_sock;                                        // endpoint socket (fd in si case) for send
719         uta_ctx_t*      ctx;
720         int                     group;                          // selected group to get socket for
721         int                     send_again;                     // true if the message must be sent again
722         rmr_mbuf_t*     clone_m;                        // cloned message for an nth send
723         int                     sock_ok;                        // got a valid socket from round robin select
724         char*           d1;
725         int                     ok_sends = 0;           // track number of ok sends
726
727         if( (ctx = (uta_ctx_t *) vctx) == NULL || msg == NULL ) {               // bad stuff, bail fast
728                 errno = EINVAL;                                                                                         // if msg is null, this is their clue
729                 if( msg != NULL ) {
730                         msg->state = RMR_ERR_BADARG;
731                         errno = EINVAL;                                                                                 // must ensure it's not eagain
732                         msg->tp_state = errno;
733                 }
734                 return msg;
735         }
736
737         errno = 0;                                                                                                      // clear; nano might set, but ensure it's not left over if it doesn't
738         if( msg->header == NULL ) {
739                 fprintf( stderr, "rmr_mtosend_msg: ERROR: message had no header\n" );
740                 msg->state = RMR_ERR_NOHDR;
741                 errno = EBADMSG;                                                                                        // must ensure it's not eagain
742                 msg->tp_state = errno;
743                 return msg;
744         }
745
746         if( max_to < 0 ) {
747                 max_to = ctx->send_retries;             // convert to retries
748         }
749
750         if( (rte = uta_get_rte( ctx->rtable, msg->sub_id, msg->mtype, TRUE )) == NULL ) {               // find the entry which matches subid/type allow fallback to type only key
751                 rmr_vlog( RMR_VL_WARN, "no route table entry for mtype=%d sub_id=%d\n", msg->mtype, msg->sub_id );
752                 msg->state = RMR_ERR_NOENDPT;
753                 errno = ENXIO;                                                                          // must ensure it's not eagain
754                 msg->tp_state = errno;
755                 return msg;                                                                                     // caller can resend (maybe) or free
756         }
757
758         send_again = 1;                                                                                 // force loop entry
759         group = 0;                                                                                              // always start with group 0
760         while( send_again ) {
761                 if( rte->nrrgroups > 0 ) {                                                      // this is a round robin entry if groups are listed
762                         sock_ok = uta_epsock_rr( ctx, rte, group, &send_again, &nn_sock, &ep );         // select endpt from rr group and set again if more groups
763                 } else {
764                         sock_ok = epsock_meid( ctx, ctx->rtable, msg, &nn_sock, &ep );
765                         send_again = 0;
766                 }
767
768                 if( DEBUG ) rmr_vlog( RMR_VL_DEBUG, "mtosend_msg: flgs=0x%04x type=%d again=%d group=%d len=%d sock_ok=%d\n",
769                                 msg->flags, msg->mtype, send_again, group, msg->len, sock_ok );
770
771                 group++;
772
773                 if( sock_ok ) {                                                                                                 // with an rte we _should_ always have a socket, but don't bet on it
774                         if( send_again ) {
775                                 clone_m = clone_msg( msg );                                                             // must make a copy as once we send this message is not available
776                                 if( clone_m == NULL ) {
777                                         msg->state = RMR_ERR_SENDFAILED;
778                                         errno = ENOMEM;
779                                         msg->tp_state = errno;
780                                         rmr_vlog( RMR_VL_WARN, "unable to clone message for multiple rr-group send\n" );
781                                         return msg;
782                                 }
783
784                                 if( DEBUG ) rmr_vlog( RMR_VL_DEBUG, "msg cloned: type=%d len=%d\n", msg->mtype, msg->len );
785                                 msg->flags |= MFL_NOALLOC;                                                              // keep send from allocating a new message; we have a clone to use
786                                 msg = send_msg( ctx, msg, nn_sock, max_to );                    // do the hard work, msg should be nil on success
787
788                                 if( msg != NULL ) {                                                                             // returned message indicates send error of some sort
789                                         rmr_free_msg( msg );                                                            // must ditchone; pick msg so we don't have to unfiddle flags
790                                         msg = clone_m;
791                                 } else {
792                                         ok_sends++;
793                                         msg = clone_m;                                                                          // clone will be the next to send
794                                         msg->state = RMR_OK;
795                                 }
796                         } else {
797                                 msg = send_msg( ctx, msg, nn_sock, max_to );                    // send the last, and allocate a new buffer; drops the clone if it was
798                                 if( DEBUG ) {
799                                         if( msg == NULL ) {
800                                                 rmr_vlog( RMR_VL_DEBUG, "mtosend_msg:  send returned nil message!\n" );
801                                         }
802                                 }
803                         }
804
805                         if( ep != NULL && msg != NULL ) {
806                                 switch( msg->state ) {
807                                         case RMR_OK:
808                                                 ep->scounts[EPSC_GOOD]++;
809                                                 break;
810         
811                                         case RMR_ERR_RETRY:
812                                                 ep->scounts[EPSC_TRANS]++;
813                                                 break;
814
815                                         default:
816                                                 ep->scounts[EPSC_FAIL]++;
817                                                 uta_ep_failed( ep );                                                            // sending to ep failed; set up to reconnect
818                                                 break;
819                                 }
820                         }
821                 } else {
822                         if( DEBUG ) rmr_vlog( RMR_VL_DEBUG, "invalid socket for rte, setting no endpoint err: mtype=%d sub_id=%d\n", msg->mtype, msg->sub_id );
823                         msg->state = RMR_ERR_NOENDPT;
824                         errno = ENXIO;
825                 }
826         }
827
828         if( msg ) {                                                     // call functions don't get a buffer back, so a nil check is required
829                 msg->flags &= ~MFL_NOALLOC;             // must return with this flag off
830                 if( ok_sends ) {                                // multiple rr-groups and one was successful; report ok
831                         msg->state = RMR_OK;
832                 }
833
834                 if( DEBUG ) rmr_vlog( RMR_VL_DEBUG, "final send stats: ok=%d group=%d state=%d\n", ok_sends, group, msg->state );
835
836                 msg->tp_state = errno;
837         }
838
839         return msg;                                                                     // last message caries the status of last/only send attempt
840 }
841
842
843 /*
844         A generic wrapper to the real send to keep wormhole stuff agnostic.
845         We assume the wormhole function vetted the buffer so we don't have to.
846 */
847 static rmr_mbuf_t* send2ep( uta_ctx_t* ctx, endpoint_t* ep, rmr_mbuf_t* msg ) {
848         return send_msg( ctx, msg, ep->nn_sock, -1 );
849 }
850
851 #endif