From 1fdd6a6e68359bccc4998e416cb2395ac8e6eaca Mon Sep 17 00:00:00 2001 From: Simo Sorce Date: Fri, 19 May 2006 21:12:26 +0000 Subject: r15725: First shot at making password_hash async The async path is not yet enabled by default so it should make no harm (This used to be commit b7d5f2325726757a4fcd0b5ac03de1b867085a89) --- source4/dsdb/samdb/ldb_modules/password_hash.c | 1087 +++++++++++++++++++++++- 1 file changed, 1076 insertions(+), 11 deletions(-) (limited to 'source4/dsdb/samdb/ldb_modules/password_hash.c') diff --git a/source4/dsdb/samdb/ldb_modules/password_hash.c b/source4/dsdb/samdb/ldb_modules/password_hash.c index 46bafeefc2..b7f4aff67a 100644 --- a/source4/dsdb/samdb/ldb_modules/password_hash.c +++ b/source4/dsdb/samdb/ldb_modules/password_hash.c @@ -121,13 +121,6 @@ static int password_hash_handle(struct ldb_module *module, struct ldb_request *r /* Do the original action */ - /* If no part of this touches the sambaPassword, then we don't - * need to make any changes. For password changes/set there should - * be a 'delete' or a 'modify' on this attribute. */ - if ((attribute = ldb_msg_find_element(msg, "sambaPassword")) == NULL ) { - return ldb_next_request(module, req); - } - mem_ctx = talloc_new(module); if (!mem_ctx) { return LDB_ERR_OPERATIONS_ERROR; @@ -326,7 +319,7 @@ static int password_hash_handle(struct ldb_module *module, struct ldb_request *r is_computer = False; } - domain_expression = talloc_asprintf(mem_ctx, "(&(objectSid=%s)(|(objectClass=domain)(objectClass=builtinDomain)))", + domain_expression = talloc_asprintf(mem_ctx, "(&(objectSid=%s)(objectClass=domain))", ldap_encode_ndr_dom_sid(mem_ctx, domain_sid)); /* Find the user's domain, then find out the domain password @@ -663,12 +656,12 @@ static int password_hash_handle(struct ldb_module *module, struct ldb_request *r } sambaNTPwdHistory_len = MIN(sambaNTPwdHistory_len + 1, pwdHistoryLength); - CHECK_RET(samdb_msg_add_hashes(module->ldb, mem_ctx, modify_msg, + CHECK_RET(samdb_msg_add_hashes(mem_ctx, modify_msg, "sambaLMPwdHistory", new_sambaLMPwdHistory, sambaLMPwdHistory_len)); - CHECK_RET(samdb_msg_add_hashes(module->ldb, mem_ctx, modify_msg, + CHECK_RET(samdb_msg_add_hashes(mem_ctx, modify_msg, "sambaNTPwdHistory", new_sambaNTPwdHistory, sambaNTPwdHistory_len)); @@ -704,6 +697,13 @@ static int password_hash_add(struct ldb_module *module, struct ldb_request *req) return ldb_next_request(module, req); } + /* If no part of this touches the sambaPassword, then we don't + * need to make any changes. For password changes/set there should + * be a 'delete' or a 'modify' on this attribute. */ + if (ldb_msg_find_element(msg, "sambaPassword") == NULL ) { + return ldb_next_request(module, req); + } + return password_hash_handle(module, req, msg); } @@ -718,9 +718,1067 @@ static int password_hash_modify(struct ldb_module *module, struct ldb_request *r return ldb_next_request(module, req); } + /* If no part of this touches the sambaPassword, then we don't + * need to make any changes. For password changes/set there should + * be a 'delete' or a 'modify' on this attribute. */ + if (ldb_msg_find_element(msg, "sambaPassword") == NULL ) { + return ldb_next_request(module, req); + } + return password_hash_handle(module, req, msg); } +enum ph_type {PH_ADD, PH_MOD}; +enum ph_step {PH_ADD_SEARCH_DOM, PH_ADD_DO_ADD, PH_MOD_DO_REQ, PH_MOD_SEARCH_SELF, PH_MOD_SEARCH_DOM, PH_MOD_DO_MOD}; + +struct ph_async_context { + + enum ph_type type; + enum ph_step step; + + struct ldb_module *module; + struct ldb_request *orig_req; + + struct ldb_request *dom_req; + struct ldb_async_result *dom_res; + + struct ldb_request *down_req; + + struct ldb_request *search_req; + struct ldb_async_result *search_res; + + struct ldb_request *mod_req; +}; + +struct domain_data { + uint_t pwdProperties; + uint_t pwdHistoryLength; + char *dnsDomain; + char *realm; +}; + +static int add_password_hashes(struct ldb_module *module, struct ldb_message *msg, int is_mod) +{ + const char *sambaPassword; + struct samr_Password tmp_hash; + + sambaPassword = ldb_msg_find_string(msg, "sambaPassword", NULL); + if (sambaPassword == NULL) { /* impossible, what happened ?! */ + return LDB_ERR_OPERATIONS_ERROR; + } + + /* compute the new nt and lm hashes */ + if (is_mod) { + if (ldb_msg_add_empty(msg, "ntPwdHash", LDB_FLAG_MOD_REPLACE) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + } + E_md4hash(sambaPassword, tmp_hash.hash); + if (samdb_msg_add_hash(module->ldb, msg, msg, "ntPwdHash", &tmp_hash) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + + if (E_deshash(sambaPassword, tmp_hash.hash)) { + if (is_mod) { + if (ldb_msg_add_empty(msg, "lmPwdHash", LDB_FLAG_MOD_REPLACE) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + } + if (samdb_msg_add_hash(module->ldb, msg, msg, "lmPwdHash", &tmp_hash) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + } + + return LDB_SUCCESS; +} + +static int add_krb5_keys_from_password(struct ldb_module *module, struct ldb_message *msg, + struct smb_krb5_context *smb_krb5_context, + struct domain_data *domain, + const char *samAccountName, + const char *user_principal_name, + int is_computer) +{ + const char *sambaPassword; + Principal *salt_principal; + krb5_error_code krb5_ret; + size_t num_keys; + Key *keys; + int i; + + /* Many, many thanks to lukeh@padl.com for this + * algorithm, described in his Nov 10 2004 mail to + * samba-technical@samba.org */ + + sambaPassword = ldb_msg_find_string(msg, "sambaPassword", NULL); + if (sambaPassword == NULL) { /* impossible, what happened ?! */ + return LDB_ERR_OPERATIONS_ERROR; + } + + if (is_computer) { + /* Determine a salting principal */ + char *name = talloc_strdup(msg, samAccountName); + char *saltbody; + if (name == NULL) { + ldb_set_errstring(module->ldb, + talloc_asprintf(msg, "password_hash_handle: " + "generation of new kerberos keys failed: %s is a computer without a samAccountName", + ldb_dn_linearize(msg, msg->dn))); + return LDB_ERR_OPERATIONS_ERROR; + } + if (name[strlen(name)-1] == '$') { + name[strlen(name)-1] = '\0'; + } + saltbody = talloc_asprintf(msg, "%s.%s", name, domain->dnsDomain); + + krb5_ret = krb5_make_principal(smb_krb5_context->krb5_context, + &salt_principal, + domain->realm, "host", + saltbody, NULL); + } else if (user_principal_name) { + char *p; + user_principal_name = talloc_strdup(msg, user_principal_name); + if (user_principal_name == NULL) { + return LDB_ERR_OPERATIONS_ERROR; + } else { + p = strchr(user_principal_name, '@'); + if (p) { + p[0] = '\0'; + } + krb5_ret = krb5_make_principal(smb_krb5_context->krb5_context, + &salt_principal, + domain->realm, user_principal_name, NULL); + } + } else { + if (!samAccountName) { + ldb_set_errstring(module->ldb, + talloc_asprintf(msg, "password_hash_handle: " + "generation of new kerberos keys failed: %s has no samAccountName", + ldb_dn_linearize(msg, msg->dn))); + return LDB_ERR_OPERATIONS_ERROR; + } + krb5_ret = krb5_make_principal(smb_krb5_context->krb5_context, + &salt_principal, + domain->realm, samAccountName, + NULL); + } + + if (krb5_ret) { + ldb_set_errstring(module->ldb, + talloc_asprintf(msg, "password_hash_handle: " + "generation of a saltking principal failed: %s", + smb_get_krb5_error_message(smb_krb5_context->krb5_context, + krb5_ret, msg))); + return LDB_ERR_OPERATIONS_ERROR; + } + + /* TODO: We may wish to control the encryption types chosen in future */ + krb5_ret = hdb_generate_key_set_password(smb_krb5_context->krb5_context, + salt_principal, sambaPassword, &keys, &num_keys); + krb5_free_principal(smb_krb5_context->krb5_context, salt_principal); + + if (krb5_ret) { + ldb_set_errstring(module->ldb, + talloc_asprintf(msg, "password_hash_handle: " + "generation of new kerberos keys failed: %s", + smb_get_krb5_error_message(smb_krb5_context->krb5_context, + krb5_ret, msg))); + return LDB_ERR_OPERATIONS_ERROR; + } + + /* Walking all the key types generated, transform each + * key into an ASN.1 blob + */ + for (i=0; i < num_keys; i++) { + unsigned char *buf; + size_t buf_size; + size_t len; + struct ldb_val val; + int ret; + + if (keys[i].key.keytype == ENCTYPE_ARCFOUR_HMAC) { + /* We might end up doing this below: + * This ensures we get the unicode + * conversion right. This should also + * be fixed in the Heimdal libs */ + continue; + } + ASN1_MALLOC_ENCODE(Key, buf, buf_size, &keys[i], &len, krb5_ret); + if (krb5_ret) { + return LDB_ERR_OPERATIONS_ERROR; + } + + val.data = talloc_memdup(msg, buf, len); + val.length = len; + free(buf); + if (!val.data || krb5_ret) { + hdb_free_keys (smb_krb5_context->krb5_context, num_keys, keys); + return LDB_ERR_OPERATIONS_ERROR; + } + ret = ldb_msg_add_value(msg, "krb5Key", &val); + if (ret != LDB_SUCCESS) { + hdb_free_keys (smb_krb5_context->krb5_context, num_keys, keys); + return ret; + } + } + + hdb_free_keys (smb_krb5_context->krb5_context, num_keys, keys); + + return LDB_SUCCESS; +} + +static int add_krb5_keys_from_NThash(struct ldb_module *module, struct ldb_message *msg, + struct smb_krb5_context *smb_krb5_context) +{ + struct samr_Password *ntPwdHash; + krb5_error_code krb5_ret; + unsigned char *buf; + size_t buf_size; + size_t len; + struct ldb_val val; + Key key; + + key.mkvno = 0; + key.salt = NULL; /* No salt for this enc type */ + + ntPwdHash = samdb_result_hash(msg, msg, "ntPwdHash"); + if (ntPwdHash == NULL) { /* what happened ?! */ + return LDB_ERR_OPERATIONS_ERROR; + } + + krb5_ret = krb5_keyblock_init(smb_krb5_context->krb5_context, + ENCTYPE_ARCFOUR_HMAC, + ntPwdHash->hash, sizeof(ntPwdHash->hash), + &key.key); + if (krb5_ret) { + return LDB_ERR_OPERATIONS_ERROR; + } + ASN1_MALLOC_ENCODE(Key, buf, buf_size, &key, &len, krb5_ret); + if (krb5_ret) { + return LDB_ERR_OPERATIONS_ERROR; + } + krb5_free_keyblock_contents(smb_krb5_context->krb5_context, + &key.key); + + val.data = talloc_memdup(msg, buf, len); + val.length = len; + free(buf); + if (!val.data) { + return LDB_ERR_OPERATIONS_ERROR; + } + if (ldb_msg_add_value(msg, "krb5Key", &val) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + + return LDB_SUCCESS; +} + +static int set_pwdLastSet(struct ldb_module *module, struct ldb_message *msg) +{ + NTTIME now_nt; + + /* set it as now */ + unix_to_nt_time(&now_nt, time(NULL)); + + /* replace or add */ + if (ldb_msg_add_empty(msg, "pwdLastSet", LDB_FLAG_MOD_REPLACE) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + + if (samdb_msg_add_uint64(module->ldb, msg, msg, "pwdLastSet", now_nt) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + + return LDB_SUCCESS; +} + +static int add_keyVersionNumber(struct ldb_module *module, struct ldb_message *msg, int previous) +{ + /* replace or add */ + if (ldb_msg_add_empty(msg, "msDS-KeyVersionNumber", LDB_FLAG_MOD_REPLACE) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + + if (samdb_msg_add_uint(module->ldb, msg, msg, "msDS-KeyVersionNumber", previous+1) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + + return LDB_SUCCESS; +} + +static int setPwdHistory(struct ldb_module *module, struct ldb_message *msg, struct ldb_message *old_msg, int hlen) +{ + struct samr_Password *nt_hash; + struct samr_Password *lm_hash; + struct samr_Password *nt_history; + struct samr_Password *lm_history; + struct samr_Password *new_nt_history; + struct samr_Password *new_lm_history; + int nt_hist_len; + int lm_hist_len; + int i; + + nt_hash = samdb_result_hash(msg, old_msg, "ntPwdHash"); + lm_hash = samdb_result_hash(msg, old_msg, "lmPwdHash"); + + /* if no previous passwords just return */ + if (nt_hash == NULL && lm_hash == NULL) return LDB_SUCCESS; + + nt_hist_len = samdb_result_hashes(msg, old_msg, "sambaNTPwdHistory", &nt_history); + lm_hist_len = samdb_result_hashes(msg, old_msg, "sambaLMPwdHistory", &lm_history); + + /* We might not have an old NT password */ + new_nt_history = talloc_array(msg, struct samr_Password, hlen); + if (new_nt_history == NULL) { + return LDB_ERR_OPERATIONS_ERROR; + } + for (i = 0; i < MIN(hlen-1, nt_hist_len); i++) { + new_nt_history[i+1] = nt_history[i]; + } + nt_hist_len = i + 1; + if (nt_hash) { + new_nt_history[0] = *nt_hash; + } else { + ZERO_STRUCT(new_nt_history[0]); + } + if (ldb_msg_add_empty(msg, "sambaNTPwdHistory", LDB_FLAG_MOD_REPLACE) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + if (samdb_msg_add_hashes(msg, msg, "sambaNTPwdHistory", new_nt_history, nt_hist_len) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + + + /* Don't store 'long' passwords in the LM history, + but make sure to 'expire' one password off the other end */ + new_lm_history = talloc_array(msg, struct samr_Password, hlen); + if (new_lm_history == NULL) { + return LDB_ERR_OPERATIONS_ERROR; + } + for (i = 0; i < MIN(hlen-1, lm_hist_len); i++) { + new_lm_history[i+1] = lm_history[i]; + } + lm_hist_len = i + 1; + if (lm_hash) { + new_lm_history[0] = *lm_hash; + } else { + ZERO_STRUCT(new_lm_history[0]); + } + if (ldb_msg_add_empty(msg, "sambaLMPwdHistory", LDB_FLAG_MOD_REPLACE) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + if (samdb_msg_add_hashes(msg, msg, "sambaLMPwdHistory", new_lm_history, lm_hist_len) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + + return LDB_SUCCESS; +} + +static struct ldb_async_handle *ph_init_handle(struct ldb_request *req, struct ldb_module *module, enum ph_type type) +{ + struct ph_async_context *ac; + struct ldb_async_handle *h; + + h = talloc_zero(req, struct ldb_async_handle); + if (h == NULL) { + ldb_set_errstring(module->ldb, talloc_asprintf(module, "Out of Memory")); + return NULL; + } + + h->module = module; + + ac = talloc_zero(h, struct ph_async_context); + if (ac == NULL) { + ldb_set_errstring(module->ldb, talloc_asprintf(module, "Out of Memory")); + talloc_free(h); + return NULL; + } + + h->private_data = (void *)ac; + + h->state = LDB_ASYNC_INIT; + h->status = LDB_SUCCESS; + + ac->type = type; + ac->module = module; + ac->orig_req = req; + + return h; +} + +static int get_domain_data_callback(struct ldb_context *ldb, void *context, struct ldb_async_result *ares) +{ + struct ph_async_context *ac; + + if (!context || !ares) { + ldb_set_errstring(ldb, talloc_asprintf(ldb, "NULL Context or Result in callback")); + return LDB_ERR_OPERATIONS_ERROR; + } + + ac = talloc_get_type(context, struct ph_async_context); + + /* we are interested only in the single reply (base search) we receive here */ + if (ares->type == LDB_REPLY_ENTRY) { + if (ac->dom_res != NULL) { + ldb_set_errstring(ldb, talloc_asprintf(ldb, "Too many results")); + talloc_free(ares); + return LDB_ERR_OPERATIONS_ERROR; + } + ac->dom_res = talloc_steal(ac, ares); + } else { + talloc_free(ares); + } + + return LDB_SUCCESS; +} + +static int build_domain_data_request(struct ph_async_context *ac, + struct dom_sid *sid) +{ + const char * const attrs[] = { "pwdProperties", "pwdHistoryLength", "dnsDomain", NULL }; + char *filter; + + ac->dom_req = talloc_zero(ac, struct ldb_request); + if (ac->dom_req == NULL) { + ldb_debug(ac->module->ldb, LDB_DEBUG_ERROR, "Out of Memory!\n"); + return LDB_ERR_OPERATIONS_ERROR; + } + ac->dom_req->operation = LDB_ASYNC_SEARCH; + ac->dom_req->op.search.base = NULL; + ac->dom_req->op.search.scope = LDB_SCOPE_SUBTREE; + + filter = talloc_asprintf(ac->dom_req, "(&(objectSid=%s)(objectClass=domain))", dom_sid_string(ac->dom_req, sid)); + if (filter == NULL) { + ldb_debug(ac->module->ldb, LDB_DEBUG_ERROR, "Out of Memory!\n"); + talloc_free(ac->dom_req); + return LDB_ERR_OPERATIONS_ERROR; + } + + ac->dom_req->op.search.tree = ldb_parse_tree(ac->module->ldb, filter); + if (ac->dom_req->op.search.tree == NULL) { + ldb_set_errstring(ac->module->ldb, talloc_asprintf(ac, "Invalid search filter")); + talloc_free(ac->dom_req); + return LDB_ERR_OPERATIONS_ERROR; + } + ac->dom_req->op.search.attrs = attrs; + ac->dom_req->controls = NULL; + ac->dom_req->creds = ac->orig_req->creds; + ac->dom_req->async.context = ac; + ac->dom_req->async.callback = get_domain_data_callback; + ac->dom_req->async.timeout = ac->orig_req->async.timeout; + + return LDB_SUCCESS; +} + +static struct domain_data *get_domain_data(struct ldb_module *module, void *mem_ctx, struct ldb_async_result *res) +{ + struct domain_data *data; + const char *tmp; + + data = talloc_zero(mem_ctx, struct domain_data); + if (data == NULL) { + return NULL; + } + + data->pwdProperties = samdb_result_uint(res->message, "pwdProperties", 0); + data->pwdHistoryLength = samdb_result_uint(res->message, "pwdHistoryLength", 0); + tmp = ldb_msg_find_string(res->message, "dnsDomain", NULL); + + if (tmp != NULL) { + data->dnsDomain = talloc_strdup(data, tmp); + if (data->dnsDomain == NULL) { + ldb_debug(module->ldb, LDB_DEBUG_ERROR, "Out of memory!\n"); + return NULL; + } + data->realm = strupper_talloc(mem_ctx, tmp); + if (data->realm == NULL) { + ldb_debug(module->ldb, LDB_DEBUG_ERROR, "Out of memory!\n"); + return NULL; + } + } + + return data; +} + +static int password_hash_add_async(struct ldb_module *module, struct ldb_request *req) +{ + struct ldb_async_handle *h; + struct ph_async_context *ac; + struct ldb_message_element *attribute; + struct dom_sid *domain_sid; + int ret; + + ldb_debug(module->ldb, LDB_DEBUG_TRACE, "password_hash_add_async\n"); + + if (ldb_dn_is_special(req->op.add.message->dn)) { /* do not manipulate our control entries */ + return ldb_next_request(module, req); + } + + /* nobody must touch password Histories */ + if (ldb_msg_find_element(req->op.add.message, "sambaNTPwdHistory") || + ldb_msg_find_element(req->op.add.message, "sambaLMPwdHistory")) { + return LDB_ERR_UNWILLING_TO_PERFORM; + } + + /* If no part of this touches the sambaPassword, then we don't + * need to make any changes. For password changes/set there should + * be a 'delete' or a 'modify' on this attribute. */ + if ((attribute = ldb_msg_find_element(req->op.add.message, "sambaPassword")) == NULL ) { + return ldb_next_request(module, req); + } + + /* if it is not an entry of type person its an error */ + /* TODO: remove this when sambaPassword will be in schema */ + if (!ldb_msg_check_string_attribute(req->op.add.message, "objectClass", "person")) { + return LDB_ERR_OBJECT_CLASS_VIOLATION; + } + + /* check sambaPassword is single valued here */ + /* TODO: remove this when sambaPassword will be single valued in schema */ + if (attribute->num_values > 1) { + ldb_set_errstring(module->ldb, talloc_asprintf(req, + "mupltiple values for sambaPassword not allowed!\n")); + return LDB_ERR_CONSTRAINT_VIOLATION; + } + + /* get user domain data */ + domain_sid = samdb_result_sid_prefix(req, req->op.add.message, "objectSid"); + if (domain_sid == NULL) { + ldb_debug(module->ldb, LDB_DEBUG_ERROR, "can't handle entry with missing objectSid!\n"); + return LDB_ERR_OPERATIONS_ERROR; + } + + h = ph_init_handle(req, module, PH_ADD); + if (!h) { + return LDB_ERR_OPERATIONS_ERROR; + } + ac = talloc_get_type(h->private_data, struct ph_async_context); + + ret = build_domain_data_request(ac, domain_sid); + if (ret != LDB_SUCCESS) { + return ret; + } + + ac->step = PH_ADD_SEARCH_DOM; + + req->async.handle = h; + + return ldb_next_request(module, ac->dom_req); +} + +static int password_hash_add_async_do_add(struct ldb_async_handle *h) { + + struct ph_async_context *ac; + struct domain_data *domain; + struct smb_krb5_context *smb_krb5_context; + struct ldb_message *msg; + + ac = talloc_get_type(h->private_data, struct ph_async_context); + + domain = get_domain_data(ac->module, ac, ac->search_res); + if (domain == NULL) { + return LDB_ERR_OPERATIONS_ERROR; + } + + ac->down_req = talloc(ac, struct ldb_request); + if (ac->down_req == NULL) { + return LDB_ERR_OPERATIONS_ERROR; + } + + *(ac->down_req) = *(ac->orig_req); + ac->down_req->op.add.message = msg = ldb_msg_copy_shallow(ac->down_req, ac->orig_req->op.add.message); + if (ac->down_req->op.add.message == NULL) { + return LDB_ERR_OPERATIONS_ERROR; + } + + /* Some operations below require kerberos contexts */ + if (smb_krb5_init_context(ac->down_req, &smb_krb5_context) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + + /* we can compute new password hashes from the unicode password */ + if (add_password_hashes(ac->module, msg, 0) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + + /* now add krb5 keys based on unicode password */ + if (add_krb5_keys_from_password(ac->module, msg, smb_krb5_context, domain, + ldb_msg_find_string(msg, "samAccountName", NULL), + ldb_msg_find_string(msg, "userPrincipalName", NULL), + ldb_msg_check_string_attribute(msg, "objectClass", "computer") + ) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + + /* add also kr5 keys based on NT the hash */ + if (add_krb5_keys_from_NThash(ac->module, msg, smb_krb5_context) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + + /* if both the domain properties and the user account controls do not permit + * clear text passwords then wipe out the sambaPassword */ + if ((!(domain->pwdProperties & DOMAIN_PASSWORD_STORE_CLEARTEXT)) || + (!(ldb_msg_find_uint(msg, "userAccountControl", 0) & UF_ENCRYPTED_TEXT_PASSWORD_ALLOWED))) { + ldb_msg_remove_attr(msg, "sambaPassword"); + } + + /* don't touch it if a value is set. It could be an incoming samsync */ + if (ldb_msg_find_uint64(msg, "pwdLastSet", 0) == 0) { + if (set_pwdLastSet(ac->module, msg) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + } + + /* don't touch it if a value is set. It could be an incoming samsync */ + if (!ldb_msg_find_element(msg, "msDS-KeyVersionNumber")) { + if (add_keyVersionNumber(ac->module, msg, 0) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + } + + h->state = LDB_ASYNC_INIT; + h->status = LDB_SUCCESS; + + ac->step = PH_ADD_DO_ADD; + + /* perform the operation */ + return ldb_next_request(ac->module, ac->down_req); +} + +static int password_hash_mod_async_search_self(struct ldb_async_handle *h); + +static int password_hash_modify_async(struct ldb_module *module, struct ldb_request *req) +{ + struct ldb_async_handle *h; + struct ph_async_context *ac; + struct ldb_message_element *sambaAttr; + struct ldb_message_element *ntAttr; + struct ldb_message_element *lmAttr; + + ldb_debug(module->ldb, LDB_DEBUG_TRACE, "password_hash_add_async\n"); + + if (ldb_dn_is_special(req->op.mod.message->dn)) { /* do not manipulate our control entries */ + return ldb_next_request(module, req); + } + + /* nobody must touch password Histories */ + if (ldb_msg_find_element(req->op.mod.message, "sambaNTPwdHistory") || + ldb_msg_find_element(req->op.mod.message, "sambaLMPwdHistory")) { + return LDB_ERR_UNWILLING_TO_PERFORM; + } + + sambaAttr = ldb_msg_find_element(req->op.mod.message, "sambaPassword"); + ntAttr = ldb_msg_find_element(req->op.mod.message, "ntPwdHash"); + lmAttr = ldb_msg_find_element(req->op.mod.message, "lmPwdHash"); + + /* check passwords are single valued here */ + /* TODO: remove this when passwords will be single valued in schema */ + if (sambaAttr && (sambaAttr->num_values > 1)) { + return LDB_ERR_CONSTRAINT_VIOLATION; + } + if (ntAttr && (ntAttr->num_values > 1)) { + return LDB_ERR_CONSTRAINT_VIOLATION; + } + if (lmAttr && (lmAttr->num_values > 1)) { + return LDB_ERR_CONSTRAINT_VIOLATION; + } + + /* If no part of this touches the sambaPassword OR ntPwdHash and/or lmPwdHash, then we don't + * need to make any changes. For password changes/set there should + * be a 'delete' or a 'modify' on this attribute. */ + /* If the only operation is the deletion of the passwords then go on */ + if ( ((!sambaAttr) || ((sambaAttr->flags & LDB_FLAG_MOD_MASK) == LDB_FLAG_MOD_DELETE)) + && ((!ntAttr) || ((ntAttr->flags & LDB_FLAG_MOD_MASK) == LDB_FLAG_MOD_DELETE)) + && ((!lmAttr) || ((lmAttr->flags & LDB_FLAG_MOD_MASK) == LDB_FLAG_MOD_DELETE)) ) { + + return ldb_next_request(module, req); + } + + h = ph_init_handle(req, module, PH_MOD); + if (!h) { + return LDB_ERR_OPERATIONS_ERROR; + } + ac = talloc_get_type(h->private_data, struct ph_async_context); + + /* return or own handle to deal with this call */ + req->async.handle = h; + + /* prepare the first operation */ + ac->down_req = talloc_zero(ac, struct ldb_request); + if (ac->down_req == NULL) { + ldb_set_errstring(module->ldb, talloc_asprintf(module->ldb, "Out of memory!")); + return LDB_ERR_OPERATIONS_ERROR; + } + + *(ac->down_req) = *req; /* copy the request */ + + /* use a new message structure so that we can modify it */ + ac->down_req->op.mod.message = ldb_msg_copy_shallow(ac->down_req, req->op.mod.message); + + /* - remove any imodification to the password from the first commit + * we will make the real modification later */ + if (sambaAttr) ldb_msg_remove_attr(ac->down_req->op.mod.message, "sambaPassword"); + if (ntAttr) ldb_msg_remove_attr(ac->down_req->op.mod.message, "ntPwdHash"); + if (lmAttr) ldb_msg_remove_attr(ac->down_req->op.mod.message, "lmPwdHash"); + + /* if there was nothing else to be modify skip to next step */ + if (ac->down_req->op.mod.message->num_elements == 0) { + talloc_free(ac->down_req); + ac->down_req = NULL; + return password_hash_mod_async_search_self(h); + } + + ac->down_req->async.context = NULL; + ac->down_req->async.callback = NULL; + + ac->step = PH_MOD_DO_REQ; + + return ldb_next_request(module, ac->down_req); +} + +static int get_self_callback(struct ldb_context *ldb, void *context, struct ldb_async_result *ares) +{ + struct ph_async_context *ac; + + if (!context || !ares) { + ldb_set_errstring(ldb, talloc_asprintf(ldb, "NULL Context or Result in callback")); + return LDB_ERR_OPERATIONS_ERROR; + } + + ac = talloc_get_type(context, struct ph_async_context); + + /* we are interested only in the single reply (base search) we receive here */ + if (ares->type == LDB_REPLY_ENTRY) { + if (ac->search_res != NULL) { + ldb_set_errstring(ldb, talloc_asprintf(ldb, "Too many results")); + talloc_free(ares); + return LDB_ERR_OPERATIONS_ERROR; + } + + /* if it is not an entry of type person this is an error */ + /* TODO: remove this when sambaPassword will be in schema */ + if (!ldb_msg_check_string_attribute(ares->message, "objectClass", "person")) { + ldb_set_errstring(ldb, talloc_asprintf(ldb, "Object class violation")); + talloc_free(ares); + return LDB_ERR_OBJECT_CLASS_VIOLATION; + } + + ac->search_res = talloc_steal(ac, ares); + } else { + talloc_free(ares); + } + + return LDB_SUCCESS; +} + +static int password_hash_mod_async_search_self(struct ldb_async_handle *h) { + + struct ph_async_context *ac; + + ac = talloc_get_type(h->private_data, struct ph_async_context); + + /* prepare the search operation */ + ac->search_req = talloc_zero(ac, struct ldb_request); + if (ac->search_req == NULL) { + ldb_debug(ac->module->ldb, LDB_DEBUG_ERROR, "Out of Memory!\n"); + return LDB_ERR_OPERATIONS_ERROR; + } + + ac->search_req->operation = LDB_ASYNC_SEARCH; + ac->search_req->op.search.base = ac->orig_req->op.mod.message->dn; + ac->search_req->op.search.scope = LDB_SCOPE_BASE; + ac->search_req->op.search.tree = ldb_parse_tree(ac->module->ldb, NULL); + if (ac->search_req->op.search.tree == NULL) { + ldb_set_errstring(ac->module->ldb, talloc_asprintf(ac, "Invalid search filter")); + return LDB_ERR_OPERATIONS_ERROR; + } + ac->search_req->op.search.attrs = NULL; + ac->search_req->controls = NULL; + ac->search_req->creds = ac->orig_req->creds; + ac->search_req->async.context = ac; + ac->search_req->async.callback = get_self_callback; + ac->search_req->async.timeout = ac->orig_req->async.timeout; + + ac->step = PH_MOD_SEARCH_SELF; + + return ldb_next_request(ac->module, ac->search_req); +} + +static int password_hash_mod_async_search_dom(struct ldb_async_handle *h) { + + struct ph_async_context *ac; + struct dom_sid *domain_sid; + int ret; + + ac = talloc_get_type(h->private_data, struct ph_async_context); + + /* get object domain sid */ + domain_sid = samdb_result_sid_prefix(ac, ac->search_res->message, "objectSid"); + if (domain_sid == NULL) { + ldb_debug(ac->module->ldb, LDB_DEBUG_ERROR, "can't handle entry with missing objectSid!\n"); + return LDB_ERR_OPERATIONS_ERROR; + } + + /* get user domain data */ + ret = build_domain_data_request(ac, domain_sid); + if (ret != LDB_SUCCESS) { + return ret; + } + + ac->step = PH_MOD_SEARCH_DOM; + + return ldb_next_request(ac->module, ac->dom_req); +} + +static int password_hash_mod_async_do_mod(struct ldb_async_handle *h) { + + struct ph_async_context *ac; + struct domain_data *domain; + struct smb_krb5_context *smb_krb5_context; + struct ldb_message_element *sambaAttr; + struct ldb_message *msg; + int phlen; + + ac = talloc_get_type(h->private_data, struct ph_async_context); + + domain = get_domain_data(ac->module, ac, ac->dom_res); + if (domain == NULL) { + return LDB_ERR_OPERATIONS_ERROR; + } + + ac->mod_req = talloc(ac, struct ldb_request); + if (ac->mod_req == NULL) { + return LDB_ERR_OPERATIONS_ERROR; + } + + *(ac->mod_req) = *(ac->orig_req); + + /* use a new message structure so that we can modify it */ + ac->mod_req->op.mod.message = msg = ldb_msg_new(ac->mod_req); + if (msg == NULL) { + return LDB_ERR_OPERATIONS_ERROR; + } + + /* modify dn */ + msg->dn = ac->orig_req->op.mod.message->dn; + + /* Some operations below require kerberos contexts */ + if (smb_krb5_init_context(ac->mod_req, &smb_krb5_context) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + + /* we are going to replace the existing krb5key or delete it */ + if (ldb_msg_add_empty(msg, "krb5key", LDB_FLAG_MOD_REPLACE) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + + /* if we have sambaPassword in the original message add the operatio on it here */ + sambaAttr = ldb_msg_find_element(ac->orig_req->op.mod.message, "sambaPassword"); + if (sambaAttr) { + + if (ldb_msg_add(msg, sambaAttr, sambaAttr->flags) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + + /* we are not deleteing it add password hashes */ + if ((sambaAttr->flags & LDB_FLAG_MOD_MASK) != LDB_FLAG_MOD_DELETE) { + + /* we can compute new password hashes from the unicode password */ + if (add_password_hashes(ac->module, msg, 1) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + + /* now add krb5 keys based on unicode password */ + if (add_krb5_keys_from_password(ac->module, msg, smb_krb5_context, domain, + ldb_msg_find_string(ac->search_res->message, "samAccountName", NULL), + ldb_msg_find_string(ac->search_res->message, "userPrincipalName", NULL), + ldb_msg_check_string_attribute(ac->search_res->message, "objectClass", "computer") + ) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + + /* if the domain properties or the user account controls do not permit + * clear text passwords then wipe out the sambaPassword */ + if ((!(domain->pwdProperties & DOMAIN_PASSWORD_STORE_CLEARTEXT)) || + (!(ldb_msg_find_uint(ac->search_res->message, "userAccountControl", 0) & UF_ENCRYPTED_TEXT_PASSWORD_ALLOWED))) { + ldb_msg_remove_attr(msg, "sambaPassword"); + } + + } + } + + /* if we don't have sambaPassword or we are trying to delete it try with nt or lm hasehs */ + if ((!sambaAttr) || ((sambaAttr->flags & LDB_FLAG_MOD_MASK) == LDB_FLAG_MOD_DELETE)) { + struct ldb_message_element *el; + + el = ldb_msg_find_element(ac->orig_req->op.mod.message, "ntPwdHash"); + if (ldb_msg_add(msg, el, el->flags) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + + el = ldb_msg_find_element(ac->orig_req->op.mod.message, "lmPwdHash"); + if (ldb_msg_add(msg, el, el->flags) != 0) { + return LDB_ERR_OPERATIONS_ERROR; + } + } + + /* add also kr5 keys based on NT the hash */ + if (add_krb5_keys_from_NThash(ac->module, msg, smb_krb5_context) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + + /* set change time */ + if (set_pwdLastSet(ac->module, msg) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + + /* don't touch it if a value is set. It could be an incoming samsync */ + if (add_keyVersionNumber(ac->module, msg, + ldb_msg_find_uint(msg, "msDS-KeyVersionNumber", 0) + ) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + + if ((phlen = samdb_result_uint(ac->dom_res->message, "pwdHistoryLength", 0)) > 0) { + if (setPwdHistory(ac->module, msg, ac->search_res->message, phlen) != LDB_SUCCESS) { + return LDB_ERR_OPERATIONS_ERROR; + } + } + + h->state = LDB_ASYNC_INIT; + h->status = LDB_SUCCESS; + + ac->step = PH_MOD_DO_MOD; + + /* perform the search */ + return ldb_next_request(ac->module, ac->mod_req); +} + +static int ph_async_wait(struct ldb_async_handle *handle) { + struct ph_async_context *ac; + int ret; + + if (!handle || !handle->private_data) { + return LDB_ERR_OPERATIONS_ERROR; + } + + if (handle->state == LDB_ASYNC_DONE) { + return handle->status; + } + + handle->state = LDB_ASYNC_PENDING; + + ac = talloc_get_type(handle->private_data, struct ph_async_context); + + switch (ac->step) { + case PH_ADD_SEARCH_DOM: + if (ac->dom_req->async.handle->status != LDB_ASYNC_DONE) { + ret = ldb_async_wait(ac->dom_req->async.handle, LDB_WAIT_NONE); + if (ret != LDB_SUCCESS) goto error; + + if (ac->dom_req->async.handle->status != LDB_ASYNC_DONE) { + return LDB_SUCCESS; + } + } + + /* domain search done, go on */ + return password_hash_add_async_do_add(handle); + + case PH_ADD_DO_ADD: + if (ac->down_req->async.handle->status != LDB_ASYNC_DONE) { + ret = ldb_async_wait(ac->down_req->async.handle, LDB_WAIT_NONE); + if (ret != LDB_SUCCESS) goto error; + + if (ac->down_req->async.handle->status != LDB_ASYNC_DONE) { + return LDB_SUCCESS; + } + } + return LDB_SUCCESS; + + case PH_MOD_DO_REQ: + if (ac->down_req->async.handle->status != LDB_ASYNC_DONE) { + ret = ldb_async_wait(ac->down_req->async.handle, LDB_WAIT_NONE); + if (ret != LDB_SUCCESS) goto error; + + if (ac->down_req->async.handle->status != LDB_ASYNC_DONE) { + return LDB_SUCCESS; + } + } + + /* non-password mods done, go on */ + return password_hash_mod_async_search_self(handle); + + case PH_MOD_SEARCH_SELF: + if (ac->search_req->async.handle->status != LDB_ASYNC_DONE) { + ret = ldb_async_wait(ac->search_req->async.handle, LDB_WAIT_NONE); + if (ret != LDB_SUCCESS) goto error; + + if (ac->search_req->async.handle->status != LDB_ASYNC_DONE) { + return LDB_SUCCESS; + } + } + + /* self search done, go on */ + return password_hash_mod_async_search_dom(handle); + + case PH_MOD_SEARCH_DOM: + if (ac->dom_req->async.handle->status != LDB_ASYNC_DONE) { + ret = ldb_async_wait(ac->dom_req->async.handle, LDB_WAIT_NONE); + if (ret != LDB_SUCCESS) goto error; + + if (ac->dom_req->async.handle->status != LDB_ASYNC_DONE) { + return LDB_SUCCESS; + } + } + + /* domain search done, go on */ + return password_hash_mod_async_do_mod(handle); + + case PH_MOD_DO_MOD: + if (ac->mod_req->async.handle->status != LDB_ASYNC_DONE) { + ret = ldb_async_wait(ac->mod_req->async.handle, LDB_WAIT_NONE); + if (ret != LDB_SUCCESS) goto error; + + if (ac->mod_req->async.handle->status != LDB_ASYNC_DONE) { + return LDB_SUCCESS; + } + } + return LDB_SUCCESS; + + default: + ret = LDB_ERR_OPERATIONS_ERROR; + goto error; + } + +error: + handle->state = LDB_ASYNC_DONE; + handle->status = ret; + return ret; +} + +static int ph_async_wait_all(struct ldb_async_handle *handle) { + + int ret; + + while (handle->state != LDB_ASYNC_DONE) { + ret = ph_async_wait(handle); + if (ret != LDB_SUCCESS) { + return ret; + } + } + + return handle->status; +} + +static int password_hash_async_wait(struct ldb_async_handle *handle, enum ldb_async_wait_type type) +{ + if (type == LDB_WAIT_ALL) { + return ph_async_wait_all(handle); + } else { + return ph_async_wait(handle); + } +} + static int password_hash_request(struct ldb_module *module, struct ldb_request *req) { switch (req->operation) { @@ -731,6 +1789,12 @@ static int password_hash_request(struct ldb_module *module, struct ldb_request * case LDB_REQ_MODIFY: return password_hash_modify(module, req); + case LDB_ASYNC_ADD: + return password_hash_add_async(module, req); + + case LDB_ASYNC_MODIFY: + return password_hash_modify_async(module, req); + default: return ldb_next_request(module, req); @@ -739,7 +1803,8 @@ static int password_hash_request(struct ldb_module *module, struct ldb_request * static const struct ldb_module_ops password_hash_ops = { .name = "password_hash", - .request = password_hash_request + .request = password_hash_request, + .async_wait = password_hash_async_wait }; -- cgit