tools/vm/page_owner: increase temporary buffer size

[mirror_ubuntu-zesty-kernel.git] / tools / vm / page_owner_sort.c

diff --git a/tools/vm/page_owner_sort.c b/tools/vm/page_owner_sort.c
index 77147b42d598be985abd087e5eb8ea5ea95875fc..f1c055f3c2436564302adebf429f239557d14a3c 100644 (file)
--- a/tools/vm/page_owner_sort.c
+++ b/tools/vm/page_owner_sort.c
@@ -79,12 +79,12 @@ static void add_list(char *buf, int len)
        }
 }
 
-#define BUF_SIZE       1024
+#define BUF_SIZE       (128 * 1024)
 
 int main(int argc, char **argv)
 {
        FILE *fin, *fout;
-       char buf[BUF_SIZE];
+       char *buf;
        int ret, i, count;
        struct block_list *list2;
        struct stat st;
@@ -107,6 +107,11 @@ int main(int argc, char **argv)
        max_size = st.st_size / 100; /* hack ... */
 
        list = malloc(max_size * sizeof(*list));
+       buf = malloc(BUF_SIZE);
+       if (!list || !buf) {
+               printf("Out of memory\n");
+               exit(1);
+       }
 
        for ( ; ; ) {
                ret = read_block(buf, BUF_SIZE, fin);