Merge branch 'v3-2-test' of ssh://git.samba.org/data/git/samba into v3-2-test
[jra/samba/.git] / source / 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 /****************************************************************
54 ****************************************************************/
55
56 static void libnet_join_set_error_string(TALLOC_CTX *mem_ctx,
57                                          struct libnet_JoinCtx *r,
58                                          const char *format, ...)
59 {
60         va_list args;
61
62         if (r->out.error_string) {
63                 return;
64         }
65
66         va_start(args, format);
67         r->out.error_string = talloc_vasprintf(mem_ctx, format, args);
68         va_end(args);
69 }
70
71 /****************************************************************
72 ****************************************************************/
73
74 static void libnet_unjoin_set_error_string(TALLOC_CTX *mem_ctx,
75                                            struct libnet_UnjoinCtx *r,
76                                            const char *format, ...)
77 {
78         va_list args;
79
80         if (r->out.error_string) {
81                 return;
82         }
83
84         va_start(args, format);
85         r->out.error_string = talloc_vasprintf(mem_ctx, format, args);
86         va_end(args);
87 }
88
89 #ifdef WITH_ADS
90
91 /****************************************************************
92 ****************************************************************/
93
94 static ADS_STATUS libnet_connect_ads(const char *dns_domain_name,
95                                      const char *netbios_domain_name,
96                                      const char *dc_name,
97                                      const char *user_name,
98                                      const char *password,
99                                      ADS_STRUCT **ads)
100 {
101         ADS_STATUS status;
102         ADS_STRUCT *my_ads = NULL;
103
104         my_ads = ads_init(dns_domain_name,
105                           netbios_domain_name,
106                           dc_name);
107         if (!my_ads) {
108                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
109         }
110
111         if (user_name) {
112                 SAFE_FREE(my_ads->auth.user_name);
113                 my_ads->auth.user_name = SMB_STRDUP(user_name);
114         }
115
116         if (password) {
117                 SAFE_FREE(my_ads->auth.password);
118                 my_ads->auth.password = SMB_STRDUP(password);
119         }
120
121         status = ads_connect(my_ads);
122         if (!ADS_ERR_OK(status)) {
123                 ads_destroy(&my_ads);
124                 return status;
125         }
126
127         *ads = my_ads;
128         return ADS_SUCCESS;
129 }
130
131 /****************************************************************
132 ****************************************************************/
133
134 static ADS_STATUS libnet_join_connect_ads(TALLOC_CTX *mem_ctx,
135                                           struct libnet_JoinCtx *r)
136 {
137         ADS_STATUS status;
138
139         status = libnet_connect_ads(r->in.domain_name,
140                                     r->in.domain_name,
141                                     r->in.dc_name,
142                                     r->in.admin_account,
143                                     r->in.admin_password,
144                                     &r->in.ads);
145         if (!ADS_ERR_OK(status)) {
146                 libnet_join_set_error_string(mem_ctx, r,
147                         "failed to connect to AD: %s",
148                         ads_errstr(status));
149         }
150
151         return status;
152 }
153
154 /****************************************************************
155 ****************************************************************/
156
157 static ADS_STATUS libnet_unjoin_connect_ads(TALLOC_CTX *mem_ctx,
158                                             struct libnet_UnjoinCtx *r)
159 {
160         ADS_STATUS status;
161
162         status = libnet_connect_ads(r->in.domain_name,
163                                     r->in.domain_name,
164                                     r->in.dc_name,
165                                     r->in.admin_account,
166                                     r->in.admin_password,
167                                     &r->in.ads);
168         if (!ADS_ERR_OK(status)) {
169                 libnet_unjoin_set_error_string(mem_ctx, r,
170                         "failed to connect to AD: %s",
171                         ads_errstr(status));
172         }
173
174         return status;
175 }
176
177 /****************************************************************
178 ****************************************************************/
179
180 static ADS_STATUS libnet_join_precreate_machine_acct(TALLOC_CTX *mem_ctx,
181                                                      struct libnet_JoinCtx *r)
182 {
183         ADS_STATUS status;
184         LDAPMessage *res = NULL;
185         const char *attrs[] = { "dn", NULL };
186
187         status = ads_search_dn(r->in.ads, &res, r->in.account_ou, attrs);
188         if (!ADS_ERR_OK(status)) {
189                 return status;
190         }
191
192         if (ads_count_replies(r->in.ads, res) != 1) {
193                 ads_msgfree(r->in.ads, res);
194                 return ADS_ERROR_LDAP(LDAP_NO_SUCH_OBJECT);
195         }
196
197         status = ads_create_machine_acct(r->in.ads,
198                                          r->in.machine_name,
199                                          r->in.account_ou);
200         ads_msgfree(r->in.ads, res);
201
202         if ((status.error_type == ENUM_ADS_ERROR_LDAP) &&
203             (status.err.rc == LDAP_ALREADY_EXISTS)) {
204                 status = ADS_SUCCESS;
205         }
206
207         return status;
208 }
209
210 /****************************************************************
211 ****************************************************************/
212
213 static ADS_STATUS libnet_unjoin_remove_machine_acct(TALLOC_CTX *mem_ctx,
214                                                     struct libnet_UnjoinCtx *r)
215 {
216         ADS_STATUS status;
217
218         if (!r->in.ads) {
219                 status = libnet_unjoin_connect_ads(mem_ctx, r);
220                 if (!ADS_ERR_OK(status)) {
221                         return status;
222                 }
223         }
224
225         status = ads_leave_realm(r->in.ads, r->in.machine_name);
226         if (!ADS_ERR_OK(status)) {
227                 libnet_unjoin_set_error_string(mem_ctx, r,
228                         "failed to leave realm: %s",
229                         ads_errstr(status));
230                 return status;
231         }
232
233         return ADS_SUCCESS;
234 }
235
236 /****************************************************************
237 ****************************************************************/
238
239 static ADS_STATUS libnet_join_find_machine_acct(TALLOC_CTX *mem_ctx,
240                                                 struct libnet_JoinCtx *r)
241 {
242         ADS_STATUS status;
243         LDAPMessage *res = NULL;
244         char *dn = NULL;
245
246         if (!r->in.machine_name) {
247                 return ADS_ERROR(LDAP_NO_MEMORY);
248         }
249
250         status = ads_find_machine_acct(r->in.ads,
251                                        &res,
252                                        r->in.machine_name);
253         if (!ADS_ERR_OK(status)) {
254                 return status;
255         }
256
257         if (ads_count_replies(r->in.ads, res) != 1) {
258                 status = ADS_ERROR_LDAP(LDAP_NO_MEMORY);
259                 goto done;
260         }
261
262         dn = ads_get_dn(r->in.ads, res);
263         if (!dn) {
264                 status = ADS_ERROR_LDAP(LDAP_NO_MEMORY);
265                 goto done;
266         }
267
268         r->out.dn = talloc_strdup(mem_ctx, dn);
269         if (!r->out.dn) {
270                 status = ADS_ERROR_LDAP(LDAP_NO_MEMORY);
271                 goto done;
272         }
273
274  done:
275         ads_msgfree(r->in.ads, res);
276         ads_memfree(r->in.ads, dn);
277
278         return status;
279 }
280
281 /****************************************************************
282 ****************************************************************/
283
284 static ADS_STATUS libnet_join_set_machine_spn(TALLOC_CTX *mem_ctx,
285                                               struct libnet_JoinCtx *r)
286 {
287         ADS_STATUS status;
288         ADS_MODLIST mods;
289         fstring my_fqdn;
290         const char *spn_array[3] = {NULL, NULL, NULL};
291         char *spn = NULL;
292
293         if (!r->in.ads) {
294                 status = libnet_join_connect_ads(mem_ctx, r);
295                 if (!ADS_ERR_OK(status)) {
296                         return status;
297                 }
298         }
299
300         status = libnet_join_find_machine_acct(mem_ctx, r);
301         if (!ADS_ERR_OK(status)) {
302                 return status;
303         }
304
305         spn = talloc_asprintf(mem_ctx, "HOST/%s", r->in.machine_name);
306         if (!spn) {
307                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
308         }
309         strupper_m(spn);
310         spn_array[0] = spn;
311
312         if (name_to_fqdn(my_fqdn, r->in.machine_name) &&
313             !strequal(my_fqdn, r->in.machine_name)) {
314
315                 strlower_m(my_fqdn);
316                 spn = talloc_asprintf(mem_ctx, "HOST/%s", my_fqdn);
317                 if (!spn) {
318                         return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
319                 }
320                 spn_array[1] = spn;
321         }
322
323         mods = ads_init_mods(mem_ctx);
324         if (!mods) {
325                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
326         }
327
328         status = ads_mod_str(mem_ctx, &mods, "dNSHostName", my_fqdn);
329         if (!ADS_ERR_OK(status)) {
330                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
331         }
332
333         status = ads_mod_strlist(mem_ctx, &mods, "servicePrincipalName",
334                                  spn_array);
335         if (!ADS_ERR_OK(status)) {
336                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
337         }
338
339         return ads_gen_mod(r->in.ads, r->out.dn, mods);
340 }
341
342 /****************************************************************
343 ****************************************************************/
344
345 static ADS_STATUS libnet_join_set_machine_upn(TALLOC_CTX *mem_ctx,
346                                               struct libnet_JoinCtx *r)
347 {
348         ADS_STATUS status;
349         ADS_MODLIST mods;
350
351         if (!r->in.create_upn) {
352                 return ADS_SUCCESS;
353         }
354
355         if (!r->in.ads) {
356                 status = libnet_join_connect_ads(mem_ctx, r);
357                 if (!ADS_ERR_OK(status)) {
358                         return status;
359                 }
360         }
361
362         status = libnet_join_find_machine_acct(mem_ctx, r);
363         if (!ADS_ERR_OK(status)) {
364                 return status;
365         }
366
367         if (!r->in.upn) {
368                 r->in.upn = talloc_asprintf(mem_ctx,
369                                             "host/%s@%s",
370                                             r->in.machine_name,
371                                             r->out.dns_domain_name);
372                 if (!r->in.upn) {
373                         return ADS_ERROR(LDAP_NO_MEMORY);
374                 }
375         }
376
377         mods = ads_init_mods(mem_ctx);
378         if (!mods) {
379                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
380         }
381
382         status = ads_mod_str(mem_ctx, &mods, "userPrincipalName", r->in.upn);
383         if (!ADS_ERR_OK(status)) {
384                 return ADS_ERROR_LDAP(LDAP_NO_MEMORY);
385         }
386
387         return ads_gen_mod(r->in.ads, r->out.dn, mods);
388 }
389
390
391 /****************************************************************
392 ****************************************************************/
393
394 static ADS_STATUS libnet_join_set_os_attributes(TALLOC_CTX *mem_ctx,
395                                                 struct libnet_JoinCtx *r)
396 {
397         ADS_STATUS status;
398         ADS_MODLIST mods;
399         char *os_sp = NULL;
400
401         if (!r->in.os_name || !r->in.os_version ) {
402                 return ADS_SUCCESS;
403         }
404
405         if (!r->in.ads) {
406                 status = libnet_join_connect_ads(mem_ctx, r);
407                 if (!ADS_ERR_OK(status)) {
408                         return status;
409                 }
410         }
411
412         status = libnet_join_find_machine_acct(mem_ctx, r);
413         if (!ADS_ERR_OK(status)) {
414                 return status;
415         }
416
417         mods = ads_init_mods(mem_ctx);
418         if (!mods) {
419                 return ADS_ERROR(LDAP_NO_MEMORY);
420         }
421
422         os_sp = talloc_asprintf(mem_ctx, "Samba %s", SAMBA_VERSION_STRING);
423         if (!os_sp) {
424                 return ADS_ERROR(LDAP_NO_MEMORY);
425         }
426
427         status = ads_mod_str(mem_ctx, &mods, "operatingSystem",
428                              r->in.os_name);
429         if (!ADS_ERR_OK(status)) {
430                 return status;
431         }
432
433         status = ads_mod_str(mem_ctx, &mods, "operatingSystemVersion",
434                              r->in.os_version);
435         if (!ADS_ERR_OK(status)) {
436                 return status;
437         }
438
439         status = ads_mod_str(mem_ctx, &mods, "operatingSystemServicePack",
440                              os_sp);
441         if (!ADS_ERR_OK(status)) {
442                 return status;
443         }
444
445         return ads_gen_mod(r->in.ads, r->out.dn, mods);
446 }
447
448 /****************************************************************
449 ****************************************************************/
450
451 static bool libnet_join_create_keytab(TALLOC_CTX *mem_ctx,
452                                       struct libnet_JoinCtx *r)
453 {
454         if (!lp_use_kerberos_keytab()) {
455                 return true;
456         }
457
458         if (!ads_keytab_create_default(r->in.ads)) {
459                 return false;
460         }
461
462         return true;
463 }
464
465 /****************************************************************
466 ****************************************************************/
467
468 static bool libnet_join_derive_salting_principal(TALLOC_CTX *mem_ctx,
469                                                  struct libnet_JoinCtx *r)
470 {
471         uint32_t domain_func;
472         ADS_STATUS status;
473         const char *salt = NULL;
474         char *std_salt = NULL;
475
476         status = ads_domain_func_level(r->in.ads, &domain_func);
477         if (!ADS_ERR_OK(status)) {
478                 libnet_join_set_error_string(mem_ctx, r,
479                         "failed to determine domain functional level: %s",
480                         ads_errstr(status));
481                 return false;
482         }
483
484         std_salt = kerberos_standard_des_salt();
485         if (!std_salt) {
486                 libnet_join_set_error_string(mem_ctx, r,
487                         "failed to obtain standard DES salt");
488                 return false;
489         }
490
491         salt = talloc_strdup(mem_ctx, std_salt);
492         if (!salt) {
493                 return false;
494         }
495
496         SAFE_FREE(std_salt);
497
498         if (domain_func == DS_DOMAIN_FUNCTION_2000) {
499                 char *upn;
500
501                 upn = ads_get_upn(r->in.ads, mem_ctx,
502                                   r->in.machine_name);
503                 if (upn) {
504                         salt = talloc_strdup(mem_ctx, upn);
505                         if (!salt) {
506                                 return false;
507                         }
508                 }
509         }
510
511         return kerberos_secrets_store_des_salt(salt);
512 }
513
514 /****************************************************************
515 ****************************************************************/
516
517 static ADS_STATUS libnet_join_post_processing_ads(TALLOC_CTX *mem_ctx,
518                                                   struct libnet_JoinCtx *r)
519 {
520         ADS_STATUS status;
521
522         status = libnet_join_set_machine_spn(mem_ctx, r);
523         if (!ADS_ERR_OK(status)) {
524                 libnet_join_set_error_string(mem_ctx, r,
525                         "failed to set machine spn: %s",
526                         ads_errstr(status));
527                 return status;
528         }
529
530         status = libnet_join_set_os_attributes(mem_ctx, r);
531         if (!ADS_ERR_OK(status)) {
532                 libnet_join_set_error_string(mem_ctx, r,
533                         "failed to set machine os attributes: %s",
534                         ads_errstr(status));
535                 return status;
536         }
537
538         status = libnet_join_set_machine_upn(mem_ctx, r);
539         if (!ADS_ERR_OK(status)) {
540                 libnet_join_set_error_string(mem_ctx, r,
541                         "failed to set machine upn: %s",
542                         ads_errstr(status));
543                 return status;
544         }
545
546         if (!libnet_join_derive_salting_principal(mem_ctx, r)) {
547                 return ADS_ERROR_NT(NT_STATUS_UNSUCCESSFUL);
548         }
549
550         if (!libnet_join_create_keytab(mem_ctx, r)) {
551                 libnet_join_set_error_string(mem_ctx, r,
552                         "failed to create kerberos keytab");
553                 return ADS_ERROR_NT(NT_STATUS_UNSUCCESSFUL);
554         }
555
556         return ADS_SUCCESS;
557 }
558 #endif /* WITH_ADS */
559
560 /****************************************************************
561 ****************************************************************/
562
563 static bool libnet_join_joindomain_store_secrets(TALLOC_CTX *mem_ctx,
564                                                  struct libnet_JoinCtx *r)
565 {
566         if (!secrets_store_domain_sid(r->out.netbios_domain_name,
567                                       r->out.domain_sid))
568         {
569                 return false;
570         }
571
572         if (!secrets_store_machine_password(r->in.machine_password,
573                                             r->out.netbios_domain_name,
574                                             SEC_CHAN_WKSTA))
575         {
576                 return false;
577         }
578
579         return true;
580 }
581
582 /****************************************************************
583 ****************************************************************/
584
585 static NTSTATUS libnet_join_joindomain_rpc(TALLOC_CTX *mem_ctx,
586                                            struct libnet_JoinCtx *r)
587 {
588         struct cli_state *cli = NULL;
589         struct rpc_pipe_client *pipe_hnd = NULL;
590         POLICY_HND sam_pol, domain_pol, user_pol, lsa_pol;
591         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
592         char *acct_name;
593         const char *const_acct_name;
594         uint32 user_rid;
595         uint32 num_rids, *name_types, *user_rids;
596         uint32 flags = 0x3e8;
597         uint32 acb_info = ACB_WSTRUST;
598         uint32 fields_present;
599         uchar pwbuf[532];
600         SAM_USERINFO_CTR ctr;
601         SAM_USER_INFO_25 p25;
602         const int infolevel = 25;
603         struct MD5Context md5ctx;
604         uchar md5buffer[16];
605         DATA_BLOB digested_session_key;
606         uchar md4_trust_password[16];
607
608         if (!r->in.machine_password) {
609                 r->in.machine_password = talloc_strdup(mem_ctx, generate_random_str(DEFAULT_TRUST_ACCOUNT_PASSWORD_LENGTH));
610                 NT_STATUS_HAVE_NO_MEMORY(r->in.machine_password);
611         }
612
613         status = cli_full_connection(&cli, NULL,
614                                      r->in.dc_name,
615                                      NULL, 0,
616                                      "IPC$", "IPC",
617                                      r->in.admin_account,
618                                      NULL,
619                                      r->in.admin_password,
620                                      0,
621                                      Undefined, NULL);
622
623         if (!NT_STATUS_IS_OK(status)) {
624                 goto done;
625         }
626
627         pipe_hnd = cli_rpc_pipe_open_noauth(cli, PI_LSARPC, &status);
628         if (!pipe_hnd) {
629                 goto done;
630         }
631
632         status = rpccli_lsa_open_policy(pipe_hnd, mem_ctx, True,
633                                         SEC_RIGHTS_MAXIMUM_ALLOWED, &lsa_pol);
634         if (!NT_STATUS_IS_OK(status)) {
635                 goto done;
636         }
637
638         status = rpccli_lsa_query_info_policy2(pipe_hnd, mem_ctx, &lsa_pol,
639                                                12,
640                                                &r->out.netbios_domain_name,
641                                                &r->out.dns_domain_name,
642                                                NULL,
643                                                NULL,
644                                                &r->out.domain_sid);
645
646         if (NT_STATUS_IS_OK(status)) {
647                 r->out.domain_is_ad = true;
648         }
649
650         if (!NT_STATUS_IS_OK(status)) {
651                 status = rpccli_lsa_query_info_policy(pipe_hnd, mem_ctx, &lsa_pol,
652                                                       5,
653                                                       &r->out.netbios_domain_name,
654                                                       &r->out.domain_sid);
655                 if (!NT_STATUS_IS_OK(status)) {
656                         goto done;
657                 }
658         }
659
660         rpccli_lsa_Close(pipe_hnd, mem_ctx, &lsa_pol);
661         cli_rpc_pipe_close(pipe_hnd);
662
663         pipe_hnd = cli_rpc_pipe_open_noauth(cli, PI_SAMR, &status);
664         if (!pipe_hnd) {
665                 goto done;
666         }
667
668         status = rpccli_samr_connect(pipe_hnd, mem_ctx,
669                                      SEC_RIGHTS_MAXIMUM_ALLOWED, &sam_pol);
670         if (!NT_STATUS_IS_OK(status)) {
671                 goto done;
672         }
673
674         status = rpccli_samr_open_domain(pipe_hnd, mem_ctx, &sam_pol,
675                                          SEC_RIGHTS_MAXIMUM_ALLOWED,
676                                          r->out.domain_sid,
677                                          &domain_pol);
678         if (!NT_STATUS_IS_OK(status)) {
679                 goto done;
680         }
681
682         acct_name = talloc_asprintf(mem_ctx, "%s$", r->in.machine_name);
683         strlower_m(acct_name);
684         const_acct_name = acct_name;
685
686         if (r->in.join_flags & WKSSVC_JOIN_FLAGS_ACCOUNT_CREATE) {
687                 status = rpccli_samr_create_dom_user(pipe_hnd, mem_ctx,
688                                                      &domain_pol,
689                                                      acct_name, ACB_WSTRUST,
690                                                      0xe005000b, &user_pol,
691                                                      &user_rid);
692                 if (NT_STATUS_EQUAL(status, NT_STATUS_USER_EXISTS)) {
693                         if (!(r->in.join_flags &
694                               WKSSVC_JOIN_FLAGS_DOMAIN_JOIN_IF_JOINED)) {
695                                 goto done;
696                         }
697                 }
698
699                 if (NT_STATUS_IS_OK(status)) {
700                         rpccli_samr_close(pipe_hnd, mem_ctx, &user_pol);
701                 }
702         }
703
704         status = rpccli_samr_lookup_names(pipe_hnd, mem_ctx,
705                                           &domain_pol, flags, 1,
706                                           &const_acct_name,
707                                           &num_rids, &user_rids, &name_types);
708         if (!NT_STATUS_IS_OK(status)) {
709                 goto done;
710         }
711
712         if (name_types[0] != SID_NAME_USER) {
713                 status = NT_STATUS_INVALID_WORKSTATION;
714                 goto done;
715         }
716
717         user_rid = user_rids[0];
718
719         status = rpccli_samr_open_user(pipe_hnd, mem_ctx, &domain_pol,
720                                        SEC_RIGHTS_MAXIMUM_ALLOWED, user_rid,
721                                        &user_pol);
722         if (!NT_STATUS_IS_OK(status)) {
723                 goto done;
724         }
725
726         E_md4hash(r->in.machine_password, md4_trust_password);
727         encode_pw_buffer(pwbuf, r->in.machine_password, STR_UNICODE);
728
729         generate_random_buffer((uint8*)md5buffer, sizeof(md5buffer));
730         digested_session_key = data_blob_talloc(mem_ctx, 0, 16);
731
732         MD5Init(&md5ctx);
733         MD5Update(&md5ctx, md5buffer, sizeof(md5buffer));
734         MD5Update(&md5ctx, cli->user_session_key.data,
735                   cli->user_session_key.length);
736         MD5Final(digested_session_key.data, &md5ctx);
737
738         SamOEMhashBlob(pwbuf, sizeof(pwbuf), &digested_session_key);
739         memcpy(&pwbuf[516], md5buffer, sizeof(md5buffer));
740
741         acb_info |= ACB_PWNOEXP;
742         if (r->out.domain_is_ad) {
743 #if !defined(ENCTYPE_ARCFOUR_HMAC)
744                 acb_info |= ACB_USE_DES_KEY_ONLY;
745 #endif
746                 ;;
747         }
748
749         ZERO_STRUCT(ctr);
750         ZERO_STRUCT(p25);
751
752         fields_present = ACCT_NT_PWD_SET | ACCT_LM_PWD_SET | ACCT_FLAGS;
753         init_sam_user_info25P(&p25, fields_present, acb_info, (char *)pwbuf);
754
755         ctr.switch_value = infolevel;
756         ctr.info.id25    = &p25;
757
758         status = rpccli_samr_set_userinfo2(pipe_hnd, mem_ctx, &user_pol,
759                                            infolevel, &cli->user_session_key,
760                                            &ctr);
761         if (!NT_STATUS_IS_OK(status)) {
762                 goto done;
763         }
764
765         rpccli_samr_close(pipe_hnd, mem_ctx, &user_pol);
766         cli_rpc_pipe_close(pipe_hnd);
767
768         status = NT_STATUS_OK;
769  done:
770         if (cli) {
771                 cli_shutdown(cli);
772         }
773
774         return status;
775 }
776
777 /****************************************************************
778 ****************************************************************/
779
780 static bool libnet_join_unjoindomain_remove_secrets(TALLOC_CTX *mem_ctx,
781                                                     struct libnet_UnjoinCtx *r)
782 {
783         if (!secrets_delete_machine_password_ex(lp_workgroup())) {
784                 return false;
785         }
786
787         if (!secrets_delete_domain_sid(lp_workgroup())) {
788                 return false;
789         }
790
791         return true;
792 }
793
794 /****************************************************************
795 ****************************************************************/
796
797 static NTSTATUS libnet_join_unjoindomain_rpc(TALLOC_CTX *mem_ctx,
798                                              struct libnet_UnjoinCtx *r)
799 {
800         struct cli_state *cli = NULL;
801         struct rpc_pipe_client *pipe_hnd = NULL;
802         POLICY_HND sam_pol, domain_pol, user_pol;
803         NTSTATUS status = NT_STATUS_UNSUCCESSFUL;
804         char *acct_name;
805         uint32 flags = 0x3e8;
806         const char *const_acct_name;
807         uint32 user_rid;
808         uint32 num_rids, *name_types, *user_rids;
809         SAM_USERINFO_CTR ctr, *qctr = NULL;
810         SAM_USER_INFO_16 p16;
811
812         status = cli_full_connection(&cli, NULL,
813                                      r->in.dc_name,
814                                      NULL, 0,
815                                      "IPC$", "IPC",
816                                      r->in.admin_account,
817                                      NULL,
818                                      r->in.admin_password,
819                                      0, Undefined, NULL);
820
821         if (!NT_STATUS_IS_OK(status)) {
822                 goto done;
823         }
824
825         pipe_hnd = cli_rpc_pipe_open_noauth(cli, PI_SAMR, &status);
826         if (!pipe_hnd) {
827                 goto done;
828         }
829
830         status = rpccli_samr_connect(pipe_hnd, mem_ctx,
831                                      SEC_RIGHTS_MAXIMUM_ALLOWED, &sam_pol);
832         if (!NT_STATUS_IS_OK(status)) {
833                 goto done;
834         }
835
836         status = rpccli_samr_open_domain(pipe_hnd, mem_ctx, &sam_pol,
837                                          SEC_RIGHTS_MAXIMUM_ALLOWED,
838                                          r->in.domain_sid,
839                                          &domain_pol);
840         if (!NT_STATUS_IS_OK(status)) {
841                 goto done;
842         }
843
844         acct_name = talloc_asprintf(mem_ctx, "%s$", r->in.machine_name);
845         strlower_m(acct_name);
846         const_acct_name = acct_name;
847
848         status = rpccli_samr_lookup_names(pipe_hnd, mem_ctx,
849                                           &domain_pol, flags, 1,
850                                           &const_acct_name,
851                                           &num_rids, &user_rids, &name_types);
852         if (!NT_STATUS_IS_OK(status)) {
853                 goto done;
854         }
855
856         if (name_types[0] != SID_NAME_USER) {
857                 status = NT_STATUS_INVALID_WORKSTATION;
858                 goto done;
859         }
860
861         user_rid = user_rids[0];
862
863         status = rpccli_samr_open_user(pipe_hnd, mem_ctx, &domain_pol,
864                                        SEC_RIGHTS_MAXIMUM_ALLOWED,
865                                        user_rid, &user_pol);
866         if (!NT_STATUS_IS_OK(status)) {
867                 goto done;
868         }
869
870         status = rpccli_samr_query_userinfo(pipe_hnd, mem_ctx,
871                                             &user_pol, 16, &qctr);
872         if (!NT_STATUS_IS_OK(status)) {
873                 rpccli_samr_close(pipe_hnd, mem_ctx, &user_pol);
874                 goto done;
875         }
876
877         ZERO_STRUCT(ctr);
878         ctr.switch_value = 16;
879         ctr.info.id16 = &p16;
880
881         p16.acb_info = qctr->info.id16->acb_info | ACB_DISABLED;
882
883         status = rpccli_samr_set_userinfo2(pipe_hnd, mem_ctx, &user_pol, 16,
884                                            &cli->user_session_key, &ctr);
885
886         rpccli_samr_close(pipe_hnd, mem_ctx, &user_pol);
887
888 done:
889         if (pipe_hnd) {
890                 rpccli_samr_close(pipe_hnd, mem_ctx, &domain_pol);
891                 rpccli_samr_close(pipe_hnd, mem_ctx, &sam_pol);
892                 cli_rpc_pipe_close(pipe_hnd);
893         }
894
895         if (cli) {
896                 cli_shutdown(cli);
897         }
898
899         return status;
900 }
901
902 /****************************************************************
903 ****************************************************************/
904
905 static WERROR do_join_modify_vals_config(struct libnet_JoinCtx *r)
906 {
907         WERROR werr;
908         struct libnet_conf_ctx *ctx;
909
910         werr = libnet_conf_open(r, &ctx);
911         if (!W_ERROR_IS_OK(werr)) {
912                 goto done;
913         }
914
915         if (!(r->in.join_flags & WKSSVC_JOIN_FLAGS_JOIN_TYPE)) {
916
917                 werr = libnet_conf_set_global_parameter(ctx, "security", "user");
918                 if (!W_ERROR_IS_OK(werr)) {
919                         goto done;
920                 }
921
922                 werr = libnet_conf_set_global_parameter(ctx, "workgroup",
923                                                         r->in.domain_name);
924                 goto done;
925         }
926
927         werr = libnet_conf_set_global_parameter(ctx, "security", "domain");
928         if (!W_ERROR_IS_OK(werr)) {
929                 goto done;
930         }
931
932         werr = libnet_conf_set_global_parameter(ctx, "workgroup",
933                                                 r->out.netbios_domain_name);
934         if (!W_ERROR_IS_OK(werr)) {
935                 goto done;
936         }
937
938         if (r->out.domain_is_ad) {
939                 werr = libnet_conf_set_global_parameter(ctx, "security", "ads");
940                 if (!W_ERROR_IS_OK(werr)) {
941                         goto done;
942                 }
943
944                 werr = libnet_conf_set_global_parameter(ctx, "realm",
945                                                         r->out.dns_domain_name);
946         }
947
948 done:
949         libnet_conf_close(ctx);
950         return werr;
951 }
952
953 /****************************************************************
954 ****************************************************************/
955
956 static WERROR do_unjoin_modify_vals_config(struct libnet_UnjoinCtx *r)
957 {
958         WERROR werr = WERR_OK;
959         struct libnet_conf_ctx *ctx;
960
961         werr = libnet_conf_open(r, &ctx);
962         if (!W_ERROR_IS_OK(werr)) {
963                 goto done;
964         }
965
966         if (r->in.unjoin_flags & WKSSVC_JOIN_FLAGS_JOIN_TYPE) {
967
968                 werr = libnet_conf_set_global_parameter(ctx, "security", "user");
969                 if (!W_ERROR_IS_OK(werr)) {
970                         goto done;
971                 }
972         }
973
974         libnet_conf_delete_global_parameter(ctx, "realm");
975
976 done:
977         libnet_conf_close(ctx);
978         return werr;
979 }
980
981 /****************************************************************
982 ****************************************************************/
983
984 static WERROR do_JoinConfig(struct libnet_JoinCtx *r)
985 {
986         WERROR werr;
987
988         if (!W_ERROR_IS_OK(r->out.result)) {
989                 return r->out.result;
990         }
991
992         if (!r->in.modify_config) {
993                 return WERR_OK;
994         }
995
996         werr = do_join_modify_vals_config(r);
997         if (!W_ERROR_IS_OK(werr)) {
998                 return werr;
999         }
1000
1001         r->out.modified_config = true;
1002         r->out.result = werr;
1003
1004         return werr;
1005 }
1006
1007 /****************************************************************
1008 ****************************************************************/
1009
1010 static WERROR do_UnjoinConfig(struct libnet_UnjoinCtx *r)
1011 {
1012         WERROR werr;
1013
1014         if (!W_ERROR_IS_OK(r->out.result)) {
1015                 return r->out.result;
1016         }
1017
1018         if (!r->in.modify_config) {
1019                 return WERR_OK;
1020         }
1021
1022         werr = do_unjoin_modify_vals_config(r);
1023         if (!W_ERROR_IS_OK(werr)) {
1024                 return werr;
1025         }
1026
1027         r->out.modified_config = true;
1028         r->out.result = werr;
1029
1030         return werr;
1031 }
1032
1033 /****************************************************************
1034 ****************************************************************/
1035
1036 static WERROR libnet_join_pre_processing(TALLOC_CTX *mem_ctx,
1037                                          struct libnet_JoinCtx *r)
1038 {
1039
1040         if (!r->in.domain_name) {
1041                 return WERR_INVALID_PARAM;
1042         }
1043
1044         if (r->in.modify_config && !lp_config_backend_is_registry()) {
1045                 return WERR_NOT_SUPPORTED;
1046         }
1047
1048         if (IS_DC) {
1049                 return WERR_SETUP_DOMAIN_CONTROLLER;
1050         }
1051
1052         if (!secrets_init()) {
1053                 libnet_join_set_error_string(mem_ctx, r,
1054                         "Unable to open secrets database");
1055                 return WERR_CAN_NOT_COMPLETE;
1056         }
1057
1058         return WERR_OK;
1059 }
1060
1061 /****************************************************************
1062 ****************************************************************/
1063
1064 static WERROR libnet_join_post_processing(TALLOC_CTX *mem_ctx,
1065                                           struct libnet_JoinCtx *r)
1066 {
1067         WERROR werr;
1068
1069         if (!W_ERROR_IS_OK(r->out.result)) {
1070                 return r->out.result;
1071         }
1072
1073         werr = do_JoinConfig(r);
1074         if (!W_ERROR_IS_OK(werr)) {
1075                 return werr;
1076         }
1077
1078         if (r->in.join_flags & WKSSVC_JOIN_FLAGS_JOIN_TYPE) {
1079                 saf_store(r->in.domain_name, r->in.dc_name);
1080         }
1081
1082         return WERR_OK;
1083 }
1084
1085 /****************************************************************
1086 ****************************************************************/
1087
1088 static int libnet_destroy_JoinCtx(struct libnet_JoinCtx *r)
1089 {
1090         if (r->in.ads) {
1091                 ads_destroy(&r->in.ads);
1092         }
1093
1094         return 0;
1095 }
1096
1097 /****************************************************************
1098 ****************************************************************/
1099
1100 static int libnet_destroy_UnjoinCtx(struct libnet_UnjoinCtx *r)
1101 {
1102         if (r->in.ads) {
1103                 ads_destroy(&r->in.ads);
1104         }
1105
1106         return 0;
1107 }
1108
1109 /****************************************************************
1110 ****************************************************************/
1111
1112 WERROR libnet_init_JoinCtx(TALLOC_CTX *mem_ctx,
1113                            struct libnet_JoinCtx **r)
1114 {
1115         struct libnet_JoinCtx *ctx;
1116
1117         ctx = talloc_zero(mem_ctx, struct libnet_JoinCtx);
1118         if (!ctx) {
1119                 return WERR_NOMEM;
1120         }
1121
1122         talloc_set_destructor(ctx, libnet_destroy_JoinCtx);
1123
1124         ctx->in.machine_name = talloc_strdup(mem_ctx, global_myname());
1125         W_ERROR_HAVE_NO_MEMORY(ctx->in.machine_name);
1126
1127         *r = ctx;
1128
1129         return WERR_OK;
1130 }
1131
1132 /****************************************************************
1133 ****************************************************************/
1134
1135 WERROR libnet_init_UnjoinCtx(TALLOC_CTX *mem_ctx,
1136                              struct libnet_UnjoinCtx **r)
1137 {
1138         struct libnet_UnjoinCtx *ctx;
1139
1140         ctx = talloc_zero(mem_ctx, struct libnet_UnjoinCtx);
1141         if (!ctx) {
1142                 return WERR_NOMEM;
1143         }
1144
1145         talloc_set_destructor(ctx, libnet_destroy_UnjoinCtx);
1146
1147         ctx->in.machine_name = talloc_strdup(mem_ctx, global_myname());
1148         W_ERROR_HAVE_NO_MEMORY(ctx->in.machine_name);
1149
1150         *r = ctx;
1151
1152         return WERR_OK;
1153 }
1154
1155 /****************************************************************
1156 ****************************************************************/
1157
1158 static WERROR libnet_DomainJoin(TALLOC_CTX *mem_ctx,
1159                                 struct libnet_JoinCtx *r)
1160 {
1161         NTSTATUS status;
1162 #ifdef WITH_ADS
1163         ADS_STATUS ads_status;
1164 #endif /* WITH_ADS */
1165
1166         if (!r->in.dc_name) {
1167                 struct DS_DOMAIN_CONTROLLER_INFO *info;
1168                 status = dsgetdcname(mem_ctx,
1169                                      NULL,
1170                                      r->in.domain_name,
1171                                      NULL,
1172                                      NULL,
1173                                      DS_DIRECTORY_SERVICE_REQUIRED |
1174                                      DS_WRITABLE_REQUIRED |
1175                                      DS_RETURN_DNS_NAME,
1176                                      &info);
1177                 if (!NT_STATUS_IS_OK(status)) {
1178                         libnet_join_set_error_string(mem_ctx, r,
1179                                 "failed to find DC for domain %s",
1180                                 r->in.domain_name,
1181                                 get_friendly_nt_error_msg(status));
1182                         return WERR_DOMAIN_CONTROLLER_NOT_FOUND;
1183                 }
1184
1185                 r->in.dc_name = talloc_strdup(mem_ctx,
1186                                               info->domain_controller_name);
1187                 W_ERROR_HAVE_NO_MEMORY(r->in.dc_name);
1188         }
1189
1190 #ifdef WITH_ADS
1191         if (r->in.account_ou) {
1192
1193                 ads_status = libnet_join_connect_ads(mem_ctx, r);
1194                 if (!ADS_ERR_OK(ads_status)) {
1195                         return WERR_DEFAULT_JOIN_REQUIRED;
1196                 }
1197
1198                 ads_status = libnet_join_precreate_machine_acct(mem_ctx, r);
1199                 if (!ADS_ERR_OK(ads_status)) {
1200                         libnet_join_set_error_string(mem_ctx, r,
1201                                 "failed to precreate account in ou %s: %s",
1202                                 r->in.account_ou,
1203                                 ads_errstr(ads_status));
1204                         return WERR_DEFAULT_JOIN_REQUIRED;
1205                 }
1206
1207                 r->in.join_flags &= ~WKSSVC_JOIN_FLAGS_ACCOUNT_CREATE;
1208         }
1209 #endif /* WITH_ADS */
1210
1211         status = libnet_join_joindomain_rpc(mem_ctx, r);
1212         if (!NT_STATUS_IS_OK(status)) {
1213                 libnet_join_set_error_string(mem_ctx, r,
1214                         "failed to join domain over rpc: %s",
1215                         get_friendly_nt_error_msg(status));
1216                 if (NT_STATUS_EQUAL(status, NT_STATUS_USER_EXISTS)) {
1217                         return WERR_SETUP_ALREADY_JOINED;
1218                 }
1219                 return ntstatus_to_werror(status);
1220         }
1221
1222         if (!libnet_join_joindomain_store_secrets(mem_ctx, r)) {
1223                 return WERR_SETUP_NOT_JOINED;
1224         }
1225
1226 #ifdef WITH_ADS
1227         if (r->out.domain_is_ad) {
1228                 ads_status  = libnet_join_post_processing_ads(mem_ctx, r);
1229                 if (!ADS_ERR_OK(ads_status)) {
1230                         return WERR_GENERAL_FAILURE;
1231                 }
1232         }
1233 #endif /* WITH_ADS */
1234
1235         return WERR_OK;
1236 }
1237
1238 /****************************************************************
1239 ****************************************************************/
1240
1241 WERROR libnet_Join(TALLOC_CTX *mem_ctx,
1242                    struct libnet_JoinCtx *r)
1243 {
1244         WERROR werr;
1245
1246         if (r->in.debug) {
1247                 LIBNET_JOIN_IN_DUMP_CTX(mem_ctx, r);
1248         }
1249
1250         werr = libnet_join_pre_processing(mem_ctx, r);
1251         if (!W_ERROR_IS_OK(werr)) {
1252                 goto done;
1253         }
1254
1255         if (r->in.join_flags & WKSSVC_JOIN_FLAGS_JOIN_TYPE) {
1256                 werr = libnet_DomainJoin(mem_ctx, r);
1257                 if (!W_ERROR_IS_OK(werr)) {
1258                         goto done;
1259                 }
1260         }
1261
1262         werr = libnet_join_post_processing(mem_ctx, r);
1263         if (!W_ERROR_IS_OK(werr)) {
1264                 goto done;
1265         }
1266  done:
1267         r->out.result = werr;
1268
1269         if (r->in.debug) {
1270                 LIBNET_JOIN_OUT_DUMP_CTX(mem_ctx, r);
1271         }
1272         return werr;
1273 }
1274
1275 /****************************************************************
1276 ****************************************************************/
1277
1278 static WERROR libnet_DomainUnjoin(TALLOC_CTX *mem_ctx,
1279                                   struct libnet_UnjoinCtx *r)
1280 {
1281         NTSTATUS status;
1282
1283         if (!r->in.domain_sid) {
1284                 struct dom_sid sid;
1285                 if (!secrets_fetch_domain_sid(lp_workgroup(), &sid)) {
1286                         libnet_unjoin_set_error_string(mem_ctx, r,
1287                                 "Unable to fetch domain sid: are we joined?");
1288                         return WERR_SETUP_NOT_JOINED;
1289                 }
1290                 r->in.domain_sid = sid_dup_talloc(mem_ctx, &sid);
1291                 W_ERROR_HAVE_NO_MEMORY(r->in.domain_sid);
1292         }
1293
1294         if (!r->in.dc_name) {
1295                 struct DS_DOMAIN_CONTROLLER_INFO *info;
1296                 status = dsgetdcname(mem_ctx,
1297                                      NULL,
1298                                      r->in.domain_name,
1299                                      NULL,
1300                                      NULL,
1301                                      DS_DIRECTORY_SERVICE_REQUIRED |
1302                                      DS_WRITABLE_REQUIRED |
1303                                      DS_RETURN_DNS_NAME,
1304                                      &info);
1305                 if (!NT_STATUS_IS_OK(status)) {
1306                         libnet_unjoin_set_error_string(mem_ctx, r,
1307                                 "failed to find DC for domain %s",
1308                                 r->in.domain_name,
1309                                 get_friendly_nt_error_msg(status));
1310                         return WERR_DOMAIN_CONTROLLER_NOT_FOUND;
1311                 }
1312
1313                 r->in.dc_name = talloc_strdup(mem_ctx,
1314                                               info->domain_controller_name);
1315                 W_ERROR_HAVE_NO_MEMORY(r->in.dc_name);
1316         }
1317
1318         status = libnet_join_unjoindomain_rpc(mem_ctx, r);
1319         if (!NT_STATUS_IS_OK(status)) {
1320                 libnet_unjoin_set_error_string(mem_ctx, r,
1321                         "failed to disable machine account via rpc: %s",
1322                         get_friendly_nt_error_msg(status));
1323                 if (NT_STATUS_EQUAL(status, NT_STATUS_NO_SUCH_USER)) {
1324                         return WERR_SETUP_NOT_JOINED;
1325                 }
1326                 return ntstatus_to_werror(status);
1327         }
1328
1329 #ifdef WITH_ADS
1330         if (r->in.unjoin_flags & WKSSVC_JOIN_FLAGS_ACCOUNT_DELETE) {
1331                 ADS_STATUS ads_status;
1332                 libnet_unjoin_connect_ads(mem_ctx, r);
1333                 ads_status = libnet_unjoin_remove_machine_acct(mem_ctx, r);
1334                 if (!ADS_ERR_OK(ads_status)) {
1335                         libnet_unjoin_set_error_string(mem_ctx, r,
1336                                 "failed to remove machine account from AD: %s",
1337                                 ads_errstr(ads_status));
1338                 }
1339         }
1340 #endif /* WITH_ADS */
1341
1342         libnet_join_unjoindomain_remove_secrets(mem_ctx, r);
1343
1344         return WERR_OK;
1345 }
1346
1347 /****************************************************************
1348 ****************************************************************/
1349
1350 static WERROR libnet_unjoin_pre_processing(TALLOC_CTX *mem_ctx,
1351                                            struct libnet_UnjoinCtx *r)
1352 {
1353         if (r->in.modify_config && !lp_config_backend_is_registry()) {
1354                 return WERR_NOT_SUPPORTED;
1355         }
1356
1357         if (!secrets_init()) {
1358                 libnet_unjoin_set_error_string(mem_ctx, r,
1359                         "Unable to open secrets database");
1360                 return WERR_CAN_NOT_COMPLETE;
1361         }
1362
1363         return WERR_OK;
1364 }
1365
1366
1367 /****************************************************************
1368 ****************************************************************/
1369
1370 WERROR libnet_Unjoin(TALLOC_CTX *mem_ctx,
1371                      struct libnet_UnjoinCtx *r)
1372 {
1373         WERROR werr;
1374
1375         if (r->in.debug) {
1376                 LIBNET_UNJOIN_IN_DUMP_CTX(mem_ctx, r);
1377         }
1378
1379         werr = libnet_unjoin_pre_processing(mem_ctx, r);
1380         if (!W_ERROR_IS_OK(werr)) {
1381                 goto done;
1382         }
1383
1384         if (r->in.unjoin_flags & WKSSVC_JOIN_FLAGS_JOIN_TYPE) {
1385                 werr = libnet_DomainUnjoin(mem_ctx, r);
1386                 if (!W_ERROR_IS_OK(werr)) {
1387                         goto done;
1388                 }
1389         }
1390
1391         werr = do_UnjoinConfig(r);
1392         if (!W_ERROR_IS_OK(werr)) {
1393                 goto done;
1394         }
1395
1396  done:
1397         r->out.result = werr;
1398
1399         if (r->in.debug) {
1400                 LIBNET_UNJOIN_OUT_DUMP_CTX(mem_ctx, r);
1401         }
1402
1403         return werr;
1404 }