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