lockd: Ensure NSM strings aren't longer than protocol allows
[safe/jmp/linux-2.6] / fs / lockd / mon.c
1 /*
2  * linux/fs/lockd/mon.c
3  *
4  * The kernel statd client.
5  *
6  * Copyright (C) 1996, Olaf Kirch <okir@monad.swb.de>
7  */
8
9 #include <linux/types.h>
10 #include <linux/utsname.h>
11 #include <linux/kernel.h>
12 #include <linux/sunrpc/clnt.h>
13 #include <linux/sunrpc/xprtsock.h>
14 #include <linux/sunrpc/svc.h>
15 #include <linux/lockd/lockd.h>
16 #include <linux/lockd/sm_inter.h>
17
18
19 #define NLMDBG_FACILITY         NLMDBG_MONITOR
20
21 static struct rpc_clnt *        nsm_create(void);
22
23 static struct rpc_program       nsm_program;
24
25 /*
26  * Local NSM state
27  */
28 int                             nsm_local_state;
29
30 /*
31  * Common procedure for SM_MON/SM_UNMON calls
32  */
33 static int
34 nsm_mon_unmon(struct nsm_handle *nsm, u32 proc, struct nsm_res *res)
35 {
36         struct rpc_clnt *clnt;
37         int             status;
38         struct nsm_args args;
39         struct rpc_message msg = {
40                 .rpc_argp       = &args,
41                 .rpc_resp       = res,
42         };
43
44         clnt = nsm_create();
45         if (IS_ERR(clnt)) {
46                 status = PTR_ERR(clnt);
47                 goto out;
48         }
49
50         memset(&args, 0, sizeof(args));
51         args.mon_name = nsm->sm_name;
52         args.addr = nsm->sm_addr.sin_addr.s_addr;
53         args.prog = NLM_PROGRAM;
54         args.vers = 3;
55         args.proc = NLMPROC_NSM_NOTIFY;
56         memset(res, 0, sizeof(*res));
57
58         msg.rpc_proc = &clnt->cl_procinfo[proc];
59         status = rpc_call_sync(clnt, &msg, 0);
60         if (status < 0)
61                 printk(KERN_DEBUG "nsm_mon_unmon: rpc failed, status=%d\n",
62                         status);
63         else
64                 status = 0;
65         rpc_shutdown_client(clnt);
66  out:
67         return status;
68 }
69
70 /*
71  * Set up monitoring of a remote host
72  */
73 int
74 nsm_monitor(struct nlm_host *host)
75 {
76         struct nsm_handle *nsm = host->h_nsmhandle;
77         struct nsm_res  res;
78         int             status;
79
80         dprintk("lockd: nsm_monitor(%s)\n", host->h_name);
81         BUG_ON(nsm == NULL);
82
83         if (nsm->sm_monitored)
84                 return 0;
85
86         status = nsm_mon_unmon(nsm, SM_MON, &res);
87
88         if (status < 0 || res.status != 0)
89                 printk(KERN_NOTICE "lockd: cannot monitor %s\n", host->h_name);
90         else
91                 nsm->sm_monitored = 1;
92         return status;
93 }
94
95 /*
96  * Cease to monitor remote host
97  */
98 int
99 nsm_unmonitor(struct nlm_host *host)
100 {
101         struct nsm_handle *nsm = host->h_nsmhandle;
102         struct nsm_res  res;
103         int             status = 0;
104
105         if (nsm == NULL)
106                 return 0;
107         host->h_nsmhandle = NULL;
108
109         if (atomic_read(&nsm->sm_count) == 1
110          && nsm->sm_monitored && !nsm->sm_sticky) {
111                 dprintk("lockd: nsm_unmonitor(%s)\n", host->h_name);
112
113                 status = nsm_mon_unmon(nsm, SM_UNMON, &res);
114                 if (status < 0)
115                         printk(KERN_NOTICE "lockd: cannot unmonitor %s\n",
116                                         host->h_name);
117                 else
118                         nsm->sm_monitored = 0;
119         }
120         nsm_release(nsm);
121         return status;
122 }
123
124 /*
125  * Create NSM client for the local host
126  */
127 static struct rpc_clnt *
128 nsm_create(void)
129 {
130         struct sockaddr_in      sin = {
131                 .sin_family     = AF_INET,
132                 .sin_addr.s_addr = htonl(INADDR_LOOPBACK),
133                 .sin_port       = 0,
134         };
135         struct rpc_create_args args = {
136                 .protocol       = XPRT_TRANSPORT_UDP,
137                 .address        = (struct sockaddr *)&sin,
138                 .addrsize       = sizeof(sin),
139                 .servername     = "localhost",
140                 .program        = &nsm_program,
141                 .version        = SM_VERSION,
142                 .authflavor     = RPC_AUTH_NULL,
143         };
144
145         return rpc_create(&args);
146 }
147
148 /*
149  * XDR functions for NSM.
150  */
151
152 static __be32 *xdr_encode_nsm_string(__be32 *p, char *string)
153 {
154         size_t len = strlen(string);
155
156         if (len > SM_MAXSTRLEN)
157                 len = SM_MAXSTRLEN;
158         return xdr_encode_opaque(p, string, len);
159 }
160
161 static __be32 *
162 xdr_encode_common(struct rpc_rqst *rqstp, __be32 *p, struct nsm_args *argp)
163 {
164         char    buffer[20], *name;
165
166         /*
167          * Use the dotted-quad IP address of the remote host as
168          * identifier. Linux statd always looks up the canonical
169          * hostname first for whatever remote hostname it receives,
170          * so this works alright.
171          */
172         if (nsm_use_hostnames) {
173                 name = argp->mon_name;
174         } else {
175                 sprintf(buffer, "%u.%u.%u.%u", NIPQUAD(argp->addr));
176                 name = buffer;
177         }
178         if (!(p = xdr_encode_string(p, name))
179          || !(p = xdr_encode_string(p, utsname()->nodename)))
180                 return ERR_PTR(-EIO);
181         *p++ = htonl(argp->prog);
182         *p++ = htonl(argp->vers);
183         *p++ = htonl(argp->proc);
184
185         return p;
186 }
187
188 static int
189 xdr_encode_mon(struct rpc_rqst *rqstp, __be32 *p, struct nsm_args *argp)
190 {
191         p = xdr_encode_common(rqstp, p, argp);
192         if (IS_ERR(p))
193                 return PTR_ERR(p);
194
195         /* Surprise - there may even be room for an IPv6 address now */
196         *p++ = argp->addr;
197         *p++ = 0;
198         *p++ = 0;
199         *p++ = 0;
200         rqstp->rq_slen = xdr_adjust_iovec(rqstp->rq_svec, p);
201         return 0;
202 }
203
204 static int
205 xdr_encode_unmon(struct rpc_rqst *rqstp, __be32 *p, struct nsm_args *argp)
206 {
207         p = xdr_encode_common(rqstp, p, argp);
208         if (IS_ERR(p))
209                 return PTR_ERR(p);
210         rqstp->rq_slen = xdr_adjust_iovec(rqstp->rq_svec, p);
211         return 0;
212 }
213
214 static int
215 xdr_decode_stat_res(struct rpc_rqst *rqstp, __be32 *p, struct nsm_res *resp)
216 {
217         resp->status = ntohl(*p++);
218         resp->state = ntohl(*p++);
219         dprintk("nsm: xdr_decode_stat_res status %d state %d\n",
220                         resp->status, resp->state);
221         return 0;
222 }
223
224 static int
225 xdr_decode_stat(struct rpc_rqst *rqstp, __be32 *p, struct nsm_res *resp)
226 {
227         resp->state = ntohl(*p++);
228         return 0;
229 }
230
231 #define SM_my_name_sz   (1+XDR_QUADLEN(SM_MAXSTRLEN))
232 #define SM_my_id_sz     (3+1+SM_my_name_sz)
233 #define SM_mon_id_sz    (1+XDR_QUADLEN(20)+SM_my_id_sz)
234 #define SM_mon_sz       (SM_mon_id_sz+4)
235 #define SM_monres_sz    2
236 #define SM_unmonres_sz  1
237
238 static struct rpc_procinfo      nsm_procedures[] = {
239 [SM_MON] = {
240                 .p_proc         = SM_MON,
241                 .p_encode       = (kxdrproc_t) xdr_encode_mon,
242                 .p_decode       = (kxdrproc_t) xdr_decode_stat_res,
243                 .p_arglen       = SM_mon_sz,
244                 .p_replen       = SM_monres_sz,
245                 .p_statidx      = SM_MON,
246                 .p_name         = "MONITOR",
247         },
248 [SM_UNMON] = {
249                 .p_proc         = SM_UNMON,
250                 .p_encode       = (kxdrproc_t) xdr_encode_unmon,
251                 .p_decode       = (kxdrproc_t) xdr_decode_stat,
252                 .p_arglen       = SM_mon_id_sz,
253                 .p_replen       = SM_unmonres_sz,
254                 .p_statidx      = SM_UNMON,
255                 .p_name         = "UNMONITOR",
256         },
257 };
258
259 static struct rpc_version       nsm_version1 = {
260                 .number         = 1,
261                 .nrprocs        = ARRAY_SIZE(nsm_procedures),
262                 .procs          = nsm_procedures
263 };
264
265 static struct rpc_version *     nsm_version[] = {
266         [1] = &nsm_version1,
267 };
268
269 static struct rpc_stat          nsm_stats;
270
271 static struct rpc_program       nsm_program = {
272                 .name           = "statd",
273                 .number         = SM_PROGRAM,
274                 .nrvers         = ARRAY_SIZE(nsm_version),
275                 .version        = nsm_version,
276                 .stats          = &nsm_stats
277 };