mirror of
https://github.com/ipxe/ipxe
synced 2025-12-11 05:51:37 +03:00
e2c0055e23
took out the contract to eventually call efree() when it called fetch(). Maintain the most recently loaded image at the start of the list, so that imgautoselect() will pick it.
145 lines
3.0 KiB
C
145 lines
3.0 KiB
C
/*
|
|
* Copyright (C) 2007 Michael Brown <mbrown@fensystems.co.uk>.
|
|
*
|
|
* This program is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU General Public License as
|
|
* published by the Free Software Foundation; either version 2 of the
|
|
* License, or any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
|
|
*/
|
|
|
|
#include <stdint.h>
|
|
#include <stdlib.h>
|
|
#include <errno.h>
|
|
#include <vsprintf.h>
|
|
#include <gpxe/image.h>
|
|
#include <gpxe/emalloc.h>
|
|
#include <usr/fetch.h>
|
|
#include <usr/imgmgmt.h>
|
|
|
|
/** @file
|
|
*
|
|
* Image management
|
|
*
|
|
*/
|
|
|
|
/**
|
|
* Fetch an image
|
|
*
|
|
* @v filename Filename for image
|
|
* @v name Name for image, or NULL
|
|
* @ret new_image Newly created image
|
|
* @ret rc Return status code
|
|
*/
|
|
int imgfetch ( const char *filename, const char *name,
|
|
struct image **new_image ) {
|
|
struct image *image;
|
|
int rc;
|
|
|
|
/* Allocate new image */
|
|
image = malloc ( sizeof ( *image ) );
|
|
if ( ! image )
|
|
return -ENOMEM;
|
|
memset ( image, 0, sizeof ( *image ) );
|
|
|
|
/* Fill in image name */
|
|
if ( name )
|
|
strncpy ( image->name, name, ( sizeof ( image->name ) - 1 ) );
|
|
|
|
/* Fetch the file */
|
|
if ( ( rc = fetch ( filename, &image->data, &image->len ) ) != 0 )
|
|
goto err;
|
|
|
|
/* Register the image */
|
|
if ( ( rc = register_image ( image ) ) != 0 )
|
|
goto err;
|
|
|
|
*new_image = image;
|
|
return 0;
|
|
|
|
err:
|
|
efree ( image->data );
|
|
free ( image );
|
|
return rc;
|
|
}
|
|
|
|
/**
|
|
* Load an image
|
|
*
|
|
* @v image Image
|
|
* @ret rc Return status code
|
|
*/
|
|
int imgload ( struct image *image ) {
|
|
int rc;
|
|
|
|
/* Try to load image */
|
|
if ( ( rc = image_autoload ( image ) ) != 0 )
|
|
return rc;
|
|
|
|
/* If we succeed, move the image to the start of the list */
|
|
promote_image ( image );
|
|
|
|
return 0;
|
|
}
|
|
|
|
/**
|
|
* Execute an image
|
|
*
|
|
* @v image Image
|
|
* @ret rc Return status code
|
|
*/
|
|
int imgexec ( struct image *image ) {
|
|
return image_exec ( image );
|
|
}
|
|
|
|
/**
|
|
* Identify the first loaded image
|
|
*
|
|
* @ret image Image, or NULL
|
|
*/
|
|
struct image * imgautoselect ( void ) {
|
|
struct image *image;
|
|
|
|
for_each_image ( image ) {
|
|
if ( image->flags & IMAGE_LOADED )
|
|
return image;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
/**
|
|
* Display status of an image
|
|
*
|
|
* @v image Executable/loadable image
|
|
*/
|
|
void imgstat ( struct image *image ) {
|
|
printf ( "%s: %zd bytes", image->name, image->len );
|
|
if ( image->type )
|
|
printf ( " [%s]", image->type->name );
|
|
if ( image->flags & IMAGE_LOADED )
|
|
printf ( " [LOADED]" );
|
|
if ( image->cmdline[0] )
|
|
printf ( " \"%s\"", image->cmdline );
|
|
printf ( "\n" );
|
|
}
|
|
|
|
/**
|
|
* Free an image
|
|
*
|
|
* @v image Executable/loadable image
|
|
*/
|
|
void imgfree ( struct image *image ) {
|
|
unregister_image ( image );
|
|
efree ( image->data );
|
|
free ( image );
|
|
}
|