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 < 0)
 988                 return;
 989
 990         if (*array == NULL) {
 991                 if (*array_size == 0) {
 992                         *array_size = 128;
 993                 }
 994
 995                 if (*array_size >= MAX_ALLOC_SIZE/element_size) {
 996                         goto error;
 997                 }
 998
 999                 if (mem_ctx != NULL)
1000                         *array = TALLOC(mem_ctx, element_size * (*array_size));
1001                 else
1002                         *array = SMB_MALLOC(element_size * (*array_size));
1003
1004                 if (*array == NULL)
1005                         goto error;
1006         }
1007
1008         if (*num_elements == *array_size) {
1009                 *array_size *= 2;
1010
1011                 if (*array_size >= MAX_ALLOC_SIZE/element_size) {
1012                         goto error;
1013                 }
1014
1015                 if (mem_ctx != NULL)
1016                         *array = TALLOC_REALLOC(mem_ctx, *array,
1017                                                 element_size * (*array_size));
1018                 else
1019                         *array = SMB_REALLOC(*array,
1020                                              element_size * (*array_size));
1021
1022                 if (*array == NULL)
1023                         goto error;
1024         }
1025
1026         memcpy((char *)(*array) + element_size*(*num_elements),
1027                element, element_size);
1028         *num_elements += 1;
1029
1030         return;
1031
1032  error:
1033         *num_elements = 0;
1034         *array_size = -1;
1035 }
1036
1037 /****************************************************************************
1038  Free memory, checks for NULL.
1039  Use directly SAFE_FREE()
1040  Exists only because we need to pass a function pointer somewhere --SSS
1041 ****************************************************************************/
1042
1043 void safe_free(void *p)
1044 {
1045         SAFE_FREE(p);
1046 }
1047
1048 /****************************************************************************
1049  Get my own name and IP.
1050 ****************************************************************************/
1051
1052 BOOL get_myname(char *my_name)
1053 {
1054         pstring hostname;
1055
1056         *hostname = 0;
1057
1058         /* get my host name */
1059         if (gethostname(hostname, sizeof(hostname)) == -1) {
1060                 DEBUG(0,("gethostname failed\n"));
1061                 return False;
1062         }
1063
1064         /* Ensure null termination. */
1065         hostname[sizeof(hostname)-1] = '\0';
1066
1067         if (my_name) {
1068                 /* split off any parts after an initial . */
1069                 char *p = strchr_m(hostname,'.');
1070
1071                 if (p)
1072                         *p = 0;
1073
1074                 fstrcpy(my_name,hostname);
1075         }
1076
1077         return(True);
1078 }
1079
1080 /****************************************************************************
1081  Get my own canonical name, including domain.
1082 ****************************************************************************/
1083
1084 BOOL get_mydnsfullname(fstring my_dnsname)
1085 {
1086         static fstring dnshostname;
1087         struct hostent *hp;
1088
1089         if (!*dnshostname) {
1090                 /* get my host name */
1091                 if (gethostname(dnshostname, sizeof(dnshostname)) == -1) {
1092                         *dnshostname = '\0';
1093                         DEBUG(0,("gethostname failed\n"));
1094                         return False;
1095                 }
1096
1097                 /* Ensure null termination. */
1098                 dnshostname[sizeof(dnshostname)-1] = '\0';
1099
1100                 /* Ensure we get the cannonical name. */
1101                 if (!(hp = sys_gethostbyname(dnshostname))) {
1102                         *dnshostname = '\0';
1103                         return False;
1104                 }
1105                 fstrcpy(dnshostname, hp->h_name);
1106         }
1107         fstrcpy(my_dnsname, dnshostname);
1108         return True;
1109 }
1110
1111 /****************************************************************************
1112  Get my own domain name.
1113 ****************************************************************************/
1114
1115 BOOL get_mydnsdomname(fstring my_domname)
1116 {
1117         fstring domname;
1118         char *p;
1119
1120         *my_domname = '\0';
1121         if (!get_mydnsfullname(domname)) {
1122                 return False;
1123         }
1124         p = strchr_m(domname, '.');
1125         if (p) {
1126                 p++;
1127                 fstrcpy(my_domname, p);
1128         }
1129
1130         return False;
1131 }
1132
1133 /****************************************************************************
1134  Interpret a protocol description string, with a default.
1135 ****************************************************************************/
1136
1137 int interpret_protocol(const char *str,int def)
1138 {
1139         if (strequal(str,"NT1"))
1140                 return(PROTOCOL_NT1);
1141         if (strequal(str,"LANMAN2"))
1142                 return(PROTOCOL_LANMAN2);
1143         if (strequal(str,"LANMAN1"))
1144                 return(PROTOCOL_LANMAN1);
1145         if (strequal(str,"CORE"))
1146                 return(PROTOCOL_CORE);
1147         if (strequal(str,"COREPLUS"))
1148                 return(PROTOCOL_COREPLUS);
1149         if (strequal(str,"CORE+"))
1150                 return(PROTOCOL_COREPLUS);
1151
1152         DEBUG(0,("Unrecognised protocol level %s\n",str));
1153
1154         return(def);
1155 }
1156
1157 /****************************************************************************
1158  Return true if a string could be a pure IP address.
1159 ****************************************************************************/
1160
1161 BOOL is_ipaddress(const char *str)
1162 {
1163         BOOL pure_address = True;
1164         int i;
1165
1166         for (i=0; pure_address && str[i]; i++)
1167                 if (!(isdigit((int)str[i]) || str[i] == '.'))
1168                         pure_address = False;
1169
1170         /* Check that a pure number is not misinterpreted as an IP */
1171         pure_address = pure_address && (strchr_m(str, '.') != NULL);
1172
1173         return pure_address;
1174 }
1175
1176 /****************************************************************************
1177  Interpret an internet address or name into an IP address in 4 byte form.
1178 ****************************************************************************/
1179
1180 uint32 interpret_addr(const char *str)
1181 {
1182         struct hostent *hp;
1183         uint32 res;
1184
1185         if (strcmp(str,"0.0.0.0") == 0)
1186                 return(0);
1187         if (strcmp(str,"255.255.255.255") == 0)
1188                 return(0xFFFFFFFF);
1189
1190   /* if it's in the form of an IP address then get the lib to interpret it */
1191         if (is_ipaddress(str)) {
1192                 res = inet_addr(str);
1193         } else {
1194                 /* otherwise assume it's a network name of some sort and use
1195                         sys_gethostbyname */
1196                 if ((hp = sys_gethostbyname(str)) == 0) {
1197                         DEBUG(3,("sys_gethostbyname: Unknown host. %s\n",str));
1198                         return 0;
1199                 }
1200
1201                 if(hp->h_addr == NULL) {
1202                         DEBUG(3,("sys_gethostbyname: host address is invalid for host %s\n",str));
1203                         return 0;
1204                 }
1205                 putip((char *)&res,(char *)hp->h_addr);
1206         }
1207
1208         if (res == (uint32)-1)
1209                 return(0);
1210
1211         return(res);
1212 }
1213
1214 /*******************************************************************
1215  A convenient addition to interpret_addr().
1216 ******************************************************************/
1217
1218 struct in_addr *interpret_addr2(const char *str)
1219 {
1220         static struct in_addr ret;
1221         uint32 a = interpret_addr(str);
1222         ret.s_addr = a;
1223         return(&ret);
1224 }
1225
1226 /*******************************************************************
1227  Check if an IP is the 0.0.0.0.
1228 ******************************************************************/
1229
1230 BOOL is_zero_ip(struct in_addr ip)
1231 {
1232         uint32 a;
1233         putip((char *)&a,(char *)&ip);
1234         return(a == 0);
1235 }
1236
1237 /*******************************************************************
1238  Set an IP to 0.0.0.0.
1239 ******************************************************************/
1240
1241 void zero_ip(struct in_addr *ip)
1242 {
1243         static BOOL init;
1244         static struct in_addr ipzero;
1245
1246         if (!init) {
1247                 ipzero = *interpret_addr2("0.0.0.0");
1248                 init = True;
1249         }
1250
1251         *ip = ipzero;
1252 }
1253
1254 #if (defined(HAVE_NETGROUP) && defined(WITH_AUTOMOUNT))
1255 /******************************************************************
1256  Remove any mount options such as -rsize=2048,wsize=2048 etc.
1257  Based on a fix from <Thomas.Hepper@icem.de>.
1258 *******************************************************************/
1259
1260 static void strip_mount_options( pstring *str)
1261 {
1262         if (**str == '-') {
1263                 char *p = *str;
1264                 while(*p && !isspace(*p))
1265                         p++;
1266                 while(*p && isspace(*p))
1267                         p++;
1268                 if(*p) {
1269                         pstring tmp_str;
1270
1271                         pstrcpy(tmp_str, p);
1272                         pstrcpy(*str, tmp_str);
1273                 }
1274         }
1275 }
1276
1277 /*******************************************************************
1278  Patch from jkf@soton.ac.uk
1279  Split Luke's automount_server into YP lookup and string splitter
1280  so can easily implement automount_path().
1281  As we may end up doing both, cache the last YP result.
1282 *******************************************************************/
1283
1284 #ifdef WITH_NISPLUS_HOME
1285 char *automount_lookup( char *user_name)
1286 {
1287         static fstring last_key = "";
1288         static pstring last_value = "";
1289
1290         char *nis_map = (char *)lp_nis_home_map_name();
1291
1292         char buffer[NIS_MAXATTRVAL + 1];
1293         nis_result *result;
1294         nis_object *object;
1295         entry_obj  *entry;
1296
1297         if (strcmp(user_name, last_key)) {
1298                 slprintf(buffer, sizeof(buffer)-1, "[key=%s],%s", user_name, nis_map);
1299                 DEBUG(5, ("NIS+ querystring: %s\n", buffer));
1300
1301                 if (result = nis_list(buffer, FOLLOW_PATH|EXPAND_NAME|HARD_LOOKUP, NULL, NULL)) {
1302                         if (result->status != NIS_SUCCESS) {
1303                                 DEBUG(3, ("NIS+ query failed: %s\n", nis_sperrno(result->status)));
1304                                 fstrcpy(last_key, ""); pstrcpy(last_value, "");
1305                         } else {
1306                                 object = result->objects.objects_val;
1307                                 if (object->zo_data.zo_type == ENTRY_OBJ) {
1308                                         entry = &object->zo_data.objdata_u.en_data;
1309                                         DEBUG(5, ("NIS+ entry type: %s\n", entry->en_type));
1310                                         DEBUG(3, ("NIS+ result: %s\n", entry->en_cols.en_cols_val[1].ec_value.ec_value_val));
1311
1312                                         pstrcpy(last_value, entry->en_cols.en_cols_val[1].ec_value.ec_value_val);
1313                                         pstring_sub(last_value, "&", user_name);
1314                                         fstrcpy(last_key, user_name);
1315                                 }
1316                         }
1317                 }
1318                 nis_freeresult(result);
1319         }
1320
1321         strip_mount_options(&last_value);
1322
1323         DEBUG(4, ("NIS+ Lookup: %s resulted in %s\n", user_name, last_value));
1324         return last_value;
1325 }
1326 #else /* WITH_NISPLUS_HOME */
1327
1328 char *automount_lookup( char *user_name)
1329 {
1330         static fstring last_key = "";
1331         static pstring last_value = "";
1332
1333         int nis_error;        /* returned by yp all functions */
1334         char *nis_result;     /* yp_match inits this */
1335         int nis_result_len;  /* and set this */
1336         char *nis_domain;     /* yp_get_default_domain inits this */
1337         char *nis_map = (char *)lp_nis_home_map_name();
1338
1339         if ((nis_error = yp_get_default_domain(&nis_domain)) != 0) {
1340                 DEBUG(3, ("YP Error: %s\n", yperr_string(nis_error)));
1341                 return last_value;
1342         }
1343
1344         DEBUG(5, ("NIS Domain: %s\n", nis_domain));
1345
1346         if (!strcmp(user_name, last_key)) {
1347                 nis_result = last_value;
1348                 nis_result_len = strlen(last_value);
1349                 nis_error = 0;
1350         } else {
1351                 if ((nis_error = yp_match(nis_domain, nis_map, user_name, strlen(user_name),
1352                                 &nis_result, &nis_result_len)) == 0) {
1353                         if (!nis_error && nis_result_len >= sizeof(pstring)) {
1354                                 nis_result_len = sizeof(pstring)-1;
1355                         }
1356                         fstrcpy(last_key, user_name);
1357                         strncpy(last_value, nis_result, nis_result_len);
1358                         last_value[nis_result_len] = '\0';
1359                         strip_mount_options(&last_value);
1360
1361                 } else if(nis_error == YPERR_KEY) {
1362
1363                         /* If Key lookup fails user home server is not in nis_map
1364                                 use default information for server, and home directory */
1365                         last_value[0] = 0;
1366                         DEBUG(3, ("YP Key not found:  while looking up \"%s\" in map \"%s\"\n",
1367                                         user_name, nis_map));
1368                         DEBUG(3, ("using defaults for server and home directory\n"));
1369                 } else {
1370                         DEBUG(3, ("YP Error: \"%s\" while looking up \"%s\" in map \"%s\"\n",
1371                                         yperr_string(nis_error), user_name, nis_map));
1372                 }
1373         }
1374
1375         DEBUG(4, ("YP Lookup: %s resulted in %s\n", user_name, last_value));
1376         return last_value;
1377 }
1378 #endif /* WITH_NISPLUS_HOME */
1379 #endif
1380
1381 /*******************************************************************
1382  Are two IPs on the same subnet?
1383 ********************************************************************/
1384
1385 BOOL same_net(struct in_addr ip1,struct in_addr ip2,struct in_addr mask)
1386 {
1387         uint32 net1,net2,nmask;
1388
1389         nmask = ntohl(mask.s_addr);
1390         net1  = ntohl(ip1.s_addr);
1391         net2  = ntohl(ip2.s_addr);
1392
1393         return((net1 & nmask) == (net2 & nmask));
1394 }
1395
1396
1397 /****************************************************************************
1398  Check if a process exists. Does this work on all unixes?
1399 ****************************************************************************/
1400
1401 BOOL process_exists(pid_t pid)
1402 {
1403         /* Doing kill with a non-positive pid causes messages to be
1404          * sent to places we don't want. */
1405         SMB_ASSERT(pid > 0);
1406         return(kill(pid,0) == 0 || errno != ESRCH);
1407 }
1408
1409 /*******************************************************************
1410  Convert a uid into a user name.
1411 ********************************************************************/
1412
1413 const char *uidtoname(uid_t uid)
1414 {
1415         static fstring name;
1416         struct passwd *pass;
1417
1418         pass = getpwuid_alloc(uid);
1419         if (pass) {
1420                 fstrcpy(name, pass->pw_name);
1421                 passwd_free(&pass);
1422         } else {
1423                 slprintf(name, sizeof(name) - 1, "%ld",(long int)uid);
1424         }
1425         return name;
1426 }
1427
1428
1429 /*******************************************************************
1430  Convert a gid into a group name.
1431 ********************************************************************/
1432
1433 char *gidtoname(gid_t gid)
1434 {
1435         static fstring name;
1436         struct group *grp;
1437
1438         grp = getgrgid(gid);
1439         if (grp)
1440                 return(grp->gr_name);
1441         slprintf(name,sizeof(name) - 1, "%d",(int)gid);
1442         return(name);
1443 }
1444
1445 /*******************************************************************
1446  Convert a user name into a uid.
1447 ********************************************************************/
1448
1449 uid_t nametouid(const char *name)
1450 {
1451         struct passwd *pass;
1452         char *p;
1453         uid_t u;
1454
1455         pass = getpwnam_alloc(name);
1456         if (pass) {
1457                 u = pass->pw_uid;
1458                 passwd_free(&pass);
1459                 return u;
1460         }
1461
1462         u = (uid_t)strtol(name, &p, 0);
1463         if ((p != name) && (*p == '\0'))
1464                 return u;
1465
1466         return (uid_t)-1;
1467 }
1468
1469 /*******************************************************************
1470  Convert a name to a gid_t if possible. Return -1 if not a group.
1471 ********************************************************************/
1472
1473 gid_t nametogid(const char *name)
1474 {
1475         struct group *grp;
1476         char *p;
1477         gid_t g;
1478
1479         g = (gid_t)strtol(name, &p, 0);
1480         if ((p != name) && (*p == '\0'))
1481                 return g;
1482
1483         grp = sys_getgrnam(name);
1484         if (grp)
1485                 return(grp->gr_gid);
1486         return (gid_t)-1;
1487 }
1488
1489 /*******************************************************************
1490  legacy wrapper for smb_panic2()
1491 ********************************************************************/
1492 void smb_panic( const char *why )
1493 {
1494         smb_panic2( why, True );
1495 }
1496
1497 /*******************************************************************
1498  Something really nasty happened - panic !
1499 ********************************************************************/
1500
1501 #ifdef HAVE_LIBEXC_H
1502 #include <libexc.h>
1503 #endif
1504
1505 void smb_panic2(const char *why, BOOL decrement_pid_count )
1506 {
1507         char *cmd;
1508         int result;
1509 #ifdef HAVE_BACKTRACE_SYMBOLS
1510         void *backtrace_stack[BACKTRACE_STACK_SIZE];
1511         size_t backtrace_size;
1512         char **backtrace_strings;
1513 #endif
1514
1515 #ifdef DEVELOPER
1516         {
1517
1518                 if (global_clobber_region_function) {
1519                         DEBUG(0,("smb_panic: clobber_region() last called from [%s(%u)]\n",
1520                                          global_clobber_region_function,
1521                                          global_clobber_region_line));
1522                 }
1523         }
1524 #endif
1525
1526         /* only smbd needs to decrement the smbd counter in connections.tdb */
1527         if ( decrement_pid_count )
1528                 decrement_smbd_process_count();
1529
1530         cmd = lp_panic_action();
1531         if (cmd && *cmd) {
1532                 DEBUG(0, ("smb_panic(): calling panic action [%s]\n", cmd));
1533                 result = system(cmd);
1534
1535                 if (result == -1)
1536                         DEBUG(0, ("smb_panic(): fork failed in panic action: %s\n",
1537                                           strerror(errno)));
1538                 else
1539                         DEBUG(0, ("smb_panic(): action returned status %d\n",
1540                                           WEXITSTATUS(result)));
1541         }
1542         DEBUG(0,("PANIC: %s\n", why));
1543
1544 #ifdef HAVE_BACKTRACE_SYMBOLS
1545         /* get the backtrace (stack frames) */
1546         backtrace_size = backtrace(backtrace_stack,BACKTRACE_STACK_SIZE);
1547         backtrace_strings = backtrace_symbols(backtrace_stack, backtrace_size);
1548
1549         DEBUG(0, ("BACKTRACE: %lu stack frames:\n",
1550                   (unsigned long)backtrace_size));
1551
1552         if (backtrace_strings) {
1553                 int i;
1554
1555                 for (i = 0; i < backtrace_size; i++)
1556                         DEBUGADD(0, (" #%u %s\n", i, backtrace_strings[i]));
1557
1558                 /* Leak the backtrace_strings, rather than risk what free() might do */
1559         }
1560
1561 #elif HAVE_LIBEXC
1562
1563 #define NAMESIZE 32 /* Arbitrary */
1564
1565         /* The IRIX libexc library provides an API for unwinding the stack. See
1566          * libexc(3) for details. Apparantly trace_back_stack leaks memory, but
1567          * since we are about to abort anyway, it hardly matters.
1568          *
1569          * Note that if we paniced due to a SIGSEGV or SIGBUS (or similar) this
1570          * will fail with a nasty message upon failing to open the /proc entry.
1571          */
1572         {
1573                 __uint64_t      addrs[BACKTRACE_STACK_SIZE];
1574                 char *          names[BACKTRACE_STACK_SIZE];
1575                 char            namebuf[BACKTRACE_STACK_SIZE * NAMESIZE];
1576
1577                 int             i;
1578                 int             levels;
1579
1580                 ZERO_ARRAY(addrs);
1581                 ZERO_ARRAY(names);
1582                 ZERO_ARRAY(namebuf);
1583
1584                 /* We need to be root so we can open our /proc entry to walk
1585                  * our stack. It also helps when we want to dump core.
1586                  */
1587                 become_root();
1588
1589                 for (i = 0; i < BACKTRACE_STACK_SIZE; i++) {
1590                         names[i] = namebuf + (i * NAMESIZE);
1591                 }
1592
1593                 levels = trace_back_stack(0, addrs, names,
1594                                 BACKTRACE_STACK_SIZE, NAMESIZE - 1);
1595
1596                 DEBUG(0, ("BACKTRACE: %d stack frames:\n", levels));
1597                 for (i = 0; i < levels; i++) {
1598                         DEBUGADD(0, (" #%d 0x%llx %s\n", i, addrs[i], names[i]));
1599                 }
1600      }
1601 #undef NAMESIZE
1602 #endif
1603
1604         dbgflush();
1605 #ifdef SIGABRT
1606         CatchSignal(SIGABRT,SIGNAL_CAST SIG_DFL);
1607 #endif
1608         abort();
1609 }
1610
1611 /*******************************************************************
1612   A readdir wrapper which just returns the file name.
1613  ********************************************************************/
1614
1615 const char *readdirname(DIR *p)
1616 {
1617         SMB_STRUCT_DIRENT *ptr;
1618         char *dname;
1619
1620         if (!p)
1621                 return(NULL);
1622
1623         ptr = (SMB_STRUCT_DIRENT *)sys_readdir(p);
1624         if (!ptr)
1625                 return(NULL);
1626
1627         dname = ptr->d_name;
1628
1629 #ifdef NEXT2
1630         if (telldir(p) < 0)
1631                 return(NULL);
1632 #endif
1633
1634 #ifdef HAVE_BROKEN_READDIR
1635         /* using /usr/ucb/cc is BAD */
1636         dname = dname - 2;
1637 #endif
1638
1639         {
1640                 static pstring buf;
1641                 int len = NAMLEN(ptr);
1642                 memcpy(buf, dname, len);
1643                 buf[len] = 0;
1644                 dname = buf;
1645         }
1646
1647         return(dname);
1648 }
1649
1650 /*******************************************************************
1651  Utility function used to decide if the last component
1652  of a path matches a (possibly wildcarded) entry in a namelist.
1653 ********************************************************************/
1654
1655 BOOL is_in_path(const char *name, name_compare_entry *namelist, BOOL case_sensitive)
1656 {
1657         pstring last_component;
1658         char *p;
1659
1660         /* if we have no list it's obviously not in the path */
1661         if((namelist == NULL ) || ((namelist != NULL) && (namelist[0].name == NULL))) {
1662                 return False;
1663         }
1664
1665         DEBUG(8, ("is_in_path: %s\n", name));
1666
1667         /* Get the last component of the unix name. */
1668         p = strrchr_m(name, '/');
1669         strncpy(last_component, p ? ++p : name, sizeof(last_component)-1);
1670         last_component[sizeof(last_component)-1] = '\0';
1671
1672         for(; namelist->name != NULL; namelist++) {
1673                 if(namelist->is_wild) {
1674                         if (mask_match(last_component, namelist->name, case_sensitive)) {
1675                                 DEBUG(8,("is_in_path: mask match succeeded\n"));
1676                                 return True;
1677                         }
1678                 } else {
1679                         if((case_sensitive && (strcmp(last_component, namelist->name) == 0))||
1680                                                 (!case_sensitive && (StrCaseCmp(last_component, namelist->name) == 0))) {
1681                                 DEBUG(8,("is_in_path: match succeeded\n"));
1682                                 return True;
1683                         }
1684                 }
1685         }
1686         DEBUG(8,("is_in_path: match not found\n"));
1687
1688         return False;
1689 }
1690
1691 /*******************************************************************
1692  Strip a '/' separated list into an array of
1693  name_compare_enties structures suitable for
1694  passing to is_in_path(). We do this for
1695  speed so we can pre-parse all the names in the list
1696  and don't do it for each call to is_in_path().
1697  namelist is modified here and is assumed to be
1698  a copy owned by the caller.
1699  We also check if the entry contains a wildcard to
1700  remove a potentially expensive call to mask_match
1701  if possible.
1702 ********************************************************************/
1703
1704 void set_namearray(name_compare_entry **ppname_array, char *namelist)
1705 {
1706         char *name_end;
1707         char *nameptr = namelist;
1708         int num_entries = 0;
1709         int i;
1710
1711         (*ppname_array) = NULL;
1712
1713         if((nameptr == NULL ) || ((nameptr != NULL) && (*nameptr == '\0')))
1714                 return;
1715
1716         /* We need to make two passes over the string. The
1717                 first to count the number of elements, the second
1718                 to split it.
1719         */
1720
1721         while(*nameptr) {
1722                 if ( *nameptr == '/' ) {
1723                         /* cope with multiple (useless) /s) */
1724                         nameptr++;
1725                         continue;
1726                 }
1727                 /* find the next / */
1728                 name_end = strchr_m(nameptr, '/');
1729
1730                 /* oops - the last check for a / didn't find one. */
1731                 if (name_end == NULL)
1732                         break;
1733
1734                 /* next segment please */
1735                 nameptr = name_end + 1;
1736                 num_entries++;
1737         }
1738
1739         if(num_entries == 0)
1740                 return;
1741
1742         if(( (*ppname_array) = SMB_MALLOC_ARRAY(name_compare_entry, num_entries + 1)) == NULL) {
1743                 DEBUG(0,("set_namearray: malloc fail\n"));
1744                 return;
1745         }
1746
1747         /* Now copy out the names */
1748         nameptr = namelist;
1749         i = 0;
1750         while(*nameptr) {
1751                 if ( *nameptr == '/' ) {
1752                         /* cope with multiple (useless) /s) */
1753                         nameptr++;
1754                         continue;
1755                 }
1756                 /* find the next / */
1757                 if ((name_end = strchr_m(nameptr, '/')) != NULL)
1758                         *name_end = 0;
1759
1760                 /* oops - the last check for a / didn't find one. */
1761                 if(name_end == NULL)
1762                         break;
1763
1764                 (*ppname_array)[i].is_wild = ms_has_wild(nameptr);
1765                 if(((*ppname_array)[i].name = SMB_STRDUP(nameptr)) == NULL) {
1766                         DEBUG(0,("set_namearray: malloc fail (1)\n"));
1767                         return;
1768                 }
1769
1770                 /* next segment please */
1771                 nameptr = name_end + 1;
1772                 i++;
1773         }
1774
1775         (*ppname_array)[i].name = NULL;
1776
1777         return;
1778 }
1779
1780 /****************************************************************************
1781  Routine to free a namearray.
1782 ****************************************************************************/
1783
1784 void free_namearray(name_compare_entry *name_array)
1785 {
1786         int i;
1787
1788         if(name_array == NULL)
1789                 return;
1790
1791         for(i=0; name_array[i].name!=NULL; i++)
1792                 SAFE_FREE(name_array[i].name);
1793         SAFE_FREE(name_array);
1794 }
1795
1796 #undef DBGC_CLASS
1797 #define DBGC_CLASS DBGC_LOCKING
1798
1799 /****************************************************************************
1800  Simple routine to do POSIX file locking. Cruft in NFS and 64->32 bit mapping
1801  is dealt with in posix.c
1802 ****************************************************************************/
1803
1804 BOOL fcntl_lock(int fd, int op, SMB_OFF_T offset, SMB_OFF_T count, int type)
1805 {
1806         SMB_STRUCT_FLOCK lock;
1807         int ret;
1808
1809         DEBUG(8,("fcntl_lock %d %d %.0f %.0f %d\n",fd,op,(double)offset,(double)count,type));
1810
1811         lock.l_type = type;
1812         lock.l_whence = SEEK_SET;
1813         lock.l_start = offset;
1814         lock.l_len = count;
1815         lock.l_pid = 0;
1816
1817         ret = sys_fcntl_ptr(fd,op,&lock);
1818
1819         if (ret == -1 && errno != 0)
1820                 DEBUG(3,("fcntl_lock: fcntl lock gave errno %d (%s)\n",errno,strerror(errno)));
1821
1822         /* a lock query */
1823         if (op == SMB_F_GETLK) {
1824                 if ((ret != -1) &&
1825                                 (lock.l_type != F_UNLCK) &&
1826                                 (lock.l_pid != 0) &&
1827                                 (lock.l_pid != sys_getpid())) {
1828                         DEBUG(3,("fcntl_lock: fd %d is locked by pid %d\n",fd,(int)lock.l_pid));
1829                         return(True);
1830                 }
1831
1832                 /* it must be not locked or locked by me */
1833                 return(False);
1834         }
1835
1836         /* a lock set or unset */
1837         if (ret == -1) {
1838                 DEBUG(3,("fcntl_lock: lock failed at offset %.0f count %.0f op %d type %d (%s)\n",
1839                         (double)offset,(double)count,op,type,strerror(errno)));
1840                 return(False);
1841         }
1842
1843         /* everything went OK */
1844         DEBUG(8,("fcntl_lock: Lock call successful\n"));
1845
1846         return(True);
1847 }
1848
1849 #undef DBGC_CLASS
1850 #define DBGC_CLASS DBGC_ALL
1851
1852 /*******************************************************************
1853  Is the name specified one of my netbios names.
1854  Returns true if it is equal, false otherwise.
1855 ********************************************************************/
1856
1857 BOOL is_myname(const char *s)
1858 {
1859         int n;
1860         BOOL ret = False;
1861
1862         for (n=0; my_netbios_names(n); n++) {
1863                 if (strequal(my_netbios_names(n), s)) {
1864                         ret=True;
1865                         break;
1866                 }
1867         }
1868         DEBUG(8, ("is_myname(\"%s\") returns %d\n", s, ret));
1869         return(ret);
1870 }
1871
1872 BOOL is_myname_or_ipaddr(const char *s)
1873 {
1874         fstring name, dnsname;
1875         char *servername;
1876
1877         if ( !s )
1878                 return False;
1879
1880         /* santize the string from '\\name' */
1881
1882         fstrcpy( name, s );
1883
1884         servername = strrchr_m( name, '\\' );
1885         if ( !servername )
1886                 servername = name;
1887         else
1888                 servername++;
1889
1890         /* optimize for the common case */
1891
1892         if (strequal(servername, global_myname()))
1893                 return True;
1894
1895         /* check for an alias */
1896
1897         if (is_myname(servername))
1898                 return True;
1899
1900         /* check for loopback */
1901
1902         if (strequal(servername, "localhost"))
1903                 return True;
1904
1905         /* maybe it's my dns name */
1906
1907         if ( get_mydnsfullname( dnsname ) )
1908                 if ( strequal( servername, dnsname ) )
1909                         return True;
1910
1911         /* handle possible CNAME records */
1912
1913         if ( !is_ipaddress( servername ) ) {
1914                 /* use DNS to resolve the name, but only the first address */
1915                 struct hostent *hp;
1916
1917                 if (((hp = sys_gethostbyname(name)) != NULL) && (hp->h_addr != NULL)) {
1918                         struct in_addr return_ip;
1919                         putip( (char*)&return_ip, (char*)hp->h_addr );
1920                         fstrcpy( name, inet_ntoa( return_ip ) );
1921                         servername = name;
1922                 }
1923         }
1924
1925         /* maybe its an IP address? */
1926         if (is_ipaddress(servername)) {
1927                 struct iface_struct nics[MAX_INTERFACES];
1928                 int i, n;
1929                 uint32 ip;
1930
1931                 ip = interpret_addr(servername);
1932                 if ((ip==0) || (ip==0xffffffff))
1933                         return False;
1934
1935                 n = get_interfaces(nics, MAX_INTERFACES);
1936                 for (i=0; i<n; i++) {
1937                         if (ip == nics[i].ip.s_addr)
1938                                 return True;
1939                 }
1940         }
1941
1942         /* no match */
1943         return False;
1944 }
1945
1946 /*******************************************************************
1947  Is the name specified our workgroup/domain.
1948  Returns true if it is equal, false otherwise.
1949 ********************************************************************/
1950
1951 BOOL is_myworkgroup(const char *s)
1952 {
1953         BOOL ret = False;
1954
1955         if (strequal(s, lp_workgroup())) {
1956                 ret=True;
1957         }
1958
1959         DEBUG(8, ("is_myworkgroup(\"%s\") returns %d\n", s, ret));
1960         return(ret);
1961 }
1962
1963 /*******************************************************************
1964  we distinguish between 2K and XP by the "Native Lan Manager" string
1965    WinXP => "Windows 2002 5.1"
1966    Win2k => "Windows 2000 5.0"
1967    NT4   => "Windows NT 4.0"
1968    Win9x => "Windows 4.0"
1969  Windows 2003 doesn't set the native lan manager string but
1970  they do set the domain to "Windows 2003 5.2" (probably a bug).
1971 ********************************************************************/
1972
1973 void ra_lanman_string( const char *native_lanman )
1974 {
1975         if ( strcmp( native_lanman, "Windows 2002 5.1" ) == 0 )
1976                 set_remote_arch( RA_WINXP );
1977         else if ( strcmp( native_lanman, "Windows Server 2003 5.2" ) == 0 )
1978                 set_remote_arch( RA_WIN2K3 );
1979 }
1980
1981 /*******************************************************************
1982  Set the horrid remote_arch string based on an enum.
1983 ********************************************************************/
1984
1985 void set_remote_arch(enum remote_arch_types type)
1986 {
1987         ra_type = type;
1988         switch( type ) {
1989         case RA_WFWG:
1990                 fstrcpy(remote_arch, "WfWg");
1991                 break;
1992         case RA_OS2:
1993                 fstrcpy(remote_arch, "OS2");
1994                 break;
1995         case RA_WIN95:
1996                 fstrcpy(remote_arch, "Win95");
1997                 break;
1998         case RA_WINNT:
1999                 fstrcpy(remote_arch, "WinNT");
2000                 break;
2001         case RA_WIN2K:
2002                 fstrcpy(remote_arch, "Win2K");
2003                 break;
2004         case RA_WINXP:
2005                 fstrcpy(remote_arch, "WinXP");
2006                 break;
2007         case RA_WIN2K3:
2008                 fstrcpy(remote_arch, "Win2K3");
2009                 break;
2010         case RA_SAMBA:
2011                 fstrcpy(remote_arch,"Samba");
2012                 break;
2013         case RA_CIFSFS:
2014                 fstrcpy(remote_arch,"CIFSFS");
2015                 break;
2016         default:
2017                 ra_type = RA_UNKNOWN;
2018                 fstrcpy(remote_arch, "UNKNOWN");
2019                 break;
2020         }
2021
2022         DEBUG(10,("set_remote_arch: Client arch is \'%s\'\n", remote_arch));
2023 }
2024
2025 /*******************************************************************
2026  Get the remote_arch type.
2027 ********************************************************************/
2028
2029 enum remote_arch_types get_remote_arch(void)
2030 {
2031         return ra_type;
2032 }
2033
2034 void print_asc(int level, const unsigned char *buf,int len)
2035 {
2036         int i;
2037         for (i=0;i<len;i++)
2038                 DEBUG(level,("%c", isprint(buf[i])?buf[i]:'.'));
2039 }
2040
2041 void dump_data(int level, const char *buf1,int len)
2042 {
2043         const unsigned char *buf = (const unsigned char *)buf1;
2044         int i=0;
2045         if (len<=0) return;
2046
2047         if (!DEBUGLVL(level)) return;
2048
2049         DEBUGADD(level,("[%03X] ",i));
2050         for (i=0;i<len;) {
2051                 DEBUGADD(level,("%02X ",(int)buf[i]));
2052                 i++;
2053                 if (i%8 == 0) DEBUGADD(level,(" "));
2054                 if (i%16 == 0) {
2055                         print_asc(level,&buf[i-16],8); DEBUGADD(level,(" "));
2056                         print_asc(level,&buf[i-8],8); DEBUGADD(level,("\n"));
2057                         if (i<len) DEBUGADD(level,("[%03X] ",i));
2058                 }
2059         }
2060         if (i%16) {
2061                 int n;
2062                 n = 16 - (i%16);
2063                 DEBUGADD(level,(" "));
2064                 if (n>8) DEBUGADD(level,(" "));
2065                 while (n--) DEBUGADD(level,("   "));
2066                 n = MIN(8,i%16);
2067                 print_asc(level,&buf[i-(i%16)],n); DEBUGADD(level,( " " ));
2068                 n = (i%16) - n;
2069                 if (n>0) print_asc(level,&buf[i-n],n);
2070                 DEBUGADD(level,("\n"));
2071         }
2072 }
2073
2074 void dump_data_pw(const char *msg, const uchar * data, size_t len)
2075 {
2076 #ifdef DEBUG_PASSWORD
2077         DEBUG(11, ("%s", msg));
2078         if (data != NULL && len > 0)
2079         {
2080                 dump_data(11, data, len);
2081         }
2082 #endif
2083 }
2084
2085 char *tab_depth(int depth)
2086 {
2087         static pstring spaces;
2088         memset(spaces, ' ', depth * 4);
2089         spaces[depth * 4] = 0;
2090         return spaces;
2091 }
2092
2093 /*****************************************************************************
2094  Provide a checksum on a string
2095
2096  Input:  s - the null-terminated character string for which the checksum
2097              will be calculated.
2098
2099   Output: The checksum value calculated for s.
2100 *****************************************************************************/
2101
2102 int str_checksum(const char *s)
2103 {
2104         int res = 0;
2105         int c;
2106         int i=0;
2107
2108         while(*s) {
2109                 c = *s;
2110                 res ^= (c << (i % 15)) ^ (c >> (15-(i%15)));
2111                 s++;
2112                 i++;
2113         }
2114         return(res);
2115 }
2116
2117 /*****************************************************************
2118  Zero a memory area then free it. Used to catch bugs faster.
2119 *****************************************************************/
2120
2121 void zero_free(void *p, size_t size)
2122 {
2123         memset(p, 0, size);
2124         SAFE_FREE(p);
2125 }
2126
2127 /*****************************************************************
2128  Set our open file limit to a requested max and return the limit.
2129 *****************************************************************/
2130
2131 int set_maxfiles(int requested_max)
2132 {
2133 #if (defined(HAVE_GETRLIMIT) && defined(RLIMIT_NOFILE))
2134         struct rlimit rlp;
2135         int saved_current_limit;
2136
2137         if(getrlimit(RLIMIT_NOFILE, &rlp)) {
2138                 DEBUG(0,("set_maxfiles: getrlimit (1) for RLIMIT_NOFILE failed with error %s\n",
2139                         strerror(errno) ));
2140                 /* just guess... */
2141                 return requested_max;
2142         }
2143
2144         /*
2145          * Set the fd limit to be real_max_open_files + MAX_OPEN_FUDGEFACTOR to
2146          * account for the extra fd we need
2147          * as well as the log files and standard
2148          * handles etc. Save the limit we want to set in case
2149          * we are running on an OS that doesn't support this limit (AIX)
2150          * which always returns RLIM_INFINITY for rlp.rlim_max.
2151          */
2152
2153         /* Try raising the hard (max) limit to the requested amount. */
2154
2155 #if defined(RLIM_INFINITY)
2156         if (rlp.rlim_max != RLIM_INFINITY) {
2157                 int orig_max = rlp.rlim_max;
2158
2159                 if ( rlp.rlim_max < requested_max )
2160                         rlp.rlim_max = requested_max;
2161
2162                 /* This failing is not an error - many systems (Linux) don't
2163                         support our default request of 10,000 open files. JRA. */
2164
2165                 if(setrlimit(RLIMIT_NOFILE, &rlp)) {
2166                         DEBUG(3,("set_maxfiles: setrlimit for RLIMIT_NOFILE for %d max files failed with error %s\n",
2167                                 (int)rlp.rlim_max, strerror(errno) ));
2168
2169                         /* Set failed - restore original value from get. */
2170                         rlp.rlim_max = orig_max;
2171                 }
2172         }
2173 #endif
2174
2175         /* Now try setting the soft (current) limit. */
2176
2177         saved_current_limit = rlp.rlim_cur = MIN(requested_max,rlp.rlim_max);
2178
2179         if(setrlimit(RLIMIT_NOFILE, &rlp)) {
2180                 DEBUG(0,("set_maxfiles: setrlimit for RLIMIT_NOFILE for %d files failed with error %s\n",
2181                         (int)rlp.rlim_cur, strerror(errno) ));
2182                 /* just guess... */
2183                 return saved_current_limit;
2184         }
2185
2186         if(getrlimit(RLIMIT_NOFILE, &rlp)) {
2187                 DEBUG(0,("set_maxfiles: getrlimit (2) for RLIMIT_NOFILE failed with error %s\n",
2188                         strerror(errno) ));
2189                 /* just guess... */
2190                 return saved_current_limit;
2191     }
2192
2193 #if defined(RLIM_INFINITY)
2194         if(rlp.rlim_cur == RLIM_INFINITY)
2195                 return saved_current_limit;
2196 #endif
2197
2198         if((int)rlp.rlim_cur > saved_current_limit)
2199                 return saved_current_limit;
2200
2201         return rlp.rlim_cur;
2202 #else /* !defined(HAVE_GETRLIMIT) || !defined(RLIMIT_NOFILE) */
2203         /*
2204          * No way to know - just guess...
2205          */
2206         return requested_max;
2207 #endif
2208 }
2209
2210 /*****************************************************************
2211  Splits out the start of the key (HKLM or HKU) and the rest of the key.
2212 *****************************************************************/
2213
2214 BOOL reg_split_key(const char *full_keyname, uint32 *reg_type, char *key_name)
2215 {
2216         pstring tmp;
2217
2218         if (!next_token(&full_keyname, tmp, "\\", sizeof(tmp)))
2219                 return False;
2220
2221         (*reg_type) = 0;
2222
2223         DEBUG(10, ("reg_split_key: hive %s\n", tmp));
2224
2225         if (strequal(tmp, "HKLM") || strequal(tmp, "HKEY_LOCAL_MACHINE"))
2226                 (*reg_type) = HKEY_LOCAL_MACHINE;
2227         else if (strequal(tmp, "HKCR") || strequal(tmp, "HKEY_CLASSES_ROOT"))
2228                 (*reg_type) = HKEY_CLASSES_ROOT;
2229         else if (strequal(tmp, "HKU") || strequal(tmp, "HKEY_USERS"))
2230                 (*reg_type) = HKEY_USERS;
2231         else if (strequal(tmp, "HKPD")||strequal(tmp, "HKEY_PERFORMANCE_DATA"))
2232                 (*reg_type) = HKEY_PERFORMANCE_DATA;
2233         else {
2234                 DEBUG(10,("reg_split_key: unrecognised hive key %s\n", tmp));
2235                 return False;
2236         }
2237
2238         if (next_token(&full_keyname, tmp, "\n\r", sizeof(tmp)))
2239                 fstrcpy(key_name, tmp);
2240         else
2241                 key_name[0] = 0;
2242
2243         DEBUG(10, ("reg_split_key: name %s\n", key_name));
2244
2245         return True;
2246 }
2247
2248 /*****************************************************************
2249  Possibly replace mkstemp if it is broken.
2250 *****************************************************************/
2251
2252 int smb_mkstemp(char *template)
2253 {
2254 #if HAVE_SECURE_MKSTEMP
2255         return mkstemp(template);
2256 #else
2257         /* have a reasonable go at emulating it. Hope that
2258            the system mktemp() isn't completly hopeless */
2259         char *p = mktemp(template);
2260         if (!p)
2261                 return -1;
2262         return open(p, O_CREAT|O_EXCL|O_RDWR, 0600);
2263 #endif
2264 }
2265
2266 /*****************************************************************
2267  malloc that aborts with smb_panic on fail or zero size.
2268  *****************************************************************/
2269
2270 void *smb_xmalloc_array(size_t size, unsigned int count)
2271 {
2272         void *p;
2273         if (size == 0)
2274                 smb_panic("smb_xmalloc_array: called with zero size.\n");
2275         if (count >= MAX_ALLOC_SIZE/size) {
2276                 smb_panic("smb_xmalloc: alloc size too large.\n");
2277         }
2278         if ((p = SMB_MALLOC(size*count)) == NULL) {
2279                 DEBUG(0, ("smb_xmalloc_array failed to allocate %lu * %lu bytes\n",
2280                         (unsigned long)size, (unsigned long)count));
2281                 smb_panic("smb_xmalloc_array: malloc fail.\n");
2282         }
2283         return p;
2284 }
2285
2286 /**
2287  Memdup with smb_panic on fail.
2288 **/
2289
2290 void *smb_xmemdup(const void *p, size_t size)
2291 {
2292         void *p2;
2293         p2 = SMB_XMALLOC_ARRAY(unsigned char,size);
2294         memcpy(p2, p, size);
2295         return p2;
2296 }
2297
2298 /**
2299  strdup that aborts on malloc fail.
2300 **/
2301
2302 char *smb_xstrdup(const char *s)
2303 {
2304 #if defined(PARANOID_MALLOC_CHECKER)
2305 #ifdef strdup
2306 #undef strdup
2307 #endif
2308 #endif
2309         char *s1 = strdup(s);
2310 #if defined(PARANOID_MALLOC_CHECKER)
2311 #define strdup(s) __ERROR_DONT_USE_STRDUP_DIRECTLY
2312 #endif
2313         if (!s1)
2314                 smb_panic("smb_xstrdup: malloc fail\n");
2315         return s1;
2316
2317 }
2318
2319 /**
2320  strndup that aborts on malloc fail.
2321 **/
2322
2323 char *smb_xstrndup(const char *s, size_t n)
2324 {
2325 #if defined(PARANOID_MALLOC_CHECKER)
2326 #ifdef strndup
2327 #undef strndup
2328 #endif
2329 #endif
2330         char *s1 = strndup(s, n);
2331 #if defined(PARANOID_MALLOC_CHECKER)
2332 #define strndup(s,n) __ERROR_DONT_USE_STRNDUP_DIRECTLY
2333 #endif
2334         if (!s1)
2335                 smb_panic("smb_xstrndup: malloc fail\n");
2336         return s1;
2337 }
2338
2339 /*
2340   vasprintf that aborts on malloc fail
2341 */
2342
2343  int smb_xvasprintf(char **ptr, const char *format, va_list ap)
2344 {
2345         int n;
2346         va_list ap2;
2347
2348         VA_COPY(ap2, ap);
2349
2350         n = vasprintf(ptr, format, ap2);
2351         if (n == -1 || ! *ptr)
2352                 smb_panic("smb_xvasprintf: out of memory");
2353         return n;
2354 }
2355
2356 /*****************************************************************
2357  Like strdup but for memory.
2358 *****************************************************************/
2359
2360 void *memdup(const void *p, size_t size)
2361 {
2362         void *p2;
2363         if (size == 0)
2364                 return NULL;
2365         p2 = SMB_MALLOC(size);
2366         if (!p2)
2367                 return NULL;
2368         memcpy(p2, p, size);
2369         return p2;
2370 }
2371
2372 /*****************************************************************
2373  Get local hostname and cache result.
2374 *****************************************************************/
2375
2376 char *myhostname(void)
2377 {
2378         static pstring ret;
2379         if (ret[0] == 0)
2380                 get_myname(ret);
2381         return ret;
2382 }
2383
2384 /*****************************************************************
2385  A useful function for returning a path in the Samba lock directory.
2386 *****************************************************************/
2387
2388 char *lock_path(const char *name)
2389 {
2390         static pstring fname;
2391
2392         pstrcpy(fname,lp_lockdir());
2393         trim_char(fname,'\0','/');
2394
2395         if (!directory_exist(fname,NULL))
2396                 mkdir(fname,0755);
2397
2398         pstrcat(fname,"/");
2399         pstrcat(fname,name);
2400
2401         return fname;
2402 }
2403
2404 /*****************************************************************
2405  A useful function for returning a path in the Samba pid directory.
2406 *****************************************************************/
2407
2408 char *pid_path(const char *name)
2409 {
2410         static pstring fname;
2411
2412         pstrcpy(fname,lp_piddir());
2413         trim_char(fname,'\0','/');
2414
2415         if (!directory_exist(fname,NULL))
2416                 mkdir(fname,0755);
2417
2418         pstrcat(fname,"/");
2419         pstrcat(fname,name);
2420
2421         return fname;
2422 }
2423
2424 /**
2425  * @brief Returns an absolute path to a file in the Samba lib directory.
2426  *
2427  * @param name File to find, relative to LIBDIR.
2428  *
2429  * @retval Pointer to a static #pstring containing the full path.
2430  **/
2431
2432 char *lib_path(const char *name)
2433 {
2434         static pstring fname;
2435         fstr_sprintf(fname, "%s/%s", dyn_LIBDIR, name);
2436         return fname;
2437 }
2438
2439 /**
2440  * @brief Returns the platform specific shared library extension.
2441  *
2442  * @retval Pointer to a static #fstring containing the extension.
2443  **/
2444
2445 const char *shlib_ext(void)
2446 {
2447   return dyn_SHLIBEXT;
2448 }
2449
2450 /*******************************************************************
2451  Given a filename - get its directory name
2452  NB: Returned in static storage.  Caveats:
2453  o  Not safe in thread environment.
2454  o  Caller must not free.
2455  o  If caller wishes to preserve, they should copy.
2456 ********************************************************************/
2457
2458 char *parent_dirname(const char *path)
2459 {
2460         static pstring dirpath;
2461         char *p;
2462
2463         if (!path)
2464                 return(NULL);
2465
2466         pstrcpy(dirpath, path);
2467         p = strrchr_m(dirpath, '/');  /* Find final '/', if any */
2468         if (!p) {
2469                 pstrcpy(dirpath, ".");    /* No final "/", so dir is "." */
2470         } else {
2471                 if (p == dirpath)
2472                         ++p;    /* For root "/", leave "/" in place */
2473                 *p = '\0';
2474         }
2475         return dirpath;
2476 }
2477
2478
2479 /*******************************************************************
2480  Determine if a pattern contains any Microsoft wildcard characters.
2481 *******************************************************************/
2482
2483 BOOL ms_has_wild(const char *s)
2484 {
2485         char c;
2486         while ((c = *s++)) {
2487                 switch (c) {
2488                 case '*':
2489                 case '?':
2490                 case '<':
2491                 case '>':
2492                 case '"':
2493                         return True;
2494                 }
2495         }
2496         return False;
2497 }
2498
2499 BOOL ms_has_wild_w(const smb_ucs2_t *s)
2500 {
2501         smb_ucs2_t c;
2502         if (!s) return False;
2503         while ((c = *s++)) {
2504                 switch (c) {
2505                 case UCS2_CHAR('*'):
2506                 case UCS2_CHAR('?'):
2507                 case UCS2_CHAR('<'):
2508                 case UCS2_CHAR('>'):
2509                 case UCS2_CHAR('"'):
2510                         return True;
2511                 }
2512         }
2513         return False;
2514 }
2515
2516 /*******************************************************************
2517  A wrapper that handles case sensitivity and the special handling
2518  of the ".." name.
2519 *******************************************************************/
2520
2521 BOOL mask_match(const char *string, char *pattern, BOOL is_case_sensitive)
2522 {
2523         if (strcmp(string,"..") == 0)
2524                 string = ".";
2525         if (strcmp(pattern,".") == 0)
2526                 return False;
2527
2528         return ms_fnmatch(pattern, string, Protocol <= PROTOCOL_LANMAN2, is_case_sensitive) == 0;
2529 }
2530
2531 /*******************************************************************
2532  A wrapper that handles case sensitivity and the special handling
2533  of the ".." name. Varient that is only called by old search code which requires
2534  pattern translation.
2535 *******************************************************************/
2536
2537 BOOL mask_match_search(const char *string, char *pattern, BOOL is_case_sensitive)
2538 {
2539         if (strcmp(string,"..") == 0)
2540                 string = ".";
2541         if (strcmp(pattern,".") == 0)
2542                 return False;
2543
2544         return ms_fnmatch(pattern, string, True, is_case_sensitive) == 0;
2545 }
2546
2547 /*******************************************************************
2548  A wrapper that handles a list of patters and calls mask_match()
2549  on each.  Returns True if any of the patterns match.
2550 *******************************************************************/
2551
2552 BOOL mask_match_list(const char *string, char **list, int listLen, BOOL is_case_sensitive)
2553 {
2554        while (listLen-- > 0) {
2555                if (mask_match(string, *list++, is_case_sensitive))
2556                        return True;
2557        }
2558        return False;
2559 }
2560
2561 /*********************************************************
2562  Recursive routine that is called by unix_wild_match.
2563 *********************************************************/
2564
2565 static BOOL unix_do_match(const char *regexp, const char *str)
2566 {
2567         const char *p;
2568
2569         for( p = regexp; *p && *str; ) {
2570
2571                 switch(*p) {
2572                         case '?':
2573                                 str++;
2574                                 p++;
2575                                 break;
2576
2577                         case '*':
2578
2579                                 /*
2580                                  * Look for a character matching
2581                                  * the one after the '*'.
2582                                  */
2583                                 p++;
2584                                 if(!*p)
2585                                         return True; /* Automatic match */
2586                                 while(*str) {
2587
2588                                         while(*str && (*p != *str))
2589                                                 str++;
2590
2591                                         /*
2592                                          * Patch from weidel@multichart.de. In the case of the regexp
2593                                          * '*XX*' we want to ensure there are at least 2 'X' characters
2594                                          * in the string after the '*' for a match to be made.
2595                                          */
2596
2597                                         {
2598                                                 int matchcount=0;
2599
2600                                                 /*
2601                                                  * Eat all the characters that match, but count how many there were.
2602                                                  */
2603
2604                                                 while(*str && (*p == *str)) {
2605                                                         str++;
2606                                                         matchcount++;
2607                                                 }
2608
2609                                                 /*
2610                                                  * Now check that if the regexp had n identical characters that
2611                                                  * matchcount had at least that many matches.
2612                                                  */
2613
2614                                                 while ( *(p+1) && (*(p+1) == *p)) {
2615                                                         p++;
2616                                                         matchcount--;
2617                                                 }
2618
2619                                                 if ( matchcount <= 0 )
2620                                                         return False;
2621                                         }
2622
2623                                         str--; /* We've eaten the match char after the '*' */
2624
2625                                         if(unix_do_match(p, str))
2626                                                 return True;
2627
2628                                         if(!*str)
2629                                                 return False;
2630                                         else
2631                                                 str++;
2632                                 }
2633                                 return False;
2634
2635                         default:
2636                                 if(*str != *p)
2637                                         return False;
2638                                 str++;
2639                                 p++;
2640                                 break;
2641                 }
2642         }
2643
2644         if(!*p && !*str)
2645                 return True;
2646
2647         if (!*p && str[0] == '.' && str[1] == 0)
2648                 return(True);
2649
2650         if (!*str && *p == '?') {
2651                 while (*p == '?')
2652                         p++;
2653                 return(!*p);
2654         }
2655
2656         if(!*str && (*p == '*' && p[1] == '\0'))
2657                 return True;
2658
2659         return False;
2660 }
2661
2662 /*******************************************************************
2663  Simple case insensitive interface to a UNIX wildcard matcher.
2664 *******************************************************************/
2665
2666 BOOL unix_wild_match(const char *pattern, const char *string)
2667 {
2668         pstring p2, s2;
2669         char *p;
2670
2671         pstrcpy(p2, pattern);
2672         pstrcpy(s2, string);
2673         strlower_m(p2);
2674         strlower_m(s2);
2675
2676         /* Remove any *? and ** from the pattern as they are meaningless */
2677         for(p = p2; *p; p++)
2678                 while( *p == '*' && (p[1] == '?' ||p[1] == '*'))
2679                         pstrcpy( &p[1], &p[2]);
2680
2681         if (strequal(p2,"*"))
2682                 return True;
2683
2684         return unix_do_match(p2, s2) == 0;
2685 }
2686
2687 /**********************************************************************
2688  Converts a name to a fully qalified domain name.
2689 ***********************************************************************/
2690
2691 void name_to_fqdn(fstring fqdn, const char *name)
2692 {
2693         struct hostent *hp = sys_gethostbyname(name);
2694         if ( hp && hp->h_name && *hp->h_name ) {
2695                 DEBUG(10,("name_to_fqdn: lookup for %s -> %s.\n", name, hp->h_name));
2696                 fstrcpy(fqdn,hp->h_name);
2697         } else {
2698                 DEBUG(10,("name_to_fqdn: lookup for %s failed.\n", name));
2699                 fstrcpy(fqdn, name);
2700         }
2701 }
2702
2703 #ifdef __INSURE__
2704
2705 /*******************************************************************
2706 This routine is a trick to immediately catch errors when debugging
2707 with insure. A xterm with a gdb is popped up when insure catches
2708 a error. It is Linux specific.
2709 ********************************************************************/
2710
2711 int _Insure_trap_error(int a1, int a2, int a3, int a4, int a5, int a6)
2712 {
2713         static int (*fn)();
2714         int ret;
2715         char pidstr[10];
2716         /* you can get /usr/bin/backtrace from
2717            http://samba.org/ftp/unpacked/junkcode/backtrace */
2718         pstring cmd = "/usr/bin/backtrace %d";
2719
2720         slprintf(pidstr, sizeof(pidstr)-1, "%d", sys_getpid());
2721         pstring_sub(cmd, "%d", pidstr);
2722
2723         if (!fn) {
2724                 static void *h;
2725                 h = dlopen("/usr/local/parasoft/insure++lite/lib.linux2/libinsure.so", RTLD_LAZY);
2726                 fn = dlsym(h, "_Insure_trap_error");
2727
2728                 if (!h || h == _Insure_trap_error) {
2729                         h = dlopen("/usr/local/parasoft/lib.linux2/libinsure.so", RTLD_LAZY);
2730                         fn = dlsym(h, "_Insure_trap_error");
2731                 }
2732         }
2733
2734         ret = fn(a1, a2, a3, a4, a5, a6);
2735
2736         system(cmd);
2737
2738         return ret;
2739 }
2740 #endif