2003-02-21 Roland McGrath <roland@redhat.com>
[kopensolaris-gnu/glibc.git] / sunrpc / rpc_thread.c
1 #include <stdio.h>
2 #include <bits/libc-lock.h>
3 #include <rpc/rpc.h>
4 #include <assert.h>
5
6 #include <bits/libc-lock.h>
7 #include <bits/libc-tsd.h>
8
9 #ifdef _RPC_THREAD_SAFE_
10
11 /* Variable used in non-threaded applications or for the first thread.  */
12 static struct rpc_thread_variables __libc_tsd_RPC_VARS_mem;
13 __libc_tsd_define (, RPC_VARS)
14
15 /*
16  * Task-variable destructor
17  */
18 void __attribute__ ((section ("__libc_thread_freeres_fn")))
19 __rpc_thread_destroy (void)
20 {
21         struct rpc_thread_variables *tvp = __libc_tsd_get (RPC_VARS);
22
23         if (tvp != NULL && tvp != &__libc_tsd_RPC_VARS_mem) {
24                 __rpc_thread_svc_cleanup ();
25                 __rpc_thread_clnt_cleanup ();
26                 __rpc_thread_key_cleanup ();
27                 free (tvp->clnt_perr_buf_s);
28                 free (tvp->clntraw_private_s);
29                 free (tvp->svcraw_private_s);
30                 free (tvp->authdes_cache_s);
31                 free (tvp->authdes_lru_s);
32                 free (tvp);
33         }
34 }
35 text_set_element (__libc_thread_subfreeres, __rpc_thread_destroy);
36
37
38 /*
39  * Initialize RPC multi-threaded operation
40  */
41 static void
42 rpc_thread_multi (void)
43 {
44   __libc_tsd_set (RPC_VARS, &__libc_tsd_RPC_VARS_mem);
45 }
46
47
48 struct rpc_thread_variables *
49 __rpc_thread_variables (void)
50 {
51         __libc_once_define (static, once);
52         struct rpc_thread_variables *tvp;
53
54         tvp = __libc_tsd_get (RPC_VARS);
55         if (tvp == NULL) {
56                 __libc_once (once, rpc_thread_multi);
57                 tvp = __libc_tsd_get (RPC_VARS);
58                 if (tvp == NULL) {
59                         tvp = calloc (1, sizeof *tvp);
60                         if (tvp != NULL)
61                                 __libc_tsd_set (RPC_VARS, tvp);
62                         else
63                                 tvp = __libc_tsd_get (RPC_VARS);
64                 }
65         }
66         return tvp;
67 }
68
69
70 /* Global variables If we're single-threaded, or if this is the first
71    thread using the variable, use the existing global variable.  This
72    provides backwards compatability for existing applications which
73    dynamically link against this code.  */
74 #undef svc_fdset
75 #undef rpc_createerr
76 #undef svc_pollfd
77 #undef svc_max_pollfd
78
79 fd_set *
80 __rpc_thread_svc_fdset (void)
81 {
82         struct rpc_thread_variables *tvp;
83
84         tvp = __rpc_thread_variables ();
85         if (tvp == &__libc_tsd_RPC_VARS_mem)
86                 return &svc_fdset;
87         return &tvp->svc_fdset_s;
88 }
89 libc_hidden_def (__rpc_thread_svc_fdset)
90
91 struct rpc_createerr *
92 __rpc_thread_createerr (void)
93 {
94         struct rpc_thread_variables *tvp;
95
96         tvp = __rpc_thread_variables ();
97         if (tvp == &__libc_tsd_RPC_VARS_mem)
98                 return &rpc_createerr;
99         return &tvp->rpc_createerr_s;
100 }
101 libc_hidden_def (__rpc_thread_createerr)
102
103 struct pollfd **
104 __rpc_thread_svc_pollfd (void)
105 {
106         struct rpc_thread_variables *tvp;
107
108         tvp = __rpc_thread_variables ();
109         if (tvp == &__libc_tsd_RPC_VARS_mem)
110                 return &svc_pollfd;
111         return &tvp->svc_pollfd_s;
112 }
113 libc_hidden_def (__rpc_thread_svc_pollfd)
114
115 int *
116 __rpc_thread_svc_max_pollfd (void)
117 {
118         struct rpc_thread_variables *tvp;
119
120         tvp = __rpc_thread_variables ();
121         if (tvp == &__libc_tsd_RPC_VARS_mem)
122                 return &svc_max_pollfd;
123         return &tvp->svc_max_pollfd_s;
124 }
125 libc_hidden_def (__rpc_thread_svc_max_pollfd)
126
127 #endif /* _RPC_THREAD_SAFE_ */