Merge branch 'selftest' of git://git.samba.org/jelmer/samba
[jra/samba/.git] / source3 / libnet / libnet_join.c
1 /*
2  *  Unix SMB/CIFS implementation.
3  *  libnet Join Support
4  *  Copyright (C) Gerald (Jerry) Carter 2006
5  *  Copyright (C) Guenther Deschner 2007-2008
6  *
7  *  This program is free software; you can redistribute it and/or modify
8  *  it under the terms of the GNU General Public License as published by
9  *  the Free Software Foundation; either version 3 of the License, or
10  *  (at your option) any later version.
11  *
12  *  This program is distributed in the hope that it will be useful,
13  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
14  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
15  *  GNU General Public License for more details.
16  *
17  *  You should have received a copy of the GNU General Public License
18  *  along with this program; if not, see <http://www.gnu.org/licenses/>.
19  */
20
21 #include "includes.h"
22 #include "libnet/libnet.h"
23
24 /****************************************************************
25 ****************************************************************/
26
27 #define LIBNET_JOIN_DUMP_CTX(ctx, r, f) \
28         do { \
29                 char *str = NULL; \
30                 str = NDR_PRINT_FUNCTION_STRING(ctx, libnet_JoinCtx, f, r); \
31                 DEBUG(1,("libnet_Join:\n%s", str)); \
32                 TALLOC_FREE(str); \
33         } while (0)
34
35 #define LIBNET_JOIN_IN_DUMP_CTX(ctx, r) \
36         LIBNET_JOIN_DUMP_CTX(ctx, r, NDR_IN | NDR_SET_VALUES)
37 #define LIBNET_JOIN_OUT_DUMP_CTX(ctx, r) \
38         LIBNET_JOIN_DUMP_CTX(ctx, r, NDR_OUT)
39
40 #define LIBNET_UNJOIN_DUMP_CTX(ctx, r, f) \
41         do { \
42                 char *str = NULL; \
43                 str = NDR_PRINT_FUNCTION_STRING(ctx, libnet_UnjoinCtx, f, r); \
44                 DEBUG(1,("libnet_Unjoin:\n%s", str)); \
45                 TALLOC_FREE(str); \
46         } while (0)
47
48 #define LIBNET_UNJOIN_IN_DUMP_CTX(ctx, r) \
49         LIBNET_UNJOIN_DUMP_CTX(ctx, r, NDR_IN | NDR_SET_VALUES)
50 #define LIBNET_UNJOIN_OUT_DUMP_CTX(ctx, r) \
51         LIBNET_UNJOIN_DUMP_CTX(ctx, r, NDR_OUT)
52
53 #define W_ERROR_NOT_OK_GOTO_DONE(x) do { \
54         if (!W_ERROR_IS_OK(x)) {\
55                 goto done;\
56         }\
57 } while (0)
58
59 /****************************************************************
60 ****************************************************************/
61
62 static void libnet_join_set_error_string(TALLOC_CTX *mem_ctx,
63                                          struct libnet_JoinCtx *r,
64                                          const char *format, ...)
65 {
66         va_list args;
67
68         if (r->out.error_string) {
69                 return;
70         }
71
72         va_start(args, format);
73         r->out.error_string = talloc_vasprintf(mem_ctx, format, args);
74         va_end(args);
75 }
76
77 /****************************************************************
78 ****************************************************************/
79
80 static void libnet_unjoin_set_error_string(TALLOC_CTX *mem_ctx,
81                                            struct libnet_UnjoinCtx *r,
82                                            const char *format, ...)
83 {
84         va_list args;
85
86         if (r->out.error_string) {
87                 return;
88         }
89
90         va_start(args, format);
91         r->out.error_string = talloc_vasprintf(mem_ctx, format, args);
92         va_end(args);
93 }
94
95 #ifdef WITH_ADS
96
97 /****************************************************************
98 ****************************************************************/
99
100 static ADS_STATUS libnet_connect_ads(const char *dns_domain_name,
101                                      const char *netbios_domain_name,
102                                      const char *dc_name,
103                                      const char *user_name,
104                                      const char *password,
105                                      ADS_STRUCT **ads)
106 {
107         ADS_STATUS status;
108         ADS_STRUCT *my_ads = NULL;
109
110         my_ads = ads_init(dns_domain_name,
111                           netbios_domain_name,
112                           dc_name);
113         if (!my_ads) {
114                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
115         }
116
117         if (user_name) {
118                 SAFE_FREE(my_ads->auth.user_name);
119                 my_ads->auth.user_name = SMB_STRDUP(user_name);
120         }
121
122         if (password) {
123                 SAFE_FREE(my_ads->auth.password);
124                 my_ads->auth.password = SMB_STRDUP(password);
125         }
126
127         status = ads_connect_user_creds(my_ads);
128         if (!ADS_ERR_OK(status)) {
129                 ads_destroy(&my_ads);
130                 return status;
131         }
132
133         *ads = my_ads;
134         return ADS_SUCCESS;
135 }
136
137 /****************************************************************
138 ****************************************************************/
139
140 static ADS_STATUS libnet_join_connect_ads(TALLOC_CTX *mem_ctx,
141                                           struct libnet_JoinCtx *r)
142 {
143         ADS_STATUS status;
144
145         status = libnet_connect_ads(r->out.dns_domain_name,
146                                     r->out.netbios_domain_name,
147                                     r->in.dc_name,
148                                     r->in.admin_account,
149                                     r->in.admin_password,
150                                     &r->in.ads);
151         if (!ADS_ERR_OK(status)) {
152                 libnet_join_set_error_string(mem_ctx, r,
153                         "failed to connect to AD: %s",
154                         ads_errstr(status));
155                 return status;
156         }
157
158         if (!r->out.netbios_domain_name) {
159                 r->out.netbios_domain_name = talloc_strdup(mem_ctx,
160                                                            r->in.ads->server.workgroup);
161                 ADS_ERROR_HAVE_NO_MEMORY(r->out.netbios_domain_name);
162         }
163
164         if (!r->out.dns_domain_name) {
165                 r->out.dns_domain_name = talloc_strdup(mem_ctx,
166                                                        r->in.ads->config.realm);
167                 ADS_ERROR_HAVE_NO_MEMORY(r->out.dns_domain_name);
168         }
169
170         r->out.domain_is_ad = true;
171
172         return ADS_SUCCESS;
173 }
174
175 /****************************************************************
176 ****************************************************************/
177
178 static ADS_STATUS libnet_unjoin_connect_ads(TALLOC_CTX *mem_ctx,
179                                             struct libnet_UnjoinCtx *r)
180 {
181         ADS_STATUS status;
182
183         status = libnet_connect_ads(r->in.domain_name,
184                                     r->in.domain_name,
185                                     r->in.dc_name,
186                                     r->in.admin_account,
187                                     r->in.admin_password,
188                                     &r->in.ads);
189         if (!ADS_ERR_OK(status)) {
190                 libnet_unjoin_set_error_string(mem_ctx, r,
191                         "failed to connect to AD: %s",
192                         ads_errstr(status));
193         }
194
195         return status;
196 }
197
198 /****************************************************************
199  join a domain using ADS (LDAP mods)
200 ****************************************************************/
201
202 static ADS_STATUS libnet_join_precreate_machine_acct(TALLOC_CTX *mem_ctx,
203                                                      struct libnet_JoinCtx *r)
204 {
205         ADS_STATUS status;
206         LDAPMessage *res = NULL;
207         const char *attrs[] = { "dn", NULL };
208         bool moved = false;
209
210         status = ads_check_ou_dn(mem_ctx, r->in.ads, &r->in.account_ou);
211         if (!ADS_ERR_OK(status)) {
212                 return status;
213         }
214
215         status = ads_search_dn(r->in.ads, &res, r->in.account_ou, attrs);
216         if (!ADS_ERR_OK(status)) {
217                 return status;
218         }
219
220         if (ads_count_replies(r->in.ads, res) != 1) {
221                 ads_msgfree(r->in.ads, res);
222                 return ADS_ERROR_LDAP(LDAP_NO_SUCH_OBJECT);
223         }
224
225         ads_msgfree(r->in.ads, res);
226
227         /* Attempt to create the machine account and bail if this fails.
228            Assume that the admin wants exactly what they requested */
229
230         status = ads_create_machine_acct(r->in.ads,
231                                          r->in.machine_name,
232                                          r->in.account_ou);
233
234         if (ADS_ERR_OK(status)) {
235                 DEBUG(1,("machine account creation created\n"));
236                 return status;
237         } else  if ((status.error_type == ENUM_ADS_ERROR_LDAP) &&
238                     (status.err.rc == LDAP_ALREADY_EXISTS)) {
239                 status = ADS_SUCCESS;
240         }
241
242         if (!ADS_ERR_OK(status)) {
243                 DEBUG(1,("machine account creation failed\n"));
244                 return status;
245         }
246
247         status = ads_move_machine_acct(r->in.ads,
248                                        r->in.machine_name,
249                                        r->in.account_ou,
250                                        &moved);
251         if (!ADS_ERR_OK(status)) {
252                 DEBUG(1,("failure to locate/move pre-existing "
253                         "machine account\n"));
254                 return status;
255         }
256
257         DEBUG(1,("The machine account %s the specified OU.\n",
258                 moved ? "was moved into" : "already exists in"));
259
260         return status;
261 }
262
263 /****************************************************************
264 ****************************************************************/
265
266 static ADS_STATUS libnet_unjoin_remove_machine_acct(TALLOC_CTX *mem_ctx,
267                                                     struct libnet_UnjoinCtx *r)
268 {
269         ADS_STATUS status;
270
271         if (!r->in.ads) {
272                 return libnet_unjoin_connect_ads(mem_ctx, r);
273         }
274
275         status = ads_leave_realm(r->in.ads, r->in.machine_name);
276         if (!ADS_ERR_OK(status)) {
277                 libnet_unjoin_set_error_string(mem_ctx, r,
278                         "failed to leave realm: %s",
279                         ads_errstr(status));
280                 return status;
281         }
282
283         return ADS_SUCCESS;
284 }
285
286 /****************************************************************
287 ****************************************************************/
288
289 static ADS_STATUS libnet_join_find_machine_acct(TALLOC_CTX *mem_ctx,
290                                                 struct libnet_JoinCtx *r)
291 {
292         ADS_STATUS status;
293         LDAPMessage *res = NULL;
294         char *dn = NULL;
295
296         if (!r->in.machine_name) {
297                 return ADS_ERROR(LDAP_NO_MEMORY);
298         }
299
300         status = ads_find_machine_acct(r->in.ads,
301                                        &res,
302                                        r->in.machine_name);
303         if (!ADS_ERR_OK(status)) {
304                 return status;
305         }
306
307         if (ads_count_replies(r->in.ads, res) != 1) {
308                 status = ADS_ERROR_LDAP(LDAP_NO_MEMORY);
309                 goto done;
310         }
311
312         dn = ads_get_dn(r->in.ads, res);
313         if (!dn) {
314                 status = ADS_ERROR_LDAP(LDAP_NO_MEMORY);
315                 goto done;
316         }
317
318         r->out.dn = talloc_strdup(mem_ctx, dn);
319         if (!r->out.dn) {
320                 status = ADS_ERROR_LDAP(LDAP_NO_MEMORY);
321                 goto done;
322         }
323
324  done:
325         ads_msgfree(r->in.ads, res);
326         ads_memfree(r->in.ads, dn);
327
328         return status;
329 }
330
331 /****************************************************************
332  Set a machines dNSHostName and servicePrincipalName attributes
333 ****************************************************************/
334
335 static ADS_STATUS libnet_join_set_machine_spn(TALLOC_CTX *mem_ctx,
336                                               struct libnet_JoinCtx *r)
337 {
338         ADS_STATUS status;
339         ADS_MODLIST mods;
340         fstring my_fqdn;
341         const char *spn_array[3] = {NULL, NULL, NULL};
342         char *spn = NULL;
343
344         /* Find our DN */
345
346         status = libnet_join_find_machine_acct(mem_ctx, r);
347         if (!ADS_ERR_OK(status)) {
348                 return status;
349         }
350
351         /* Windows only creates HOST/shortname & HOST/fqdn. */
352
353         spn = talloc_asprintf(mem_ctx, "HOST/%s", r->in.machine_name);
354         if (!spn) {
355                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
356         }
357         strupper_m(spn);
358         spn_array[0] = spn;
359
360         if (!name_to_fqdn(my_fqdn, r->in.machine_name)
361             || (strchr(my_fqdn, '.') == NULL)) {
362                 fstr_sprintf(my_fqdn, "%s.%s", r->in.machine_name,
363                              r->out.dns_domain_name);
364         }
365
366         strlower_m(my_fqdn);
367
368         if (!strequal(my_fqdn, r->in.machine_name)) {
369                 spn = talloc_asprintf(mem_ctx, "HOST/%s", my_fqdn);
370                 if (!spn) {
371                         return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
372                 }
373                 spn_array[1] = spn;
374         }
375
376         mods = ads_init_mods(mem_ctx);
377         if (!mods) {
378                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
379         }
380
381         /* fields of primary importance */
382
383         status = ads_mod_str(mem_ctx, &mods, "dNSHostName", my_fqdn);
384         if (!ADS_ERR_OK(status)) {
385                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
386         }
387
388         status = ads_mod_strlist(mem_ctx, &mods, "servicePrincipalName",
389                                  spn_array);
390         if (!ADS_ERR_OK(status)) {
391                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
392         }
393
394         return ads_gen_mod(r->in.ads, r->out.dn, mods);
395 }
396
397 /****************************************************************
398 ****************************************************************/
399
400 static ADS_STATUS libnet_join_set_machine_upn(TALLOC_CTX *mem_ctx,
401                                               struct libnet_JoinCtx *r)
402 {
403         ADS_STATUS status;
404         ADS_MODLIST mods;
405
406         if (!r->in.create_upn) {
407                 return ADS_SUCCESS;
408         }
409
410         /* Find our DN */
411
412         status = libnet_join_find_machine_acct(mem_ctx, r);
413         if (!ADS_ERR_OK(status)) {
414                 return status;
415         }
416
417         if (!r->in.upn) {
418                 r->in.upn = talloc_asprintf(mem_ctx,
419                                             "host/%s@%s",
420                                             r->in.machine_name,
421                                             r->out.dns_domain_name);
422                 if (!r->in.upn) {
423                         return ADS_ERROR(LDAP_NO_MEMORY);
424                 }
425         }
426
427         /* now do the mods */
428
429         mods = ads_init_mods(mem_ctx);
430         if (!mods) {
431                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
432         }
433
434         /* fields of primary importance */
435
436         status = ads_mod_str(mem_ctx, &mods, "userPrincipalName", r->in.upn);
437         if (!ADS_ERR_OK(status)) {
438                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
439         }
440
441         return ads_gen_mod(r->in.ads, r->out.dn, mods);
442 }
443
444
445 /****************************************************************
446 ****************************************************************/
447
448 static ADS_STATUS libnet_join_set_os_attributes(TALLOC_CTX *mem_ctx,
449                                                 struct libnet_JoinCtx *r)
450 {
451         ADS_STATUS status;
452         ADS_MODLIST mods;
453         char *os_sp = NULL;
454
455         if (!r->in.os_name || !r->in.os_version ) {
456                 return ADS_SUCCESS;
457         }
458
459         /* Find our DN */
460
461         status = libnet_join_find_machine_acct(mem_ctx, r);
462         if (!ADS_ERR_OK(status)) {
463                 return status;
464         }
465
466         /* now do the mods */
467
468         mods = ads_init_mods(mem_ctx);
469         if (!mods) {
470                 return ADS_ERROR(LDAP_NO_MEMORY);
471         }
472
473         os_sp = talloc_asprintf(mem_ctx, "Samba %s", SAMBA_VERSION_STRING);
474         if (!os_sp) {
475                 return ADS_ERROR(LDAP_NO_MEMORY);
476         }
477
478         /* fields of primary importance */
479
480         status = ads_mod_str(mem_ctx, &mods, "operatingSystem",
481                              r->in.os_name);
482         if (!ADS_ERR_OK(status)) {
483                 return status;
484         }
485
486         status = ads_mod_str(mem_ctx, &mods, "operatingSystemVersion",
487                              r->in.os_version);
488         if (!ADS_ERR_OK(status)) {
489                 return status;
490         }
491
492         status = ads_mod_str(mem_ctx, &mods, "operatingSystemServicePack",
493                              os_sp);
494         if (!ADS_ERR_OK(status)) {
495                 return status;
496         }
497
498         return ads_gen_mod(r->in.ads, r->out.dn, mods);
499 }
500
501 /****************************************************************
502 ****************************************************************/
503
504 static bool libnet_join_create_keytab(TALLOC_CTX *mem_ctx,
505                                       struct libnet_JoinCtx *r)
506 {
507         if (!lp_use_kerberos_keytab()) {
508                 return true;
509         }
510
511         if (ads_keytab_create_default(r->in.ads) != 0) {
512                 return false;
513         }
514
515         return true;
516 }
517
518 /****************************************************************
519 ****************************************************************/
520
521 static bool libnet_join_derive_salting_principal(TALLOC_CTX *mem_ctx,
522                                                  struct libnet_JoinCtx *r)
523 {
524         uint32_t domain_func;
525         ADS_STATUS status;
526         const char *salt = NULL;
527         char *std_salt = NULL;
528
529         status = ads_domain_func_level(r->in.ads, &domain_func);
530         if (!ADS_ERR_OK(status)) {
531                 libnet_join_set_error_string(mem_ctx, r,
532                         "failed to determine domain functional level: %s",
533                         ads_errstr(status));
534                 return false;
535         }
536
537         /* go ahead and setup the default salt */
538
539         std_salt = kerberos_standard_des_salt();
540         if (!std_salt) {
541                 libnet_join_set_error_string(mem_ctx, r,
542                         "failed to obtain standard DES salt");
543                 return false;
544         }
545
546         salt = talloc_strdup(mem_ctx, std_salt);
547         if (!salt) {
548                 return false;
549         }
550
551         SAFE_FREE(std_salt);
552
553         /* if it's a Windows functional domain, we have to look for the UPN */
554
555         if (domain_func == DS_DOMAIN_FUNCTION_2000) {
556                 char *upn;
557
558                 upn = ads_get_upn(r->in.ads, mem_ctx,
559                                   r->in.machine_name);
560                 if (upn) {
561                         salt = talloc_strdup(mem_ctx, upn);
562                         if (!salt) {
563                                 return false;
564                         }
565                 }
566         }
567
568         return kerberos_secrets_store_des_salt(salt);
569 }
570
571 /****************************************************************
572 ****************************************************************/
573
574 static ADS_STATUS libnet_join_post_processing_ads(TALLOC_CTX *mem_ctx,
575                                                   struct libnet_JoinCtx *r)
576 {
577         ADS_STATUS status;
578
579         if (!r->in.ads) {
580                 status = libnet_join_connect_ads(mem_ctx, r);
581                 if (!ADS_ERR_OK(status)) {
582                         return status;
583                 }
584         }
585
586         status = libnet_join_set_machine_spn(mem_ctx, r);
587         if (!ADS_ERR_OK(status)) {
588                 libnet_join_set_error_string(mem_ctx, r,
589                         "failed to set machine spn: %s",
590                         ads_errstr(status));
591                 return status;
592         }
593
594         status = libnet_join_set_os_attributes(mem_ctx, r);
595         if (!ADS_ERR_OK(status)) {
596                 libnet_join_set_error_string(mem_ctx, r,
597                         "failed to set machine os attributes: %s",
598                         ads_errstr(status));
599                 return status;
600         }
601
602         status = libnet_join_set_machine_upn(mem_ctx, r);
603         if (!ADS_ERR_OK(status)) {
604                 libnet_join_set_error_string(mem_ctx, r,
605                         "failed to set machine upn: %s",
606                         ads_errstr(status));
607                 return status;
608         }
609
610         if (!libnet_join_derive_salting_principal(mem_ctx, r)) {
611                 return ADS_ERROR_NT(NT_STATUS_UNSUCCESSFUL);
612         }
613
614         if (!libnet_join_create_keytab(mem_ctx, r)) {
615                 libnet_join_set_error_string(mem_ctx, r,
616                         "failed to create kerberos keytab");
617                 return ADS_ERROR_NT(NT_STATUS_UNSUCCESSFUL);
618         }
619
620         return ADS_SUCCESS;
621 }
622 #endif /* WITH_ADS */
623
624 /****************************************************************
625  Store the machine password and domain SID
626 ****************************************************************/
627
628 static bool libnet_join_joindomain_store_secrets(TALLOC_CTX *mem_ctx,
629                                                  struct libnet_JoinCtx *r)
630 {
631         if (!secrets_store_domain_sid(r->out.netbios_domain_name,
632                                       r->out.domain_sid))
633         {
634                 DEBUG(1,("Failed to save domain sid\n"));
635                 return false;
636         }
637
638         if (!secrets_store_machine_password(r->in.machine_password,
639                                             r->out.netbios_domain_name,
640                                             r->in.secure_channel_type))
641         {
642                 DEBUG(1,("Failed to save machine password\n"));
643                 return false;
644         }
645
646         return true;
647 }
648
649 /****************************************************************
650  Connect dc's IPC$ share
651 ****************************************************************/
652
653 static NTSTATUS libnet_join_connect_dc_ipc(const char *dc,
654                                            const char *user,
655                                            const char *pass,
656                                            bool use_kerberos,
657                                            struct cli_state **cli)
658 {
659         int flags = 0;
660
661         if (use_kerberos) {
662                 flags |= CLI_FULL_CONNECTION_USE_KERBEROS;
663         }
664
665         if (use_kerberos && pass) {
666                 flags |= CLI_FULL_CONNECTION_FALLBACK_AFTER_KERBEROS;
667         }
668
669         return cli_full_connection(cli, NULL,
670                                    dc,
671                                    NULL, 0,
672                                    "IPC$", "IPC",
673                                    user,
674                                    NULL,
675                                    pass,
676                                    flags,
677                                    Undefined, NULL);
678 }
679
680 /****************************************************************
681  Lookup domain dc's info
682 ****************************************************************/
683
684 static NTSTATUS libnet_join_lookup_dc_rpc(TALLOC_CTX *mem_ctx,
685                                           struct libnet_JoinCtx *r,
686                                           struct cli_state **cli)
687 {
688         struct rpc_pipe_client *pipe_hnd = NULL;
689         POLICY_HND lsa_pol;
690         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
691         union lsa_PolicyInformation *info = NULL;
692
693         status = libnet_join_connect_dc_ipc(r->in.dc_name,
694                                             r->in.admin_account,
695                                             r->in.admin_password,
696                                             r->in.use_kerberos,
697                                             cli);
698         if (!NT_STATUS_IS_OK(status)) {
699                 goto done;
700         }
701
702         status = cli_rpc_pipe_open_noauth(*cli, &ndr_table_lsarpc.syntax_id,
703                                           &pipe_hnd);
704         if (!NT_STATUS_IS_OK(status)) {
705                 DEBUG(0,("Error connecting to LSA pipe. Error was %s\n",
706                         nt_errstr(status)));
707                 goto done;
708         }
709
710         status = rpccli_lsa_open_policy(pipe_hnd, mem_ctx, true,
711                                         SEC_RIGHTS_MAXIMUM_ALLOWED, &lsa_pol);
712         if (!NT_STATUS_IS_OK(status)) {
713                 goto done;
714         }
715
716         status = rpccli_lsa_QueryInfoPolicy2(pipe_hnd, mem_ctx,
717                                              &lsa_pol,
718                                              LSA_POLICY_INFO_DNS,
719                                              &info);
720         if (NT_STATUS_IS_OK(status)) {
721                 r->out.domain_is_ad = true;
722                 r->out.netbios_domain_name = info->dns.name.string;
723                 r->out.dns_domain_name = info->dns.dns_domain.string;
724                 r->out.forest_name = info->dns.dns_forest.string;
725                 r->out.domain_sid = sid_dup_talloc(mem_ctx, info->dns.sid);
726                 NT_STATUS_HAVE_NO_MEMORY(r->out.domain_sid);
727         }
728
729         if (!NT_STATUS_IS_OK(status)) {
730                 status = rpccli_lsa_QueryInfoPolicy(pipe_hnd, mem_ctx,
731                                                     &lsa_pol,
732                                                     LSA_POLICY_INFO_ACCOUNT_DOMAIN,
733                                                     &info);
734                 if (!NT_STATUS_IS_OK(status)) {
735                         goto done;
736                 }
737
738                 r->out.netbios_domain_name = info->account_domain.name.string;
739                 r->out.domain_sid = sid_dup_talloc(mem_ctx, info->account_domain.sid);
740                 NT_STATUS_HAVE_NO_MEMORY(r->out.domain_sid);
741         }
742
743         rpccli_lsa_Close(pipe_hnd, mem_ctx, &lsa_pol);
744         TALLOC_FREE(pipe_hnd);
745
746  done:
747         return status;
748 }
749
750 /****************************************************************
751  Do the domain join
752 ****************************************************************/
753
754 static NTSTATUS libnet_join_joindomain_rpc(TALLOC_CTX *mem_ctx,
755                                            struct libnet_JoinCtx *r,
756                                            struct cli_state *cli)
757 {
758         struct rpc_pipe_client *pipe_hnd = NULL;
759         POLICY_HND sam_pol, domain_pol, user_pol;
760         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
761         char *acct_name;
762         struct lsa_String lsa_acct_name;
763         uint32_t user_rid;
764         uint32_t acct_flags = ACB_WSTRUST;
765         uchar md4_trust_password[16];
766         struct samr_Ids user_rids;
767         struct samr_Ids name_types;
768         union samr_UserInfo user_info;
769
770         struct samr_CryptPassword crypt_pwd;
771         struct samr_CryptPasswordEx crypt_pwd_ex;
772
773         ZERO_STRUCT(sam_pol);
774         ZERO_STRUCT(domain_pol);
775         ZERO_STRUCT(user_pol);
776
777         if (!r->in.machine_password) {
778                 r->in.machine_password = generate_random_str(mem_ctx, DEFAULT_TRUST_ACCOUNT_PASSWORD_LENGTH);
779                 NT_STATUS_HAVE_NO_MEMORY(r->in.machine_password);
780         }
781
782         /* Open the domain */
783
784         status = cli_rpc_pipe_open_noauth(cli, &ndr_table_samr.syntax_id,
785                                           &pipe_hnd);
786         if (!NT_STATUS_IS_OK(status)) {
787                 DEBUG(0,("Error connecting to SAM pipe. Error was %s\n",
788                         nt_errstr(status)));
789                 goto done;
790         }
791
792         status = rpccli_samr_Connect2(pipe_hnd, mem_ctx,
793                                       pipe_hnd->desthost,
794                                       SEC_RIGHTS_MAXIMUM_ALLOWED,
795                                       &sam_pol);
796         if (!NT_STATUS_IS_OK(status)) {
797                 goto done;
798         }
799
800         status = rpccli_samr_OpenDomain(pipe_hnd, mem_ctx,
801                                         &sam_pol,
802                                         SEC_RIGHTS_MAXIMUM_ALLOWED,
803                                         r->out.domain_sid,
804                                         &domain_pol);
805         if (!NT_STATUS_IS_OK(status)) {
806                 goto done;
807         }
808
809         /* Create domain user */
810
811         acct_name = talloc_asprintf(mem_ctx, "%s$", r->in.machine_name);
812         strlower_m(acct_name);
813
814         init_lsa_String(&lsa_acct_name, acct_name);
815
816         if (r->in.join_flags & WKSSVC_JOIN_FLAGS_ACCOUNT_CREATE) {
817                 uint32_t access_desired =
818                         SEC_GENERIC_READ | SEC_GENERIC_WRITE | SEC_GENERIC_EXECUTE |
819                         SEC_STD_WRITE_DAC | SEC_STD_DELETE |
820                         SAMR_USER_ACCESS_SET_PASSWORD |
821                         SAMR_USER_ACCESS_GET_ATTRIBUTES |
822                         SAMR_USER_ACCESS_SET_ATTRIBUTES;
823                 uint32_t access_granted = 0;
824
825                 /* Don't try to set any acct_flags flags other than ACB_WSTRUST */
826
827                 DEBUG(10,("Creating account with desired access mask: %d\n",
828                         access_desired));
829
830                 status = rpccli_samr_CreateUser2(pipe_hnd, mem_ctx,
831                                                  &domain_pol,
832                                                  &lsa_acct_name,
833                                                  ACB_WSTRUST,
834                                                  access_desired,
835                                                  &user_pol,
836                                                  &access_granted,
837                                                  &user_rid);
838                 if (!NT_STATUS_IS_OK(status) &&
839                     !NT_STATUS_EQUAL(status, NT_STATUS_USER_EXISTS)) {
840
841                         DEBUG(10,("Creation of workstation account failed: %s\n",
842                                 nt_errstr(status)));
843
844                         /* If NT_STATUS_ACCESS_DENIED then we have a valid
845                            username/password combo but the user does not have
846                            administrator access. */
847
848                         if (NT_STATUS_EQUAL(status, NT_STATUS_ACCESS_DENIED)) {
849                                 libnet_join_set_error_string(mem_ctx, r,
850                                         "User specified does not have "
851                                         "administrator privileges");
852                         }
853
854                         goto done;
855                 }
856
857                 if (NT_STATUS_EQUAL(status, NT_STATUS_USER_EXISTS)) {
858                         if (!(r->in.join_flags &
859                               WKSSVC_JOIN_FLAGS_DOMAIN_JOIN_IF_JOINED)) {
860                                 goto done;
861                         }
862                 }
863
864                 /* We *must* do this.... don't ask... */
865
866                 if (NT_STATUS_IS_OK(status)) {
867                         rpccli_samr_Close(pipe_hnd, mem_ctx, &user_pol);
868                 }
869         }
870
871         status = rpccli_samr_LookupNames(pipe_hnd, mem_ctx,
872                                          &domain_pol,
873                                          1,
874                                          &lsa_acct_name,
875                                          &user_rids,
876                                          &name_types);
877         if (!NT_STATUS_IS_OK(status)) {
878                 goto done;
879         }
880
881         if (name_types.ids[0] != SID_NAME_USER) {
882                 DEBUG(0,("%s is not a user account (type=%d)\n",
883                         acct_name, name_types.ids[0]));
884                 status = NT_STATUS_INVALID_WORKSTATION;
885                 goto done;
886         }
887
888         user_rid = user_rids.ids[0];
889
890         /* Open handle on user */
891
892         status = rpccli_samr_OpenUser(pipe_hnd, mem_ctx,
893                                       &domain_pol,
894                                       SEC_RIGHTS_MAXIMUM_ALLOWED,
895                                       user_rid,
896                                       &user_pol);
897         if (!NT_STATUS_IS_OK(status)) {
898                 goto done;
899         }
900
901         /* Create a random machine account password and generate the hash */
902
903         E_md4hash(r->in.machine_password, md4_trust_password);
904
905         init_samr_CryptPasswordEx(r->in.machine_password,
906                                   &cli->user_session_key,
907                                   &crypt_pwd_ex);
908
909         /* Fill in the additional account flags now */
910
911         acct_flags |= ACB_PWNOEXP;
912         if (r->out.domain_is_ad) {
913 #if !defined(ENCTYPE_ARCFOUR_HMAC)
914                 acct_flags |= ACB_USE_DES_KEY_ONLY;
915 #endif
916                 ;;
917         }
918
919         /* Set password and account flags on machine account */
920
921         ZERO_STRUCT(user_info.info25);
922
923         user_info.info25.info.fields_present = ACCT_NT_PWD_SET |
924                                                ACCT_LM_PWD_SET |
925                                                SAMR_FIELD_ACCT_FLAGS;
926
927         user_info.info25.info.acct_flags = acct_flags;
928         memcpy(&user_info.info25.password.data, crypt_pwd_ex.data,
929                sizeof(crypt_pwd_ex.data));
930
931         status = rpccli_samr_SetUserInfo(pipe_hnd, mem_ctx,
932                                          &user_pol,
933                                          25,
934                                          &user_info);
935
936         if (NT_STATUS_EQUAL(status, NT_STATUS(DCERPC_FAULT_INVALID_TAG))) {
937
938                 /* retry with level 24 */
939
940                 init_samr_CryptPassword(r->in.machine_password,
941                                         &cli->user_session_key,
942                                         &crypt_pwd);
943
944                 init_samr_user_info24(&user_info.info24, crypt_pwd.data, 24);
945
946                 status = rpccli_samr_SetUserInfo2(pipe_hnd, mem_ctx,
947                                                   &user_pol,
948                                                   24,
949                                                   &user_info);
950         }
951
952         if (!NT_STATUS_IS_OK(status)) {
953
954                 rpccli_samr_DeleteUser(pipe_hnd, mem_ctx,
955                                        &user_pol);
956
957                 libnet_join_set_error_string(mem_ctx, r,
958                         "Failed to set password for machine account (%s)\n",
959                         nt_errstr(status));
960                 goto done;
961         }
962
963         status = NT_STATUS_OK;
964
965  done:
966         if (!pipe_hnd) {
967                 return status;
968         }
969
970         if (is_valid_policy_hnd(&sam_pol)) {
971                 rpccli_samr_Close(pipe_hnd, mem_ctx, &sam_pol);
972         }
973         if (is_valid_policy_hnd(&domain_pol)) {
974                 rpccli_samr_Close(pipe_hnd, mem_ctx, &domain_pol);
975         }
976         if (is_valid_policy_hnd(&user_pol)) {
977                 rpccli_samr_Close(pipe_hnd, mem_ctx, &user_pol);
978         }
979         TALLOC_FREE(pipe_hnd);
980
981         return status;
982 }
983
984 /****************************************************************
985 ****************************************************************/
986
987 NTSTATUS libnet_join_ok(const char *netbios_domain_name,
988                         const char *machine_name,
989                         const char *dc_name)
990 {
991         uint32_t neg_flags = NETLOGON_NEG_AUTH2_ADS_FLAGS;
992         struct cli_state *cli = NULL;
993         struct rpc_pipe_client *pipe_hnd = NULL;
994         struct rpc_pipe_client *netlogon_pipe = NULL;
995         NTSTATUS status;
996         char *machine_password = NULL;
997         char *machine_account = NULL;
998
999         if (!dc_name) {
1000                 return NT_STATUS_INVALID_PARAMETER;
1001         }
1002
1003         if (!secrets_init()) {
1004                 return NT_STATUS_CANT_ACCESS_DOMAIN_INFO;
1005         }
1006
1007         machine_password = secrets_fetch_machine_password(netbios_domain_name,
1008                                                           NULL, NULL);
1009         if (!machine_password) {
1010                 return NT_STATUS_NO_TRUST_LSA_SECRET;
1011         }
1012
1013         asprintf(&machine_account, "%s$", machine_name);
1014         if (!machine_account) {
1015                 SAFE_FREE(machine_password);
1016                 return NT_STATUS_NO_MEMORY;
1017         }
1018
1019         status = cli_full_connection(&cli, NULL,
1020                                      dc_name,
1021                                      NULL, 0,
1022                                      "IPC$", "IPC",
1023                                      machine_account,
1024                                      NULL,
1025                                      machine_password,
1026                                      0,
1027                                      Undefined, NULL);
1028         free(machine_account);
1029         free(machine_password);
1030
1031         if (!NT_STATUS_IS_OK(status)) {
1032                 status = cli_full_connection(&cli, NULL,
1033                                              dc_name,
1034                                              NULL, 0,
1035                                              "IPC$", "IPC",
1036                                              "",
1037                                              NULL,
1038                                              "",
1039                                              0,
1040                                              Undefined, NULL);
1041         }
1042
1043         if (!NT_STATUS_IS_OK(status)) {
1044                 return status;
1045         }
1046
1047         status = get_schannel_session_key(cli, netbios_domain_name,
1048                                           &neg_flags, &netlogon_pipe);
1049         if (!NT_STATUS_IS_OK(status)) {
1050                 if (NT_STATUS_EQUAL(status, NT_STATUS_INVALID_NETWORK_RESPONSE)) {
1051                         cli_shutdown(cli);
1052                         return NT_STATUS_OK;
1053                 }
1054
1055                 DEBUG(0,("libnet_join_ok: failed to get schannel session "
1056                         "key from server %s for domain %s. Error was %s\n",
1057                 cli->desthost, netbios_domain_name, nt_errstr(status)));
1058                 cli_shutdown(cli);
1059                 return status;
1060         }
1061
1062         if (!lp_client_schannel()) {
1063                 cli_shutdown(cli);
1064                 return NT_STATUS_OK;
1065         }
1066
1067         status = cli_rpc_pipe_open_schannel_with_key(
1068                 cli, &ndr_table_netlogon.syntax_id, PIPE_AUTH_LEVEL_PRIVACY,
1069                 netbios_domain_name, netlogon_pipe->dc, &pipe_hnd);
1070
1071         cli_shutdown(cli);
1072
1073         if (!NT_STATUS_IS_OK(status)) {
1074                 DEBUG(0,("libnet_join_ok: failed to open schannel session "
1075                         "on netlogon pipe to server %s for domain %s. "
1076                         "Error was %s\n",
1077                         cli->desthost, netbios_domain_name, nt_errstr(status)));
1078                 return status;
1079         }
1080
1081         return NT_STATUS_OK;
1082 }
1083
1084 /****************************************************************
1085 ****************************************************************/
1086
1087 static WERROR libnet_join_post_verify(TALLOC_CTX *mem_ctx,
1088                                       struct libnet_JoinCtx *r)
1089 {
1090         NTSTATUS status;
1091
1092         status = libnet_join_ok(r->out.netbios_domain_name,
1093                                 r->in.machine_name,
1094                                 r->in.dc_name);
1095         if (!NT_STATUS_IS_OK(status)) {
1096                 libnet_join_set_error_string(mem_ctx, r,
1097                         "failed to verify domain membership after joining: %s",
1098                         get_friendly_nt_error_msg(status));
1099                 return WERR_SETUP_NOT_JOINED;
1100         }
1101
1102         return WERR_OK;
1103 }
1104
1105 /****************************************************************
1106 ****************************************************************/
1107
1108 static bool libnet_join_unjoindomain_remove_secrets(TALLOC_CTX *mem_ctx,
1109                                                     struct libnet_UnjoinCtx *r)
1110 {
1111         if (!secrets_delete_machine_password_ex(lp_workgroup())) {
1112                 return false;
1113         }
1114
1115         if (!secrets_delete_domain_sid(lp_workgroup())) {
1116                 return false;
1117         }
1118
1119         return true;
1120 }
1121
1122 /****************************************************************
1123 ****************************************************************/
1124
1125 static NTSTATUS libnet_join_unjoindomain_rpc(TALLOC_CTX *mem_ctx,
1126                                              struct libnet_UnjoinCtx *r)
1127 {
1128         struct cli_state *cli = NULL;
1129         struct rpc_pipe_client *pipe_hnd = NULL;
1130         POLICY_HND sam_pol, domain_pol, user_pol;
1131         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
1132         char *acct_name;
1133         uint32_t user_rid;
1134         struct lsa_String lsa_acct_name;
1135         struct samr_Ids user_rids;
1136         struct samr_Ids name_types;
1137         union samr_UserInfo *info = NULL;
1138
1139         ZERO_STRUCT(sam_pol);
1140         ZERO_STRUCT(domain_pol);
1141         ZERO_STRUCT(user_pol);
1142
1143         status = libnet_join_connect_dc_ipc(r->in.dc_name,
1144                                             r->in.admin_account,
1145                                             r->in.admin_password,
1146                                             r->in.use_kerberos,
1147                                             &cli);
1148         if (!NT_STATUS_IS_OK(status)) {
1149                 goto done;
1150         }
1151
1152         /* Open the domain */
1153
1154         status = cli_rpc_pipe_open_noauth(cli, &ndr_table_samr.syntax_id,
1155                                           &pipe_hnd);
1156         if (!NT_STATUS_IS_OK(status)) {
1157                 DEBUG(0,("Error connecting to SAM pipe. Error was %s\n",
1158                         nt_errstr(status)));
1159                 goto done;
1160         }
1161
1162         status = rpccli_samr_Connect2(pipe_hnd, mem_ctx,
1163                                       pipe_hnd->desthost,
1164                                       SEC_RIGHTS_MAXIMUM_ALLOWED,
1165                                       &sam_pol);
1166         if (!NT_STATUS_IS_OK(status)) {
1167                 goto done;
1168         }
1169
1170         status = rpccli_samr_OpenDomain(pipe_hnd, mem_ctx,
1171                                         &sam_pol,
1172                                         SEC_RIGHTS_MAXIMUM_ALLOWED,
1173                                         r->in.domain_sid,
1174                                         &domain_pol);
1175         if (!NT_STATUS_IS_OK(status)) {
1176                 goto done;
1177         }
1178
1179         /* Create domain user */
1180
1181         acct_name = talloc_asprintf(mem_ctx, "%s$", r->in.machine_name);
1182         strlower_m(acct_name);
1183
1184         init_lsa_String(&lsa_acct_name, acct_name);
1185
1186         status = rpccli_samr_LookupNames(pipe_hnd, mem_ctx,
1187                                          &domain_pol,
1188                                          1,
1189                                          &lsa_acct_name,
1190                                          &user_rids,
1191                                          &name_types);
1192
1193         if (!NT_STATUS_IS_OK(status)) {
1194                 goto done;
1195         }
1196
1197         if (name_types.ids[0] != SID_NAME_USER) {
1198                 DEBUG(0, ("%s is not a user account (type=%d)\n", acct_name,
1199                         name_types.ids[0]));
1200                 status = NT_STATUS_INVALID_WORKSTATION;
1201                 goto done;
1202         }
1203
1204         user_rid = user_rids.ids[0];
1205
1206         /* Open handle on user */
1207
1208         status = rpccli_samr_OpenUser(pipe_hnd, mem_ctx,
1209                                       &domain_pol,
1210                                       SEC_RIGHTS_MAXIMUM_ALLOWED,
1211                                       user_rid,
1212                                       &user_pol);
1213         if (!NT_STATUS_IS_OK(status)) {
1214                 goto done;
1215         }
1216
1217         /* Get user info */
1218
1219         status = rpccli_samr_QueryUserInfo(pipe_hnd, mem_ctx,
1220                                            &user_pol,
1221                                            16,
1222                                            &info);
1223         if (!NT_STATUS_IS_OK(status)) {
1224                 rpccli_samr_Close(pipe_hnd, mem_ctx, &user_pol);
1225                 goto done;
1226         }
1227
1228         /* now disable and setuser info */
1229
1230         info->info16.acct_flags |= ACB_DISABLED;
1231
1232         status = rpccli_samr_SetUserInfo(pipe_hnd, mem_ctx,
1233                                          &user_pol,
1234                                          16,
1235                                          info);
1236
1237         rpccli_samr_Close(pipe_hnd, mem_ctx, &user_pol);
1238
1239 done:
1240         if (pipe_hnd) {
1241                 if (is_valid_policy_hnd(&domain_pol)) {
1242                         rpccli_samr_Close(pipe_hnd, mem_ctx, &domain_pol);
1243                 }
1244                 if (is_valid_policy_hnd(&sam_pol)) {
1245                         rpccli_samr_Close(pipe_hnd, mem_ctx, &sam_pol);
1246                 }
1247                 TALLOC_FREE(pipe_hnd);
1248         }
1249
1250         if (cli) {
1251                 cli_shutdown(cli);
1252         }
1253
1254         return status;
1255 }
1256
1257 /****************************************************************
1258 ****************************************************************/
1259
1260 static WERROR do_join_modify_vals_config(struct libnet_JoinCtx *r)
1261 {
1262         WERROR werr;
1263         struct smbconf_ctx *ctx;
1264
1265         werr = smbconf_init_reg(r, &ctx, NULL);
1266         if (!W_ERROR_IS_OK(werr)) {
1267                 goto done;
1268         }
1269
1270         if (!(r->in.join_flags & WKSSVC_JOIN_FLAGS_JOIN_TYPE)) {
1271
1272                 werr = smbconf_set_global_parameter(ctx, "security", "user");
1273                 W_ERROR_NOT_OK_GOTO_DONE(werr);
1274
1275                 werr = smbconf_set_global_parameter(ctx, "workgroup",
1276                                                     r->in.domain_name);
1277
1278                 smbconf_delete_global_parameter(ctx, "realm");
1279                 goto done;
1280         }
1281
1282         werr = smbconf_set_global_parameter(ctx, "security", "domain");
1283         W_ERROR_NOT_OK_GOTO_DONE(werr);
1284
1285         werr = smbconf_set_global_parameter(ctx, "workgroup",
1286                                             r->out.netbios_domain_name);
1287         W_ERROR_NOT_OK_GOTO_DONE(werr);
1288
1289         if (r->out.domain_is_ad) {
1290                 werr = smbconf_set_global_parameter(ctx, "security", "ads");
1291                 W_ERROR_NOT_OK_GOTO_DONE(werr);
1292
1293                 werr = smbconf_set_global_parameter(ctx, "realm",
1294                                                     r->out.dns_domain_name);
1295                 W_ERROR_NOT_OK_GOTO_DONE(werr);
1296         }
1297
1298  done:
1299         smbconf_shutdown(ctx);
1300         return werr;
1301 }
1302
1303 /****************************************************************
1304 ****************************************************************/
1305
1306 static WERROR do_unjoin_modify_vals_config(struct libnet_UnjoinCtx *r)
1307 {
1308         WERROR werr = WERR_OK;
1309         struct smbconf_ctx *ctx;
1310
1311         werr = smbconf_init_reg(r, &ctx, NULL);
1312         if (!W_ERROR_IS_OK(werr)) {
1313                 goto done;
1314         }
1315
1316         if (r->in.unjoin_flags & WKSSVC_JOIN_FLAGS_JOIN_TYPE) {
1317
1318                 werr = smbconf_set_global_parameter(ctx, "security", "user");
1319                 W_ERROR_NOT_OK_GOTO_DONE(werr);
1320
1321                 werr = smbconf_delete_global_parameter(ctx, "workgroup");
1322                 W_ERROR_NOT_OK_GOTO_DONE(werr);
1323
1324                 smbconf_delete_global_parameter(ctx, "realm");
1325         }
1326
1327  done:
1328         smbconf_shutdown(ctx);
1329         return werr;
1330 }
1331
1332 /****************************************************************
1333 ****************************************************************/
1334
1335 static WERROR do_JoinConfig(struct libnet_JoinCtx *r)
1336 {
1337         WERROR werr;
1338
1339         if (!W_ERROR_IS_OK(r->out.result)) {
1340                 return r->out.result;
1341         }
1342
1343         if (!r->in.modify_config) {
1344                 return WERR_OK;
1345         }
1346
1347         werr = do_join_modify_vals_config(r);
1348         if (!W_ERROR_IS_OK(werr)) {
1349                 return werr;
1350         }
1351
1352         lp_load(get_dyn_CONFIGFILE(),true,false,false,true);
1353
1354         r->out.modified_config = true;
1355         r->out.result = werr;
1356
1357         return werr;
1358 }
1359
1360 /****************************************************************
1361 ****************************************************************/
1362
1363 static WERROR libnet_unjoin_config(struct libnet_UnjoinCtx *r)
1364 {
1365         WERROR werr;
1366
1367         if (!W_ERROR_IS_OK(r->out.result)) {
1368                 return r->out.result;
1369         }
1370
1371         if (!r->in.modify_config) {
1372                 return WERR_OK;
1373         }
1374
1375         werr = do_unjoin_modify_vals_config(r);
1376         if (!W_ERROR_IS_OK(werr)) {
1377                 return werr;
1378         }
1379
1380         lp_load(get_dyn_CONFIGFILE(),true,false,false,true);
1381
1382         r->out.modified_config = true;
1383         r->out.result = werr;
1384
1385         return werr;
1386 }
1387
1388 /****************************************************************
1389 ****************************************************************/
1390
1391 static bool libnet_parse_domain_dc(TALLOC_CTX *mem_ctx,
1392                                    const char *domain_str,
1393                                    const char **domain_p,
1394                                    const char **dc_p)
1395 {
1396         char *domain = NULL;
1397         char *dc = NULL;
1398         const char *p = NULL;
1399
1400         if (!domain_str || !domain_p || !dc_p) {
1401                 return false;
1402         }
1403
1404         p = strchr_m(domain_str, '\\');
1405
1406         if (p != NULL) {
1407                 domain = talloc_strndup(mem_ctx, domain_str,
1408                                          PTR_DIFF(p, domain_str));
1409                 dc = talloc_strdup(mem_ctx, p+1);
1410                 if (!dc) {
1411                         return false;
1412                 }
1413         } else {
1414                 domain = talloc_strdup(mem_ctx, domain_str);
1415                 dc = NULL;
1416         }
1417         if (!domain) {
1418                 return false;
1419         }
1420
1421         *domain_p = domain;
1422
1423         if (!*dc_p && dc) {
1424                 *dc_p = dc;
1425         }
1426
1427         return true;
1428 }
1429
1430 /****************************************************************
1431 ****************************************************************/
1432
1433 static WERROR libnet_join_pre_processing(TALLOC_CTX *mem_ctx,
1434                                          struct libnet_JoinCtx *r)
1435 {
1436         if (!r->in.domain_name) {
1437                 libnet_join_set_error_string(mem_ctx, r,
1438                         "No domain name defined");
1439                 return WERR_INVALID_PARAM;
1440         }
1441
1442         if (!libnet_parse_domain_dc(mem_ctx, r->in.domain_name,
1443                                     &r->in.domain_name,
1444                                     &r->in.dc_name)) {
1445                 libnet_join_set_error_string(mem_ctx, r,
1446                         "Failed to parse domain name");
1447                 return WERR_INVALID_PARAM;
1448         }
1449
1450         if (IS_DC) {
1451                 return WERR_SETUP_DOMAIN_CONTROLLER;
1452         }
1453
1454         if (!secrets_init()) {
1455                 libnet_join_set_error_string(mem_ctx, r,
1456                         "Unable to open secrets database");
1457                 return WERR_CAN_NOT_COMPLETE;
1458         }
1459
1460         return WERR_OK;
1461 }
1462
1463 /****************************************************************
1464 ****************************************************************/
1465
1466 static void libnet_join_add_dom_rids_to_builtins(struct dom_sid *domain_sid)
1467 {
1468         NTSTATUS status;
1469
1470         /* Try adding dom admins to builtin\admins. Only log failures. */
1471         status = create_builtin_administrators(domain_sid);
1472         if (NT_STATUS_EQUAL(status, NT_STATUS_PROTOCOL_UNREACHABLE)) {
1473                 DEBUG(10,("Unable to auto-add domain administrators to "
1474                           "BUILTIN\\Administrators during join because "
1475                           "winbindd must be running."));
1476         } else if (!NT_STATUS_IS_OK(status)) {
1477                 DEBUG(5, ("Failed to auto-add domain administrators to "
1478                           "BUILTIN\\Administrators during join: %s\n",
1479                           nt_errstr(status)));
1480         }
1481
1482         /* Try adding dom users to builtin\users. Only log failures. */
1483         status = create_builtin_users(domain_sid);
1484         if (NT_STATUS_EQUAL(status, NT_STATUS_PROTOCOL_UNREACHABLE)) {
1485                 DEBUG(10,("Unable to auto-add domain users to BUILTIN\\users "
1486                           "during join because winbindd must be running."));
1487         } else if (!NT_STATUS_IS_OK(status)) {
1488                 DEBUG(5, ("Failed to auto-add domain administrators to "
1489                           "BUILTIN\\Administrators during join: %s\n",
1490                           nt_errstr(status)));
1491         }
1492 }
1493
1494 /****************************************************************
1495 ****************************************************************/
1496
1497 static WERROR libnet_join_post_processing(TALLOC_CTX *mem_ctx,
1498                                           struct libnet_JoinCtx *r)
1499 {
1500         WERROR werr;
1501
1502         if (!W_ERROR_IS_OK(r->out.result)) {
1503                 return r->out.result;
1504         }
1505
1506         werr = do_JoinConfig(r);
1507         if (!W_ERROR_IS_OK(werr)) {
1508                 return werr;
1509         }
1510
1511         if (!(r->in.join_flags & WKSSVC_JOIN_FLAGS_JOIN_TYPE)) {
1512                 return WERR_OK;
1513         }
1514
1515         saf_store(r->in.domain_name, r->in.dc_name);
1516
1517 #ifdef WITH_ADS
1518         if (r->out.domain_is_ad) {
1519                 ADS_STATUS ads_status;
1520
1521                 ads_status  = libnet_join_post_processing_ads(mem_ctx, r);
1522                 if (!ADS_ERR_OK(ads_status)) {
1523                         return WERR_GENERAL_FAILURE;
1524                 }
1525         }
1526 #endif /* WITH_ADS */
1527
1528         libnet_join_add_dom_rids_to_builtins(r->out.domain_sid);
1529
1530         return WERR_OK;
1531 }
1532
1533 /****************************************************************
1534 ****************************************************************/
1535
1536 static int libnet_destroy_JoinCtx(struct libnet_JoinCtx *r)
1537 {
1538         const char *krb5_cc_env = NULL;
1539
1540         if (r->in.ads) {
1541                 ads_destroy(&r->in.ads);
1542         }
1543
1544         krb5_cc_env = getenv(KRB5_ENV_CCNAME);
1545         if (krb5_cc_env && StrCaseCmp(krb5_cc_env, "MEMORY:libnetjoin")) {
1546                 unsetenv(KRB5_ENV_CCNAME);
1547         }
1548
1549         return 0;
1550 }
1551
1552 /****************************************************************
1553 ****************************************************************/
1554
1555 static int libnet_destroy_UnjoinCtx(struct libnet_UnjoinCtx *r)
1556 {
1557         const char *krb5_cc_env = NULL;
1558
1559         if (r->in.ads) {
1560                 ads_destroy(&r->in.ads);
1561         }
1562
1563         krb5_cc_env = getenv(KRB5_ENV_CCNAME);
1564         if (krb5_cc_env && StrCaseCmp(krb5_cc_env, "MEMORY:libnetjoin")) {
1565                 unsetenv(KRB5_ENV_CCNAME);
1566         }
1567
1568         return 0;
1569 }
1570
1571 /****************************************************************
1572 ****************************************************************/
1573
1574 WERROR libnet_init_JoinCtx(TALLOC_CTX *mem_ctx,
1575                            struct libnet_JoinCtx **r)
1576 {
1577         struct libnet_JoinCtx *ctx;
1578         const char *krb5_cc_env = NULL;
1579
1580         ctx = talloc_zero(mem_ctx, struct libnet_JoinCtx);
1581         if (!ctx) {
1582                 return WERR_NOMEM;
1583         }
1584
1585         talloc_set_destructor(ctx, libnet_destroy_JoinCtx);
1586
1587         ctx->in.machine_name = talloc_strdup(mem_ctx, global_myname());
1588         W_ERROR_HAVE_NO_MEMORY(ctx->in.machine_name);
1589
1590         krb5_cc_env = getenv(KRB5_ENV_CCNAME);
1591         if (!krb5_cc_env || (strlen(krb5_cc_env) == 0)) {
1592                 krb5_cc_env = talloc_strdup(mem_ctx, "MEMORY:libnetjoin");
1593                 W_ERROR_HAVE_NO_MEMORY(krb5_cc_env);
1594                 setenv(KRB5_ENV_CCNAME, krb5_cc_env, 1);
1595         }
1596
1597         ctx->in.secure_channel_type = SEC_CHAN_WKSTA;
1598
1599         *r = ctx;
1600
1601         return WERR_OK;
1602 }
1603
1604 /****************************************************************
1605 ****************************************************************/
1606
1607 WERROR libnet_init_UnjoinCtx(TALLOC_CTX *mem_ctx,
1608                              struct libnet_UnjoinCtx **r)
1609 {
1610         struct libnet_UnjoinCtx *ctx;
1611         const char *krb5_cc_env = NULL;
1612
1613         ctx = talloc_zero(mem_ctx, struct libnet_UnjoinCtx);
1614         if (!ctx) {
1615                 return WERR_NOMEM;
1616         }
1617
1618         talloc_set_destructor(ctx, libnet_destroy_UnjoinCtx);
1619
1620         ctx->in.machine_name = talloc_strdup(mem_ctx, global_myname());
1621         W_ERROR_HAVE_NO_MEMORY(ctx->in.machine_name);
1622
1623         krb5_cc_env = getenv(KRB5_ENV_CCNAME);
1624         if (!krb5_cc_env || (strlen(krb5_cc_env) == 0)) {
1625                 krb5_cc_env = talloc_strdup(mem_ctx, "MEMORY:libnetjoin");
1626                 W_ERROR_HAVE_NO_MEMORY(krb5_cc_env);
1627                 setenv(KRB5_ENV_CCNAME, krb5_cc_env, 1);
1628         }
1629
1630         *r = ctx;
1631
1632         return WERR_OK;
1633 }
1634
1635 /****************************************************************
1636 ****************************************************************/
1637
1638 static WERROR libnet_join_check_config(TALLOC_CTX *mem_ctx,
1639                                        struct libnet_JoinCtx *r)
1640 {
1641         /* check if configuration is already set correctly */
1642
1643         switch (r->out.domain_is_ad) {
1644                 case false:
1645                         if ((strequal(lp_workgroup(),
1646                                       r->out.netbios_domain_name)) &&
1647                             (lp_security() == SEC_DOMAIN)) {
1648                                 /* nothing to be done */
1649                                 return WERR_OK;
1650                         }
1651                         break;
1652                 case true:
1653                         if ((strequal(lp_workgroup(),
1654                                       r->out.netbios_domain_name)) &&
1655                             (strequal(lp_realm(),
1656                                       r->out.dns_domain_name)) &&
1657                             ((lp_security() == SEC_ADS) ||
1658                              (lp_security() == SEC_DOMAIN))) {
1659                                 /* nothing to be done */
1660                                 return WERR_OK;
1661                         }
1662                         break;
1663         }
1664
1665         /* check if we are supposed to manipulate configuration */
1666
1667         if (!r->in.modify_config) {
1668                 libnet_join_set_error_string(mem_ctx, r,
1669                         "Invalid configuration and configuration modification "
1670                         "was not requested");
1671                 return WERR_CAN_NOT_COMPLETE;
1672         }
1673
1674         /* check if we are able to manipulate configuration */
1675
1676         if (!lp_config_backend_is_registry()) {
1677                 libnet_join_set_error_string(mem_ctx, r,
1678                         "Configuration manipulation requested but not "
1679                         "supported by backend");
1680                 return WERR_NOT_SUPPORTED;
1681         }
1682
1683         return WERR_OK;
1684 }
1685
1686 /****************************************************************
1687 ****************************************************************/
1688
1689 static WERROR libnet_DomainJoin(TALLOC_CTX *mem_ctx,
1690                                 struct libnet_JoinCtx *r)
1691 {
1692         NTSTATUS status;
1693         WERROR werr;
1694         struct cli_state *cli = NULL;
1695 #ifdef WITH_ADS
1696         ADS_STATUS ads_status;
1697 #endif /* WITH_ADS */
1698
1699         if (!r->in.dc_name) {
1700                 struct netr_DsRGetDCNameInfo *info;
1701                 const char *dc;
1702                 status = dsgetdcname(mem_ctx,
1703                                      r->in.msg_ctx,
1704                                      r->in.domain_name,
1705                                      NULL,
1706                                      NULL,
1707                                      DS_DIRECTORY_SERVICE_REQUIRED |
1708                                      DS_WRITABLE_REQUIRED |
1709                                      DS_RETURN_DNS_NAME,
1710                                      &info);
1711                 if (!NT_STATUS_IS_OK(status)) {
1712                         libnet_join_set_error_string(mem_ctx, r,
1713                                 "failed to find DC for domain %s",
1714                                 r->in.domain_name,
1715                                 get_friendly_nt_error_msg(status));
1716                         return WERR_DOMAIN_CONTROLLER_NOT_FOUND;
1717                 }
1718
1719                 dc = strip_hostname(info->dc_unc);
1720                 r->in.dc_name = talloc_strdup(mem_ctx, dc);
1721                 W_ERROR_HAVE_NO_MEMORY(r->in.dc_name);
1722         }
1723
1724         status = libnet_join_lookup_dc_rpc(mem_ctx, r, &cli);
1725         if (!NT_STATUS_IS_OK(status)) {
1726                 libnet_join_set_error_string(mem_ctx, r,
1727                         "failed to lookup DC info for domain '%s' over rpc: %s",
1728                         r->in.domain_name, get_friendly_nt_error_msg(status));
1729                 return ntstatus_to_werror(status);
1730         }
1731
1732         werr = libnet_join_check_config(mem_ctx, r);
1733         if (!W_ERROR_IS_OK(werr)) {
1734                 goto done;
1735         }
1736
1737 #ifdef WITH_ADS
1738         if (r->out.domain_is_ad && r->in.account_ou) {
1739
1740                 ads_status = libnet_join_connect_ads(mem_ctx, r);
1741                 if (!ADS_ERR_OK(ads_status)) {
1742                         return WERR_DEFAULT_JOIN_REQUIRED;
1743                 }
1744
1745                 ads_status = libnet_join_precreate_machine_acct(mem_ctx, r);
1746                 if (!ADS_ERR_OK(ads_status)) {
1747                         libnet_join_set_error_string(mem_ctx, r,
1748                                 "failed to precreate account in ou %s: %s",
1749                                 r->in.account_ou,
1750                                 ads_errstr(ads_status));
1751                         return WERR_DEFAULT_JOIN_REQUIRED;
1752                 }
1753
1754                 r->in.join_flags &= ~WKSSVC_JOIN_FLAGS_ACCOUNT_CREATE;
1755         }
1756 #endif /* WITH_ADS */
1757
1758         status = libnet_join_joindomain_rpc(mem_ctx, r, cli);
1759         if (!NT_STATUS_IS_OK(status)) {
1760                 libnet_join_set_error_string(mem_ctx, r,
1761                         "failed to join domain '%s' over rpc: %s",
1762                         r->in.domain_name, get_friendly_nt_error_msg(status));
1763                 if (NT_STATUS_EQUAL(status, NT_STATUS_USER_EXISTS)) {
1764                         return WERR_SETUP_ALREADY_JOINED;
1765                 }
1766                 werr = ntstatus_to_werror(status);
1767                 goto done;
1768         }
1769
1770         if (!libnet_join_joindomain_store_secrets(mem_ctx, r)) {
1771                 werr = WERR_SETUP_NOT_JOINED;
1772                 goto done;
1773         }
1774
1775         werr = WERR_OK;
1776
1777  done:
1778         if (cli) {
1779                 cli_shutdown(cli);
1780         }
1781
1782         return werr;
1783 }
1784
1785 /****************************************************************
1786 ****************************************************************/
1787
1788 static WERROR libnet_join_rollback(TALLOC_CTX *mem_ctx,
1789                                    struct libnet_JoinCtx *r)
1790 {
1791         WERROR werr;
1792         struct libnet_UnjoinCtx *u = NULL;
1793
1794         werr = libnet_init_UnjoinCtx(mem_ctx, &u);
1795         if (!W_ERROR_IS_OK(werr)) {
1796                 return werr;
1797         }
1798
1799         u->in.debug             = r->in.debug;
1800         u->in.dc_name           = r->in.dc_name;
1801         u->in.domain_name       = r->in.domain_name;
1802         u->in.admin_account     = r->in.admin_account;
1803         u->in.admin_password    = r->in.admin_password;
1804         u->in.modify_config     = r->in.modify_config;
1805         u->in.unjoin_flags      = WKSSVC_JOIN_FLAGS_JOIN_TYPE |
1806                                   WKSSVC_JOIN_FLAGS_ACCOUNT_DELETE;
1807
1808         werr = libnet_Unjoin(mem_ctx, u);
1809         TALLOC_FREE(u);
1810
1811         return werr;
1812 }
1813
1814 /****************************************************************
1815 ****************************************************************/
1816
1817 WERROR libnet_Join(TALLOC_CTX *mem_ctx,
1818                    struct libnet_JoinCtx *r)
1819 {
1820         WERROR werr;
1821
1822         if (r->in.debug) {
1823                 LIBNET_JOIN_IN_DUMP_CTX(mem_ctx, r);
1824         }
1825
1826         werr = libnet_join_pre_processing(mem_ctx, r);
1827         if (!W_ERROR_IS_OK(werr)) {
1828                 goto done;
1829         }
1830
1831         if (r->in.join_flags & WKSSVC_JOIN_FLAGS_JOIN_TYPE) {
1832                 werr = libnet_DomainJoin(mem_ctx, r);
1833                 if (!W_ERROR_IS_OK(werr)) {
1834                         goto done;
1835                 }
1836         }
1837
1838         werr = libnet_join_post_processing(mem_ctx, r);
1839         if (!W_ERROR_IS_OK(werr)) {
1840                 goto done;
1841         }
1842
1843         if (r->in.join_flags & WKSSVC_JOIN_FLAGS_JOIN_TYPE) {
1844                 werr = libnet_join_post_verify(mem_ctx, r);
1845                 if (!W_ERROR_IS_OK(werr)) {
1846                         libnet_join_rollback(mem_ctx, r);
1847                 }
1848         }
1849
1850  done:
1851         r->out.result = werr;
1852
1853         if (r->in.debug) {
1854                 LIBNET_JOIN_OUT_DUMP_CTX(mem_ctx, r);
1855         }
1856         return werr;
1857 }
1858
1859 /****************************************************************
1860 ****************************************************************/
1861
1862 static WERROR libnet_DomainUnjoin(TALLOC_CTX *mem_ctx,
1863                                   struct libnet_UnjoinCtx *r)
1864 {
1865         NTSTATUS status;
1866
1867         if (!r->in.domain_sid) {
1868                 struct dom_sid sid;
1869                 if (!secrets_fetch_domain_sid(lp_workgroup(), &sid)) {
1870                         libnet_unjoin_set_error_string(mem_ctx, r,
1871                                 "Unable to fetch domain sid: are we joined?");
1872                         return WERR_SETUP_NOT_JOINED;
1873                 }
1874                 r->in.domain_sid = sid_dup_talloc(mem_ctx, &sid);
1875                 W_ERROR_HAVE_NO_MEMORY(r->in.domain_sid);
1876         }
1877
1878         if (!r->in.dc_name) {
1879                 struct netr_DsRGetDCNameInfo *info;
1880                 const char *dc;
1881                 status = dsgetdcname(mem_ctx,
1882                                      r->in.msg_ctx,
1883                                      r->in.domain_name,
1884                                      NULL,
1885                                      NULL,
1886                                      DS_DIRECTORY_SERVICE_REQUIRED |
1887                                      DS_WRITABLE_REQUIRED |
1888                                      DS_RETURN_DNS_NAME,
1889                                      &info);
1890                 if (!NT_STATUS_IS_OK(status)) {
1891                         libnet_unjoin_set_error_string(mem_ctx, r,
1892                                 "failed to find DC for domain %s",
1893                                 r->in.domain_name,
1894                                 get_friendly_nt_error_msg(status));
1895                         return WERR_DOMAIN_CONTROLLER_NOT_FOUND;
1896                 }
1897
1898                 dc = strip_hostname(info->dc_unc);
1899                 r->in.dc_name = talloc_strdup(mem_ctx, dc);
1900                 W_ERROR_HAVE_NO_MEMORY(r->in.dc_name);
1901         }
1902
1903         status = libnet_join_unjoindomain_rpc(mem_ctx, r);
1904         if (!NT_STATUS_IS_OK(status)) {
1905                 libnet_unjoin_set_error_string(mem_ctx, r,
1906                         "failed to disable machine account via rpc: %s",
1907                         get_friendly_nt_error_msg(status));
1908                 if (NT_STATUS_EQUAL(status, NT_STATUS_NO_SUCH_USER)) {
1909                         return WERR_SETUP_NOT_JOINED;
1910                 }
1911                 return ntstatus_to_werror(status);
1912         }
1913
1914         r->out.disabled_machine_account = true;
1915
1916 #ifdef WITH_ADS
1917         if (r->in.unjoin_flags & WKSSVC_JOIN_FLAGS_ACCOUNT_DELETE) {
1918                 ADS_STATUS ads_status;
1919                 libnet_unjoin_connect_ads(mem_ctx, r);
1920                 ads_status = libnet_unjoin_remove_machine_acct(mem_ctx, r);
1921                 if (!ADS_ERR_OK(ads_status)) {
1922                         libnet_unjoin_set_error_string(mem_ctx, r,
1923                                 "failed to remove machine account from AD: %s",
1924                                 ads_errstr(ads_status));
1925                 } else {
1926                         r->out.deleted_machine_account = true;
1927                         /* dirty hack */
1928                         r->out.dns_domain_name = talloc_strdup(mem_ctx,
1929                                                                r->in.ads->server.realm);
1930                         W_ERROR_HAVE_NO_MEMORY(r->out.dns_domain_name);
1931                 }
1932         }
1933 #endif /* WITH_ADS */
1934
1935         libnet_join_unjoindomain_remove_secrets(mem_ctx, r);
1936
1937         return WERR_OK;
1938 }
1939
1940 /****************************************************************
1941 ****************************************************************/
1942
1943 static WERROR libnet_unjoin_pre_processing(TALLOC_CTX *mem_ctx,
1944                                            struct libnet_UnjoinCtx *r)
1945 {
1946         if (!r->in.domain_name) {
1947                 libnet_unjoin_set_error_string(mem_ctx, r,
1948                         "No domain name defined");
1949                 return WERR_INVALID_PARAM;
1950         }
1951
1952         if (!libnet_parse_domain_dc(mem_ctx, r->in.domain_name,
1953                                     &r->in.domain_name,
1954                                     &r->in.dc_name)) {
1955                 libnet_unjoin_set_error_string(mem_ctx, r,
1956                         "Failed to parse domain name");
1957                 return WERR_INVALID_PARAM;
1958         }
1959
1960         if (IS_DC) {
1961                 return WERR_SETUP_DOMAIN_CONTROLLER;
1962         }
1963
1964         if (!secrets_init()) {
1965                 libnet_unjoin_set_error_string(mem_ctx, r,
1966                         "Unable to open secrets database");
1967                 return WERR_CAN_NOT_COMPLETE;
1968         }
1969
1970         return WERR_OK;
1971 }
1972
1973 /****************************************************************
1974 ****************************************************************/
1975
1976 static WERROR libnet_unjoin_post_processing(TALLOC_CTX *mem_ctx,
1977                                             struct libnet_UnjoinCtx *r)
1978 {
1979         saf_delete(r->out.netbios_domain_name);
1980         saf_delete(r->out.dns_domain_name);
1981
1982         return libnet_unjoin_config(r);
1983 }
1984
1985 /****************************************************************
1986 ****************************************************************/
1987
1988 WERROR libnet_Unjoin(TALLOC_CTX *mem_ctx,
1989                      struct libnet_UnjoinCtx *r)
1990 {
1991         WERROR werr;
1992
1993         if (r->in.debug) {
1994                 LIBNET_UNJOIN_IN_DUMP_CTX(mem_ctx, r);
1995         }
1996
1997         werr = libnet_unjoin_pre_processing(mem_ctx, r);
1998         if (!W_ERROR_IS_OK(werr)) {
1999                 goto done;
2000         }
2001
2002         if (r->in.unjoin_flags & WKSSVC_JOIN_FLAGS_JOIN_TYPE) {
2003                 werr = libnet_DomainUnjoin(mem_ctx, r);
2004                 if (!W_ERROR_IS_OK(werr)) {
2005                         libnet_unjoin_config(r);
2006                         goto done;
2007                 }
2008         }
2009
2010         werr = libnet_unjoin_post_processing(mem_ctx, r);
2011         if (!W_ERROR_IS_OK(werr)) {
2012                 goto done;
2013         }
2014
2015  done:
2016         r->out.result = werr;
2017
2018         if (r->in.debug) {
2019                 LIBNET_UNJOIN_OUT_DUMP_CTX(mem_ctx, r);
2020         }
2021
2022         return werr;
2023 }