RIP BOOL. Convert BOOL -> bool. I found a few interesting
[sfrench/samba-autobuild/.git] / source / lib / util_file.c
index 77c0d7888e89cc00905d541a8c210def2283d999..673a15df3806619f4ed90678e108fd1f1dde4d05 100644 (file)
@@ -1,10 +1,11 @@
 /*
- * Unix SMB/Netbios implementation. Version 1.9. SMB parameters and setup
+ * Unix SMB/CIFS implementation.
+ * SMB parameters and setup
  * Copyright (C) Andrew Tridgell 1992-1998 Modified by Jeremy Allison 1995.
  * 
  * This program is free software; you can redistribute it and/or modify it under
  * the terms of the GNU General Public License as published by the Free
- * Software Foundation; either version 2 of the License, or (at your option)
+ * Software Foundation; either version 3 of the License, or (at your option)
  * any later version.
  * 
  * This program is distributed in the hope that it will be useful, but WITHOUT
  * more details.
  * 
  * You should have received a copy of the GNU General Public License along with
- * this program; if not, write to the Free Software Foundation, Inc., 675
- * Mass Ave, Cambridge, MA 02139, USA.
+ * this program; if not, see <http://www.gnu.org/licenses/>.
  */
 
 #include "includes.h"
 
-static int gotalarm;
-
-/***************************************************************
- Signal function to tell us we timed out.
-****************************************************************/
-
-static void gotalarm_sig(void)
-{
-  gotalarm = 1;
-}
-
-/***************************************************************
- Lock or unlock a fd for a known lock type. Abandon after waitsecs 
- seconds.
-****************************************************************/
-
-BOOL do_file_lock(int fd, int waitsecs, int type)
-{
-  SMB_STRUCT_FLOCK lock;
-  int             ret;
-
-  gotalarm = 0;
-  CatchSignal(SIGALRM, SIGNAL_CAST gotalarm_sig);
-
-  lock.l_type = type;
-  lock.l_whence = SEEK_SET;
-  lock.l_start = 0;
-  lock.l_len = 1;
-  lock.l_pid = 0;
-
-  alarm(waitsecs);
-  ret = fcntl(fd, SMB_F_SETLKW, &lock);
-  alarm(0);
-  CatchSignal(SIGALRM, SIGNAL_CAST SIG_DFL);
-
-  if (gotalarm) {
-    DEBUG(0, ("do_file_lock: failed to %s file.\n",
-                type == F_UNLCK ? "unlock" : "lock"));
-    return False;
-  }
-
-  return (ret == 0);
-}
-
-
-/***************************************************************
- Lock an fd. Abandon after waitsecs seconds.
-****************************************************************/
-
-BOOL file_lock(int fd, int type, int secs, int *plock_depth)
-{
-  if (fd < 0)
-    return False;
-
-  (*plock_depth)++;
-
-  if ((*plock_depth) == 0)
-  {
-    if (!do_file_lock(fd, secs, type)) {
-      DEBUG(10,("file_lock: locking file failed, error = %s.\n",
-                 strerror(errno)));
-      return False;
-    }
-  }
-
-  return True;
-}
-
-/***************************************************************
- Unlock an fd. Abandon after waitsecs seconds.
-****************************************************************/
-
-BOOL file_unlock(int fd, int *plock_depth)
-{
-  BOOL ret=True;
-
-  if(*plock_depth == 1)
-    ret = do_file_lock(fd, 5, F_UNLCK);
-
-  (*plock_depth)--;
-
-  if(!ret)
-    DEBUG(10,("file_unlock: unlocking file failed, error = %s.\n",
-                 strerror(errno)));
-  return ret;
-}
-
-/***************************************************************
- locks a file for enumeration / modification.
- update to be set = True if modification is required.
-****************************************************************/
-
-void *startfilepwent(char *pfile, char *s_readbuf, int bufsize,
-                               int *file_lock_depth, BOOL update)
-{
-  FILE *fp = NULL;
-
-  if (!*pfile)
- {
-    DEBUG(0, ("startfilepwent: No file set\n"));
-    return (NULL);
-  }
-  DEBUG(10, ("startfilepwent: opening file %s\n", pfile));
-
-  fp = sys_fopen(pfile, update ? "r+b" : "rb");
-
-  if (fp == NULL) {
-    DEBUG(0, ("startfilepwent: unable to open file %s\n", pfile));
-    return NULL;
-  }
-
-  /* Set a buffer to do more efficient reads */
-  setvbuf(fp, s_readbuf, _IOFBF, bufsize);
-
-  if (!file_lock(fileno(fp), (update ? F_WRLCK : F_RDLCK), 5, file_lock_depth))
-  {
-    DEBUG(0, ("startfilepwent: unable to lock file %s\n", pfile));
-    fclose(fp);
-    return NULL;
-  }
-
-  /* Make sure it is only rw by the owner */
-  chmod(pfile, 0600);
+#ifndef MAP_FAILED
+#define MAP_FAILED ((void *)-1)
+#endif
 
-  /* We have a lock on the file. */
-  return (void *)fp;
-}
+/****************************************************************************
+ Read a line from a file with possible \ continuation chars. 
+ Blanks at the start or end of a line are stripped.
+ The string will be allocated if s2 is NULL.
+****************************************************************************/
 
-/***************************************************************
- End enumeration of the file.
-****************************************************************/
-void endfilepwent(void *vp, int *file_lock_depth)
+char *fgets_slash(char *s2,int maxlen,XFILE *f)
 {
-  FILE *fp = (FILE *)vp;
+       char *s=s2;
+       int len = 0;
+       int c;
+       bool start_of_line = True;
 
-  file_unlock(fileno(fp), file_lock_depth);
-  fclose(fp);
-  DEBUG(7, ("endfilepwent: closed file.\n"));
-}
-
-/*************************************************************************
- Return the current position in the file list as an SMB_BIG_UINT.
- This must be treated as an opaque token.
-*************************************************************************/
-SMB_BIG_UINT getfilepwpos(void *vp)
-{
-  return (SMB_BIG_UINT)sys_ftell((FILE *)vp);
-}
+       if (x_feof(f)) {
+               return(NULL);
+       }
 
-/*************************************************************************
- Set the current position in the file list from an SMB_BIG_UINT.
- This must be treated as an opaque token.
-*************************************************************************/
-BOOL setfilepwpos(void *vp, SMB_BIG_UINT tok)
-{
-  return !sys_fseek((FILE *)vp, (SMB_OFF_T)tok, SEEK_SET);
-}
+       if (maxlen <2) {
+               return(NULL);
+       }
 
-/*************************************************************************
- gets a line out of a file.
- line is of format "xxxx:xxxxxx:xxxxx:".
- lines with "#" at the front are ignored.
-*************************************************************************/
-int getfileline(void *vp, char *linebuf, int linebuf_size)
-{
-       /* Static buffers we will return. */
-       FILE *fp = (FILE *)vp;
-       unsigned char   c;
-       unsigned char  *p;
-       size_t            linebuf_len;
-
-       if (fp == NULL)
-       {
-               DEBUG(0,("getfileline: Bad file pointer.\n"));
-               return -1;
-       }
-
-       /*
-        * Scan the file, a line at a time.
-        */
-       while (!feof(fp))
-       {
-               linebuf[0] = '\0';
-
-               fgets(linebuf, linebuf_size, fp);
-               if (ferror(fp))
-               {
-                       return -1;
-               }
+       if (!s2) {
+               maxlen = MIN(maxlen,8);
+               s = (char *)SMB_MALLOC(maxlen);
+       }
 
-               /*
-                * Check if the string is terminated with a newline - if not
-                * then we must keep reading and discard until we get one.
-                */
+       if (!s) {
+               return(NULL);
+       }
 
-               linebuf_len = strlen(linebuf);
-               if (linebuf_len == 0)
-               {
-                       linebuf[0] = '\0';
-                       return 0;
-               }
+       *s = 0;
 
-               if (linebuf[linebuf_len - 1] != '\n')
-               {
-                       c = '\0';
-                       while (!ferror(fp) && !feof(fp))
-                       {
-                               c = fgetc(fp);
-                               if (c == '\n')
-                               {
+       while (len < maxlen-1) {
+               c = x_getc(f);
+               switch (c) {
+                       case '\r':
+                               break;
+                       case '\n':
+                               while (len > 0 && s[len-1] == ' ') {
+                                       s[--len] = 0;
+                               }
+                               if (len > 0 && s[len-1] == '\\') {
+                                       s[--len] = 0;
+                                       start_of_line = True;
                                        break;
                                }
-                       }
-               }
-               else
-               {
-                       linebuf[linebuf_len - 1] = '\0';
-               }
-
-#ifdef DEBUG_PASSWORD
-               DEBUG(100, ("getfileline: got line |%s|\n", linebuf));
-#endif
-               if ((linebuf[0] == 0) && feof(fp))
-               {
-                       DEBUG(4, ("getfileline: end of file reached\n"));
-                       return 0;
-               }
-
-               if (linebuf[0] == '#' || linebuf[0] == '\0')
-               {
-                       DEBUG(6, ("getfileline: skipping comment or blank line\n"));
-                       continue;
+                               return(s);
+                       case EOF:
+                               if (len <= 0 && !s2)  {
+                                       SAFE_FREE(s);
+                               }
+                               return(len>0?s:NULL);
+                       case ' ':
+                               if (start_of_line) {
+                                       break;
+                               }
+                       default:
+                               start_of_line = False;
+                               s[len++] = c;
+                               s[len] = 0;
                }
 
-               p = (unsigned char *) strchr_m(linebuf, ':');
-               if (p == NULL)
-               {
-                       DEBUG(0, ("getfileline: malformed line entry (no :)\n"));
-                       continue;
+               if (!s2 && len > maxlen-3) {
+                       maxlen *= 2;
+                       s = (char *)SMB_REALLOC(s,maxlen);
+                       if (!s) {
+                               DEBUG(0,("fgets_slash: failed to expand buffer!\n"));
+                               return(NULL);
+                       }
                }
-               return linebuf_len;
        }
-       return -1;
+       return(s);
 }
 
-
 /****************************************************************************
-read a line from a file with possible \ continuation chars. 
-Blanks at the start or end of a line are stripped.
-The string will be allocated if s2 is NULL
+ Load from a pipe into memory.
 ****************************************************************************/
-char *fgets_slash(char *s2,int maxlen,XFILE *f)
-{
-  char *s=s2;
-  int len = 0;
-  int c;
-  BOOL start_of_line = True;
-
-  if (x_feof(f))
-    return(NULL);
-
-  if (maxlen <2) return(NULL);
-
-  if (!s2)
-    {
-      maxlen = MIN(maxlen,8);
-      s = (char *)malloc(maxlen);
-    }
-
-  if (!s) return(NULL);
-
-  *s = 0;
-
-  while (len < maxlen-1)
-    {
-      c = x_getc(f);
-      switch (c)
-       {
-       case '\r':
-         break;
-       case '\n':
-         while (len > 0 && s[len-1] == ' ')
-           {
-             s[--len] = 0;
-           }
-         if (len > 0 && s[len-1] == '\\')
-           {
-             s[--len] = 0;
-             start_of_line = True;
-             break;
-           }
-         return(s);
-       case EOF:
-         if (len <= 0 && !s2) 
-           SAFE_FREE(s);
-         return(len>0?s:NULL);
-       case ' ':
-         if (start_of_line)
-           break;
-       default:
-         start_of_line = False;
-         s[len++] = c;
-         s[len] = 0;
-       }
-      if (!s2 && len > maxlen-3)
-       {
-         char *t;
-         
-         maxlen *= 2;
-         t = (char *)Realloc(s,maxlen);
-         if (!t) {
-           DEBUG(0,("fgets_slash: failed to expand buffer!\n"));
-           SAFE_FREE(s);
-           return(NULL);
-         } else s = t;
-       }
-    }
-  return(s);
-}
-
 
-/****************************************************************************
-load from a pipe into memory
-****************************************************************************/
-char *file_pload(char *syscmd, size_t *size)
+static char *file_pload(char *syscmd, size_t *size)
 {
        int fd, n;
-       char *p, *tp;
+       char *p;
        pstring buf;
        size_t total;
        
        fd = sys_popen(syscmd);
-       if (fd == -1) return NULL;
+       if (fd == -1) {
+               return NULL;
+       }
 
        p = NULL;
        total = 0;
 
        while ((n = read(fd, buf, sizeof(buf))) > 0) {
-               tp = Realloc(p, total + n + 1);
-               if (!tp) {
-                       DEBUG(0,("file_pload: failed to exand buffer!\n"));
+               p = (char *)SMB_REALLOC(p, total + n + 1);
+               if (!p) {
+                       DEBUG(0,("file_pload: failed to expand buffer!\n"));
                        close(fd);
-                       SAFE_FREE(p);
                        return NULL;
-               } else p = tp;
+               }
                memcpy(p+total, buf, n);
                total += n;
        }
-       if (p) p[total] = 0;
 
+       if (p) {
+               p[total] = 0;
+       }
+
+       /* FIXME: Perhaps ought to check that the command completed
+        * successfully (returned 0); if not the data may be
+        * truncated. */
        sys_pclose(fd);
 
-       if (size) *size = total;
+       if (size) {
+               *size = total;
+       }
 
        return p;
 }
 
 /****************************************************************************
-load a file into memory from a fd.
+ Load a file into memory from a fd.
+ Truncate at maxsize. If maxsize == 0 - no limit.
 ****************************************************************************/ 
 
-char *fd_load(int fd, size_t *size)
+char *fd_load(int fd, size_t *psize, size_t maxsize)
 {
        SMB_STRUCT_STAT sbuf;
+       size_t size;
        char *p;
 
-       if (sys_fstat(fd, &sbuf) != 0) return NULL;
+       if (sys_fstat(fd, &sbuf) != 0) {
+               return NULL;
+       }
 
-       p = (char *)malloc(sbuf.st_size+1);
-       if (!p) return NULL;
+       size = sbuf.st_size;
+       if (maxsize) {
+               size = MIN(size, maxsize);
+       }
 
-       if (read(fd, p, sbuf.st_size) != sbuf.st_size) {
+       p = (char *)SMB_MALLOC(size+1);
+       if (!p) {
+               return NULL;
+       }
+
+       if (read(fd, p, size) != size) {
                SAFE_FREE(p);
                return NULL;
        }
-       p[sbuf.st_size] = 0;
+       p[size] = 0;
 
-       if (size) *size = sbuf.st_size;
+       if (psize) {
+               *psize = size;
+       }
 
        return p;
 }
 
 /****************************************************************************
-load a file into memory
+ Load a file into memory.
 ****************************************************************************/
-char *file_load(const char *fname, size_t *size)
+
+char *file_load(const char *fname, size_t *size, size_t maxsize)
 {
        int fd;
        char *p;
 
-       if (!fname || !*fname) return NULL;
+       if (!fname || !*fname) {
+               return NULL;
+       }
        
        fd = open(fname,O_RDONLY);
-       if (fd == -1) return NULL;
-
-       p = fd_load(fd, size);
+       if (fd == -1) {
+               return NULL;
+       }
 
+       p = fd_load(fd, size, maxsize);
        close(fd);
-
        return p;
 }
 
+/*******************************************************************
+ unmap or free memory
+*******************************************************************/
+
+bool unmap_file(void* start, size_t size)
+{
+#ifdef HAVE_MMAP
+       if ( munmap( start, size ) != 0 ) {
+               DEBUG( 1, ("map_file: Failed to unmap address %p "
+                       "of size %u - %s\n", 
+                       start, (unsigned int)size, strerror(errno) ));
+               return False;
+       }
+       return True;
+#else
+       SAFE_FREE( start );
+       return True;
+#endif
+}
 
 /*******************************************************************
-mmap (if possible) or read a file
+ mmap (if possible) or read a file.
 ********************************************************************/
+
 void *map_file(char *fname, size_t size)
 {
        size_t s2 = 0;
@@ -433,51 +236,54 @@ void *map_file(char *fname, size_t size)
        int fd;
        fd = open(fname, O_RDONLY, 0);
        if (fd == -1) {
-               DEBUG(1,("Failed to load %s - %s\n", fname, strerror(errno)));
+               DEBUG(2,("map_file: Failed to load %s - %s\n", fname, strerror(errno)));
                return NULL;
        }
        p = mmap(NULL, size, PROT_READ, MAP_SHARED|MAP_FILE, fd, 0);
        close(fd);
        if (p == MAP_FAILED) {
-               DEBUG(1,("Failed to mmap %s - %s\n", fname, strerror(errno)));
+               DEBUG(1,("map_file: Failed to mmap %s - %s\n", fname, strerror(errno)));
                return NULL;
        }
 #endif
        if (!p) {
-               p = file_load(fname, &s2);
-               if (!p || s2 != size) {
-                       DEBUG(1,("incorrect size for %s - got %d expected %d\n",
-                                fname, s2, size));
-                       if (p) free(p);
+               p = file_load(fname, &s2, 0);
+               if (!p) {
+                       return NULL;
+               }
+               if (s2 != size) {
+                       DEBUG(1,("map_file: incorrect size for %s - got %lu expected %lu\n",
+                                fname, (unsigned long)s2, (unsigned long)size));
+                       SAFE_FREE(p);   
                        return NULL;
                }
        }
-
        return p;
 }
 
-
 /****************************************************************************
-parse a buffer into lines
+ Parse a buffer into lines.
 ****************************************************************************/
+
 static char **file_lines_parse(char *p, size_t size, int *numlines)
 {
        int i;
        char *s, **ret;
 
-       if (!p) return NULL;
+       if (!p) {
+               return NULL;
+       }
 
        for (s = p, i=0; s < p+size; s++) {
                if (s[0] == '\n') i++;
        }
 
-       ret = (char **)malloc(sizeof(ret[0])*(i+2));
+       ret = SMB_MALLOC_ARRAY(char *, i+2);
        if (!ret) {
                SAFE_FREE(p);
                return NULL;
        }       
        memset(ret, 0, sizeof(ret[0])*(i+2));
-       if (numlines) *numlines = i;
 
        ret[0] = p;
        for (s = p, i=0; s < p+size; s++) {
@@ -486,75 +292,96 @@ static char **file_lines_parse(char *p, size_t size, int *numlines)
                        i++;
                        ret[i] = s+1;
                }
-               if (s[0] == '\r') s[0] = 0;
+               if (s[0] == '\r') {
+                       s[0] = 0;
+               }
+       }
+
+       /* remove any blank lines at the end */
+       while (i > 0 && ret[i-1][0] == 0) {
+               i--;
+       }
+
+       if (numlines) {
+               *numlines = i;
        }
 
        return ret;
 }
 
-
 /****************************************************************************
-load a file into memory and return an array of pointers to lines in the file
-must be freed with file_lines_free(). 
+ Load a file into memory and return an array of pointers to lines in the file
+ must be freed with file_lines_free(). 
 ****************************************************************************/
-char **file_lines_load(const char *fname, int *numlines)
+
+char **file_lines_load(const char *fname, int *numlines, size_t maxsize)
 {
        char *p;
-       size_t size;
+       size_t size = 0;
 
-       p = file_load(fname, &size);
-       if (!p) return NULL;
+       p = file_load(fname, &size, maxsize);
+       if (!p) {
+               return NULL;
+       }
 
        return file_lines_parse(p, size, numlines);
 }
 
 /****************************************************************************
-load a fd into memory and return an array of pointers to lines in the file
-must be freed with file_lines_free(). If convert is true calls unix_to_dos on
-the list.
+ Load a fd into memory and return an array of pointers to lines in the file
+ must be freed with file_lines_free(). If convert is true calls unix_to_dos on
+ the list.
 ****************************************************************************/
-char **fd_lines_load(int fd, int *numlines)
+
+char **fd_lines_load(int fd, int *numlines, size_t maxsize)
 {
        char *p;
        size_t size;
 
-       p = fd_load(fd, &size);
-       if (!p) return NULL;
+       p = fd_load(fd, &size, maxsize);
+       if (!p) {
+               return NULL;
+       }
 
        return file_lines_parse(p, size, numlines);
 }
 
-
 /****************************************************************************
-load a pipe into memory and return an array of pointers to lines in the data
-must be freed with file_lines_free(). 
+ Load a pipe into memory and return an array of pointers to lines in the data
+ must be freed with file_lines_free(). 
 ****************************************************************************/
+
 char **file_lines_pload(char *syscmd, int *numlines)
 {
        char *p;
        size_t size;
 
        p = file_pload(syscmd, &size);
-       if (!p) return NULL;
+       if (!p) {
+               return NULL;
+       }
 
        return file_lines_parse(p, size, numlines);
 }
 
 /****************************************************************************
-free lines loaded with file_lines_load
+ Free lines loaded with file_lines_load.
 ****************************************************************************/
+
 void file_lines_free(char **lines)
 {
-       if (!lines) return;
+       if (!lines) {
+               return;
+       }
        SAFE_FREE(lines[0]);
        SAFE_FREE(lines);
 }
 
-
 /****************************************************************************
-take a lislist of lines and modify them to produce a list where \ continues
-a line
+ Take a list of lines and modify them to produce a list where \ continues
+ a line.
 ****************************************************************************/
+
 void file_lines_slashcont(char **lines)
 {
        int i, j;
@@ -565,11 +392,33 @@ void file_lines_slashcont(char **lines)
                        lines[i][len-1] = ' ';
                        if (lines[i+1]) {
                                char *p = &lines[i][len];
-                               while (p < lines[i+1]) *p++ = ' ';
-                               for (j = i+1; lines[j]; j++) lines[j] = lines[j+1];
+                               while (p < lines[i+1]) {
+                                       *p++ = ' ';
+                               }
+                               for (j = i+1; lines[j]; j++) {
+                                       lines[j] = lines[j+1];
+                               }
                        }
                } else {
                        i++;
                }
        }
 }
+
+/****************************************************************************
+ Save a lump of data into a file. Mostly used for debugging.
+****************************************************************************/
+
+bool file_save(const char *fname, void *packet, size_t length)
+{
+       int fd;
+       fd = open(fname, O_WRONLY|O_CREAT|O_TRUNC, 0644);
+       if (fd == -1) {
+               return False;
+       }
+       if (write(fd, packet, length) != (size_t)length) {
+               return False;
+       }
+       close(fd);
+       return True;
+}