lib: Add fgets_slash

[martins/samba.git] / lib / util / util_file.c

diff --git a/lib/util/util_file.c b/lib/util/util_file.c
index 4948afb16b62046a148a26f83edf8054c5a6e024..ac8206008a3ccf9d23ad8f14c8534cdb8b9d0ef9 100644 (file)
--- a/lib/util/util_file.c
+++ b/lib/util/util_file.c
@@ -79,6 +79,88 @@ _PUBLIC_ char *afdgets(int fd, TALLOC_CTX *mem_ctx, size_t hint)
        return data;
 }
 
+char *fgets_slash(TALLOC_CTX *mem_ctx, char *s2, int maxlen, FILE *f)
+{
+       char *s = s2;
+       int len = 0;
+       int c;
+       bool start_of_line = true;
+
+       if (feof(f)) {
+               return NULL;
+       }
+
+       if (maxlen < 2) {
+               return NULL;
+       }
+
+       if (s2 == NULL) {
+               maxlen = MIN(maxlen,8);
+               s = talloc_array(mem_ctx, char, maxlen);
+       }
+
+       if (s == NULL) {
+               return NULL;
+       }
+
+       *s = 0;
+
+       while (len < maxlen-1) {
+               c = 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 == NULL)) {
+                                   TALLOC_FREE(s);
+                           }
+                           return (len>0) ? s : NULL;
+                   case ' ':
+                           if (start_of_line) {
+                                   break;
+                           }
+                           /* fall through */
+                   default:
+                           start_of_line = false;
+                           s[len++] = c;
+                           s[len] = 0;
+               }
+               if ((s2 == NULL) && (len > maxlen-3)) {
+                       int m;
+                       char *t;
+
+                       m = maxlen * 2;
+                       if (m < maxlen) {
+                               DBG_ERR("length overflow");
+                               TALLOC_FREE(s);
+                               return NULL;
+                       }
+                       maxlen = m;
+
+                       t = talloc_realloc(mem_ctx, s, char, maxlen);
+                       if (t == NULL) {
+                               DBG_ERR("failed to expand buffer!\n");
+                               TALLOC_FREE(s);
+                               return NULL;
+                       }
+
+                       s = t;
+               }
+       }
+
+       return s;
+}
 
 /**
 load a file into memory from a fd.