ctdbd_conn: Only poll if there's a timeout
[Samba.git] / source3 / lib / substitute.c
blobce4fbbab73d4d915f5816eca4e8d815e7327289b
1 /*
2 Unix SMB/CIFS implementation.
3 string substitution functions
4 Copyright (C) Andrew Tridgell 1992-2000
5 Copyright (C) Gerald Carter 2006
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 "system/passwd.h"
24 #include "secrets.h"
25 #include "auth.h"
27 userdom_struct current_user_info;
28 fstring remote_proto="UNKNOWN";
30 /**
31 * Set the 'local' machine name
32 * @param local_name the name we are being called
33 * @param if this is the 'final' name for us, not be be changed again
36 static char *local_machine;
38 void free_local_machine_name(void)
40 TALLOC_FREE(local_machine);
43 bool set_local_machine_name(const char *local_name, bool perm)
45 static bool already_perm = false;
46 char *tmp_local_machine = NULL;
47 size_t len;
49 if (already_perm) {
50 return true;
53 tmp_local_machine = talloc_strdup(NULL, local_name);
54 if (!tmp_local_machine) {
55 return false;
57 trim_char(tmp_local_machine,' ',' ');
59 TALLOC_FREE(local_machine);
60 len = strlen(tmp_local_machine);
61 local_machine = (char *)TALLOC_ZERO(NULL, len+1);
62 if (!local_machine) {
63 TALLOC_FREE(tmp_local_machine);
64 return false;
66 /* alpha_strcpy includes the space for the terminating nul. */
67 alpha_strcpy(local_machine,tmp_local_machine,
68 SAFE_NETBIOS_CHARS,len+1);
69 if (!strlower_m(local_machine)) {
70 TALLOC_FREE(tmp_local_machine);
71 return false;
73 TALLOC_FREE(tmp_local_machine);
75 already_perm = perm;
77 return true;
80 const char *get_local_machine_name(void)
82 if (!local_machine || !*local_machine) {
83 return lp_netbios_name();
86 return local_machine;
89 /**
90 * Set the 'remote' machine name
91 * @param remote_name the name our client wants to be called by
92 * @param if this is the 'final' name for them, not be be changed again
95 static char *remote_machine;
97 bool set_remote_machine_name(const char *remote_name, bool perm)
99 static bool already_perm = False;
100 char *tmp_remote_machine;
101 size_t len;
103 if (already_perm) {
104 return true;
107 tmp_remote_machine = talloc_strdup(NULL, remote_name);
108 if (!tmp_remote_machine) {
109 return false;
111 trim_char(tmp_remote_machine,' ',' ');
113 TALLOC_FREE(remote_machine);
114 len = strlen(tmp_remote_machine);
115 remote_machine = (char *)TALLOC_ZERO(NULL, len+1);
116 if (!remote_machine) {
117 TALLOC_FREE(tmp_remote_machine);
118 return false;
121 /* alpha_strcpy includes the space for the terminating nul. */
122 alpha_strcpy(remote_machine,tmp_remote_machine,
123 SAFE_NETBIOS_CHARS,len+1);
124 if (!strlower_m(remote_machine)) {
125 TALLOC_FREE(tmp_remote_machine);
126 return false;
128 TALLOC_FREE(tmp_remote_machine);
130 already_perm = perm;
132 return true;
135 const char *get_remote_machine_name(void)
137 return remote_machine ? remote_machine : "";
140 /*******************************************************************
141 Setup the string used by %U substitution.
142 ********************************************************************/
144 static char *smb_user_name;
146 void sub_set_smb_name(const char *name)
148 char *tmp;
149 size_t len;
150 bool is_machine_account = false;
152 /* don't let anonymous logins override the name */
153 if (!name || !*name) {
154 return;
157 tmp = talloc_strdup(NULL, name);
158 if (!tmp) {
159 return;
161 trim_char(tmp, ' ', ' ');
162 if (!strlower_m(tmp)) {
163 TALLOC_FREE(tmp);
164 return;
167 len = strlen(tmp);
169 if (len == 0) {
170 TALLOC_FREE(tmp);
171 return;
174 /* long story but here goes....we have to allow usernames
175 ending in '$' as they are valid machine account names.
176 So check for a machine account and re-add the '$'
177 at the end after the call to alpha_strcpy(). --jerry */
179 if (tmp[len-1] == '$') {
180 is_machine_account = True;
183 TALLOC_FREE(smb_user_name);
184 smb_user_name = (char *)TALLOC_ZERO(NULL, len+1);
185 if (!smb_user_name) {
186 TALLOC_FREE(tmp);
187 return;
190 /* alpha_strcpy includes the space for the terminating nul. */
191 alpha_strcpy(smb_user_name, tmp,
192 SAFE_NETBIOS_CHARS,
193 len+1);
195 TALLOC_FREE(tmp);
197 if (is_machine_account) {
198 len = strlen(smb_user_name);
199 smb_user_name[len-1] = '$';
203 static char sub_peeraddr[INET6_ADDRSTRLEN];
204 static const char *sub_peername = NULL;
205 static char sub_sockaddr[INET6_ADDRSTRLEN];
207 void sub_set_socket_ids(const char *peeraddr, const char *peername,
208 const char *sockaddr)
210 const char *addr = peeraddr;
212 if (strnequal(addr, "::ffff:", 7)) {
213 addr += 7;
215 strlcpy(sub_peeraddr, addr, sizeof(sub_peeraddr));
217 if (sub_peername != NULL &&
218 sub_peername != sub_peeraddr) {
219 talloc_free(discard_const_p(char,sub_peername));
220 sub_peername = NULL;
222 sub_peername = talloc_strdup(NULL, peername);
223 if (sub_peername == NULL) {
224 sub_peername = sub_peeraddr;
228 * Shouldn't we do the ::ffff: cancellation here as well? The
229 * original code in talloc_sub_basic() did not do it, so I'm
230 * leaving it out here as well for compatibility.
232 strlcpy(sub_sockaddr, sockaddr, sizeof(sub_sockaddr));
235 static const char *get_smb_user_name(void)
237 return smb_user_name ? smb_user_name : "";
240 /*******************************************************************
241 Setup the strings used by substitutions. Called per packet. Ensure
242 %U name is set correctly also.
244 smb_name must be sanitized by alpha_strcpy
245 ********************************************************************/
247 void set_current_user_info(const char *smb_name, const char *unix_name,
248 const char *domain)
250 fstrcpy(current_user_info.smb_name, smb_name);
251 fstrcpy(current_user_info.unix_name, unix_name);
252 fstrcpy(current_user_info.domain, domain);
254 /* The following is safe as current_user_info.smb_name
255 * has already been sanitised in register_existing_vuid. */
257 sub_set_smb_name(current_user_info.smb_name);
260 /*******************************************************************
261 Return the current active user name.
262 *******************************************************************/
264 const char *get_current_username(void)
266 if (current_user_info.smb_name[0] == '\0' ) {
267 return get_smb_user_name();
270 return current_user_info.smb_name;
273 /*******************************************************************
274 Given a pointer to a %$(NAME) in p and the whole string in str
275 expand it as an environment variable.
276 str must be a talloced string.
277 Return a new allocated and expanded string.
278 Based on code by Branko Cibej <branko.cibej@hermes.si>
279 When this is called p points at the '%' character.
280 May substitute multiple occurrencies of the same env var.
281 ********************************************************************/
283 static char *realloc_expand_env_var(char *str, char *p)
285 char *envname;
286 char *envval;
287 char *q, *r;
288 int copylen;
290 if (p[0] != '%' || p[1] != '$' || p[2] != '(') {
291 return str;
295 * Look for the terminating ')'.
298 if ((q = strchr_m(p,')')) == NULL) {
299 DEBUG(0,("expand_env_var: Unterminated environment variable [%s]\n", p));
300 return str;
304 * Extract the name from within the %$(NAME) string.
307 r = p + 3;
308 copylen = q - r;
310 /* reserve space for use later add %$() chars */
311 if ( (envname = talloc_array(talloc_tos(), char, copylen + 1 + 4)) == NULL ) {
312 return NULL;
315 strncpy(envname,r,copylen);
316 envname[copylen] = '\0';
318 if ((envval = getenv(envname)) == NULL) {
319 DEBUG(0,("expand_env_var: Environment variable [%s] not set\n", envname));
320 TALLOC_FREE(envname);
321 return str;
325 * Copy the full %$(NAME) into envname so it
326 * can be replaced.
329 copylen = q + 1 - p;
330 strncpy(envname,p,copylen);
331 envname[copylen] = '\0';
332 r = realloc_string_sub(str, envname, envval);
333 TALLOC_FREE(envname);
335 return r;
338 /*******************************************************************
339 Patch from jkf@soton.ac.uk
340 Added this to implement %p (NIS auto-map version of %H)
341 *******************************************************************/
343 static const char *automount_path(const char *user_name)
345 TALLOC_CTX *ctx = talloc_tos();
346 const char *server_path;
348 /* use the passwd entry as the default */
349 /* this will be the default if WITH_AUTOMOUNT is not used or fails */
351 server_path = talloc_strdup(ctx, get_user_home_dir(ctx, user_name));
352 if (!server_path) {
353 return "";
356 #if (defined(HAVE_NETGROUP) && defined (WITH_AUTOMOUNT))
358 if (lp_nis_homedir()) {
359 const char *home_path_start;
360 char *automount_value = automount_lookup(ctx, user_name);
362 if(automount_value && strlen(automount_value) > 0) {
363 home_path_start = strchr_m(automount_value,':');
364 if (home_path_start != NULL) {
365 DEBUG(5, ("NIS lookup succeeded. "
366 "Home path is: %s\n",
367 home_path_start ?
368 (home_path_start+1):""));
369 server_path = talloc_strdup(ctx,
370 home_path_start+1);
371 if (!server_path) {
372 server_path = "";
375 } else {
376 /* NIS key lookup failed: default to
377 * user home directory from password file */
378 DEBUG(5, ("NIS lookup failed. Using Home path from "
379 "passwd file. Home path is: %s\n", server_path ));
382 #endif
384 DEBUG(4,("Home server path: %s\n", server_path));
385 return server_path;
388 /*******************************************************************
389 Patch from jkf@soton.ac.uk
390 This is Luke's original function with the NIS lookup code
391 moved out to a separate function.
392 *******************************************************************/
394 static const char *automount_server(const char *user_name)
396 TALLOC_CTX *ctx = talloc_tos();
397 const char *server_name;
398 const char *local_machine_name = get_local_machine_name();
400 /* use the local machine name as the default */
401 /* this will be the default if WITH_AUTOMOUNT is not used or fails */
402 if (local_machine_name && *local_machine_name) {
403 server_name = talloc_strdup(ctx, local_machine_name);
404 } else {
405 server_name = talloc_strdup(ctx, lp_netbios_name());
408 if (!server_name) {
409 return "";
412 #if (defined(HAVE_NETGROUP) && defined (WITH_AUTOMOUNT))
413 if (lp_nis_homedir()) {
414 char *p;
415 char *srv;
416 char *automount_value = automount_lookup(ctx, user_name);
417 if (!automount_value) {
418 return "";
420 srv = talloc_strdup(ctx, automount_value);
421 if (!srv) {
422 return "";
424 p = strchr_m(srv, ':');
425 if (!p) {
426 return "";
428 *p = '\0';
429 server_name = srv;
430 DEBUG(5, ("NIS lookup succeeded. Home server %s\n",
431 server_name));
433 #endif
435 DEBUG(4,("Home server: %s\n", server_name));
436 return server_name;
439 /****************************************************************************
440 Do some standard substitutions in a string.
441 len is the length in bytes of the space allowed in string str. If zero means
442 don't allow expansions.
443 ****************************************************************************/
445 void standard_sub_basic(const char *smb_name, const char *domain_name,
446 char *str, size_t len)
448 char *s;
450 if ( (s = talloc_sub_basic(talloc_tos(), smb_name, domain_name, str )) != NULL ) {
451 strncpy( str, s, len );
454 TALLOC_FREE( s );
457 /****************************************************************************
458 Do some standard substitutions in a string.
459 This function will return an talloced string that has to be freed.
460 ****************************************************************************/
462 char *talloc_sub_basic(TALLOC_CTX *mem_ctx,
463 const char *smb_name,
464 const char *domain_name,
465 const char *str)
467 char *b, *p, *s, *r, *a_string;
468 fstring pidstr, vnnstr;
469 const char *local_machine_name = get_local_machine_name();
470 TALLOC_CTX *tmp_ctx = NULL;
472 /* workaround to prevent a crash while looking at bug #687 */
474 if (!str) {
475 DEBUG(0,("talloc_sub_basic: NULL source string! This should not happen\n"));
476 return NULL;
479 a_string = talloc_strdup(mem_ctx, str);
480 if (a_string == NULL) {
481 DEBUG(0, ("talloc_sub_basic: Out of memory!\n"));
482 return NULL;
485 tmp_ctx = talloc_stackframe();
487 for (b = s = a_string; (p = strchr_m(s, '%')); s = a_string + (p - b)) {
489 r = NULL;
490 b = a_string;
492 switch (*(p+1)) {
493 case 'U' :
494 r = strlower_talloc(tmp_ctx, smb_name);
495 if (r == NULL) {
496 goto error;
498 a_string = realloc_string_sub(a_string, "%U", r);
499 break;
500 case 'G' : {
501 struct passwd *pass;
503 if (domain_name != NULL && domain_name[0] != '\0') {
504 r = talloc_asprintf(tmp_ctx,
505 "%s%c%s",
506 domain_name,
507 *lp_winbind_separator(),
508 smb_name);
509 } else {
510 r = talloc_strdup(tmp_ctx, smb_name);
512 if (r == NULL) {
513 goto error;
516 pass = Get_Pwnam_alloc(tmp_ctx, r);
517 if (pass != NULL) {
518 a_string = realloc_string_sub(
519 a_string, "%G",
520 gidtoname(pass->pw_gid));
522 TALLOC_FREE(pass);
523 break;
525 case 'D' :
526 r = strupper_talloc(tmp_ctx, domain_name);
527 if (r == NULL) {
528 goto error;
530 a_string = realloc_string_sub(a_string, "%D", r);
531 break;
532 case 'I' : {
533 a_string = realloc_string_sub(
534 a_string, "%I",
535 sub_peeraddr[0] ? sub_peeraddr : "0.0.0.0");
536 break;
538 case 'i':
539 a_string = realloc_string_sub(
540 a_string, "%i",
541 sub_sockaddr[0] ? sub_sockaddr : "0.0.0.0");
542 break;
543 case 'L' :
544 if ( strncasecmp_m(p, "%LOGONSERVER%", strlen("%LOGONSERVER%")) == 0 ) {
545 break;
547 if (local_machine_name && *local_machine_name) {
548 a_string = realloc_string_sub(a_string, "%L", local_machine_name);
549 } else {
550 a_string = realloc_string_sub(a_string, "%L", lp_netbios_name());
552 break;
553 case 'N':
554 a_string = realloc_string_sub(a_string, "%N", automount_server(smb_name));
555 break;
556 case 'M' :
557 a_string = realloc_string_sub(a_string, "%M",
558 sub_peername ? sub_peername : "");
559 break;
560 case 'R' :
561 a_string = realloc_string_sub(a_string, "%R", remote_proto);
562 break;
563 case 'T' :
564 a_string = realloc_string_sub(a_string, "%T", current_timestring(tmp_ctx, False));
565 break;
566 case 'a' :
567 a_string = realloc_string_sub(a_string, "%a",
568 get_remote_arch_str());
569 break;
570 case 'd' :
571 slprintf(pidstr,sizeof(pidstr)-1, "%d",(int)getpid());
572 a_string = realloc_string_sub(a_string, "%d", pidstr);
573 break;
574 case 'h' :
575 a_string = realloc_string_sub(a_string, "%h", myhostname());
576 break;
577 case 'm' :
578 a_string = realloc_string_sub(a_string, "%m",
579 remote_machine
580 ? remote_machine
581 : "");
582 break;
583 case 'v' :
584 a_string = realloc_string_sub(a_string, "%v", samba_version_string());
585 break;
586 case 'w' :
587 a_string = realloc_string_sub(a_string, "%w", lp_winbind_separator());
588 break;
589 case '$' :
590 a_string = realloc_expand_env_var(a_string, p); /* Expand environment variables */
591 break;
592 case 'V' :
593 slprintf(vnnstr,sizeof(vnnstr)-1, "%u", get_my_vnn());
594 a_string = realloc_string_sub(a_string, "%V", vnnstr);
595 break;
596 default:
597 break;
600 p++;
601 TALLOC_FREE(r);
603 if (a_string == NULL) {
604 goto done;
608 goto done;
610 error:
611 TALLOC_FREE(a_string);
613 done:
614 TALLOC_FREE(tmp_ctx);
615 return a_string;
618 /****************************************************************************
619 Do some specific substitutions in a string.
620 This function will return an allocated string that have to be freed.
621 ****************************************************************************/
623 char *talloc_sub_specified(TALLOC_CTX *mem_ctx,
624 const char *input_string,
625 const char *username,
626 const char *grpname,
627 const char *domain,
628 uid_t uid,
629 gid_t gid)
631 char *a_string;
632 char *ret_string = NULL;
633 char *b, *p, *s;
634 TALLOC_CTX *tmp_ctx;
636 if (!(tmp_ctx = talloc_new(mem_ctx))) {
637 DEBUG(0, ("talloc_new failed\n"));
638 return NULL;
641 a_string = talloc_strdup(tmp_ctx, input_string);
642 if (a_string == NULL) {
643 DEBUG(0, ("talloc_sub_specified: Out of memory!\n"));
644 goto done;
647 for (b = s = a_string; (p = strchr_m(s, '%')); s = a_string + (p - b)) {
649 b = a_string;
651 switch (*(p+1)) {
652 case 'U' :
653 a_string = talloc_string_sub(
654 tmp_ctx, a_string, "%U", username);
655 break;
656 case 'u' :
657 a_string = talloc_string_sub(
658 tmp_ctx, a_string, "%u", username);
659 break;
660 case 'G' :
661 if (gid != -1) {
662 const char *name;
664 if (grpname != NULL) {
665 name = grpname;
666 } else {
667 name = gidtoname(gid);
670 a_string = talloc_string_sub(tmp_ctx,
671 a_string,
672 "%G",
673 name);
674 } else {
675 a_string = talloc_string_sub(
676 tmp_ctx, a_string,
677 "%G", "NO_GROUP");
679 break;
680 case 'g' :
681 if (gid != -1) {
682 const char *name;
684 if (grpname != NULL) {
685 name = grpname;
686 } else {
687 name = gidtoname(gid);
690 a_string = talloc_string_sub(tmp_ctx,
691 a_string,
692 "%g",
693 name);
694 } else {
695 a_string = talloc_string_sub(
696 tmp_ctx, a_string, "%g", "NO_GROUP");
698 break;
699 case 'D' :
700 a_string = talloc_string_sub(tmp_ctx, a_string,
701 "%D", domain);
702 break;
703 case 'N' :
704 a_string = talloc_string_sub(
705 tmp_ctx, a_string, "%N",
706 automount_server(username));
707 break;
708 default:
709 break;
712 p++;
713 if (a_string == NULL) {
714 goto done;
718 /* Watch out, using "mem_ctx" here, so all intermediate stuff goes
719 * away with the TALLOC_FREE(tmp_ctx) further down. */
721 ret_string = talloc_sub_basic(mem_ctx, username, domain, a_string);
723 done:
724 TALLOC_FREE(tmp_ctx);
725 return ret_string;
728 /****************************************************************************
729 ****************************************************************************/
731 char *talloc_sub_advanced(TALLOC_CTX *ctx,
732 const char *servicename,
733 const char *user,
734 const char *connectpath,
735 gid_t gid,
736 const char *smb_name,
737 const char *domain_name,
738 const char *str)
740 char *a_string, *ret_string;
741 char *b, *p, *s;
743 a_string = talloc_strdup(talloc_tos(), str);
744 if (a_string == NULL) {
745 DEBUG(0, ("talloc_sub_advanced: Out of memory!\n"));
746 return NULL;
749 for (b = s = a_string; (p = strchr_m(s, '%')); s = a_string + (p - b)) {
751 b = a_string;
753 switch (*(p+1)) {
754 case 'N' :
755 a_string = realloc_string_sub(a_string, "%N", automount_server(user));
756 break;
757 case 'H': {
758 char *h;
759 if ((h = get_user_home_dir(talloc_tos(), user)))
760 a_string = realloc_string_sub(a_string, "%H", h);
761 TALLOC_FREE(h);
762 break;
764 case 'P':
765 a_string = realloc_string_sub(a_string, "%P", connectpath);
766 break;
767 case 'S':
768 a_string = realloc_string_sub(a_string, "%S", servicename);
769 break;
770 case 'g':
771 a_string = realloc_string_sub(a_string, "%g", gidtoname(gid));
772 break;
773 case 'u':
774 a_string = realloc_string_sub(a_string, "%u", user);
775 break;
777 /* Patch from jkf@soton.ac.uk Left the %N (NIS
778 * server name) in standard_sub_basic as it is
779 * a feature for logon servers, hence uses the
780 * username. The %p (NIS server path) code is
781 * here as it is used instead of the default
782 * "path =" string in [homes] and so needs the
783 * service name, not the username. */
784 case 'p':
785 a_string = realloc_string_sub(a_string, "%p",
786 automount_path(servicename));
787 break;
789 default:
790 break;
793 p++;
794 if (a_string == NULL) {
795 return NULL;
799 ret_string = talloc_sub_basic(ctx, smb_name, domain_name, a_string);
800 TALLOC_FREE(a_string);
801 return ret_string;
804 void standard_sub_advanced(const char *servicename, const char *user,
805 const char *connectpath, gid_t gid,
806 const char *smb_name, const char *domain_name,
807 char *str, size_t len)
809 char *s = talloc_sub_advanced(talloc_tos(),
810 servicename, user, connectpath,
811 gid, smb_name, domain_name, str);
813 if (!s) {
814 return;
816 strlcpy( str, s, len );
817 TALLOC_FREE( s );
820 /******************************************************************************
821 version of standard_sub_basic() for string lists; uses talloc_sub_basic()
822 for the work
823 *****************************************************************************/
825 bool str_list_sub_basic( char **list, const char *smb_name,
826 const char *domain_name )
828 TALLOC_CTX *ctx = list;
829 char *s, *tmpstr;
831 while ( *list ) {
832 s = *list;
833 tmpstr = talloc_sub_basic(ctx, smb_name, domain_name, s);
834 if ( !tmpstr ) {
835 DEBUG(0,("str_list_sub_basic: "
836 "talloc_sub_basic() return NULL!\n"));
837 return false;
840 TALLOC_FREE(*list);
841 *list = tmpstr;
843 list++;
846 return true;