diff options
Diffstat (limited to 'src/usr/autoboot.c')
| -rw-r--r-- | src/usr/autoboot.c | 10 |
1 files changed, 9 insertions, 1 deletions
diff --git a/src/usr/autoboot.c b/src/usr/autoboot.c index 5d017229d..3936e4ca5 100644 --- a/src/usr/autoboot.c +++ b/src/usr/autoboot.c @@ -99,19 +99,27 @@ void netboot ( struct net_device *netdev ) { return; } printf ( "Booting \"%s\"\n", filename ); - if ( ( rc = imgfetch ( filename, NULL, &image ) ) != 0 ) { + image = alloc_image(); + if ( ! image ) { + printf ( "Out of memory\n" ); + return; + } + if ( ( rc = imgfetch ( image, filename, 0 ) ) != 0 ) { printf ( "Could not retrieve %s: %s\n", filename, strerror ( rc ) ); + image_put ( image ); return; } if ( ( rc = imgload ( image ) ) != 0 ) { printf ( "Could not load %s: %s\n", image->name, strerror ( rc ) ); + image_put ( image ); return; } if ( ( rc = imgexec ( image ) ) != 0 ) { printf ( "Could not execute %s: %s\n", image->name, strerror ( rc ) ); + image_put ( image ); return; } } |