Merge branch 'master' of ssh://git.samba.org/data/git/samba
[nivanova/samba-autobuild/.git] / source3 / utils / net_util.c
1 /*
2  *  Unix SMB/CIFS implementation.
3  *  Helper routines for net
4  *  Copyright (C) Volker Lendecke 2006
5  *  Copyright (C) Kai Blin 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
22 #include "includes.h"
23 #include "utils/net.h"
24
25 NTSTATUS net_rpc_lookup_name(struct net_context *c,
26                              TALLOC_CTX *mem_ctx, struct cli_state *cli,
27                              const char *name, const char **ret_domain,
28                              const char **ret_name, DOM_SID *ret_sid,
29                              enum lsa_SidType *ret_type)
30 {
31         struct rpc_pipe_client *lsa_pipe;
32         struct policy_handle pol;
33         NTSTATUS result = NT_STATUS_OK;
34         const char **dom_names;
35         DOM_SID *sids;
36         enum lsa_SidType *types;
37
38         ZERO_STRUCT(pol);
39
40         result = cli_rpc_pipe_open_noauth(cli, &ndr_table_lsarpc.syntax_id,
41                                           &lsa_pipe);
42         if (!NT_STATUS_IS_OK(result)) {
43                 d_fprintf(stderr, "Could not initialise lsa pipe\n");
44                 return result;
45         }
46
47         result = rpccli_lsa_open_policy(lsa_pipe, mem_ctx, false,
48                                         SEC_FLAG_MAXIMUM_ALLOWED,
49                                         &pol);
50         if (!NT_STATUS_IS_OK(result)) {
51                 d_fprintf(stderr, "open_policy failed: %s\n",
52                           nt_errstr(result));
53                 return result;
54         }
55
56         result = rpccli_lsa_lookup_names(lsa_pipe, mem_ctx, &pol, 1,
57                                          &name, &dom_names, 1, &sids, &types);
58
59         if (!NT_STATUS_IS_OK(result)) {
60                 /* This can happen easily, don't log an error */
61                 goto done;
62         }
63
64         if (ret_domain != NULL) {
65                 *ret_domain = dom_names[0];
66         }
67         if (ret_name != NULL) {
68                 *ret_name = talloc_strdup(mem_ctx, name);
69         }
70         if (ret_sid != NULL) {
71                 sid_copy(ret_sid, &sids[0]);
72         }
73         if (ret_type != NULL) {
74                 *ret_type = types[0];
75         }
76
77  done:
78         if (is_valid_policy_hnd(&pol)) {
79                 rpccli_lsa_Close(lsa_pipe, mem_ctx, &pol);
80         }
81         TALLOC_FREE(lsa_pipe);
82
83         return result;
84 }
85
86 /****************************************************************************
87  Connect to \\server\service.
88 ****************************************************************************/
89
90 NTSTATUS connect_to_service(struct net_context *c,
91                                         struct cli_state **cli_ctx,
92                                         struct sockaddr_storage *server_ss,
93                                         const char *server_name,
94                                         const char *service_name,
95                                         const char *service_type)
96 {
97         NTSTATUS nt_status;
98         int flags = 0;
99         struct user_auth_info *ai = c->auth_info;
100
101         set_cmdline_auth_info_getpass(ai);
102
103         if (get_cmdline_auth_info_use_kerberos(ai)) {
104                 flags |= CLI_FULL_CONNECTION_USE_KERBEROS |
105                          CLI_FULL_CONNECTION_FALLBACK_AFTER_KERBEROS;
106         }
107
108         nt_status = cli_full_connection(cli_ctx, NULL, server_name,
109                                         server_ss, c->opt_port,
110                                         service_name, service_type,
111                                         get_cmdline_auth_info_username(ai),
112                                         c->opt_workgroup,
113                                         get_cmdline_auth_info_password(ai),
114                                         flags, Undefined, NULL);
115         if (!NT_STATUS_IS_OK(nt_status)) {
116                 d_fprintf(stderr, "Could not connect to server %s\n", server_name);
117
118                 /* Display a nicer message depending on the result */
119
120                 if (NT_STATUS_V(nt_status) ==
121                     NT_STATUS_V(NT_STATUS_LOGON_FAILURE))
122                         d_fprintf(stderr, "The username or password was not correct.\n");
123
124                 if (NT_STATUS_V(nt_status) ==
125                     NT_STATUS_V(NT_STATUS_ACCOUNT_LOCKED_OUT))
126                         d_fprintf(stderr, "The account was locked out.\n");
127
128                 if (NT_STATUS_V(nt_status) ==
129                     NT_STATUS_V(NT_STATUS_ACCOUNT_DISABLED))
130                         d_fprintf(stderr, "The account was disabled.\n");
131                 return nt_status;
132         }
133
134         if (get_cmdline_auth_info_smb_encrypt(ai)) {
135                 nt_status = cli_force_encryption(*cli_ctx,
136                                         get_cmdline_auth_info_username(ai),
137                                         get_cmdline_auth_info_password(ai),
138                                         c->opt_workgroup);
139
140                 if (NT_STATUS_EQUAL(nt_status,NT_STATUS_NOT_SUPPORTED)) {
141                         d_printf("Encryption required and "
142                                 "server that doesn't support "
143                                 "UNIX extensions - failing connect\n");
144                 } else if (NT_STATUS_EQUAL(nt_status,NT_STATUS_UNKNOWN_REVISION)) {
145                         d_printf("Encryption required and "
146                                 "can't get UNIX CIFS extensions "
147                                 "version from server.\n");
148                 } else if (NT_STATUS_EQUAL(nt_status,NT_STATUS_UNSUPPORTED_COMPRESSION)) {
149                         d_printf("Encryption required and "
150                                 "share %s doesn't support "
151                                 "encryption.\n", service_name);
152                 } else if (!NT_STATUS_IS_OK(nt_status)) {
153                         d_printf("Encryption required and "
154                                 "setup failed with error %s.\n",
155                                 nt_errstr(nt_status));
156                 }
157
158                 if (!NT_STATUS_IS_OK(nt_status)) {
159                         cli_shutdown(*cli_ctx);
160                         *cli_ctx = NULL;
161                 }
162         }
163
164         return nt_status;
165 }
166
167 /****************************************************************************
168  Connect to \\server\ipc$.
169 ****************************************************************************/
170
171 NTSTATUS connect_to_ipc(struct net_context *c,
172                         struct cli_state **cli_ctx,
173                         struct sockaddr_storage *server_ss,
174                         const char *server_name)
175 {
176         return connect_to_service(c, cli_ctx, server_ss, server_name, "IPC$",
177                                   "IPC");
178 }
179
180 /****************************************************************************
181  Connect to \\server\ipc$ anonymously.
182 ****************************************************************************/
183
184 NTSTATUS connect_to_ipc_anonymous(struct net_context *c,
185                                 struct cli_state **cli_ctx,
186                                 struct sockaddr_storage *server_ss,
187                                 const char *server_name)
188 {
189         NTSTATUS nt_status;
190
191         nt_status = cli_full_connection(cli_ctx, c->opt_requester_name,
192                                         server_name, server_ss, c->opt_port,
193                                         "IPC$", "IPC",
194                                         "", "",
195                                         "", 0, Undefined, NULL);
196
197         if (NT_STATUS_IS_OK(nt_status)) {
198                 return nt_status;
199         } else {
200                 DEBUG(1,("Cannot connect to server (anonymously).  Error was %s\n", nt_errstr(nt_status)));
201                 return nt_status;
202         }
203 }
204
205 /****************************************************************************
206  Return malloced user@realm for krb5 login.
207 ****************************************************************************/
208
209 static char *get_user_and_realm(const char *username)
210 {
211         char *user_and_realm = NULL;
212
213         if (!username) {
214                 return NULL;
215         }
216         if (strchr_m(username, '@')) {
217                 user_and_realm = SMB_STRDUP(username);
218         } else {
219                 if (asprintf(&user_and_realm, "%s@%s", username, lp_realm()) == -1) {
220                         user_and_realm = NULL;
221                 }
222         }
223         return user_and_realm;
224 }
225
226 /****************************************************************************
227  Connect to \\server\ipc$ using KRB5.
228 ****************************************************************************/
229
230 NTSTATUS connect_to_ipc_krb5(struct net_context *c,
231                         struct cli_state **cli_ctx,
232                         struct sockaddr_storage *server_ss,
233                         const char *server_name)
234 {
235         NTSTATUS nt_status;
236         char *user_and_realm = NULL;
237         struct user_auth_info *ai = c->auth_info;
238
239         /* FIXME: Should get existing kerberos ticket if possible. */
240         set_cmdline_auth_info_getpass(ai);
241
242         user_and_realm = get_user_and_realm(get_cmdline_auth_info_username(ai));
243         if (!user_and_realm) {
244                 return NT_STATUS_NO_MEMORY;
245         }
246
247         nt_status = cli_full_connection(cli_ctx, NULL, server_name,
248                                         server_ss, c->opt_port,
249                                         "IPC$", "IPC",
250                                         user_and_realm, c->opt_workgroup,
251                                         get_cmdline_auth_info_password(ai),
252                                         CLI_FULL_CONNECTION_USE_KERBEROS,
253                                         Undefined, NULL);
254
255         SAFE_FREE(user_and_realm);
256
257         if (!NT_STATUS_IS_OK(nt_status)) {
258                 DEBUG(1,("Cannot connect to server using kerberos.  Error was %s\n", nt_errstr(nt_status)));
259                 return nt_status;
260         }
261
262         if (get_cmdline_auth_info_smb_encrypt(ai)) {
263                 nt_status = cli_cm_force_encryption(*cli_ctx,
264                                         user_and_realm,
265                                         get_cmdline_auth_info_password(ai),
266                                         c->opt_workgroup,
267                                         "IPC$");
268                 if (!NT_STATUS_IS_OK(nt_status)) {
269                         cli_shutdown(*cli_ctx);
270                         *cli_ctx = NULL;
271                 }
272         }
273
274         return nt_status;
275 }
276
277 /**
278  * Connect a server and open a given pipe
279  *
280  * @param cli_dst               A cli_state
281  * @param pipe                  The pipe to open
282  * @param got_pipe              boolean that stores if we got a pipe
283  *
284  * @return Normal NTSTATUS return.
285  **/
286 NTSTATUS connect_dst_pipe(struct net_context *c, struct cli_state **cli_dst,
287                           struct rpc_pipe_client **pp_pipe_hnd,
288                           const struct ndr_syntax_id *interface)
289 {
290         NTSTATUS nt_status;
291         char *server_name = SMB_STRDUP("127.0.0.1");
292         struct cli_state *cli_tmp = NULL;
293         struct rpc_pipe_client *pipe_hnd = NULL;
294
295         if (server_name == NULL) {
296                 return NT_STATUS_NO_MEMORY;
297         }
298
299         if (c->opt_destination) {
300                 SAFE_FREE(server_name);
301                 if ((server_name = SMB_STRDUP(c->opt_destination)) == NULL) {
302                         return NT_STATUS_NO_MEMORY;
303                 }
304         }
305
306         /* make a connection to a named pipe */
307         nt_status = connect_to_ipc(c, &cli_tmp, NULL, server_name);
308         if (!NT_STATUS_IS_OK(nt_status)) {
309                 SAFE_FREE(server_name);
310                 return nt_status;
311         }
312
313         nt_status = cli_rpc_pipe_open_noauth(cli_tmp, interface,
314                                              &pipe_hnd);
315         if (!NT_STATUS_IS_OK(nt_status)) {
316                 DEBUG(0, ("couldn't not initialize pipe\n"));
317                 cli_shutdown(cli_tmp);
318                 SAFE_FREE(server_name);
319                 return nt_status;
320         }
321
322         *cli_dst = cli_tmp;
323         *pp_pipe_hnd = pipe_hnd;
324         SAFE_FREE(server_name);
325
326         return nt_status;
327 }
328
329 bool net_find_server(struct net_context *c,
330                         const char *domain,
331                         unsigned flags,
332                         struct sockaddr_storage *server_ss,
333                         char **server_name)
334 {
335         const char *d = domain ? domain : c->opt_target_workgroup;
336
337         if (c->opt_host) {
338                 *server_name = SMB_STRDUP(c->opt_host);
339         }
340
341         if (c->opt_have_ip) {
342                 *server_ss = c->opt_dest_ip;
343                 if (!*server_name) {
344                         char addr[INET6_ADDRSTRLEN];
345                         print_sockaddr(addr, sizeof(addr), &c->opt_dest_ip);
346                         *server_name = SMB_STRDUP(addr);
347                 }
348         } else if (*server_name) {
349                 /* resolve the IP address */
350                 if (!resolve_name(*server_name, server_ss, 0x20))  {
351                         DEBUG(1,("Unable to resolve server name\n"));
352                         return false;
353                 }
354         } else if (flags & NET_FLAGS_PDC) {
355                 fstring dc_name;
356                 struct sockaddr_storage pdc_ss;
357
358                 if (!get_pdc_ip(d, &pdc_ss)) {
359                         DEBUG(1,("Unable to resolve PDC server address\n"));
360                         return false;
361                 }
362
363                 if (is_zero_addr((struct sockaddr *)&pdc_ss)) {
364                         return false;
365                 }
366
367                 if (!name_status_find(d, 0x1b, 0x20, &pdc_ss, dc_name)) {
368                         return false;
369                 }
370
371                 *server_name = SMB_STRDUP(dc_name);
372                 *server_ss = pdc_ss;
373         } else if (flags & NET_FLAGS_DMB) {
374                 struct sockaddr_storage msbrow_ss;
375                 char addr[INET6_ADDRSTRLEN];
376
377                 /*  if (!resolve_name(MSBROWSE, &msbrow_ip, 1)) */
378                 if (!resolve_name(d, &msbrow_ss, 0x1B))  {
379                         DEBUG(1,("Unable to resolve domain browser via name lookup\n"));
380                         return false;
381                 }
382                 *server_ss = msbrow_ss;
383                 print_sockaddr(addr, sizeof(addr), server_ss);
384                 *server_name = SMB_STRDUP(addr);
385         } else if (flags & NET_FLAGS_MASTER) {
386                 struct sockaddr_storage brow_ss;
387                 char addr[INET6_ADDRSTRLEN];
388                 if (!resolve_name(d, &brow_ss, 0x1D))  {
389                                 /* go looking for workgroups */
390                         DEBUG(1,("Unable to resolve master browser via name lookup\n"));
391                         return false;
392                 }
393                 *server_ss = brow_ss;
394                 print_sockaddr(addr, sizeof(addr), server_ss);
395                 *server_name = SMB_STRDUP(addr);
396         } else if (!(flags & NET_FLAGS_LOCALHOST_DEFAULT_INSANE)) {
397                 if (!interpret_string_addr(server_ss,
398                                         "127.0.0.1", AI_NUMERICHOST)) {
399                         DEBUG(1,("Unable to resolve 127.0.0.1\n"));
400                         return false;
401                 }
402                 *server_name = SMB_STRDUP("127.0.0.1");
403         }
404
405         if (!*server_name) {
406                 DEBUG(1,("no server to connect to\n"));
407                 return false;
408         }
409
410         return true;
411 }
412
413 bool net_find_pdc(struct sockaddr_storage *server_ss,
414                 fstring server_name,
415                 const char *domain_name)
416 {
417         if (!get_pdc_ip(domain_name, server_ss)) {
418                 return false;
419         }
420         if (is_zero_addr((struct sockaddr *)server_ss)) {
421                 return false;
422         }
423
424         if (!name_status_find(domain_name, 0x1b, 0x20, server_ss, server_name)) {
425                 return false;
426         }
427
428         return true;
429 }
430
431 NTSTATUS net_make_ipc_connection(struct net_context *c, unsigned flags,
432                                  struct cli_state **pcli)
433 {
434         return net_make_ipc_connection_ex(c, c->opt_workgroup, NULL, NULL, flags, pcli);
435 }
436
437 NTSTATUS net_make_ipc_connection_ex(struct net_context *c ,const char *domain,
438                                     const char *server,
439                                     struct sockaddr_storage *pss,
440                                     unsigned flags, struct cli_state **pcli)
441 {
442         char *server_name = NULL;
443         struct sockaddr_storage server_ss;
444         struct cli_state *cli = NULL;
445         NTSTATUS nt_status;
446
447         if ( !server || !pss ) {
448                 if (!net_find_server(c, domain, flags, &server_ss,
449                                      &server_name)) {
450                         d_fprintf(stderr, "Unable to find a suitable server "
451                                 "for domain %s\n", domain);
452                         nt_status = NT_STATUS_UNSUCCESSFUL;
453                         goto done;
454                 }
455         } else {
456                 server_name = SMB_STRDUP( server );
457                 server_ss = *pss;
458         }
459
460         if (flags & NET_FLAGS_ANONYMOUS) {
461                 nt_status = connect_to_ipc_anonymous(c, &cli, &server_ss,
462                                                      server_name);
463         } else {
464                 nt_status = connect_to_ipc(c, &cli, &server_ss,
465                                            server_name);
466         }
467
468         /* store the server in the affinity cache if it was a PDC */
469
470         if ( (flags & NET_FLAGS_PDC) && NT_STATUS_IS_OK(nt_status) )
471                 saf_store( cli->server_domain, cli->desthost );
472
473         SAFE_FREE(server_name);
474         if (!NT_STATUS_IS_OK(nt_status)) {
475                 d_fprintf(stderr, "Connection failed: %s\n",
476                           nt_errstr(nt_status));
477                 cli = NULL;
478         } else if (c->opt_request_timeout) {
479                 cli_set_timeout(cli, c->opt_request_timeout * 1000);
480         }
481
482 done:
483         if (pcli != NULL) {
484                 *pcli = cli;
485         }
486         return nt_status;
487 }
488
489 /****************************************************************************
490 ****************************************************************************/
491
492 int net_run_function(struct net_context *c, int argc, const char **argv,
493                       const char *whoami, struct functable *table)
494 {
495         int i;
496
497         if (argc != 0) {
498                 for (i=0; table[i].funcname != NULL; i++) {
499                         if (StrCaseCmp(argv[0], table[i].funcname) == 0)
500                                 return table[i].fn(c, argc-1, argv+1);
501                 }
502         }
503
504         if (c->display_usage == false) {
505                 d_fprintf(stderr, "Invalid command: %s %s\n", whoami,
506                           (argc > 0)?argv[0]:"");
507         }
508         d_printf("Usage:\n");
509         for (i=0; table[i].funcname != NULL; i++) {
510                 if(c->display_usage == false)
511                         d_printf("%s %-15s %s\n", whoami, table[i].funcname,
512                                  table[i].description);
513                 else
514                         d_printf("%s\n", table[i].usage);
515         }
516
517         return c->display_usage?0:-1;
518 }
519
520 void net_display_usage_from_functable(struct functable *table)
521 {
522         int i;
523         for (i=0; table[i].funcname != NULL; i++) {
524                 d_printf("%s\n", table[i].usage);
525         }
526 }
527
528 const char *net_share_type_str(int num_type)
529 {
530         switch(num_type) {
531                 case 0: return "Disk";
532                 case 1: return "Print";
533                 case 2: return "Dev";
534                 case 3: return "IPC";
535                 default: return "Unknown";
536         }
537 }
538
539 NTSTATUS net_scan_dc(struct net_context *c,
540                      struct cli_state *cli,
541                      struct net_dc_info *dc_info)
542 {
543         TALLOC_CTX *mem_ctx = talloc_tos();
544         struct rpc_pipe_client *dssetup_pipe = NULL;
545         union dssetup_DsRoleInfo info;
546         NTSTATUS status;
547
548         ZERO_STRUCTP(dc_info);
549
550         status = cli_rpc_pipe_open_noauth(cli, &ndr_table_dssetup.syntax_id,
551                                           &dssetup_pipe);
552         if (!NT_STATUS_IS_OK(status)) {
553                 return status;
554         }
555
556         status = rpccli_dssetup_DsRoleGetPrimaryDomainInformation(dssetup_pipe, mem_ctx,
557                                                                   DS_ROLE_BASIC_INFORMATION,
558                                                                   &info,
559                                                                   NULL);
560         TALLOC_FREE(dssetup_pipe);
561
562         if (!NT_STATUS_IS_OK(status)) {
563                 return status;
564         }
565
566         dc_info->is_dc  = (info.basic.role & (DS_ROLE_PRIMARY_DC|DS_ROLE_BACKUP_DC));
567         dc_info->is_pdc = (info.basic.role & DS_ROLE_PRIMARY_DC);
568         dc_info->is_ad  = (info.basic.flags & DS_ROLE_PRIMARY_DS_RUNNING);
569         dc_info->is_mixed_mode = (info.basic.flags & DS_ROLE_PRIMARY_DS_MIXED_MODE);
570         dc_info->netbios_domain_name = talloc_strdup(mem_ctx, info.basic.domain);
571         dc_info->dns_domain_name = talloc_strdup(mem_ctx, info.basic.dns_domain);
572         dc_info->forest_name = talloc_strdup(mem_ctx, info.basic.forest);
573
574         return NT_STATUS_OK;
575 }