source4/libnet/libnet_join.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3
   4    Copyright (C) Stefan Metzmacher      2004
   5    Copyright (C) Andrew Bartlett <abartlet@samba.org> 2005
   6    Copyright (C) Brad Henry 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    You should have received a copy of the GNU General Public License
  19    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  20 */
  21
  22 #include "includes.h"
  23 #include "libnet/libnet.h"
  24 #include "librpc/gen_ndr/ndr_drsuapi_c.h"
  25 #include "lib/ldb/include/ldb.h"
  26 #include "lib/ldb/include/ldb_errors.h"
  27 #include "param/secrets.h"
  28 #include "dsdb/samdb/samdb.h"
  29 #include "ldb_wrap.h"
  30 #include "util/util_ldb.h"
  31 #include "libcli/security/security.h"
  32 #include "auth/credentials/credentials.h"
  33 #include "auth/credentials/credentials_krb5.h"
  34 #include "librpc/gen_ndr/ndr_samr_c.h"
  35 #include "param/param.h"
  36
  37 /*
  38  * complete a domain join, when joining to a AD domain:
  39  * 1.) connect and bind to the DRSUAPI pipe
  40  * 2.) do a DsCrackNames() to find the machine account dn
  41  * 3.) connect to LDAP
  42  * 4.) do an ldap search to find the "msDS-KeyVersionNumber" of the machine account
  43  * 5.) set the servicePrincipalName's of the machine account via LDAP, (maybe we should use DsWriteAccountSpn()...)
  44  * 6.) do a DsCrackNames() to find the domain dn
  45  * 7.) find out Site specific stuff, look at libnet_JoinSite() for details
  46  */
  47 static NTSTATUS libnet_JoinADSDomain(struct libnet_context *ctx, struct libnet_JoinDomain *r)
  48 {
  49         NTSTATUS status;
  50
  51         TALLOC_CTX *tmp_ctx;
  52
  53         const char *realm = r->out.realm;
  54
  55         struct dcerpc_binding *samr_binding = r->out.samr_binding;
  56
  57         struct dcerpc_pipe *drsuapi_pipe;
  58         struct dcerpc_binding *drsuapi_binding;
  59         struct drsuapi_DsBind r_drsuapi_bind;
  60         struct drsuapi_DsCrackNames r_crack_names;
  61         struct drsuapi_DsNameString names[1];
  62         struct policy_handle drsuapi_bind_handle;
  63         struct GUID drsuapi_bind_guid;
  64
  65         struct ldb_context *remote_ldb;
  66         struct ldb_dn *account_dn;
  67         const char *account_dn_str;
  68         const char *remote_ldb_url;
  69         struct ldb_result *res;
  70         struct ldb_message *msg;
  71
  72         int ret, rtn;
  73
  74         const char * const attrs[] = {
  75                 "msDS-KeyVersionNumber",
  76                 "servicePrincipalName",
  77                 "dNSHostName",
  78                 "objectGUID",
  79                 NULL,
  80         };
  81
  82         r->out.error_string = NULL;
  83
  84         /* We need to convert between a samAccountName and domain to a
  85          * DN in the directory.  The correct way to do this is with
  86          * DRSUAPI CrackNames */
  87
  88         /* Fiddle with the bindings, so get to DRSUAPI on
  89          * NCACN_IP_TCP, sealed */
  90         tmp_ctx = talloc_named(r, 0, "libnet_JoinADSDomain temp context");
  91         if (!tmp_ctx) {
  92                 r->out.error_string = NULL;
  93                 return NT_STATUS_NO_MEMORY;
  94         }
  95
  96         drsuapi_binding = talloc(tmp_ctx, struct dcerpc_binding);
  97         if (!drsuapi_binding) {
  98                 r->out.error_string = NULL;
  99                 talloc_free(tmp_ctx);
 100                 return NT_STATUS_NO_MEMORY;
 101         }
 102
 103         *drsuapi_binding = *samr_binding;
 104
 105         /* DRSUAPI is only available on IP_TCP, and locally on NCALRPC */
 106         if (drsuapi_binding->transport != NCALRPC) {
 107                 drsuapi_binding->transport = NCACN_IP_TCP;
 108         }
 109         drsuapi_binding->endpoint = NULL;
 110         drsuapi_binding->flags |= DCERPC_SEAL;
 111
 112         status = dcerpc_pipe_connect_b(tmp_ctx,
 113                                        &drsuapi_pipe,
 114                                        drsuapi_binding,
 115                                        &ndr_table_drsuapi,
 116                                        ctx->cred,
 117                                        ctx->event_ctx,
 118                                        ctx->lp_ctx);
 119         if (!NT_STATUS_IS_OK(status)) {
 120                 r->out.error_string = talloc_asprintf(r,
 121                                         "Connection to DRSUAPI pipe of PDC of domain '%s' failed: %s",
 122                                         r->out.domain_name,
 123                                         nt_errstr(status));
 124                 talloc_free(tmp_ctx);
 125                 return status;
 126         }
 127
 128         /* get a DRSUAPI pipe handle */
 129         GUID_from_string(DRSUAPI_DS_BIND_GUID, &drsuapi_bind_guid);
 130
 131         r_drsuapi_bind.in.bind_guid = &drsuapi_bind_guid;
 132         r_drsuapi_bind.in.bind_info = NULL;
 133         r_drsuapi_bind.out.bind_handle = &drsuapi_bind_handle;
 134
 135         status = dcerpc_drsuapi_DsBind(drsuapi_pipe, tmp_ctx, &r_drsuapi_bind);
 136         if (!NT_STATUS_IS_OK(status)) {
 137                 if (NT_STATUS_EQUAL(status, NT_STATUS_NET_WRITE_FAULT)) {
 138                         r->out.error_string
 139                                 = talloc_asprintf(r,
 140                                                   "dcerpc_drsuapi_DsBind failed - %s",
 141                                                   dcerpc_errstr(tmp_ctx, drsuapi_pipe->last_fault_code));
 142                         talloc_free(tmp_ctx);
 143                         return status;
 144                 } else {
 145                         r->out.error_string
 146                                 = talloc_asprintf(r,
 147                                                   "dcerpc_drsuapi_DsBind failed - %s",
 148                                                   nt_errstr(status));
 149                         talloc_free(tmp_ctx);
 150                         return status;
 151                 }
 152         } else if (!W_ERROR_IS_OK(r_drsuapi_bind.out.result)) {
 153                 r->out.error_string
 154                                 = talloc_asprintf(r,
 155                                                   "DsBind failed - %s",
 156                                                   win_errstr(r_drsuapi_bind.out.result));
 157                         talloc_free(tmp_ctx);
 158                 return NT_STATUS_UNSUCCESSFUL;
 159         }
 160
 161         /* Actually 'crack' the names */
 162         ZERO_STRUCT(r_crack_names);
 163         r_crack_names.in.bind_handle            = &drsuapi_bind_handle;
 164         r_crack_names.in.level                  = 1;
 165         r_crack_names.in.req.req1.codepage      = 1252; /* western european */
 166         r_crack_names.in.req.req1.language      = 0x00000407; /* german */
 167         r_crack_names.in.req.req1.count         = 1;
 168         r_crack_names.in.req.req1.names         = names;
 169         r_crack_names.in.req.req1.format_flags  = DRSUAPI_DS_NAME_FLAG_NO_FLAGS;
 170         r_crack_names.in.req.req1.format_offered= DRSUAPI_DS_NAME_FORMAT_SID_OR_SID_HISTORY;
 171         r_crack_names.in.req.req1.format_desired= DRSUAPI_DS_NAME_FORMAT_FQDN_1779;
 172         names[0].str = dom_sid_string(tmp_ctx, r->out.account_sid);
 173         if (!names[0].str) {
 174                 r->out.error_string = NULL;
 175                 talloc_free(tmp_ctx);
 176                 return NT_STATUS_NO_MEMORY;
 177         }
 178
 179         status = dcerpc_drsuapi_DsCrackNames(drsuapi_pipe, tmp_ctx, &r_crack_names);
 180         if (!NT_STATUS_IS_OK(status)) {
 181                 if (NT_STATUS_EQUAL(status, NT_STATUS_NET_WRITE_FAULT)) {
 182                         r->out.error_string
 183                                 = talloc_asprintf(r,
 184                                                   "dcerpc_drsuapi_DsCrackNames for [%s] failed - %s",
 185                                                   names[0].str,
 186                                                   dcerpc_errstr(tmp_ctx, drsuapi_pipe->last_fault_code));
 187                         talloc_free(tmp_ctx);
 188                         return status;
 189                 } else {
 190                         r->out.error_string
 191                                 = talloc_asprintf(r,
 192                                                   "dcerpc_drsuapi_DsCrackNames for [%s] failed - %s",
 193                                                   names[0].str,
 194                                                   nt_errstr(status));
 195                         talloc_free(tmp_ctx);
 196                         return status;
 197                 }
 198         } else if (!W_ERROR_IS_OK(r_crack_names.out.result)) {
 199                 r->out.error_string
 200                                 = talloc_asprintf(r,
 201                                                   "DsCrackNames failed - %s", win_errstr(r_crack_names.out.result));
 202                 talloc_free(tmp_ctx);
 203                 return NT_STATUS_UNSUCCESSFUL;
 204         } else if (r_crack_names.out.level != 1
 205                    || !r_crack_names.out.ctr.ctr1
 206                    || r_crack_names.out.ctr.ctr1->count != 1) {
 207                 r->out.error_string = talloc_asprintf(r, "DsCrackNames failed");
 208                 talloc_free(tmp_ctx);
 209                 return NT_STATUS_INVALID_PARAMETER;
 210         } else if (r_crack_names.out.ctr.ctr1->array[0].status != DRSUAPI_DS_NAME_STATUS_OK) {
 211                 r->out.error_string = talloc_asprintf(r, "DsCrackNames failed: %d", r_crack_names.out.ctr.ctr1->array[0].status);
 212                 talloc_free(tmp_ctx);
 213                 return NT_STATUS_UNSUCCESSFUL;
 214         } else if (r_crack_names.out.ctr.ctr1->array[0].result_name == NULL) {
 215                 r->out.error_string = talloc_asprintf(r, "DsCrackNames failed: no result name");
 216                 talloc_free(tmp_ctx);
 217                 return NT_STATUS_INVALID_PARAMETER;
 218         }
 219
 220         /* Store the DN of our machine account. */
 221         account_dn_str = r_crack_names.out.ctr.ctr1->array[0].result_name;
 222
 223         /* Now we know the user's DN, open with LDAP, read and modify a few things */
 224
 225         remote_ldb_url = talloc_asprintf(tmp_ctx, "ldap://%s",
 226                                          drsuapi_binding->target_hostname);
 227         if (!remote_ldb_url) {
 228                 r->out.error_string = NULL;
 229                 talloc_free(tmp_ctx);
 230                 return NT_STATUS_NO_MEMORY;
 231         }
 232
 233         remote_ldb = ldb_wrap_connect(tmp_ctx, ctx->lp_ctx,
 234                                       remote_ldb_url,
 235                                       NULL, ctx->cred, 0, NULL);
 236         if (!remote_ldb) {
 237                 r->out.error_string = NULL;
 238                 talloc_free(tmp_ctx);
 239                 return NT_STATUS_UNSUCCESSFUL;
 240         }
 241
 242         account_dn = ldb_dn_new(tmp_ctx, remote_ldb, account_dn_str);
 243         if (! ldb_dn_validate(account_dn)) {
 244                 r->out.error_string = talloc_asprintf(r, "Invalid account dn: %s",
 245                                                       account_dn_str);
 246                 talloc_free(tmp_ctx);
 247                 return NT_STATUS_UNSUCCESSFUL;
 248         }
 249
 250         /* search for the user's record */
 251         ret = ldb_search(remote_ldb, account_dn, LDB_SCOPE_BASE,
 252                          NULL, attrs, &res);
 253         if (ret != LDB_SUCCESS) {
 254                 r->out.error_string = talloc_asprintf(r, "ldb_search for %s failed - %s",
 255                                                       account_dn_str, ldb_errstring(remote_ldb));
 256                 talloc_free(tmp_ctx);
 257                 return NT_STATUS_UNSUCCESSFUL;
 258         }
 259
 260         talloc_steal(tmp_ctx, res);
 261
 262         if (res->count != 1) {
 263                 r->out.error_string = talloc_asprintf(r, "ldb_search for %s failed - found %d entries",
 264                                                       account_dn_str, res->count);
 265                 talloc_free(tmp_ctx);
 266                 return NT_STATUS_UNSUCCESSFUL;
 267         }
 268
 269         /* Prepare a new message, for the modify */
 270         msg = ldb_msg_new(tmp_ctx);
 271         if (!msg) {
 272                 r->out.error_string = NULL;
 273                 talloc_free(tmp_ctx);
 274                 return NT_STATUS_NO_MEMORY;
 275         }
 276         msg->dn = res->msgs[0]->dn;
 277
 278         {
 279                 int i;
 280                 const char *service_principal_name[6];
 281                 const char *dns_host_name = strlower_talloc(tmp_ctx,
 282                                                             talloc_asprintf(tmp_ctx,
 283                                                                             "%s.%s",
 284                                                                             r->in.netbios_name,
 285                                                                             realm));
 286
 287                 if (!dns_host_name) {
 288                         r->out.error_string = NULL;
 289                         talloc_free(tmp_ctx);
 290                         return NT_STATUS_NO_MEMORY;
 291                 }
 292
 293                 service_principal_name[0] = talloc_asprintf(tmp_ctx, "host/%s", dns_host_name);
 294                 service_principal_name[1] = talloc_asprintf(tmp_ctx, "host/%s", strlower_talloc(tmp_ctx, r->in.netbios_name));
 295                 service_principal_name[2] = talloc_asprintf(tmp_ctx, "host/%s/%s", dns_host_name, realm);
 296                 service_principal_name[3] = talloc_asprintf(tmp_ctx, "host/%s/%s", strlower_talloc(tmp_ctx, r->in.netbios_name), realm);
 297                 service_principal_name[4] = talloc_asprintf(tmp_ctx, "host/%s/%s", dns_host_name, r->out.domain_name);
 298                 service_principal_name[5] = talloc_asprintf(tmp_ctx, "host/%s/%s", strlower_talloc(tmp_ctx, r->in.netbios_name), r->out.domain_name);
 299
 300                 for (i=0; i < ARRAY_SIZE(service_principal_name); i++) {
 301                         if (!service_principal_name[i]) {
 302                                 r->out.error_string = NULL;
 303                                 talloc_free(tmp_ctx);
 304                                 return NT_STATUS_NO_MEMORY;
 305                         }
 306                         rtn = samdb_msg_add_string(remote_ldb, tmp_ctx, msg, "servicePrincipalName", service_principal_name[i]);
 307                         if (rtn == -1) {
 308                                 r->out.error_string = NULL;
 309                                 talloc_free(tmp_ctx);
 310                                 return NT_STATUS_NO_MEMORY;
 311                         }
 312                 }
 313
 314                 rtn = samdb_msg_add_string(remote_ldb, tmp_ctx, msg, "dNSHostName", dns_host_name);
 315                 if (rtn == -1) {
 316                         r->out.error_string = NULL;
 317                         talloc_free(tmp_ctx);
 318                         return NT_STATUS_NO_MEMORY;
 319                 }
 320
 321                 rtn = samdb_replace(remote_ldb, tmp_ctx, msg);
 322                 if (rtn != 0) {
 323                         r->out.error_string
 324                                 = talloc_asprintf(r,
 325                                                   "Failed to replace entries on %s",
 326                                                   ldb_dn_get_linearized(msg->dn));
 327                         talloc_free(tmp_ctx);
 328                         return NT_STATUS_INTERNAL_DB_CORRUPTION;
 329                 }
 330         }
 331
 332         /* DsCrackNames to find out the DN of the domain. */
 333         r_crack_names.in.req.req1.format_offered = DRSUAPI_DS_NAME_FORMAT_NT4_ACCOUNT;
 334         r_crack_names.in.req.req1.format_desired = DRSUAPI_DS_NAME_FORMAT_FQDN_1779;
 335         names[0].str = talloc_asprintf(tmp_ctx, "%s\\", r->out.domain_name);
 336         if (!names[0].str) {
 337                 r->out.error_string = NULL;
 338                 talloc_free(tmp_ctx);
 339                 return NT_STATUS_NO_MEMORY;
 340         }
 341
 342         status = dcerpc_drsuapi_DsCrackNames(drsuapi_pipe, tmp_ctx, &r_crack_names);
 343         if (!NT_STATUS_IS_OK(status)) {
 344                 if (NT_STATUS_EQUAL(status, NT_STATUS_NET_WRITE_FAULT)) {
 345                         r->out.error_string
 346                                 = talloc_asprintf(r,
 347                                                   "dcerpc_drsuapi_DsCrackNames for [%s] failed - %s",
 348                                                   r->in.domain_name,
 349                                                   dcerpc_errstr(tmp_ctx, drsuapi_pipe->last_fault_code));
 350                         talloc_free(tmp_ctx);
 351                         return status;
 352                 } else {
 353                         r->out.error_string
 354                                 = talloc_asprintf(r,
 355                                                   "dcerpc_drsuapi_DsCrackNames for [%s] failed - %s",
 356                                                   r->in.domain_name,
 357                                                   nt_errstr(status));
 358                         talloc_free(tmp_ctx);
 359                         return status;
 360                 }
 361         } else if (!W_ERROR_IS_OK(r_crack_names.out.result)) {
 362                 r->out.error_string
 363                         = talloc_asprintf(r,
 364                                           "DsCrackNames failed - %s", win_errstr(r_crack_names.out.result));
 365                 talloc_free(tmp_ctx);
 366                 return NT_STATUS_UNSUCCESSFUL;
 367         } else if (r_crack_names.out.level != 1
 368                    || !r_crack_names.out.ctr.ctr1
 369                    || r_crack_names.out.ctr.ctr1->count != 1
 370                    || !r_crack_names.out.ctr.ctr1->array[0].result_name
 371                    || r_crack_names.out.ctr.ctr1->array[0].status != DRSUAPI_DS_NAME_STATUS_OK) {
 372                 r->out.error_string = talloc_asprintf(r, "DsCrackNames failed");
 373                 talloc_free(tmp_ctx);
 374                 return NT_STATUS_UNSUCCESSFUL;
 375         }
 376
 377         /* Store the account DN. */
 378         r->out.account_dn_str = account_dn_str;
 379         talloc_steal(r, account_dn_str);
 380
 381         /* Store the domain DN. */
 382         r->out.domain_dn_str = r_crack_names.out.ctr.ctr1->array[0].result_name;
 383         talloc_steal(r, r_crack_names.out.ctr.ctr1->array[0].result_name);
 384
 385         /* Store the KVNO of the account, critical for some kerberos
 386          * operations */
 387         r->out.kvno = ldb_msg_find_attr_as_uint(res->msgs[0], "msDS-KeyVersionNumber", 0);
 388
 389         /* Store the account GUID. */
 390         r->out.account_guid = samdb_result_guid(res->msgs[0], "objectGUID");
 391
 392         if (r->in.acct_type == ACB_SVRTRUST) {
 393                 status = libnet_JoinSite(ctx, remote_ldb, r);
 394         }
 395         talloc_free(tmp_ctx);
 396
 397         return status;
 398 }
 399
 400 /*
 401  * do a domain join using DCERPC/SAMR calls
 402  * - connect to the LSA pipe, to try and find out information about the domain
 403  * - create a secondary connection to SAMR pipe
 404  * - do a samr_Connect to get a policy handle
 405  * - do a samr_LookupDomain to get the domain sid
 406  * - do a samr_OpenDomain to get a domain handle
 407  * - do a samr_CreateAccount to try and get a new account
 408  *
 409  * If that fails, do:
 410  * - do a samr_LookupNames to get the users rid
 411  * - do a samr_OpenUser to get a user handle
 412  * - potentially delete and recreate the user
 413  * - assert the account is of the right type with samrQueryUserInfo
 414  *
 415  * - call libnet_SetPassword_samr_handle to set the password,
 416  *   and pass a samr_UserInfo21 struct to set full_name and the account flags
 417  *
 418  * - do some ADS specific things when we join as Domain Controller,
 419  *    look at libnet_joinADSDomain() for the details
 420  */
 421 NTSTATUS libnet_JoinDomain(struct libnet_context *ctx, TALLOC_CTX *mem_ctx, struct libnet_JoinDomain *r)
 422 {
 423         TALLOC_CTX *tmp_ctx;
 424
 425         NTSTATUS status, cu_status;
 426
 427         struct libnet_RpcConnect *connect_with_info;
 428         struct dcerpc_pipe *samr_pipe;
 429
 430         struct samr_Connect sc;
 431         struct policy_handle p_handle;
 432         struct samr_OpenDomain od;
 433         struct policy_handle d_handle;
 434         struct samr_LookupNames ln;
 435         struct samr_OpenUser ou;
 436         struct samr_CreateUser2 cu;
 437         struct policy_handle *u_handle = NULL;
 438         struct samr_QueryUserInfo qui;
 439         struct samr_UserInfo21 u_info21;
 440         union libnet_SetPassword r2;
 441         struct samr_GetUserPwInfo pwp;
 442         struct lsa_String samr_account_name;
 443
 444         uint32_t acct_flags, old_acct_flags;
 445         uint32_t rid, access_granted;
 446         int policy_min_pw_len = 0;
 447
 448         struct dom_sid *account_sid = NULL;
 449         const char *password_str = NULL;
 450
 451         r->out.error_string = NULL;
 452         r2.samr_handle.out.error_string = NULL;
 453
 454         tmp_ctx = talloc_named(mem_ctx, 0, "libnet_Join temp context");
 455         if (!tmp_ctx) {
 456                 r->out.error_string = NULL;
 457                 return NT_STATUS_NO_MEMORY;
 458         }
 459
 460         u_handle = talloc(tmp_ctx, struct policy_handle);
 461         if (!u_handle) {
 462                 r->out.error_string = NULL;
 463                 talloc_free(tmp_ctx);
 464                 return NT_STATUS_NO_MEMORY;
 465         }
 466
 467         connect_with_info = talloc(tmp_ctx, struct libnet_RpcConnect);
 468         if (!connect_with_info) {
 469                 r->out.error_string = NULL;
 470                 talloc_free(tmp_ctx);
 471                 return NT_STATUS_NO_MEMORY;
 472         }
 473
 474         /* prepare connect to the SAMR pipe of PDC */
 475         if (r->in.level == LIBNET_JOINDOMAIN_AUTOMATIC) {
 476                 connect_with_info->in.binding = NULL;
 477                 connect_with_info->in.name    = r->in.domain_name;
 478         } else {
 479                 connect_with_info->in.binding = r->in.binding;
 480                 connect_with_info->in.name    = NULL;
 481         }
 482
 483         /* This level makes a connection to the LSA pipe on the way,
 484          * to get some useful bits of information about the domain */
 485         connect_with_info->level              = LIBNET_RPC_CONNECT_DC_INFO;
 486         connect_with_info->in.dcerpc_iface    = &ndr_table_samr;
 487
 488         /*
 489           establish the SAMR connection
 490         */
 491         status = libnet_RpcConnect(ctx, tmp_ctx, connect_with_info);
 492         if (!NT_STATUS_IS_OK(status)) {
 493                 if (r->in.binding) {
 494                         r->out.error_string = talloc_asprintf(mem_ctx,
 495                                                               "Connection to SAMR pipe of DC %s failed: %s",
 496                                                               r->in.binding, connect_with_info->out.error_string);
 497                 } else {
 498                         r->out.error_string = talloc_asprintf(mem_ctx,
 499                                                               "Connection to SAMR pipe of PDC for %s failed: %s",
 500                                                               r->in.domain_name, connect_with_info->out.error_string);
 501                 }
 502                 talloc_free(tmp_ctx);
 503                 return status;
 504         }
 505
 506         samr_pipe = connect_with_info->out.dcerpc_pipe;
 507
 508         status = dcerpc_pipe_auth(tmp_ctx, &samr_pipe,
 509                                   connect_with_info->out.dcerpc_pipe->binding,
 510                                   &ndr_table_samr, ctx->cred, ctx->lp_ctx);
 511         if (!NT_STATUS_IS_OK(status)) {
 512                 r->out.error_string = talloc_asprintf(mem_ctx,
 513                                                 "SAMR bind failed: %s",
 514                                                 nt_errstr(status));
 515                 talloc_free(tmp_ctx);
 516                 return status;
 517         }
 518
 519         /* prepare samr_Connect */
 520         ZERO_STRUCT(p_handle);
 521         sc.in.system_name = NULL;
 522         sc.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
 523         sc.out.connect_handle = &p_handle;
 524
 525         /* 2. do a samr_Connect to get a policy handle */
 526         status = dcerpc_samr_Connect(samr_pipe, tmp_ctx, &sc);
 527         if (!NT_STATUS_IS_OK(status)) {
 528                 r->out.error_string = talloc_asprintf(mem_ctx,
 529                                                 "samr_Connect failed: %s",
 530                                                 nt_errstr(status));
 531                 talloc_free(tmp_ctx);
 532                 return status;
 533         }
 534
 535         /* If this is a connection on ncacn_ip_tcp to Win2k3 SP1, we don't get back this useful info */
 536         if (!connect_with_info->out.domain_name) {
 537                 if (r->in.level == LIBNET_JOINDOMAIN_AUTOMATIC) {
 538                         connect_with_info->out.domain_name = talloc_strdup(tmp_ctx, r->in.domain_name);
 539                 } else {
 540                         /* Bugger, we just lost our way to automaticly find the domain name */
 541                         connect_with_info->out.domain_name = talloc_strdup(tmp_ctx, lp_workgroup(ctx->lp_ctx));
 542                         connect_with_info->out.realm = talloc_strdup(tmp_ctx, lp_realm(ctx->lp_ctx));
 543                 }
 544         }
 545
 546         /* Perhaps we didn't get a SID above, because we are against ncacn_ip_tcp */
 547         if (!connect_with_info->out.domain_sid) {
 548                 struct lsa_String name;
 549                 struct samr_LookupDomain l;
 550                 name.string = connect_with_info->out.domain_name;
 551                 l.in.connect_handle = &p_handle;
 552                 l.in.domain_name = &name;
 553
 554                 status = dcerpc_samr_LookupDomain(samr_pipe, tmp_ctx, &l);
 555                 if (!NT_STATUS_IS_OK(status)) {
 556                         r->out.error_string = talloc_asprintf(mem_ctx,
 557                                                               "SAMR LookupDomain failed: %s",
 558                                                               nt_errstr(status));
 559                         talloc_free(tmp_ctx);
 560                         return status;
 561                 }
 562                 connect_with_info->out.domain_sid = l.out.sid;
 563         }
 564
 565         /* prepare samr_OpenDomain */
 566         ZERO_STRUCT(d_handle);
 567         od.in.connect_handle = &p_handle;
 568         od.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
 569         od.in.sid = connect_with_info->out.domain_sid;
 570         od.out.domain_handle = &d_handle;
 571
 572         /* do a samr_OpenDomain to get a domain handle */
 573         status = dcerpc_samr_OpenDomain(samr_pipe, tmp_ctx, &od);
 574         if (!NT_STATUS_IS_OK(status)) {
 575                 r->out.error_string = talloc_asprintf(mem_ctx,
 576                                                       "samr_OpenDomain for [%s] failed: %s",
 577                                                       dom_sid_string(tmp_ctx, connect_with_info->out.domain_sid),
 578                                                       nt_errstr(status));
 579                 talloc_free(tmp_ctx);
 580                 return status;
 581         }
 582
 583         /* prepare samr_CreateUser2 */
 584         ZERO_STRUCTP(u_handle);
 585         cu.in.domain_handle  = &d_handle;
 586         cu.in.access_mask     = SEC_FLAG_MAXIMUM_ALLOWED;
 587         samr_account_name.string = r->in.account_name;
 588         cu.in.account_name    = &samr_account_name;
 589         cu.in.acct_flags      = r->in.acct_type;
 590         cu.out.user_handle    = u_handle;
 591         cu.out.rid            = &rid;
 592         cu.out.access_granted = &access_granted;
 593
 594         /* do a samr_CreateUser2 to get an account handle, or an error */
 595         cu_status = dcerpc_samr_CreateUser2(samr_pipe, tmp_ctx, &cu);
 596         status = cu_status;
 597         if (NT_STATUS_EQUAL(status, NT_STATUS_USER_EXISTS)) {
 598                 /* prepare samr_LookupNames */
 599                 ln.in.domain_handle = &d_handle;
 600                 ln.in.num_names = 1;
 601                 ln.in.names = talloc_array(tmp_ctx, struct lsa_String, 1);
 602                 if (!ln.in.names) {
 603                         r->out.error_string = NULL;
 604                         talloc_free(tmp_ctx);
 605                         return NT_STATUS_NO_MEMORY;
 606                 }
 607                 ln.in.names[0].string = r->in.account_name;
 608
 609                 /* 5. do a samr_LookupNames to get the users rid */
 610                 status = dcerpc_samr_LookupNames(samr_pipe, tmp_ctx, &ln);
 611                 if (!NT_STATUS_IS_OK(status)) {
 612                         r->out.error_string = talloc_asprintf(mem_ctx,
 613                                                 "samr_LookupNames for [%s] failed: %s",
 614                                                 r->in.account_name,
 615                                                 nt_errstr(status));
 616                         talloc_free(tmp_ctx);
 617                         return status;
 618                 }
 619
 620                 /* check if we got one RID for the user */
 621                 if (ln.out.rids.count != 1) {
 622                         r->out.error_string = talloc_asprintf(mem_ctx,
 623                                                               "samr_LookupNames for [%s] returns %d RIDs",
 624                                                               r->in.account_name, ln.out.rids.count);
 625                         talloc_free(tmp_ctx);
 626                         return NT_STATUS_INVALID_PARAMETER;
 627                 }
 628
 629                 /* prepare samr_OpenUser */
 630                 ZERO_STRUCTP(u_handle);
 631                 ou.in.domain_handle = &d_handle;
 632                 ou.in.access_mask = SEC_FLAG_MAXIMUM_ALLOWED;
 633                 ou.in.rid = ln.out.rids.ids[0];
 634                 rid = ou.in.rid;
 635                 ou.out.user_handle = u_handle;
 636
 637                 /* 6. do a samr_OpenUser to get a user handle */
 638                 status = dcerpc_samr_OpenUser(samr_pipe, tmp_ctx, &ou);
 639                 if (!NT_STATUS_IS_OK(status)) {
 640                         r->out.error_string = talloc_asprintf(mem_ctx,
 641                                                         "samr_OpenUser for [%s] failed: %s",
 642                                                         r->in.account_name,
 643                                                         nt_errstr(status));
 644                         talloc_free(tmp_ctx);
 645                         return status;
 646                 }
 647
 648                 if (r->in.recreate_account) {
 649                         struct samr_DeleteUser d;
 650                         d.in.user_handle = u_handle;
 651                         d.out.user_handle = u_handle;
 652                         status = dcerpc_samr_DeleteUser(samr_pipe, mem_ctx, &d);
 653                         if (!NT_STATUS_IS_OK(status)) {
 654                                 r->out.error_string = talloc_asprintf(mem_ctx,
 655                                                                       "samr_DeleteUser (for recreate) of [%s] failed: %s",
 656                                                                       r->in.account_name,
 657                                                                       nt_errstr(status));
 658                                 talloc_free(tmp_ctx);
 659                                 return status;
 660                         }
 661
 662                         /* We want to recreate, so delete and another samr_CreateUser2 */
 663
 664                         /* &cu filled in above */
 665                         status = dcerpc_samr_CreateUser2(samr_pipe, tmp_ctx, &cu);
 666                         if (!NT_STATUS_IS_OK(status)) {
 667                                 r->out.error_string = talloc_asprintf(mem_ctx,
 668                                                                       "samr_CreateUser2 (recreate) for [%s] failed: %s",
 669                                                                       r->in.account_name, nt_errstr(status));
 670                                 talloc_free(tmp_ctx);
 671                                 return status;
 672                         }
 673                 }
 674         } else if (!NT_STATUS_IS_OK(status)) {
 675                 r->out.error_string = talloc_asprintf(mem_ctx,
 676                                                       "samr_CreateUser2 for [%s] failed: %s",
 677                                                       r->in.account_name, nt_errstr(status));
 678                 talloc_free(tmp_ctx);
 679                 return status;
 680         }
 681
 682         /* prepare samr_QueryUserInfo (get flags) */
 683         qui.in.user_handle = u_handle;
 684         qui.in.level = 16;
 685
 686         status = dcerpc_samr_QueryUserInfo(samr_pipe, tmp_ctx, &qui);
 687         if (!NT_STATUS_IS_OK(status)) {
 688                 r->out.error_string = talloc_asprintf(mem_ctx,
 689                                                 "samr_QueryUserInfo for [%s] failed: %s",
 690                                                 r->in.account_name,
 691                                                 nt_errstr(status));
 692                 talloc_free(tmp_ctx);
 693                 return status;
 694         }
 695
 696         if (!qui.out.info) {
 697                 status = NT_STATUS_INVALID_PARAMETER;
 698                 r->out.error_string
 699                         = talloc_asprintf(mem_ctx,
 700                                           "samr_QueryUserInfo failed to return qui.out.info for [%s]: %s",
 701                                           r->in.account_name, nt_errstr(status));
 702                 talloc_free(tmp_ctx);
 703                 return status;
 704         }
 705
 706         old_acct_flags = (qui.out.info->info16.acct_flags & (ACB_WSTRUST | ACB_SVRTRUST | ACB_DOMTRUST));
 707         /* Possibly bail if the account is of the wrong type */
 708         if (old_acct_flags
 709             != r->in.acct_type) {
 710                 const char *old_account_type, *new_account_type;
 711                 switch (old_acct_flags) {
 712                 case ACB_WSTRUST:
 713                         old_account_type = "domain member (member)";
 714                         break;
 715                 case ACB_SVRTRUST:
 716                         old_account_type = "domain controller (bdc)";
 717                         break;
 718                 case ACB_DOMTRUST:
 719                         old_account_type = "trusted domain";
 720                         break;
 721                 default:
 722                         return NT_STATUS_INVALID_PARAMETER;
 723                 }
 724                 switch (r->in.acct_type) {
 725                 case ACB_WSTRUST:
 726                         new_account_type = "domain member (member)";
 727                         break;
 728                 case ACB_SVRTRUST:
 729                         new_account_type = "domain controller (bdc)";
 730                         break;
 731                 case ACB_DOMTRUST:
 732                         new_account_type = "trusted domain";
 733                         break;
 734                 default:
 735                         return NT_STATUS_INVALID_PARAMETER;
 736                 }
 737
 738                 if (!NT_STATUS_EQUAL(cu_status, NT_STATUS_USER_EXISTS)) {
 739                         /* We created a new user, but they didn't come out the right type?!? */
 740                         r->out.error_string
 741                                 = talloc_asprintf(mem_ctx,
 742                                                   "We asked to create a new machine account (%s) of type %s, "
 743                                                   "but we got an account of type %s.  This is unexpected.  "
 744                                                   "Perhaps delete the account and try again.",
 745                                                   r->in.account_name, new_account_type, old_account_type);
 746                         talloc_free(tmp_ctx);
 747                         return NT_STATUS_INVALID_PARAMETER;
 748                 } else {
 749                         /* The account is of the wrong type, so bail */
 750
 751                         /* TODO: We should allow a --force option to override, and redo this from the top setting r.in.recreate_account */
 752                         r->out.error_string
 753                                 = talloc_asprintf(mem_ctx,
 754                                                   "The machine account (%s) already exists in the domain %s, "
 755                                                   "but is a %s.  You asked to join as a %s.  Please delete "
 756                                                   "the account and try again.",
 757                                                   r->in.account_name, connect_with_info->out.domain_name, old_account_type, new_account_type);
 758                         talloc_free(tmp_ctx);
 759                         return NT_STATUS_USER_EXISTS;
 760                 }
 761         } else {
 762                 acct_flags = qui.out.info->info16.acct_flags;
 763         }
 764
 765         acct_flags = (acct_flags & ~(ACB_DISABLED|ACB_PWNOTREQ));
 766
 767         /* Find out what password policy this user has */
 768         pwp.in.user_handle = u_handle;
 769
 770         status = dcerpc_samr_GetUserPwInfo(samr_pipe, tmp_ctx, &pwp);
 771         if (NT_STATUS_IS_OK(status)) {
 772                 policy_min_pw_len = pwp.out.info.min_password_length;
 773         }
 774
 775         /* Grab a password of that minimum length */
 776
 777         password_str = generate_random_str(tmp_ctx, MAX(8, policy_min_pw_len));
 778
 779         /* set full_name and reset flags */
 780         ZERO_STRUCT(u_info21);
 781         u_info21.full_name.string       = r->in.account_name;
 782         u_info21.acct_flags             = acct_flags;
 783         u_info21.fields_present         = SAMR_FIELD_FULL_NAME | SAMR_FIELD_ACCT_FLAGS;
 784
 785         r2.samr_handle.level            = LIBNET_SET_PASSWORD_SAMR_HANDLE;
 786         r2.samr_handle.in.account_name  = r->in.account_name;
 787         r2.samr_handle.in.newpassword   = password_str;
 788         r2.samr_handle.in.user_handle   = u_handle;
 789         r2.samr_handle.in.dcerpc_pipe   = samr_pipe;
 790         r2.samr_handle.in.info21        = &u_info21;
 791
 792         status = libnet_SetPassword(ctx, tmp_ctx, &r2);
 793         if (!NT_STATUS_IS_OK(status)) {
 794                 r->out.error_string = talloc_steal(mem_ctx, r2.samr_handle.out.error_string);
 795                 talloc_free(tmp_ctx);
 796                 return status;
 797         }
 798
 799         account_sid = dom_sid_add_rid(mem_ctx, connect_with_info->out.domain_sid, rid);
 800         if (!account_sid) {
 801                 r->out.error_string = NULL;
 802                 talloc_free(tmp_ctx);
 803                 return NT_STATUS_NO_MEMORY;
 804         }
 805
 806         /* Finish out by pushing various bits of status data out for the caller to use */
 807         r->out.join_password = password_str;
 808         talloc_steal(mem_ctx, r->out.join_password);
 809
 810         r->out.domain_sid = connect_with_info->out.domain_sid;
 811         talloc_steal(mem_ctx, r->out.domain_sid);
 812
 813         r->out.account_sid = account_sid;
 814         talloc_steal(mem_ctx, r->out.account_sid);
 815
 816         r->out.domain_name = connect_with_info->out.domain_name;
 817         talloc_steal(mem_ctx, r->out.domain_name);
 818         r->out.realm = connect_with_info->out.realm;
 819         talloc_steal(mem_ctx, r->out.realm);
 820         r->out.samr_pipe = samr_pipe;
 821         talloc_steal(mem_ctx, samr_pipe);
 822         r->out.samr_binding = samr_pipe->binding;
 823         talloc_steal(mem_ctx, r->out.samr_binding);
 824         r->out.user_handle = u_handle;
 825         talloc_steal(mem_ctx, u_handle);
 826         r->out.error_string = r2.samr_handle.out.error_string;
 827         talloc_steal(mem_ctx, r2.samr_handle.out.error_string);
 828         r->out.kvno = 0;
 829         r->out.server_dn_str = NULL;
 830         talloc_free(tmp_ctx);
 831
 832         /* Now, if it was AD, then we want to start looking changing a
 833          * few more things.  Otherwise, we are done. */
 834         if (r->out.realm) {
 835                 status = libnet_JoinADSDomain(ctx, r);
 836                 return status;
 837         }
 838
 839         return status;
 840 }
 841
 842 NTSTATUS libnet_set_join_secrets(struct libnet_context *ctx,
 843                                  TALLOC_CTX *mem_ctx,
 844                                  struct libnet_set_join_secrets *r)
 845 {
 846         TALLOC_CTX *tmp_mem;
 847         int ret, rtn;
 848         struct ldb_context *ldb;
 849         struct ldb_dn *base_dn;
 850         struct ldb_message **msgs, *msg;
 851         const char *sct;
 852         const char * const attrs[] = {
 853                 "whenChanged",
 854                 "secret",
 855                 "priorSecret",
 856                 "priorChanged",
 857                 "krb5Keytab",
 858                 "privateKeytab",
 859                 NULL
 860         };
 861
 862         tmp_mem = talloc_new(mem_ctx);
 863         if (!tmp_mem) {
 864                 return NT_STATUS_NO_MEMORY;
 865         }
 866
 867         /* Open the secrets database */
 868         ldb = secrets_db_connect(tmp_mem, ctx->lp_ctx);
 869         if (!ldb) {
 870                 r->out.error_string
 871                         = talloc_asprintf(mem_ctx,
 872                                           "Could not open secrets database");
 873                 talloc_free(tmp_mem);
 874                 return NT_STATUS_CANT_ACCESS_DOMAIN_INFO;
 875         }
 876
 877         /*
 878          * now prepare the record for secrets.ldb
 879          */
 880         sct = talloc_asprintf(tmp_mem, "%d", r->in.join_type);
 881         if (!sct) {
 882                 r->out.error_string = NULL;
 883                 talloc_free(tmp_mem);
 884                 return NT_STATUS_NO_MEMORY;
 885         }
 886
 887         msg = ldb_msg_new(tmp_mem);
 888         if (!msg) {
 889                 r->out.error_string = NULL;
 890                 talloc_free(tmp_mem);
 891                 return NT_STATUS_NO_MEMORY;
 892         }
 893
 894         base_dn = ldb_dn_new(tmp_mem, ldb, "cn=Primary Domains");
 895         if (!base_dn) {
 896                 r->out.error_string = NULL;
 897                 talloc_free(tmp_mem);
 898                 return NT_STATUS_NO_MEMORY;
 899         }
 900
 901         msg->dn = ldb_dn_copy(tmp_mem, base_dn);
 902         if ( ! ldb_dn_add_child_fmt(msg->dn, "flatname=%s", r->in.domain_name)) {
 903                 r->out.error_string = NULL;
 904                 talloc_free(tmp_mem);
 905                 return NT_STATUS_NO_MEMORY;
 906         }
 907
 908         rtn = samdb_msg_add_string(ldb, tmp_mem, msg, "flatname", r->in.domain_name);
 909         if (rtn == -1) {
 910                 r->out.error_string = NULL;
 911                 talloc_free(tmp_mem);
 912                 return NT_STATUS_NO_MEMORY;
 913         }
 914
 915         if (r->in.realm) {
 916                 rtn = samdb_msg_add_string(ldb, tmp_mem, msg, "realm", r->in.realm);
 917                 if (rtn == -1) {
 918                         r->out.error_string = NULL;
 919                         talloc_free(tmp_mem);
 920                         return NT_STATUS_NO_MEMORY;
 921                 }
 922
 923                 rtn = samdb_msg_add_string(ldb, tmp_mem, msg, "objectClass", "primaryDomain");
 924                 if (rtn == -1) {
 925                         r->out.error_string = NULL;
 926                         talloc_free(tmp_mem);
 927                         return NT_STATUS_NO_MEMORY;
 928                 }
 929         }
 930
 931         rtn = samdb_msg_add_string(ldb, tmp_mem, msg, "objectClass", "primaryDomain");
 932         if (rtn == -1) {
 933                 r->out.error_string = NULL;
 934                 talloc_free(tmp_mem);
 935                 return NT_STATUS_NO_MEMORY;
 936         }
 937
 938         rtn = samdb_msg_add_string(ldb, tmp_mem, msg, "secret", r->in.join_password);
 939         if (rtn == -1) {
 940                 r->out.error_string = NULL;
 941                 talloc_free(tmp_mem);
 942                 return NT_STATUS_NO_MEMORY;
 943         }
 944
 945         rtn = samdb_msg_add_string(ldb, tmp_mem, msg, "samAccountName", r->in.account_name);
 946         if (rtn == -1) {
 947                 r->out.error_string = NULL;
 948                 talloc_free(tmp_mem);
 949                 return NT_STATUS_NO_MEMORY;
 950         }
 951
 952         rtn = samdb_msg_add_string(ldb, tmp_mem, msg, "secureChannelType", sct);
 953         if (rtn == -1) {
 954                 r->out.error_string = NULL;
 955                 talloc_free(tmp_mem);
 956                 return NT_STATUS_NO_MEMORY;
 957         }
 958
 959         if (r->in.kvno) {
 960                 rtn = samdb_msg_add_uint(ldb, tmp_mem, msg, "msDS-KeyVersionNumber",
 961                                          r->in.kvno);
 962                 if (rtn == -1) {
 963                         r->out.error_string = NULL;
 964                         talloc_free(tmp_mem);
 965                         return NT_STATUS_NO_MEMORY;
 966                 }
 967         }
 968
 969         if (r->in.domain_sid) {
 970                 rtn = samdb_msg_add_dom_sid(ldb, tmp_mem, msg, "objectSid",
 971                                             r->in.domain_sid);
 972                 if (rtn == -1) {
 973                         r->out.error_string = NULL;
 974                         talloc_free(tmp_mem);
 975                         return NT_STATUS_NO_MEMORY;
 976                 }
 977         }
 978
 979         /*
 980          * search for the secret record
 981          * - remove the records we find
 982          * - and fetch the old secret and store it under priorSecret
 983          */
 984         ret = gendb_search(ldb,
 985                            tmp_mem, base_dn,
 986                            &msgs, attrs,
 987                            "(|" SECRETS_PRIMARY_DOMAIN_FILTER "(realm=%s))",
 988                            r->in.domain_name, r->in.realm);
 989         if (ret == 0) {
 990                 rtn = samdb_msg_set_string(ldb, tmp_mem, msg, "secretsKeytab", "secrets.keytab");
 991                 if (rtn == -1) {
 992                         r->out.error_string = NULL;
 993                         talloc_free(tmp_mem);
 994                         return NT_STATUS_NO_MEMORY;
 995                 }
 996         } else if (ret == -1) {
 997                 r->out.error_string
 998                         = talloc_asprintf(mem_ctx,
 999                                           "Search for domain: %s and realm: %s failed: %s",
1000                                           r->in.domain_name, r->in.realm, ldb_errstring(ldb));
1001                 talloc_free(tmp_mem);
1002                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
1003         } else {
1004                 const struct ldb_val *private_keytab;
1005                 const struct ldb_val *krb5_main_keytab;
1006                 const struct ldb_val *prior_secret;
1007                 const struct ldb_val *prior_modified_time;
1008                 int i;
1009
1010                 for (i = 0; i < ret; i++) {
1011                         ldb_delete(ldb, msgs[i]->dn);
1012                 }
1013
1014                 prior_secret = ldb_msg_find_ldb_val(msgs[0], "secret");
1015                 if (prior_secret) {
1016                         rtn = samdb_msg_set_value(ldb, tmp_mem, msg, "priorSecret", prior_secret);
1017                         if (rtn == -1) {
1018                                 r->out.error_string = NULL;
1019                                 talloc_free(tmp_mem);
1020                                 return NT_STATUS_NO_MEMORY;
1021                         }
1022                 }
1023                 rtn = samdb_msg_set_string(ldb, tmp_mem, msg, "secret", r->in.join_password);
1024                 if (rtn == -1) {
1025                         r->out.error_string = NULL;
1026                         talloc_free(tmp_mem);
1027                         return NT_STATUS_NO_MEMORY;
1028                 }
1029
1030                 prior_modified_time = ldb_msg_find_ldb_val(msgs[0],
1031                                                            "whenChanged");
1032                 if (prior_modified_time) {
1033                         rtn = samdb_msg_set_value(ldb, tmp_mem, msg, "priorWhenChanged",
1034                                                   prior_modified_time);
1035                         if (rtn == -1) {
1036                                 r->out.error_string = NULL;
1037                                 talloc_free(tmp_mem);
1038                                 return NT_STATUS_NO_MEMORY;
1039                         }
1040                 }
1041
1042                 rtn = samdb_msg_set_string(ldb, tmp_mem, msg, "samAccountName", r->in.account_name);
1043                 if (rtn == -1) {
1044                         r->out.error_string = NULL;
1045                         talloc_free(tmp_mem);
1046                         return NT_STATUS_NO_MEMORY;
1047                 }
1048
1049                 rtn = samdb_msg_set_string(ldb, tmp_mem, msg, "secureChannelType", sct);
1050                 if (rtn == -1) {
1051                         r->out.error_string = NULL;
1052                         talloc_free(tmp_mem);
1053                         return NT_STATUS_NO_MEMORY;
1054                 }
1055
1056                 /* We will want to keep the keytab names */
1057                 private_keytab = ldb_msg_find_ldb_val(msgs[0], "privateKeytab");
1058                 if (private_keytab) {
1059                         rtn = samdb_msg_set_value(ldb, tmp_mem, msg, "privateKeytab", private_keytab);
1060                         if (rtn == -1) {
1061                                 r->out.error_string = NULL;
1062                                 talloc_free(tmp_mem);
1063                                 return NT_STATUS_NO_MEMORY;
1064                         }
1065                 }
1066                 krb5_main_keytab = ldb_msg_find_ldb_val(msgs[0], "krb5Keytab");
1067                 if (krb5_main_keytab) {
1068                         rtn = samdb_msg_set_value(ldb, tmp_mem, msg,
1069                                         "krb5Keytab", krb5_main_keytab);
1070                         if (rtn == -1) {
1071                                 r->out.error_string = NULL;
1072                                 talloc_free(tmp_mem);
1073                                 return NT_STATUS_NO_MEMORY;
1074                         }
1075                 }
1076         }
1077
1078         /* create the secret */
1079         ret = ldb_add(ldb, msg);
1080         if (ret != 0) {
1081                 r->out.error_string = talloc_asprintf(mem_ctx, "Failed to create secret record %s",
1082                                                       ldb_dn_get_linearized(msg->dn));
1083                 talloc_free(tmp_mem);
1084                 return NT_STATUS_INTERNAL_DB_CORRUPTION;
1085         }
1086
1087         return NT_STATUS_OK;
1088 }
1089
1090 static NTSTATUS libnet_Join_primary_domain(struct libnet_context *ctx,
1091                                            TALLOC_CTX *mem_ctx,
1092                                            struct libnet_Join *r)
1093 {
1094         NTSTATUS status;
1095         TALLOC_CTX *tmp_mem;
1096         struct libnet_JoinDomain *r2;
1097         struct libnet_set_join_secrets *r3;
1098         uint32_t acct_type = 0;
1099         const char *account_name;
1100         const char *netbios_name;
1101
1102         r->out.error_string = NULL;
1103
1104         tmp_mem = talloc_new(mem_ctx);
1105         if (!tmp_mem) {
1106                 return NT_STATUS_NO_MEMORY;
1107         }
1108
1109         r2 = talloc(tmp_mem, struct libnet_JoinDomain);
1110         if (!r2) {
1111                 r->out.error_string = NULL;
1112                 talloc_free(tmp_mem);
1113                 return NT_STATUS_NO_MEMORY;
1114         }
1115
1116         if (r->in.join_type == SEC_CHAN_BDC) {
1117                 acct_type = ACB_SVRTRUST;
1118         } else if (r->in.join_type == SEC_CHAN_WKSTA) {
1119                 acct_type = ACB_WSTRUST;
1120         } else {
1121                 r->out.error_string = NULL;
1122                 talloc_free(tmp_mem);
1123                 return NT_STATUS_INVALID_PARAMETER;
1124         }
1125
1126         if (r->in.netbios_name != NULL) {
1127                 netbios_name = r->in.netbios_name;
1128         } else {
1129                 netbios_name = talloc_reference(tmp_mem, lp_netbios_name(ctx->lp_ctx));
1130                 if (!netbios_name) {
1131                         r->out.error_string = NULL;
1132                         talloc_free(tmp_mem);
1133                         return NT_STATUS_NO_MEMORY;
1134                 }
1135         }
1136
1137         account_name = talloc_asprintf(tmp_mem, "%s$", netbios_name);
1138         if (!account_name) {
1139                 r->out.error_string = NULL;
1140                 talloc_free(tmp_mem);
1141                 return NT_STATUS_NO_MEMORY;
1142         }
1143
1144         /*
1145          * join the domain
1146          */
1147         ZERO_STRUCTP(r2);
1148         r2->in.domain_name      = r->in.domain_name;
1149         r2->in.account_name     = account_name;
1150         r2->in.netbios_name     = netbios_name;
1151         r2->in.level            = LIBNET_JOINDOMAIN_AUTOMATIC;
1152         r2->in.acct_type        = acct_type;
1153         r2->in.recreate_account = false;
1154         status = libnet_JoinDomain(ctx, r2, r2);
1155         if (!NT_STATUS_IS_OK(status)) {
1156                 r->out.error_string = talloc_steal(mem_ctx, r2->out.error_string);
1157                 talloc_free(tmp_mem);
1158                 return status;
1159         }
1160
1161         r3 = talloc(tmp_mem, struct libnet_set_join_secrets);
1162         if (!r3) {
1163                 r->out.error_string = NULL;
1164                 talloc_free(tmp_mem);
1165                 return NT_STATUS_NO_MEMORY;
1166         }
1167
1168         ZERO_STRUCTP(r3);
1169         r3->in.domain_name = r2->out.domain_name;
1170         r3->in.realm = r2->out.realm;
1171         r3->in.account_name = account_name;
1172         r3->in.netbios_name = netbios_name;
1173         r3->in.join_type = r->in.join_type;
1174         r3->in.join_password = r2->out.join_password;
1175         r3->in.kvno = r2->out.kvno;
1176         r3->in.domain_sid = r2->out.domain_sid;
1177
1178         status = libnet_set_join_secrets(ctx, r3, r3);
1179         if (!NT_STATUS_IS_OK(status)) {
1180                 r->out.error_string = talloc_steal(mem_ctx, r3->out.error_string);
1181                 talloc_free(tmp_mem);
1182                 return status;
1183         }
1184
1185         /* move all out parameter to the callers TALLOC_CTX */
1186         r->out.error_string     = NULL;
1187         r->out.join_password    = r2->out.join_password;
1188         talloc_steal(mem_ctx, r2->out.join_password);
1189         r->out.domain_sid       = r2->out.domain_sid;
1190         talloc_steal(mem_ctx, r2->out.domain_sid);
1191         r->out.domain_name      = r2->out.domain_name;
1192         talloc_steal(mem_ctx, r2->out.domain_name);
1193         talloc_free(tmp_mem);
1194         return NT_STATUS_OK;
1195 }
1196
1197 NTSTATUS libnet_Join(struct libnet_context *ctx, TALLOC_CTX *mem_ctx, struct libnet_Join *r)
1198 {
1199         switch (r->in.join_type) {
1200                 case SEC_CHAN_WKSTA:
1201                         return libnet_Join_primary_domain(ctx, mem_ctx, r);
1202                 case SEC_CHAN_BDC:
1203                         return libnet_Join_primary_domain(ctx, mem_ctx, r);
1204                 case SEC_CHAN_DOMAIN:
1205                         break;
1206         }
1207
1208         r->out.error_string = talloc_asprintf(mem_ctx,
1209                                               "Invalid join type specified (%08X) attempting to join domain %s",
1210                                               r->in.join_type, r->in.domain_name);
1211         return NT_STATUS_INVALID_PARAMETER;
1212 }