kdc: we don't need any *_locl.h header from heimdal in the kdc
[ira/wip.git] / source4 / kdc / pac-glue.c
1 /* 
2    Unix SMB/CIFS implementation.
3
4    PAC Glue between Samba and the KDC
5    
6    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2005
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 3 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    
19    You should have received a copy of the GNU General Public License
20    along with this program.  If not, see <http://www.gnu.org/licenses/>.
21 */
22
23 #include "includes.h"
24 #include "dsdb/common/flags.h"
25 #include "lib/ldb/include/ldb.h"
26 #include "librpc/gen_ndr/ndr_krb5pac.h"
27 #include "librpc/gen_ndr/krb5pac.h"
28 #include "auth/auth.h"
29 #include "auth/auth_sam.h"
30 #include "auth/auth_sam_reply.h"
31 #include "param/param.h"
32 #include "kdc/kdc.h"
33
34 struct krb5_dh_moduli;
35 struct _krb5_krb_auth_data;
36
37 krb5_error_code samba_kdc_plugin_init(krb5_context context, void **ptr) 
38 {
39         *ptr = NULL;
40         return 0;
41 }
42
43 void    samba_kdc_plugin_fini(void *ptr) 
44 {
45         return;
46 }
47
48 static krb5_error_code make_pac(krb5_context context,
49                                 TALLOC_CTX *mem_ctx, 
50                                 struct smb_iconv_convenience *iconv_convenience,
51                                 struct auth_serversupplied_info *server_info,
52                                 krb5_pac *pac) 
53 {
54         struct PAC_LOGON_INFO_CTR logon_info;
55         struct netr_SamInfo3 *info3;
56         krb5_data pac_data;
57         NTSTATUS nt_status;
58         enum ndr_err_code ndr_err;
59         DATA_BLOB pac_out;
60         krb5_error_code ret;
61
62         ZERO_STRUCT(logon_info);
63
64         nt_status = auth_convert_server_info_saminfo3(mem_ctx, server_info, &info3);
65         if (!NT_STATUS_IS_OK(nt_status)) {
66                 DEBUG(1, ("Getting Samba info failed: %s\n", nt_errstr(nt_status)));
67                 return EINVAL;
68         }
69
70         logon_info.info = talloc_zero(mem_ctx, struct PAC_LOGON_INFO);
71         if (!mem_ctx) {
72                 return ENOMEM;
73         }
74
75         logon_info.info->info3 = *info3;
76
77         ndr_err = ndr_push_struct_blob(&pac_out, mem_ctx, iconv_convenience, &logon_info,
78                                        (ndr_push_flags_fn_t)ndr_push_PAC_LOGON_INFO_CTR);
79         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
80                 nt_status = ndr_map_error2ntstatus(ndr_err);
81                 DEBUG(1, ("PAC (presig) push failed: %s\n", nt_errstr(nt_status)));
82                 return EINVAL;
83         }
84
85         ret = krb5_data_copy(&pac_data, pac_out.data, pac_out.length);
86         if (ret != 0) {
87                 return ret;
88         }
89
90         ret = krb5_pac_init(context, pac);
91         if (ret != 0) {
92                 krb5_data_free(&pac_data);
93                 return ret;
94         }
95
96         ret = krb5_pac_add_buffer(context, *pac, PAC_TYPE_LOGON_INFO, &pac_data);
97         krb5_data_free(&pac_data);
98         if (ret != 0) {
99                 return ret;
100         }
101
102         return ret;
103 }
104
105 /* Given the right private pointer from hdb_ldb, get a PAC from the attached ldb messages */
106 krb5_error_code samba_kdc_get_pac(void *priv,
107                                   krb5_context context, 
108                                   struct hdb_entry_ex *client,
109                                   krb5_pac *pac)
110 {
111         krb5_error_code ret;
112         NTSTATUS nt_status;
113         struct auth_serversupplied_info *server_info;
114         struct hdb_ldb_private *private = talloc_get_type(client->ctx, struct hdb_ldb_private);
115         TALLOC_CTX *mem_ctx = talloc_named(private, 0, "samba_get_pac context");
116         unsigned int userAccountControl;
117
118         if (!mem_ctx) {
119                 return ENOMEM;
120         }
121
122         /* The user account may be set not to want the PAC */
123         userAccountControl = ldb_msg_find_attr_as_uint(private->msg, "userAccountControl", 0);
124         if (userAccountControl & UF_NO_AUTH_DATA_REQUIRED) {
125                 *pac = NULL;
126                 return 0;
127         }
128
129         nt_status = authsam_make_server_info(mem_ctx, private->samdb, 
130                                              private->netbios_name,
131                                              private->msg, 
132                                              private->realm_ref_msg,
133                                              data_blob(NULL, 0),
134                                              data_blob(NULL, 0),
135                                              &server_info);
136         if (!NT_STATUS_IS_OK(nt_status)) {
137                 DEBUG(0, ("Getting user info for PAC failed: %s\n",
138                           nt_errstr(nt_status)));
139                 return ENOMEM;
140         }
141
142         ret = make_pac(context, mem_ctx, private->iconv_convenience, server_info, pac);
143
144         talloc_free(mem_ctx);
145         return ret;
146 }
147
148 /* Resign (and reform, including possibly new groups) a PAC */
149
150 krb5_error_code samba_kdc_reget_pac(void *priv, krb5_context context,
151                                 const krb5_principal client_principal,
152                                 struct hdb_entry_ex *client,  
153                                 struct hdb_entry_ex *server, krb5_pac *pac)
154 {
155         NTSTATUS nt_status;
156         enum ndr_err_code ndr_err;
157         krb5_error_code ret;
158
159         unsigned int userAccountControl;
160
161         struct hdb_ldb_private *private = talloc_get_type(server->ctx, struct hdb_ldb_private);
162         krb5_data k5pac_in;
163         DATA_BLOB pac_in;
164
165         struct PAC_LOGON_INFO_CTR logon_info;
166         union netr_Validation validation;
167         struct auth_serversupplied_info *server_info_out;
168
169         TALLOC_CTX *mem_ctx = talloc_named(private, 0, "samba_get_pac context");
170         
171         if (!mem_ctx) {
172                 return ENOMEM;
173         }
174
175         /* The service account may be set not to want the PAC */
176         userAccountControl = ldb_msg_find_attr_as_uint(private->msg, "userAccountControl", 0);
177         if (userAccountControl & UF_NO_AUTH_DATA_REQUIRED) {
178                 *pac = NULL;
179                 return 0;
180         }
181
182         ret = krb5_pac_get_buffer(context, *pac, PAC_TYPE_LOGON_INFO, &k5pac_in);
183         if (ret != 0) {
184                 return ret;
185         }
186
187         pac_in = data_blob_talloc(mem_ctx, k5pac_in.data, k5pac_in.length);
188         krb5_data_free(&k5pac_in);
189         if (!pac_in.data) {
190                 talloc_free(mem_ctx);
191                 return ENOMEM;
192         }
193                 
194         ndr_err = ndr_pull_struct_blob(&pac_in, mem_ctx, private->iconv_convenience, &logon_info,
195                                        (ndr_pull_flags_fn_t)ndr_pull_PAC_LOGON_INFO_CTR);
196         if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err) || !logon_info.info) {
197                 nt_status = ndr_map_error2ntstatus(ndr_err);
198                 DEBUG(0,("can't parse the PAC LOGON_INFO: %s\n", nt_errstr(nt_status)));
199                 talloc_free(mem_ctx);
200                 return EINVAL;
201         }
202
203         /* Pull this right into the normal auth sysstem structures */
204         validation.sam3 = &logon_info.info->info3;
205         nt_status = make_server_info_netlogon_validation(mem_ctx,
206                                                          "",
207                                                          3, &validation,
208                                                          &server_info_out); 
209         if (!NT_STATUS_IS_OK(nt_status)) {
210                 talloc_free(mem_ctx);
211                 return ENOMEM;
212         }
213
214         /* We will compleatly regenerate this pac */
215         krb5_pac_free(context, *pac);
216
217         ret = make_pac(context, mem_ctx, private->iconv_convenience, server_info_out, pac);
218
219         talloc_free(mem_ctx);
220         return ret;
221 }
222
223 static void samba_kdc_build_edata_reply(TALLOC_CTX *tmp_ctx, krb5_data *e_data,
224                                        NTSTATUS nt_status)
225 {
226         PA_DATA pa;
227         unsigned char *buf;
228         size_t len;
229         krb5_error_code ret = 0;
230
231         if (!e_data)
232                 return;
233
234         pa.padata_type          = KRB5_PADATA_PW_SALT;
235         pa.padata_value.length  = 12;
236         pa.padata_value.data    = malloc(pa.padata_value.length);
237         if (!pa.padata_value.data) {
238                 e_data->length = 0;
239                 e_data->data = NULL;
240                 return;
241         }
242
243         SIVAL(pa.padata_value.data, 0, NT_STATUS_V(nt_status));
244         SIVAL(pa.padata_value.data, 4, 0);
245         SIVAL(pa.padata_value.data, 8, 1);
246
247         ASN1_MALLOC_ENCODE(PA_DATA, buf, len, &pa, &len, ret);
248         free(pa.padata_value.data);
249
250         e_data->data   = buf;
251         e_data->length = len;
252
253         return;
254 }
255
256 /* Given an hdb entry (and in particular it's private member), consult
257  * the account_ok routine in auth/auth_sam.c for consistancy */
258
259
260 krb5_error_code samba_kdc_check_client_access(void *priv, 
261                                               krb5_context context, hdb_entry_ex *entry_ex, 
262                                               KDC_REQ *req,
263                                               krb5_data *e_data)
264 {
265         krb5_error_code ret;
266         NTSTATUS nt_status;
267         TALLOC_CTX *tmp_ctx = talloc_new(entry_ex->ctx);
268         struct hdb_ldb_private *private = talloc_get_type(entry_ex->ctx, struct hdb_ldb_private);
269         char *name, *workstation = NULL;
270         HostAddresses *addresses = req->req_body.addresses;
271         int i;
272
273         if (!tmp_ctx) {
274                 return ENOMEM;
275         }
276         
277         ret = krb5_unparse_name(context, entry_ex->entry.principal, &name);
278         if (ret != 0) {
279                 talloc_free(tmp_ctx);
280                 return ret;
281         }
282
283         if (addresses) {
284                 for (i=0; i < addresses->len; i++) {
285                         if (addresses->val->addr_type == KRB5_ADDRESS_NETBIOS) {
286                                 workstation = talloc_strndup(tmp_ctx, addresses->val->address.data, MIN(addresses->val->address.length, 15));
287                                 if (workstation) {
288                                         break;
289                                 }
290                         }
291                 }
292         }
293
294         /* Strip space padding */
295         if (workstation) {
296                 i = MIN(strlen(workstation), 15);
297                 for (; i > 0 && workstation[i - 1] == ' '; i--) {
298                         workstation[i - 1] = '\0';
299                 }
300         }
301
302         nt_status = authsam_account_ok(tmp_ctx, 
303                                        private->samdb, 
304                                        MSV1_0_ALLOW_SERVER_TRUST_ACCOUNT | MSV1_0_ALLOW_WORKSTATION_TRUST_ACCOUNT,
305                                        private->msg,
306                                        private->realm_ref_msg,
307                                        workstation,
308                                        name);
309         free(name);
310
311         if (NT_STATUS_IS_OK(nt_status))
312                 return 0;
313
314         if (NT_STATUS_EQUAL(nt_status, NT_STATUS_PASSWORD_MUST_CHANGE))
315                 ret = KRB5KDC_ERR_KEY_EXPIRED;
316         else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_PASSWORD_EXPIRED))
317                 ret = KRB5KDC_ERR_KEY_EXPIRED;
318         else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_ACCOUNT_EXPIRED))
319                 ret = KRB5KDC_ERR_CLIENT_REVOKED;
320         else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_ACCOUNT_DISABLED))
321                 ret = KRB5KDC_ERR_CLIENT_REVOKED;
322         else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_LOGON_HOURS))
323                 ret = KRB5KDC_ERR_CLIENT_REVOKED;
324         else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_ACCOUNT_LOCKED_OUT))
325                 ret = KRB5KDC_ERR_CLIENT_REVOKED;
326         else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_WORKSTATION))
327                 ret = KRB5KDC_ERR_POLICY;
328         else
329                 ret = KRB5KDC_ERR_POLICY;
330
331         samba_kdc_build_edata_reply(tmp_ctx, e_data, nt_status);
332
333         return ret;
334 }
335