Fix inode generation so nautilus can count total dir size correctly
[Samba.git] / source3 / lib / util.c
blob46be349a2c22d0253642eaecc72ea96e35f078af
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 value = talloc_strdup(ctx, nis_result);
1316 if (!value) {
1317 return NULL;
1319 value = strip_mount_options(ctx, value);
1320 } else if(nis_error == YPERR_KEY) {
1321 DEBUG(3, ("YP Key not found: while looking up \"%s\" in map \"%s\"\n",
1322 user_name, nis_map));
1323 DEBUG(3, ("using defaults for server and home directory\n"));
1324 } else {
1325 DEBUG(3, ("YP Error: \"%s\" while looking up \"%s\" in map \"%s\"\n",
1326 yperr_string(nis_error), user_name, nis_map));
1329 if (value) {
1330 DEBUG(4, ("YP Lookup: %s resulted in %s\n", user_name, value));
1332 return value;
1334 #endif /* WITH_NISPLUS_HOME */
1335 #endif
1337 /****************************************************************************
1338 Check if a process exists. Does this work on all unixes?
1339 ****************************************************************************/
1341 bool process_exists(const struct server_id pid)
1343 if (procid_is_me(&pid)) {
1344 return True;
1347 if (procid_is_local(&pid)) {
1348 return (kill(pid.pid,0) == 0 || errno != ESRCH);
1351 #ifdef CLUSTER_SUPPORT
1352 return ctdbd_process_exists(messaging_ctdbd_connection(), pid.vnn,
1353 pid.pid);
1354 #else
1355 return False;
1356 #endif
1359 /*******************************************************************
1360 Convert a uid into a user name.
1361 ********************************************************************/
1363 const char *uidtoname(uid_t uid)
1365 TALLOC_CTX *ctx = talloc_tos();
1366 char *name = NULL;
1367 struct passwd *pass = NULL;
1369 pass = getpwuid_alloc(ctx,uid);
1370 if (pass) {
1371 name = talloc_strdup(ctx,pass->pw_name);
1372 TALLOC_FREE(pass);
1373 } else {
1374 name = talloc_asprintf(ctx,
1375 "%ld",
1376 (long int)uid);
1378 return name;
1381 /*******************************************************************
1382 Convert a gid into a group name.
1383 ********************************************************************/
1385 char *gidtoname(gid_t gid)
1387 struct group *grp;
1389 grp = getgrgid(gid);
1390 if (grp) {
1391 return talloc_strdup(talloc_tos(), grp->gr_name);
1393 else {
1394 return talloc_asprintf(talloc_tos(),
1395 "%d",
1396 (int)gid);
1400 /*******************************************************************
1401 Convert a user name into a uid.
1402 ********************************************************************/
1404 uid_t nametouid(const char *name)
1406 struct passwd *pass;
1407 char *p;
1408 uid_t u;
1410 pass = Get_Pwnam_alloc(talloc_autofree_context(), name);
1411 if (pass) {
1412 u = pass->pw_uid;
1413 TALLOC_FREE(pass);
1414 return u;
1417 u = (uid_t)strtol(name, &p, 0);
1418 if ((p != name) && (*p == '\0'))
1419 return u;
1421 return (uid_t)-1;
1424 /*******************************************************************
1425 Convert a name to a gid_t if possible. Return -1 if not a group.
1426 ********************************************************************/
1428 gid_t nametogid(const char *name)
1430 struct group *grp;
1431 char *p;
1432 gid_t g;
1434 g = (gid_t)strtol(name, &p, 0);
1435 if ((p != name) && (*p == '\0'))
1436 return g;
1438 grp = sys_getgrnam(name);
1439 if (grp)
1440 return(grp->gr_gid);
1441 return (gid_t)-1;
1444 /*******************************************************************
1445 Something really nasty happened - panic !
1446 ********************************************************************/
1448 void smb_panic(const char *const why)
1450 char *cmd;
1451 int result;
1453 #ifdef DEVELOPER
1456 if (global_clobber_region_function) {
1457 DEBUG(0,("smb_panic: clobber_region() last called from [%s(%u)]\n",
1458 global_clobber_region_function,
1459 global_clobber_region_line));
1462 #endif
1464 DEBUG(0,("PANIC (pid %llu): %s\n",
1465 (unsigned long long)sys_getpid(), why));
1466 log_stack_trace();
1468 cmd = lp_panic_action();
1469 if (cmd && *cmd) {
1470 DEBUG(0, ("smb_panic(): calling panic action [%s]\n", cmd));
1471 result = system(cmd);
1473 if (result == -1)
1474 DEBUG(0, ("smb_panic(): fork failed in panic action: %s\n",
1475 strerror(errno)));
1476 else
1477 DEBUG(0, ("smb_panic(): action returned status %d\n",
1478 WEXITSTATUS(result)));
1481 dump_core();
1484 /*******************************************************************
1485 Print a backtrace of the stack to the debug log. This function
1486 DELIBERATELY LEAKS MEMORY. The expectation is that you should
1487 exit shortly after calling it.
1488 ********************************************************************/
1490 #ifdef HAVE_LIBUNWIND_H
1491 #include <libunwind.h>
1492 #endif
1494 #ifdef HAVE_EXECINFO_H
1495 #include <execinfo.h>
1496 #endif
1498 #ifdef HAVE_LIBEXC_H
1499 #include <libexc.h>
1500 #endif
1502 void log_stack_trace(void)
1504 #ifdef HAVE_LIBUNWIND
1505 /* Try to use libunwind before any other technique since on ia64
1506 * libunwind correctly walks the stack in more circumstances than
1507 * backtrace.
1509 unw_cursor_t cursor;
1510 unw_context_t uc;
1511 unsigned i = 0;
1513 char procname[256];
1514 unw_word_t ip, sp, off;
1516 procname[sizeof(procname) - 1] = '\0';
1518 if (unw_getcontext(&uc) != 0) {
1519 goto libunwind_failed;
1522 if (unw_init_local(&cursor, &uc) != 0) {
1523 goto libunwind_failed;
1526 DEBUG(0, ("BACKTRACE:\n"));
1528 do {
1529 ip = sp = 0;
1530 unw_get_reg(&cursor, UNW_REG_IP, &ip);
1531 unw_get_reg(&cursor, UNW_REG_SP, &sp);
1533 switch (unw_get_proc_name(&cursor,
1534 procname, sizeof(procname) - 1, &off) ) {
1535 case 0:
1536 /* Name found. */
1537 case -UNW_ENOMEM:
1538 /* Name truncated. */
1539 DEBUGADD(0, (" #%u %s + %#llx [ip=%#llx] [sp=%#llx]\n",
1540 i, procname, (long long)off,
1541 (long long)ip, (long long) sp));
1542 break;
1543 default:
1544 /* case -UNW_ENOINFO: */
1545 /* case -UNW_EUNSPEC: */
1546 /* No symbol name found. */
1547 DEBUGADD(0, (" #%u %s [ip=%#llx] [sp=%#llx]\n",
1548 i, "<unknown symbol>",
1549 (long long)ip, (long long) sp));
1551 ++i;
1552 } while (unw_step(&cursor) > 0);
1554 return;
1556 libunwind_failed:
1557 DEBUG(0, ("unable to produce a stack trace with libunwind\n"));
1559 #elif HAVE_BACKTRACE_SYMBOLS
1560 void *backtrace_stack[BACKTRACE_STACK_SIZE];
1561 size_t backtrace_size;
1562 char **backtrace_strings;
1564 /* get the backtrace (stack frames) */
1565 backtrace_size = backtrace(backtrace_stack,BACKTRACE_STACK_SIZE);
1566 backtrace_strings = backtrace_symbols(backtrace_stack, backtrace_size);
1568 DEBUG(0, ("BACKTRACE: %lu stack frames:\n",
1569 (unsigned long)backtrace_size));
1571 if (backtrace_strings) {
1572 int i;
1574 for (i = 0; i < backtrace_size; i++)
1575 DEBUGADD(0, (" #%u %s\n", i, backtrace_strings[i]));
1577 /* Leak the backtrace_strings, rather than risk what free() might do */
1580 #elif HAVE_LIBEXC
1582 /* The IRIX libexc library provides an API for unwinding the stack. See
1583 * libexc(3) for details. Apparantly trace_back_stack leaks memory, but
1584 * since we are about to abort anyway, it hardly matters.
1587 #define NAMESIZE 32 /* Arbitrary */
1589 __uint64_t addrs[BACKTRACE_STACK_SIZE];
1590 char * names[BACKTRACE_STACK_SIZE];
1591 char namebuf[BACKTRACE_STACK_SIZE * NAMESIZE];
1593 int i;
1594 int levels;
1596 ZERO_ARRAY(addrs);
1597 ZERO_ARRAY(names);
1598 ZERO_ARRAY(namebuf);
1600 /* We need to be root so we can open our /proc entry to walk
1601 * our stack. It also helps when we want to dump core.
1603 become_root();
1605 for (i = 0; i < BACKTRACE_STACK_SIZE; i++) {
1606 names[i] = namebuf + (i * NAMESIZE);
1609 levels = trace_back_stack(0, addrs, names,
1610 BACKTRACE_STACK_SIZE, NAMESIZE - 1);
1612 DEBUG(0, ("BACKTRACE: %d stack frames:\n", levels));
1613 for (i = 0; i < levels; i++) {
1614 DEBUGADD(0, (" #%d 0x%llx %s\n", i, addrs[i], names[i]));
1616 #undef NAMESIZE
1618 #else
1619 DEBUG(0, ("unable to produce a stack trace on this platform\n"));
1620 #endif
1623 /*******************************************************************
1624 A readdir wrapper which just returns the file name.
1625 ********************************************************************/
1627 const char *readdirname(SMB_STRUCT_DIR *p)
1629 SMB_STRUCT_DIRENT *ptr;
1630 char *dname;
1632 if (!p)
1633 return(NULL);
1635 ptr = (SMB_STRUCT_DIRENT *)sys_readdir(p);
1636 if (!ptr)
1637 return(NULL);
1639 dname = ptr->d_name;
1641 #ifdef NEXT2
1642 if (telldir(p) < 0)
1643 return(NULL);
1644 #endif
1646 #ifdef HAVE_BROKEN_READDIR_NAME
1647 /* using /usr/ucb/cc is BAD */
1648 dname = dname - 2;
1649 #endif
1651 return talloc_strdup(talloc_tos(), dname);
1654 /*******************************************************************
1655 Utility function used to decide if the last component
1656 of a path matches a (possibly wildcarded) entry in a namelist.
1657 ********************************************************************/
1659 bool is_in_path(const char *name, name_compare_entry *namelist, bool case_sensitive)
1661 const char *last_component;
1663 /* if we have no list it's obviously not in the path */
1664 if((namelist == NULL ) || ((namelist != NULL) && (namelist[0].name == NULL))) {
1665 return False;
1668 DEBUG(8, ("is_in_path: %s\n", name));
1670 /* Get the last component of the unix name. */
1671 last_component = strrchr_m(name, '/');
1672 if (!last_component) {
1673 last_component = name;
1674 } else {
1675 last_component++; /* Go past '/' */
1678 for(; namelist->name != NULL; namelist++) {
1679 if(namelist->is_wild) {
1680 if (mask_match(last_component, namelist->name, case_sensitive)) {
1681 DEBUG(8,("is_in_path: mask match succeeded\n"));
1682 return True;
1684 } else {
1685 if((case_sensitive && (strcmp(last_component, namelist->name) == 0))||
1686 (!case_sensitive && (StrCaseCmp(last_component, namelist->name) == 0))) {
1687 DEBUG(8,("is_in_path: match succeeded\n"));
1688 return True;
1692 DEBUG(8,("is_in_path: match not found\n"));
1693 return False;
1696 /*******************************************************************
1697 Strip a '/' separated list into an array of
1698 name_compare_enties structures suitable for
1699 passing to is_in_path(). We do this for
1700 speed so we can pre-parse all the names in the list
1701 and don't do it for each call to is_in_path().
1702 namelist is modified here and is assumed to be
1703 a copy owned by the caller.
1704 We also check if the entry contains a wildcard to
1705 remove a potentially expensive call to mask_match
1706 if possible.
1707 ********************************************************************/
1709 void set_namearray(name_compare_entry **ppname_array, const char *namelist)
1711 char *name_end;
1712 char *nameptr = (char *)namelist;
1713 int num_entries = 0;
1714 int i;
1716 (*ppname_array) = NULL;
1718 if((nameptr == NULL ) || ((nameptr != NULL) && (*nameptr == '\0')))
1719 return;
1721 /* We need to make two passes over the string. The
1722 first to count the number of elements, the second
1723 to split it.
1726 while(*nameptr) {
1727 if ( *nameptr == '/' ) {
1728 /* cope with multiple (useless) /s) */
1729 nameptr++;
1730 continue;
1732 /* anything left? */
1733 if ( *nameptr == '\0' )
1734 break;
1736 /* find the next '/' or consume remaining */
1737 name_end = strchr_m(nameptr, '/');
1738 if (name_end == NULL)
1739 name_end = (char *)nameptr + strlen(nameptr);
1741 /* next segment please */
1742 nameptr = name_end + 1;
1743 num_entries++;
1746 if(num_entries == 0)
1747 return;
1749 if(( (*ppname_array) = SMB_MALLOC_ARRAY(name_compare_entry, num_entries + 1)) == NULL) {
1750 DEBUG(0,("set_namearray: malloc fail\n"));
1751 return;
1754 /* Now copy out the names */
1755 nameptr = (char *)namelist;
1756 i = 0;
1757 while(*nameptr) {
1758 if ( *nameptr == '/' ) {
1759 /* cope with multiple (useless) /s) */
1760 nameptr++;
1761 continue;
1763 /* anything left? */
1764 if ( *nameptr == '\0' )
1765 break;
1767 /* find the next '/' or consume remaining */
1768 name_end = strchr_m(nameptr, '/');
1769 if (name_end)
1770 *name_end = '\0';
1771 else
1772 name_end = nameptr + strlen(nameptr);
1774 (*ppname_array)[i].is_wild = ms_has_wild(nameptr);
1775 if(((*ppname_array)[i].name = SMB_STRDUP(nameptr)) == NULL) {
1776 DEBUG(0,("set_namearray: malloc fail (1)\n"));
1777 return;
1780 /* next segment please */
1781 nameptr = name_end + 1;
1782 i++;
1785 (*ppname_array)[i].name = NULL;
1787 return;
1790 /****************************************************************************
1791 Routine to free a namearray.
1792 ****************************************************************************/
1794 void free_namearray(name_compare_entry *name_array)
1796 int i;
1798 if(name_array == NULL)
1799 return;
1801 for(i=0; name_array[i].name!=NULL; i++)
1802 SAFE_FREE(name_array[i].name);
1803 SAFE_FREE(name_array);
1806 #undef DBGC_CLASS
1807 #define DBGC_CLASS DBGC_LOCKING
1809 /****************************************************************************
1810 Simple routine to query existing file locks. Cruft in NFS and 64->32 bit mapping
1811 is dealt with in posix.c
1812 Returns True if we have information regarding this lock region (and returns
1813 F_UNLCK in *ptype if the region is unlocked). False if the call failed.
1814 ****************************************************************************/
1816 bool fcntl_getlock(int fd, SMB_OFF_T *poffset, SMB_OFF_T *pcount, int *ptype, pid_t *ppid)
1818 SMB_STRUCT_FLOCK lock;
1819 int ret;
1821 DEBUG(8,("fcntl_getlock fd=%d offset=%.0f count=%.0f type=%d\n",
1822 fd,(double)*poffset,(double)*pcount,*ptype));
1824 lock.l_type = *ptype;
1825 lock.l_whence = SEEK_SET;
1826 lock.l_start = *poffset;
1827 lock.l_len = *pcount;
1828 lock.l_pid = 0;
1830 ret = sys_fcntl_ptr(fd,SMB_F_GETLK,&lock);
1832 if (ret == -1) {
1833 int sav = errno;
1834 DEBUG(3,("fcntl_getlock: lock request failed at offset %.0f count %.0f type %d (%s)\n",
1835 (double)*poffset,(double)*pcount,*ptype,strerror(errno)));
1836 errno = sav;
1837 return False;
1840 *ptype = lock.l_type;
1841 *poffset = lock.l_start;
1842 *pcount = lock.l_len;
1843 *ppid = lock.l_pid;
1845 DEBUG(3,("fcntl_getlock: fd %d is returned info %d pid %u\n",
1846 fd, (int)lock.l_type, (unsigned int)lock.l_pid));
1847 return True;
1850 #undef DBGC_CLASS
1851 #define DBGC_CLASS DBGC_ALL
1853 /*******************************************************************
1854 Is the name specified one of my netbios names.
1855 Returns true if it is equal, false otherwise.
1856 ********************************************************************/
1858 bool is_myname(const char *s)
1860 int n;
1861 bool ret = False;
1863 for (n=0; my_netbios_names(n); n++) {
1864 if (strequal(my_netbios_names(n), s)) {
1865 ret=True;
1866 break;
1869 DEBUG(8, ("is_myname(\"%s\") returns %d\n", s, ret));
1870 return(ret);
1873 /*******************************************************************
1874 Is the name specified our workgroup/domain.
1875 Returns true if it is equal, false otherwise.
1876 ********************************************************************/
1878 bool is_myworkgroup(const char *s)
1880 bool ret = False;
1882 if (strequal(s, lp_workgroup())) {
1883 ret=True;
1886 DEBUG(8, ("is_myworkgroup(\"%s\") returns %d\n", s, ret));
1887 return(ret);
1890 /*******************************************************************
1891 we distinguish between 2K and XP by the "Native Lan Manager" string
1892 WinXP => "Windows 2002 5.1"
1893 WinXP 64bit => "Windows XP 5.2"
1894 Win2k => "Windows 2000 5.0"
1895 NT4 => "Windows NT 4.0"
1896 Win9x => "Windows 4.0"
1897 Windows 2003 doesn't set the native lan manager string but
1898 they do set the domain to "Windows 2003 5.2" (probably a bug).
1899 ********************************************************************/
1901 void ra_lanman_string( const char *native_lanman )
1903 if ( strcmp( native_lanman, "Windows 2002 5.1" ) == 0 )
1904 set_remote_arch( RA_WINXP );
1905 else if ( strcmp( native_lanman, "Windows XP 5.2" ) == 0 )
1906 set_remote_arch( RA_WINXP64 );
1907 else if ( strcmp( native_lanman, "Windows Server 2003 5.2" ) == 0 )
1908 set_remote_arch( RA_WIN2K3 );
1911 static const char *remote_arch_str;
1913 const char *get_remote_arch_str(void)
1915 if (!remote_arch_str) {
1916 return "UNKNOWN";
1918 return remote_arch_str;
1921 /*******************************************************************
1922 Set the horrid remote_arch string based on an enum.
1923 ********************************************************************/
1925 void set_remote_arch(enum remote_arch_types type)
1927 ra_type = type;
1928 switch( type ) {
1929 case RA_WFWG:
1930 remote_arch_str = "WfWg";
1931 break;
1932 case RA_OS2:
1933 remote_arch_str = "OS2";
1934 break;
1935 case RA_WIN95:
1936 remote_arch_str = "Win95";
1937 break;
1938 case RA_WINNT:
1939 remote_arch_str = "WinNT";
1940 break;
1941 case RA_WIN2K:
1942 remote_arch_str = "Win2K";
1943 break;
1944 case RA_WINXP:
1945 remote_arch_str = "WinXP";
1946 break;
1947 case RA_WINXP64:
1948 remote_arch_str = "WinXP64";
1949 break;
1950 case RA_WIN2K3:
1951 remote_arch_str = "Win2K3";
1952 break;
1953 case RA_VISTA:
1954 remote_arch_str = "Vista";
1955 break;
1956 case RA_SAMBA:
1957 remote_arch_str = "Samba";
1958 break;
1959 case RA_CIFSFS:
1960 remote_arch_str = "CIFSFS";
1961 break;
1962 default:
1963 ra_type = RA_UNKNOWN;
1964 remote_arch_str = "UNKNOWN";
1965 break;
1968 DEBUG(10,("set_remote_arch: Client arch is \'%s\'\n",
1969 remote_arch_str));
1972 /*******************************************************************
1973 Get the remote_arch type.
1974 ********************************************************************/
1976 enum remote_arch_types get_remote_arch(void)
1978 return ra_type;
1981 const char *tab_depth(int level, int depth)
1983 if( CHECK_DEBUGLVL(level) ) {
1984 dbgtext("%*s", depth*4, "");
1986 return "";
1989 /*****************************************************************************
1990 Provide a checksum on a string
1992 Input: s - the null-terminated character string for which the checksum
1993 will be calculated.
1995 Output: The checksum value calculated for s.
1996 *****************************************************************************/
1998 int str_checksum(const char *s)
2000 TDB_DATA key = string_tdb_data(s);
2001 return jenkins_hash(&key);
2004 /*****************************************************************
2005 Zero a memory area then free it. Used to catch bugs faster.
2006 *****************************************************************/
2008 void zero_free(void *p, size_t size)
2010 memset(p, 0, size);
2011 SAFE_FREE(p);
2014 /*****************************************************************
2015 Set our open file limit to a requested max and return the limit.
2016 *****************************************************************/
2018 int set_maxfiles(int requested_max)
2020 #if (defined(HAVE_GETRLIMIT) && defined(RLIMIT_NOFILE))
2021 struct rlimit rlp;
2022 int saved_current_limit;
2024 if(getrlimit(RLIMIT_NOFILE, &rlp)) {
2025 DEBUG(0,("set_maxfiles: getrlimit (1) for RLIMIT_NOFILE failed with error %s\n",
2026 strerror(errno) ));
2027 /* just guess... */
2028 return requested_max;
2032 * Set the fd limit to be real_max_open_files + MAX_OPEN_FUDGEFACTOR to
2033 * account for the extra fd we need
2034 * as well as the log files and standard
2035 * handles etc. Save the limit we want to set in case
2036 * we are running on an OS that doesn't support this limit (AIX)
2037 * which always returns RLIM_INFINITY for rlp.rlim_max.
2040 /* Try raising the hard (max) limit to the requested amount. */
2042 #if defined(RLIM_INFINITY)
2043 if (rlp.rlim_max != RLIM_INFINITY) {
2044 int orig_max = rlp.rlim_max;
2046 if ( rlp.rlim_max < requested_max )
2047 rlp.rlim_max = requested_max;
2049 /* This failing is not an error - many systems (Linux) don't
2050 support our default request of 10,000 open files. JRA. */
2052 if(setrlimit(RLIMIT_NOFILE, &rlp)) {
2053 DEBUG(3,("set_maxfiles: setrlimit for RLIMIT_NOFILE for %d max files failed with error %s\n",
2054 (int)rlp.rlim_max, strerror(errno) ));
2056 /* Set failed - restore original value from get. */
2057 rlp.rlim_max = orig_max;
2060 #endif
2062 /* Now try setting the soft (current) limit. */
2064 saved_current_limit = rlp.rlim_cur = MIN(requested_max,rlp.rlim_max);
2066 if(setrlimit(RLIMIT_NOFILE, &rlp)) {
2067 DEBUG(0,("set_maxfiles: setrlimit for RLIMIT_NOFILE for %d files failed with error %s\n",
2068 (int)rlp.rlim_cur, strerror(errno) ));
2069 /* just guess... */
2070 return saved_current_limit;
2073 if(getrlimit(RLIMIT_NOFILE, &rlp)) {
2074 DEBUG(0,("set_maxfiles: getrlimit (2) for RLIMIT_NOFILE failed with error %s\n",
2075 strerror(errno) ));
2076 /* just guess... */
2077 return saved_current_limit;
2080 #if defined(RLIM_INFINITY)
2081 if(rlp.rlim_cur == RLIM_INFINITY)
2082 return saved_current_limit;
2083 #endif
2085 if((int)rlp.rlim_cur > saved_current_limit)
2086 return saved_current_limit;
2088 return rlp.rlim_cur;
2089 #else /* !defined(HAVE_GETRLIMIT) || !defined(RLIMIT_NOFILE) */
2091 * No way to know - just guess...
2093 return requested_max;
2094 #endif
2097 /*****************************************************************
2098 malloc that aborts with smb_panic on fail or zero size.
2099 *****************************************************************/
2101 void *smb_xmalloc_array(size_t size, unsigned int count)
2103 void *p;
2104 if (size == 0) {
2105 smb_panic("smb_xmalloc_array: called with zero size");
2107 if (count >= MAX_ALLOC_SIZE/size) {
2108 smb_panic("smb_xmalloc_array: alloc size too large");
2110 if ((p = SMB_MALLOC(size*count)) == NULL) {
2111 DEBUG(0, ("smb_xmalloc_array failed to allocate %lu * %lu bytes\n",
2112 (unsigned long)size, (unsigned long)count));
2113 smb_panic("smb_xmalloc_array: malloc failed");
2115 return p;
2119 vasprintf that aborts on malloc fail
2122 int smb_xvasprintf(char **ptr, const char *format, va_list ap)
2124 int n;
2125 va_list ap2;
2127 va_copy(ap2, ap);
2129 n = vasprintf(ptr, format, ap2);
2130 va_end(ap2);
2131 if (n == -1 || ! *ptr) {
2132 smb_panic("smb_xvasprintf: out of memory");
2134 return n;
2137 /*****************************************************************
2138 Get local hostname and cache result.
2139 *****************************************************************/
2141 char *myhostname(void)
2143 static char *ret;
2144 if (ret == NULL) {
2145 /* This is cached forever so
2146 * use talloc_autofree_context() ctx. */
2147 ret = get_myname(talloc_autofree_context());
2149 return ret;
2153 * @brief Returns an absolute path to a file concatenating the provided
2154 * @a rootpath and @a basename
2156 * @param name Filename, relative to @a rootpath
2158 * @retval Pointer to a string containing the full path.
2161 static char *xx_path(const char *name, const char *rootpath)
2163 char *fname = NULL;
2165 fname = talloc_strdup(talloc_tos(), rootpath);
2166 if (!fname) {
2167 return NULL;
2169 trim_string(fname,"","/");
2171 if (!directory_exist(fname)) {
2172 if (!mkdir(fname,0755))
2173 DEBUG(1, ("Unable to create directory %s for file %s. "
2174 "Error was %s\n", fname, name, strerror(errno)));
2177 return talloc_asprintf(talloc_tos(),
2178 "%s/%s",
2179 fname,
2180 name);
2184 * @brief Returns an absolute path to a file in the Samba lock directory.
2186 * @param name File to find, relative to LOCKDIR.
2188 * @retval Pointer to a talloc'ed string containing the full path.
2191 char *lock_path(const char *name)
2193 return xx_path(name, lp_lockdir());
2197 * @brief Returns an absolute path to a file in the Samba pid directory.
2199 * @param name File to find, relative to PIDDIR.
2201 * @retval Pointer to a talloc'ed string containing the full path.
2204 char *pid_path(const char *name)
2206 return xx_path(name, lp_piddir());
2210 * @brief Returns an absolute path to a file in the Samba lib directory.
2212 * @param name File to find, relative to LIBDIR.
2214 * @retval Pointer to a string containing the full path.
2217 char *lib_path(const char *name)
2219 return talloc_asprintf(talloc_tos(), "%s/%s", get_dyn_LIBDIR(), name);
2223 * @brief Returns an absolute path to a file in the Samba modules directory.
2225 * @param name File to find, relative to MODULESDIR.
2227 * @retval Pointer to a string containing the full path.
2230 char *modules_path(const char *name)
2232 return talloc_asprintf(talloc_tos(), "%s/%s", get_dyn_MODULESDIR(), name);
2236 * @brief Returns an absolute path to a file in the Samba data directory.
2238 * @param name File to find, relative to CODEPAGEDIR.
2240 * @retval Pointer to a talloc'ed string containing the full path.
2243 char *data_path(const char *name)
2245 return talloc_asprintf(talloc_tos(), "%s/%s", get_dyn_CODEPAGEDIR(), name);
2249 * @brief Returns an absolute path to a file in the Samba state directory.
2251 * @param name File to find, relative to STATEDIR.
2253 * @retval Pointer to a talloc'ed string containing the full path.
2256 char *state_path(const char *name)
2258 return xx_path(name, lp_statedir());
2262 * @brief Returns an absolute path to a file in the Samba cache directory.
2264 * @param name File to find, relative to CACHEDIR.
2266 * @retval Pointer to a talloc'ed string containing the full path.
2269 char *cache_path(const char *name)
2271 return xx_path(name, lp_cachedir());
2275 * @brief Returns the platform specific shared library extension.
2277 * @retval Pointer to a const char * containing the extension.
2280 const char *shlib_ext(void)
2282 return get_dyn_SHLIBEXT();
2285 /*******************************************************************
2286 Given a filename - get its directory name
2287 ********************************************************************/
2289 bool parent_dirname(TALLOC_CTX *mem_ctx, const char *dir, char **parent,
2290 const char **name)
2292 char *p;
2293 ptrdiff_t len;
2295 p = strrchr_m(dir, '/'); /* Find final '/', if any */
2297 if (p == NULL) {
2298 if (!(*parent = talloc_strdup(mem_ctx, "."))) {
2299 return False;
2301 if (name) {
2302 *name = dir;
2304 return True;
2307 len = p-dir;
2309 if (!(*parent = (char *)TALLOC_MEMDUP(mem_ctx, dir, len+1))) {
2310 return False;
2312 (*parent)[len] = '\0';
2314 if (name) {
2315 *name = p+1;
2317 return True;
2320 /*******************************************************************
2321 Determine if a pattern contains any Microsoft wildcard characters.
2322 *******************************************************************/
2324 bool ms_has_wild(const char *s)
2326 char c;
2328 if (lp_posix_pathnames()) {
2329 /* With posix pathnames no characters are wild. */
2330 return False;
2333 while ((c = *s++)) {
2334 switch (c) {
2335 case '*':
2336 case '?':
2337 case '<':
2338 case '>':
2339 case '"':
2340 return True;
2343 return False;
2346 bool ms_has_wild_w(const smb_ucs2_t *s)
2348 smb_ucs2_t c;
2349 if (!s) return False;
2350 while ((c = *s++)) {
2351 switch (c) {
2352 case UCS2_CHAR('*'):
2353 case UCS2_CHAR('?'):
2354 case UCS2_CHAR('<'):
2355 case UCS2_CHAR('>'):
2356 case UCS2_CHAR('"'):
2357 return True;
2360 return False;
2363 /*******************************************************************
2364 A wrapper that handles case sensitivity and the special handling
2365 of the ".." name.
2366 *******************************************************************/
2368 bool mask_match(const char *string, const char *pattern, bool is_case_sensitive)
2370 if (strcmp(string,"..") == 0)
2371 string = ".";
2372 if (strcmp(pattern,".") == 0)
2373 return False;
2375 return ms_fnmatch(pattern, string, Protocol <= PROTOCOL_LANMAN2, is_case_sensitive) == 0;
2378 /*******************************************************************
2379 A wrapper that handles case sensitivity and the special handling
2380 of the ".." name. Varient that is only called by old search code which requires
2381 pattern translation.
2382 *******************************************************************/
2384 bool mask_match_search(const char *string, const char *pattern, bool is_case_sensitive)
2386 if (strcmp(string,"..") == 0)
2387 string = ".";
2388 if (strcmp(pattern,".") == 0)
2389 return False;
2391 return ms_fnmatch(pattern, string, True, is_case_sensitive) == 0;
2394 /*******************************************************************
2395 A wrapper that handles a list of patters and calls mask_match()
2396 on each. Returns True if any of the patterns match.
2397 *******************************************************************/
2399 bool mask_match_list(const char *string, char **list, int listLen, bool is_case_sensitive)
2401 while (listLen-- > 0) {
2402 if (mask_match(string, *list++, is_case_sensitive))
2403 return True;
2405 return False;
2408 /*********************************************************
2409 Recursive routine that is called by unix_wild_match.
2410 *********************************************************/
2412 static bool unix_do_match(const char *regexp, const char *str)
2414 const char *p;
2416 for( p = regexp; *p && *str; ) {
2418 switch(*p) {
2419 case '?':
2420 str++;
2421 p++;
2422 break;
2424 case '*':
2427 * Look for a character matching
2428 * the one after the '*'.
2430 p++;
2431 if(!*p)
2432 return true; /* Automatic match */
2433 while(*str) {
2435 while(*str && (*p != *str))
2436 str++;
2439 * Patch from weidel@multichart.de. In the case of the regexp
2440 * '*XX*' we want to ensure there are at least 2 'X' characters
2441 * in the string after the '*' for a match to be made.
2445 int matchcount=0;
2448 * Eat all the characters that match, but count how many there were.
2451 while(*str && (*p == *str)) {
2452 str++;
2453 matchcount++;
2457 * Now check that if the regexp had n identical characters that
2458 * matchcount had at least that many matches.
2461 while ( *(p+1) && (*(p+1) == *p)) {
2462 p++;
2463 matchcount--;
2466 if ( matchcount <= 0 )
2467 return false;
2470 str--; /* We've eaten the match char after the '*' */
2472 if(unix_do_match(p, str))
2473 return true;
2475 if(!*str)
2476 return false;
2477 else
2478 str++;
2480 return false;
2482 default:
2483 if(*str != *p)
2484 return false;
2485 str++;
2486 p++;
2487 break;
2491 if(!*p && !*str)
2492 return true;
2494 if (!*p && str[0] == '.' && str[1] == 0)
2495 return true;
2497 if (!*str && *p == '?') {
2498 while (*p == '?')
2499 p++;
2500 return(!*p);
2503 if(!*str && (*p == '*' && p[1] == '\0'))
2504 return true;
2506 return false;
2509 /*******************************************************************
2510 Simple case insensitive interface to a UNIX wildcard matcher.
2511 Returns True if match, False if not.
2512 *******************************************************************/
2514 bool unix_wild_match(const char *pattern, const char *string)
2516 TALLOC_CTX *ctx = talloc_stackframe();
2517 char *p2;
2518 char *s2;
2519 char *p;
2520 bool ret = false;
2522 p2 = talloc_strdup(ctx,pattern);
2523 s2 = talloc_strdup(ctx,string);
2524 if (!p2 || !s2) {
2525 TALLOC_FREE(ctx);
2526 return false;
2528 strlower_m(p2);
2529 strlower_m(s2);
2531 /* Remove any *? and ** from the pattern as they are meaningless */
2532 for(p = p2; *p; p++) {
2533 while( *p == '*' && (p[1] == '?' ||p[1] == '*')) {
2534 memmove(&p[1], &p[2], strlen(&p[2])+1);
2538 if (strequal(p2,"*")) {
2539 TALLOC_FREE(ctx);
2540 return true;
2543 ret = unix_do_match(p2, s2);
2544 TALLOC_FREE(ctx);
2545 return ret;
2548 /**********************************************************************
2549 Converts a name to a fully qualified domain name.
2550 Returns true if lookup succeeded, false if not (then fqdn is set to name)
2551 Note we deliberately use gethostbyname here, not getaddrinfo as we want
2552 to examine the h_aliases and I don't know how to do that with getaddrinfo.
2553 ***********************************************************************/
2555 bool name_to_fqdn(fstring fqdn, const char *name)
2557 char *full = NULL;
2558 struct hostent *hp = gethostbyname(name);
2560 if (!hp || !hp->h_name || !*hp->h_name) {
2561 DEBUG(10,("name_to_fqdn: lookup for %s failed.\n", name));
2562 fstrcpy(fqdn, name);
2563 return false;
2566 /* Find out if the fqdn is returned as an alias
2567 * to cope with /etc/hosts files where the first
2568 * name is not the fqdn but the short name */
2569 if (hp->h_aliases && (! strchr_m(hp->h_name, '.'))) {
2570 int i;
2571 for (i = 0; hp->h_aliases[i]; i++) {
2572 if (strchr_m(hp->h_aliases[i], '.')) {
2573 full = hp->h_aliases[i];
2574 break;
2578 if (full && (StrCaseCmp(full, "localhost.localdomain") == 0)) {
2579 DEBUG(1, ("WARNING: your /etc/hosts file may be broken!\n"));
2580 DEBUGADD(1, (" Specifing the machine hostname for address 127.0.0.1 may lead\n"));
2581 DEBUGADD(1, (" to Kerberos authentication problems as localhost.localdomain\n"));
2582 DEBUGADD(1, (" may end up being used instead of the real machine FQDN.\n"));
2583 full = hp->h_name;
2585 if (!full) {
2586 full = hp->h_name;
2589 DEBUG(10,("name_to_fqdn: lookup for %s -> %s.\n", name, full));
2590 fstrcpy(fqdn, full);
2591 return true;
2594 /**********************************************************************
2595 Append a DATA_BLOB to a talloc'ed object
2596 ***********************************************************************/
2598 void *talloc_append_blob(TALLOC_CTX *mem_ctx, void *buf, DATA_BLOB blob)
2600 size_t old_size = 0;
2601 char *result;
2603 if (blob.length == 0) {
2604 return buf;
2607 if (buf != NULL) {
2608 old_size = talloc_get_size(buf);
2611 result = (char *)TALLOC_REALLOC(mem_ctx, buf, old_size + blob.length);
2612 if (result == NULL) {
2613 return NULL;
2616 memcpy(result + old_size, blob.data, blob.length);
2617 return result;
2620 uint32 map_share_mode_to_deny_mode(uint32 share_access, uint32 private_options)
2622 switch (share_access & ~FILE_SHARE_DELETE) {
2623 case FILE_SHARE_NONE:
2624 return DENY_ALL;
2625 case FILE_SHARE_READ:
2626 return DENY_WRITE;
2627 case FILE_SHARE_WRITE:
2628 return DENY_READ;
2629 case FILE_SHARE_READ|FILE_SHARE_WRITE:
2630 return DENY_NONE;
2632 if (private_options & NTCREATEX_OPTIONS_PRIVATE_DENY_DOS) {
2633 return DENY_DOS;
2634 } else if (private_options & NTCREATEX_OPTIONS_PRIVATE_DENY_FCB) {
2635 return DENY_FCB;
2638 return (uint32)-1;
2641 pid_t procid_to_pid(const struct server_id *proc)
2643 return proc->pid;
2646 static uint32 my_vnn = NONCLUSTER_VNN;
2648 void set_my_vnn(uint32 vnn)
2650 DEBUG(10, ("vnn pid %d = %u\n", (int)sys_getpid(), (unsigned int)vnn));
2651 my_vnn = vnn;
2654 uint32 get_my_vnn(void)
2656 return my_vnn;
2659 struct server_id pid_to_procid(pid_t pid)
2661 struct server_id result;
2662 result.pid = pid;
2663 #ifdef CLUSTER_SUPPORT
2664 result.vnn = my_vnn;
2665 #endif
2666 return result;
2669 struct server_id procid_self(void)
2671 return pid_to_procid(sys_getpid());
2674 struct server_id server_id_self(void)
2676 return procid_self();
2679 bool procid_equal(const struct server_id *p1, const struct server_id *p2)
2681 if (p1->pid != p2->pid)
2682 return False;
2683 #ifdef CLUSTER_SUPPORT
2684 if (p1->vnn != p2->vnn)
2685 return False;
2686 #endif
2687 return True;
2690 bool cluster_id_equal(const struct server_id *id1,
2691 const struct server_id *id2)
2693 return procid_equal(id1, id2);
2696 bool procid_is_me(const struct server_id *pid)
2698 if (pid->pid != sys_getpid())
2699 return False;
2700 #ifdef CLUSTER_SUPPORT
2701 if (pid->vnn != my_vnn)
2702 return False;
2703 #endif
2704 return True;
2707 struct server_id interpret_pid(const char *pid_string)
2709 struct server_id result;
2710 int pid;
2711 #ifdef CLUSTER_SUPPORT
2712 unsigned int vnn;
2713 if (sscanf(pid_string, "%u:%d", &vnn, &pid) == 2) {
2714 result.vnn = vnn;
2715 result.pid = pid;
2717 else if (sscanf(pid_string, "%d", &pid) == 1) {
2718 result.vnn = get_my_vnn();
2719 result.pid = pid;
2721 else {
2722 result.vnn = NONCLUSTER_VNN;
2723 result.pid = -1;
2725 #else
2726 if (sscanf(pid_string, "%d", &pid) != 1) {
2727 result.pid = -1;
2728 } else {
2729 result.pid = pid;
2731 #endif
2732 /* Assigning to result.pid may have overflowed
2733 Map negative pid to -1: i.e. error */
2734 if (result.pid < 0) {
2735 result.pid = -1;
2737 return result;
2740 char *procid_str(TALLOC_CTX *mem_ctx, const struct server_id *pid)
2742 #ifdef CLUSTER_SUPPORT
2743 if (pid->vnn == NONCLUSTER_VNN) {
2744 return talloc_asprintf(mem_ctx,
2745 "%d",
2746 (int)pid->pid);
2748 else {
2749 return talloc_asprintf(mem_ctx,
2750 "%u:%d",
2751 (unsigned)pid->vnn,
2752 (int)pid->pid);
2754 #else
2755 return talloc_asprintf(mem_ctx,
2756 "%d",
2757 (int)pid->pid);
2758 #endif
2761 char *procid_str_static(const struct server_id *pid)
2763 return procid_str(talloc_tos(), pid);
2766 bool procid_valid(const struct server_id *pid)
2768 return (pid->pid != -1);
2771 bool procid_is_local(const struct server_id *pid)
2773 #ifdef CLUSTER_SUPPORT
2774 return pid->vnn == my_vnn;
2775 #else
2776 return True;
2777 #endif
2780 int this_is_smp(void)
2782 #if defined(HAVE_SYSCONF)
2784 #if defined(SYSCONF_SC_NPROC_ONLN)
2785 return (sysconf(_SC_NPROC_ONLN) > 1) ? 1 : 0;
2786 #elif defined(SYSCONF_SC_NPROCESSORS_ONLN)
2787 return (sysconf(_SC_NPROCESSORS_ONLN) > 1) ? 1 : 0;
2788 #else
2789 return 0;
2790 #endif
2792 #else
2793 return 0;
2794 #endif
2797 /****************************************************************
2798 Check if offset/length fit into bufsize. Should probably be
2799 merged with is_offset_safe, but this would require a rewrite
2800 of lanman.c. Later :-)
2801 ****************************************************************/
2803 bool trans_oob(uint32_t bufsize, uint32_t offset, uint32_t length)
2805 if ((offset + length < offset) || (offset + length < length)) {
2806 /* wrap */
2807 return true;
2809 if ((offset > bufsize) || (offset + length > bufsize)) {
2810 /* overflow */
2811 return true;
2813 return false;
2816 /****************************************************************
2817 Check if an offset into a buffer is safe.
2818 If this returns True it's safe to indirect into the byte at
2819 pointer ptr+off.
2820 ****************************************************************/
2822 bool is_offset_safe(const char *buf_base, size_t buf_len, char *ptr, size_t off)
2824 const char *end_base = buf_base + buf_len;
2825 char *end_ptr = ptr + off;
2827 if (!buf_base || !ptr) {
2828 return False;
2831 if (end_base < buf_base || end_ptr < ptr) {
2832 return False; /* wrap. */
2835 if (end_ptr < end_base) {
2836 return True;
2838 return False;
2841 /****************************************************************
2842 Return a safe pointer into a buffer, or NULL.
2843 ****************************************************************/
2845 char *get_safe_ptr(const char *buf_base, size_t buf_len, char *ptr, size_t off)
2847 return is_offset_safe(buf_base, buf_len, ptr, off) ?
2848 ptr + off : NULL;
2851 /****************************************************************
2852 Return a safe pointer into a string within a buffer, or NULL.
2853 ****************************************************************/
2855 char *get_safe_str_ptr(const char *buf_base, size_t buf_len, char *ptr, size_t off)
2857 if (!is_offset_safe(buf_base, buf_len, ptr, off)) {
2858 return NULL;
2860 /* Check if a valid string exists at this offset. */
2861 if (skip_string(buf_base,buf_len, ptr + off) == NULL) {
2862 return NULL;
2864 return ptr + off;
2867 /****************************************************************
2868 Return an SVAL at a pointer, or failval if beyond the end.
2869 ****************************************************************/
2871 int get_safe_SVAL(const char *buf_base, size_t buf_len, char *ptr, size_t off, int failval)
2874 * Note we use off+1 here, not off+2 as SVAL accesses ptr[0] and ptr[1],
2875 * NOT ptr[2].
2877 if (!is_offset_safe(buf_base, buf_len, ptr, off+1)) {
2878 return failval;
2880 return SVAL(ptr,off);
2883 /****************************************************************
2884 Return an IVAL at a pointer, or failval if beyond the end.
2885 ****************************************************************/
2887 int get_safe_IVAL(const char *buf_base, size_t buf_len, char *ptr, size_t off, int failval)
2890 * Note we use off+3 here, not off+4 as IVAL accesses
2891 * ptr[0] ptr[1] ptr[2] ptr[3] NOT ptr[4].
2893 if (!is_offset_safe(buf_base, buf_len, ptr, off+3)) {
2894 return failval;
2896 return IVAL(ptr,off);
2899 /****************************************************************
2900 Split DOM\user into DOM and user. Do not mix with winbind variants of that
2901 call (they take care of winbind separator and other winbind specific settings).
2902 ****************************************************************/
2904 void split_domain_user(TALLOC_CTX *mem_ctx,
2905 const char *full_name,
2906 char **domain,
2907 char **user)
2909 const char *p = NULL;
2911 p = strchr_m(full_name, '\\');
2913 if (p != NULL) {
2914 *domain = talloc_strndup(mem_ctx, full_name,
2915 PTR_DIFF(p, full_name));
2916 *user = talloc_strdup(mem_ctx, p+1);
2917 } else {
2918 *domain = talloc_strdup(mem_ctx, "");
2919 *user = talloc_strdup(mem_ctx, full_name);
2923 #if 0
2925 Disable these now we have checked all code paths and ensured
2926 NULL returns on zero request. JRA.
2928 /****************************************************************
2929 talloc wrapper functions that guarentee a null pointer return
2930 if size == 0.
2931 ****************************************************************/
2933 #ifndef MAX_TALLOC_SIZE
2934 #define MAX_TALLOC_SIZE 0x10000000
2935 #endif
2938 * talloc and zero memory.
2939 * - returns NULL if size is zero.
2942 void *_talloc_zero_zeronull(const void *ctx, size_t size, const char *name)
2944 void *p;
2946 if (size == 0) {
2947 return NULL;
2950 p = talloc_named_const(ctx, size, name);
2952 if (p) {
2953 memset(p, '\0', size);
2956 return p;
2960 * memdup with a talloc.
2961 * - returns NULL if size is zero.
2964 void *_talloc_memdup_zeronull(const void *t, const void *p, size_t size, const char *name)
2966 void *newp;
2968 if (size == 0) {
2969 return NULL;
2972 newp = talloc_named_const(t, size, name);
2973 if (newp) {
2974 memcpy(newp, p, size);
2977 return newp;
2981 * alloc an array, checking for integer overflow in the array size.
2982 * - returns NULL if count or el_size are zero.
2985 void *_talloc_array_zeronull(const void *ctx, size_t el_size, unsigned count, const char *name)
2987 if (count >= MAX_TALLOC_SIZE/el_size) {
2988 return NULL;
2991 if (el_size == 0 || count == 0) {
2992 return NULL;
2995 return talloc_named_const(ctx, el_size * count, name);
2999 * alloc an zero array, checking for integer overflow in the array size
3000 * - returns NULL if count or el_size are zero.
3003 void *_talloc_zero_array_zeronull(const void *ctx, size_t el_size, unsigned count, const char *name)
3005 if (count >= MAX_TALLOC_SIZE/el_size) {
3006 return NULL;
3009 if (el_size == 0 || count == 0) {
3010 return NULL;
3013 return _talloc_zero(ctx, el_size * count, name);
3017 * Talloc wrapper that returns NULL if size == 0.
3019 void *talloc_zeronull(const void *context, size_t size, const char *name)
3021 if (size == 0) {
3022 return NULL;
3024 return talloc_named_const(context, size, name);
3026 #endif
3028 bool is_valid_policy_hnd(const struct policy_handle *hnd)
3030 struct policy_handle tmp;
3031 ZERO_STRUCT(tmp);
3032 return (memcmp(&tmp, hnd, sizeof(tmp)) != 0);
3035 bool policy_hnd_equal(const struct policy_handle *hnd1,
3036 const struct policy_handle *hnd2)
3038 if (!hnd1 || !hnd2) {
3039 return false;
3042 return (memcmp(hnd1, hnd2, sizeof(*hnd1)) == 0);
3045 /****************************************************************
3046 strip off leading '\\' from a hostname
3047 ****************************************************************/
3049 const char *strip_hostname(const char *s)
3051 if (!s) {
3052 return NULL;
3055 if (strlen_m(s) < 3) {
3056 return s;
3059 if (s[0] == '\\') s++;
3060 if (s[0] == '\\') s++;
3062 return s;
3065 bool tevent_req_poll_ntstatus(struct tevent_req *req,
3066 struct tevent_context *ev,
3067 NTSTATUS *status)
3069 bool ret = tevent_req_poll(req, ev);
3070 if (!ret) {
3071 *status = map_nt_error_from_unix(errno);
3073 return ret;