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