source/lib/util.c

   1 /*
   2    Unix SMB/CIFS implementation.
   3    Samba utility functions
   4    Copyright (C) Andrew Tridgell 1992-1998
   5    Copyright (C) Jeremy Allison 2001-2002
   6    Copyright (C) Simo Sorce 2001
   7    Copyright (C) Jim McDonough <jmcd@us.ibm.com> 2003
   8
   9    This program is free software; you can redistribute it and/or modify
  10    it under the terms of the GNU General Public License as published by
  11    the Free Software Foundation; either version 2 of the License, or
  12    (at your option) any later version.
  13
  14    This program is distributed in the hope that it will be useful,
  15    but WITHOUT ANY WARRANTY; without even the implied warranty of
  16    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  17    GNU General Public License for more details.
  18
  19    You should have received a copy of the GNU General Public License
  20    along with this program; if not, write to the Free Software
  21    Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  22 */
  23
  24 #include "includes.h"
  25
  26 extern fstring local_machine;
  27 extern char *global_clobber_region_function;
  28 extern unsigned int global_clobber_region_line;
  29 extern fstring remote_arch;
  30
  31 /* Max allowable allococation - 256mb - 0x10000000 */
  32 #define MAX_ALLOC_SIZE (1024*1024*256)
  33
  34 #if (defined(HAVE_NETGROUP) && defined (WITH_AUTOMOUNT))
  35 #ifdef WITH_NISPLUS_HOME
  36 #ifdef BROKEN_NISPLUS_INCLUDE_FILES
  37 /*
  38  * The following lines are needed due to buggy include files
  39  * in Solaris 2.6 which define GROUP in both /usr/include/sys/acl.h and
  40  * also in /usr/include/rpcsvc/nis.h. The definitions conflict. JRA.
  41  * Also GROUP_OBJ is defined as 0x4 in /usr/include/sys/acl.h and as
  42  * an enum in /usr/include/rpcsvc/nis.h.
  43  */
  44
  45 #if defined(GROUP)
  46 #undef GROUP
  47 #endif
  48
  49 #if defined(GROUP_OBJ)
  50 #undef GROUP_OBJ
  51 #endif
  52
  53 #endif /* BROKEN_NISPLUS_INCLUDE_FILES */
  54
  55 #include <rpcsvc/nis.h>
  56
  57 #endif /* WITH_NISPLUS_HOME */
  58 #endif /* HAVE_NETGROUP && WITH_AUTOMOUNT */
  59
  60 enum protocol_types Protocol = PROTOCOL_COREPLUS;
  61
  62 /* a default finfo structure to ensure all fields are sensible */
  63 file_info def_finfo = {-1,0,0,0,0,0,0,"",""};
  64
  65 /* this is used by the chaining code */
  66 int chain_size = 0;
  67
  68 int trans_num = 0;
  69
  70 static enum remote_arch_types ra_type = RA_UNKNOWN;
  71 pstring user_socket_options=DEFAULT_SOCKET_OPTIONS;
  72
  73 /***********************************************************************
  74  Definitions for all names.
  75 ***********************************************************************/
  76
  77 static char *smb_myname;
  78 static char *smb_myworkgroup;
  79 static char *smb_scope;
  80 static int smb_num_netbios_names;
  81 static char **smb_my_netbios_names;
  82
  83 /***********************************************************************
  84  Allocate and set myname. Ensure upper case.
  85 ***********************************************************************/
  86
  87 BOOL set_global_myname(const char *myname)
  88 {
  89         SAFE_FREE(smb_myname);
  90         smb_myname = SMB_STRDUP(myname);
  91         if (!smb_myname)
  92                 return False;
  93         strupper_m(smb_myname);
  94         return True;
  95 }
  96
  97 const char *global_myname(void)
  98 {
  99         return smb_myname;
 100 }
 101
 102 /***********************************************************************
 103  Allocate and set myworkgroup. Ensure upper case.
 104 ***********************************************************************/
 105
 106 BOOL set_global_myworkgroup(const char *myworkgroup)
 107 {
 108         SAFE_FREE(smb_myworkgroup);
 109         smb_myworkgroup = SMB_STRDUP(myworkgroup);
 110         if (!smb_myworkgroup)
 111                 return False;
 112         strupper_m(smb_myworkgroup);
 113         return True;
 114 }
 115
 116 const char *lp_workgroup(void)
 117 {
 118         return smb_myworkgroup;
 119 }
 120
 121 /***********************************************************************
 122  Allocate and set scope. Ensure upper case.
 123 ***********************************************************************/
 124
 125 BOOL set_global_scope(const char *scope)
 126 {
 127         SAFE_FREE(smb_scope);
 128         smb_scope = SMB_STRDUP(scope);
 129         if (!smb_scope)
 130                 return False;
 131         strupper_m(smb_scope);
 132         return True;
 133 }
 134
 135 /*********************************************************************
 136  Ensure scope is never null string.
 137 *********************************************************************/
 138
 139 const char *global_scope(void)
 140 {
 141         if (!smb_scope)
 142                 set_global_scope("");
 143         return smb_scope;
 144 }
 145
 146 static void free_netbios_names_array(void)
 147 {
 148         int i;
 149
 150         for (i = 0; i < smb_num_netbios_names; i++)
 151                 SAFE_FREE(smb_my_netbios_names[i]);
 152
 153         SAFE_FREE(smb_my_netbios_names);
 154         smb_num_netbios_names = 0;
 155 }
 156
 157 static BOOL allocate_my_netbios_names_array(size_t number)
 158 {
 159         free_netbios_names_array();
 160
 161         smb_num_netbios_names = number + 1;
 162         smb_my_netbios_names = SMB_MALLOC_ARRAY( char *, smb_num_netbios_names );
 163
 164         if (!smb_my_netbios_names)
 165                 return False;
 166
 167         memset(smb_my_netbios_names, '\0', sizeof(char *) * smb_num_netbios_names);
 168         return True;
 169 }
 170
 171 static BOOL set_my_netbios_names(const char *name, int i)
 172 {
 173         SAFE_FREE(smb_my_netbios_names[i]);
 174
 175         smb_my_netbios_names[i] = SMB_STRDUP(name);
 176         if (!smb_my_netbios_names[i])
 177                 return False;
 178         strupper_m(smb_my_netbios_names[i]);
 179         return True;
 180 }
 181
 182 const char *my_netbios_names(int i)
 183 {
 184         return smb_my_netbios_names[i];
 185 }
 186
 187 BOOL set_netbios_aliases(const char **str_array)
 188 {
 189         size_t namecount;
 190
 191         /* Work out the max number of netbios aliases that we have */
 192         for( namecount=0; str_array && (str_array[namecount] != NULL); namecount++ )
 193                 ;
 194
 195         if ( global_myname() && *global_myname())
 196                 namecount++;
 197
 198         /* Allocate space for the netbios aliases */
 199         if (!allocate_my_netbios_names_array(namecount))
 200                 return False;
 201
 202         /* Use the global_myname string first */
 203         namecount=0;
 204         if ( global_myname() && *global_myname()) {
 205                 set_my_netbios_names( global_myname(), namecount );
 206                 namecount++;
 207         }
 208
 209         if (str_array) {
 210                 size_t i;
 211                 for ( i = 0; str_array[i] != NULL; i++) {
 212                         size_t n;
 213                         BOOL duplicate = False;
 214
 215                         /* Look for duplicates */
 216                         for( n=0; n<namecount; n++ ) {
 217                                 if( strequal( str_array[i], my_netbios_names(n) ) ) {
 218                                         duplicate = True;
 219                                         break;
 220                                 }
 221                         }
 222                         if (!duplicate) {
 223                                 if (!set_my_netbios_names(str_array[i], namecount))
 224                                         return False;
 225                                 namecount++;
 226                         }
 227                 }
 228         }
 229         return True;
 230 }
 231
 232 /****************************************************************************
 233   Common name initialization code.
 234 ****************************************************************************/
 235
 236 BOOL init_names(void)
 237 {
 238         char *p;
 239         int n;
 240
 241         if (global_myname() == NULL || *global_myname() == '\0') {
 242                 if (!set_global_myname(myhostname())) {
 243                         DEBUG( 0, ( "init_structs: malloc fail.\n" ) );
 244                         return False;
 245                 }
 246         }
 247
 248         if (!set_netbios_aliases(lp_netbios_aliases())) {
 249                 DEBUG( 0, ( "init_structs: malloc fail.\n" ) );
 250                 return False;
 251         }
 252
 253         fstrcpy( local_machine, global_myname() );
 254         trim_char( local_machine, ' ', ' ' );
 255         p = strchr( local_machine, ' ' );
 256         if (p)
 257                 *p = 0;
 258         strlower_m( local_machine );
 259
 260         DEBUG( 5, ("Netbios name list:-\n") );
 261         for( n=0; my_netbios_names(n); n++ )
 262                 DEBUGADD( 5, ( "my_netbios_names[%d]=\"%s\"\n", n, my_netbios_names(n) ) );
 263
 264         return( True );
 265 }
 266
 267 /**************************************************************************n
 268  Find a suitable temporary directory. The result should be copied immediately
 269  as it may be overwritten by a subsequent call.
 270 ****************************************************************************/
 271
 272 const char *tmpdir(void)
 273 {
 274         char *p;
 275         if ((p = getenv("TMPDIR")))
 276                 return p;
 277         return "/tmp";
 278 }
 279
 280 /****************************************************************************
 281  Add a gid to an array of gids if it's not already there.
 282 ****************************************************************************/
 283
 284 void add_gid_to_array_unique(TALLOC_CTX *mem_ctx, gid_t gid,
 285                              gid_t **gids, int *num)
 286 {
 287         int i;
 288
 289         for (i=0; i<*num; i++) {
 290                 if ((*gids)[i] == gid)
 291                         return;
 292         }
 293
 294         if (mem_ctx != NULL)
 295                 *gids = TALLOC_REALLOC_ARRAY(mem_ctx, *gids, gid_t, *num+1);
 296         else
 297                 *gids = SMB_REALLOC_ARRAY(*gids, gid_t, *num+1);
 298
 299         if (*gids == NULL)
 300                 return;
 301
 302         (*gids)[*num] = gid;
 303         *num += 1;
 304 }
 305
 306 /****************************************************************************
 307  Like atoi but gets the value up to the separator character.
 308 ****************************************************************************/
 309
 310 static const char *Atoic(const char *p, int *n, const char *c)
 311 {
 312         if (!isdigit((int)*p)) {
 313                 DEBUG(5, ("Atoic: malformed number\n"));
 314                 return NULL;
 315         }
 316
 317         (*n) = atoi(p);
 318
 319         while ((*p) && isdigit((int)*p))
 320                 p++;
 321
 322         if (strchr_m(c, *p) == NULL) {
 323                 DEBUG(5, ("Atoic: no separator characters (%s) not found\n", c));
 324                 return NULL;
 325         }
 326
 327         return p;
 328 }
 329
 330 /*************************************************************************
 331  Reads a list of numbers.
 332  *************************************************************************/
 333
 334 const char *get_numlist(const char *p, uint32 **num, int *count)
 335 {
 336         int val;
 337
 338         if (num == NULL || count == NULL)
 339                 return NULL;
 340
 341         (*count) = 0;
 342         (*num  ) = NULL;
 343
 344         while ((p = Atoic(p, &val, ":,")) != NULL && (*p) != ':') {
 345                 uint32 *tn;
 346
 347                 tn = SMB_REALLOC_ARRAY((*num), uint32, (*count)+1);
 348                 if (tn == NULL) {
 349                         SAFE_FREE(*num);
 350                         return NULL;
 351                 } else
 352                         (*num) = tn;
 353                 (*num)[(*count)] = val;
 354                 (*count)++;
 355                 p++;
 356         }
 357
 358         return p;
 359 }
 360
 361 /*******************************************************************
 362  Check if a file exists - call vfs_file_exist for samba files.
 363 ********************************************************************/
 364
 365 BOOL file_exist(const char *fname,SMB_STRUCT_STAT *sbuf)
 366 {
 367         SMB_STRUCT_STAT st;
 368         if (!sbuf)
 369                 sbuf = &st;
 370
 371         if (sys_stat(fname,sbuf) != 0)
 372                 return(False);
 373
 374         return((S_ISREG(sbuf->st_mode)) || (S_ISFIFO(sbuf->st_mode)));
 375 }
 376
 377 /*******************************************************************
 378  Check a files mod time.
 379 ********************************************************************/
 380
 381 time_t file_modtime(const char *fname)
 382 {
 383         SMB_STRUCT_STAT st;
 384
 385         if (sys_stat(fname,&st) != 0)
 386                 return(0);
 387
 388         return(st.st_mtime);
 389 }
 390
 391 /*******************************************************************
 392  Check if a directory exists.
 393 ********************************************************************/
 394
 395 BOOL directory_exist(char *dname,SMB_STRUCT_STAT *st)
 396 {
 397         SMB_STRUCT_STAT st2;
 398         BOOL ret;
 399
 400         if (!st)
 401                 st = &st2;
 402
 403         if (sys_stat(dname,st) != 0)
 404                 return(False);
 405
 406         ret = S_ISDIR(st->st_mode);
 407         if(!ret)
 408                 errno = ENOTDIR;
 409         return ret;
 410 }
 411
 412 /*******************************************************************
 413  Returns the size in bytes of the named file.
 414 ********************************************************************/
 415
 416 SMB_OFF_T get_file_size(char *file_name)
 417 {
 418         SMB_STRUCT_STAT buf;
 419         buf.st_size = 0;
 420         if(sys_stat(file_name,&buf) != 0)
 421                 return (SMB_OFF_T)-1;
 422         return(buf.st_size);
 423 }
 424
 425 /*******************************************************************
 426  Return a string representing an attribute for a file.
 427 ********************************************************************/
 428
 429 char *attrib_string(uint16 mode)
 430 {
 431         static fstring attrstr;
 432
 433         attrstr[0] = 0;
 434
 435         if (mode & aVOLID) fstrcat(attrstr,"V");
 436         if (mode & aDIR) fstrcat(attrstr,"D");
 437         if (mode & aARCH) fstrcat(attrstr,"A");
 438         if (mode & aHIDDEN) fstrcat(attrstr,"H");
 439         if (mode & aSYSTEM) fstrcat(attrstr,"S");
 440         if (mode & aRONLY) fstrcat(attrstr,"R");
 441
 442         return(attrstr);
 443 }
 444
 445 /*******************************************************************
 446  Show a smb message structure.
 447 ********************************************************************/
 448
 449 void show_msg(char *buf)
 450 {
 451         int i;
 452         int bcc=0;
 453
 454         if (!DEBUGLVL(5))
 455                 return;
 456
 457         DEBUG(5,("size=%d\nsmb_com=0x%x\nsmb_rcls=%d\nsmb_reh=%d\nsmb_err=%d\nsmb_flg=%d\nsmb_flg2=%d\n",
 458                         smb_len(buf),
 459                         (int)CVAL(buf,smb_com),
 460                         (int)CVAL(buf,smb_rcls),
 461                         (int)CVAL(buf,smb_reh),
 462                         (int)SVAL(buf,smb_err),
 463                         (int)CVAL(buf,smb_flg),
 464                         (int)SVAL(buf,smb_flg2)));
 465         DEBUGADD(5,("smb_tid=%d\nsmb_pid=%d\nsmb_uid=%d\nsmb_mid=%d\n",
 466                         (int)SVAL(buf,smb_tid),
 467                         (int)SVAL(buf,smb_pid),
 468                         (int)SVAL(buf,smb_uid),
 469                         (int)SVAL(buf,smb_mid)));
 470         DEBUGADD(5,("smt_wct=%d\n",(int)CVAL(buf,smb_wct)));
 471
 472         for (i=0;i<(int)CVAL(buf,smb_wct);i++)
 473                 DEBUGADD(5,("smb_vwv[%2d]=%5d (0x%X)\n",i,
 474                         SVAL(buf,smb_vwv+2*i),SVAL(buf,smb_vwv+2*i)));
 475
 476         bcc = (int)SVAL(buf,smb_vwv+2*(CVAL(buf,smb_wct)));
 477
 478         DEBUGADD(5,("smb_bcc=%d\n",bcc));
 479
 480         if (DEBUGLEVEL < 10)
 481                 return;
 482
 483         if (DEBUGLEVEL < 50)
 484                 bcc = MIN(bcc, 512);
 485
 486         dump_data(10, smb_buf(buf), bcc);
 487 }
 488
 489 /*******************************************************************
 490  Set the length and marker of an smb packet.
 491 ********************************************************************/
 492
 493 void smb_setlen(char *buf,int len)
 494 {
 495         _smb_setlen(buf,len);
 496
 497         SCVAL(buf,4,0xFF);
 498         SCVAL(buf,5,'S');
 499         SCVAL(buf,6,'M');
 500         SCVAL(buf,7,'B');
 501 }
 502
 503 /*******************************************************************
 504  Setup the word count and byte count for a smb message.
 505 ********************************************************************/
 506
 507 int set_message(char *buf,int num_words,int num_bytes,BOOL zero)
 508 {
 509         if (zero)
 510                 memset(buf + smb_size,'\0',num_words*2 + num_bytes);
 511         SCVAL(buf,smb_wct,num_words);
 512         SSVAL(buf,smb_vwv + num_words*SIZEOFWORD,num_bytes);
 513         smb_setlen(buf,smb_size + num_words*2 + num_bytes - 4);
 514         return (smb_size + num_words*2 + num_bytes);
 515 }
 516
 517 /*******************************************************************
 518  Setup only the byte count for a smb message.
 519 ********************************************************************/
 520
 521 int set_message_bcc(char *buf,int num_bytes)
 522 {
 523         int num_words = CVAL(buf,smb_wct);
 524         SSVAL(buf,smb_vwv + num_words*SIZEOFWORD,num_bytes);
 525         smb_setlen(buf,smb_size + num_words*2 + num_bytes - 4);
 526         return (smb_size + num_words*2 + num_bytes);
 527 }
 528
 529 /*******************************************************************
 530  Setup only the byte count for a smb message, using the end of the
 531  message as a marker.
 532 ********************************************************************/
 533
 534 int set_message_end(void *outbuf,void *end_ptr)
 535 {
 536         return set_message_bcc((char *)outbuf,PTR_DIFF(end_ptr,smb_buf((char *)outbuf)));
 537 }
 538
 539 /*******************************************************************
 540  Reduce a file name, removing .. elements.
 541 ********************************************************************/
 542
 543 void dos_clean_name(char *s)
 544 {
 545         char *p=NULL;
 546
 547         DEBUG(3,("dos_clean_name [%s]\n",s));
 548
 549         /* remove any double slashes */
 550         all_string_sub(s, "\\\\", "\\", 0);
 551
 552         while ((p = strstr_m(s,"\\..\\")) != NULL) {
 553                 pstring s1;
 554
 555                 *p = 0;
 556                 pstrcpy(s1,p+3);
 557
 558                 if ((p=strrchr_m(s,'\\')) != NULL)
 559                         *p = 0;
 560                 else
 561                         *s = 0;
 562                 pstrcat(s,s1);
 563         }
 564
 565         trim_string(s,NULL,"\\..");
 566
 567         all_string_sub(s, "\\.\\", "\\", 0);
 568 }
 569
 570 /*******************************************************************
 571  Reduce a file name, removing .. elements.
 572 ********************************************************************/
 573
 574 void unix_clean_name(char *s)
 575 {
 576         char *p=NULL;
 577
 578         DEBUG(3,("unix_clean_name [%s]\n",s));
 579
 580         /* remove any double slashes */
 581         all_string_sub(s, "//","/", 0);
 582
 583         /* Remove leading ./ characters */
 584         if(strncmp(s, "./", 2) == 0) {
 585                 trim_string(s, "./", NULL);
 586                 if(*s == 0)
 587                         pstrcpy(s,"./");
 588         }
 589
 590         while ((p = strstr_m(s,"/../")) != NULL) {
 591                 pstring s1;
 592
 593                 *p = 0;
 594                 pstrcpy(s1,p+3);
 595
 596                 if ((p=strrchr_m(s,'/')) != NULL)
 597                         *p = 0;
 598                 else
 599                         *s = 0;
 600                 pstrcat(s,s1);
 601         }
 602
 603         trim_string(s,NULL,"/..");
 604 }
 605
 606 /****************************************************************************
 607  Make a dir struct.
 608 ****************************************************************************/
 609
 610 void make_dir_struct(char *buf, const char *mask, const char *fname,SMB_OFF_T size,int mode,time_t date, BOOL uc)
 611 {
 612         char *p;
 613         pstring mask2;
 614
 615         pstrcpy(mask2,mask);
 616
 617         if ((mode & aDIR) != 0)
 618                 size = 0;
 619
 620         memset(buf+1,' ',11);
 621         if ((p = strchr_m(mask2,'.')) != NULL) {
 622                 *p = 0;
 623                 push_ascii(buf+1,mask2,8, 0);
 624                 push_ascii(buf+9,p+1,3, 0);
 625                 *p = '.';
 626         } else
 627                 push_ascii(buf+1,mask2,11, 0);
 628
 629         memset(buf+21,'\0',DIR_STRUCT_SIZE-21);
 630         SCVAL(buf,21,mode);
 631         put_dos_date(buf,22,date);
 632         SSVAL(buf,26,size & 0xFFFF);
 633         SSVAL(buf,28,(size >> 16)&0xFFFF);
 634         /* We only uppercase if FLAGS2_LONG_PATH_COMPONENTS is zero in the input buf.
 635            Strange, but verified on W2K3. Needed for OS/2. JRA. */
 636         push_ascii(buf+30,fname,12, uc ? STR_UPPER : 0);
 637         DEBUG(8,("put name [%s] from [%s] into dir struct\n",buf+30, fname));
 638 }
 639
 640 /*******************************************************************
 641  Close the low 3 fd's and open dev/null in their place.
 642 ********************************************************************/
 643
 644 void close_low_fds(BOOL stderr_too)
 645 {
 646 #ifndef VALGRIND
 647         int fd;
 648         int i;
 649
 650         close(0);
 651         close(1);
 652
 653         if (stderr_too)
 654                 close(2);
 655
 656         /* try and use up these file descriptors, so silly
 657                 library routines writing to stdout etc won't cause havoc */
 658         for (i=0;i<3;i++) {
 659                 if (i == 2 && !stderr_too)
 660                         continue;
 661
 662                 fd = sys_open("/dev/null",O_RDWR,0);
 663                 if (fd < 0)
 664                         fd = sys_open("/dev/null",O_WRONLY,0);
 665                 if (fd < 0) {
 666                         DEBUG(0,("Can't open /dev/null\n"));
 667                         return;
 668                 }
 669                 if (fd != i) {
 670                         DEBUG(0,("Didn't get file descriptor %d\n",i));
 671                         return;
 672                 }
 673         }
 674 #endif
 675 }
 676
 677 /****************************************************************************
 678  Set a fd into blocking/nonblocking mode. Uses POSIX O_NONBLOCK if available,
 679  else
 680   if SYSV use O_NDELAY
 681   if BSD use FNDELAY
 682 ****************************************************************************/
 683
 684 int set_blocking(int fd, BOOL set)
 685 {
 686         int val;
 687 #ifdef O_NONBLOCK
 688 #define FLAG_TO_SET O_NONBLOCK
 689 #else
 690 #ifdef SYSV
 691 #define FLAG_TO_SET O_NDELAY
 692 #else /* BSD */
 693 #define FLAG_TO_SET FNDELAY
 694 #endif
 695 #endif
 696
 697         if((val = sys_fcntl_long(fd, F_GETFL, 0)) == -1)
 698                 return -1;
 699         if(set) /* Turn blocking on - ie. clear nonblock flag */
 700                 val &= ~FLAG_TO_SET;
 701         else
 702                 val |= FLAG_TO_SET;
 703         return sys_fcntl_long( fd, F_SETFL, val);
 704 #undef FLAG_TO_SET
 705 }
 706
 707 /****************************************************************************
 708  Transfer some data between two fd's.
 709 ****************************************************************************/
 710
 711 #ifndef TRANSFER_BUF_SIZE
 712 #define TRANSFER_BUF_SIZE 65536
 713 #endif
 714
 715 ssize_t transfer_file_internal(int infd, int outfd, size_t n, ssize_t (*read_fn)(int, void *, size_t),
 716                                                 ssize_t (*write_fn)(int, const void *, size_t))
 717 {
 718         char *buf;
 719         size_t total = 0;
 720         ssize_t read_ret;
 721         ssize_t write_ret;
 722         size_t num_to_read_thistime;
 723         size_t num_written = 0;
 724
 725         if ((buf = SMB_MALLOC(TRANSFER_BUF_SIZE)) == NULL)
 726                 return -1;
 727
 728         while (total < n) {
 729                 num_to_read_thistime = MIN((n - total), TRANSFER_BUF_SIZE);
 730
 731                 read_ret = (*read_fn)(infd, buf, num_to_read_thistime);
 732                 if (read_ret == -1) {
 733                         DEBUG(0,("transfer_file_internal: read failure. Error = %s\n", strerror(errno) ));
 734                         SAFE_FREE(buf);
 735                         return -1;
 736                 }
 737                 if (read_ret == 0)
 738                         break;
 739
 740                 num_written = 0;
 741
 742                 while (num_written < read_ret) {
 743                         write_ret = (*write_fn)(outfd,buf + num_written, read_ret - num_written);
 744
 745                         if (write_ret == -1) {
 746                                 DEBUG(0,("transfer_file_internal: write failure. Error = %s\n", strerror(errno) ));
 747                                 SAFE_FREE(buf);
 748                                 return -1;
 749                         }
 750                         if (write_ret == 0)
 751                                 return (ssize_t)total;
 752
 753                         num_written += (size_t)write_ret;
 754                 }
 755
 756                 total += (size_t)read_ret;
 757         }
 758
 759         SAFE_FREE(buf);
 760         return (ssize_t)total;
 761 }
 762
 763 SMB_OFF_T transfer_file(int infd,int outfd,SMB_OFF_T n)
 764 {
 765         return (SMB_OFF_T)transfer_file_internal(infd, outfd, (size_t)n, sys_read, sys_write);
 766 }
 767
 768 /*******************************************************************
 769  Sleep for a specified number of milliseconds.
 770 ********************************************************************/
 771
 772 void smb_msleep(unsigned int t)
 773 {
 774 #if defined(HAVE_NANOSLEEP)
 775         struct timespec tval;
 776         int ret;
 777
 778         tval.tv_sec = t/1000;
 779         tval.tv_nsec = 1000000*(t%1000);
 780
 781         do {
 782                 errno = 0;
 783                 ret = nanosleep(&tval, &tval);
 784         } while (ret < 0 && errno == EINTR && (tval.tv_sec > 0 || tval.tv_nsec > 0));
 785 #else
 786         unsigned int tdiff=0;
 787         struct timeval tval,t1,t2;
 788         fd_set fds;
 789
 790         GetTimeOfDay(&t1);
 791         t2 = t1;
 792
 793         while (tdiff < t) {
 794                 tval.tv_sec = (t-tdiff)/1000;
 795                 tval.tv_usec = 1000*((t-tdiff)%1000);
 796
 797                 /* Never wait for more than 1 sec. */
 798                 if (tval.tv_sec > 1) {
 799                         tval.tv_sec = 1;
 800                         tval.tv_usec = 0;
 801                 }
 802
 803                 FD_ZERO(&fds);
 804                 errno = 0;
 805                 sys_select_intr(0,&fds,NULL,NULL,&tval);
 806
 807                 GetTimeOfDay(&t2);
 808                 if (t2.tv_sec < t1.tv_sec) {
 809                         /* Someone adjusted time... */
 810                         t1 = t2;
 811                 }
 812
 813                 tdiff = TvalDiff(&t1,&t2);
 814         }
 815 #endif
 816 }
 817
 818 /****************************************************************************
 819  Become a daemon, discarding the controlling terminal.
 820 ****************************************************************************/
 821
 822 void become_daemon(BOOL Fork)
 823 {
 824         if (Fork) {
 825                 if (sys_fork()) {
 826                         _exit(0);
 827                 }
 828         }
 829
 830   /* detach from the terminal */
 831 #ifdef HAVE_SETSID
 832         setsid();
 833 #elif defined(TIOCNOTTY)
 834         {
 835                 int i = sys_open("/dev/tty", O_RDWR, 0);
 836                 if (i != -1) {
 837                         ioctl(i, (int) TIOCNOTTY, (char *)0);
 838                         close(i);
 839                 }
 840         }
 841 #endif /* HAVE_SETSID */
 842
 843         /* Close fd's 0,1,2. Needed if started by rsh */
 844         close_low_fds(False);  /* Don't close stderr, let the debug system
 845                                   attach it to the logfile */
 846 }
 847
 848 /****************************************************************************
 849  Put up a yes/no prompt.
 850 ****************************************************************************/
 851
 852 BOOL yesno(char *p)
 853 {
 854         pstring ans;
 855         printf("%s",p);
 856
 857         if (!fgets(ans,sizeof(ans)-1,stdin))
 858                 return(False);
 859
 860         if (*ans == 'y' || *ans == 'Y')
 861                 return(True);
 862
 863         return(False);
 864 }
 865
 866 #if defined(PARANOID_MALLOC_CHECKER)
 867
 868 /****************************************************************************
 869  Internal malloc wrapper. Externally visible.
 870 ****************************************************************************/
 871
 872 void *malloc_(size_t size)
 873 {
 874 #undef malloc
 875         return malloc(size);
 876 #define malloc(s) __ERROR_DONT_USE_MALLOC_DIRECTLY
 877 }
 878
 879 /****************************************************************************
 880  Internal calloc wrapper. Not externally visible.
 881 ****************************************************************************/
 882
 883 static void *calloc_(size_t count, size_t size)
 884 {
 885 #undef calloc
 886         return calloc(count, size);
 887 #define calloc(n,s) __ERROR_DONT_USE_CALLOC_DIRECTLY
 888 }
 889
 890 /****************************************************************************
 891  Internal realloc wrapper. Not externally visible.
 892 ****************************************************************************/
 893
 894 static void *realloc_(void *ptr, size_t size)
 895 {
 896 #undef realloc
 897         return realloc(ptr, size);
 898 #define realloc(p,s) __ERROR_DONT_USE_RELLOC_DIRECTLY
 899 }
 900
 901 #endif /* PARANOID_MALLOC_CHECKER */
 902
 903 /****************************************************************************
 904  Type-safe malloc.
 905 ****************************************************************************/
 906
 907 void *malloc_array(size_t el_size, unsigned int count)
 908 {
 909         if (count >= MAX_ALLOC_SIZE/el_size) {
 910                 return NULL;
 911         }
 912
 913 #if defined(PARANOID_MALLOC_CHECKER)
 914         return malloc_(el_size*count);
 915 #else
 916         return malloc(el_size*count);
 917 #endif
 918 }
 919
 920 /****************************************************************************
 921  Type-safe calloc.
 922 ****************************************************************************/
 923
 924 void *calloc_array(size_t size, size_t nmemb)
 925 {
 926         if (nmemb >= MAX_ALLOC_SIZE/size) {
 927                 return NULL;
 928         }
 929 #if defined(PARANOID_MALLOC_CHECKER)
 930         return calloc_(nmemb, size);
 931 #else
 932         return calloc(nmemb, size);
 933 #endif
 934 }
 935
 936 /****************************************************************************
 937  Expand a pointer to be a particular size.
 938 ****************************************************************************/
 939
 940 void *Realloc(void *p,size_t size)
 941 {
 942         void *ret=NULL;
 943
 944         if (size == 0) {
 945                 SAFE_FREE(p);
 946                 DEBUG(5,("Realloc asked for 0 bytes\n"));
 947                 return NULL;
 948         }
 949
 950 #if defined(PARANOID_MALLOC_CHECKER)
 951         if (!p)
 952                 ret = (void *)malloc_(size);
 953         else
 954                 ret = (void *)realloc_(p,size);
 955 #else
 956         if (!p)
 957                 ret = (void *)malloc(size);
 958         else
 959                 ret = (void *)realloc(p,size);
 960 #endif
 961
 962         if (!ret)
 963                 DEBUG(0,("Memory allocation error: failed to expand to %d bytes\n",(int)size));
 964
 965         return(ret);
 966 }
 967
 968 /****************************************************************************
 969  Type-safe realloc.
 970 ****************************************************************************/
 971
 972 void *realloc_array(void *p,size_t el_size, unsigned int count)
 973 {
 974         if (count >= MAX_ALLOC_SIZE/el_size) {
 975                 return NULL;
 976         }
 977         return Realloc(p,el_size*count);
 978 }
 979
 980 /****************************************************************************
 981  (Hopefully) efficient array append
 982 ****************************************************************************/
 983 void add_to_large_array(TALLOC_CTX *mem_ctx, size_t element_size,
 984                         void *element, void **array, uint32 *num_elements,
 985                         ssize_t *array_size)
 986 {
 987         if (*array_size == -1)
 988                 return;
 989
 990         if (*array == NULL) {
 991                 if (*array_size == 0)
 992                         *array_size = 128;
 993
 994                 if (mem_ctx != NULL)
 995                         *array = talloc_array(mem_ctx, element_size,
 996                                               *array_size);
 997                 else
 998                         *array = malloc_array(element_size, *array_size);
 999
1000                 if (*array == NULL)
1001                         goto error;
1002         }
1003
1004         if (*num_elements == *array_size) {
1005                 *array_size *= 2;
1006
1007                 if (mem_ctx != NULL)
1008                         *array = talloc_realloc_array(mem_ctx, *array,
1009                                                       element_size,
1010                                                       *array_size);
1011                 else
1012                         *array = realloc_array(*array, element_size,
1013                                                *array_size);
1014
1015                 if (*array == NULL)
1016                         goto error;
1017         }
1018
1019         memcpy((char *)(*array) + element_size*(*num_elements),
1020                element, element_size);
1021         *num_elements += 1;
1022
1023         return;
1024
1025  error:
1026         *num_elements = 0;
1027         *array_size = -1;
1028 }
1029
1030 /****************************************************************************
1031  Free memory, checks for NULL.
1032  Use directly SAFE_FREE()
1033  Exists only because we need to pass a function pointer somewhere --SSS
1034 ****************************************************************************/
1035
1036 void safe_free(void *p)
1037 {
1038         SAFE_FREE(p);
1039 }
1040
1041 /****************************************************************************
1042  Get my own name and IP.
1043 ****************************************************************************/
1044
1045 BOOL get_myname(char *my_name)
1046 {
1047         pstring hostname;
1048
1049         *hostname = 0;
1050
1051         /* get my host name */
1052         if (gethostname(hostname, sizeof(hostname)) == -1) {
1053                 DEBUG(0,("gethostname failed\n"));
1054                 return False;
1055         }
1056
1057         /* Ensure null termination. */
1058         hostname[sizeof(hostname)-1] = '\0';
1059
1060         if (my_name) {
1061                 /* split off any parts after an initial . */
1062                 char *p = strchr_m(hostname,'.');
1063
1064                 if (p)
1065                         *p = 0;
1066
1067                 fstrcpy(my_name,hostname);
1068         }
1069
1070         return(True);
1071 }
1072
1073 /****************************************************************************
1074  Get my own canonical name, including domain.
1075 ****************************************************************************/
1076
1077 BOOL get_mydnsfullname(fstring my_dnsname)
1078 {
1079         static fstring dnshostname;
1080         struct hostent *hp;
1081
1082         if (!*dnshostname) {
1083                 /* get my host name */
1084                 if (gethostname(dnshostname, sizeof(dnshostname)) == -1) {
1085                         *dnshostname = '\0';
1086                         DEBUG(0,("gethostname failed\n"));
1087                         return False;
1088                 }
1089
1090                 /* Ensure null termination. */
1091                 dnshostname[sizeof(dnshostname)-1] = '\0';
1092
1093                 /* Ensure we get the cannonical name. */
1094                 if (!(hp = sys_gethostbyname(dnshostname))) {
1095                         *dnshostname = '\0';
1096                         return False;
1097                 }
1098                 fstrcpy(dnshostname, hp->h_name);
1099         }
1100         fstrcpy(my_dnsname, dnshostname);
1101         return True;
1102 }
1103
1104 /****************************************************************************
1105  Get my own domain name.
1106 ****************************************************************************/
1107
1108 BOOL get_mydnsdomname(fstring my_domname)
1109 {
1110         fstring domname;
1111         char *p;
1112
1113         *my_domname = '\0';
1114         if (!get_mydnsfullname(domname)) {
1115                 return False;
1116         }
1117         p = strchr_m(domname, '.');
1118         if (p) {
1119                 p++;
1120                 fstrcpy(my_domname, p);
1121         }
1122
1123         return False;
1124 }
1125
1126 /****************************************************************************
1127  Interpret a protocol description string, with a default.
1128 ****************************************************************************/
1129
1130 int interpret_protocol(const char *str,int def)
1131 {
1132         if (strequal(str,"NT1"))
1133                 return(PROTOCOL_NT1);
1134         if (strequal(str,"LANMAN2"))
1135                 return(PROTOCOL_LANMAN2);
1136         if (strequal(str,"LANMAN1"))
1137                 return(PROTOCOL_LANMAN1);
1138         if (strequal(str,"CORE"))
1139                 return(PROTOCOL_CORE);
1140         if (strequal(str,"COREPLUS"))
1141                 return(PROTOCOL_COREPLUS);
1142         if (strequal(str,"CORE+"))
1143                 return(PROTOCOL_COREPLUS);
1144
1145         DEBUG(0,("Unrecognised protocol level %s\n",str));
1146
1147         return(def);
1148 }
1149
1150 /****************************************************************************
1151  Return true if a string could be a pure IP address.
1152 ****************************************************************************/
1153
1154 BOOL is_ipaddress(const char *str)
1155 {
1156         BOOL pure_address = True;
1157         int i;
1158
1159         for (i=0; pure_address && str[i]; i++)
1160                 if (!(isdigit((int)str[i]) || str[i] == '.'))
1161                         pure_address = False;
1162
1163         /* Check that a pure number is not misinterpreted as an IP */
1164         pure_address = pure_address && (strchr_m(str, '.') != NULL);
1165
1166         return pure_address;
1167 }
1168
1169 /****************************************************************************
1170  Interpret an internet address or name into an IP address in 4 byte form.
1171 ****************************************************************************/
1172
1173 uint32 interpret_addr(const char *str)
1174 {
1175         struct hostent *hp;
1176         uint32 res;
1177
1178         if (strcmp(str,"0.0.0.0") == 0)
1179                 return(0);
1180         if (strcmp(str,"255.255.255.255") == 0)
1181                 return(0xFFFFFFFF);
1182
1183   /* if it's in the form of an IP address then get the lib to interpret it */
1184         if (is_ipaddress(str)) {
1185                 res = inet_addr(str);
1186         } else {
1187                 /* otherwise assume it's a network name of some sort and use
1188                         sys_gethostbyname */
1189                 if ((hp = sys_gethostbyname(str)) == 0) {
1190                         DEBUG(3,("sys_gethostbyname: Unknown host. %s\n",str));
1191                         return 0;
1192                 }
1193
1194                 if(hp->h_addr == NULL) {
1195                         DEBUG(3,("sys_gethostbyname: host address is invalid for host %s\n",str));
1196                         return 0;
1197                 }
1198                 putip((char *)&res,(char *)hp->h_addr);
1199         }
1200
1201         if (res == (uint32)-1)
1202                 return(0);
1203
1204         return(res);
1205 }
1206
1207 /*******************************************************************
1208  A convenient addition to interpret_addr().
1209 ******************************************************************/
1210
1211 struct in_addr *interpret_addr2(const char *str)
1212 {
1213         static struct in_addr ret;
1214         uint32 a = interpret_addr(str);
1215         ret.s_addr = a;
1216         return(&ret);
1217 }
1218
1219 /*******************************************************************
1220  Check if an IP is the 0.0.0.0.
1221 ******************************************************************/
1222
1223 BOOL is_zero_ip(struct in_addr ip)
1224 {
1225         uint32 a;
1226         putip((char *)&a,(char *)&ip);
1227         return(a == 0);
1228 }
1229
1230 /*******************************************************************
1231  Set an IP to 0.0.0.0.
1232 ******************************************************************/
1233
1234 void zero_ip(struct in_addr *ip)
1235 {
1236         static BOOL init;
1237         static struct in_addr ipzero;
1238
1239         if (!init) {
1240                 ipzero = *interpret_addr2("0.0.0.0");
1241                 init = True;
1242         }
1243
1244         *ip = ipzero;
1245 }
1246
1247 #if (defined(HAVE_NETGROUP) && defined(WITH_AUTOMOUNT))
1248 /******************************************************************
1249  Remove any mount options such as -rsize=2048,wsize=2048 etc.
1250  Based on a fix from <Thomas.Hepper@icem.de>.
1251 *******************************************************************/
1252
1253 static void strip_mount_options( pstring *str)
1254 {
1255         if (**str == '-') {
1256                 char *p = *str;
1257                 while(*p && !isspace(*p))
1258                         p++;
1259                 while(*p && isspace(*p))
1260                         p++;
1261                 if(*p) {
1262                         pstring tmp_str;
1263
1264                         pstrcpy(tmp_str, p);
1265                         pstrcpy(*str, tmp_str);
1266                 }
1267         }
1268 }
1269
1270 /*******************************************************************
1271  Patch from jkf@soton.ac.uk
1272  Split Luke's automount_server into YP lookup and string splitter
1273  so can easily implement automount_path().
1274  As we may end up doing both, cache the last YP result.
1275 *******************************************************************/
1276
1277 #ifdef WITH_NISPLUS_HOME
1278 char *automount_lookup( char *user_name)
1279 {
1280         static fstring last_key = "";
1281         static pstring last_value = "";
1282
1283         char *nis_map = (char *)lp_nis_home_map_name();
1284
1285         char buffer[NIS_MAXATTRVAL + 1];
1286         nis_result *result;
1287         nis_object *object;
1288         entry_obj  *entry;
1289
1290         if (strcmp(user_name, last_key)) {
1291                 slprintf(buffer, sizeof(buffer)-1, "[key=%s],%s", user_name, nis_map);
1292                 DEBUG(5, ("NIS+ querystring: %s\n", buffer));
1293
1294                 if (result = nis_list(buffer, FOLLOW_PATH|EXPAND_NAME|HARD_LOOKUP, NULL, NULL)) {
1295                         if (result->status != NIS_SUCCESS) {
1296                                 DEBUG(3, ("NIS+ query failed: %s\n", nis_sperrno(result->status)));
1297                                 fstrcpy(last_key, ""); pstrcpy(last_value, "");
1298                         } else {
1299                                 object = result->objects.objects_val;
1300                                 if (object->zo_data.zo_type == ENTRY_OBJ) {
1301                                         entry = &object->zo_data.objdata_u.en_data;
1302                                         DEBUG(5, ("NIS+ entry type: %s\n", entry->en_type));
1303                                         DEBUG(3, ("NIS+ result: %s\n", entry->en_cols.en_cols_val[1].ec_value.ec_value_val));
1304
1305                                         pstrcpy(last_value, entry->en_cols.en_cols_val[1].ec_value.ec_value_val);
1306                                         pstring_sub(last_value, "&", user_name);
1307                                         fstrcpy(last_key, user_name);
1308                                 }
1309                         }
1310                 }
1311                 nis_freeresult(result);
1312         }
1313
1314         strip_mount_options(&last_value);
1315
1316         DEBUG(4, ("NIS+ Lookup: %s resulted in %s\n", user_name, last_value));
1317         return last_value;
1318 }
1319 #else /* WITH_NISPLUS_HOME */
1320
1321 char *automount_lookup( char *user_name)
1322 {
1323         static fstring last_key = "";
1324         static pstring last_value = "";
1325
1326         int nis_error;        /* returned by yp all functions */
1327         char *nis_result;     /* yp_match inits this */
1328         int nis_result_len;  /* and set this */
1329         char *nis_domain;     /* yp_get_default_domain inits this */
1330         char *nis_map = (char *)lp_nis_home_map_name();
1331
1332         if ((nis_error = yp_get_default_domain(&nis_domain)) != 0) {
1333                 DEBUG(3, ("YP Error: %s\n", yperr_string(nis_error)));
1334                 return last_value;
1335         }
1336
1337         DEBUG(5, ("NIS Domain: %s\n", nis_domain));
1338
1339         if (!strcmp(user_name, last_key)) {
1340                 nis_result = last_value;
1341                 nis_result_len = strlen(last_value);
1342                 nis_error = 0;
1343         } else {
1344                 if ((nis_error = yp_match(nis_domain, nis_map, user_name, strlen(user_name),
1345                                 &nis_result, &nis_result_len)) == 0) {
1346                         if (!nis_error && nis_result_len >= sizeof(pstring)) {
1347                                 nis_result_len = sizeof(pstring)-1;
1348                         }
1349                         fstrcpy(last_key, user_name);
1350                         strncpy(last_value, nis_result, nis_result_len);
1351                         last_value[nis_result_len] = '\0';
1352                         strip_mount_options(&last_value);
1353
1354                 } else if(nis_error == YPERR_KEY) {
1355
1356                         /* If Key lookup fails user home server is not in nis_map
1357                                 use default information for server, and home directory */
1358                         last_value[0] = 0;
1359                         DEBUG(3, ("YP Key not found:  while looking up \"%s\" in map \"%s\"\n",
1360                                         user_name, nis_map));
1361                         DEBUG(3, ("using defaults for server and home directory\n"));
1362                 } else {
1363                         DEBUG(3, ("YP Error: \"%s\" while looking up \"%s\" in map \"%s\"\n",
1364                                         yperr_string(nis_error), user_name, nis_map));
1365                 }
1366         }
1367
1368         DEBUG(4, ("YP Lookup: %s resulted in %s\n", user_name, last_value));
1369         return last_value;
1370 }
1371 #endif /* WITH_NISPLUS_HOME */
1372 #endif
1373
1374 /*******************************************************************
1375  Are two IPs on the same subnet?
1376 ********************************************************************/
1377
1378 BOOL same_net(struct in_addr ip1,struct in_addr ip2,struct in_addr mask)
1379 {
1380         uint32 net1,net2,nmask;
1381
1382         nmask = ntohl(mask.s_addr);
1383         net1  = ntohl(ip1.s_addr);
1384         net2  = ntohl(ip2.s_addr);
1385
1386         return((net1 & nmask) == (net2 & nmask));
1387 }
1388
1389
1390 /****************************************************************************
1391  Check if a process exists. Does this work on all unixes?
1392 ****************************************************************************/
1393
1394 BOOL process_exists(pid_t pid)
1395 {
1396         /* Doing kill with a non-positive pid causes messages to be
1397          * sent to places we don't want. */
1398         SMB_ASSERT(pid > 0);
1399         return(kill(pid,0) == 0 || errno != ESRCH);
1400 }
1401
1402 /*******************************************************************
1403  Convert a uid into a user name.
1404 ********************************************************************/
1405
1406 const char *uidtoname(uid_t uid)
1407 {
1408         static fstring name;
1409         struct passwd *pass;
1410
1411         pass = getpwuid_alloc(uid);
1412         if (pass) {
1413                 fstrcpy(name, pass->pw_name);
1414                 passwd_free(&pass);
1415         } else {
1416                 slprintf(name, sizeof(name) - 1, "%ld",(long int)uid);
1417         }
1418         return name;
1419 }
1420
1421
1422 /*******************************************************************
1423  Convert a gid into a group name.
1424 ********************************************************************/
1425
1426 char *gidtoname(gid_t gid)
1427 {
1428         static fstring name;
1429         struct group *grp;
1430
1431         grp = getgrgid(gid);
1432         if (grp)
1433                 return(grp->gr_name);
1434         slprintf(name,sizeof(name) - 1, "%d",(int)gid);
1435         return(name);
1436 }
1437
1438 /*******************************************************************
1439  Convert a user name into a uid.
1440 ********************************************************************/
1441
1442 uid_t nametouid(const char *name)
1443 {
1444         struct passwd *pass;
1445         char *p;
1446         uid_t u;
1447
1448         pass = getpwnam_alloc(name);
1449         if (pass) {
1450                 u = pass->pw_uid;
1451                 passwd_free(&pass);
1452                 return u;
1453         }
1454
1455         u = (uid_t)strtol(name, &p, 0);
1456         if ((p != name) && (*p == '\0'))
1457                 return u;
1458
1459         return (uid_t)-1;
1460 }
1461
1462 /*******************************************************************
1463  Convert a name to a gid_t if possible. Return -1 if not a group.
1464 ********************************************************************/
1465
1466 gid_t nametogid(const char *name)
1467 {
1468         struct group *grp;
1469         char *p;
1470         gid_t g;
1471
1472         g = (gid_t)strtol(name, &p, 0);
1473         if ((p != name) && (*p == '\0'))
1474                 return g;
1475
1476         grp = sys_getgrnam(name);
1477         if (grp)
1478                 return(grp->gr_gid);
1479         return (gid_t)-1;
1480 }
1481
1482 /*******************************************************************
1483  legacy wrapper for smb_panic2()
1484 ********************************************************************/
1485 void smb_panic( const char *why )
1486 {
1487         smb_panic2( why, True );
1488 }
1489
1490 /*******************************************************************
1491  Something really nasty happened - panic !
1492 ********************************************************************/
1493
1494 #ifdef HAVE_LIBEXC_H
1495 #include <libexc.h>
1496 #endif
1497
1498 void smb_panic2(const char *why, BOOL decrement_pid_count )
1499 {
1500         char *cmd;
1501         int result;
1502 #ifdef HAVE_BACKTRACE_SYMBOLS
1503         void *backtrace_stack[BACKTRACE_STACK_SIZE];
1504         size_t backtrace_size;
1505         char **backtrace_strings;
1506 #endif
1507
1508 #ifdef DEVELOPER
1509         {
1510
1511                 if (global_clobber_region_function) {
1512                         DEBUG(0,("smb_panic: clobber_region() last called from [%s(%u)]\n",
1513                                          global_clobber_region_function,
1514                                          global_clobber_region_line));
1515                 }
1516         }
1517 #endif
1518
1519         /* only smbd needs to decrement the smbd counter in connections.tdb */
1520         if ( decrement_pid_count )
1521                 decrement_smbd_process_count();
1522
1523         cmd = lp_panic_action();
1524         if (cmd && *cmd) {
1525                 DEBUG(0, ("smb_panic(): calling panic action [%s]\n", cmd));
1526                 result = system(cmd);
1527
1528                 if (result == -1)
1529                         DEBUG(0, ("smb_panic(): fork failed in panic action: %s\n",
1530                                           strerror(errno)));
1531                 else
1532                         DEBUG(0, ("smb_panic(): action returned status %d\n",
1533                                           WEXITSTATUS(result)));
1534         }
1535         DEBUG(0,("PANIC: %s\n", why));
1536
1537 #ifdef HAVE_BACKTRACE_SYMBOLS
1538         /* get the backtrace (stack frames) */
1539         backtrace_size = backtrace(backtrace_stack,BACKTRACE_STACK_SIZE);
1540         backtrace_strings = backtrace_symbols(backtrace_stack, backtrace_size);
1541
1542         DEBUG(0, ("BACKTRACE: %lu stack frames:\n",
1543                   (unsigned long)backtrace_size));
1544
1545         if (backtrace_strings) {
1546                 int i;
1547
1548                 for (i = 0; i < backtrace_size; i++)
1549                         DEBUGADD(0, (" #%u %s\n", i, backtrace_strings[i]));
1550
1551                 /* Leak the backtrace_strings, rather than risk what free() might do */
1552         }
1553
1554 #elif HAVE_LIBEXC
1555
1556 #define NAMESIZE 32 /* Arbitrary */
1557
1558         /* The IRIX libexc library provides an API for unwinding the stack. See
1559          * libexc(3) for details. Apparantly trace_back_stack leaks memory, but
1560          * since we are about to abort anyway, it hardly matters.
1561          *
1562          * Note that if we paniced due to a SIGSEGV or SIGBUS (or similar) this
1563          * will fail with a nasty message upon failing to open the /proc entry.
1564          */
1565         {
1566                 __uint64_t      addrs[BACKTRACE_STACK_SIZE];
1567                 char *          names[BACKTRACE_STACK_SIZE];
1568                 char            namebuf[BACKTRACE_STACK_SIZE * NAMESIZE];
1569
1570                 int             i;
1571                 int             levels;
1572
1573                 ZERO_ARRAY(addrs);
1574                 ZERO_ARRAY(names);
1575                 ZERO_ARRAY(namebuf);
1576
1577                 /* We need to be root so we can open our /proc entry to walk
1578                  * our stack. It also helps when we want to dump core.
1579                  */
1580                 become_root();
1581
1582                 for (i = 0; i < BACKTRACE_STACK_SIZE; i++) {
1583                         names[i] = namebuf + (i * NAMESIZE);
1584                 }
1585
1586                 levels = trace_back_stack(0, addrs, names,
1587                                 BACKTRACE_STACK_SIZE, NAMESIZE);
1588
1589                 DEBUG(0, ("BACKTRACE: %d stack frames:\n", levels));
1590                 for (i = 0; i < levels; i++) {
1591                         DEBUGADD(0, (" #%d 0x%llx %s\n", i, addrs[i], names[i]));
1592                 }
1593      }
1594 #undef NAMESIZE
1595 #endif
1596
1597         dbgflush();
1598 #ifdef SIGABRT
1599         CatchSignal(SIGABRT,SIGNAL_CAST SIG_DFL);
1600 #endif
1601         abort();
1602 }
1603
1604 /*******************************************************************
1605   A readdir wrapper which just returns the file name.
1606  ********************************************************************/
1607
1608 const char *readdirname(DIR *p)
1609 {
1610         SMB_STRUCT_DIRENT *ptr;
1611         char *dname;
1612
1613         if (!p)
1614                 return(NULL);
1615
1616         ptr = (SMB_STRUCT_DIRENT *)sys_readdir(p);
1617         if (!ptr)
1618                 return(NULL);
1619
1620         dname = ptr->d_name;
1621
1622 #ifdef NEXT2
1623         if (telldir(p) < 0)
1624                 return(NULL);
1625 #endif
1626
1627 #ifdef HAVE_BROKEN_READDIR
1628         /* using /usr/ucb/cc is BAD */
1629         dname = dname - 2;
1630 #endif
1631
1632         {
1633                 static pstring buf;
1634                 int len = NAMLEN(ptr);
1635                 memcpy(buf, dname, len);
1636                 buf[len] = 0;
1637                 dname = buf;
1638         }
1639
1640         return(dname);
1641 }
1642
1643 /*******************************************************************
1644  Utility function used to decide if the last component
1645  of a path matches a (possibly wildcarded) entry in a namelist.
1646 ********************************************************************/
1647
1648 BOOL is_in_path(const char *name, name_compare_entry *namelist, BOOL case_sensitive)
1649 {
1650         pstring last_component;
1651         char *p;
1652
1653         /* if we have no list it's obviously not in the path */
1654         if((namelist == NULL ) || ((namelist != NULL) && (namelist[0].name == NULL))) {
1655                 return False;
1656         }
1657
1658         DEBUG(8, ("is_in_path: %s\n", name));
1659
1660         /* Get the last component of the unix name. */
1661         p = strrchr_m(name, '/');
1662         strncpy(last_component, p ? ++p : name, sizeof(last_component)-1);
1663         last_component[sizeof(last_component)-1] = '\0';
1664
1665         for(; namelist->name != NULL; namelist++) {
1666                 if(namelist->is_wild) {
1667                         if (mask_match(last_component, namelist->name, case_sensitive)) {
1668                                 DEBUG(8,("is_in_path: mask match succeeded\n"));
1669                                 return True;
1670                         }
1671                 } else {
1672                         if((case_sensitive && (strcmp(last_component, namelist->name) == 0))||
1673                                                 (!case_sensitive && (StrCaseCmp(last_component, namelist->name) == 0))) {
1674                                 DEBUG(8,("is_in_path: match succeeded\n"));
1675                                 return True;
1676                         }
1677                 }
1678         }
1679         DEBUG(8,("is_in_path: match not found\n"));
1680
1681         return False;
1682 }
1683
1684 /*******************************************************************
1685  Strip a '/' separated list into an array of
1686  name_compare_enties structures suitable for
1687  passing to is_in_path(). We do this for
1688  speed so we can pre-parse all the names in the list
1689  and don't do it for each call to is_in_path().
1690  namelist is modified here and is assumed to be
1691  a copy owned by the caller.
1692  We also check if the entry contains a wildcard to
1693  remove a potentially expensive call to mask_match
1694  if possible.
1695 ********************************************************************/
1696
1697 void set_namearray(name_compare_entry **ppname_array, char *namelist)
1698 {
1699         char *name_end;
1700         char *nameptr = namelist;
1701         int num_entries = 0;
1702         int i;
1703
1704         (*ppname_array) = NULL;
1705
1706         if((nameptr == NULL ) || ((nameptr != NULL) && (*nameptr == '\0')))
1707                 return;
1708
1709         /* We need to make two passes over the string. The
1710                 first to count the number of elements, the second
1711                 to split it.
1712         */
1713
1714         while(*nameptr) {
1715                 if ( *nameptr == '/' ) {
1716                         /* cope with multiple (useless) /s) */
1717                         nameptr++;
1718                         continue;
1719                 }
1720                 /* find the next / */
1721                 name_end = strchr_m(nameptr, '/');
1722
1723                 /* oops - the last check for a / didn't find one. */
1724                 if (name_end == NULL)
1725                         break;
1726
1727                 /* next segment please */
1728                 nameptr = name_end + 1;
1729                 num_entries++;
1730         }
1731
1732         if(num_entries == 0)
1733                 return;
1734
1735         if(( (*ppname_array) = SMB_MALLOC_ARRAY(name_compare_entry, num_entries + 1)) == NULL) {
1736                 DEBUG(0,("set_namearray: malloc fail\n"));
1737                 return;
1738         }
1739
1740         /* Now copy out the names */
1741         nameptr = namelist;
1742         i = 0;
1743         while(*nameptr) {
1744                 if ( *nameptr == '/' ) {
1745                         /* cope with multiple (useless) /s) */
1746                         nameptr++;
1747                         continue;
1748                 }
1749                 /* find the next / */
1750                 if ((name_end = strchr_m(nameptr, '/')) != NULL)
1751                         *name_end = 0;
1752
1753                 /* oops - the last check for a / didn't find one. */
1754                 if(name_end == NULL)
1755                         break;
1756
1757                 (*ppname_array)[i].is_wild = ms_has_wild(nameptr);
1758                 if(((*ppname_array)[i].name = SMB_STRDUP(nameptr)) == NULL) {
1759                         DEBUG(0,("set_namearray: malloc fail (1)\n"));
1760                         return;
1761                 }
1762
1763                 /* next segment please */
1764                 nameptr = name_end + 1;
1765                 i++;
1766         }
1767
1768         (*ppname_array)[i].name = NULL;
1769
1770         return;
1771 }
1772
1773 /****************************************************************************
1774  Routine to free a namearray.
1775 ****************************************************************************/
1776
1777 void free_namearray(name_compare_entry *name_array)
1778 {
1779         int i;
1780
1781         if(name_array == NULL)
1782                 return;
1783
1784         for(i=0; name_array[i].name!=NULL; i++)
1785                 SAFE_FREE(name_array[i].name);
1786         SAFE_FREE(name_array);
1787 }
1788
1789 /****************************************************************************
1790  Simple routine to do POSIX file locking. Cruft in NFS and 64->32 bit mapping
1791  is dealt with in posix.c
1792 ****************************************************************************/
1793
1794 BOOL fcntl_lock(int fd, int op, SMB_OFF_T offset, SMB_OFF_T count, int type)
1795 {
1796         SMB_STRUCT_FLOCK lock;
1797         int ret;
1798
1799         DEBUG(8,("fcntl_lock %d %d %.0f %.0f %d\n",fd,op,(double)offset,(double)count,type));
1800
1801         lock.l_type = type;
1802         lock.l_whence = SEEK_SET;
1803         lock.l_start = offset;
1804         lock.l_len = count;
1805         lock.l_pid = 0;
1806
1807         ret = sys_fcntl_ptr(fd,op,&lock);
1808
1809         if (ret == -1 && errno != 0)
1810                 DEBUG(3,("fcntl_lock: fcntl lock gave errno %d (%s)\n",errno,strerror(errno)));
1811
1812         /* a lock query */
1813         if (op == SMB_F_GETLK) {
1814                 if ((ret != -1) &&
1815                                 (lock.l_type != F_UNLCK) &&
1816                                 (lock.l_pid != 0) &&
1817                                 (lock.l_pid != sys_getpid())) {
1818                         DEBUG(3,("fcntl_lock: fd %d is locked by pid %d\n",fd,(int)lock.l_pid));
1819                         return(True);
1820                 }
1821
1822                 /* it must be not locked or locked by me */
1823                 return(False);
1824         }
1825
1826         /* a lock set or unset */
1827         if (ret == -1) {
1828                 DEBUG(3,("fcntl_lock: lock failed at offset %.0f count %.0f op %d type %d (%s)\n",
1829                         (double)offset,(double)count,op,type,strerror(errno)));
1830                 return(False);
1831         }
1832
1833         /* everything went OK */
1834         DEBUG(8,("fcntl_lock: Lock call successful\n"));
1835
1836         return(True);
1837 }
1838
1839 /*******************************************************************
1840  Is the name specified one of my netbios names.
1841  Returns true if it is equal, false otherwise.
1842 ********************************************************************/
1843
1844 BOOL is_myname(const char *s)
1845 {
1846         int n;
1847         BOOL ret = False;
1848
1849         for (n=0; my_netbios_names(n); n++) {
1850                 if (strequal(my_netbios_names(n), s)) {
1851                         ret=True;
1852                         break;
1853                 }
1854         }
1855         DEBUG(8, ("is_myname(\"%s\") returns %d\n", s, ret));
1856         return(ret);
1857 }
1858
1859 BOOL is_myname_or_ipaddr(const char *s)
1860 {
1861         fstring name, dnsname;
1862         char *servername;
1863
1864         if ( !s )
1865                 return False;
1866
1867         /* santize the string from '\\name' */
1868
1869         fstrcpy( name, s );
1870
1871         servername = strrchr_m( name, '\\' );
1872         if ( !servername )
1873                 servername = name;
1874         else
1875                 servername++;
1876
1877         /* optimize for the common case */
1878
1879         if (strequal(servername, global_myname()))
1880                 return True;
1881
1882         /* check for an alias */
1883
1884         if (is_myname(servername))
1885                 return True;
1886
1887         /* check for loopback */
1888
1889         if (strequal(servername, "localhost"))
1890                 return True;
1891
1892         /* maybe it's my dns name */
1893
1894         if ( get_mydnsfullname( dnsname ) )
1895                 if ( strequal( servername, dnsname ) )
1896                         return True;
1897
1898         /* handle possible CNAME records */
1899
1900         if ( !is_ipaddress( servername ) ) {
1901                 /* use DNS to resolve the name, but only the first address */
1902                 struct hostent *hp;
1903
1904                 if (((hp = sys_gethostbyname(name)) != NULL) && (hp->h_addr != NULL)) {
1905                         struct in_addr return_ip;
1906                         putip( (char*)&return_ip, (char*)hp->h_addr );
1907                         fstrcpy( name, inet_ntoa( return_ip ) );
1908                         servername = name;
1909                 }
1910         }
1911
1912         /* maybe its an IP address? */
1913         if (is_ipaddress(servername)) {
1914                 struct iface_struct nics[MAX_INTERFACES];
1915                 int i, n;
1916                 uint32 ip;
1917
1918                 ip = interpret_addr(servername);
1919                 if ((ip==0) || (ip==0xffffffff))
1920                         return False;
1921
1922                 n = get_interfaces(nics, MAX_INTERFACES);
1923                 for (i=0; i<n; i++) {
1924                         if (ip == nics[i].ip.s_addr)
1925                                 return True;
1926                 }
1927         }
1928
1929         /* no match */
1930         return False;
1931 }
1932
1933 /*******************************************************************
1934  Is the name specified our workgroup/domain.
1935  Returns true if it is equal, false otherwise.
1936 ********************************************************************/
1937
1938 BOOL is_myworkgroup(const char *s)
1939 {
1940         BOOL ret = False;
1941
1942         if (strequal(s, lp_workgroup())) {
1943                 ret=True;
1944         }
1945
1946         DEBUG(8, ("is_myworkgroup(\"%s\") returns %d\n", s, ret));
1947         return(ret);
1948 }
1949
1950 /*******************************************************************
1951  we distinguish between 2K and XP by the "Native Lan Manager" string
1952    WinXP => "Windows 2002 5.1"
1953    Win2k => "Windows 2000 5.0"
1954    NT4   => "Windows NT 4.0"
1955    Win9x => "Windows 4.0"
1956  Windows 2003 doesn't set the native lan manager string but
1957  they do set the domain to "Windows 2003 5.2" (probably a bug).
1958 ********************************************************************/
1959
1960 void ra_lanman_string( const char *native_lanman )
1961 {
1962         if ( strcmp( native_lanman, "Windows 2002 5.1" ) == 0 )
1963                 set_remote_arch( RA_WINXP );
1964         else if ( strcmp( native_lanman, "Windows Server 2003 5.2" ) == 0 )
1965                 set_remote_arch( RA_WIN2K3 );
1966 }
1967
1968 /*******************************************************************
1969  Set the horrid remote_arch string based on an enum.
1970 ********************************************************************/
1971
1972 void set_remote_arch(enum remote_arch_types type)
1973 {
1974         ra_type = type;
1975         switch( type ) {
1976         case RA_WFWG:
1977                 fstrcpy(remote_arch, "WfWg");
1978                 break;
1979         case RA_OS2:
1980                 fstrcpy(remote_arch, "OS2");
1981                 break;
1982         case RA_WIN95:
1983                 fstrcpy(remote_arch, "Win95");
1984                 break;
1985         case RA_WINNT:
1986                 fstrcpy(remote_arch, "WinNT");
1987                 break;
1988         case RA_WIN2K:
1989                 fstrcpy(remote_arch, "Win2K");
1990                 break;
1991         case RA_WINXP:
1992                 fstrcpy(remote_arch, "WinXP");
1993                 break;
1994         case RA_WIN2K3:
1995                 fstrcpy(remote_arch, "Win2K3");
1996                 break;
1997         case RA_SAMBA:
1998                 fstrcpy(remote_arch,"Samba");
1999                 break;
2000         case RA_CIFSFS:
2001                 fstrcpy(remote_arch,"CIFSFS");
2002                 break;
2003         default:
2004                 ra_type = RA_UNKNOWN;
2005                 fstrcpy(remote_arch, "UNKNOWN");
2006                 break;
2007         }
2008
2009         DEBUG(10,("set_remote_arch: Client arch is \'%s\'\n", remote_arch));
2010 }
2011
2012 /*******************************************************************
2013  Get the remote_arch type.
2014 ********************************************************************/
2015
2016 enum remote_arch_types get_remote_arch(void)
2017 {
2018         return ra_type;
2019 }
2020
2021 void print_asc(int level, const unsigned char *buf,int len)
2022 {
2023         int i;
2024         for (i=0;i<len;i++)
2025                 DEBUG(level,("%c", isprint(buf[i])?buf[i]:'.'));
2026 }
2027
2028 void dump_data(int level, const char *buf1,int len)
2029 {
2030         const unsigned char *buf = (const unsigned char *)buf1;
2031         int i=0;
2032         if (len<=0) return;
2033
2034         if (!DEBUGLVL(level)) return;
2035
2036         DEBUGADD(level,("[%03X] ",i));
2037         for (i=0;i<len;) {
2038                 DEBUGADD(level,("%02X ",(int)buf[i]));
2039                 i++;
2040                 if (i%8 == 0) DEBUGADD(level,(" "));
2041                 if (i%16 == 0) {
2042                         print_asc(level,&buf[i-16],8); DEBUGADD(level,(" "));
2043                         print_asc(level,&buf[i-8],8); DEBUGADD(level,("\n"));
2044                         if (i<len) DEBUGADD(level,("[%03X] ",i));
2045                 }
2046         }
2047         if (i%16) {
2048                 int n;
2049                 n = 16 - (i%16);
2050                 DEBUGADD(level,(" "));
2051                 if (n>8) DEBUGADD(level,(" "));
2052                 while (n--) DEBUGADD(level,("   "));
2053                 n = MIN(8,i%16);
2054                 print_asc(level,&buf[i-(i%16)],n); DEBUGADD(level,( " " ));
2055                 n = (i%16) - n;
2056                 if (n>0) print_asc(level,&buf[i-n],n);
2057                 DEBUGADD(level,("\n"));
2058         }
2059 }
2060
2061 void dump_data_pw(const char *msg, const uchar * data, size_t len)
2062 {
2063 #ifdef DEBUG_PASSWORD
2064         DEBUG(11, ("%s", msg));
2065         if (data != NULL && len > 0)
2066         {
2067                 dump_data(11, data, len);
2068         }
2069 #endif
2070 }
2071
2072 char *tab_depth(int depth)
2073 {
2074         static pstring spaces;
2075         memset(spaces, ' ', depth * 4);
2076         spaces[depth * 4] = 0;
2077         return spaces;
2078 }
2079
2080 /*****************************************************************************
2081  Provide a checksum on a string
2082
2083  Input:  s - the null-terminated character string for which the checksum
2084              will be calculated.
2085
2086   Output: The checksum value calculated for s.
2087 *****************************************************************************/
2088
2089 int str_checksum(const char *s)
2090 {
2091         int res = 0;
2092         int c;
2093         int i=0;
2094
2095         while(*s) {
2096                 c = *s;
2097                 res ^= (c << (i % 15)) ^ (c >> (15-(i%15)));
2098                 s++;
2099                 i++;
2100         }
2101         return(res);
2102 }
2103
2104 /*****************************************************************
2105  Zero a memory area then free it. Used to catch bugs faster.
2106 *****************************************************************/
2107
2108 void zero_free(void *p, size_t size)
2109 {
2110         memset(p, 0, size);
2111         SAFE_FREE(p);
2112 }
2113
2114 /*****************************************************************
2115  Set our open file limit to a requested max and return the limit.
2116 *****************************************************************/
2117
2118 int set_maxfiles(int requested_max)
2119 {
2120 #if (defined(HAVE_GETRLIMIT) && defined(RLIMIT_NOFILE))
2121         struct rlimit rlp;
2122         int saved_current_limit;
2123
2124         if(getrlimit(RLIMIT_NOFILE, &rlp)) {
2125                 DEBUG(0,("set_maxfiles: getrlimit (1) for RLIMIT_NOFILE failed with error %s\n",
2126                         strerror(errno) ));
2127                 /* just guess... */
2128                 return requested_max;
2129         }
2130
2131         /*
2132          * Set the fd limit to be real_max_open_files + MAX_OPEN_FUDGEFACTOR to
2133          * account for the extra fd we need
2134          * as well as the log files and standard
2135          * handles etc. Save the limit we want to set in case
2136          * we are running on an OS that doesn't support this limit (AIX)
2137          * which always returns RLIM_INFINITY for rlp.rlim_max.
2138          */
2139
2140         /* Try raising the hard (max) limit to the requested amount. */
2141
2142 #if defined(RLIM_INFINITY)
2143         if (rlp.rlim_max != RLIM_INFINITY) {
2144                 int orig_max = rlp.rlim_max;
2145
2146                 if ( rlp.rlim_max < requested_max )
2147                         rlp.rlim_max = requested_max;
2148
2149                 /* This failing is not an error - many systems (Linux) don't
2150                         support our default request of 10,000 open files. JRA. */
2151
2152                 if(setrlimit(RLIMIT_NOFILE, &rlp)) {
2153                         DEBUG(3,("set_maxfiles: setrlimit for RLIMIT_NOFILE for %d max files failed with error %s\n",
2154                                 (int)rlp.rlim_max, strerror(errno) ));
2155
2156                         /* Set failed - restore original value from get. */
2157                         rlp.rlim_max = orig_max;
2158                 }
2159         }
2160 #endif
2161
2162         /* Now try setting the soft (current) limit. */
2163
2164         saved_current_limit = rlp.rlim_cur = MIN(requested_max,rlp.rlim_max);
2165
2166         if(setrlimit(RLIMIT_NOFILE, &rlp)) {
2167                 DEBUG(0,("set_maxfiles: setrlimit for RLIMIT_NOFILE for %d files failed with error %s\n",
2168                         (int)rlp.rlim_cur, strerror(errno) ));
2169                 /* just guess... */
2170                 return saved_current_limit;
2171         }
2172
2173         if(getrlimit(RLIMIT_NOFILE, &rlp)) {
2174                 DEBUG(0,("set_maxfiles: getrlimit (2) for RLIMIT_NOFILE failed with error %s\n",
2175                         strerror(errno) ));
2176                 /* just guess... */
2177                 return saved_current_limit;
2178     }
2179
2180 #if defined(RLIM_INFINITY)
2181         if(rlp.rlim_cur == RLIM_INFINITY)
2182                 return saved_current_limit;
2183 #endif
2184
2185         if((int)rlp.rlim_cur > saved_current_limit)
2186                 return saved_current_limit;
2187
2188         return rlp.rlim_cur;
2189 #else /* !defined(HAVE_GETRLIMIT) || !defined(RLIMIT_NOFILE) */
2190         /*
2191          * No way to know - just guess...
2192          */
2193         return requested_max;
2194 #endif
2195 }
2196
2197 /*****************************************************************
2198  Splits out the start of the key (HKLM or HKU) and the rest of the key.
2199 *****************************************************************/
2200
2201 BOOL reg_split_key(const char *full_keyname, uint32 *reg_type, char *key_name)
2202 {
2203         pstring tmp;
2204
2205         if (!next_token(&full_keyname, tmp, "\\", sizeof(tmp)))
2206                 return False;
2207
2208         (*reg_type) = 0;
2209
2210         DEBUG(10, ("reg_split_key: hive %s\n", tmp));
2211
2212         if (strequal(tmp, "HKLM") || strequal(tmp, "HKEY_LOCAL_MACHINE"))
2213                 (*reg_type) = HKEY_LOCAL_MACHINE;
2214         else if (strequal(tmp, "HKCR") || strequal(tmp, "HKEY_CLASSES_ROOT"))
2215                 (*reg_type) = HKEY_CLASSES_ROOT;
2216         else if (strequal(tmp, "HKU") || strequal(tmp, "HKEY_USERS"))
2217                 (*reg_type) = HKEY_USERS;
2218         else if (strequal(tmp, "HKPD")||strequal(tmp, "HKEY_PERFORMANCE_DATA"))
2219                 (*reg_type) = HKEY_PERFORMANCE_DATA;
2220         else {
2221                 DEBUG(10,("reg_split_key: unrecognised hive key %s\n", tmp));
2222                 return False;
2223         }
2224
2225         if (next_token(&full_keyname, tmp, "\n\r", sizeof(tmp)))
2226                 fstrcpy(key_name, tmp);
2227         else
2228                 key_name[0] = 0;
2229
2230         DEBUG(10, ("reg_split_key: name %s\n", key_name));
2231
2232         return True;
2233 }
2234
2235 /*****************************************************************
2236  Possibly replace mkstemp if it is broken.
2237 *****************************************************************/
2238
2239 int smb_mkstemp(char *template)
2240 {
2241 #if HAVE_SECURE_MKSTEMP
2242         return mkstemp(template);
2243 #else
2244         /* have a reasonable go at emulating it. Hope that
2245            the system mktemp() isn't completly hopeless */
2246         char *p = mktemp(template);
2247         if (!p)
2248                 return -1;
2249         return open(p, O_CREAT|O_EXCL|O_RDWR, 0600);
2250 #endif
2251 }
2252
2253 /*****************************************************************
2254  malloc that aborts with smb_panic on fail or zero size.
2255  *****************************************************************/
2256
2257 void *smb_xmalloc_array(size_t size, unsigned int count)
2258 {
2259         void *p;
2260         if (size == 0)
2261                 smb_panic("smb_xmalloc_array: called with zero size.\n");
2262         if (count >= MAX_ALLOC_SIZE/size) {
2263                 smb_panic("smb_xmalloc: alloc size too large.\n");
2264         }
2265         if ((p = SMB_MALLOC(size*count)) == NULL) {
2266                 DEBUG(0, ("smb_xmalloc_array failed to allocate %lu * %lu bytes\n",
2267                         (unsigned long)size, (unsigned long)count));
2268                 smb_panic("smb_xmalloc_array: malloc fail.\n");
2269         }
2270         return p;
2271 }
2272
2273 /**
2274  Memdup with smb_panic on fail.
2275 **/
2276
2277 void *smb_xmemdup(const void *p, size_t size)
2278 {
2279         void *p2;
2280         p2 = SMB_XMALLOC_ARRAY(unsigned char,size);
2281         memcpy(p2, p, size);
2282         return p2;
2283 }
2284
2285 /**
2286  strdup that aborts on malloc fail.
2287 **/
2288
2289 char *smb_xstrdup(const char *s)
2290 {
2291 #if defined(PARANOID_MALLOC_CHECKER)
2292 #ifdef strdup
2293 #undef strdup
2294 #endif
2295 #endif
2296         char *s1 = strdup(s);
2297 #if defined(PARANOID_MALLOC_CHECKER)
2298 #define strdup(s) __ERROR_DONT_USE_STRDUP_DIRECTLY
2299 #endif
2300         if (!s1)
2301                 smb_panic("smb_xstrdup: malloc fail\n");
2302         return s1;
2303
2304 }
2305
2306 /**
2307  strndup that aborts on malloc fail.
2308 **/
2309
2310 char *smb_xstrndup(const char *s, size_t n)
2311 {
2312 #if defined(PARANOID_MALLOC_CHECKER)
2313 #ifdef strndup
2314 #undef strndup
2315 #endif
2316 #endif
2317         char *s1 = strndup(s, n);
2318 #if defined(PARANOID_MALLOC_CHECKER)
2319 #define strndup(s,n) __ERROR_DONT_USE_STRNDUP_DIRECTLY
2320 #endif
2321         if (!s1)
2322                 smb_panic("smb_xstrndup: malloc fail\n");
2323         return s1;
2324 }
2325
2326 /*
2327   vasprintf that aborts on malloc fail
2328 */
2329
2330  int smb_xvasprintf(char **ptr, const char *format, va_list ap)
2331 {
2332         int n;
2333         va_list ap2;
2334
2335         VA_COPY(ap2, ap);
2336
2337         n = vasprintf(ptr, format, ap2);
2338         if (n == -1 || ! *ptr)
2339                 smb_panic("smb_xvasprintf: out of memory");
2340         return n;
2341 }
2342
2343 /*****************************************************************
2344  Like strdup but for memory.
2345 *****************************************************************/
2346
2347 void *memdup(const void *p, size_t size)
2348 {
2349         void *p2;
2350         if (size == 0)
2351                 return NULL;
2352         p2 = SMB_MALLOC(size);
2353         if (!p2)
2354                 return NULL;
2355         memcpy(p2, p, size);
2356         return p2;
2357 }
2358
2359 /*****************************************************************
2360  Get local hostname and cache result.
2361 *****************************************************************/
2362
2363 char *myhostname(void)
2364 {
2365         static pstring ret;
2366         if (ret[0] == 0)
2367                 get_myname(ret);
2368         return ret;
2369 }
2370
2371 /*****************************************************************
2372  A useful function for returning a path in the Samba lock directory.
2373 *****************************************************************/
2374
2375 char *lock_path(const char *name)
2376 {
2377         static pstring fname;
2378
2379         pstrcpy(fname,lp_lockdir());
2380         trim_char(fname,'\0','/');
2381
2382         if (!directory_exist(fname,NULL))
2383                 mkdir(fname,0755);
2384
2385         pstrcat(fname,"/");
2386         pstrcat(fname,name);
2387
2388         return fname;
2389 }
2390
2391 /*****************************************************************
2392  A useful function for returning a path in the Samba pid directory.
2393 *****************************************************************/
2394
2395 char *pid_path(const char *name)
2396 {
2397         static pstring fname;
2398
2399         pstrcpy(fname,lp_piddir());
2400         trim_char(fname,'\0','/');
2401
2402         if (!directory_exist(fname,NULL))
2403                 mkdir(fname,0755);
2404
2405         pstrcat(fname,"/");
2406         pstrcat(fname,name);
2407
2408         return fname;
2409 }
2410
2411 /**
2412  * @brief Returns an absolute path to a file in the Samba lib directory.
2413  *
2414  * @param name File to find, relative to LIBDIR.
2415  *
2416  * @retval Pointer to a static #pstring containing the full path.
2417  **/
2418
2419 char *lib_path(const char *name)
2420 {
2421         static pstring fname;
2422         fstr_sprintf(fname, "%s/%s", dyn_LIBDIR, name);
2423         return fname;
2424 }
2425
2426 /**
2427  * @brief Returns the platform specific shared library extension.
2428  *
2429  * @retval Pointer to a static #fstring containing the extension.
2430  **/
2431
2432 const char *shlib_ext(void)
2433 {
2434   return dyn_SHLIBEXT;
2435 }
2436
2437 /*******************************************************************
2438  Given a filename - get its directory name
2439  NB: Returned in static storage.  Caveats:
2440  o  Not safe in thread environment.
2441  o  Caller must not free.
2442  o  If caller wishes to preserve, they should copy.
2443 ********************************************************************/
2444
2445 char *parent_dirname(const char *path)
2446 {
2447         static pstring dirpath;
2448         char *p;
2449
2450         if (!path)
2451                 return(NULL);
2452
2453         pstrcpy(dirpath, path);
2454         p = strrchr_m(dirpath, '/');  /* Find final '/', if any */
2455         if (!p) {
2456                 pstrcpy(dirpath, ".");    /* No final "/", so dir is "." */
2457         } else {
2458                 if (p == dirpath)
2459                         ++p;    /* For root "/", leave "/" in place */
2460                 *p = '\0';
2461         }
2462         return dirpath;
2463 }
2464
2465
2466 /*******************************************************************
2467  Determine if a pattern contains any Microsoft wildcard characters.
2468 *******************************************************************/
2469
2470 BOOL ms_has_wild(const char *s)
2471 {
2472         char c;
2473         while ((c = *s++)) {
2474                 switch (c) {
2475                 case '*':
2476                 case '?':
2477                 case '<':
2478                 case '>':
2479                 case '"':
2480                         return True;
2481                 }
2482         }
2483         return False;
2484 }
2485
2486 BOOL ms_has_wild_w(const smb_ucs2_t *s)
2487 {
2488         smb_ucs2_t c;
2489         if (!s) return False;
2490         while ((c = *s++)) {
2491                 switch (c) {
2492                 case UCS2_CHAR('*'):
2493                 case UCS2_CHAR('?'):
2494                 case UCS2_CHAR('<'):
2495                 case UCS2_CHAR('>'):
2496                 case UCS2_CHAR('"'):
2497                         return True;
2498                 }
2499         }
2500         return False;
2501 }
2502
2503 /*******************************************************************
2504  A wrapper that handles case sensitivity and the special handling
2505  of the ".." name.
2506 *******************************************************************/
2507
2508 BOOL mask_match(const char *string, char *pattern, BOOL is_case_sensitive)
2509 {
2510         if (strcmp(string,"..") == 0)
2511                 string = ".";
2512         if (strcmp(pattern,".") == 0)
2513                 return False;
2514
2515         return ms_fnmatch(pattern, string, Protocol <= PROTOCOL_LANMAN2, is_case_sensitive) == 0;
2516 }
2517
2518 /*******************************************************************
2519  A wrapper that handles case sensitivity and the special handling
2520  of the ".." name. Varient that is only called by old search code which requires
2521  pattern translation.
2522 *******************************************************************/
2523
2524 BOOL mask_match_search(const char *string, char *pattern, BOOL is_case_sensitive)
2525 {
2526         if (strcmp(string,"..") == 0)
2527                 string = ".";
2528         if (strcmp(pattern,".") == 0)
2529                 return False;
2530
2531         return ms_fnmatch(pattern, string, True, is_case_sensitive) == 0;
2532 }
2533
2534 /*******************************************************************
2535  A wrapper that handles a list of patters and calls mask_match()
2536  on each.  Returns True if any of the patterns match.
2537 *******************************************************************/
2538
2539 BOOL mask_match_list(const char *string, char **list, int listLen, BOOL is_case_sensitive)
2540 {
2541        while (listLen-- > 0) {
2542                if (mask_match(string, *list++, is_case_sensitive))
2543                        return True;
2544        }
2545        return False;
2546 }
2547
2548 /*********************************************************
2549  Recursive routine that is called by unix_wild_match.
2550 *********************************************************/
2551
2552 static BOOL unix_do_match(const char *regexp, const char *str)
2553 {
2554         const char *p;
2555
2556         for( p = regexp; *p && *str; ) {
2557
2558                 switch(*p) {
2559                         case '?':
2560                                 str++;
2561                                 p++;
2562                                 break;
2563
2564                         case '*':
2565
2566                                 /*
2567                                  * Look for a character matching
2568                                  * the one after the '*'.
2569                                  */
2570                                 p++;
2571                                 if(!*p)
2572                                         return True; /* Automatic match */
2573                                 while(*str) {
2574
2575                                         while(*str && (*p != *str))
2576                                                 str++;
2577
2578                                         /*
2579                                          * Patch from weidel@multichart.de. In the case of the regexp
2580                                          * '*XX*' we want to ensure there are at least 2 'X' characters
2581                                          * in the string after the '*' for a match to be made.
2582                                          */
2583
2584                                         {
2585                                                 int matchcount=0;
2586
2587                                                 /*
2588                                                  * Eat all the characters that match, but count how many there were.
2589                                                  */
2590
2591                                                 while(*str && (*p == *str)) {
2592                                                         str++;
2593                                                         matchcount++;
2594                                                 }
2595
2596                                                 /*
2597                                                  * Now check that if the regexp had n identical characters that
2598                                                  * matchcount had at least that many matches.
2599                                                  */
2600
2601                                                 while ( *(p+1) && (*(p+1) == *p)) {
2602                                                         p++;
2603                                                         matchcount--;
2604                                                 }
2605
2606                                                 if ( matchcount <= 0 )
2607                                                         return False;
2608                                         }
2609
2610                                         str--; /* We've eaten the match char after the '*' */
2611
2612                                         if(unix_do_match(p, str))
2613                                                 return True;
2614
2615                                         if(!*str)
2616                                                 return False;
2617                                         else
2618                                                 str++;
2619                                 }
2620                                 return False;
2621
2622                         default:
2623                                 if(*str != *p)
2624                                         return False;
2625                                 str++;
2626                                 p++;
2627                                 break;
2628                 }
2629         }
2630
2631         if(!*p && !*str)
2632                 return True;
2633
2634         if (!*p && str[0] == '.' && str[1] == 0)
2635                 return(True);
2636
2637         if (!*str && *p == '?') {
2638                 while (*p == '?')
2639                         p++;
2640                 return(!*p);
2641         }
2642
2643         if(!*str && (*p == '*' && p[1] == '\0'))
2644                 return True;
2645
2646         return False;
2647 }
2648
2649 /*******************************************************************
2650  Simple case insensitive interface to a UNIX wildcard matcher.
2651 *******************************************************************/
2652
2653 BOOL unix_wild_match(const char *pattern, const char *string)
2654 {
2655         pstring p2, s2;
2656         char *p;
2657
2658         pstrcpy(p2, pattern);
2659         pstrcpy(s2, string);
2660         strlower_m(p2);
2661         strlower_m(s2);
2662
2663         /* Remove any *? and ** from the pattern as they are meaningless */
2664         for(p = p2; *p; p++)
2665                 while( *p == '*' && (p[1] == '?' ||p[1] == '*'))
2666                         pstrcpy( &p[1], &p[2]);
2667
2668         if (strequal(p2,"*"))
2669                 return True;
2670
2671         return unix_do_match(p2, s2) == 0;
2672 }
2673
2674 /**********************************************************************
2675  Converts a name to a fully qalified domain name.
2676 ***********************************************************************/
2677
2678 void name_to_fqdn(fstring fqdn, const char *name)
2679 {
2680         struct hostent *hp = sys_gethostbyname(name);
2681         if ( hp && hp->h_name && *hp->h_name ) {
2682                 DEBUG(10,("name_to_fqdn: lookup for %s -> %s.\n", name, hp->h_name));
2683                 fstrcpy(fqdn,hp->h_name);
2684         } else {
2685                 DEBUG(10,("name_to_fqdn: lookup for %s failed.\n", name));
2686                 fstrcpy(fqdn, name);
2687         }
2688 }
2689
2690 #ifdef __INSURE__
2691
2692 /*******************************************************************
2693 This routine is a trick to immediately catch errors when debugging
2694 with insure. A xterm with a gdb is popped up when insure catches
2695 a error. It is Linux specific.
2696 ********************************************************************/
2697
2698 int _Insure_trap_error(int a1, int a2, int a3, int a4, int a5, int a6)
2699 {
2700         static int (*fn)();
2701         int ret;
2702         char pidstr[10];
2703         /* you can get /usr/bin/backtrace from
2704            http://samba.org/ftp/unpacked/junkcode/backtrace */
2705         pstring cmd = "/usr/bin/backtrace %d";
2706
2707         slprintf(pidstr, sizeof(pidstr)-1, "%d", sys_getpid());
2708         pstring_sub(cmd, "%d", pidstr);
2709
2710         if (!fn) {
2711                 static void *h;
2712                 h = dlopen("/usr/local/parasoft/insure++lite/lib.linux2/libinsure.so", RTLD_LAZY);
2713                 fn = dlsym(h, "_Insure_trap_error");
2714
2715                 if (!h || h == _Insure_trap_error) {
2716                         h = dlopen("/usr/local/parasoft/lib.linux2/libinsure.so", RTLD_LAZY);
2717                         fn = dlsym(h, "_Insure_trap_error");
2718                 }
2719         }
2720
2721         ret = fn(a1, a2, a3, a4, a5, a6);
2722
2723         system(cmd);
2724
2725         return ret;
2726 }
2727 #endif