pac: Fix wrong memory allocation check
[idra/samba.git] / source4 / kdc / pac-glue.c
index 21ae7091a691700e09fbd1db34c90e69916d3177..5718452cc00c1ac9379e3b1f95ddaa8c7f3c3805 100644 (file)
-/* 
+/*
    Unix SMB/CIFS implementation.
 
    PAC Glue between Samba and the KDC
-   
+
    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2005-2009
+   Copyright (C) Simo Sorce <idra@samba.org> 2010
 
    This program is free software; you can redistribute it and/or modify
    it under the terms of the GNU General Public License as published by
    the Free Software Foundation; either version 3 of the License, or
    (at your option) any later version.
-   
+
    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU General Public License for more details.
 
-   
+
    You should have received a copy of the GNU General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
 #include "includes.h"
 #include "../libds/common/flags.h"
-#include "lib/ldb/include/ldb.h"
-#include "librpc/gen_ndr/ndr_krb5pac.h"
-#include "librpc/gen_ndr/krb5pac.h"
+#include <ldb.h>
 #include "auth/auth.h"
-#include "auth/auth_sam.h"
 #include "auth/auth_sam_reply.h"
-#include "kdc/kdc.h"
+#include "kdc/kdc-glue.h"
+#include "kdc/pac-glue.h"
 #include "param/param.h"
+#include "librpc/gen_ndr/ndr_krb5pac.h"
 
-struct krb5_dh_moduli;
-struct _krb5_krb_auth_data;
-
-krb5_error_code        samba_kdc_plugin_init(krb5_context context, void **ptr) 
-{
-       *ptr = NULL;
-       return 0;
-}
-
-void   samba_kdc_plugin_fini(void *ptr) 
-{
-       return;
-}
-
-static krb5_error_code make_pac(krb5_context context,
-                               TALLOC_CTX *mem_ctx, 
-                               struct smb_iconv_convenience *iconv_convenience,
-                               struct auth_serversupplied_info *server_info,
-                               krb5_pac *pac) 
+static
+NTSTATUS samba_get_logon_info_pac_blob(TALLOC_CTX *mem_ctx,
+                                      struct auth_user_info_dc *info,
+                                      DATA_BLOB *pac_data)
 {
-       union PAC_INFO info;
        struct netr_SamInfo3 *info3;
-       krb5_data pac_data;
-       NTSTATUS nt_status;
+       union PAC_INFO pac_info;
        enum ndr_err_code ndr_err;
-       DATA_BLOB pac_out;
-       krb5_error_code ret;
+       NTSTATUS nt_status;
 
-       ZERO_STRUCT(info);
+       ZERO_STRUCT(pac_info);
 
-       nt_status = auth_convert_server_info_saminfo3(mem_ctx, server_info, &info3);
+       nt_status = auth_convert_user_info_dc_saminfo3(mem_ctx, info, &info3);
        if (!NT_STATUS_IS_OK(nt_status)) {
-               DEBUG(1, ("Getting Samba info failed: %s\n", nt_errstr(nt_status)));
-               return EINVAL;
+               DEBUG(1, ("Getting Samba info failed: %s\n",
+                         nt_errstr(nt_status)));
+               return nt_status;
        }
 
-       info.logon_info.info = talloc_zero(mem_ctx, struct PAC_LOGON_INFO);
-       if (!mem_ctx) {
-               return ENOMEM;
+       pac_info.logon_info.info = talloc_zero(mem_ctx, struct PAC_LOGON_INFO);
+       if (!pac_info.logon_info.info) {
+               return NT_STATUS_NO_MEMORY;
        }
 
-       info.logon_info.info->info3 = *info3;
+       pac_info.logon_info.info->info3 = *info3;
 
-       ndr_err = ndr_push_union_blob(&pac_out, mem_ctx, iconv_convenience, &info,
+       ndr_err = ndr_push_union_blob(pac_data, mem_ctx, &pac_info,
                                      PAC_TYPE_LOGON_INFO,
                                      (ndr_push_flags_fn_t)ndr_push_PAC_INFO);
        if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
                nt_status = ndr_map_error2ntstatus(ndr_err);
-               DEBUG(1, ("PAC (presig) push failed: %s\n", nt_errstr(nt_status)));
-               return EINVAL;
+               DEBUG(1, ("PAC (presig) push failed: %s\n",
+                         nt_errstr(nt_status)));
+               return nt_status;
        }
 
-       ret = krb5_data_copy(&pac_data, pac_out.data, pac_out.length);
+       return NT_STATUS_OK;
+}
+
+krb5_error_code samba_make_krb5_pac(krb5_context context,
+                                   DATA_BLOB *pac_blob,
+                                   DATA_BLOB *deleg_blob,
+                                   krb5_pac *pac)
+{
+       krb5_data pac_data;
+       krb5_data deleg_data;
+       krb5_error_code ret;
+
+        /* The user account may be set not to want the PAC */
+       if (!pac_blob) {
+               return 0;
+       }
+
+       ret = krb5_data_copy(&pac_data, pac_blob->data, pac_blob->length);
        if (ret != 0) {
                return ret;
        }
 
+       ZERO_STRUCT(deleg_data);
+       if (deleg_blob) {
+               ret = krb5_data_copy(&deleg_data,
+                                    deleg_blob->data,
+                                    deleg_blob->length);
+               if (ret != 0) {
+                       krb5_data_free(&pac_data);
+                       return ret;
+               }
+       }
+
        ret = krb5_pac_init(context, pac);
        if (ret != 0) {
                krb5_data_free(&pac_data);
+               krb5_data_free(&deleg_data);
                return ret;
        }
 
        ret = krb5_pac_add_buffer(context, *pac, PAC_TYPE_LOGON_INFO, &pac_data);
        krb5_data_free(&pac_data);
        if (ret != 0) {
+               krb5_data_free(&deleg_data);
                return ret;
        }
 
+       if (deleg_blob) {
+               ret = krb5_pac_add_buffer(context, *pac,
+                                         PAC_TYPE_CONSTRAINED_DELEGATION,
+                                         &deleg_data);
+               krb5_data_free(&deleg_data);
+               if (ret != 0) {
+                       return ret;
+               }
+       }
+
        return ret;
 }
 
-/* Given the right private pointer from hdb_ldb, get a PAC from the attached ldb messages */
-krb5_error_code samba_kdc_get_pac(void *priv,
-                                 krb5_context context, 
-                                 struct hdb_entry_ex *client,
-                                 krb5_pac *pac)
+bool samba_princ_needs_pac(struct hdb_entry_ex *princ)
 {
-       krb5_error_code ret;
-       NTSTATUS nt_status;
-       struct auth_serversupplied_info *server_info;
-       struct hdb_ldb_private *p = talloc_get_type(client->ctx, struct hdb_ldb_private);
-       TALLOC_CTX *mem_ctx = talloc_named(p, 0, "samba_get_pac context");
-       unsigned int userAccountControl;
 
-       if (!mem_ctx) {
-               return ENOMEM;
-       }
+       struct samba_kdc_entry *p = talloc_get_type(princ->ctx, struct samba_kdc_entry);
+       uint32_t userAccountControl;
 
-       /* The user account may be set not to want the PAC */
+
+       /* The service account may be set not to want the PAC */
        userAccountControl = ldb_msg_find_attr_as_uint(p->msg, "userAccountControl", 0);
        if (userAccountControl & UF_NO_AUTH_DATA_REQUIRED) {
-               *pac = NULL;
-               return 0;
+               return false;
        }
 
-       nt_status = authsam_make_server_info(mem_ctx, p->samdb,
-                                            lp_netbios_name(p->lp_ctx),
-                                            lp_sam_name(p->lp_ctx),
+       return true;
+}
+
+/* Was the krbtgt an RODC (and we are not) */
+bool samba_krbtgt_was_untrusted_rodc(struct hdb_entry_ex *princ)
+{
+
+       struct samba_kdc_entry *p = talloc_get_type(princ->ctx, struct samba_kdc_entry);
+       int rodc_krbtgt_number;
+
+       /* Determine if this was printed by an RODC */
+       rodc_krbtgt_number = ldb_msg_find_attr_as_int(p->msg, "msDS-SecondaryKrbTgtNumber", -1);
+       if (rodc_krbtgt_number == -1) {
+               return false;
+       } else if (rodc_krbtgt_number != p->kdc_db_ctx->my_krbtgt_number) {
+               return true;
+       }
+
+       return false;
+}
+
+NTSTATUS samba_kdc_get_pac_blob(TALLOC_CTX *mem_ctx,
+                               struct hdb_entry_ex *client,
+                               DATA_BLOB **_pac_blob)
+{
+       struct samba_kdc_entry *p = talloc_get_type(client->ctx, struct samba_kdc_entry);
+       struct auth_user_info_dc *user_info_dc;
+       DATA_BLOB *pac_blob;
+       NTSTATUS nt_status;
+
+       /* The user account may be set not to want the PAC */
+       if ( ! samba_princ_needs_pac(client)) {
+               *_pac_blob = NULL;
+               return NT_STATUS_OK;
+       }
+
+       pac_blob = talloc_zero(mem_ctx, DATA_BLOB);
+       if (!pac_blob) {
+               return NT_STATUS_NO_MEMORY;
+       }
+
+       nt_status = authsam_make_user_info_dc(mem_ctx, p->kdc_db_ctx->samdb,
+                                            lpcfg_netbios_name(p->kdc_db_ctx->lp_ctx),
+                                            lpcfg_sam_name(p->kdc_db_ctx->lp_ctx),
                                             p->realm_dn,
                                             p->msg,
                                             data_blob(NULL, 0),
                                             data_blob(NULL, 0),
-                                            &server_info);
+                                            &user_info_dc);
        if (!NT_STATUS_IS_OK(nt_status)) {
                DEBUG(0, ("Getting user info for PAC failed: %s\n",
                          nt_errstr(nt_status)));
-               return ENOMEM;
+               return nt_status;
        }
 
-       ret = make_pac(context, mem_ctx, p->iconv_convenience, server_info, pac);
+       nt_status = samba_get_logon_info_pac_blob(mem_ctx, user_info_dc, pac_blob);
+       if (!NT_STATUS_IS_OK(nt_status)) {
+               DEBUG(0, ("Building PAC failed: %s\n",
+                         nt_errstr(nt_status)));
+               return nt_status;
+       }
 
-       talloc_free(mem_ctx);
-       return ret;
+       *_pac_blob = pac_blob;
+       return NT_STATUS_OK;
 }
 
-/* Resign (and reform, including possibly new groups) a PAC */
-
-krb5_error_code samba_kdc_reget_pac(void *priv, krb5_context context,
-                               const krb5_principal client_principal,
-                               struct hdb_entry_ex *client,  
-                               struct hdb_entry_ex *server, krb5_pac *pac)
+NTSTATUS samba_kdc_update_pac_blob(TALLOC_CTX *mem_ctx,
+                                  krb5_context context,
+                                  const krb5_pac pac, DATA_BLOB *pac_blob)
 {
+       struct auth_user_info_dc *user_info_dc;
        krb5_error_code ret;
+       NTSTATUS nt_status;
 
-       unsigned int userAccountControl;
+       ret = kerberos_pac_to_user_info_dc(mem_ctx, pac,
+                                          context, &user_info_dc, NULL, NULL);
+       if (ret) {
+               return NT_STATUS_UNSUCCESSFUL;
+       }
 
-       struct hdb_ldb_private *p = talloc_get_type(server->ctx, struct hdb_ldb_private);
+       nt_status = samba_get_logon_info_pac_blob(mem_ctx, 
+                                                 user_info_dc, pac_blob);
 
-       struct auth_serversupplied_info *server_info_out;
+       return nt_status;
+}
 
-       TALLOC_CTX *mem_ctx = talloc_named(p, 0, "samba_get_pac context");
-       
-       if (!mem_ctx) {
-               return ENOMEM;
+NTSTATUS samba_kdc_update_delegation_info_blob(TALLOC_CTX *mem_ctx,
+                               krb5_context context,
+                               const krb5_pac pac,
+                               const krb5_principal server_principal,
+                               const krb5_principal proxy_principal,
+                               DATA_BLOB *new_blob)
+{
+       krb5_data old_data;
+       DATA_BLOB old_blob;
+       krb5_error_code ret;
+       NTSTATUS nt_status;
+       enum ndr_err_code ndr_err;
+       union PAC_INFO info;
+       struct PAC_CONSTRAINED_DELEGATION _d;
+       struct PAC_CONSTRAINED_DELEGATION *d = NULL;
+       char *server = NULL;
+       char *proxy = NULL;
+       uint32_t i;
+       TALLOC_CTX *tmp_ctx = talloc_new(mem_ctx);
+
+       if (tmp_ctx == NULL) {
+               return NT_STATUS_NO_MEMORY;
        }
 
-       /* The service account may be set not to want the PAC */
-       userAccountControl = ldb_msg_find_attr_as_uint(p->msg, "userAccountControl", 0);
-       if (userAccountControl & UF_NO_AUTH_DATA_REQUIRED) {
-               talloc_free(mem_ctx);
-               *pac = NULL;
-               return 0;
+       ret = krb5_pac_get_buffer(context, pac, PAC_TYPE_CONSTRAINED_DELEGATION, &old_data);
+       if (ret == ENOENT) {
+               ZERO_STRUCT(old_data);
+       } else if (ret) {
+               talloc_free(tmp_ctx);
+               return NT_STATUS_UNSUCCESSFUL;
        }
 
-       ret = kerberos_pac_to_server_info(mem_ctx, p->iconv_convenience,
-                                         *pac, context, &server_info_out);
+       old_blob.length = old_data.length;
+       old_blob.data = (uint8_t *)old_data.data;
+
+       ZERO_STRUCT(info);
+       if (old_blob.length > 0) {
+               ndr_err = ndr_pull_union_blob(&old_blob, mem_ctx,
+                               &info, PAC_TYPE_CONSTRAINED_DELEGATION,
+                               (ndr_pull_flags_fn_t)ndr_pull_PAC_INFO);
+               if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+                       krb5_data_free(&old_data);
+                       nt_status = ndr_map_error2ntstatus(ndr_err);
+                       DEBUG(0,("can't parse the PAC LOGON_INFO: %s\n", nt_errstr(nt_status)));
+                       talloc_free(tmp_ctx);
+                       return nt_status;
+               }
+       } else {
+               ZERO_STRUCT(_d);
+               info.constrained_delegation.info = &_d;
+       }
+       krb5_data_free(&old_data);
 
-       /* We will compleatly regenerate this pac */
-       krb5_pac_free(context, *pac);
+       ret = krb5_unparse_name(context, server_principal, &server);
+       if (ret) {
+               talloc_free(tmp_ctx);
+               return NT_STATUS_INTERNAL_ERROR;
+       }
 
+       ret = krb5_unparse_name_flags(context, proxy_principal,
+                                     KRB5_PRINCIPAL_UNPARSE_NO_REALM, &proxy);
        if (ret) {
-               talloc_free(mem_ctx);
-               return ret;
+               SAFE_FREE(server);
+               talloc_free(tmp_ctx);
+               return NT_STATUS_INTERNAL_ERROR;
        }
 
-       ret = make_pac(context, mem_ctx, p->iconv_convenience, server_info_out, pac);
+       d = info.constrained_delegation.info;
+       i = d->num_transited_services;
+       d->proxy_target.string = server;
+       d->transited_services = talloc_realloc(mem_ctx, d->transited_services,
+                                              struct lsa_String, i + 1);
+       d->transited_services[i].string = proxy;
+       d->num_transited_services = i + 1;
+
+       ndr_err = ndr_push_union_blob(new_blob, mem_ctx,
+                               &info, PAC_TYPE_CONSTRAINED_DELEGATION,
+                               (ndr_push_flags_fn_t)ndr_push_PAC_INFO);
+       SAFE_FREE(server);
+       SAFE_FREE(proxy);
+       if (!NDR_ERR_CODE_IS_SUCCESS(ndr_err)) {
+               krb5_data_free(&old_data);
+               nt_status = ndr_map_error2ntstatus(ndr_err);
+               DEBUG(0,("can't parse the PAC LOGON_INFO: %s\n", nt_errstr(nt_status)));
+               talloc_free(tmp_ctx);
+               return nt_status;
+       }
 
-       talloc_free(mem_ctx);
-       return ret;
+       talloc_free(tmp_ctx);
+       return NT_STATUS_OK;
 }
 
-static void samba_kdc_build_edata_reply(TALLOC_CTX *tmp_ctx, krb5_data *e_data,
-                                      NTSTATUS nt_status)
+/* this function allocates 'data' using malloc.
+ * The caller is responsible for freeing it */
+void samba_kdc_build_edata_reply(NTSTATUS nt_status, DATA_BLOB *e_data)
 {
        PA_DATA pa;
        unsigned char *buf;
@@ -226,91 +352,56 @@ static void samba_kdc_build_edata_reply(TALLOC_CTX *tmp_ctx, krb5_data *e_data,
        return;
 }
 
-/* Given an hdb entry (and in particular it's private member), consult
- * the account_ok routine in auth/auth_sam.c for consistancy */
+/* function to map policy errors */
+krb5_error_code samba_kdc_map_policy_err(NTSTATUS nt_status)
+{
+       krb5_error_code ret;
 
+       if (NT_STATUS_EQUAL(nt_status, NT_STATUS_PASSWORD_MUST_CHANGE))
+               ret = KRB5KDC_ERR_KEY_EXPIRED;
+       else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_PASSWORD_EXPIRED))
+               ret = KRB5KDC_ERR_KEY_EXPIRED;
+       else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_ACCOUNT_EXPIRED))
+               ret = KRB5KDC_ERR_CLIENT_REVOKED;
+       else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_ACCOUNT_DISABLED))
+               ret = KRB5KDC_ERR_CLIENT_REVOKED;
+       else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_LOGON_HOURS))
+               ret = KRB5KDC_ERR_CLIENT_REVOKED;
+       else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_ACCOUNT_LOCKED_OUT))
+               ret = KRB5KDC_ERR_CLIENT_REVOKED;
+       else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_WORKSTATION))
+               ret = KRB5KDC_ERR_POLICY;
+       else
+               ret = KRB5KDC_ERR_POLICY;
 
-krb5_error_code samba_kdc_check_client_access(void *priv, 
-                                             krb5_context context, 
-                                             krb5_kdc_configuration *config,
-                                             hdb_entry_ex *client_ex, const char *client_name,
-                                             hdb_entry_ex *server_ex, const char *server_name,
-                                             KDC_REQ *req,
-                                             krb5_data *e_data)
+       return ret;
+}
+
+/* Given a kdc entry, consult the account_ok routine in auth/auth_sam.c
+ * for consistency */
+NTSTATUS samba_kdc_check_client_access(struct samba_kdc_entry *kdc_entry,
+                                      const char *client_name,
+                                      const char *workstation,
+                                      bool password_change)
 {
-       krb5_error_code ret;
-       NTSTATUS nt_status;
        TALLOC_CTX *tmp_ctx;
-       struct hdb_ldb_private *p;
-       char *workstation = NULL;
-       HostAddresses *addresses = req->req_body.addresses;
-       int i;
-       bool password_change;
-
-       tmp_ctx = talloc_new(client_ex->ctx);
-       p = talloc_get_type(client_ex->ctx, struct hdb_ldb_private);
+       NTSTATUS nt_status;
 
+       tmp_ctx = talloc_named(NULL, 0, "samba_kdc_check_client_access");
        if (!tmp_ctx) {
-               return ENOMEM;
-       }
-       
-       if (addresses) {
-               for (i=0; i < addresses->len; i++) {
-                       if (addresses->val->addr_type == KRB5_ADDRESS_NETBIOS) {
-                               workstation = talloc_strndup(tmp_ctx, addresses->val->address.data, MIN(addresses->val->address.length, 15));
-                               if (workstation) {
-                                       break;
-                               }
-                       }
-               }
+               return NT_STATUS_NO_MEMORY;
        }
 
-       /* Strip space padding */
-       if (workstation) {
-               i = MIN(strlen(workstation), 15);
-               for (; i > 0 && workstation[i - 1] == ' '; i--) {
-                       workstation[i - 1] = '\0';
-               }
-       }
-
-       password_change = (server_ex && server_ex->entry.flags.change_pw);
-
        /* we allow all kinds of trusts here */
-       nt_status = authsam_account_ok(tmp_ctx, 
-                                      p->samdb,
-                                      MSV1_0_ALLOW_SERVER_TRUST_ACCOUNT | MSV1_0_ALLOW_WORKSTATION_TRUST_ACCOUNT,
-                                      p->realm_dn,
-                                      p->msg,
-                                      workstation,
-                                      client_name, true, password_change);
-
-       if (NT_STATUS_IS_OK(nt_status)) {
-               /* Now do the standard Heimdal check */
-               ret = kdc_check_flags(context, config,
-                                     client_ex, client_name,
-                                     server_ex, server_name,
-                                     req->msg_type == krb_as_req);
-       } else {
-               if (NT_STATUS_EQUAL(nt_status, NT_STATUS_PASSWORD_MUST_CHANGE))
-                       ret = KRB5KDC_ERR_KEY_EXPIRED;
-               else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_PASSWORD_EXPIRED))
-                       ret = KRB5KDC_ERR_KEY_EXPIRED;
-               else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_ACCOUNT_EXPIRED))
-                       ret = KRB5KDC_ERR_CLIENT_REVOKED;
-               else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_ACCOUNT_DISABLED))
-                       ret = KRB5KDC_ERR_CLIENT_REVOKED;
-               else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_LOGON_HOURS))
-                       ret = KRB5KDC_ERR_CLIENT_REVOKED;
-               else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_ACCOUNT_LOCKED_OUT))
-                       ret = KRB5KDC_ERR_CLIENT_REVOKED;
-               else if (NT_STATUS_EQUAL(nt_status, NT_STATUS_INVALID_WORKSTATION))
-                       ret = KRB5KDC_ERR_POLICY;
-               else
-                       ret = KRB5KDC_ERR_POLICY;
-
-               samba_kdc_build_edata_reply(tmp_ctx, e_data, nt_status);
-       }
-
-       return ret;
+       nt_status = authsam_account_ok(tmp_ctx,
+                                      kdc_entry->kdc_db_ctx->samdb,
+                                      MSV1_0_ALLOW_SERVER_TRUST_ACCOUNT |
+                                      MSV1_0_ALLOW_WORKSTATION_TRUST_ACCOUNT,
+                                      kdc_entry->realm_dn, kdc_entry->msg,
+                                      workstation, client_name,
+                                      true, password_change);
+
+       talloc_free(tmp_ctx);
+       return nt_status;
 }