diff options
Diffstat (limited to 'net/sunrpc/pmap_clnt.c')
-rw-r--r-- | net/sunrpc/pmap_clnt.c | 298 |
1 files changed, 298 insertions, 0 deletions
diff --git a/net/sunrpc/pmap_clnt.c b/net/sunrpc/pmap_clnt.c new file mode 100644 index 000000000000..d0b1d2c34a4d --- /dev/null +++ b/net/sunrpc/pmap_clnt.c | |||
@@ -0,0 +1,298 @@ | |||
1 | /* | ||
2 | * linux/net/sunrpc/pmap.c | ||
3 | * | ||
4 | * Portmapper client. | ||
5 | * | ||
6 | * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de> | ||
7 | */ | ||
8 | |||
9 | #include <linux/config.h> | ||
10 | #include <linux/types.h> | ||
11 | #include <linux/socket.h> | ||
12 | #include <linux/kernel.h> | ||
13 | #include <linux/errno.h> | ||
14 | #include <linux/uio.h> | ||
15 | #include <linux/in.h> | ||
16 | #include <linux/sunrpc/clnt.h> | ||
17 | #include <linux/sunrpc/xprt.h> | ||
18 | #include <linux/sunrpc/sched.h> | ||
19 | |||
20 | #ifdef RPC_DEBUG | ||
21 | # define RPCDBG_FACILITY RPCDBG_PMAP | ||
22 | #endif | ||
23 | |||
24 | #define PMAP_SET 1 | ||
25 | #define PMAP_UNSET 2 | ||
26 | #define PMAP_GETPORT 3 | ||
27 | |||
28 | static struct rpc_procinfo pmap_procedures[]; | ||
29 | static struct rpc_clnt * pmap_create(char *, struct sockaddr_in *, int); | ||
30 | static void pmap_getport_done(struct rpc_task *); | ||
31 | static struct rpc_program pmap_program; | ||
32 | static DEFINE_SPINLOCK(pmap_lock); | ||
33 | |||
34 | /* | ||
35 | * Obtain the port for a given RPC service on a given host. This one can | ||
36 | * be called for an ongoing RPC request. | ||
37 | */ | ||
38 | void | ||
39 | rpc_getport(struct rpc_task *task, struct rpc_clnt *clnt) | ||
40 | { | ||
41 | struct rpc_portmap *map = clnt->cl_pmap; | ||
42 | struct sockaddr_in *sap = &clnt->cl_xprt->addr; | ||
43 | struct rpc_message msg = { | ||
44 | .rpc_proc = &pmap_procedures[PMAP_GETPORT], | ||
45 | .rpc_argp = map, | ||
46 | .rpc_resp = &clnt->cl_port, | ||
47 | .rpc_cred = NULL | ||
48 | }; | ||
49 | struct rpc_clnt *pmap_clnt; | ||
50 | struct rpc_task *child; | ||
51 | |||
52 | dprintk("RPC: %4d rpc_getport(%s, %d, %d, %d)\n", | ||
53 | task->tk_pid, clnt->cl_server, | ||
54 | map->pm_prog, map->pm_vers, map->pm_prot); | ||
55 | |||
56 | spin_lock(&pmap_lock); | ||
57 | if (map->pm_binding) { | ||
58 | rpc_sleep_on(&map->pm_bindwait, task, NULL, NULL); | ||
59 | spin_unlock(&pmap_lock); | ||
60 | return; | ||
61 | } | ||
62 | map->pm_binding = 1; | ||
63 | spin_unlock(&pmap_lock); | ||
64 | |||
65 | pmap_clnt = pmap_create(clnt->cl_server, sap, map->pm_prot); | ||
66 | if (IS_ERR(pmap_clnt)) { | ||
67 | task->tk_status = PTR_ERR(pmap_clnt); | ||
68 | goto bailout; | ||
69 | } | ||
70 | task->tk_status = 0; | ||
71 | |||
72 | /* | ||
73 | * Note: rpc_new_child will release client after a failure. | ||
74 | */ | ||
75 | if (!(child = rpc_new_child(pmap_clnt, task))) | ||
76 | goto bailout; | ||
77 | |||
78 | /* Setup the call info struct */ | ||
79 | rpc_call_setup(child, &msg, 0); | ||
80 | |||
81 | /* ... and run the child task */ | ||
82 | rpc_run_child(task, child, pmap_getport_done); | ||
83 | return; | ||
84 | |||
85 | bailout: | ||
86 | spin_lock(&pmap_lock); | ||
87 | map->pm_binding = 0; | ||
88 | rpc_wake_up(&map->pm_bindwait); | ||
89 | spin_unlock(&pmap_lock); | ||
90 | task->tk_status = -EIO; | ||
91 | task->tk_action = NULL; | ||
92 | } | ||
93 | |||
94 | #ifdef CONFIG_ROOT_NFS | ||
95 | int | ||
96 | rpc_getport_external(struct sockaddr_in *sin, __u32 prog, __u32 vers, int prot) | ||
97 | { | ||
98 | struct rpc_portmap map = { | ||
99 | .pm_prog = prog, | ||
100 | .pm_vers = vers, | ||
101 | .pm_prot = prot, | ||
102 | .pm_port = 0 | ||
103 | }; | ||
104 | struct rpc_clnt *pmap_clnt; | ||
105 | char hostname[32]; | ||
106 | int status; | ||
107 | |||
108 | dprintk("RPC: rpc_getport_external(%u.%u.%u.%u, %d, %d, %d)\n", | ||
109 | NIPQUAD(sin->sin_addr.s_addr), prog, vers, prot); | ||
110 | |||
111 | sprintf(hostname, "%u.%u.%u.%u", NIPQUAD(sin->sin_addr.s_addr)); | ||
112 | pmap_clnt = pmap_create(hostname, sin, prot); | ||
113 | if (IS_ERR(pmap_clnt)) | ||
114 | return PTR_ERR(pmap_clnt); | ||
115 | |||
116 | /* Setup the call info struct */ | ||
117 | status = rpc_call(pmap_clnt, PMAP_GETPORT, &map, &map.pm_port, 0); | ||
118 | |||
119 | if (status >= 0) { | ||
120 | if (map.pm_port != 0) | ||
121 | return map.pm_port; | ||
122 | status = -EACCES; | ||
123 | } | ||
124 | return status; | ||
125 | } | ||
126 | #endif | ||
127 | |||
128 | static void | ||
129 | pmap_getport_done(struct rpc_task *task) | ||
130 | { | ||
131 | struct rpc_clnt *clnt = task->tk_client; | ||
132 | struct rpc_portmap *map = clnt->cl_pmap; | ||
133 | |||
134 | dprintk("RPC: %4d pmap_getport_done(status %d, port %d)\n", | ||
135 | task->tk_pid, task->tk_status, clnt->cl_port); | ||
136 | if (task->tk_status < 0) { | ||
137 | /* Make the calling task exit with an error */ | ||
138 | task->tk_action = NULL; | ||
139 | } else if (clnt->cl_port == 0) { | ||
140 | /* Program not registered */ | ||
141 | task->tk_status = -EACCES; | ||
142 | task->tk_action = NULL; | ||
143 | } else { | ||
144 | /* byte-swap port number first */ | ||
145 | clnt->cl_port = htons(clnt->cl_port); | ||
146 | clnt->cl_xprt->addr.sin_port = clnt->cl_port; | ||
147 | } | ||
148 | spin_lock(&pmap_lock); | ||
149 | map->pm_binding = 0; | ||
150 | rpc_wake_up(&map->pm_bindwait); | ||
151 | spin_unlock(&pmap_lock); | ||
152 | } | ||
153 | |||
154 | /* | ||
155 | * Set or unset a port registration with the local portmapper. | ||
156 | * port == 0 means unregister, port != 0 means register. | ||
157 | */ | ||
158 | int | ||
159 | rpc_register(u32 prog, u32 vers, int prot, unsigned short port, int *okay) | ||
160 | { | ||
161 | struct sockaddr_in sin; | ||
162 | struct rpc_portmap map; | ||
163 | struct rpc_clnt *pmap_clnt; | ||
164 | int error = 0; | ||
165 | |||
166 | dprintk("RPC: registering (%d, %d, %d, %d) with portmapper.\n", | ||
167 | prog, vers, prot, port); | ||
168 | |||
169 | sin.sin_family = AF_INET; | ||
170 | sin.sin_addr.s_addr = htonl(INADDR_LOOPBACK); | ||
171 | pmap_clnt = pmap_create("localhost", &sin, IPPROTO_UDP); | ||
172 | if (IS_ERR(pmap_clnt)) { | ||
173 | error = PTR_ERR(pmap_clnt); | ||
174 | dprintk("RPC: couldn't create pmap client. Error = %d\n", error); | ||
175 | return error; | ||
176 | } | ||
177 | |||
178 | map.pm_prog = prog; | ||
179 | map.pm_vers = vers; | ||
180 | map.pm_prot = prot; | ||
181 | map.pm_port = port; | ||
182 | |||
183 | error = rpc_call(pmap_clnt, port? PMAP_SET : PMAP_UNSET, | ||
184 | &map, okay, 0); | ||
185 | |||
186 | if (error < 0) { | ||
187 | printk(KERN_WARNING | ||
188 | "RPC: failed to contact portmap (errno %d).\n", | ||
189 | error); | ||
190 | } | ||
191 | dprintk("RPC: registration status %d/%d\n", error, *okay); | ||
192 | |||
193 | /* Client deleted automatically because cl_oneshot == 1 */ | ||
194 | return error; | ||
195 | } | ||
196 | |||
197 | static struct rpc_clnt * | ||
198 | pmap_create(char *hostname, struct sockaddr_in *srvaddr, int proto) | ||
199 | { | ||
200 | struct rpc_xprt *xprt; | ||
201 | struct rpc_clnt *clnt; | ||
202 | |||
203 | /* printk("pmap: create xprt\n"); */ | ||
204 | xprt = xprt_create_proto(proto, srvaddr, NULL); | ||
205 | if (IS_ERR(xprt)) | ||
206 | return (struct rpc_clnt *)xprt; | ||
207 | xprt->addr.sin_port = htons(RPC_PMAP_PORT); | ||
208 | |||
209 | /* printk("pmap: create clnt\n"); */ | ||
210 | clnt = rpc_create_client(xprt, hostname, | ||
211 | &pmap_program, RPC_PMAP_VERSION, | ||
212 | RPC_AUTH_UNIX); | ||
213 | if (IS_ERR(clnt)) { | ||
214 | xprt_destroy(xprt); | ||
215 | } else { | ||
216 | clnt->cl_softrtry = 1; | ||
217 | clnt->cl_chatty = 1; | ||
218 | clnt->cl_oneshot = 1; | ||
219 | } | ||
220 | return clnt; | ||
221 | } | ||
222 | |||
223 | /* | ||
224 | * XDR encode/decode functions for PMAP | ||
225 | */ | ||
226 | static int | ||
227 | xdr_encode_mapping(struct rpc_rqst *req, u32 *p, struct rpc_portmap *map) | ||
228 | { | ||
229 | dprintk("RPC: xdr_encode_mapping(%d, %d, %d, %d)\n", | ||
230 | map->pm_prog, map->pm_vers, map->pm_prot, map->pm_port); | ||
231 | *p++ = htonl(map->pm_prog); | ||
232 | *p++ = htonl(map->pm_vers); | ||
233 | *p++ = htonl(map->pm_prot); | ||
234 | *p++ = htonl(map->pm_port); | ||
235 | |||
236 | req->rq_slen = xdr_adjust_iovec(req->rq_svec, p); | ||
237 | return 0; | ||
238 | } | ||
239 | |||
240 | static int | ||
241 | xdr_decode_port(struct rpc_rqst *req, u32 *p, unsigned short *portp) | ||
242 | { | ||
243 | *portp = (unsigned short) ntohl(*p++); | ||
244 | return 0; | ||
245 | } | ||
246 | |||
247 | static int | ||
248 | xdr_decode_bool(struct rpc_rqst *req, u32 *p, unsigned int *boolp) | ||
249 | { | ||
250 | *boolp = (unsigned int) ntohl(*p++); | ||
251 | return 0; | ||
252 | } | ||
253 | |||
254 | static struct rpc_procinfo pmap_procedures[] = { | ||
255 | [PMAP_SET] = { | ||
256 | .p_proc = PMAP_SET, | ||
257 | .p_encode = (kxdrproc_t) xdr_encode_mapping, | ||
258 | .p_decode = (kxdrproc_t) xdr_decode_bool, | ||
259 | .p_bufsiz = 4, | ||
260 | .p_count = 1, | ||
261 | }, | ||
262 | [PMAP_UNSET] = { | ||
263 | .p_proc = PMAP_UNSET, | ||
264 | .p_encode = (kxdrproc_t) xdr_encode_mapping, | ||
265 | .p_decode = (kxdrproc_t) xdr_decode_bool, | ||
266 | .p_bufsiz = 4, | ||
267 | .p_count = 1, | ||
268 | }, | ||
269 | [PMAP_GETPORT] = { | ||
270 | .p_proc = PMAP_GETPORT, | ||
271 | .p_encode = (kxdrproc_t) xdr_encode_mapping, | ||
272 | .p_decode = (kxdrproc_t) xdr_decode_port, | ||
273 | .p_bufsiz = 4, | ||
274 | .p_count = 1, | ||
275 | }, | ||
276 | }; | ||
277 | |||
278 | static struct rpc_version pmap_version2 = { | ||
279 | .number = 2, | ||
280 | .nrprocs = 4, | ||
281 | .procs = pmap_procedures | ||
282 | }; | ||
283 | |||
284 | static struct rpc_version * pmap_version[] = { | ||
285 | NULL, | ||
286 | NULL, | ||
287 | &pmap_version2 | ||
288 | }; | ||
289 | |||
290 | static struct rpc_stat pmap_stats; | ||
291 | |||
292 | static struct rpc_program pmap_program = { | ||
293 | .name = "portmap", | ||
294 | .number = RPC_PMAP_PROGRAM, | ||
295 | .nrvers = ARRAY_SIZE(pmap_version), | ||
296 | .version = pmap_version, | ||
297 | .stats = &pmap_stats, | ||
298 | }; | ||