build: format the header check for netinet/ip.h more nicely
[Samba/bb.git] / source / libsmb / libsmb_server.c
blob6880f5c3426eed4156c9f10e67ebd3b180b2cd3f
1 /*
2 Unix SMB/Netbios implementation.
3 SMB client library implementation
4 Copyright (C) Andrew Tridgell 1998
5 Copyright (C) Richard Sharpe 2000, 2002
6 Copyright (C) John Terpstra 2000
7 Copyright (C) Tom Jansen (Ninja ISD) 2002
8 Copyright (C) Derrell Lipman 2003-2008
9 Copyright (C) Jeremy Allison 2007, 2008
11 This program is free software; you can redistribute it and/or modify
12 it under the terms of the GNU General Public License as published by
13 the Free Software Foundation; either version 3 of the License, or
14 (at your option) any later version.
16 This program is distributed in the hope that it will be useful,
17 but WITHOUT ANY WARRANTY; without even the implied warranty of
18 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
19 GNU General Public License for more details.
21 You should have received a copy of the GNU General Public License
22 along with this program. If not, see <http://www.gnu.org/licenses/>.
25 #include "includes.h"
26 #include "libsmbclient.h"
27 #include "libsmb_internal.h"
30 /*
31 * Check a server for being alive and well.
32 * returns 0 if the server is in shape. Returns 1 on error
34 * Also useable outside libsmbclient to enable external cache
35 * to do some checks too.
37 int
38 SMBC_check_server(SMBCCTX * context,
39 SMBCSRV * server)
41 socklen_t size;
42 struct sockaddr addr;
44 size = sizeof(addr);
45 return (getpeername(server->cli->fd, &addr, &size) == -1);
48 /*
49 * Remove a server from the cached server list it's unused.
50 * On success, 0 is returned. 1 is returned if the server could not be removed.
52 * Also useable outside libsmbclient
54 int
55 SMBC_remove_unused_server(SMBCCTX * context,
56 SMBCSRV * srv)
58 SMBCFILE * file;
60 /* are we being fooled ? */
61 if (!context || !context->internal->initialized || !srv) {
62 return 1;
65 /* Check all open files/directories for a relation with this server */
66 for (file = context->internal->files; file; file = file->next) {
67 if (file->srv == srv) {
68 /* Still used */
69 DEBUG(3, ("smbc_remove_usused_server: "
70 "%p still used by %p.\n",
71 srv, file));
72 return 1;
76 DLIST_REMOVE(context->internal->servers, srv);
78 cli_shutdown(srv->cli);
79 srv->cli = NULL;
81 DEBUG(3, ("smbc_remove_usused_server: %p removed.\n", srv));
83 smbc_getFunctionRemoveCachedServer(context)(context, srv);
85 SAFE_FREE(srv);
86 return 0;
89 /****************************************************************
90 * Call the auth_fn with fixed size (fstring) buffers.
91 ***************************************************************/
92 void
93 SMBC_call_auth_fn(TALLOC_CTX *ctx,
94 SMBCCTX *context,
95 const char *server,
96 const char *share,
97 char **pp_workgroup,
98 char **pp_username,
99 char **pp_password)
101 fstring workgroup;
102 fstring username;
103 fstring password;
104 smbc_get_auth_data_with_context_fn auth_with_context_fn;
106 strlcpy(workgroup, *pp_workgroup, sizeof(workgroup));
107 strlcpy(username, *pp_username, sizeof(username));
108 strlcpy(password, *pp_password, sizeof(password));
110 /* See if there's an authentication with context function provided */
111 auth_with_context_fn = smbc_getFunctionAuthDataWithContext(context);
112 if (auth_with_context_fn)
114 (* auth_with_context_fn)(context,
115 server, share,
116 workgroup, sizeof(workgroup),
117 username, sizeof(username),
118 password, sizeof(password));
120 else
122 smbc_getFunctionAuthData(context)(server, share,
123 workgroup, sizeof(workgroup),
124 username, sizeof(username),
125 password, sizeof(password));
128 TALLOC_FREE(*pp_workgroup);
129 TALLOC_FREE(*pp_username);
130 TALLOC_FREE(*pp_password);
132 *pp_workgroup = talloc_strdup(ctx, workgroup);
133 *pp_username = talloc_strdup(ctx, username);
134 *pp_password = talloc_strdup(ctx, password);
138 void
139 SMBC_get_auth_data(const char *server, const char *share,
140 char *workgroup_buf, int workgroup_buf_len,
141 char *username_buf, int username_buf_len,
142 char *password_buf, int password_buf_len)
144 /* Default function just uses provided data. Nothing to do. */
149 SMBCSRV *
150 SMBC_find_server(TALLOC_CTX *ctx,
151 SMBCCTX *context,
152 const char *server,
153 const char *share,
154 char **pp_workgroup,
155 char **pp_username,
156 char **pp_password)
158 SMBCSRV *srv;
159 int auth_called = 0;
161 if (!pp_workgroup || !pp_username || !pp_password) {
162 return NULL;
165 check_server_cache:
167 srv = smbc_getFunctionGetCachedServer(context)(context,
168 server, share,
169 *pp_workgroup,
170 *pp_username);
172 if (!auth_called && !srv && (!*pp_username || !(*pp_username)[0] ||
173 !*pp_password || !(*pp_password)[0])) {
174 SMBC_call_auth_fn(ctx, context, server, share,
175 pp_workgroup, pp_username, pp_password);
178 * However, smbc_auth_fn may have picked up info relating to
179 * an existing connection, so try for an existing connection
180 * again ...
182 auth_called = 1;
183 goto check_server_cache;
187 if (srv) {
188 if (smbc_getFunctionCheckServer(context)(context, srv)) {
190 * This server is no good anymore
191 * Try to remove it and check for more possible
192 * servers in the cache
194 if (smbc_getFunctionRemoveUnusedServer(context)(context,
195 srv)) {
197 * We could not remove the server completely,
198 * remove it from the cache so we will not get
199 * it again. It will be removed when the last
200 * file/dir is closed.
202 smbc_getFunctionRemoveCachedServer(context)(context,
203 srv);
207 * Maybe there are more cached connections to this
208 * server
210 goto check_server_cache;
213 return srv;
216 return NULL;
220 * Connect to a server, possibly on an existing connection
222 * Here, what we want to do is: If the server and username
223 * match an existing connection, reuse that, otherwise, establish a
224 * new connection.
226 * If we have to create a new connection, call the auth_fn to get the
227 * info we need, unless the username and password were passed in.
230 SMBCSRV *
231 SMBC_server(TALLOC_CTX *ctx,
232 SMBCCTX *context,
233 bool connect_if_not_found,
234 const char *server,
235 const char *share,
236 char **pp_workgroup,
237 char **pp_username,
238 char **pp_password)
240 SMBCSRV *srv=NULL;
241 char *workgroup = NULL;
242 struct cli_state *c;
243 struct nmb_name called, calling;
244 const char *server_n = server;
245 struct sockaddr_storage ss;
246 int tried_reverse = 0;
247 int port_try_first;
248 int port_try_next;
249 int is_ipc = (share != NULL && strcmp(share, "IPC$") == 0);
250 uint32 fs_attrs = 0;
251 const char *username_used;
252 NTSTATUS status;
254 zero_sockaddr(&ss);
255 ZERO_STRUCT(c);
257 if (server[0] == 0) {
258 errno = EPERM;
259 return NULL;
262 /* Look for a cached connection */
263 srv = SMBC_find_server(ctx, context, server, share,
264 pp_workgroup, pp_username, pp_password);
267 * If we found a connection and we're only allowed one share per
268 * server...
270 if (srv &&
271 *share != '\0' &&
272 smbc_getOptionOneSharePerServer(context)) {
275 * ... then if there's no current connection to the share,
276 * connect to it. SMBC_find_server(), or rather the function
277 * pointed to by context->get_cached_srv_fn which
278 * was called by SMBC_find_server(), will have issued a tree
279 * disconnect if the requested share is not the same as the
280 * one that was already connected.
282 if (srv->cli->cnum == (uint16) -1) {
283 /* Ensure we have accurate auth info */
284 SMBC_call_auth_fn(ctx, context, server, share,
285 pp_workgroup,
286 pp_username,
287 pp_password);
289 if (!*pp_workgroup || !*pp_username || !*pp_password) {
290 errno = ENOMEM;
291 cli_shutdown(srv->cli);
292 srv->cli = NULL;
293 smbc_getFunctionRemoveCachedServer(context)(context,
294 srv);
295 return NULL;
299 * We don't need to renegotiate encryption
300 * here as the encryption context is not per
301 * tid.
304 if (!cli_send_tconX(srv->cli, share, "?????",
305 *pp_password,
306 strlen(*pp_password)+1)) {
308 errno = SMBC_errno(context, srv->cli);
309 cli_shutdown(srv->cli);
310 srv->cli = NULL;
311 smbc_getFunctionRemoveCachedServer(context)(context,
312 srv);
313 srv = NULL;
316 /* Determine if this share supports case sensitivity */
317 if (is_ipc) {
318 DEBUG(4,
319 ("IPC$ so ignore case sensitivity\n"));
320 } else if (!cli_get_fs_attr_info(c, &fs_attrs)) {
321 DEBUG(4, ("Could not retrieve "
322 "case sensitivity flag: %s.\n",
323 cli_errstr(c)));
326 * We can't determine the case sensitivity of
327 * the share. We have no choice but to use the
328 * user-specified case sensitivity setting.
330 if (smbc_getOptionCaseSensitive(context)) {
331 cli_set_case_sensitive(c, True);
332 } else {
333 cli_set_case_sensitive(c, False);
335 } else {
336 DEBUG(4,
337 ("Case sensitive: %s\n",
338 (fs_attrs & FILE_CASE_SENSITIVE_SEARCH
339 ? "True"
340 : "False")));
341 cli_set_case_sensitive(
343 (fs_attrs & FILE_CASE_SENSITIVE_SEARCH
344 ? True
345 : False));
349 * Regenerate the dev value since it's based on both
350 * server and share
352 if (srv) {
353 srv->dev = (dev_t)(str_checksum(server) ^
354 str_checksum(share));
359 /* If we have a connection... */
360 if (srv) {
362 /* ... then we're done here. Give 'em what they came for. */
363 goto done;
366 /* If we're not asked to connect when a connection doesn't exist... */
367 if (! connect_if_not_found) {
368 /* ... then we're done here. */
369 return NULL;
372 if (!*pp_workgroup || !*pp_username || !*pp_password) {
373 errno = ENOMEM;
374 return NULL;
377 make_nmb_name(&calling, smbc_getNetbiosName(context), 0x0);
378 make_nmb_name(&called , server, 0x20);
380 DEBUG(4,("SMBC_server: server_n=[%s] server=[%s]\n", server_n, server));
382 DEBUG(4,(" -> server_n=[%s] server=[%s]\n", server_n, server));
384 again:
386 zero_sockaddr(&ss);
388 /* have to open a new connection */
389 if ((c = cli_initialise()) == NULL) {
390 errno = ENOMEM;
391 return NULL;
394 if (smbc_getOptionUseKerberos(context)) {
395 c->use_kerberos = True;
398 if (smbc_getOptionFallbackAfterKerberos(context)) {
399 c->fallback_after_kerberos = True;
402 c->timeout = smbc_getTimeout(context);
405 * Force use of port 139 for first try if share is $IPC, empty, or
406 * null, so browse lists can work
408 if (share == NULL || *share == '\0' || is_ipc) {
409 port_try_first = 139;
410 port_try_next = 445;
411 } else {
412 port_try_first = 445;
413 port_try_next = 139;
416 c->port = port_try_first;
418 status = cli_connect(c, server_n, &ss);
419 if (!NT_STATUS_IS_OK(status)) {
421 /* First connection attempt failed. Try alternate port. */
422 c->port = port_try_next;
424 status = cli_connect(c, server_n, &ss);
425 if (!NT_STATUS_IS_OK(status)) {
426 cli_shutdown(c);
427 errno = ETIMEDOUT;
428 return NULL;
432 if (!cli_session_request(c, &calling, &called)) {
433 cli_shutdown(c);
434 if (strcmp(called.name, "*SMBSERVER")) {
435 make_nmb_name(&called , "*SMBSERVER", 0x20);
436 goto again;
437 } else { /* Try one more time, but ensure we don't loop */
439 /* Only try this if server is an IP address ... */
441 if (is_ipaddress(server) && !tried_reverse) {
442 fstring remote_name;
443 struct sockaddr_storage rem_ss;
445 if (!interpret_string_addr(&rem_ss, server,
446 NI_NUMERICHOST)) {
447 DEBUG(4, ("Could not convert IP address "
448 "%s to struct sockaddr_storage\n",
449 server));
450 errno = ETIMEDOUT;
451 return NULL;
454 tried_reverse++; /* Yuck */
456 if (name_status_find("*", 0, 0,
457 &rem_ss, remote_name)) {
458 make_nmb_name(&called,
459 remote_name,
460 0x20);
461 goto again;
465 errno = ETIMEDOUT;
466 return NULL;
469 DEBUG(4,(" session request ok\n"));
471 if (!cli_negprot(c)) {
472 cli_shutdown(c);
473 errno = ETIMEDOUT;
474 return NULL;
477 username_used = *pp_username;
479 if (!NT_STATUS_IS_OK(cli_session_setup(c, username_used,
480 *pp_password,
481 strlen(*pp_password),
482 *pp_password,
483 strlen(*pp_password),
484 *pp_workgroup))) {
486 /* Failed. Try an anonymous login, if allowed by flags. */
487 username_used = "";
489 if (smbc_getOptionNoAutoAnonymousLogin(context) ||
490 !NT_STATUS_IS_OK(cli_session_setup(c, username_used,
491 *pp_password, 1,
492 *pp_password, 0,
493 *pp_workgroup))) {
495 cli_shutdown(c);
496 errno = EPERM;
497 return NULL;
501 DEBUG(4,(" session setup ok\n"));
503 if (!cli_send_tconX(c, share, "?????",
504 *pp_password, strlen(*pp_password)+1)) {
505 errno = SMBC_errno(context, c);
506 cli_shutdown(c);
507 return NULL;
510 DEBUG(4,(" tconx ok\n"));
512 /* Determine if this share supports case sensitivity */
513 if (is_ipc) {
514 DEBUG(4, ("IPC$ so ignore case sensitivity\n"));
515 } else if (!cli_get_fs_attr_info(c, &fs_attrs)) {
516 DEBUG(4, ("Could not retrieve case sensitivity flag: %s.\n",
517 cli_errstr(c)));
520 * We can't determine the case sensitivity of the share. We
521 * have no choice but to use the user-specified case
522 * sensitivity setting.
524 if (smbc_getOptionCaseSensitive(context)) {
525 cli_set_case_sensitive(c, True);
526 } else {
527 cli_set_case_sensitive(c, False);
529 } else {
530 DEBUG(4, ("Case sensitive: %s\n",
531 (fs_attrs & FILE_CASE_SENSITIVE_SEARCH
532 ? "True"
533 : "False")));
534 cli_set_case_sensitive(c,
535 (fs_attrs & FILE_CASE_SENSITIVE_SEARCH
536 ? True
537 : False));
540 if (context->internal->smb_encryption_level) {
541 /* Attempt UNIX smb encryption. */
542 if (!NT_STATUS_IS_OK(cli_force_encryption(c,
543 username_used,
544 *pp_password,
545 *pp_workgroup))) {
548 * context->smb_encryption_level == 1
549 * means don't fail if encryption can't be negotiated,
550 * == 2 means fail if encryption can't be negotiated.
553 DEBUG(4,(" SMB encrypt failed\n"));
555 if (context->internal->smb_encryption_level == 2) {
556 cli_shutdown(c);
557 errno = EPERM;
558 return NULL;
561 DEBUG(4,(" SMB encrypt ok\n"));
565 * Ok, we have got a nice connection
566 * Let's allocate a server structure.
569 srv = SMB_MALLOC_P(SMBCSRV);
570 if (!srv) {
571 errno = ENOMEM;
572 goto failed;
575 ZERO_STRUCTP(srv);
576 srv->cli = c;
577 srv->dev = (dev_t)(str_checksum(server) ^ str_checksum(share));
578 srv->no_pathinfo = False;
579 srv->no_pathinfo2 = False;
580 srv->no_nt_session = False;
582 /* now add it to the cache (internal or external) */
583 /* Let the cache function set errno if it wants to */
584 errno = 0;
585 if (smbc_getFunctionAddCachedServer(context)(context, srv,
586 server, share,
587 *pp_workgroup,
588 *pp_username)) {
589 int saved_errno = errno;
590 DEBUG(3, (" Failed to add server to cache\n"));
591 errno = saved_errno;
592 if (errno == 0) {
593 errno = ENOMEM;
595 goto failed;
598 DEBUG(2, ("Server connect ok: //%s/%s: %p\n",
599 server, share, srv));
601 DLIST_ADD(context->internal->servers, srv);
602 done:
603 if (!pp_workgroup || !*pp_workgroup || !**pp_workgroup) {
604 workgroup = talloc_strdup(ctx, smbc_getWorkgroup(context));
605 } else {
606 workgroup = *pp_workgroup;
608 if(!workgroup) {
609 return NULL;
612 /* set the credentials to make DFS work */
613 smbc_set_credentials_with_fallback(context,
614 workgroup,
615 *pp_username,
616 *pp_password);
618 return srv;
620 failed:
621 cli_shutdown(c);
622 if (!srv) {
623 return NULL;
626 SAFE_FREE(srv);
627 return NULL;
631 * Connect to a server for getting/setting attributes, possibly on an existing
632 * connection. This works similarly to SMBC_server().
634 SMBCSRV *
635 SMBC_attr_server(TALLOC_CTX *ctx,
636 SMBCCTX *context,
637 const char *server,
638 const char *share,
639 char **pp_workgroup,
640 char **pp_username,
641 char **pp_password)
643 int flags;
644 struct sockaddr_storage ss;
645 struct cli_state *ipc_cli;
646 struct rpc_pipe_client *pipe_hnd;
647 NTSTATUS nt_status;
648 SMBCSRV *ipc_srv=NULL;
651 * See if we've already created this special connection. Reference
652 * our "special" share name '*IPC$', which is an impossible real share
653 * name due to the leading asterisk.
655 ipc_srv = SMBC_find_server(ctx, context, server, "*IPC$",
656 pp_workgroup, pp_username, pp_password);
657 if (!ipc_srv) {
659 /* We didn't find a cached connection. Get the password */
660 if (!*pp_password || (*pp_password)[0] == '\0') {
661 /* ... then retrieve it now. */
662 SMBC_call_auth_fn(ctx, context, server, share,
663 pp_workgroup,
664 pp_username,
665 pp_password);
666 if (!*pp_workgroup || !*pp_username || !*pp_password) {
667 errno = ENOMEM;
668 return NULL;
672 flags = 0;
673 if (smbc_getOptionUseKerberos(context)) {
674 flags |= CLI_FULL_CONNECTION_USE_KERBEROS;
677 zero_sockaddr(&ss);
678 nt_status = cli_full_connection(&ipc_cli,
679 global_myname(), server,
680 &ss, 0, "IPC$", "?????",
681 *pp_username,
682 *pp_workgroup,
683 *pp_password,
684 flags,
685 Undefined, NULL);
686 if (! NT_STATUS_IS_OK(nt_status)) {
687 DEBUG(1,("cli_full_connection failed! (%s)\n",
688 nt_errstr(nt_status)));
689 errno = ENOTSUP;
690 return NULL;
693 if (context->internal->smb_encryption_level) {
694 /* Attempt UNIX smb encryption. */
695 if (!NT_STATUS_IS_OK(cli_force_encryption(ipc_cli,
696 *pp_username,
697 *pp_password,
698 *pp_workgroup))) {
701 * context->smb_encryption_level ==
702 * 1 means don't fail if encryption can't be
703 * negotiated, == 2 means fail if encryption
704 * can't be negotiated.
707 DEBUG(4,(" SMB encrypt failed on IPC$\n"));
709 if (context->internal->smb_encryption_level == 2) {
710 cli_shutdown(ipc_cli);
711 errno = EPERM;
712 return NULL;
715 DEBUG(4,(" SMB encrypt ok on IPC$\n"));
718 ipc_srv = SMB_MALLOC_P(SMBCSRV);
719 if (!ipc_srv) {
720 errno = ENOMEM;
721 cli_shutdown(ipc_cli);
722 return NULL;
725 ZERO_STRUCTP(ipc_srv);
726 ipc_srv->cli = ipc_cli;
728 nt_status = cli_rpc_pipe_open_noauth(
729 ipc_srv->cli, &ndr_table_lsarpc.syntax_id, &pipe_hnd);
730 if (!NT_STATUS_IS_OK(nt_status)) {
731 DEBUG(1, ("cli_nt_session_open fail!\n"));
732 errno = ENOTSUP;
733 cli_shutdown(ipc_srv->cli);
734 free(ipc_srv);
735 return NULL;
739 * Some systems don't support
740 * SEC_RIGHTS_MAXIMUM_ALLOWED, but NT sends 0x2000000
741 * so we might as well do it too.
744 nt_status = rpccli_lsa_open_policy(
745 pipe_hnd,
746 talloc_tos(),
747 True,
748 GENERIC_EXECUTE_ACCESS,
749 &ipc_srv->pol);
751 if (!NT_STATUS_IS_OK(nt_status)) {
752 errno = SMBC_errno(context, ipc_srv->cli);
753 cli_shutdown(ipc_srv->cli);
754 return NULL;
757 /* now add it to the cache (internal or external) */
759 errno = 0; /* let cache function set errno if it likes */
760 if (smbc_getFunctionAddCachedServer(context)(context, ipc_srv,
761 server,
762 "*IPC$",
763 *pp_workgroup,
764 *pp_username)) {
765 DEBUG(3, (" Failed to add server to cache\n"));
766 if (errno == 0) {
767 errno = ENOMEM;
769 cli_shutdown(ipc_srv->cli);
770 free(ipc_srv);
771 return NULL;
774 DLIST_ADD(context->internal->servers, ipc_srv);
777 return ipc_srv;