b45b59b17ae1b21e56c7cd72db4ef0f1a074fb18
[safe/jmp/linux-2.6] / net / sunrpc / auth_gss / gss_krb5_wrap.c
1 #include <linux/types.h>
2 #include <linux/jiffies.h>
3 #include <linux/sunrpc/gss_krb5.h>
4 #include <linux/random.h>
5 #include <linux/pagemap.h>
6 #include <linux/crypto.h>
7
8 #ifdef RPC_DEBUG
9 # define RPCDBG_FACILITY        RPCDBG_AUTH
10 #endif
11
12 static inline int
13 gss_krb5_padding(int blocksize, int length)
14 {
15         return blocksize - (length % blocksize);
16 }
17
18 static inline void
19 gss_krb5_add_padding(struct xdr_buf *buf, int offset, int blocksize)
20 {
21         int padding = gss_krb5_padding(blocksize, buf->len - offset);
22         char *p;
23         struct kvec *iov;
24
25         if (buf->page_len || buf->tail[0].iov_len)
26                 iov = &buf->tail[0];
27         else
28                 iov = &buf->head[0];
29         p = iov->iov_base + iov->iov_len;
30         iov->iov_len += padding;
31         buf->len += padding;
32         memset(p, padding, padding);
33 }
34
35 static inline int
36 gss_krb5_remove_padding(struct xdr_buf *buf, int blocksize)
37 {
38         u8 *ptr;
39         u8 pad;
40         size_t len = buf->len;
41
42         if (len <= buf->head[0].iov_len) {
43                 pad = *(u8 *)(buf->head[0].iov_base + len - 1);
44                 if (pad > buf->head[0].iov_len)
45                         return -EINVAL;
46                 buf->head[0].iov_len -= pad;
47                 goto out;
48         } else
49                 len -= buf->head[0].iov_len;
50         if (len <= buf->page_len) {
51                 unsigned int last = (buf->page_base + len - 1)
52                                         >>PAGE_CACHE_SHIFT;
53                 unsigned int offset = (buf->page_base + len - 1)
54                                         & (PAGE_CACHE_SIZE - 1);
55                 ptr = kmap_atomic(buf->pages[last], KM_USER0);
56                 pad = *(ptr + offset);
57                 kunmap_atomic(ptr, KM_USER0);
58                 goto out;
59         } else
60                 len -= buf->page_len;
61         BUG_ON(len > buf->tail[0].iov_len);
62         pad = *(u8 *)(buf->tail[0].iov_base + len - 1);
63 out:
64         /* XXX: NOTE: we do not adjust the page lengths--they represent
65          * a range of data in the real filesystem page cache, and we need
66          * to know that range so the xdr code can properly place read data.
67          * However adjusting the head length, as we do above, is harmless.
68          * In the case of a request that fits into a single page, the server
69          * also uses length and head length together to determine the original
70          * start of the request to copy the request for deferal; so it's
71          * easier on the server if we adjust head and tail length in tandem.
72          * It's not really a problem that we don't fool with the page and
73          * tail lengths, though--at worst badly formed xdr might lead the
74          * server to attempt to parse the padding.
75          * XXX: Document all these weird requirements for gss mechanism
76          * wrap/unwrap functions. */
77         if (pad > blocksize)
78                 return -EINVAL;
79         if (buf->len > pad)
80                 buf->len -= pad;
81         else
82                 return -EINVAL;
83         return 0;
84 }
85
86 static void
87 make_confounder(char *p, u32 conflen)
88 {
89         static u64 i = 0;
90         u64 *q = (u64 *)p;
91
92         /* rfc1964 claims this should be "random".  But all that's really
93          * necessary is that it be unique.  And not even that is necessary in
94          * our case since our "gssapi" implementation exists only to support
95          * rpcsec_gss, so we know that the only buffers we will ever encrypt
96          * already begin with a unique sequence number.  Just to hedge my bets
97          * I'll make a half-hearted attempt at something unique, but ensuring
98          * uniqueness would mean worrying about atomicity and rollover, and I
99          * don't care enough. */
100
101         /* initialize to random value */
102         if (i == 0) {
103                 i = random32();
104                 i = (i << 32) | random32();
105         }
106
107         switch (conflen) {
108         case 16:
109                 *q++ = i++;
110                 /* fall through */
111         case 8:
112                 *q++ = i++;
113                 break;
114         default:
115                 BUG();
116         }
117 }
118
119 /* Assumptions: the head and tail of inbuf are ours to play with.
120  * The pages, however, may be real pages in the page cache and we replace
121  * them with scratch pages from **pages before writing to them. */
122 /* XXX: obviously the above should be documentation of wrap interface,
123  * and shouldn't be in this kerberos-specific file. */
124
125 /* XXX factor out common code with seal/unseal. */
126
127 static u32
128 gss_wrap_kerberos_v1(struct krb5_ctx *kctx, int offset,
129                 struct xdr_buf *buf, struct page **pages)
130 {
131         char                    cksumdata[16];
132         struct xdr_netobj       md5cksum = {.len = 0, .data = cksumdata};
133         int                     blocksize = 0, plainlen;
134         unsigned char           *ptr, *msg_start;
135         s32                     now;
136         int                     headlen;
137         struct page             **tmp_pages;
138         u32                     seq_send;
139
140         dprintk("RPC:       gss_wrap_kerberos\n");
141
142         now = get_seconds();
143
144         blocksize = crypto_blkcipher_blocksize(kctx->enc);
145         gss_krb5_add_padding(buf, offset, blocksize);
146         BUG_ON((buf->len - offset) % blocksize);
147         plainlen = blocksize + buf->len - offset;
148
149         headlen = g_token_size(&kctx->mech_used, 24 + plainlen) -
150                                                 (buf->len - offset);
151
152         ptr = buf->head[0].iov_base + offset;
153         /* shift data to make room for header. */
154         xdr_extend_head(buf, offset, headlen);
155
156         /* XXX Would be cleverer to encrypt while copying. */
157         BUG_ON((buf->len - offset - headlen) % blocksize);
158
159         g_make_token_header(&kctx->mech_used,
160                                 GSS_KRB5_TOK_HDR_LEN + 8 + plainlen, &ptr);
161
162
163         /* ptr now at header described in rfc 1964, section 1.2.1: */
164         ptr[0] = (unsigned char) ((KG_TOK_WRAP_MSG >> 8) & 0xff);
165         ptr[1] = (unsigned char) (KG_TOK_WRAP_MSG & 0xff);
166
167         msg_start = ptr + 24;
168
169         *(__be16 *)(ptr + 2) = htons(SGN_ALG_DES_MAC_MD5);
170         memset(ptr + 4, 0xff, 4);
171         *(__be16 *)(ptr + 4) = htons(SEAL_ALG_DES);
172
173         make_confounder(msg_start, blocksize);
174
175         /* XXXJBF: UGH!: */
176         tmp_pages = buf->pages;
177         buf->pages = pages;
178         if (make_checksum("md5", ptr, 8, buf,
179                                 offset + headlen - blocksize, &md5cksum))
180                 return GSS_S_FAILURE;
181         buf->pages = tmp_pages;
182
183         if (krb5_encrypt(kctx->seq, NULL, md5cksum.data,
184                           md5cksum.data, md5cksum.len))
185                 return GSS_S_FAILURE;
186         memcpy(ptr + GSS_KRB5_TOK_HDR_LEN, md5cksum.data + md5cksum.len - 8, 8);
187
188         spin_lock(&krb5_seq_lock);
189         seq_send = kctx->seq_send++;
190         spin_unlock(&krb5_seq_lock);
191
192         /* XXX would probably be more efficient to compute checksum
193          * and encrypt at the same time: */
194         if ((krb5_make_seq_num(kctx->seq, kctx->initiate ? 0 : 0xff,
195                                seq_send, ptr + GSS_KRB5_TOK_HDR_LEN, ptr + 8)))
196                 return GSS_S_FAILURE;
197
198         if (gss_encrypt_xdr_buf(kctx->enc, buf, offset + headlen - blocksize,
199                                                                         pages))
200                 return GSS_S_FAILURE;
201
202         return (kctx->endtime < now) ? GSS_S_CONTEXT_EXPIRED : GSS_S_COMPLETE;
203 }
204
205 static u32
206 gss_unwrap_kerberos_v1(struct krb5_ctx *kctx, int offset, struct xdr_buf *buf)
207 {
208         int                     signalg;
209         int                     sealalg;
210         char                    cksumdata[16];
211         struct xdr_netobj       md5cksum = {.len = 0, .data = cksumdata};
212         s32                     now;
213         int                     direction;
214         s32                     seqnum;
215         unsigned char           *ptr;
216         int                     bodysize;
217         void                    *data_start, *orig_start;
218         int                     data_len;
219         int                     blocksize;
220
221         dprintk("RPC:       gss_unwrap_kerberos\n");
222
223         ptr = (u8 *)buf->head[0].iov_base + offset;
224         if (g_verify_token_header(&kctx->mech_used, &bodysize, &ptr,
225                                         buf->len - offset))
226                 return GSS_S_DEFECTIVE_TOKEN;
227
228         if ((ptr[0] != ((KG_TOK_WRAP_MSG >> 8) & 0xff)) ||
229             (ptr[1] !=  (KG_TOK_WRAP_MSG & 0xff)))
230                 return GSS_S_DEFECTIVE_TOKEN;
231
232         /* XXX sanity-check bodysize?? */
233
234         /* get the sign and seal algorithms */
235
236         signalg = ptr[2] + (ptr[3] << 8);
237         if (signalg != SGN_ALG_DES_MAC_MD5)
238                 return GSS_S_DEFECTIVE_TOKEN;
239
240         sealalg = ptr[4] + (ptr[5] << 8);
241         if (sealalg != SEAL_ALG_DES)
242                 return GSS_S_DEFECTIVE_TOKEN;
243
244         if ((ptr[6] != 0xff) || (ptr[7] != 0xff))
245                 return GSS_S_DEFECTIVE_TOKEN;
246
247         if (gss_decrypt_xdr_buf(kctx->enc, buf,
248                         ptr + GSS_KRB5_TOK_HDR_LEN + 8 - (unsigned char *)buf->head[0].iov_base))
249                 return GSS_S_DEFECTIVE_TOKEN;
250
251         if (make_checksum("md5", ptr, 8, buf,
252                  ptr + GSS_KRB5_TOK_HDR_LEN + 8 - (unsigned char *)buf->head[0].iov_base, &md5cksum))
253                 return GSS_S_FAILURE;
254
255         if (krb5_encrypt(kctx->seq, NULL, md5cksum.data,
256                            md5cksum.data, md5cksum.len))
257                 return GSS_S_FAILURE;
258
259         if (memcmp(md5cksum.data + 8, ptr + GSS_KRB5_TOK_HDR_LEN, 8))
260                 return GSS_S_BAD_SIG;
261
262         /* it got through unscathed.  Make sure the context is unexpired */
263
264         now = get_seconds();
265
266         if (now > kctx->endtime)
267                 return GSS_S_CONTEXT_EXPIRED;
268
269         /* do sequencing checks */
270
271         if (krb5_get_seq_num(kctx->seq, ptr + GSS_KRB5_TOK_HDR_LEN, ptr + 8,
272                                     &direction, &seqnum))
273                 return GSS_S_BAD_SIG;
274
275         if ((kctx->initiate && direction != 0xff) ||
276             (!kctx->initiate && direction != 0))
277                 return GSS_S_BAD_SIG;
278
279         /* Copy the data back to the right position.  XXX: Would probably be
280          * better to copy and encrypt at the same time. */
281
282         blocksize = crypto_blkcipher_blocksize(kctx->enc);
283         data_start = ptr + GSS_KRB5_TOK_HDR_LEN + 8 + blocksize;
284         orig_start = buf->head[0].iov_base + offset;
285         data_len = (buf->head[0].iov_base + buf->head[0].iov_len) - data_start;
286         memmove(orig_start, data_start, data_len);
287         buf->head[0].iov_len -= (data_start - orig_start);
288         buf->len -= (data_start - orig_start);
289
290         if (gss_krb5_remove_padding(buf, blocksize))
291                 return GSS_S_DEFECTIVE_TOKEN;
292
293         return GSS_S_COMPLETE;
294 }
295
296 u32
297 gss_wrap_kerberos(struct gss_ctx *gctx, int offset,
298                   struct xdr_buf *buf, struct page **pages)
299 {
300         struct krb5_ctx *kctx = gctx->internal_ctx_id;
301
302         switch (kctx->enctype) {
303         default:
304                 BUG();
305         case ENCTYPE_DES_CBC_RAW:
306                 return gss_wrap_kerberos_v1(kctx, offset, buf, pages);
307         }
308 }
309
310 u32
311 gss_unwrap_kerberos(struct gss_ctx *gctx, int offset, struct xdr_buf *buf)
312 {
313         struct krb5_ctx *kctx = gctx->internal_ctx_id;
314
315         switch (kctx->enctype) {
316         default:
317                 BUG();
318         case ENCTYPE_DES_CBC_RAW:
319                 return gss_unwrap_kerberos_v1(kctx, offset, buf);
320         }
321 }
322