1 /*	$NetBSD: digest.c,v 1.1.1.1 2011/04/13 18:15:33 elric Exp $	*/
2 
3 /*
4  * Copyright (c) 2006 Kungliga Tekniska Högskolan
5  * (Royal Institute of Technology, Stockholm, Sweden).
6  * All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  *
12  * 1. Redistributions of source code must retain the above copyright
13  *    notice, this list of conditions and the following disclaimer.
14  *
15  * 2. Redistributions in binary form must reproduce the above copyright
16  *    notice, this list of conditions and the following disclaimer in the
17  *    documentation and/or other materials provided with the distribution.
18  *
19  * 3. Neither the name of the Institute nor the names of its contributors
20  *    may be used to endorse or promote products derived from this software
21  *    without specific prior written permission.
22  *
23  * THIS SOFTWARE IS PROVIDED BY THE INSTITUTE AND CONTRIBUTORS ``AS IS'' AND
24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE INSTITUTE OR CONTRIBUTORS BE LIABLE
27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33  * SUCH DAMAGE.
34  */
35 
36 #include "krb5_locl.h"
37 #include <krb5/digest_asn1.h>
38 
39 #ifndef HEIMDAL_SMALLER
40 
41 struct krb5_digest_data {
42     char *cbtype;
43     char *cbbinding;
44 
45     DigestInit init;
46     DigestInitReply initReply;
47     DigestRequest request;
48     DigestResponse response;
49 };
50 
51 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
52 krb5_digest_alloc(krb5_context context, krb5_digest *digest)
53 {
54     krb5_digest d;
55 
56     d = calloc(1, sizeof(*d));
57     if (d == NULL) {
58 	*digest = NULL;
59 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
60 	return ENOMEM;
61     }
62     *digest = d;
63 
64     return 0;
65 }
66 
67 KRB5_LIB_FUNCTION void KRB5_LIB_CALL
68 krb5_digest_free(krb5_digest digest)
69 {
70     if (digest == NULL)
71 	return;
72     free_DigestInit(&digest->init);
73     free_DigestInitReply(&digest->initReply);
74     free_DigestRequest(&digest->request);
75     free_DigestResponse(&digest->response);
76     memset(digest, 0, sizeof(*digest));
77     free(digest);
78     return;
79 }
80 
81 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
82 krb5_digest_set_server_cb(krb5_context context,
83 			  krb5_digest digest,
84 			  const char *type,
85 			  const char *binding)
86 {
87     if (digest->init.channel) {
88 	krb5_set_error_message(context, EINVAL,
89 			       N_("server channel binding already set", ""));
90 	return EINVAL;
91     }
92     digest->init.channel = calloc(1, sizeof(*digest->init.channel));
93     if (digest->init.channel == NULL)
94 	goto error;
95 
96     digest->init.channel->cb_type = strdup(type);
97     if (digest->init.channel->cb_type == NULL)
98 	goto error;
99 
100     digest->init.channel->cb_binding = strdup(binding);
101     if (digest->init.channel->cb_binding == NULL)
102 	goto error;
103     return 0;
104  error:
105     if (digest->init.channel) {
106 	free(digest->init.channel->cb_type);
107 	free(digest->init.channel->cb_binding);
108 	free(digest->init.channel);
109 	digest->init.channel = NULL;
110     }
111     krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
112     return ENOMEM;
113 }
114 
115 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
116 krb5_digest_set_type(krb5_context context,
117 		     krb5_digest digest,
118 		     const char *type)
119 {
120     if (digest->init.type) {
121 	krb5_set_error_message(context, EINVAL, "client type already set");
122 	return EINVAL;
123     }
124     digest->init.type = strdup(type);
125     if (digest->init.type == NULL) {
126 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
127 	return ENOMEM;
128     }
129     return 0;
130 }
131 
132 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
133 krb5_digest_set_hostname(krb5_context context,
134 			 krb5_digest digest,
135 			 const char *hostname)
136 {
137     if (digest->init.hostname) {
138 	krb5_set_error_message(context, EINVAL, "server hostname already set");
139 	return EINVAL;
140     }
141     digest->init.hostname = malloc(sizeof(*digest->init.hostname));
142     if (digest->init.hostname == NULL) {
143 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
144 	return ENOMEM;
145     }
146     *digest->init.hostname = strdup(hostname);
147     if (*digest->init.hostname == NULL) {
148 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
149 	free(digest->init.hostname);
150 	digest->init.hostname = NULL;
151 	return ENOMEM;
152     }
153     return 0;
154 }
155 
156 KRB5_LIB_FUNCTION const char * KRB5_LIB_CALL
157 krb5_digest_get_server_nonce(krb5_context context,
158 			     krb5_digest digest)
159 {
160     return digest->initReply.nonce;
161 }
162 
163 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
164 krb5_digest_set_server_nonce(krb5_context context,
165 			     krb5_digest digest,
166 			     const char *nonce)
167 {
168     if (digest->request.serverNonce) {
169 	krb5_set_error_message(context, EINVAL, N_("nonce already set", ""));
170 	return EINVAL;
171     }
172     digest->request.serverNonce = strdup(nonce);
173     if (digest->request.serverNonce == NULL) {
174 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
175 	return ENOMEM;
176     }
177     return 0;
178 }
179 
180 KRB5_LIB_FUNCTION const char * KRB5_LIB_CALL
181 krb5_digest_get_opaque(krb5_context context,
182 		       krb5_digest digest)
183 {
184     return digest->initReply.opaque;
185 }
186 
187 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
188 krb5_digest_set_opaque(krb5_context context,
189 		       krb5_digest digest,
190 		       const char *opaque)
191 {
192     if (digest->request.opaque) {
193 	krb5_set_error_message(context, EINVAL, "opaque already set");
194 	return EINVAL;
195     }
196     digest->request.opaque = strdup(opaque);
197     if (digest->request.opaque == NULL) {
198 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
199 	return ENOMEM;
200     }
201     return 0;
202 }
203 
204 KRB5_LIB_FUNCTION const char * KRB5_LIB_CALL
205 krb5_digest_get_identifier(krb5_context context,
206 			   krb5_digest digest)
207 {
208     if (digest->initReply.identifier == NULL)
209 	return NULL;
210     return *digest->initReply.identifier;
211 }
212 
213 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
214 krb5_digest_set_identifier(krb5_context context,
215 			   krb5_digest digest,
216 			   const char *id)
217 {
218     if (digest->request.identifier) {
219 	krb5_set_error_message(context, EINVAL, N_("identifier already set", ""));
220 	return EINVAL;
221     }
222     digest->request.identifier = calloc(1, sizeof(*digest->request.identifier));
223     if (digest->request.identifier == NULL) {
224 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
225 	return ENOMEM;
226     }
227     *digest->request.identifier = strdup(id);
228     if (*digest->request.identifier == NULL) {
229 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
230 	free(digest->request.identifier);
231 	digest->request.identifier = NULL;
232 	return ENOMEM;
233     }
234     return 0;
235 }
236 
237 static krb5_error_code
238 digest_request(krb5_context context,
239 	       krb5_realm realm,
240 	       krb5_ccache ccache,
241 	       krb5_key_usage usage,
242 	       const DigestReqInner *ireq,
243 	       DigestRepInner *irep)
244 {
245     DigestREQ req;
246     DigestREP rep;
247     krb5_error_code ret;
248     krb5_data data, data2;
249     size_t size;
250     krb5_crypto crypto = NULL;
251     krb5_auth_context ac = NULL;
252     krb5_principal principal = NULL;
253     krb5_ccache id = NULL;
254     krb5_realm r = NULL;
255 
256     krb5_data_zero(&data);
257     krb5_data_zero(&data2);
258     memset(&req, 0, sizeof(req));
259     memset(&rep, 0, sizeof(rep));
260 
261     if (ccache == NULL) {
262 	ret = krb5_cc_default(context, &id);
263 	if (ret)
264 	    goto out;
265     } else
266 	id = ccache;
267 
268     if (realm == NULL) {
269 	ret = krb5_get_default_realm(context, &r);
270 	if (ret)
271 	    goto out;
272     } else
273 	r = realm;
274 
275     /*
276      *
277      */
278 
279     ret = krb5_make_principal(context, &principal,
280 			      r, KRB5_DIGEST_NAME, r, NULL);
281     if (ret)
282 	goto out;
283 
284     ASN1_MALLOC_ENCODE(DigestReqInner, data.data, data.length,
285 		       ireq, &size, ret);
286     if (ret) {
287 	krb5_set_error_message(context, ret,
288 			       N_("Failed to encode digest inner request", ""));
289 	goto out;
290     }
291     if (size != data.length)
292 	krb5_abortx(context, "ASN.1 internal encoder error");
293 
294     ret = krb5_mk_req_exact(context, &ac,
295 			    AP_OPTS_USE_SUBKEY|AP_OPTS_MUTUAL_REQUIRED,
296 			    principal, NULL, id, &req.apReq);
297     if (ret)
298 	goto out;
299 
300     {
301 	krb5_keyblock *key;
302 
303 	ret = krb5_auth_con_getlocalsubkey(context, ac, &key);
304 	if (ret)
305 	    goto out;
306 	if (key == NULL) {
307 	    ret = EINVAL;
308 	    krb5_set_error_message(context, ret,
309 				   N_("Digest failed to get local subkey", ""));
310 	    goto out;
311 	}
312 
313 	ret = krb5_crypto_init(context, key, 0, &crypto);
314 	krb5_free_keyblock (context, key);
315 	if (ret)
316 	    goto out;
317     }
318 
319     ret = krb5_encrypt_EncryptedData(context, crypto, usage,
320 				     data.data, data.length, 0,
321 				     &req.innerReq);
322     if (ret)
323 	goto out;
324 
325     krb5_data_free(&data);
326 
327     ASN1_MALLOC_ENCODE(DigestREQ, data.data, data.length,
328 		       &req, &size, ret);
329     if (ret) {
330 	krb5_set_error_message(context, ret,
331 			       N_("Failed to encode DigestREQest", ""));
332 	goto out;
333     }
334     if (size != data.length)
335 	krb5_abortx(context, "ASN.1 internal encoder error");
336 
337     ret = krb5_sendto_kdc(context, &data, &r, &data2);
338     if (ret)
339 	goto out;
340 
341     ret = decode_DigestREP(data2.data, data2.length, &rep, NULL);
342     if (ret) {
343 	krb5_set_error_message(context, ret,
344 			       N_("Failed to parse digest response", ""));
345 	goto out;
346     }
347 
348     {
349 	krb5_ap_rep_enc_part *repl;
350 
351 	ret = krb5_rd_rep(context, ac, &rep.apRep, &repl);
352 	if (ret)
353 	    goto out;
354 
355 	krb5_free_ap_rep_enc_part(context, repl);
356     }
357     {
358 	krb5_keyblock *key;
359 
360 	ret = krb5_auth_con_getremotesubkey(context, ac, &key);
361 	if (ret)
362 	    goto out;
363 	if (key == NULL) {
364 	    ret = EINVAL;
365 	    krb5_set_error_message(context, ret,
366 				   N_("Digest reply have no remote subkey", ""));
367 	    goto out;
368 	}
369 
370 	krb5_crypto_destroy(context, crypto);
371 	ret = krb5_crypto_init(context, key, 0, &crypto);
372 	krb5_free_keyblock (context, key);
373 	if (ret)
374 	    goto out;
375     }
376 
377     krb5_data_free(&data);
378     ret = krb5_decrypt_EncryptedData(context, crypto, usage,
379 				     &rep.innerRep, &data);
380     if (ret)
381 	goto out;
382 
383     ret = decode_DigestRepInner(data.data, data.length, irep, NULL);
384     if (ret) {
385 	krb5_set_error_message(context, ret,
386 			       N_("Failed to decode digest inner reply", ""));
387 	goto out;
388     }
389 
390  out:
391     if (ccache == NULL && id)
392 	krb5_cc_close(context, id);
393     if (realm == NULL && r)
394 	free(r);
395     if (crypto)
396 	krb5_crypto_destroy(context, crypto);
397     if (ac)
398 	krb5_auth_con_free(context, ac);
399     if (principal)
400 	krb5_free_principal(context, principal);
401 
402     krb5_data_free(&data);
403     krb5_data_free(&data2);
404 
405     free_DigestREQ(&req);
406     free_DigestREP(&rep);
407 
408     return ret;
409 }
410 
411 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
412 krb5_digest_init_request(krb5_context context,
413 			 krb5_digest digest,
414 			 krb5_realm realm,
415 			 krb5_ccache ccache)
416 {
417     DigestReqInner ireq;
418     DigestRepInner irep;
419     krb5_error_code ret;
420 
421     memset(&ireq, 0, sizeof(ireq));
422     memset(&irep, 0, sizeof(irep));
423 
424     if (digest->init.type == NULL) {
425 	krb5_set_error_message(context, EINVAL,
426 			       N_("Type missing from init req", ""));
427 	return EINVAL;
428     }
429 
430     ireq.element = choice_DigestReqInner_init;
431     ireq.u.init = digest->init;
432 
433     ret = digest_request(context, realm, ccache,
434 			 KRB5_KU_DIGEST_ENCRYPT, &ireq, &irep);
435     if (ret)
436 	goto out;
437 
438     if (irep.element == choice_DigestRepInner_error) {
439 	ret = irep.u.error.code;
440 	krb5_set_error_message(context, ret, N_("Digest init error: %s", ""),
441 			       irep.u.error.reason);
442 	goto out;
443     }
444 
445     if (irep.element != choice_DigestRepInner_initReply) {
446 	ret = EINVAL;
447 	krb5_set_error_message(context, ret,
448 			       N_("digest reply not an initReply", ""));
449 	goto out;
450     }
451 
452     ret = copy_DigestInitReply(&irep.u.initReply, &digest->initReply);
453     if (ret) {
454 	krb5_set_error_message(context, ret,
455 			       N_("Failed to copy initReply", ""));
456 	goto out;
457     }
458 
459  out:
460     free_DigestRepInner(&irep);
461 
462     return ret;
463 }
464 
465 
466 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
467 krb5_digest_set_client_nonce(krb5_context context,
468 			     krb5_digest digest,
469 			     const char *nonce)
470 {
471     if (digest->request.clientNonce) {
472 	krb5_set_error_message(context, EINVAL,
473 			       N_("clientNonce already set", ""));
474 	return EINVAL;
475     }
476     digest->request.clientNonce =
477 	calloc(1, sizeof(*digest->request.clientNonce));
478     if (digest->request.clientNonce == NULL) {
479 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
480 	return ENOMEM;
481     }
482     *digest->request.clientNonce = strdup(nonce);
483     if (*digest->request.clientNonce == NULL) {
484 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
485 	free(digest->request.clientNonce);
486 	digest->request.clientNonce = NULL;
487 	return ENOMEM;
488     }
489     return 0;
490 }
491 
492 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
493 krb5_digest_set_digest(krb5_context context,
494 		       krb5_digest digest,
495 		       const char *dgst)
496 {
497     if (digest->request.digest) {
498 	krb5_set_error_message(context, EINVAL,
499 			       N_("digest already set", ""));
500 	return EINVAL;
501     }
502     digest->request.digest = strdup(dgst);
503     if (digest->request.digest == NULL) {
504 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
505 	return ENOMEM;
506     }
507     return 0;
508 }
509 
510 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
511 krb5_digest_set_username(krb5_context context,
512 			 krb5_digest digest,
513 			 const char *username)
514 {
515     if (digest->request.username) {
516 	krb5_set_error_message(context, EINVAL, "username already set");
517 	return EINVAL;
518     }
519     digest->request.username = strdup(username);
520     if (digest->request.username == NULL) {
521 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
522 	return ENOMEM;
523     }
524     return 0;
525 }
526 
527 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
528 krb5_digest_set_authid(krb5_context context,
529 		       krb5_digest digest,
530 		       const char *authid)
531 {
532     if (digest->request.authid) {
533 	krb5_set_error_message(context, EINVAL, "authid already set");
534 	return EINVAL;
535     }
536     digest->request.authid = malloc(sizeof(*digest->request.authid));
537     if (digest->request.authid == NULL) {
538 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
539 	return ENOMEM;
540     }
541     *digest->request.authid = strdup(authid);
542     if (*digest->request.authid == NULL) {
543 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
544 	free(digest->request.authid);
545 	digest->request.authid = NULL;
546 	return ENOMEM;
547     }
548     return 0;
549 }
550 
551 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
552 krb5_digest_set_authentication_user(krb5_context context,
553 				    krb5_digest digest,
554 				    krb5_principal authentication_user)
555 {
556     krb5_error_code ret;
557 
558     if (digest->request.authentication_user) {
559 	krb5_set_error_message(context, EINVAL,
560 			       N_("authentication_user already set", ""));
561 	return EINVAL;
562     }
563     ret = krb5_copy_principal(context,
564 			      authentication_user,
565 			      &digest->request.authentication_user);
566     if (ret)
567 	return ret;
568     return 0;
569 }
570 
571 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
572 krb5_digest_set_realm(krb5_context context,
573 		      krb5_digest digest,
574 		      const char *realm)
575 {
576     if (digest->request.realm) {
577 	krb5_set_error_message(context, EINVAL, "realm already set");
578 	return EINVAL;
579     }
580     digest->request.realm = malloc(sizeof(*digest->request.realm));
581     if (digest->request.realm == NULL) {
582 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
583 	return ENOMEM;
584     }
585     *digest->request.realm = strdup(realm);
586     if (*digest->request.realm == NULL) {
587 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
588 	free(digest->request.realm);
589 	digest->request.realm = NULL;
590 	return ENOMEM;
591     }
592     return 0;
593 }
594 
595 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
596 krb5_digest_set_method(krb5_context context,
597 		       krb5_digest digest,
598 		       const char *method)
599 {
600     if (digest->request.method) {
601 	krb5_set_error_message(context, EINVAL,
602 			       N_("method already set", ""));
603 	return EINVAL;
604     }
605     digest->request.method = malloc(sizeof(*digest->request.method));
606     if (digest->request.method == NULL) {
607 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
608 	return ENOMEM;
609     }
610     *digest->request.method = strdup(method);
611     if (*digest->request.method == NULL) {
612 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
613 	free(digest->request.method);
614 	digest->request.method = NULL;
615 	return ENOMEM;
616     }
617     return 0;
618 }
619 
620 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
621 krb5_digest_set_uri(krb5_context context,
622 		    krb5_digest digest,
623 		    const char *uri)
624 {
625     if (digest->request.uri) {
626 	krb5_set_error_message(context, EINVAL, N_("uri already set", ""));
627 	return EINVAL;
628     }
629     digest->request.uri = malloc(sizeof(*digest->request.uri));
630     if (digest->request.uri == NULL) {
631 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
632 	return ENOMEM;
633     }
634     *digest->request.uri = strdup(uri);
635     if (*digest->request.uri == NULL) {
636 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
637 	free(digest->request.uri);
638 	digest->request.uri = NULL;
639 	return ENOMEM;
640     }
641     return 0;
642 }
643 
644 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
645 krb5_digest_set_nonceCount(krb5_context context,
646 			   krb5_digest digest,
647 			   const char *nonce_count)
648 {
649     if (digest->request.nonceCount) {
650 	krb5_set_error_message(context, EINVAL,
651 			       N_("nonceCount already set", ""));
652 	return EINVAL;
653     }
654     digest->request.nonceCount =
655 	malloc(sizeof(*digest->request.nonceCount));
656     if (digest->request.nonceCount == NULL) {
657 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
658 	return ENOMEM;
659     }
660     *digest->request.nonceCount = strdup(nonce_count);
661     if (*digest->request.nonceCount == NULL) {
662 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
663 	free(digest->request.nonceCount);
664 	digest->request.nonceCount = NULL;
665 	return ENOMEM;
666     }
667     return 0;
668 }
669 
670 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
671 krb5_digest_set_qop(krb5_context context,
672 		    krb5_digest digest,
673 		    const char *qop)
674 {
675     if (digest->request.qop) {
676 	krb5_set_error_message(context, EINVAL, "qop already set");
677 	return EINVAL;
678     }
679     digest->request.qop = malloc(sizeof(*digest->request.qop));
680     if (digest->request.qop == NULL) {
681 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
682 	return ENOMEM;
683     }
684     *digest->request.qop = strdup(qop);
685     if (*digest->request.qop == NULL) {
686 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
687 	free(digest->request.qop);
688 	digest->request.qop = NULL;
689 	return ENOMEM;
690     }
691     return 0;
692 }
693 
694 KRB5_LIB_FUNCTION int KRB5_LIB_CALL
695 krb5_digest_set_responseData(krb5_context context,
696 			     krb5_digest digest,
697 			     const char *response)
698 {
699     digest->request.responseData = strdup(response);
700     if (digest->request.responseData == NULL) {
701 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
702 	return ENOMEM;
703     }
704     return 0;
705 }
706 
707 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
708 krb5_digest_request(krb5_context context,
709 		    krb5_digest digest,
710 		    krb5_realm realm,
711 		    krb5_ccache ccache)
712 {
713     DigestReqInner ireq;
714     DigestRepInner irep;
715     krb5_error_code ret;
716 
717     memset(&ireq, 0, sizeof(ireq));
718     memset(&irep, 0, sizeof(irep));
719 
720     ireq.element = choice_DigestReqInner_digestRequest;
721     ireq.u.digestRequest = digest->request;
722 
723     if (digest->request.type == NULL) {
724 	if (digest->init.type == NULL) {
725 	    krb5_set_error_message(context, EINVAL,
726 				   N_("Type missing from req", ""));
727 	    return EINVAL;
728 	}
729 	ireq.u.digestRequest.type = digest->init.type;
730     }
731 
732     if (ireq.u.digestRequest.digest == NULL)
733 	ireq.u.digestRequest.digest = "md5";
734 
735     ret = digest_request(context, realm, ccache,
736 			 KRB5_KU_DIGEST_ENCRYPT, &ireq, &irep);
737     if (ret)
738 	return ret;
739 
740     if (irep.element == choice_DigestRepInner_error) {
741 	ret = irep.u.error.code;
742 	krb5_set_error_message(context, ret,
743 			       N_("Digest response error: %s", ""),
744 			       irep.u.error.reason);
745 	goto out;
746     }
747 
748     if (irep.element != choice_DigestRepInner_response) {
749 	krb5_set_error_message(context, EINVAL,
750 			       N_("digest reply not an DigestResponse", ""));
751 	ret = EINVAL;
752 	goto out;
753     }
754 
755     ret = copy_DigestResponse(&irep.u.response, &digest->response);
756     if (ret) {
757 	krb5_set_error_message(context, ret,
758 			       N_("Failed to copy initReply,", ""));
759 	goto out;
760     }
761 
762  out:
763     free_DigestRepInner(&irep);
764 
765     return ret;
766 }
767 
768 KRB5_LIB_FUNCTION krb5_boolean KRB5_LIB_CALL
769 krb5_digest_rep_get_status(krb5_context context,
770 			   krb5_digest digest)
771 {
772     return digest->response.success ? TRUE : FALSE;
773 }
774 
775 KRB5_LIB_FUNCTION const char * KRB5_LIB_CALL
776 krb5_digest_get_rsp(krb5_context context,
777 		    krb5_digest digest)
778 {
779     if (digest->response.rsp == NULL)
780 	return NULL;
781     return *digest->response.rsp;
782 }
783 
784 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
785 krb5_digest_get_tickets(krb5_context context,
786 			krb5_digest digest,
787 			Ticket **tickets)
788 {
789     *tickets = NULL;
790     return 0;
791 }
792 
793 
794 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
795 krb5_digest_get_client_binding(krb5_context context,
796 			       krb5_digest digest,
797 			       char **type,
798 			       char **binding)
799 {
800     if (digest->response.channel) {
801 	*type = strdup(digest->response.channel->cb_type);
802 	*binding = strdup(digest->response.channel->cb_binding);
803 	if (*type == NULL || *binding == NULL) {
804 	    free(*type);
805 	    free(*binding);
806 	    krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
807 	    return ENOMEM;
808 	}
809     } else {
810 	*type = NULL;
811 	*binding = NULL;
812     }
813     return 0;
814 }
815 
816 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
817 krb5_digest_get_session_key(krb5_context context,
818 			    krb5_digest digest,
819 			    krb5_data *data)
820 {
821     krb5_error_code ret;
822 
823     krb5_data_zero(data);
824     if (digest->response.session_key == NULL)
825 	return 0;
826     ret = der_copy_octet_string(digest->response.session_key, data);
827     if (ret)
828 	krb5_clear_error_message(context);
829 
830     return ret;
831 }
832 
833 struct krb5_ntlm_data {
834     NTLMInit init;
835     NTLMInitReply initReply;
836     NTLMRequest request;
837     NTLMResponse response;
838 };
839 
840 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
841 krb5_ntlm_alloc(krb5_context context,
842 		krb5_ntlm *ntlm)
843 {
844     *ntlm = calloc(1, sizeof(**ntlm));
845     if (*ntlm == NULL) {
846 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
847 	return ENOMEM;
848     }
849     return 0;
850 }
851 
852 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
853 krb5_ntlm_free(krb5_context context, krb5_ntlm ntlm)
854 {
855     free_NTLMInit(&ntlm->init);
856     free_NTLMInitReply(&ntlm->initReply);
857     free_NTLMRequest(&ntlm->request);
858     free_NTLMResponse(&ntlm->response);
859     memset(ntlm, 0, sizeof(*ntlm));
860     free(ntlm);
861     return 0;
862 }
863 
864 
865 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
866 krb5_ntlm_init_request(krb5_context context,
867 		       krb5_ntlm ntlm,
868 		       krb5_realm realm,
869 		       krb5_ccache ccache,
870 		       uint32_t flags,
871 		       const char *hostname,
872 		       const char *domainname)
873 {
874     DigestReqInner ireq;
875     DigestRepInner irep;
876     krb5_error_code ret;
877 
878     memset(&ireq, 0, sizeof(ireq));
879     memset(&irep, 0, sizeof(irep));
880 
881     ntlm->init.flags = flags;
882     if (hostname) {
883 	ALLOC(ntlm->init.hostname, 1);
884 	*ntlm->init.hostname = strdup(hostname);
885     }
886     if (domainname) {
887 	ALLOC(ntlm->init.domain, 1);
888 	*ntlm->init.domain = strdup(domainname);
889     }
890 
891     ireq.element = choice_DigestReqInner_ntlmInit;
892     ireq.u.ntlmInit = ntlm->init;
893 
894     ret = digest_request(context, realm, ccache,
895 			 KRB5_KU_DIGEST_ENCRYPT, &ireq, &irep);
896     if (ret)
897 	goto out;
898 
899     if (irep.element == choice_DigestRepInner_error) {
900 	ret = irep.u.error.code;
901 	krb5_set_error_message(context, ret, N_("Digest init error: %s", ""),
902 			       irep.u.error.reason);
903 	goto out;
904     }
905 
906     if (irep.element != choice_DigestRepInner_ntlmInitReply) {
907 	ret = EINVAL;
908 	krb5_set_error_message(context, ret,
909 			       N_("ntlm reply not an initReply", ""));
910 	goto out;
911     }
912 
913     ret = copy_NTLMInitReply(&irep.u.ntlmInitReply, &ntlm->initReply);
914     if (ret) {
915 	krb5_set_error_message(context, ret,
916 			       N_("Failed to copy initReply", ""));
917 	goto out;
918     }
919 
920  out:
921     free_DigestRepInner(&irep);
922 
923     return ret;
924 }
925 
926 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
927 krb5_ntlm_init_get_flags(krb5_context context,
928 			 krb5_ntlm ntlm,
929 			 uint32_t *flags)
930 {
931     *flags = ntlm->initReply.flags;
932     return 0;
933 }
934 
935 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
936 krb5_ntlm_init_get_challange(krb5_context context,
937 			     krb5_ntlm ntlm,
938 			     krb5_data *challange)
939 {
940     krb5_error_code ret;
941 
942     ret = der_copy_octet_string(&ntlm->initReply.challange, challange);
943     if (ret)
944 	krb5_clear_error_message(context);
945 
946     return ret;
947 }
948 
949 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
950 krb5_ntlm_init_get_opaque(krb5_context context,
951 			  krb5_ntlm ntlm,
952 			  krb5_data *opaque)
953 {
954     krb5_error_code ret;
955 
956     ret = der_copy_octet_string(&ntlm->initReply.opaque, opaque);
957     if (ret)
958 	krb5_clear_error_message(context);
959 
960     return ret;
961 }
962 
963 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
964 krb5_ntlm_init_get_targetname(krb5_context context,
965 			      krb5_ntlm ntlm,
966 			      char **name)
967 {
968     *name = strdup(ntlm->initReply.targetname);
969     if (*name == NULL) {
970 	krb5_clear_error_message(context);
971 	return ENOMEM;
972     }
973     return 0;
974 }
975 
976 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
977 krb5_ntlm_init_get_targetinfo(krb5_context context,
978 			      krb5_ntlm ntlm,
979 			      krb5_data *data)
980 {
981     krb5_error_code ret;
982 
983     if (ntlm->initReply.targetinfo == NULL) {
984 	krb5_data_zero(data);
985 	return 0;
986     }
987 
988     ret = krb5_data_copy(data,
989 			 ntlm->initReply.targetinfo->data,
990 			 ntlm->initReply.targetinfo->length);
991     if (ret) {
992 	krb5_clear_error_message(context);
993 	return ret;
994     }
995     return 0;
996 }
997 
998 
999 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
1000 krb5_ntlm_request(krb5_context context,
1001 		  krb5_ntlm ntlm,
1002 		  krb5_realm realm,
1003 		  krb5_ccache ccache)
1004 {
1005     DigestReqInner ireq;
1006     DigestRepInner irep;
1007     krb5_error_code ret;
1008 
1009     memset(&ireq, 0, sizeof(ireq));
1010     memset(&irep, 0, sizeof(irep));
1011 
1012     ireq.element = choice_DigestReqInner_ntlmRequest;
1013     ireq.u.ntlmRequest = ntlm->request;
1014 
1015     ret = digest_request(context, realm, ccache,
1016 			 KRB5_KU_DIGEST_ENCRYPT, &ireq, &irep);
1017     if (ret)
1018 	return ret;
1019 
1020     if (irep.element == choice_DigestRepInner_error) {
1021 	ret = irep.u.error.code;
1022 	krb5_set_error_message(context, ret,
1023 			       N_("NTLM response error: %s", ""),
1024 			       irep.u.error.reason);
1025 	goto out;
1026     }
1027 
1028     if (irep.element != choice_DigestRepInner_ntlmResponse) {
1029 	ret = EINVAL;
1030 	krb5_set_error_message(context, ret,
1031 			       N_("NTLM reply not an NTLMResponse", ""));
1032 	goto out;
1033     }
1034 
1035     ret = copy_NTLMResponse(&irep.u.ntlmResponse, &ntlm->response);
1036     if (ret) {
1037 	krb5_set_error_message(context, ret,
1038 			       N_("Failed to copy NTLMResponse", ""));
1039 	goto out;
1040     }
1041 
1042  out:
1043     free_DigestRepInner(&irep);
1044 
1045     return ret;
1046 }
1047 
1048 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
1049 krb5_ntlm_req_set_flags(krb5_context context,
1050 			krb5_ntlm ntlm,
1051 			uint32_t flags)
1052 {
1053     ntlm->request.flags = flags;
1054     return 0;
1055 }
1056 
1057 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
1058 krb5_ntlm_req_set_username(krb5_context context,
1059 			   krb5_ntlm ntlm,
1060 			   const char *username)
1061 {
1062     ntlm->request.username = strdup(username);
1063     if (ntlm->request.username == NULL) {
1064 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
1065 	return ENOMEM;
1066     }
1067     return 0;
1068 }
1069 
1070 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
1071 krb5_ntlm_req_set_targetname(krb5_context context,
1072 			     krb5_ntlm ntlm,
1073 			     const char *targetname)
1074 {
1075     ntlm->request.targetname = strdup(targetname);
1076     if (ntlm->request.targetname == NULL) {
1077 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
1078 	return ENOMEM;
1079     }
1080     return 0;
1081 }
1082 
1083 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
1084 krb5_ntlm_req_set_lm(krb5_context context,
1085 		     krb5_ntlm ntlm,
1086 		     void *hash, size_t len)
1087 {
1088     ntlm->request.lm.data = malloc(len);
1089     if (ntlm->request.lm.data == NULL && len != 0) {
1090 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
1091 	return ENOMEM;
1092     }
1093     ntlm->request.lm.length = len;
1094     memcpy(ntlm->request.lm.data, hash, len);
1095     return 0;
1096 }
1097 
1098 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
1099 krb5_ntlm_req_set_ntlm(krb5_context context,
1100 		       krb5_ntlm ntlm,
1101 		       void *hash, size_t len)
1102 {
1103     ntlm->request.ntlm.data = malloc(len);
1104     if (ntlm->request.ntlm.data == NULL && len != 0) {
1105 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
1106 	return ENOMEM;
1107     }
1108     ntlm->request.ntlm.length = len;
1109     memcpy(ntlm->request.ntlm.data, hash, len);
1110     return 0;
1111 }
1112 
1113 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
1114 krb5_ntlm_req_set_opaque(krb5_context context,
1115 			 krb5_ntlm ntlm,
1116 			 krb5_data *opaque)
1117 {
1118     ntlm->request.opaque.data = malloc(opaque->length);
1119     if (ntlm->request.opaque.data == NULL && opaque->length != 0) {
1120 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
1121 	return ENOMEM;
1122     }
1123     ntlm->request.opaque.length = opaque->length;
1124     memcpy(ntlm->request.opaque.data, opaque->data, opaque->length);
1125     return 0;
1126 }
1127 
1128 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
1129 krb5_ntlm_req_set_session(krb5_context context,
1130 			  krb5_ntlm ntlm,
1131 			  void *sessionkey, size_t length)
1132 {
1133     ntlm->request.sessionkey = calloc(1, sizeof(*ntlm->request.sessionkey));
1134     if (ntlm->request.sessionkey == NULL) {
1135 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
1136 	return ENOMEM;
1137     }
1138     ntlm->request.sessionkey->data = malloc(length);
1139     if (ntlm->request.sessionkey->data == NULL && length != 0) {
1140 	krb5_set_error_message(context, ENOMEM, N_("malloc: out of memory", ""));
1141 	return ENOMEM;
1142     }
1143     memcpy(ntlm->request.sessionkey->data, sessionkey, length);
1144     ntlm->request.sessionkey->length = length;
1145     return 0;
1146 }
1147 
1148 KRB5_LIB_FUNCTION krb5_boolean KRB5_LIB_CALL
1149 krb5_ntlm_rep_get_status(krb5_context context,
1150 			 krb5_ntlm ntlm)
1151 {
1152     return ntlm->response.success ? TRUE : FALSE;
1153 }
1154 
1155 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
1156 krb5_ntlm_rep_get_sessionkey(krb5_context context,
1157 			     krb5_ntlm ntlm,
1158 			     krb5_data *data)
1159 {
1160     if (ntlm->response.sessionkey == NULL) {
1161 	krb5_set_error_message(context, EINVAL,
1162 			       N_("no ntlm session key", ""));
1163 	return EINVAL;
1164     }
1165     krb5_clear_error_message(context);
1166     return krb5_data_copy(data,
1167 			  ntlm->response.sessionkey->data,
1168 			  ntlm->response.sessionkey->length);
1169 }
1170 
1171 /**
1172  * Get the supported/allowed mechanism for this principal.
1173  *
1174  * @param context A Keberos context.
1175  * @param realm The realm of the KDC.
1176  * @param ccache The credential cache to use when talking to the KDC.
1177  * @param flags The supported mechanism.
1178  *
1179  * @return Return an error code or 0.
1180  *
1181  * @ingroup krb5_digest
1182  */
1183 
1184 KRB5_LIB_FUNCTION krb5_error_code KRB5_LIB_CALL
1185 krb5_digest_probe(krb5_context context,
1186 		  krb5_realm realm,
1187 		  krb5_ccache ccache,
1188 		  unsigned *flags)
1189 {
1190     DigestReqInner ireq;
1191     DigestRepInner irep;
1192     krb5_error_code ret;
1193 
1194     memset(&ireq, 0, sizeof(ireq));
1195     memset(&irep, 0, sizeof(irep));
1196 
1197     ireq.element = choice_DigestReqInner_supportedMechs;
1198 
1199     ret = digest_request(context, realm, ccache,
1200 			 KRB5_KU_DIGEST_ENCRYPT, &ireq, &irep);
1201     if (ret)
1202 	goto out;
1203 
1204     if (irep.element == choice_DigestRepInner_error) {
1205 	ret = irep.u.error.code;
1206 	krb5_set_error_message(context, ret, "Digest probe error: %s",
1207 			       irep.u.error.reason);
1208 	goto out;
1209     }
1210 
1211     if (irep.element != choice_DigestRepInner_supportedMechs) {
1212 	ret = EINVAL;
1213 	krb5_set_error_message(context, ret, "Digest reply not an probe");
1214 	goto out;
1215     }
1216 
1217     *flags = DigestTypes2int(irep.u.supportedMechs);
1218 
1219  out:
1220     free_DigestRepInner(&irep);
1221 
1222     return ret;
1223 }
1224 
1225 #endif /* HEIMDAL_SMALLER */
1226