Fix major overstep with last commit

[vg.git] / src / vg / vg_io.h

diff --git a/src/vg/vg_io.h b/src/vg/vg_io.h
index 19d468b8e1ce39634cdbe31dcdac7e1447c6a363..95e16231d4a998dfbabf74a851f095e1ef1d6a28 100644 (file)
--- a/src/vg/vg_io.h
+++ b/src/vg/vg_io.h
@@ -25,17 +25,14 @@ static void *vg_disk_open_read( const char *path, int reserve_end, i64 *size )
        if( f )
        {
                i64 fsize = vg_file_size( f );
-               void *buf = malloc( fsize + reserve_end );
+               void *buf = vg_alloc( fsize + reserve_end );
                
-               if( buf )
-               {
-                       /* Invalid / corrupt read */
-                       if( fread( buf, 1, fsize, f ) != fsize )
-                       {
-                               free( buf );
-                               buf = NULL;
-                       }
-               }
+      /* Invalid / corrupt read */
+      if( fread( buf, 1, fsize, f ) != fsize )
+      {
+         vg_free( buf );
+         buf = NULL;
+      }
                
                *size = fsize;