Mem alloc checks.
[abartlet/samba.git/.git] / source3 / libads / krb5_setpw.c
1 /* 
2    Unix SMB/CIFS implementation.
3    krb5 set password implementation
4    Copyright (C) Andrew Tridgell 2001
5    Copyright (C) Remus Koos 2001 (remuskoos@yahoo.com)
6    
7    This program is free software; you can redistribute it and/or modify
8    it under the terms of the GNU General Public License as published by
9    the Free Software Foundation; either version 2 of the License, or
10    (at your option) any later version.
11    
12    This program is distributed in the hope that it will be useful,
13    but WITHOUT ANY WARRANTY; without even the implied warranty of
14    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15    GNU General Public License for more details.
16    
17    You should have received a copy of the GNU General Public License
18    along with this program; if not, write to the Free Software
19    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
20 */
21
22 #include "includes.h"
23
24 #ifdef HAVE_KRB5
25
26 #define DEFAULT_KPASSWD_PORT    464
27 #define KRB5_KPASSWD_VERS_CHANGEPW      1
28 #define KRB5_KPASSWD_VERS_SETPW         0xff80
29 #define KRB5_KPASSWD_ACCESSDENIED       5
30 #define KRB5_KPASSWD_BAD_VERSION        6
31
32 /* This implements the Kerb password change protocol as specifed in
33  * kerb-chg-password-02.txt
34  */
35 static DATA_BLOB encode_krb5_setpw(const char *principal, const char *password)
36 {
37         char* princ_part1 = NULL;
38         char* princ_part2 = NULL;
39         char* realm = NULL;
40         char* c;
41         char* princ;
42
43         ASN1_DATA req;
44         DATA_BLOB ret;
45
46
47         princ = strdup(principal);
48
49         if ((c = strchr(princ, '/')) == NULL) {
50             c = princ; 
51         } else {
52             *c = '\0';
53             c++;
54             princ_part1 = princ;
55         }
56
57         princ_part2 = c;
58
59         if ((c = strchr(c, '@')) != NULL) {
60             *c = '\0';
61             c++;
62             realm = c;
63         }
64
65         memset(&req, 0, sizeof(req));
66         
67         asn1_push_tag(&req, ASN1_SEQUENCE(0));
68         asn1_push_tag(&req, ASN1_CONTEXT(0));
69         asn1_write_OctetString(&req, password, strlen(password));
70         asn1_pop_tag(&req);
71
72         asn1_push_tag(&req, ASN1_CONTEXT(1));
73         asn1_push_tag(&req, ASN1_SEQUENCE(0));
74
75         asn1_push_tag(&req, ASN1_CONTEXT(0));
76         asn1_write_Integer(&req, 1);
77         asn1_pop_tag(&req);
78
79         asn1_push_tag(&req, ASN1_CONTEXT(1));
80         asn1_push_tag(&req, ASN1_SEQUENCE(0));
81
82         if (princ_part1) 
83             asn1_write_GeneralString(&req, princ_part1);
84         
85         asn1_write_GeneralString(&req, princ_part2);
86         asn1_pop_tag(&req);
87         asn1_pop_tag(&req);
88         asn1_pop_tag(&req);
89         asn1_pop_tag(&req);
90
91         asn1_push_tag(&req, ASN1_CONTEXT(2));
92         asn1_write_GeneralString(&req, realm);
93         asn1_pop_tag(&req);
94         asn1_pop_tag(&req);
95
96         ret = data_blob(req.data, req.length);
97         asn1_free(&req);
98
99         free(princ);
100
101         return ret;
102 }       
103
104 static krb5_error_code build_setpw_request(krb5_context context,
105                                            krb5_auth_context auth_context,
106                                            krb5_data *ap_req,
107                                            const char *princ,
108                                            const char *passwd,
109                                            krb5_data *packet)
110 {
111         krb5_error_code ret;
112         krb5_data cipherpw;
113         krb5_data encoded_setpw;
114         krb5_replay_data replay;
115         char *p;
116         DATA_BLOB setpw;
117
118         ret = krb5_auth_con_setflags(context,
119                                      auth_context,KRB5_AUTH_CONTEXT_DO_SEQUENCE);
120         if (ret) {
121                 DEBUG(1,("krb5_auth_con_setflags failed (%s)\n",
122                          error_message(ret)));
123                 return ret;
124         }
125
126         setpw = encode_krb5_setpw(princ, passwd);
127
128         encoded_setpw.data = setpw.data;
129         encoded_setpw.length = setpw.length;
130
131         ret = krb5_mk_priv(context, auth_context,
132                            &encoded_setpw, &cipherpw, &replay);
133         
134         data_blob_free(&setpw);         /*from 'encode_krb5_setpw(...)' */
135         
136         if (ret) {
137                 DEBUG(1,("krb5_mk_priv failed (%s)\n", error_message(ret)));
138                 return ret;
139         }
140
141         packet->data = (char *)malloc(ap_req->length + cipherpw.length + 6);
142         if (!packet->data)
143                 return -1;
144
145         /* see the RFC for details */
146         p = ((char *)packet->data) + 2;
147         RSSVAL(p, 0, 0xff80);
148         p += 2;
149         RSSVAL(p, 0, ap_req->length);
150         p += 2;
151         memcpy(p, ap_req->data, ap_req->length);
152         p += ap_req->length;
153         memcpy(p, cipherpw.data, cipherpw.length);
154         p += cipherpw.length;
155         packet->length = PTR_DIFF(p,packet->data);
156         RSSVAL(packet->data, 0, packet->length);
157         
158         free(cipherpw.data);    /* from 'krb5_mk_priv(...)' */
159
160         return 0;
161 }
162
163 static krb5_error_code parse_setpw_reply(krb5_context context, 
164                                          krb5_auth_context auth_context,
165                                          krb5_data *packet)
166 {
167         krb5_data ap_rep;
168         char *p;
169         int vnum, ret, res_code;
170         krb5_data cipherresult;
171         krb5_data clearresult;
172         krb5_ap_rep_enc_part *ap_rep_enc;
173         krb5_replay_data replay;
174         
175         if (packet->length < 4) {
176                 return KRB5KRB_AP_ERR_MODIFIED;
177         }
178         
179         p = packet->data;
180         
181         if (((char *)packet->data)[0] == 0x7e || ((char *)packet->data)[0] == 0x5e) {
182                 /* it's an error packet. We should parse it ... */
183                 DEBUG(1,("Got error packet 0x%x from kpasswd server\n",
184                          ((char *)packet->data)[0]));
185                 return KRB5KRB_AP_ERR_MODIFIED;
186         }
187         
188         if (RSVAL(p, 0) != packet->length) {
189                 DEBUG(1,("Bad packet length (%d/%d) from kpasswd server\n",
190                          RSVAL(p, 0), packet->length));
191                 return KRB5KRB_AP_ERR_MODIFIED;
192         }
193
194         p += 2;
195
196         vnum = RSVAL(p, 0); p += 2;
197         
198         if (vnum != KRB5_KPASSWD_VERS_SETPW && vnum != KRB5_KPASSWD_VERS_CHANGEPW) {
199                 DEBUG(1,("Bad vnum (%d) from kpasswd server\n", vnum));
200                 return KRB5KDC_ERR_BAD_PVNO;
201         }
202         
203         ap_rep.length = RSVAL(p, 0); p += 2;
204         
205         if (p + ap_rep.length >= (char *)packet->data + packet->length) {
206                 DEBUG(1,("ptr beyond end of packet from kpasswd server\n"));
207                 return KRB5KRB_AP_ERR_MODIFIED;
208         }
209         
210         if (ap_rep.length == 0) {
211                 DEBUG(1,("got unencrypted setpw result?!\n"));
212                 return KRB5KRB_AP_ERR_MODIFIED;
213         }
214
215         /* verify ap_rep */
216         ap_rep.data = p;
217         p += ap_rep.length;
218         
219         ret = krb5_rd_rep(context, auth_context, &ap_rep, &ap_rep_enc);
220         if (ret) {
221                 DEBUG(1,("failed to rd setpw reply (%s)\n", error_message(ret)));
222                 return KRB5KRB_AP_ERR_MODIFIED;
223         }
224         
225         krb5_free_ap_rep_enc_part(context, ap_rep_enc);
226         
227         cipherresult.data = p;
228         cipherresult.length = ((char *)packet->data + packet->length) - p;
229                 
230         ret = krb5_rd_priv(context, auth_context, &cipherresult, &clearresult,
231                            &replay);
232         if (ret) {
233                 DEBUG(1,("failed to decrypt setpw reply (%s)\n", error_message(ret)));
234                 return KRB5KRB_AP_ERR_MODIFIED;
235         }
236
237         if (clearresult.length < 2) {
238                 free(clearresult.data);
239                 ret = KRB5KRB_AP_ERR_MODIFIED;
240                 return KRB5KRB_AP_ERR_MODIFIED;
241         }
242         
243         p = clearresult.data;
244         
245         res_code = RSVAL(p, 0);
246         
247         free(clearresult.data);
248
249         if ((res_code < KRB5_KPASSWD_SUCCESS) || 
250             (res_code >= KRB5_KPASSWD_ACCESSDENIED)) {
251                 return KRB5KRB_AP_ERR_MODIFIED;
252         }
253         
254         return 0;
255 }
256
257 ADS_STATUS krb5_set_password(const char *kdc_host, const char *princ, const char *newpw, 
258                              int time_offset)
259 {
260         krb5_context context;
261         krb5_auth_context auth_context = NULL;
262         krb5_principal principal;
263         char *princ_name;
264         char *realm;
265         krb5_creds creds, *credsp;
266         krb5_ccache ccache;
267         krb5_data ap_req, chpw_req, chpw_rep;
268         int ret, sock, addr_len;
269         struct sockaddr remote_addr, local_addr;
270         krb5_address local_kaddr, remote_kaddr;
271
272         ret = krb5_init_context(&context);
273         if (ret) {
274                 DEBUG(1,("Failed to init krb5 context (%s)\n", error_message(ret)));
275                 return ADS_ERROR_KRB5(ret);
276         }
277         
278         if (time_offset != 0) {
279                 krb5_set_real_time(context, time(NULL) + time_offset, 0);
280         }
281
282         ret = krb5_cc_default(context, &ccache);
283         if (ret) {
284                 krb5_free_context(context);
285                 DEBUG(1,("Failed to get default creds (%s)\n", error_message(ret)));
286                 return ADS_ERROR_KRB5(ret);
287         }
288
289         ZERO_STRUCT(creds);
290         
291         realm = strchr(princ, '@');
292         realm++;
293
294         asprintf(&princ_name, "kadmin/changepw@%s", realm);
295         ret = krb5_parse_name(context, princ_name, &creds.server);
296         if (ret) {
297                 krb5_free_context(context);
298                 DEBUG(1,("Failed to parse kadmin/changepw (%s)\n", error_message(ret)));
299                 return ADS_ERROR_KRB5(ret);
300         }
301         free(princ_name);
302
303         /* parse the principal we got as a function argument */
304         ret = krb5_parse_name(context, princ, &principal);
305         if (ret) {
306                 krb5_free_context(context);
307                 DEBUG(1,("Failed to parse %s (%s)\n", princ_name, error_message(ret)));
308                 return ADS_ERROR_KRB5(ret);
309         }
310
311         krb5_princ_set_realm(context, creds.server,
312                              krb5_princ_realm(context, principal));
313         
314         ret = krb5_cc_get_principal(context, ccache, &creds.client);
315         if (ret) {
316                 krb5_free_principal(context, principal);
317                 krb5_free_context(context);
318                 DEBUG(1,("Failed to get principal from ccache (%s)\n", 
319                          error_message(ret)));
320                 return ADS_ERROR_KRB5(ret);
321         }
322         
323         ret = krb5_get_credentials(context, 0, ccache, &creds, &credsp);
324         if (ret) {
325                 krb5_free_principal(context, creds.client);
326                 krb5_free_principal(context, principal);
327                 krb5_free_context(context);
328                 DEBUG(1,("krb5_get_credentials failed (%s)\n", error_message(ret)));
329                 return ADS_ERROR_KRB5(ret);
330         }
331         
332         /* we might have to call krb5_free_creds(...) from now on ... */
333         ret = krb5_mk_req_extended(context, &auth_context, AP_OPTS_USE_SUBKEY,
334                                    NULL, credsp, &ap_req);
335         if (ret) {
336                 krb5_free_creds(context, credsp);
337                 krb5_free_principal(context, creds.client);
338                 krb5_free_principal(context, principal);
339                 krb5_free_context(context);
340                 DEBUG(1,("krb5_mk_req_extended failed (%s)\n", error_message(ret)));
341                 return ADS_ERROR_KRB5(ret);
342         }
343         
344         sock = open_udp_socket(kdc_host, DEFAULT_KPASSWD_PORT);
345         if (sock == -1) {
346                 int rc = errno;
347                 free(ap_req.data);
348                 krb5_free_creds(context, credsp);
349                 krb5_free_principal(context, creds.client);
350                 krb5_free_principal(context, principal);
351                 krb5_free_context(context);
352                 DEBUG(1,("failed to open kpasswd socket to %s (%s)\n", 
353                          kdc_host, strerror(errno)));
354                 return ADS_ERROR_SYSTEM(rc);
355         }
356         
357         addr_len = sizeof(remote_addr);
358         getpeername(sock, &remote_addr, &addr_len);
359         addr_len = sizeof(local_addr);
360         getsockname(sock, &local_addr, &addr_len);
361         
362         setup_kaddr(&remote_kaddr, &remote_addr);
363         setup_kaddr(&local_kaddr, &local_addr);
364
365         ret = krb5_auth_con_setaddrs(context, auth_context, &local_kaddr, NULL);
366         if (ret) {
367                 close(sock);
368                 free(ap_req.data);
369                 krb5_free_creds(context, credsp);
370                 krb5_free_principal(context, creds.client);
371                 krb5_free_principal(context, principal);
372                 krb5_free_context(context);
373                 DEBUG(1,("krb5_auth_con_setaddrs failed (%s)\n", error_message(ret)));
374                 return ADS_ERROR_KRB5(ret);
375         }
376
377         ret = build_setpw_request(context, auth_context, &ap_req,
378                                   princ, newpw, &chpw_req);
379         if (ret) {
380                 close(sock);
381                 free(ap_req.data);
382                 krb5_free_creds(context, credsp);
383                 krb5_free_principal(context, creds.client);
384                 krb5_free_principal(context, principal);
385                 krb5_free_context(context);
386                 DEBUG(1,("build_setpw_request failed (%s)\n", error_message(ret)));
387                 return ADS_ERROR_KRB5(ret);
388         }
389
390         if (write(sock, chpw_req.data, chpw_req.length) != chpw_req.length) {
391                 close(sock);
392                 free(chpw_req.data);
393                 free(ap_req.data);
394                 krb5_free_creds(context, credsp);
395                 krb5_free_principal(context, creds.client);
396                 krb5_free_principal(context, principal);
397                 krb5_free_context(context);
398                 DEBUG(1,("send of chpw failed (%s)\n", strerror(errno)));
399                 return ADS_ERROR_SYSTEM(errno);
400         }
401
402         free(chpw_req.data);
403
404         chpw_rep.length = 1500;
405         chpw_rep.data = (char *) malloc(chpw_rep.length);
406         if (!chpw_rep.data) {
407                 close(sock);
408                 free(ap_req.data);
409                 krb5_free_creds(context, credsp);
410                 krb5_free_principal(context, creds.client);
411                 krb5_free_principal(context, principal);
412                 krb5_free_context(context);
413                 DEBUG(1,("send of chpw failed (%s)\n", strerror(errno)));
414                 errno = ENOMEM;
415                 return ADS_ERROR_SYSTEM(errno);
416         }
417
418         ret = read(sock, chpw_rep.data, chpw_rep.length);
419         if (ret < 0) {
420                 close(sock);
421                 free(chpw_rep.data);
422                 free(ap_req.data);
423                 krb5_free_creds(context, credsp);
424                 krb5_free_principal(context, creds.client);
425                 krb5_free_principal(context, principal);
426                 krb5_free_context(context);
427                 DEBUG(1,("recv of chpw reply failed (%s)\n", strerror(errno)));
428                 return ADS_ERROR_SYSTEM(errno);
429         }
430
431         close(sock);
432         chpw_rep.length = ret;
433
434         ret = krb5_auth_con_setaddrs(context, auth_context, NULL,&remote_kaddr);
435         if (ret) {
436                 free(chpw_rep.data);
437                 free(ap_req.data);
438                 krb5_free_creds(context, credsp);
439                 krb5_free_principal(context, creds.client);
440                 krb5_free_principal(context, principal);
441                 krb5_free_context(context);
442                 DEBUG(1,("krb5_auth_con_setaddrs on reply failed (%s)\n", 
443                          error_message(ret)));
444                 return ADS_ERROR_KRB5(ret);
445         }
446
447         ret = parse_setpw_reply(context, auth_context, &chpw_rep);
448         free(chpw_rep.data);
449
450         if (ret) {
451                 free(ap_req.data);
452                 krb5_free_creds(context, credsp);
453                 krb5_free_principal(context, creds.client);
454                 krb5_free_principal(context, principal);
455                 krb5_free_context(context);
456                 DEBUG(1,("parse_setpw_reply failed (%s)\n", 
457                          error_message(ret)));
458                 return ADS_ERROR_KRB5(ret);
459         }
460
461         free(ap_req.data);
462         krb5_free_creds(context, credsp);
463         krb5_free_principal(context, creds.client);
464         krb5_free_principal(context, principal);
465         krb5_free_context(context);
466
467         return ADS_SUCCESS;
468 }
469
470
471 ADS_STATUS kerberos_set_password(const char *kpasswd_server, 
472                                  const char *auth_principal, const char *auth_password,
473                                  const char *target_principal, const char *new_password,
474                                  int time_offset)
475 {
476     int ret;
477
478     if ((ret = kerberos_kinit_password(auth_principal, auth_password, time_offset))) {
479         DEBUG(1,("Failed kinit for principal %s (%s)\n", auth_principal, error_message(ret)));
480         return ADS_ERROR_KRB5(ret);
481     }
482
483     return krb5_set_password(kpasswd_server, target_principal, new_password, time_offset);
484 }
485
486
487 /**
488  * Set the machine account password
489  * @param ads connection to ads server
490  * @param hostname machine whose password is being set
491  * @param password new password
492  * @return status of password change
493  **/
494 ADS_STATUS ads_set_machine_password(ADS_STRUCT *ads,
495                                     const char *hostname, 
496                                     const char *password)
497 {
498         ADS_STATUS status;
499         char *host = strdup(hostname);
500         char *principal; 
501
502         strlower(host);
503
504         /*
505           we need to use the '$' form of the name here, as otherwise the
506           server might end up setting the password for a user instead
507          */
508         asprintf(&principal, "%s$@%s", host, ads->auth.realm);
509         
510         status = krb5_set_password(ads->auth.kdc_server, principal, password, ads->auth.time_offset);
511         
512         free(host);
513         free(principal);
514
515         return status;
516 }
517
518 #endif