]> git.ozlabs.org Git - ccan/commitdiff
Fix multiply by three error, stat to get initial size.
authorRusty Russell <rusty@rustcorp.com.au>
Tue, 11 Aug 2009 11:57:57 +0000 (21:27 +0930)
committerRusty Russell <rusty@rustcorp.com.au>
Tue, 11 Aug 2009 11:57:57 +0000 (21:27 +0930)
ccan/grab_file/grab_file.c

index f21e1e5e3ac5356172d7406ff0f3b6c9faff8f18..b3a2f54d512ea07f5c10b9e95c5cafbb59bd0771 100644 (file)
@@ -9,18 +9,33 @@
 void *grab_fd(const void *ctx, int fd, size_t *size)
 {
        int ret;
-       size_t max = 16384, s;
+       size_t max, s;
        char *buffer;
+       struct stat st;
 
        if (!size)
                size = &s;
        *size = 0;
 
+       if (fstat(fd, &st) == 0)
+               max = st.st_size;
+       else
+               max = 16384;
+
        buffer = talloc_array(ctx, char, max+1);
        while ((ret = read(fd, buffer + *size, max - *size)) > 0) {
                *size += ret;
-               if (*size == max)
-                       buffer = talloc_realloc(ctx, buffer, char, max*=2 + 1);
+               if (*size == max) {
+                       buffer = talloc_realloc(ctx, buffer, char, max*2+1);
+                       if (!buffer) {
+                               buffer = talloc_realloc(ctx, buffer, char,
+                                                       max + 1024*1024 + 1);
+                               if (!buffer)
+                                       return NULL;
+                               max += 1024*1024;
+                       } else
+                               max *= 2;
+               }
        }
        if (ret < 0) {
                talloc_free(buffer);