diff options
Diffstat (limited to 'sunrpc/rpc_thread.c')
-rw-r--r-- | sunrpc/rpc_thread.c | 85 |
1 files changed, 79 insertions, 6 deletions
diff --git a/sunrpc/rpc_thread.c b/sunrpc/rpc_thread.c index 0b004c403f..1fd1c143ea 100644 --- a/sunrpc/rpc_thread.c +++ b/sunrpc/rpc_thread.c @@ -14,6 +14,10 @@ static struct rpc_thread_variables __libc_tsd_RPC_VARS_mem; static struct rpc_thread_variables *__libc_tsd_RPC_VARS_data = &__libc_tsd_RPC_VARS_mem; + +/* This is the variable used for the first thread. */ +static struct rpc_thread_variables rpc_default; + /* * Task-variable destructor */ @@ -22,7 +26,7 @@ __rpc_thread_destroy (void) { struct rpc_thread_variables *tvp = __rpc_thread_variables(); - if (tvp != NULL) { + if (tvp != NULL && tvp != &rpc_default) { __rpc_thread_svc_cleanup (); __rpc_thread_clnt_cleanup (); __rpc_thread_key_cleanup (); @@ -37,19 +41,88 @@ __rpc_thread_destroy (void) } +/* + * Initialize RPC multi-threaded operation + */ +static void +rpc_thread_multi (void) +{ + __libc_tsd_set (RPC_VARS, &rpc_default); +} + + struct rpc_thread_variables * __rpc_thread_variables (void) { + __libc_once_define (static, once); struct rpc_thread_variables *tvp; tvp = __libc_tsd_get (RPC_VARS); if (tvp == NULL) { - tvp = calloc (1, sizeof *tvp); - if (tvp != NULL) - __libc_tsd_set (RPC_VARS, tvp); - else - tvp = __libc_tsd_RPC_VARS_data; + __libc_once (once, rpc_thread_multi); + tvp = __libc_tsd_get (RPC_VARS); + if (tvp == NULL) { + tvp = calloc (1, sizeof *tvp); + if (tvp != NULL) + __libc_tsd_set (RPC_VARS, tvp); + else + tvp = __libc_tsd_RPC_VARS_data; + } } return tvp; } + + +/* Global variables If we're single-threaded, or if this is the first + thread using the variable, use the existing global variable. This + provides backwards compatability for existing applications which + dynamically link against this code. */ +#undef svc_fdset +#undef rpc_createerr +#undef svc_pollfd +#undef svc_max_pollfd + +fd_set * +__rpc_thread_svc_fdset (void) +{ + struct rpc_thread_variables *tvp; + + tvp = __rpc_thread_variables (); + if (tvp == &rpc_default) + return &svc_fdset; + return &tvp->svc_fdset_s; +} + +struct rpc_createerr * +__rpc_thread_createerr (void) +{ + struct rpc_thread_variables *tvp; + + tvp = __rpc_thread_variables (); + if (tvp == &rpc_default) + return &rpc_createerr; + return &tvp->rpc_createerr_s; +} + +struct pollfd ** +__rpc_thread_svc_pollfd (void) +{ + struct rpc_thread_variables *tvp; + + tvp = __rpc_thread_variables (); + if (tvp == &rpc_default) + return &svc_pollfd; + return &tvp->svc_pollfd_s; +} + +int * +__rpc_thread_svc_max_pollfd (void) +{ + struct rpc_thread_variables *tvp; + + tvp = __rpc_thread_variables (); + if (tvp == &rpc_default) + return &svc_max_pollfd; + return &tvp->svc_max_pollfd_s; +} #endif /* _RPC_THREAD_SAFE_ */ |