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