s3: smbd - smb1 - fix read of deleted memory in reply_writeclose().
[Samba.git] / source3 / utils / net_util.c
blob13a0ef12d869fc938f92f952a584603624ccc270
1 /*
2 * Unix SMB/CIFS implementation.
3 * Helper routines for net
4 * Copyright (C) Volker Lendecke 2006
5 * Copyright (C) Kai Blin 2008
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.
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.
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/>.
22 #include "includes.h"
23 #include "utils/net.h"
24 #include "rpc_client/cli_pipe.h"
25 #include "../librpc/gen_ndr/ndr_lsa_c.h"
26 #include "rpc_client/cli_lsarpc.h"
27 #include "../librpc/gen_ndr/ndr_dssetup_c.h"
28 #include "secrets.h"
29 #include "../libcli/security/security.h"
30 #include "libsmb/libsmb.h"
32 NTSTATUS net_rpc_lookup_name(struct net_context *c,
33 TALLOC_CTX *mem_ctx, struct cli_state *cli,
34 const char *name, const char **ret_domain,
35 const char **ret_name, struct dom_sid *ret_sid,
36 enum lsa_SidType *ret_type)
38 struct rpc_pipe_client *lsa_pipe = NULL;
39 struct policy_handle pol;
40 NTSTATUS status, result;
41 const char **dom_names;
42 struct dom_sid *sids;
43 enum lsa_SidType *types;
44 struct dcerpc_binding_handle *b;
46 ZERO_STRUCT(pol);
48 status = cli_rpc_pipe_open_noauth(cli, &ndr_table_lsarpc,
49 &lsa_pipe);
50 if (!NT_STATUS_IS_OK(status)) {
51 d_fprintf(stderr, _("Could not initialise lsa pipe\n"));
52 return status;
55 b = lsa_pipe->binding_handle;
57 status = rpccli_lsa_open_policy(lsa_pipe, mem_ctx, false,
58 SEC_FLAG_MAXIMUM_ALLOWED,
59 &pol);
60 if (!NT_STATUS_IS_OK(status)) {
61 d_fprintf(stderr, "open_policy %s: %s\n", _("failed"),
62 nt_errstr(status));
63 return status;
66 status = rpccli_lsa_lookup_names(lsa_pipe, mem_ctx, &pol, 1,
67 &name, &dom_names, 1, &sids, &types);
69 if (!NT_STATUS_IS_OK(status)) {
70 /* This can happen easily, don't log an error */
71 goto done;
74 if (ret_domain != NULL) {
75 *ret_domain = dom_names[0];
77 if (ret_name != NULL) {
78 *ret_name = talloc_strdup(mem_ctx, name);
80 if (ret_sid != NULL) {
81 sid_copy(ret_sid, &sids[0]);
83 if (ret_type != NULL) {
84 *ret_type = types[0];
87 done:
88 if (is_valid_policy_hnd(&pol)) {
89 dcerpc_lsa_Close(b, mem_ctx, &pol, &result);
91 TALLOC_FREE(lsa_pipe);
93 return status;
96 /****************************************************************************
97 Connect to \\server\service.
98 ****************************************************************************/
100 NTSTATUS connect_to_service(struct net_context *c,
101 struct cli_state **cli_ctx,
102 const struct sockaddr_storage *server_ss,
103 const char *server_name,
104 const char *service_name,
105 const char *service_type)
107 NTSTATUS nt_status;
108 int flags = 0;
110 c->opt_password = net_prompt_pass(c, c->opt_user_name);
112 if (c->opt_kerberos) {
113 flags |= CLI_FULL_CONNECTION_USE_KERBEROS;
116 if (c->opt_kerberos && c->opt_password) {
117 flags |= CLI_FULL_CONNECTION_FALLBACK_AFTER_KERBEROS;
120 if (c->opt_ccache) {
121 flags |= CLI_FULL_CONNECTION_USE_CCACHE;
124 nt_status = cli_full_connection(cli_ctx, NULL, server_name,
125 server_ss, c->opt_port,
126 service_name, service_type,
127 c->opt_user_name, c->opt_workgroup,
128 c->opt_password, flags,
129 SMB_SIGNING_DEFAULT);
130 if (!NT_STATUS_IS_OK(nt_status)) {
131 d_fprintf(stderr, _("Could not connect to server %s\n"),
132 server_name);
134 /* Display a nicer message depending on the result */
136 if (NT_STATUS_V(nt_status) ==
137 NT_STATUS_V(NT_STATUS_LOGON_FAILURE))
138 d_fprintf(stderr,
139 _("The username or password was not "
140 "correct.\n"));
142 if (NT_STATUS_V(nt_status) ==
143 NT_STATUS_V(NT_STATUS_ACCOUNT_LOCKED_OUT))
144 d_fprintf(stderr, _("The account was locked out.\n"));
146 if (NT_STATUS_V(nt_status) ==
147 NT_STATUS_V(NT_STATUS_ACCOUNT_DISABLED))
148 d_fprintf(stderr, _("The account was disabled.\n"));
149 return nt_status;
152 if (c->smb_encrypt) {
153 nt_status = cli_force_encryption(*cli_ctx,
154 c->opt_user_name,
155 c->opt_password,
156 c->opt_workgroup);
158 if (NT_STATUS_EQUAL(nt_status,NT_STATUS_NOT_SUPPORTED)) {
159 d_printf(_("Encryption required and "
160 "server that doesn't support "
161 "UNIX extensions - failing connect\n"));
162 } else if (NT_STATUS_EQUAL(nt_status,NT_STATUS_UNKNOWN_REVISION)) {
163 d_printf(_("Encryption required and "
164 "can't get UNIX CIFS extensions "
165 "version from server.\n"));
166 } else if (NT_STATUS_EQUAL(nt_status,NT_STATUS_UNSUPPORTED_COMPRESSION)) {
167 d_printf(_("Encryption required and "
168 "share %s doesn't support "
169 "encryption.\n"), service_name);
170 } else if (!NT_STATUS_IS_OK(nt_status)) {
171 d_printf(_("Encryption required and "
172 "setup failed with error %s.\n"),
173 nt_errstr(nt_status));
176 if (!NT_STATUS_IS_OK(nt_status)) {
177 cli_shutdown(*cli_ctx);
178 *cli_ctx = NULL;
182 return nt_status;
185 /****************************************************************************
186 Connect to \\server\ipc$.
187 ****************************************************************************/
189 NTSTATUS connect_to_ipc(struct net_context *c,
190 struct cli_state **cli_ctx,
191 const struct sockaddr_storage *server_ss,
192 const char *server_name)
194 return connect_to_service(c, cli_ctx, server_ss, server_name, "IPC$",
195 "IPC");
198 /****************************************************************************
199 Connect to \\server\ipc$ anonymously.
200 ****************************************************************************/
202 NTSTATUS connect_to_ipc_anonymous(struct net_context *c,
203 struct cli_state **cli_ctx,
204 const struct sockaddr_storage *server_ss,
205 const char *server_name)
207 NTSTATUS nt_status;
209 nt_status = cli_full_connection(cli_ctx, c->opt_requester_name,
210 server_name, server_ss, c->opt_port,
211 "IPC$", "IPC",
212 "", "",
213 "", 0, SMB_SIGNING_DEFAULT);
215 if (NT_STATUS_IS_OK(nt_status)) {
216 return nt_status;
217 } else {
218 DEBUG(1,("Cannot connect to server (anonymously). Error was %s\n", nt_errstr(nt_status)));
219 return nt_status;
224 * Connect a server and open a given pipe
226 * @param cli_dst A cli_state
227 * @param pipe The pipe to open
228 * @param got_pipe boolean that stores if we got a pipe
230 * @return Normal NTSTATUS return.
232 NTSTATUS connect_dst_pipe(struct net_context *c, struct cli_state **cli_dst,
233 struct rpc_pipe_client **pp_pipe_hnd,
234 const struct ndr_interface_table *table)
236 NTSTATUS nt_status;
237 char *server_name = SMB_STRDUP("127.0.0.1");
238 struct cli_state *cli_tmp = NULL;
239 struct rpc_pipe_client *pipe_hnd = NULL;
241 if (server_name == NULL) {
242 return NT_STATUS_NO_MEMORY;
245 if (c->opt_destination) {
246 SAFE_FREE(server_name);
247 if ((server_name = SMB_STRDUP(c->opt_destination)) == NULL) {
248 return NT_STATUS_NO_MEMORY;
252 /* make a connection to a named pipe */
253 nt_status = connect_to_ipc(c, &cli_tmp, NULL, server_name);
254 if (!NT_STATUS_IS_OK(nt_status)) {
255 SAFE_FREE(server_name);
256 return nt_status;
259 nt_status = cli_rpc_pipe_open_noauth(cli_tmp, table,
260 &pipe_hnd);
261 if (!NT_STATUS_IS_OK(nt_status)) {
262 DEBUG(0, ("couldn't not initialize pipe\n"));
263 cli_shutdown(cli_tmp);
264 SAFE_FREE(server_name);
265 return nt_status;
268 *cli_dst = cli_tmp;
269 *pp_pipe_hnd = pipe_hnd;
270 SAFE_FREE(server_name);
272 return nt_status;
275 /****************************************************************************
276 Use the local machine account (krb) and password for this session.
277 ****************************************************************************/
279 int net_use_krb_machine_account(struct net_context *c)
281 char *user_name = NULL;
283 if (!secrets_init()) {
284 d_fprintf(stderr,_("ERROR: Unable to open secrets database\n"));
285 exit(1);
288 c->opt_password = secrets_fetch_machine_password(
289 c->opt_target_workgroup, NULL, NULL);
290 if (asprintf(&user_name, "%s$@%s", lp_netbios_name(), lp_realm()) == -1) {
291 return -1;
293 c->opt_user_name = user_name;
294 return 0;
297 /****************************************************************************
298 Use the machine account name and password for this session.
299 ****************************************************************************/
301 int net_use_machine_account(struct net_context *c)
303 char *user_name = NULL;
305 if (!secrets_init()) {
306 d_fprintf(stderr,_("ERROR: Unable to open secrets database\n"));
307 exit(1);
310 c->opt_password = secrets_fetch_machine_password(
311 c->opt_target_workgroup, NULL, NULL);
312 if (asprintf(&user_name, "%s$", lp_netbios_name()) == -1) {
313 return -1;
315 c->opt_user_name = user_name;
316 return 0;
319 bool net_find_server(struct net_context *c,
320 const char *domain,
321 unsigned flags,
322 struct sockaddr_storage *server_ss,
323 char **server_name)
325 const char *d = domain ? domain : c->opt_target_workgroup;
327 if (c->opt_host) {
328 *server_name = SMB_STRDUP(c->opt_host);
331 if (c->opt_have_ip) {
332 *server_ss = c->opt_dest_ip;
333 if (!*server_name) {
334 char addr[INET6_ADDRSTRLEN];
335 print_sockaddr(addr, sizeof(addr), &c->opt_dest_ip);
336 *server_name = SMB_STRDUP(addr);
338 } else if (*server_name) {
339 /* resolve the IP address */
340 if (!resolve_name(*server_name, server_ss, 0x20, false)) {
341 DEBUG(1,("Unable to resolve server name\n"));
342 return false;
344 } else if (flags & NET_FLAGS_PDC) {
345 fstring dc_name;
346 struct sockaddr_storage pdc_ss;
348 if (!get_pdc_ip(d, &pdc_ss)) {
349 DEBUG(1,("Unable to resolve PDC server address\n"));
350 return false;
353 if (is_zero_addr(&pdc_ss)) {
354 return false;
357 if (!name_status_find(d, 0x1b, 0x20, &pdc_ss, dc_name)) {
358 return false;
361 *server_name = SMB_STRDUP(dc_name);
362 *server_ss = pdc_ss;
363 } else if (flags & NET_FLAGS_DMB) {
364 struct sockaddr_storage msbrow_ss;
365 char addr[INET6_ADDRSTRLEN];
367 /* if (!resolve_name(MSBROWSE, &msbrow_ip, 1, false)) */
368 if (!resolve_name(d, &msbrow_ss, 0x1B, false)) {
369 DEBUG(1,("Unable to resolve domain browser via name lookup\n"));
370 return false;
372 *server_ss = msbrow_ss;
373 print_sockaddr(addr, sizeof(addr), server_ss);
374 *server_name = SMB_STRDUP(addr);
375 } else if (flags & NET_FLAGS_MASTER) {
376 struct sockaddr_storage brow_ss;
377 char addr[INET6_ADDRSTRLEN];
378 if (!resolve_name(d, &brow_ss, 0x1D, false)) {
379 /* go looking for workgroups */
380 DEBUG(1,("Unable to resolve master browser via name lookup\n"));
381 return false;
383 *server_ss = brow_ss;
384 print_sockaddr(addr, sizeof(addr), server_ss);
385 *server_name = SMB_STRDUP(addr);
386 } else if (!(flags & NET_FLAGS_LOCALHOST_DEFAULT_INSANE)) {
387 if (!interpret_string_addr(server_ss,
388 "127.0.0.1", AI_NUMERICHOST)) {
389 DEBUG(1,("Unable to resolve 127.0.0.1\n"));
390 return false;
392 *server_name = SMB_STRDUP("127.0.0.1");
395 if (!*server_name) {
396 DEBUG(1,("no server to connect to\n"));
397 return false;
400 return true;
403 bool net_find_pdc(struct sockaddr_storage *server_ss,
404 fstring server_name,
405 const char *domain_name)
407 if (!get_pdc_ip(domain_name, server_ss)) {
408 return false;
410 if (is_zero_addr(server_ss)) {
411 return false;
414 if (!name_status_find(domain_name, 0x1b, 0x20, server_ss, server_name)) {
415 return false;
418 return true;
421 NTSTATUS net_make_ipc_connection(struct net_context *c, unsigned flags,
422 struct cli_state **pcli)
424 return net_make_ipc_connection_ex(c, c->opt_workgroup, NULL, NULL, flags, pcli);
427 NTSTATUS net_make_ipc_connection_ex(struct net_context *c ,const char *domain,
428 const char *server,
429 const struct sockaddr_storage *pss,
430 unsigned flags, struct cli_state **pcli)
432 char *server_name = NULL;
433 struct sockaddr_storage server_ss;
434 struct cli_state *cli = NULL;
435 NTSTATUS nt_status;
437 if ( !server || !pss ) {
438 if (!net_find_server(c, domain, flags, &server_ss,
439 &server_name)) {
440 d_fprintf(stderr, _("Unable to find a suitable server "
441 "for domain %s\n"), domain);
442 nt_status = NT_STATUS_UNSUCCESSFUL;
443 goto done;
445 } else {
446 server_name = SMB_STRDUP( server );
447 server_ss = *pss;
450 if (flags & NET_FLAGS_ANONYMOUS) {
451 nt_status = connect_to_ipc_anonymous(c, &cli, &server_ss,
452 server_name);
453 } else {
454 nt_status = connect_to_ipc(c, &cli, &server_ss,
455 server_name);
458 /* store the server in the affinity cache if it was a PDC */
460 if ( (flags & NET_FLAGS_PDC) && NT_STATUS_IS_OK(nt_status) )
461 saf_store(cli->server_domain, server_name);
463 SAFE_FREE(server_name);
464 if (!NT_STATUS_IS_OK(nt_status)) {
465 d_fprintf(stderr, _("Connection failed: %s\n"),
466 nt_errstr(nt_status));
467 cli = NULL;
468 } else if (c->opt_request_timeout) {
469 cli_set_timeout(cli, c->opt_request_timeout * 1000);
472 done:
473 if (pcli != NULL) {
474 *pcli = cli;
476 return nt_status;
479 /****************************************************************************
480 ****************************************************************************/
482 const char *net_prompt_pass(struct net_context *c, const char *user)
484 char *prompt = NULL;
485 char pwd[256] = {0};
486 int rc;
488 if (c->opt_password) {
489 return c->opt_password;
492 if (c->opt_machine_pass) {
493 return NULL;
496 if (c->opt_kerberos && !c->opt_user_specified) {
497 return NULL;
500 if (asprintf(&prompt, _("Enter %s's password:"), user) == -1) {
501 return NULL;
504 rc = samba_getpass(prompt, pwd, sizeof(pwd), false, false);
505 SAFE_FREE(prompt);
506 if (rc < 0) {
507 return NULL;
510 return SMB_STRDUP(pwd);
513 int net_run_function(struct net_context *c, int argc, const char **argv,
514 const char *whoami, struct functable *table)
516 int i;
518 if (argc != 0) {
519 for (i=0; table[i].funcname != NULL; i++) {
520 if (strcasecmp_m(argv[0], table[i].funcname) == 0)
521 return table[i].fn(c, argc-1, argv+1);
525 if (c->display_usage == false) {
526 d_fprintf(stderr, _("Invalid command: %s %s\n"), whoami,
527 (argc > 0)?argv[0]:"");
529 d_printf(_("Usage:\n"));
530 for (i=0; table[i].funcname != NULL; i++) {
531 if(c->display_usage == false)
532 d_printf("%s %-15s %s\n", whoami, table[i].funcname,
533 _(table[i].description));
534 else
535 d_printf("%s\n", _(table[i].usage));
538 return c->display_usage?0:-1;
541 void net_display_usage_from_functable(struct functable *table)
543 int i;
544 for (i=0; table[i].funcname != NULL; i++) {
545 d_printf("%s\n", _(table[i].usage));
549 const char *net_share_type_str(int num_type)
551 switch(num_type) {
552 case 0: return _("Disk");
553 case 1: return _("Print");
554 case 2: return _("Dev");
555 case 3: return _("IPC");
556 default: return _("Unknown");
560 static NTSTATUS net_scan_dc_noad(struct net_context *c,
561 struct cli_state *cli,
562 struct net_dc_info *dc_info)
564 TALLOC_CTX *mem_ctx = talloc_tos();
565 struct rpc_pipe_client *pipe_hnd = NULL;
566 struct dcerpc_binding_handle *b;
567 NTSTATUS status, result;
568 struct policy_handle pol;
569 union lsa_PolicyInformation *info;
571 ZERO_STRUCTP(dc_info);
572 ZERO_STRUCT(pol);
574 status = cli_rpc_pipe_open_noauth(cli, &ndr_table_lsarpc,
575 &pipe_hnd);
576 if (!NT_STATUS_IS_OK(status)) {
577 return status;
580 b = pipe_hnd->binding_handle;
582 status = dcerpc_lsa_open_policy(b, mem_ctx,
583 false,
584 SEC_FLAG_MAXIMUM_ALLOWED,
585 &pol,
586 &result);
587 if (!NT_STATUS_IS_OK(status)) {
588 goto done;
590 if (!NT_STATUS_IS_OK(result)) {
591 status = result;
592 goto done;
595 status = dcerpc_lsa_QueryInfoPolicy(b, mem_ctx,
596 &pol,
597 LSA_POLICY_INFO_ACCOUNT_DOMAIN,
598 &info,
599 &result);
600 if (!NT_STATUS_IS_OK(status)) {
601 goto done;
603 if (!NT_STATUS_IS_OK(result)) {
604 status = result;
605 goto done;
608 dc_info->netbios_domain_name = talloc_strdup(mem_ctx, info->account_domain.name.string);
609 if (dc_info->netbios_domain_name == NULL) {
610 status = NT_STATUS_NO_MEMORY;
611 goto done;
614 done:
615 if (is_valid_policy_hnd(&pol)) {
616 dcerpc_lsa_Close(b, mem_ctx, &pol, &result);
619 TALLOC_FREE(pipe_hnd);
621 return status;
624 NTSTATUS net_scan_dc(struct net_context *c,
625 struct cli_state *cli,
626 struct net_dc_info *dc_info)
628 TALLOC_CTX *mem_ctx = talloc_tos();
629 struct rpc_pipe_client *dssetup_pipe = NULL;
630 struct dcerpc_binding_handle *dssetup_handle = NULL;
631 union dssetup_DsRoleInfo info;
632 NTSTATUS status;
633 WERROR werr;
635 ZERO_STRUCTP(dc_info);
637 status = cli_rpc_pipe_open_noauth(cli, &ndr_table_dssetup,
638 &dssetup_pipe);
639 if (!NT_STATUS_IS_OK(status)) {
640 DEBUG(10,("net_scan_dc: failed to open dssetup pipe with %s, "
641 "retrying with lsa pipe\n", nt_errstr(status)));
642 return net_scan_dc_noad(c, cli, dc_info);
644 dssetup_handle = dssetup_pipe->binding_handle;
646 status = dcerpc_dssetup_DsRoleGetPrimaryDomainInformation(dssetup_handle, mem_ctx,
647 DS_ROLE_BASIC_INFORMATION,
648 &info,
649 &werr);
650 TALLOC_FREE(dssetup_pipe);
652 if (NT_STATUS_IS_OK(status)) {
653 status = werror_to_ntstatus(werr);
655 if (!NT_STATUS_IS_OK(status)) {
656 return status;
659 dc_info->is_dc = (info.basic.role & (DS_ROLE_PRIMARY_DC|DS_ROLE_BACKUP_DC));
660 dc_info->is_pdc = (info.basic.role & DS_ROLE_PRIMARY_DC);
661 dc_info->is_ad = (info.basic.flags & DS_ROLE_PRIMARY_DS_RUNNING);
662 dc_info->is_mixed_mode = (info.basic.flags & DS_ROLE_PRIMARY_DS_MIXED_MODE);
663 dc_info->netbios_domain_name = talloc_strdup(mem_ctx, info.basic.domain);
664 dc_info->dns_domain_name = talloc_strdup(mem_ctx, info.basic.dns_domain);
665 dc_info->forest_name = talloc_strdup(mem_ctx, info.basic.forest);
667 return NT_STATUS_OK;