s3: Fix bug #9085.
[Samba.git] / source3 / lib / util.c
blobc7dfa287304d5ac1a836118be235b48f9647e904
1 /*
2 Unix SMB/CIFS implementation.
3 Samba utility functions
4 Copyright (C) Andrew Tridgell 1992-1998
5 Copyright (C) Jeremy Allison 2001-2007
6 Copyright (C) Simo Sorce 2001
7 Copyright (C) Jim McDonough <jmcd@us.ibm.com> 2003
8 Copyright (C) James Peach 2006
10 This program is free software; you can redistribute it and/or modify
11 it under the terms of the GNU General Public License as published by
12 the Free Software Foundation; either version 3 of the License, or
13 (at your option) any later version.
15 This program is distributed in the hope that it will be useful,
16 but WITHOUT ANY WARRANTY; without even the implied warranty of
17 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 GNU General Public License for more details.
20 You should have received a copy of the GNU General Public License
21 along with this program. If not, see <http://www.gnu.org/licenses/>.
24 #include "includes.h"
26 extern char *global_clobber_region_function;
27 extern unsigned int global_clobber_region_line;
29 /* Max allowable allococation - 256mb - 0x10000000 */
30 #define MAX_ALLOC_SIZE (1024*1024*256)
32 #if (defined(HAVE_NETGROUP) && defined (WITH_AUTOMOUNT))
33 #ifdef WITH_NISPLUS_HOME
34 #ifdef BROKEN_NISPLUS_INCLUDE_FILES
36 * The following lines are needed due to buggy include files
37 * in Solaris 2.6 which define GROUP in both /usr/include/sys/acl.h and
38 * also in /usr/include/rpcsvc/nis.h. The definitions conflict. JRA.
39 * Also GROUP_OBJ is defined as 0x4 in /usr/include/sys/acl.h and as
40 * an enum in /usr/include/rpcsvc/nis.h.
43 #if defined(GROUP)
44 #undef GROUP
45 #endif
47 #if defined(GROUP_OBJ)
48 #undef GROUP_OBJ
49 #endif
51 #endif /* BROKEN_NISPLUS_INCLUDE_FILES */
53 #include <rpcsvc/nis.h>
55 #endif /* WITH_NISPLUS_HOME */
56 #endif /* HAVE_NETGROUP && WITH_AUTOMOUNT */
58 static enum protocol_types Protocol = PROTOCOL_COREPLUS;
60 enum protocol_types get_Protocol(void)
62 return Protocol;
65 void set_Protocol(enum protocol_types p)
67 Protocol = p;
70 static enum remote_arch_types ra_type = RA_UNKNOWN;
72 /***********************************************************************
73 Definitions for all names.
74 ***********************************************************************/
76 static char *smb_myname;
77 static char *smb_myworkgroup;
78 static char *smb_scope;
79 static int smb_num_netbios_names;
80 static char **smb_my_netbios_names;
82 /***********************************************************************
83 Allocate and set myname. Ensure upper case.
84 ***********************************************************************/
86 bool set_global_myname(const char *myname)
88 SAFE_FREE(smb_myname);
89 smb_myname = SMB_STRDUP(myname);
90 if (!smb_myname)
91 return False;
92 strupper_m(smb_myname);
93 return True;
96 const char *global_myname(void)
98 return smb_myname;
101 /***********************************************************************
102 Allocate and set myworkgroup. Ensure upper case.
103 ***********************************************************************/
105 bool set_global_myworkgroup(const char *myworkgroup)
107 SAFE_FREE(smb_myworkgroup);
108 smb_myworkgroup = SMB_STRDUP(myworkgroup);
109 if (!smb_myworkgroup)
110 return False;
111 strupper_m(smb_myworkgroup);
112 return True;
115 const char *lp_workgroup(void)
117 return smb_myworkgroup;
120 /***********************************************************************
121 Allocate and set scope. Ensure upper case.
122 ***********************************************************************/
124 bool set_global_scope(const char *scope)
126 SAFE_FREE(smb_scope);
127 smb_scope = SMB_STRDUP(scope);
128 if (!smb_scope)
129 return False;
130 strupper_m(smb_scope);
131 return True;
134 /*********************************************************************
135 Ensure scope is never null string.
136 *********************************************************************/
138 const char *global_scope(void)
140 if (!smb_scope)
141 set_global_scope("");
142 return smb_scope;
145 static void free_netbios_names_array(void)
147 int i;
149 for (i = 0; i < smb_num_netbios_names; i++)
150 SAFE_FREE(smb_my_netbios_names[i]);
152 SAFE_FREE(smb_my_netbios_names);
153 smb_num_netbios_names = 0;
156 static bool allocate_my_netbios_names_array(size_t number)
158 free_netbios_names_array();
160 smb_num_netbios_names = number + 1;
161 smb_my_netbios_names = SMB_MALLOC_ARRAY( char *, smb_num_netbios_names );
163 if (!smb_my_netbios_names)
164 return False;
166 memset(smb_my_netbios_names, '\0', sizeof(char *) * smb_num_netbios_names);
167 return True;
170 static bool set_my_netbios_names(const char *name, int i)
172 SAFE_FREE(smb_my_netbios_names[i]);
174 smb_my_netbios_names[i] = SMB_STRDUP(name);
175 if (!smb_my_netbios_names[i])
176 return False;
177 strupper_m(smb_my_netbios_names[i]);
178 return True;
181 /***********************************************************************
182 Free memory allocated to global objects
183 ***********************************************************************/
185 void gfree_names(void)
187 SAFE_FREE( smb_myname );
188 SAFE_FREE( smb_myworkgroup );
189 SAFE_FREE( smb_scope );
190 free_netbios_names_array();
191 free_local_machine_name();
194 void gfree_all( void )
196 gfree_names();
197 gfree_loadparm();
198 gfree_case_tables();
199 gfree_charcnv();
200 gfree_interfaces();
201 gfree_debugsyms();
204 const char *my_netbios_names(int i)
206 return smb_my_netbios_names[i];
209 bool set_netbios_aliases(const char **str_array)
211 size_t namecount;
213 /* Work out the max number of netbios aliases that we have */
214 for( namecount=0; str_array && (str_array[namecount] != NULL); namecount++ )
217 if ( global_myname() && *global_myname())
218 namecount++;
220 /* Allocate space for the netbios aliases */
221 if (!allocate_my_netbios_names_array(namecount))
222 return False;
224 /* Use the global_myname string first */
225 namecount=0;
226 if ( global_myname() && *global_myname()) {
227 set_my_netbios_names( global_myname(), namecount );
228 namecount++;
231 if (str_array) {
232 size_t i;
233 for ( i = 0; str_array[i] != NULL; i++) {
234 size_t n;
235 bool duplicate = False;
237 /* Look for duplicates */
238 for( n=0; n<namecount; n++ ) {
239 if( strequal( str_array[i], my_netbios_names(n) ) ) {
240 duplicate = True;
241 break;
244 if (!duplicate) {
245 if (!set_my_netbios_names(str_array[i], namecount))
246 return False;
247 namecount++;
251 return True;
254 /****************************************************************************
255 Common name initialization code.
256 ****************************************************************************/
258 bool init_names(void)
260 int n;
262 if (global_myname() == NULL || *global_myname() == '\0') {
263 if (!set_global_myname(myhostname())) {
264 DEBUG( 0, ( "init_structs: malloc fail.\n" ) );
265 return False;
269 if (!set_netbios_aliases(lp_netbios_aliases())) {
270 DEBUG( 0, ( "init_structs: malloc fail.\n" ) );
271 return False;
274 set_local_machine_name(global_myname(),false);
276 DEBUG( 5, ("Netbios name list:-\n") );
277 for( n=0; my_netbios_names(n); n++ ) {
278 DEBUGADD( 5, ("my_netbios_names[%d]=\"%s\"\n",
279 n, my_netbios_names(n) ) );
282 return( True );
285 /**************************************************************************n
286 Code to cope with username/password auth options from the commandline.
287 Used mainly in client tools.
288 ****************************************************************************/
290 struct user_auth_info *user_auth_info_init(TALLOC_CTX *mem_ctx)
292 struct user_auth_info *result;
294 result = TALLOC_ZERO_P(mem_ctx, struct user_auth_info);
295 if (result == NULL) {
296 return NULL;
299 result->signing_state = Undefined;
300 return result;
303 const char *get_cmdline_auth_info_username(const struct user_auth_info *auth_info)
305 if (!auth_info->username) {
306 return "";
308 return auth_info->username;
311 void set_cmdline_auth_info_username(struct user_auth_info *auth_info,
312 const char *username)
314 TALLOC_FREE(auth_info->username);
315 auth_info->username = talloc_strdup(auth_info, username);
316 if (!auth_info->username) {
317 exit(ENOMEM);
321 const char *get_cmdline_auth_info_domain(const struct user_auth_info *auth_info)
323 if (!auth_info->domain) {
324 return "";
326 return auth_info->domain;
329 void set_cmdline_auth_info_domain(struct user_auth_info *auth_info,
330 const char *domain)
332 TALLOC_FREE(auth_info->domain);
333 auth_info->domain = talloc_strdup(auth_info, domain);
334 if (!auth_info->domain) {
335 exit(ENOMEM);
339 const char *get_cmdline_auth_info_password(const struct user_auth_info *auth_info)
341 if (!auth_info->password) {
342 return "";
344 return auth_info->password;
347 void set_cmdline_auth_info_password(struct user_auth_info *auth_info,
348 const char *password)
350 TALLOC_FREE(auth_info->password);
351 if (password == NULL) {
352 password = "";
354 auth_info->password = talloc_strdup(auth_info, password);
355 if (!auth_info->password) {
356 exit(ENOMEM);
358 auth_info->got_pass = true;
361 bool set_cmdline_auth_info_signing_state(struct user_auth_info *auth_info,
362 const char *arg)
364 auth_info->signing_state = -1;
365 if (strequal(arg, "off") || strequal(arg, "no") ||
366 strequal(arg, "false")) {
367 auth_info->signing_state = false;
368 } else if (strequal(arg, "on") || strequal(arg, "yes") ||
369 strequal(arg, "true") || strequal(arg, "auto")) {
370 auth_info->signing_state = true;
371 } else if (strequal(arg, "force") || strequal(arg, "required") ||
372 strequal(arg, "forced")) {
373 auth_info->signing_state = Required;
374 } else {
375 return false;
377 return true;
380 int get_cmdline_auth_info_signing_state(const struct user_auth_info *auth_info)
382 return auth_info->signing_state;
385 void set_cmdline_auth_info_use_ccache(struct user_auth_info *auth_info, bool b)
387 auth_info->use_ccache = b;
390 bool get_cmdline_auth_info_use_ccache(const struct user_auth_info *auth_info)
392 return auth_info->use_ccache;
395 void set_cmdline_auth_info_use_kerberos(struct user_auth_info *auth_info,
396 bool b)
398 auth_info->use_kerberos = b;
401 bool get_cmdline_auth_info_use_kerberos(const struct user_auth_info *auth_info)
403 return auth_info->use_kerberos;
406 void set_cmdline_auth_info_fallback_after_kerberos(struct user_auth_info *auth_info,
407 bool b)
409 auth_info->fallback_after_kerberos = b;
412 bool get_cmdline_auth_info_fallback_after_kerberos(const struct user_auth_info *auth_info)
414 return auth_info->fallback_after_kerberos;
417 /* This should only be used by lib/popt_common.c JRA */
418 void set_cmdline_auth_info_use_krb5_ticket(struct user_auth_info *auth_info)
420 auth_info->use_kerberos = true;
421 auth_info->got_pass = true;
424 /* This should only be used by lib/popt_common.c JRA */
425 void set_cmdline_auth_info_smb_encrypt(struct user_auth_info *auth_info)
427 auth_info->smb_encrypt = true;
430 void set_cmdline_auth_info_use_machine_account(struct user_auth_info *auth_info)
432 auth_info->use_machine_account = true;
435 bool get_cmdline_auth_info_got_pass(const struct user_auth_info *auth_info)
437 return auth_info->got_pass;
440 bool get_cmdline_auth_info_smb_encrypt(const struct user_auth_info *auth_info)
442 return auth_info->smb_encrypt;
445 bool get_cmdline_auth_info_use_machine_account(const struct user_auth_info *auth_info)
447 return auth_info->use_machine_account;
450 struct user_auth_info *get_cmdline_auth_info_copy(TALLOC_CTX *mem_ctx,
451 const struct user_auth_info *src)
453 struct user_auth_info *result;
455 result = user_auth_info_init(mem_ctx);
456 if (result == NULL) {
457 return NULL;
460 *result = *src;
462 result->username = talloc_strdup(
463 result, get_cmdline_auth_info_username(src));
464 result->password = talloc_strdup(
465 result, get_cmdline_auth_info_password(src));
466 if ((result->username == NULL) || (result->password == NULL)) {
467 TALLOC_FREE(result);
468 return NULL;
471 return result;
474 bool set_cmdline_auth_info_machine_account_creds(struct user_auth_info *auth_info)
476 char *pass = NULL;
477 char *account = NULL;
479 if (!get_cmdline_auth_info_use_machine_account(auth_info)) {
480 return false;
483 if (!secrets_init()) {
484 d_printf("ERROR: Unable to open secrets database\n");
485 return false;
488 if (asprintf(&account, "%s$@%s", global_myname(), lp_realm()) < 0) {
489 return false;
492 pass = secrets_fetch_machine_password(lp_workgroup(), NULL, NULL);
493 if (!pass) {
494 d_printf("ERROR: Unable to fetch machine password for "
495 "%s in domain %s\n",
496 account, lp_workgroup());
497 SAFE_FREE(account);
498 return false;
501 set_cmdline_auth_info_username(auth_info, account);
502 set_cmdline_auth_info_password(auth_info, pass);
504 SAFE_FREE(account);
505 SAFE_FREE(pass);
507 return true;
510 /****************************************************************************
511 Ensure we have a password if one not given.
512 ****************************************************************************/
514 void set_cmdline_auth_info_getpass(struct user_auth_info *auth_info)
516 char *label = NULL;
517 char *pass;
518 TALLOC_CTX *frame;
520 if (get_cmdline_auth_info_got_pass(auth_info) ||
521 get_cmdline_auth_info_use_kerberos(auth_info)) {
522 /* Already got one... */
523 return;
526 frame = talloc_stackframe();
527 label = talloc_asprintf(frame, "Enter %s's password: ",
528 get_cmdline_auth_info_username(auth_info));
529 pass = getpass(label);
530 if (pass) {
531 set_cmdline_auth_info_password(auth_info, pass);
533 TALLOC_FREE(frame);
536 /*******************************************************************
537 Check if a file exists - call vfs_file_exist for samba files.
538 ********************************************************************/
540 bool file_exist_stat(const char *fname,SMB_STRUCT_STAT *sbuf,
541 bool fake_dir_create_times)
543 SMB_STRUCT_STAT st;
544 if (!sbuf)
545 sbuf = &st;
547 if (sys_stat(fname, sbuf, fake_dir_create_times) != 0)
548 return(False);
550 return((S_ISREG(sbuf->st_ex_mode)) || (S_ISFIFO(sbuf->st_ex_mode)));
553 /*******************************************************************
554 Check if a unix domain socket exists - call vfs_file_exist for samba files.
555 ********************************************************************/
557 bool socket_exist(const char *fname)
559 SMB_STRUCT_STAT st;
560 if (sys_stat(fname, &st, false) != 0)
561 return(False);
563 return S_ISSOCK(st.st_ex_mode);
566 /*******************************************************************
567 Returns the size in bytes of the named given the stat struct.
568 ********************************************************************/
570 uint64_t get_file_size_stat(const SMB_STRUCT_STAT *sbuf)
572 return sbuf->st_ex_size;
575 /*******************************************************************
576 Returns the size in bytes of the named file.
577 ********************************************************************/
579 SMB_OFF_T get_file_size(char *file_name)
581 SMB_STRUCT_STAT buf;
582 buf.st_ex_size = 0;
583 if (sys_stat(file_name, &buf, false) != 0)
584 return (SMB_OFF_T)-1;
585 return get_file_size_stat(&buf);
588 /*******************************************************************
589 Return a string representing an attribute for a file.
590 ********************************************************************/
592 char *attrib_string(uint16 mode)
594 fstring attrstr;
596 attrstr[0] = 0;
598 if (mode & aVOLID) fstrcat(attrstr,"V");
599 if (mode & aDIR) fstrcat(attrstr,"D");
600 if (mode & aARCH) fstrcat(attrstr,"A");
601 if (mode & aHIDDEN) fstrcat(attrstr,"H");
602 if (mode & aSYSTEM) fstrcat(attrstr,"S");
603 if (mode & aRONLY) fstrcat(attrstr,"R");
605 return talloc_strdup(talloc_tos(), attrstr);
608 /*******************************************************************
609 Show a smb message structure.
610 ********************************************************************/
612 void show_msg(char *buf)
614 int i;
615 int bcc=0;
617 if (!DEBUGLVL(5))
618 return;
620 DEBUG(5,("size=%d\nsmb_com=0x%x\nsmb_rcls=%d\nsmb_reh=%d\nsmb_err=%d\nsmb_flg=%d\nsmb_flg2=%d\n",
621 smb_len(buf),
622 (int)CVAL(buf,smb_com),
623 (int)CVAL(buf,smb_rcls),
624 (int)CVAL(buf,smb_reh),
625 (int)SVAL(buf,smb_err),
626 (int)CVAL(buf,smb_flg),
627 (int)SVAL(buf,smb_flg2)));
628 DEBUGADD(5,("smb_tid=%d\nsmb_pid=%d\nsmb_uid=%d\nsmb_mid=%d\n",
629 (int)SVAL(buf,smb_tid),
630 (int)SVAL(buf,smb_pid),
631 (int)SVAL(buf,smb_uid),
632 (int)SVAL(buf,smb_mid)));
633 DEBUGADD(5,("smt_wct=%d\n",(int)CVAL(buf,smb_wct)));
635 for (i=0;i<(int)CVAL(buf,smb_wct);i++)
636 DEBUGADD(5,("smb_vwv[%2d]=%5d (0x%X)\n",i,
637 SVAL(buf,smb_vwv+2*i),SVAL(buf,smb_vwv+2*i)));
639 bcc = (int)SVAL(buf,smb_vwv+2*(CVAL(buf,smb_wct)));
641 DEBUGADD(5,("smb_bcc=%d\n",bcc));
643 if (DEBUGLEVEL < 10)
644 return;
646 if (DEBUGLEVEL < 50)
647 bcc = MIN(bcc, 512);
649 dump_data(10, (uint8 *)smb_buf(buf), bcc);
652 /*******************************************************************
653 Set the length and marker of an encrypted smb packet.
654 ********************************************************************/
656 void smb_set_enclen(char *buf,int len,uint16 enc_ctx_num)
658 _smb_setlen(buf,len);
660 SCVAL(buf,4,0xFF);
661 SCVAL(buf,5,'E');
662 SSVAL(buf,6,enc_ctx_num);
665 /*******************************************************************
666 Set the length and marker of an smb packet.
667 ********************************************************************/
669 void smb_setlen(char *buf,int len)
671 _smb_setlen(buf,len);
673 SCVAL(buf,4,0xFF);
674 SCVAL(buf,5,'S');
675 SCVAL(buf,6,'M');
676 SCVAL(buf,7,'B');
679 /*******************************************************************
680 Setup only the byte count for a smb message.
681 ********************************************************************/
683 int set_message_bcc(char *buf,int num_bytes)
685 int num_words = CVAL(buf,smb_wct);
686 SSVAL(buf,smb_vwv + num_words*SIZEOFWORD,num_bytes);
687 _smb_setlen(buf,smb_size + num_words*2 + num_bytes - 4);
688 return (smb_size + num_words*2 + num_bytes);
691 /*******************************************************************
692 Add a data blob to the end of a smb_buf, adjusting bcc and smb_len.
693 Return the bytes added
694 ********************************************************************/
696 ssize_t message_push_blob(uint8 **outbuf, DATA_BLOB blob)
698 size_t newlen = smb_len(*outbuf) + 4 + blob.length;
699 uint8 *tmp;
701 if (!(tmp = TALLOC_REALLOC_ARRAY(NULL, *outbuf, uint8, newlen))) {
702 DEBUG(0, ("talloc failed\n"));
703 return -1;
705 *outbuf = tmp;
707 memcpy(tmp + smb_len(tmp) + 4, blob.data, blob.length);
708 set_message_bcc((char *)tmp, smb_buflen(tmp) + blob.length);
709 return blob.length;
712 /*******************************************************************
713 Reduce a file name, removing .. elements.
714 ********************************************************************/
716 static char *dos_clean_name(TALLOC_CTX *ctx, const char *s)
718 char *p = NULL;
719 char *str = NULL;
721 DEBUG(3,("dos_clean_name [%s]\n",s));
723 /* remove any double slashes */
724 str = talloc_all_string_sub(ctx, s, "\\\\", "\\");
725 if (!str) {
726 return NULL;
729 /* Remove leading .\\ characters */
730 if(strncmp(str, ".\\", 2) == 0) {
731 trim_string(str, ".\\", NULL);
732 if(*str == 0) {
733 str = talloc_strdup(ctx, ".\\");
734 if (!str) {
735 return NULL;
740 while ((p = strstr_m(str,"\\..\\")) != NULL) {
741 char *s1;
743 *p = 0;
744 s1 = p+3;
746 if ((p=strrchr_m(str,'\\')) != NULL) {
747 *p = 0;
748 } else {
749 *str = 0;
751 str = talloc_asprintf(ctx,
752 "%s%s",
753 str,
754 s1);
755 if (!str) {
756 return NULL;
760 trim_string(str,NULL,"\\..");
761 return talloc_all_string_sub(ctx, str, "\\.\\", "\\");
764 /*******************************************************************
765 Reduce a file name, removing .. elements.
766 ********************************************************************/
768 char *unix_clean_name(TALLOC_CTX *ctx, const char *s)
770 char *p = NULL;
771 char *str = NULL;
773 DEBUG(3,("unix_clean_name [%s]\n",s));
775 /* remove any double slashes */
776 str = talloc_all_string_sub(ctx, s, "//","/");
777 if (!str) {
778 return NULL;
781 /* Remove leading ./ characters */
782 if(strncmp(str, "./", 2) == 0) {
783 trim_string(str, "./", NULL);
784 if(*str == 0) {
785 str = talloc_strdup(ctx, "./");
786 if (!str) {
787 return NULL;
792 while ((p = strstr_m(str,"/../")) != NULL) {
793 char *s1;
795 *p = 0;
796 s1 = p+3;
798 if ((p=strrchr_m(str,'/')) != NULL) {
799 *p = 0;
800 } else {
801 *str = 0;
803 str = talloc_asprintf(ctx,
804 "%s%s",
805 str,
806 s1);
807 if (!str) {
808 return NULL;
812 trim_string(str,NULL,"/..");
813 return talloc_all_string_sub(ctx, str, "/./", "/");
816 char *clean_name(TALLOC_CTX *ctx, const char *s)
818 char *str = dos_clean_name(ctx, s);
819 if (!str) {
820 return NULL;
822 return unix_clean_name(ctx, str);
825 /*******************************************************************
826 Write data into an fd at a given offset. Ignore seek errors.
827 ********************************************************************/
829 ssize_t write_data_at_offset(int fd, const char *buffer, size_t N, SMB_OFF_T pos)
831 size_t total=0;
832 ssize_t ret;
834 if (pos == (SMB_OFF_T)-1) {
835 return write_data(fd, buffer, N);
837 #if defined(HAVE_PWRITE) || defined(HAVE_PRWITE64)
838 while (total < N) {
839 ret = sys_pwrite(fd,buffer + total,N - total, pos);
840 if (ret == -1 && errno == ESPIPE) {
841 return write_data(fd, buffer + total,N - total);
843 if (ret == -1) {
844 DEBUG(0,("write_data_at_offset: write failure. Error = %s\n", strerror(errno) ));
845 return -1;
847 if (ret == 0) {
848 return total;
850 total += ret;
851 pos += ret;
853 return (ssize_t)total;
854 #else
855 /* Use lseek and write_data. */
856 if (sys_lseek(fd, pos, SEEK_SET) == -1) {
857 if (errno != ESPIPE) {
858 return -1;
861 return write_data(fd, buffer, N);
862 #endif
865 /*******************************************************************
866 Sleep for a specified number of milliseconds.
867 ********************************************************************/
869 void smb_msleep(unsigned int t)
871 #if defined(HAVE_NANOSLEEP)
872 struct timespec tval;
873 int ret;
875 tval.tv_sec = t/1000;
876 tval.tv_nsec = 1000000*(t%1000);
878 do {
879 errno = 0;
880 ret = nanosleep(&tval, &tval);
881 } while (ret < 0 && errno == EINTR && (tval.tv_sec > 0 || tval.tv_nsec > 0));
882 #else
883 unsigned int tdiff=0;
884 struct timeval tval,t1,t2;
885 fd_set fds;
887 GetTimeOfDay(&t1);
888 t2 = t1;
890 while (tdiff < t) {
891 tval.tv_sec = (t-tdiff)/1000;
892 tval.tv_usec = 1000*((t-tdiff)%1000);
894 /* Never wait for more than 1 sec. */
895 if (tval.tv_sec > 1) {
896 tval.tv_sec = 1;
897 tval.tv_usec = 0;
900 FD_ZERO(&fds);
901 errno = 0;
902 sys_select_intr(0,&fds,NULL,NULL,&tval);
904 GetTimeOfDay(&t2);
905 if (t2.tv_sec < t1.tv_sec) {
906 /* Someone adjusted time... */
907 t1 = t2;
910 tdiff = TvalDiff(&t1,&t2);
912 #endif
915 NTSTATUS reinit_after_fork(struct messaging_context *msg_ctx,
916 struct event_context *ev_ctx,
917 bool parent_longlived)
919 NTSTATUS status = NT_STATUS_OK;
921 /* Reset the state of the random
922 * number generation system, so
923 * children do not get the same random
924 * numbers as each other */
925 set_need_random_reseed();
927 /* tdb needs special fork handling */
928 if (tdb_reopen_all(parent_longlived ? 1 : 0) == -1) {
929 DEBUG(0,("tdb_reopen_all failed.\n"));
930 status = NT_STATUS_OPEN_FAILED;
931 goto done;
934 if (ev_ctx) {
935 event_context_reinit(ev_ctx);
938 if (msg_ctx) {
940 * For clustering, we need to re-init our ctdbd connection after the
941 * fork
943 status = messaging_reinit(msg_ctx);
944 if (!NT_STATUS_IS_OK(status)) {
945 DEBUG(0,("messaging_reinit() failed: %s\n",
946 nt_errstr(status)));
949 done:
950 return status;
953 /****************************************************************************
954 Put up a yes/no prompt.
955 ****************************************************************************/
957 bool yesno(const char *p)
959 char ans[20];
960 printf("%s",p);
962 if (!fgets(ans,sizeof(ans)-1,stdin))
963 return(False);
965 if (*ans == 'y' || *ans == 'Y')
966 return(True);
968 return(False);
971 #if defined(PARANOID_MALLOC_CHECKER)
973 /****************************************************************************
974 Internal malloc wrapper. Externally visible.
975 ****************************************************************************/
977 void *malloc_(size_t size)
979 if (size == 0) {
980 return NULL;
982 #undef malloc
983 return malloc(size);
984 #define malloc(s) __ERROR_DONT_USE_MALLOC_DIRECTLY
987 /****************************************************************************
988 Internal calloc wrapper. Not externally visible.
989 ****************************************************************************/
991 static void *calloc_(size_t count, size_t size)
993 if (size == 0 || count == 0) {
994 return NULL;
996 #undef calloc
997 return calloc(count, size);
998 #define calloc(n,s) __ERROR_DONT_USE_CALLOC_DIRECTLY
1001 /****************************************************************************
1002 Internal realloc wrapper. Not externally visible.
1003 ****************************************************************************/
1005 static void *realloc_(void *ptr, size_t size)
1007 #undef realloc
1008 return realloc(ptr, size);
1009 #define realloc(p,s) __ERROR_DONT_USE_RELLOC_DIRECTLY
1012 #endif /* PARANOID_MALLOC_CHECKER */
1014 /****************************************************************************
1015 Type-safe memalign
1016 ****************************************************************************/
1018 void *memalign_array(size_t el_size, size_t align, unsigned int count)
1020 if (count >= MAX_ALLOC_SIZE/el_size) {
1021 return NULL;
1024 return sys_memalign(align, el_size*count);
1027 /****************************************************************************
1028 Type-safe calloc.
1029 ****************************************************************************/
1031 void *calloc_array(size_t size, size_t nmemb)
1033 if (nmemb >= MAX_ALLOC_SIZE/size) {
1034 return NULL;
1036 if (size == 0 || nmemb == 0) {
1037 return NULL;
1039 #if defined(PARANOID_MALLOC_CHECKER)
1040 return calloc_(nmemb, size);
1041 #else
1042 return calloc(nmemb, size);
1043 #endif
1046 /****************************************************************************
1047 Expand a pointer to be a particular size.
1048 Note that this version of Realloc has an extra parameter that decides
1049 whether to free the passed in storage on allocation failure or if the
1050 new size is zero.
1052 This is designed for use in the typical idiom of :
1054 p = SMB_REALLOC(p, size)
1055 if (!p) {
1056 return error;
1059 and not to have to keep track of the old 'p' contents to free later, nor
1060 to worry if the size parameter was zero. In the case where NULL is returned
1061 we guarentee that p has been freed.
1063 If free later semantics are desired, then pass 'free_old_on_error' as False which
1064 guarentees that the old contents are not freed on error, even if size == 0. To use
1065 this idiom use :
1067 tmp = SMB_REALLOC_KEEP_OLD_ON_ERROR(p, size);
1068 if (!tmp) {
1069 SAFE_FREE(p);
1070 return error;
1071 } else {
1072 p = tmp;
1075 Changes were instigated by Coverity error checking. JRA.
1076 ****************************************************************************/
1078 void *Realloc(void *p, size_t size, bool free_old_on_error)
1080 void *ret=NULL;
1082 if (size == 0) {
1083 if (free_old_on_error) {
1084 SAFE_FREE(p);
1086 DEBUG(2,("Realloc asked for 0 bytes\n"));
1087 return NULL;
1090 #if defined(PARANOID_MALLOC_CHECKER)
1091 if (!p) {
1092 ret = (void *)malloc_(size);
1093 } else {
1094 ret = (void *)realloc_(p,size);
1096 #else
1097 if (!p) {
1098 ret = (void *)malloc(size);
1099 } else {
1100 ret = (void *)realloc(p,size);
1102 #endif
1104 if (!ret) {
1105 if (free_old_on_error && p) {
1106 SAFE_FREE(p);
1108 DEBUG(0,("Memory allocation error: failed to expand to %d bytes\n",(int)size));
1111 return(ret);
1114 /****************************************************************************
1115 (Hopefully) efficient array append.
1116 ****************************************************************************/
1118 void add_to_large_array(TALLOC_CTX *mem_ctx, size_t element_size,
1119 void *element, void *_array, uint32 *num_elements,
1120 ssize_t *array_size)
1122 void **array = (void **)_array;
1124 if (*array_size < 0) {
1125 return;
1128 if (*array == NULL) {
1129 if (*array_size == 0) {
1130 *array_size = 128;
1133 if (*array_size >= MAX_ALLOC_SIZE/element_size) {
1134 goto error;
1137 *array = TALLOC(mem_ctx, element_size * (*array_size));
1138 if (*array == NULL) {
1139 goto error;
1143 if (*num_elements == *array_size) {
1144 *array_size *= 2;
1146 if (*array_size >= MAX_ALLOC_SIZE/element_size) {
1147 goto error;
1150 *array = TALLOC_REALLOC(mem_ctx, *array,
1151 element_size * (*array_size));
1153 if (*array == NULL) {
1154 goto error;
1158 memcpy((char *)(*array) + element_size*(*num_elements),
1159 element, element_size);
1160 *num_elements += 1;
1162 return;
1164 error:
1165 *num_elements = 0;
1166 *array_size = -1;
1169 /****************************************************************************
1170 Get my own domain name, or "" if we have none.
1171 ****************************************************************************/
1173 char *get_mydnsdomname(TALLOC_CTX *ctx)
1175 const char *domname;
1176 char *p;
1178 domname = get_mydnsfullname();
1179 if (!domname) {
1180 return NULL;
1183 p = strchr_m(domname, '.');
1184 if (p) {
1185 p++;
1186 return talloc_strdup(ctx, p);
1187 } else {
1188 return talloc_strdup(ctx, "");
1192 /****************************************************************************
1193 Interpret a protocol description string, with a default.
1194 ****************************************************************************/
1196 int interpret_protocol(const char *str,int def)
1198 if (strequal(str,"NT1"))
1199 return(PROTOCOL_NT1);
1200 if (strequal(str,"LANMAN2"))
1201 return(PROTOCOL_LANMAN2);
1202 if (strequal(str,"LANMAN1"))
1203 return(PROTOCOL_LANMAN1);
1204 if (strequal(str,"CORE"))
1205 return(PROTOCOL_CORE);
1206 if (strequal(str,"COREPLUS"))
1207 return(PROTOCOL_COREPLUS);
1208 if (strequal(str,"CORE+"))
1209 return(PROTOCOL_COREPLUS);
1211 DEBUG(0,("Unrecognised protocol level %s\n",str));
1213 return(def);
1217 #if (defined(HAVE_NETGROUP) && defined(WITH_AUTOMOUNT))
1218 /******************************************************************
1219 Remove any mount options such as -rsize=2048,wsize=2048 etc.
1220 Based on a fix from <Thomas.Hepper@icem.de>.
1221 Returns a malloc'ed string.
1222 *******************************************************************/
1224 static char *strip_mount_options(TALLOC_CTX *ctx, const char *str)
1226 if (*str == '-') {
1227 const char *p = str;
1228 while(*p && !isspace(*p))
1229 p++;
1230 while(*p && isspace(*p))
1231 p++;
1232 if(*p) {
1233 return talloc_strdup(ctx, p);
1236 return NULL;
1239 /*******************************************************************
1240 Patch from jkf@soton.ac.uk
1241 Split Luke's automount_server into YP lookup and string splitter
1242 so can easily implement automount_path().
1243 Returns a malloc'ed string.
1244 *******************************************************************/
1246 #ifdef WITH_NISPLUS_HOME
1247 char *automount_lookup(TALLOC_CTX *ctx, const char *user_name)
1249 char *value = NULL;
1251 char *nis_map = (char *)lp_nis_home_map_name();
1253 char buffer[NIS_MAXATTRVAL + 1];
1254 nis_result *result;
1255 nis_object *object;
1256 entry_obj *entry;
1258 snprintf(buffer, sizeof(buffer), "[key=%s],%s", user_name, nis_map);
1259 DEBUG(5, ("NIS+ querystring: %s\n", buffer));
1261 if (result = nis_list(buffer, FOLLOW_PATH|EXPAND_NAME|HARD_LOOKUP, NULL, NULL)) {
1262 if (result->status != NIS_SUCCESS) {
1263 DEBUG(3, ("NIS+ query failed: %s\n", nis_sperrno(result->status)));
1264 } else {
1265 object = result->objects.objects_val;
1266 if (object->zo_data.zo_type == ENTRY_OBJ) {
1267 entry = &object->zo_data.objdata_u.en_data;
1268 DEBUG(5, ("NIS+ entry type: %s\n", entry->en_type));
1269 DEBUG(3, ("NIS+ result: %s\n", entry->en_cols.en_cols_val[1].ec_value.ec_value_val));
1271 value = talloc_strdup(ctx,
1272 entry->en_cols.en_cols_val[1].ec_value.ec_value_val);
1273 if (!value) {
1274 nis_freeresult(result);
1275 return NULL;
1277 value = talloc_string_sub(ctx,
1278 value,
1279 "&",
1280 user_name);
1284 nis_freeresult(result);
1286 if (value) {
1287 value = strip_mount_options(ctx, value);
1288 DEBUG(4, ("NIS+ Lookup: %s resulted in %s\n",
1289 user_name, value));
1291 return value;
1293 #else /* WITH_NISPLUS_HOME */
1295 char *automount_lookup(TALLOC_CTX *ctx, const char *user_name)
1297 char *value = NULL;
1299 int nis_error; /* returned by yp all functions */
1300 char *nis_result; /* yp_match inits this */
1301 int nis_result_len; /* and set this */
1302 char *nis_domain; /* yp_get_default_domain inits this */
1303 char *nis_map = (char *)lp_nis_home_map_name();
1305 if ((nis_error = yp_get_default_domain(&nis_domain)) != 0) {
1306 DEBUG(3, ("YP Error: %s\n", yperr_string(nis_error)));
1307 return NULL;
1310 DEBUG(5, ("NIS Domain: %s\n", nis_domain));
1312 if ((nis_error = yp_match(nis_domain, nis_map, user_name,
1313 strlen(user_name), &nis_result,
1314 &nis_result_len)) == 0) {
1315 if (nis_result_len > 0 && nis_result[nis_result_len] == '\n') {
1316 nis_result[nis_result_len] = '\0';
1318 value = talloc_strdup(ctx, nis_result);
1319 if (!value) {
1320 return NULL;
1322 value = strip_mount_options(ctx, value);
1323 } else if(nis_error == YPERR_KEY) {
1324 DEBUG(3, ("YP Key not found: while looking up \"%s\" in map \"%s\"\n",
1325 user_name, nis_map));
1326 DEBUG(3, ("using defaults for server and home directory\n"));
1327 } else {
1328 DEBUG(3, ("YP Error: \"%s\" while looking up \"%s\" in map \"%s\"\n",
1329 yperr_string(nis_error), user_name, nis_map));
1332 if (value) {
1333 DEBUG(4, ("YP Lookup: %s resulted in %s\n", user_name, value));
1335 return value;
1337 #endif /* WITH_NISPLUS_HOME */
1338 #endif
1340 /****************************************************************************
1341 Check if a process exists. Does this work on all unixes?
1342 ****************************************************************************/
1344 bool process_exists(const struct server_id pid)
1346 if (procid_is_me(&pid)) {
1347 return True;
1350 if (procid_is_local(&pid)) {
1351 return (kill(pid.pid,0) == 0 || errno != ESRCH);
1354 #ifdef CLUSTER_SUPPORT
1355 return ctdbd_process_exists(messaging_ctdbd_connection(), pid.vnn,
1356 pid.pid);
1357 #else
1358 return False;
1359 #endif
1362 /*******************************************************************
1363 Convert a uid into a user name.
1364 ********************************************************************/
1366 const char *uidtoname(uid_t uid)
1368 TALLOC_CTX *ctx = talloc_tos();
1369 char *name = NULL;
1370 struct passwd *pass = NULL;
1372 pass = getpwuid_alloc(ctx,uid);
1373 if (pass) {
1374 name = talloc_strdup(ctx,pass->pw_name);
1375 TALLOC_FREE(pass);
1376 } else {
1377 name = talloc_asprintf(ctx,
1378 "%ld",
1379 (long int)uid);
1381 return name;
1384 /*******************************************************************
1385 Convert a gid into a group name.
1386 ********************************************************************/
1388 char *gidtoname(gid_t gid)
1390 struct group *grp;
1392 grp = getgrgid(gid);
1393 if (grp) {
1394 return talloc_strdup(talloc_tos(), grp->gr_name);
1396 else {
1397 return talloc_asprintf(talloc_tos(),
1398 "%d",
1399 (int)gid);
1403 /*******************************************************************
1404 Convert a user name into a uid.
1405 ********************************************************************/
1407 uid_t nametouid(const char *name)
1409 struct passwd *pass;
1410 char *p;
1411 uid_t u;
1413 pass = Get_Pwnam_alloc(talloc_autofree_context(), name);
1414 if (pass) {
1415 u = pass->pw_uid;
1416 TALLOC_FREE(pass);
1417 return u;
1420 u = (uid_t)strtol(name, &p, 0);
1421 if ((p != name) && (*p == '\0'))
1422 return u;
1424 return (uid_t)-1;
1427 /*******************************************************************
1428 Convert a name to a gid_t if possible. Return -1 if not a group.
1429 ********************************************************************/
1431 gid_t nametogid(const char *name)
1433 struct group *grp;
1434 char *p;
1435 gid_t g;
1437 g = (gid_t)strtol(name, &p, 0);
1438 if ((p != name) && (*p == '\0'))
1439 return g;
1441 grp = sys_getgrnam(name);
1442 if (grp)
1443 return(grp->gr_gid);
1444 return (gid_t)-1;
1447 /*******************************************************************
1448 Something really nasty happened - panic !
1449 ********************************************************************/
1451 void smb_panic(const char *const why)
1453 char *cmd;
1454 int result;
1456 #ifdef DEVELOPER
1459 if (global_clobber_region_function) {
1460 DEBUG(0,("smb_panic: clobber_region() last called from [%s(%u)]\n",
1461 global_clobber_region_function,
1462 global_clobber_region_line));
1465 #endif
1467 DEBUG(0,("PANIC (pid %llu): %s\n",
1468 (unsigned long long)sys_getpid(), why));
1469 log_stack_trace();
1471 cmd = lp_panic_action();
1472 if (cmd && *cmd) {
1473 DEBUG(0, ("smb_panic(): calling panic action [%s]\n", cmd));
1474 result = system(cmd);
1476 if (result == -1)
1477 DEBUG(0, ("smb_panic(): fork failed in panic action: %s\n",
1478 strerror(errno)));
1479 else
1480 DEBUG(0, ("smb_panic(): action returned status %d\n",
1481 WEXITSTATUS(result)));
1484 dump_core();
1487 /*******************************************************************
1488 Print a backtrace of the stack to the debug log. This function
1489 DELIBERATELY LEAKS MEMORY. The expectation is that you should
1490 exit shortly after calling it.
1491 ********************************************************************/
1493 #ifdef HAVE_LIBUNWIND_H
1494 #include <libunwind.h>
1495 #endif
1497 #ifdef HAVE_EXECINFO_H
1498 #include <execinfo.h>
1499 #endif
1501 #ifdef HAVE_LIBEXC_H
1502 #include <libexc.h>
1503 #endif
1505 void log_stack_trace(void)
1507 #ifdef HAVE_LIBUNWIND
1508 /* Try to use libunwind before any other technique since on ia64
1509 * libunwind correctly walks the stack in more circumstances than
1510 * backtrace.
1512 unw_cursor_t cursor;
1513 unw_context_t uc;
1514 unsigned i = 0;
1516 char procname[256];
1517 unw_word_t ip, sp, off;
1519 procname[sizeof(procname) - 1] = '\0';
1521 if (unw_getcontext(&uc) != 0) {
1522 goto libunwind_failed;
1525 if (unw_init_local(&cursor, &uc) != 0) {
1526 goto libunwind_failed;
1529 DEBUG(0, ("BACKTRACE:\n"));
1531 do {
1532 ip = sp = 0;
1533 unw_get_reg(&cursor, UNW_REG_IP, &ip);
1534 unw_get_reg(&cursor, UNW_REG_SP, &sp);
1536 switch (unw_get_proc_name(&cursor,
1537 procname, sizeof(procname) - 1, &off) ) {
1538 case 0:
1539 /* Name found. */
1540 case -UNW_ENOMEM:
1541 /* Name truncated. */
1542 DEBUGADD(0, (" #%u %s + %#llx [ip=%#llx] [sp=%#llx]\n",
1543 i, procname, (long long)off,
1544 (long long)ip, (long long) sp));
1545 break;
1546 default:
1547 /* case -UNW_ENOINFO: */
1548 /* case -UNW_EUNSPEC: */
1549 /* No symbol name found. */
1550 DEBUGADD(0, (" #%u %s [ip=%#llx] [sp=%#llx]\n",
1551 i, "<unknown symbol>",
1552 (long long)ip, (long long) sp));
1554 ++i;
1555 } while (unw_step(&cursor) > 0);
1557 return;
1559 libunwind_failed:
1560 DEBUG(0, ("unable to produce a stack trace with libunwind\n"));
1562 #elif HAVE_BACKTRACE_SYMBOLS
1563 void *backtrace_stack[BACKTRACE_STACK_SIZE];
1564 size_t backtrace_size;
1565 char **backtrace_strings;
1567 /* get the backtrace (stack frames) */
1568 backtrace_size = backtrace(backtrace_stack,BACKTRACE_STACK_SIZE);
1569 backtrace_strings = backtrace_symbols(backtrace_stack, backtrace_size);
1571 DEBUG(0, ("BACKTRACE: %lu stack frames:\n",
1572 (unsigned long)backtrace_size));
1574 if (backtrace_strings) {
1575 int i;
1577 for (i = 0; i < backtrace_size; i++)
1578 DEBUGADD(0, (" #%u %s\n", i, backtrace_strings[i]));
1580 /* Leak the backtrace_strings, rather than risk what free() might do */
1583 #elif HAVE_LIBEXC
1585 /* The IRIX libexc library provides an API for unwinding the stack. See
1586 * libexc(3) for details. Apparantly trace_back_stack leaks memory, but
1587 * since we are about to abort anyway, it hardly matters.
1590 #define NAMESIZE 32 /* Arbitrary */
1592 __uint64_t addrs[BACKTRACE_STACK_SIZE];
1593 char * names[BACKTRACE_STACK_SIZE];
1594 char namebuf[BACKTRACE_STACK_SIZE * NAMESIZE];
1596 int i;
1597 int levels;
1599 ZERO_ARRAY(addrs);
1600 ZERO_ARRAY(names);
1601 ZERO_ARRAY(namebuf);
1603 /* We need to be root so we can open our /proc entry to walk
1604 * our stack. It also helps when we want to dump core.
1606 become_root();
1608 for (i = 0; i < BACKTRACE_STACK_SIZE; i++) {
1609 names[i] = namebuf + (i * NAMESIZE);
1612 levels = trace_back_stack(0, addrs, names,
1613 BACKTRACE_STACK_SIZE, NAMESIZE - 1);
1615 DEBUG(0, ("BACKTRACE: %d stack frames:\n", levels));
1616 for (i = 0; i < levels; i++) {
1617 DEBUGADD(0, (" #%d 0x%llx %s\n", i, addrs[i], names[i]));
1619 #undef NAMESIZE
1621 #else
1622 DEBUG(0, ("unable to produce a stack trace on this platform\n"));
1623 #endif
1626 /*******************************************************************
1627 A readdir wrapper which just returns the file name.
1628 ********************************************************************/
1630 const char *readdirname(SMB_STRUCT_DIR *p)
1632 SMB_STRUCT_DIRENT *ptr;
1633 char *dname;
1635 if (!p)
1636 return(NULL);
1638 ptr = (SMB_STRUCT_DIRENT *)sys_readdir(p);
1639 if (!ptr)
1640 return(NULL);
1642 dname = ptr->d_name;
1644 #ifdef NEXT2
1645 if (telldir(p) < 0)
1646 return(NULL);
1647 #endif
1649 #ifdef HAVE_BROKEN_READDIR_NAME
1650 /* using /usr/ucb/cc is BAD */
1651 dname = dname - 2;
1652 #endif
1654 return talloc_strdup(talloc_tos(), dname);
1657 /*******************************************************************
1658 Utility function used to decide if the last component
1659 of a path matches a (possibly wildcarded) entry in a namelist.
1660 ********************************************************************/
1662 bool is_in_path(const char *name, name_compare_entry *namelist, bool case_sensitive)
1664 const char *last_component;
1666 /* if we have no list it's obviously not in the path */
1667 if((namelist == NULL ) || ((namelist != NULL) && (namelist[0].name == NULL))) {
1668 return False;
1671 DEBUG(8, ("is_in_path: %s\n", name));
1673 /* Get the last component of the unix name. */
1674 last_component = strrchr_m(name, '/');
1675 if (!last_component) {
1676 last_component = name;
1677 } else {
1678 last_component++; /* Go past '/' */
1681 for(; namelist->name != NULL; namelist++) {
1682 if(namelist->is_wild) {
1683 if (mask_match(last_component, namelist->name, case_sensitive)) {
1684 DEBUG(8,("is_in_path: mask match succeeded\n"));
1685 return True;
1687 } else {
1688 if((case_sensitive && (strcmp(last_component, namelist->name) == 0))||
1689 (!case_sensitive && (StrCaseCmp(last_component, namelist->name) == 0))) {
1690 DEBUG(8,("is_in_path: match succeeded\n"));
1691 return True;
1695 DEBUG(8,("is_in_path: match not found\n"));
1696 return False;
1699 /*******************************************************************
1700 Strip a '/' separated list into an array of
1701 name_compare_enties structures suitable for
1702 passing to is_in_path(). We do this for
1703 speed so we can pre-parse all the names in the list
1704 and don't do it for each call to is_in_path().
1705 namelist is modified here and is assumed to be
1706 a copy owned by the caller.
1707 We also check if the entry contains a wildcard to
1708 remove a potentially expensive call to mask_match
1709 if possible.
1710 ********************************************************************/
1712 void set_namearray(name_compare_entry **ppname_array, const char *namelist)
1714 char *name_end;
1715 char *nameptr = (char *)namelist;
1716 int num_entries = 0;
1717 int i;
1719 (*ppname_array) = NULL;
1721 if((nameptr == NULL ) || ((nameptr != NULL) && (*nameptr == '\0')))
1722 return;
1724 /* We need to make two passes over the string. The
1725 first to count the number of elements, the second
1726 to split it.
1729 while(*nameptr) {
1730 if ( *nameptr == '/' ) {
1731 /* cope with multiple (useless) /s) */
1732 nameptr++;
1733 continue;
1735 /* anything left? */
1736 if ( *nameptr == '\0' )
1737 break;
1739 /* find the next '/' or consume remaining */
1740 name_end = strchr_m(nameptr, '/');
1741 if (name_end == NULL)
1742 name_end = (char *)nameptr + strlen(nameptr);
1744 /* next segment please */
1745 nameptr = name_end + 1;
1746 num_entries++;
1749 if(num_entries == 0)
1750 return;
1752 if(( (*ppname_array) = SMB_MALLOC_ARRAY(name_compare_entry, num_entries + 1)) == NULL) {
1753 DEBUG(0,("set_namearray: malloc fail\n"));
1754 return;
1757 /* Now copy out the names */
1758 nameptr = (char *)namelist;
1759 i = 0;
1760 while(*nameptr) {
1761 if ( *nameptr == '/' ) {
1762 /* cope with multiple (useless) /s) */
1763 nameptr++;
1764 continue;
1766 /* anything left? */
1767 if ( *nameptr == '\0' )
1768 break;
1770 /* find the next '/' or consume remaining */
1771 name_end = strchr_m(nameptr, '/');
1772 if (name_end)
1773 *name_end = '\0';
1774 else
1775 name_end = nameptr + strlen(nameptr);
1777 (*ppname_array)[i].is_wild = ms_has_wild(nameptr);
1778 if(((*ppname_array)[i].name = SMB_STRDUP(nameptr)) == NULL) {
1779 DEBUG(0,("set_namearray: malloc fail (1)\n"));
1780 return;
1783 /* next segment please */
1784 nameptr = name_end + 1;
1785 i++;
1788 (*ppname_array)[i].name = NULL;
1790 return;
1793 /****************************************************************************
1794 Routine to free a namearray.
1795 ****************************************************************************/
1797 void free_namearray(name_compare_entry *name_array)
1799 int i;
1801 if(name_array == NULL)
1802 return;
1804 for(i=0; name_array[i].name!=NULL; i++)
1805 SAFE_FREE(name_array[i].name);
1806 SAFE_FREE(name_array);
1809 #undef DBGC_CLASS
1810 #define DBGC_CLASS DBGC_LOCKING
1812 /****************************************************************************
1813 Simple routine to query existing file locks. Cruft in NFS and 64->32 bit mapping
1814 is dealt with in posix.c
1815 Returns True if we have information regarding this lock region (and returns
1816 F_UNLCK in *ptype if the region is unlocked). False if the call failed.
1817 ****************************************************************************/
1819 bool fcntl_getlock(int fd, SMB_OFF_T *poffset, SMB_OFF_T *pcount, int *ptype, pid_t *ppid)
1821 SMB_STRUCT_FLOCK lock;
1822 int ret;
1824 DEBUG(8,("fcntl_getlock fd=%d offset=%.0f count=%.0f type=%d\n",
1825 fd,(double)*poffset,(double)*pcount,*ptype));
1827 lock.l_type = *ptype;
1828 lock.l_whence = SEEK_SET;
1829 lock.l_start = *poffset;
1830 lock.l_len = *pcount;
1831 lock.l_pid = 0;
1833 ret = sys_fcntl_ptr(fd,SMB_F_GETLK,&lock);
1835 if (ret == -1) {
1836 int sav = errno;
1837 DEBUG(3,("fcntl_getlock: lock request failed at offset %.0f count %.0f type %d (%s)\n",
1838 (double)*poffset,(double)*pcount,*ptype,strerror(errno)));
1839 errno = sav;
1840 return False;
1843 *ptype = lock.l_type;
1844 *poffset = lock.l_start;
1845 *pcount = lock.l_len;
1846 *ppid = lock.l_pid;
1848 DEBUG(3,("fcntl_getlock: fd %d is returned info %d pid %u\n",
1849 fd, (int)lock.l_type, (unsigned int)lock.l_pid));
1850 return True;
1853 #undef DBGC_CLASS
1854 #define DBGC_CLASS DBGC_ALL
1856 /*******************************************************************
1857 Is the name specified one of my netbios names.
1858 Returns true if it is equal, false otherwise.
1859 ********************************************************************/
1861 bool is_myname(const char *s)
1863 int n;
1864 bool ret = False;
1866 for (n=0; my_netbios_names(n); n++) {
1867 if (strequal(my_netbios_names(n), s)) {
1868 ret=True;
1869 break;
1872 DEBUG(8, ("is_myname(\"%s\") returns %d\n", s, ret));
1873 return(ret);
1876 /*******************************************************************
1877 Is the name specified our workgroup/domain.
1878 Returns true if it is equal, false otherwise.
1879 ********************************************************************/
1881 bool is_myworkgroup(const char *s)
1883 bool ret = False;
1885 if (strequal(s, lp_workgroup())) {
1886 ret=True;
1889 DEBUG(8, ("is_myworkgroup(\"%s\") returns %d\n", s, ret));
1890 return(ret);
1893 /*******************************************************************
1894 we distinguish between 2K and XP by the "Native Lan Manager" string
1895 WinXP => "Windows 2002 5.1"
1896 WinXP 64bit => "Windows XP 5.2"
1897 Win2k => "Windows 2000 5.0"
1898 NT4 => "Windows NT 4.0"
1899 Win9x => "Windows 4.0"
1900 Windows 2003 doesn't set the native lan manager string but
1901 they do set the domain to "Windows 2003 5.2" (probably a bug).
1902 ********************************************************************/
1904 void ra_lanman_string( const char *native_lanman )
1906 if ( strcmp( native_lanman, "Windows 2002 5.1" ) == 0 )
1907 set_remote_arch( RA_WINXP );
1908 else if ( strcmp( native_lanman, "Windows XP 5.2" ) == 0 )
1909 set_remote_arch( RA_WINXP64 );
1910 else if ( strcmp( native_lanman, "Windows Server 2003 5.2" ) == 0 )
1911 set_remote_arch( RA_WIN2K3 );
1914 static const char *remote_arch_str;
1916 const char *get_remote_arch_str(void)
1918 if (!remote_arch_str) {
1919 return "UNKNOWN";
1921 return remote_arch_str;
1924 /*******************************************************************
1925 Set the horrid remote_arch string based on an enum.
1926 ********************************************************************/
1928 void set_remote_arch(enum remote_arch_types type)
1930 ra_type = type;
1931 switch( type ) {
1932 case RA_WFWG:
1933 remote_arch_str = "WfWg";
1934 break;
1935 case RA_OS2:
1936 remote_arch_str = "OS2";
1937 break;
1938 case RA_WIN95:
1939 remote_arch_str = "Win95";
1940 break;
1941 case RA_WINNT:
1942 remote_arch_str = "WinNT";
1943 break;
1944 case RA_WIN2K:
1945 remote_arch_str = "Win2K";
1946 break;
1947 case RA_WINXP:
1948 remote_arch_str = "WinXP";
1949 break;
1950 case RA_WINXP64:
1951 remote_arch_str = "WinXP64";
1952 break;
1953 case RA_WIN2K3:
1954 remote_arch_str = "Win2K3";
1955 break;
1956 case RA_VISTA:
1957 remote_arch_str = "Vista";
1958 break;
1959 case RA_SAMBA:
1960 remote_arch_str = "Samba";
1961 break;
1962 case RA_CIFSFS:
1963 remote_arch_str = "CIFSFS";
1964 break;
1965 default:
1966 ra_type = RA_UNKNOWN;
1967 remote_arch_str = "UNKNOWN";
1968 break;
1971 DEBUG(10,("set_remote_arch: Client arch is \'%s\'\n",
1972 remote_arch_str));
1975 /*******************************************************************
1976 Get the remote_arch type.
1977 ********************************************************************/
1979 enum remote_arch_types get_remote_arch(void)
1981 return ra_type;
1984 const char *tab_depth(int level, int depth)
1986 if( CHECK_DEBUGLVL(level) ) {
1987 dbgtext("%*s", depth*4, "");
1989 return "";
1992 /*****************************************************************************
1993 Provide a checksum on a string
1995 Input: s - the null-terminated character string for which the checksum
1996 will be calculated.
1998 Output: The checksum value calculated for s.
1999 *****************************************************************************/
2001 int str_checksum(const char *s)
2003 TDB_DATA key = string_tdb_data(s);
2004 return jenkins_hash(&key);
2007 /*****************************************************************
2008 Zero a memory area then free it. Used to catch bugs faster.
2009 *****************************************************************/
2011 void zero_free(void *p, size_t size)
2013 memset(p, 0, size);
2014 SAFE_FREE(p);
2017 /*****************************************************************
2018 Set our open file limit to a requested max and return the limit.
2019 *****************************************************************/
2021 int set_maxfiles(int requested_max)
2023 #if (defined(HAVE_GETRLIMIT) && defined(RLIMIT_NOFILE))
2024 struct rlimit rlp;
2025 int saved_current_limit;
2027 if(getrlimit(RLIMIT_NOFILE, &rlp)) {
2028 DEBUG(0,("set_maxfiles: getrlimit (1) for RLIMIT_NOFILE failed with error %s\n",
2029 strerror(errno) ));
2030 /* just guess... */
2031 return requested_max;
2035 * Set the fd limit to be real_max_open_files + MAX_OPEN_FUDGEFACTOR to
2036 * account for the extra fd we need
2037 * as well as the log files and standard
2038 * handles etc. Save the limit we want to set in case
2039 * we are running on an OS that doesn't support this limit (AIX)
2040 * which always returns RLIM_INFINITY for rlp.rlim_max.
2043 /* Try raising the hard (max) limit to the requested amount. */
2045 #if defined(RLIM_INFINITY)
2046 if (rlp.rlim_max != RLIM_INFINITY) {
2047 int orig_max = rlp.rlim_max;
2049 if ( rlp.rlim_max < requested_max )
2050 rlp.rlim_max = requested_max;
2052 /* This failing is not an error - many systems (Linux) don't
2053 support our default request of 10,000 open files. JRA. */
2055 if(setrlimit(RLIMIT_NOFILE, &rlp)) {
2056 DEBUG(3,("set_maxfiles: setrlimit for RLIMIT_NOFILE for %d max files failed with error %s\n",
2057 (int)rlp.rlim_max, strerror(errno) ));
2059 /* Set failed - restore original value from get. */
2060 rlp.rlim_max = orig_max;
2063 #endif
2065 /* Now try setting the soft (current) limit. */
2067 saved_current_limit = rlp.rlim_cur = MIN(requested_max,rlp.rlim_max);
2069 if(setrlimit(RLIMIT_NOFILE, &rlp)) {
2070 DEBUG(0,("set_maxfiles: setrlimit for RLIMIT_NOFILE for %d files failed with error %s\n",
2071 (int)rlp.rlim_cur, strerror(errno) ));
2072 /* just guess... */
2073 return saved_current_limit;
2076 if(getrlimit(RLIMIT_NOFILE, &rlp)) {
2077 DEBUG(0,("set_maxfiles: getrlimit (2) for RLIMIT_NOFILE failed with error %s\n",
2078 strerror(errno) ));
2079 /* just guess... */
2080 return saved_current_limit;
2083 #if defined(RLIM_INFINITY)
2084 if(rlp.rlim_cur == RLIM_INFINITY)
2085 return saved_current_limit;
2086 #endif
2088 if((int)rlp.rlim_cur > saved_current_limit)
2089 return saved_current_limit;
2091 return rlp.rlim_cur;
2092 #else /* !defined(HAVE_GETRLIMIT) || !defined(RLIMIT_NOFILE) */
2094 * No way to know - just guess...
2096 return requested_max;
2097 #endif
2100 /*****************************************************************
2101 malloc that aborts with smb_panic on fail or zero size.
2102 *****************************************************************/
2104 void *smb_xmalloc_array(size_t size, unsigned int count)
2106 void *p;
2107 if (size == 0) {
2108 smb_panic("smb_xmalloc_array: called with zero size");
2110 if (count >= MAX_ALLOC_SIZE/size) {
2111 smb_panic("smb_xmalloc_array: alloc size too large");
2113 if ((p = SMB_MALLOC(size*count)) == NULL) {
2114 DEBUG(0, ("smb_xmalloc_array failed to allocate %lu * %lu bytes\n",
2115 (unsigned long)size, (unsigned long)count));
2116 smb_panic("smb_xmalloc_array: malloc failed");
2118 return p;
2122 vasprintf that aborts on malloc fail
2125 int smb_xvasprintf(char **ptr, const char *format, va_list ap)
2127 int n;
2128 va_list ap2;
2130 va_copy(ap2, ap);
2132 n = vasprintf(ptr, format, ap2);
2133 va_end(ap2);
2134 if (n == -1 || ! *ptr) {
2135 smb_panic("smb_xvasprintf: out of memory");
2137 return n;
2140 /*****************************************************************
2141 Get local hostname and cache result.
2142 *****************************************************************/
2144 char *myhostname(void)
2146 static char *ret;
2147 if (ret == NULL) {
2148 /* This is cached forever so
2149 * use talloc_autofree_context() ctx. */
2150 ret = get_myname(talloc_autofree_context());
2152 return ret;
2156 * @brief Returns an absolute path to a file concatenating the provided
2157 * @a rootpath and @a basename
2159 * @param name Filename, relative to @a rootpath
2161 * @retval Pointer to a string containing the full path.
2164 static char *xx_path(const char *name, const char *rootpath)
2166 char *fname = NULL;
2168 fname = talloc_strdup(talloc_tos(), rootpath);
2169 if (!fname) {
2170 return NULL;
2172 trim_string(fname,"","/");
2174 if (!directory_exist(fname)) {
2175 if (!mkdir(fname,0755))
2176 DEBUG(1, ("Unable to create directory %s for file %s. "
2177 "Error was %s\n", fname, name, strerror(errno)));
2180 return talloc_asprintf(talloc_tos(),
2181 "%s/%s",
2182 fname,
2183 name);
2187 * @brief Returns an absolute path to a file in the Samba lock directory.
2189 * @param name File to find, relative to LOCKDIR.
2191 * @retval Pointer to a talloc'ed string containing the full path.
2194 char *lock_path(const char *name)
2196 return xx_path(name, lp_lockdir());
2200 * @brief Returns an absolute path to a file in the Samba pid directory.
2202 * @param name File to find, relative to PIDDIR.
2204 * @retval Pointer to a talloc'ed string containing the full path.
2207 char *pid_path(const char *name)
2209 return xx_path(name, lp_piddir());
2213 * @brief Returns an absolute path to a file in the Samba lib directory.
2215 * @param name File to find, relative to LIBDIR.
2217 * @retval Pointer to a string containing the full path.
2220 char *lib_path(const char *name)
2222 return talloc_asprintf(talloc_tos(), "%s/%s", get_dyn_LIBDIR(), name);
2226 * @brief Returns an absolute path to a file in the Samba modules directory.
2228 * @param name File to find, relative to MODULESDIR.
2230 * @retval Pointer to a string containing the full path.
2233 char *modules_path(const char *name)
2235 return talloc_asprintf(talloc_tos(), "%s/%s", get_dyn_MODULESDIR(), name);
2239 * @brief Returns an absolute path to a file in the Samba data directory.
2241 * @param name File to find, relative to CODEPAGEDIR.
2243 * @retval Pointer to a talloc'ed string containing the full path.
2246 char *data_path(const char *name)
2248 return talloc_asprintf(talloc_tos(), "%s/%s", get_dyn_CODEPAGEDIR(), name);
2252 * @brief Returns an absolute path to a file in the Samba state directory.
2254 * @param name File to find, relative to STATEDIR.
2256 * @retval Pointer to a talloc'ed string containing the full path.
2259 char *state_path(const char *name)
2261 return xx_path(name, lp_statedir());
2265 * @brief Returns an absolute path to a file in the Samba cache directory.
2267 * @param name File to find, relative to CACHEDIR.
2269 * @retval Pointer to a talloc'ed string containing the full path.
2272 char *cache_path(const char *name)
2274 return xx_path(name, lp_cachedir());
2278 * @brief Returns the platform specific shared library extension.
2280 * @retval Pointer to a const char * containing the extension.
2283 const char *shlib_ext(void)
2285 return get_dyn_SHLIBEXT();
2288 /*******************************************************************
2289 Given a filename - get its directory name
2290 ********************************************************************/
2292 bool parent_dirname(TALLOC_CTX *mem_ctx, const char *dir, char **parent,
2293 const char **name)
2295 char *p;
2296 ptrdiff_t len;
2298 p = strrchr_m(dir, '/'); /* Find final '/', if any */
2300 if (p == NULL) {
2301 if (!(*parent = talloc_strdup(mem_ctx, "."))) {
2302 return False;
2304 if (name) {
2305 *name = dir;
2307 return True;
2310 len = p-dir;
2312 if (!(*parent = (char *)TALLOC_MEMDUP(mem_ctx, dir, len+1))) {
2313 return False;
2315 (*parent)[len] = '\0';
2317 if (name) {
2318 *name = p+1;
2320 return True;
2323 /*******************************************************************
2324 Determine if a pattern contains any Microsoft wildcard characters.
2325 *******************************************************************/
2327 bool ms_has_wild(const char *s)
2329 char c;
2331 if (lp_posix_pathnames()) {
2332 /* With posix pathnames no characters are wild. */
2333 return False;
2336 while ((c = *s++)) {
2337 switch (c) {
2338 case '*':
2339 case '?':
2340 case '<':
2341 case '>':
2342 case '"':
2343 return True;
2346 return False;
2349 bool ms_has_wild_w(const smb_ucs2_t *s)
2351 smb_ucs2_t c;
2352 if (!s) return False;
2353 while ((c = *s++)) {
2354 switch (c) {
2355 case UCS2_CHAR('*'):
2356 case UCS2_CHAR('?'):
2357 case UCS2_CHAR('<'):
2358 case UCS2_CHAR('>'):
2359 case UCS2_CHAR('"'):
2360 return True;
2363 return False;
2366 /*******************************************************************
2367 A wrapper that handles case sensitivity and the special handling
2368 of the ".." name.
2369 *******************************************************************/
2371 bool mask_match(const char *string, const char *pattern, bool is_case_sensitive)
2373 if (strcmp(string,"..") == 0)
2374 string = ".";
2375 if (strcmp(pattern,".") == 0)
2376 return False;
2378 return ms_fnmatch(pattern, string, Protocol <= PROTOCOL_LANMAN2, is_case_sensitive) == 0;
2381 /*******************************************************************
2382 A wrapper that handles case sensitivity and the special handling
2383 of the ".." name. Varient that is only called by old search code which requires
2384 pattern translation.
2385 *******************************************************************/
2387 bool mask_match_search(const char *string, const char *pattern, bool is_case_sensitive)
2389 if (strcmp(string,"..") == 0)
2390 string = ".";
2391 if (strcmp(pattern,".") == 0)
2392 return False;
2394 return ms_fnmatch(pattern, string, True, is_case_sensitive) == 0;
2397 /*******************************************************************
2398 A wrapper that handles a list of patters and calls mask_match()
2399 on each. Returns True if any of the patterns match.
2400 *******************************************************************/
2402 bool mask_match_list(const char *string, char **list, int listLen, bool is_case_sensitive)
2404 while (listLen-- > 0) {
2405 if (mask_match(string, *list++, is_case_sensitive))
2406 return True;
2408 return False;
2411 /*********************************************************
2412 Recursive routine that is called by unix_wild_match.
2413 *********************************************************/
2415 static bool unix_do_match(const char *regexp, const char *str)
2417 const char *p;
2419 for( p = regexp; *p && *str; ) {
2421 switch(*p) {
2422 case '?':
2423 str++;
2424 p++;
2425 break;
2427 case '*':
2430 * Look for a character matching
2431 * the one after the '*'.
2433 p++;
2434 if(!*p)
2435 return true; /* Automatic match */
2436 while(*str) {
2438 while(*str && (*p != *str))
2439 str++;
2442 * Patch from weidel@multichart.de. In the case of the regexp
2443 * '*XX*' we want to ensure there are at least 2 'X' characters
2444 * in the string after the '*' for a match to be made.
2448 int matchcount=0;
2451 * Eat all the characters that match, but count how many there were.
2454 while(*str && (*p == *str)) {
2455 str++;
2456 matchcount++;
2460 * Now check that if the regexp had n identical characters that
2461 * matchcount had at least that many matches.
2464 while ( *(p+1) && (*(p+1) == *p)) {
2465 p++;
2466 matchcount--;
2469 if ( matchcount <= 0 )
2470 return false;
2473 str--; /* We've eaten the match char after the '*' */
2475 if(unix_do_match(p, str))
2476 return true;
2478 if(!*str)
2479 return false;
2480 else
2481 str++;
2483 return false;
2485 default:
2486 if(*str != *p)
2487 return false;
2488 str++;
2489 p++;
2490 break;
2494 if(!*p && !*str)
2495 return true;
2497 if (!*p && str[0] == '.' && str[1] == 0)
2498 return true;
2500 if (!*str && *p == '?') {
2501 while (*p == '?')
2502 p++;
2503 return(!*p);
2506 if(!*str && (*p == '*' && p[1] == '\0'))
2507 return true;
2509 return false;
2512 /*******************************************************************
2513 Simple case insensitive interface to a UNIX wildcard matcher.
2514 Returns True if match, False if not.
2515 *******************************************************************/
2517 bool unix_wild_match(const char *pattern, const char *string)
2519 TALLOC_CTX *ctx = talloc_stackframe();
2520 char *p2;
2521 char *s2;
2522 char *p;
2523 bool ret = false;
2525 p2 = talloc_strdup(ctx,pattern);
2526 s2 = talloc_strdup(ctx,string);
2527 if (!p2 || !s2) {
2528 TALLOC_FREE(ctx);
2529 return false;
2531 strlower_m(p2);
2532 strlower_m(s2);
2534 /* Remove any *? and ** from the pattern as they are meaningless */
2535 for(p = p2; *p; p++) {
2536 while( *p == '*' && (p[1] == '?' ||p[1] == '*')) {
2537 memmove(&p[1], &p[2], strlen(&p[2])+1);
2541 if (strequal(p2,"*")) {
2542 TALLOC_FREE(ctx);
2543 return true;
2546 ret = unix_do_match(p2, s2);
2547 TALLOC_FREE(ctx);
2548 return ret;
2551 /**********************************************************************
2552 Converts a name to a fully qualified domain name.
2553 Returns true if lookup succeeded, false if not (then fqdn is set to name)
2554 Note we deliberately use gethostbyname here, not getaddrinfo as we want
2555 to examine the h_aliases and I don't know how to do that with getaddrinfo.
2556 ***********************************************************************/
2558 bool name_to_fqdn(fstring fqdn, const char *name)
2560 char *full = NULL;
2561 struct hostent *hp = gethostbyname(name);
2563 if (!hp || !hp->h_name || !*hp->h_name) {
2564 DEBUG(10,("name_to_fqdn: lookup for %s failed.\n", name));
2565 fstrcpy(fqdn, name);
2566 return false;
2569 /* Find out if the fqdn is returned as an alias
2570 * to cope with /etc/hosts files where the first
2571 * name is not the fqdn but the short name */
2572 if (hp->h_aliases && (! strchr_m(hp->h_name, '.'))) {
2573 int i;
2574 for (i = 0; hp->h_aliases[i]; i++) {
2575 if (strchr_m(hp->h_aliases[i], '.')) {
2576 full = hp->h_aliases[i];
2577 break;
2581 if (full && (StrCaseCmp(full, "localhost.localdomain") == 0)) {
2582 DEBUG(1, ("WARNING: your /etc/hosts file may be broken!\n"));
2583 DEBUGADD(1, (" Specifing the machine hostname for address 127.0.0.1 may lead\n"));
2584 DEBUGADD(1, (" to Kerberos authentication problems as localhost.localdomain\n"));
2585 DEBUGADD(1, (" may end up being used instead of the real machine FQDN.\n"));
2586 full = hp->h_name;
2588 if (!full) {
2589 full = hp->h_name;
2592 DEBUG(10,("name_to_fqdn: lookup for %s -> %s.\n", name, full));
2593 fstrcpy(fqdn, full);
2594 return true;
2597 /**********************************************************************
2598 Append a DATA_BLOB to a talloc'ed object
2599 ***********************************************************************/
2601 void *talloc_append_blob(TALLOC_CTX *mem_ctx, void *buf, DATA_BLOB blob)
2603 size_t old_size = 0;
2604 char *result;
2606 if (blob.length == 0) {
2607 return buf;
2610 if (buf != NULL) {
2611 old_size = talloc_get_size(buf);
2614 result = (char *)TALLOC_REALLOC(mem_ctx, buf, old_size + blob.length);
2615 if (result == NULL) {
2616 return NULL;
2619 memcpy(result + old_size, blob.data, blob.length);
2620 return result;
2623 uint32 map_share_mode_to_deny_mode(uint32 share_access, uint32 private_options)
2625 switch (share_access & ~FILE_SHARE_DELETE) {
2626 case FILE_SHARE_NONE:
2627 return DENY_ALL;
2628 case FILE_SHARE_READ:
2629 return DENY_WRITE;
2630 case FILE_SHARE_WRITE:
2631 return DENY_READ;
2632 case FILE_SHARE_READ|FILE_SHARE_WRITE:
2633 return DENY_NONE;
2635 if (private_options & NTCREATEX_OPTIONS_PRIVATE_DENY_DOS) {
2636 return DENY_DOS;
2637 } else if (private_options & NTCREATEX_OPTIONS_PRIVATE_DENY_FCB) {
2638 return DENY_FCB;
2641 return (uint32)-1;
2644 pid_t procid_to_pid(const struct server_id *proc)
2646 return proc->pid;
2649 static uint32 my_vnn = NONCLUSTER_VNN;
2651 void set_my_vnn(uint32 vnn)
2653 DEBUG(10, ("vnn pid %d = %u\n", (int)sys_getpid(), (unsigned int)vnn));
2654 my_vnn = vnn;
2657 uint32 get_my_vnn(void)
2659 return my_vnn;
2662 struct server_id pid_to_procid(pid_t pid)
2664 struct server_id result;
2665 result.pid = pid;
2666 #ifdef CLUSTER_SUPPORT
2667 result.vnn = my_vnn;
2668 #endif
2669 return result;
2672 struct server_id procid_self(void)
2674 return pid_to_procid(sys_getpid());
2677 struct server_id server_id_self(void)
2679 return procid_self();
2682 bool procid_equal(const struct server_id *p1, const struct server_id *p2)
2684 if (p1->pid != p2->pid)
2685 return False;
2686 #ifdef CLUSTER_SUPPORT
2687 if (p1->vnn != p2->vnn)
2688 return False;
2689 #endif
2690 return True;
2693 bool cluster_id_equal(const struct server_id *id1,
2694 const struct server_id *id2)
2696 return procid_equal(id1, id2);
2699 bool procid_is_me(const struct server_id *pid)
2701 if (pid->pid != sys_getpid())
2702 return False;
2703 #ifdef CLUSTER_SUPPORT
2704 if (pid->vnn != my_vnn)
2705 return False;
2706 #endif
2707 return True;
2710 struct server_id interpret_pid(const char *pid_string)
2712 struct server_id result;
2713 int pid;
2714 #ifdef CLUSTER_SUPPORT
2715 unsigned int vnn;
2716 if (sscanf(pid_string, "%u:%d", &vnn, &pid) == 2) {
2717 result.vnn = vnn;
2718 result.pid = pid;
2720 else if (sscanf(pid_string, "%d", &pid) == 1) {
2721 result.vnn = get_my_vnn();
2722 result.pid = pid;
2724 else {
2725 result.vnn = NONCLUSTER_VNN;
2726 result.pid = -1;
2728 #else
2729 if (sscanf(pid_string, "%d", &pid) != 1) {
2730 result.pid = -1;
2731 } else {
2732 result.pid = pid;
2734 #endif
2735 /* Assigning to result.pid may have overflowed
2736 Map negative pid to -1: i.e. error */
2737 if (result.pid < 0) {
2738 result.pid = -1;
2740 return result;
2743 char *procid_str(TALLOC_CTX *mem_ctx, const struct server_id *pid)
2745 #ifdef CLUSTER_SUPPORT
2746 if (pid->vnn == NONCLUSTER_VNN) {
2747 return talloc_asprintf(mem_ctx,
2748 "%d",
2749 (int)pid->pid);
2751 else {
2752 return talloc_asprintf(mem_ctx,
2753 "%u:%d",
2754 (unsigned)pid->vnn,
2755 (int)pid->pid);
2757 #else
2758 return talloc_asprintf(mem_ctx,
2759 "%d",
2760 (int)pid->pid);
2761 #endif
2764 char *procid_str_static(const struct server_id *pid)
2766 return procid_str(talloc_tos(), pid);
2769 bool procid_valid(const struct server_id *pid)
2771 return (pid->pid != -1);
2774 bool procid_is_local(const struct server_id *pid)
2776 #ifdef CLUSTER_SUPPORT
2777 return pid->vnn == my_vnn;
2778 #else
2779 return True;
2780 #endif
2783 int this_is_smp(void)
2785 #if defined(HAVE_SYSCONF)
2787 #if defined(SYSCONF_SC_NPROC_ONLN)
2788 return (sysconf(_SC_NPROC_ONLN) > 1) ? 1 : 0;
2789 #elif defined(SYSCONF_SC_NPROCESSORS_ONLN)
2790 return (sysconf(_SC_NPROCESSORS_ONLN) > 1) ? 1 : 0;
2791 #else
2792 return 0;
2793 #endif
2795 #else
2796 return 0;
2797 #endif
2800 /****************************************************************
2801 Check if offset/length fit into bufsize. Should probably be
2802 merged with is_offset_safe, but this would require a rewrite
2803 of lanman.c. Later :-)
2804 ****************************************************************/
2806 bool trans_oob(uint32_t bufsize, uint32_t offset, uint32_t length)
2808 if ((offset + length < offset) || (offset + length < length)) {
2809 /* wrap */
2810 return true;
2812 if ((offset > bufsize) || (offset + length > bufsize)) {
2813 /* overflow */
2814 return true;
2816 return false;
2819 /****************************************************************
2820 Check if an offset into a buffer is safe.
2821 If this returns True it's safe to indirect into the byte at
2822 pointer ptr+off.
2823 ****************************************************************/
2825 bool is_offset_safe(const char *buf_base, size_t buf_len, char *ptr, size_t off)
2827 const char *end_base = buf_base + buf_len;
2828 char *end_ptr = ptr + off;
2830 if (!buf_base || !ptr) {
2831 return False;
2834 if (end_base < buf_base || end_ptr < ptr) {
2835 return False; /* wrap. */
2838 if (end_ptr < end_base) {
2839 return True;
2841 return False;
2844 /****************************************************************
2845 Return a safe pointer into a buffer, or NULL.
2846 ****************************************************************/
2848 char *get_safe_ptr(const char *buf_base, size_t buf_len, char *ptr, size_t off)
2850 return is_offset_safe(buf_base, buf_len, ptr, off) ?
2851 ptr + off : NULL;
2854 /****************************************************************
2855 Return a safe pointer into a string within a buffer, or NULL.
2856 ****************************************************************/
2858 char *get_safe_str_ptr(const char *buf_base, size_t buf_len, char *ptr, size_t off)
2860 if (!is_offset_safe(buf_base, buf_len, ptr, off)) {
2861 return NULL;
2863 /* Check if a valid string exists at this offset. */
2864 if (skip_string(buf_base,buf_len, ptr + off) == NULL) {
2865 return NULL;
2867 return ptr + off;
2870 /****************************************************************
2871 Return an SVAL at a pointer, or failval if beyond the end.
2872 ****************************************************************/
2874 int get_safe_SVAL(const char *buf_base, size_t buf_len, char *ptr, size_t off, int failval)
2877 * Note we use off+1 here, not off+2 as SVAL accesses ptr[0] and ptr[1],
2878 * NOT ptr[2].
2880 if (!is_offset_safe(buf_base, buf_len, ptr, off+1)) {
2881 return failval;
2883 return SVAL(ptr,off);
2886 /****************************************************************
2887 Return an IVAL at a pointer, or failval if beyond the end.
2888 ****************************************************************/
2890 int get_safe_IVAL(const char *buf_base, size_t buf_len, char *ptr, size_t off, int failval)
2893 * Note we use off+3 here, not off+4 as IVAL accesses
2894 * ptr[0] ptr[1] ptr[2] ptr[3] NOT ptr[4].
2896 if (!is_offset_safe(buf_base, buf_len, ptr, off+3)) {
2897 return failval;
2899 return IVAL(ptr,off);
2902 /****************************************************************
2903 Split DOM\user into DOM and user. Do not mix with winbind variants of that
2904 call (they take care of winbind separator and other winbind specific settings).
2905 ****************************************************************/
2907 void split_domain_user(TALLOC_CTX *mem_ctx,
2908 const char *full_name,
2909 char **domain,
2910 char **user)
2912 const char *p = NULL;
2914 p = strchr_m(full_name, '\\');
2916 if (p != NULL) {
2917 *domain = talloc_strndup(mem_ctx, full_name,
2918 PTR_DIFF(p, full_name));
2919 *user = talloc_strdup(mem_ctx, p+1);
2920 } else {
2921 *domain = talloc_strdup(mem_ctx, "");
2922 *user = talloc_strdup(mem_ctx, full_name);
2926 #if 0
2928 Disable these now we have checked all code paths and ensured
2929 NULL returns on zero request. JRA.
2931 /****************************************************************
2932 talloc wrapper functions that guarentee a null pointer return
2933 if size == 0.
2934 ****************************************************************/
2936 #ifndef MAX_TALLOC_SIZE
2937 #define MAX_TALLOC_SIZE 0x10000000
2938 #endif
2941 * talloc and zero memory.
2942 * - returns NULL if size is zero.
2945 void *_talloc_zero_zeronull(const void *ctx, size_t size, const char *name)
2947 void *p;
2949 if (size == 0) {
2950 return NULL;
2953 p = talloc_named_const(ctx, size, name);
2955 if (p) {
2956 memset(p, '\0', size);
2959 return p;
2963 * memdup with a talloc.
2964 * - returns NULL if size is zero.
2967 void *_talloc_memdup_zeronull(const void *t, const void *p, size_t size, const char *name)
2969 void *newp;
2971 if (size == 0) {
2972 return NULL;
2975 newp = talloc_named_const(t, size, name);
2976 if (newp) {
2977 memcpy(newp, p, size);
2980 return newp;
2984 * alloc an array, checking for integer overflow in the array size.
2985 * - returns NULL if count or el_size are zero.
2988 void *_talloc_array_zeronull(const void *ctx, size_t el_size, unsigned count, const char *name)
2990 if (count >= MAX_TALLOC_SIZE/el_size) {
2991 return NULL;
2994 if (el_size == 0 || count == 0) {
2995 return NULL;
2998 return talloc_named_const(ctx, el_size * count, name);
3002 * alloc an zero array, checking for integer overflow in the array size
3003 * - returns NULL if count or el_size are zero.
3006 void *_talloc_zero_array_zeronull(const void *ctx, size_t el_size, unsigned count, const char *name)
3008 if (count >= MAX_TALLOC_SIZE/el_size) {
3009 return NULL;
3012 if (el_size == 0 || count == 0) {
3013 return NULL;
3016 return _talloc_zero(ctx, el_size * count, name);
3020 * Talloc wrapper that returns NULL if size == 0.
3022 void *talloc_zeronull(const void *context, size_t size, const char *name)
3024 if (size == 0) {
3025 return NULL;
3027 return talloc_named_const(context, size, name);
3029 #endif
3031 bool is_valid_policy_hnd(const struct policy_handle *hnd)
3033 struct policy_handle tmp;
3034 ZERO_STRUCT(tmp);
3035 return (memcmp(&tmp, hnd, sizeof(tmp)) != 0);
3038 bool policy_hnd_equal(const struct policy_handle *hnd1,
3039 const struct policy_handle *hnd2)
3041 if (!hnd1 || !hnd2) {
3042 return false;
3045 return (memcmp(hnd1, hnd2, sizeof(*hnd1)) == 0);
3048 /****************************************************************
3049 strip off leading '\\' from a hostname
3050 ****************************************************************/
3052 const char *strip_hostname(const char *s)
3054 if (!s) {
3055 return NULL;
3058 if (strlen_m(s) < 3) {
3059 return s;
3062 if (s[0] == '\\') s++;
3063 if (s[0] == '\\') s++;
3065 return s;
3068 bool tevent_req_poll_ntstatus(struct tevent_req *req,
3069 struct tevent_context *ev,
3070 NTSTATUS *status)
3072 bool ret = tevent_req_poll(req, ev);
3073 if (!ret) {
3074 *status = map_nt_error_from_unix(errno);
3076 return ret;
3079 /*******************************************************************
3080 Return True if the filename is one of the special executable types.
3081 ********************************************************************/
3083 bool is_executable(const char *fname)
3085 if ((fname = strrchr_m(fname,'.'))) {
3086 if (strequal(fname,".com") ||
3087 strequal(fname,".dll") ||
3088 strequal(fname,".exe") ||
3089 strequal(fname,".sym")) {
3090 return True;
3093 return False;
3096 /****************************************************************************
3097 Open a file with a share mode - old openX method - map into NTCreate.
3098 ****************************************************************************/
3100 bool map_open_params_to_ntcreate(const char *smb_base_fname,
3101 int deny_mode, int open_func,
3102 uint32 *paccess_mask,
3103 uint32 *pshare_mode,
3104 uint32 *pcreate_disposition,
3105 uint32 *pcreate_options)
3107 uint32 access_mask;
3108 uint32 share_mode;
3109 uint32 create_disposition;
3110 uint32 create_options = FILE_NON_DIRECTORY_FILE;
3112 DEBUG(10,("map_open_params_to_ntcreate: fname = %s, deny_mode = 0x%x, "
3113 "open_func = 0x%x\n",
3114 smb_base_fname, (unsigned int)deny_mode,
3115 (unsigned int)open_func ));
3117 /* Create the NT compatible access_mask. */
3118 switch (GET_OPENX_MODE(deny_mode)) {
3119 case DOS_OPEN_EXEC: /* Implies read-only - used to be FILE_READ_DATA */
3120 case DOS_OPEN_RDONLY:
3121 access_mask = FILE_GENERIC_READ;
3122 break;
3123 case DOS_OPEN_WRONLY:
3124 access_mask = FILE_GENERIC_WRITE;
3125 break;
3126 case DOS_OPEN_RDWR:
3127 case DOS_OPEN_FCB:
3128 access_mask = FILE_GENERIC_READ|FILE_GENERIC_WRITE;
3129 break;
3130 default:
3131 DEBUG(10,("map_open_params_to_ntcreate: bad open mode = 0x%x\n",
3132 (unsigned int)GET_OPENX_MODE(deny_mode)));
3133 return False;
3136 /* Create the NT compatible create_disposition. */
3137 switch (open_func) {
3138 case OPENX_FILE_EXISTS_FAIL|OPENX_FILE_CREATE_IF_NOT_EXIST:
3139 create_disposition = FILE_CREATE;
3140 break;
3142 case OPENX_FILE_EXISTS_OPEN:
3143 create_disposition = FILE_OPEN;
3144 break;
3146 case OPENX_FILE_EXISTS_OPEN|OPENX_FILE_CREATE_IF_NOT_EXIST:
3147 create_disposition = FILE_OPEN_IF;
3148 break;
3150 case OPENX_FILE_EXISTS_TRUNCATE:
3151 create_disposition = FILE_OVERWRITE;
3152 break;
3154 case OPENX_FILE_EXISTS_TRUNCATE|OPENX_FILE_CREATE_IF_NOT_EXIST:
3155 create_disposition = FILE_OVERWRITE_IF;
3156 break;
3158 default:
3159 /* From samba4 - to be confirmed. */
3160 if (GET_OPENX_MODE(deny_mode) == DOS_OPEN_EXEC) {
3161 create_disposition = FILE_CREATE;
3162 break;
3164 DEBUG(10,("map_open_params_to_ntcreate: bad "
3165 "open_func 0x%x\n", (unsigned int)open_func));
3166 return False;
3169 /* Create the NT compatible share modes. */
3170 switch (GET_DENY_MODE(deny_mode)) {
3171 case DENY_ALL:
3172 share_mode = FILE_SHARE_NONE;
3173 break;
3175 case DENY_WRITE:
3176 share_mode = FILE_SHARE_READ;
3177 break;
3179 case DENY_READ:
3180 share_mode = FILE_SHARE_WRITE;
3181 break;
3183 case DENY_NONE:
3184 share_mode = FILE_SHARE_READ|FILE_SHARE_WRITE;
3185 break;
3187 case DENY_DOS:
3188 create_options |= NTCREATEX_OPTIONS_PRIVATE_DENY_DOS;
3189 if (is_executable(smb_base_fname)) {
3190 share_mode = FILE_SHARE_READ|FILE_SHARE_WRITE;
3191 } else {
3192 if (GET_OPENX_MODE(deny_mode) == DOS_OPEN_RDONLY) {
3193 share_mode = FILE_SHARE_READ;
3194 } else {
3195 share_mode = FILE_SHARE_NONE;
3198 break;
3200 case DENY_FCB:
3201 create_options |= NTCREATEX_OPTIONS_PRIVATE_DENY_FCB;
3202 share_mode = FILE_SHARE_NONE;
3203 break;
3205 default:
3206 DEBUG(10,("map_open_params_to_ntcreate: bad deny_mode 0x%x\n",
3207 (unsigned int)GET_DENY_MODE(deny_mode) ));
3208 return False;
3211 DEBUG(10,("map_open_params_to_ntcreate: file %s, access_mask = 0x%x, "
3212 "share_mode = 0x%x, create_disposition = 0x%x, "
3213 "create_options = 0x%x\n",
3214 smb_base_fname,
3215 (unsigned int)access_mask,
3216 (unsigned int)share_mode,
3217 (unsigned int)create_disposition,
3218 (unsigned int)create_options ));
3220 if (paccess_mask) {
3221 *paccess_mask = access_mask;
3223 if (pshare_mode) {
3224 *pshare_mode = share_mode;
3226 if (pcreate_disposition) {
3227 *pcreate_disposition = create_disposition;
3229 if (pcreate_options) {
3230 *pcreate_options = create_options;
3233 return True;