diff options
| -rw-r--r-- | src/rpcb_svc_com.c | 23 |
1 files changed, 22 insertions, 1 deletions
diff --git a/src/rpcb_svc_com.c b/src/rpcb_svc_com.c index ff9ce6b..4ae93f1 100644 --- a/src/rpcb_svc_com.c +++ b/src/rpcb_svc_com.c @@ -1183,12 +1183,33 @@ check_rmtcalls(struct pollfd *pfds, int nfds) return (ncallbacks_found); } +/* + * This is really a helper function defined in libtirpc, + * but unfortunately, it hasn't been exported yet. + */ +static struct netbuf * +__rpc_set_netbuf(struct netbuf *nb, const void *ptr, size_t len) +{ + if (nb->len != len) { + if (nb->len) + mem_free(nb->buf, nb->len); + nb->buf = mem_alloc(len); + if (nb->buf == NULL) + return NULL; + + nb->maxlen = nb->len = len; + } + memcpy(nb->buf, ptr, len); + return nb; +} + static void xprt_set_caller(SVCXPRT *xprt, struct finfo *fi) { + const struct netbuf *caller = fi->caller_addr; u_int32_t *xidp; - *(svc_getrpccaller(xprt)) = *(fi->caller_addr); + __rpc_set_netbuf(svc_getrpccaller(xprt), caller->buf, caller->len); xidp = __rpcb_get_dg_xidp(xprt); *xidp = fi->caller_xid; } |