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