2007-01-12 06:05:27 +00:00
|
|
|
/*
|
|
|
|
* 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>
|
2007-01-12 07:22:20 +00:00
|
|
|
#include <stdlib.h>
|
2007-01-19 01:13:12 +00:00
|
|
|
#include <stdio.h>
|
2007-01-12 07:22:20 +00:00
|
|
|
#include <errno.h>
|
2007-01-12 06:05:27 +00:00
|
|
|
#include <gpxe/image.h>
|
2007-06-28 16:55:29 +00:00
|
|
|
#include <gpxe/downloader.h>
|
|
|
|
#include <gpxe/monojob.h>
|
|
|
|
#include <gpxe/open.h>
|
2007-08-02 19:18:32 +00:00
|
|
|
#include <gpxe/uri.h>
|
2007-01-12 06:05:27 +00:00
|
|
|
#include <usr/imgmgmt.h>
|
|
|
|
|
|
|
|
/** @file
|
|
|
|
*
|
|
|
|
* Image management
|
|
|
|
*
|
|
|
|
*/
|
|
|
|
|
2007-01-12 07:22:20 +00:00
|
|
|
/**
|
|
|
|
* Fetch an image
|
|
|
|
*
|
2007-01-18 03:39:45 +00:00
|
|
|
* @v uri_string URI as a string (e.g. "http://www.nowhere.com/vmlinuz")
|
2007-01-12 07:22:20 +00:00
|
|
|
* @v name Name for image, or NULL
|
2007-08-02 19:18:32 +00:00
|
|
|
* @v register_image Image registration routine
|
2007-01-12 07:22:20 +00:00
|
|
|
* @ret rc Return status code
|
|
|
|
*/
|
2007-08-02 19:18:32 +00:00
|
|
|
int imgfetch ( struct image *image, const char *uri_string,
|
|
|
|
int ( * image_register ) ( struct image *image ) ) {
|
|
|
|
struct uri *uri;
|
2007-01-12 07:22:20 +00:00
|
|
|
int rc;
|
|
|
|
|
2007-08-02 19:18:32 +00:00
|
|
|
if ( ! ( uri = parse_uri ( uri_string ) ) )
|
|
|
|
return -ENOMEM;
|
|
|
|
|
|
|
|
image_set_uri ( image, uri );
|
|
|
|
|
|
|
|
if ( ( rc = create_downloader ( &monojob, image, image_register,
|
|
|
|
LOCATION_URI, uri ) ) == 0 )
|
2007-08-03 11:49:21 +00:00
|
|
|
rc = monojob_wait ( uri_string );
|
2007-01-12 07:22:20 +00:00
|
|
|
|
2007-08-02 19:18:32 +00:00
|
|
|
uri_put ( uri );
|
2007-01-12 07:22:20 +00:00
|
|
|
return rc;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Load an image
|
|
|
|
*
|
|
|
|
* @v image Image
|
|
|
|
* @ret rc Return status code
|
|
|
|
*/
|
|
|
|
int imgload ( struct image *image ) {
|
2007-01-12 09:46:10 +00:00
|
|
|
int rc;
|
|
|
|
|
|
|
|
/* Try to load image */
|
|
|
|
if ( ( rc = image_autoload ( image ) ) != 0 )
|
|
|
|
return rc;
|
|
|
|
|
|
|
|
return 0;
|
2007-01-12 07:22:20 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* Execute an image
|
|
|
|
*
|
|
|
|
* @v image Image
|
|
|
|
* @ret rc Return status code
|
|
|
|
*/
|
|
|
|
int imgexec ( struct image *image ) {
|
|
|
|
return image_exec ( image );
|
|
|
|
}
|
|
|
|
|
2007-01-12 07:32:46 +00:00
|
|
|
/**
|
|
|
|
* 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;
|
|
|
|
}
|
|
|
|
|
2007-01-12 06:05:27 +00:00
|
|
|
/**
|
|
|
|
* Display status of an image
|
|
|
|
*
|
|
|
|
* @v image Executable/loadable image
|
|
|
|
*/
|
|
|
|
void imgstat ( struct image *image ) {
|
2007-01-12 07:22:20 +00:00
|
|
|
printf ( "%s: %zd bytes", image->name, image->len );
|
2007-01-12 06:05:27 +00:00
|
|
|
if ( image->type )
|
|
|
|
printf ( " [%s]", image->type->name );
|
|
|
|
if ( image->flags & IMAGE_LOADED )
|
|
|
|
printf ( " [LOADED]" );
|
2007-08-02 19:18:32 +00:00
|
|
|
if ( image->cmdline )
|
2007-01-12 07:22:20 +00:00
|
|
|
printf ( " \"%s\"", image->cmdline );
|
2007-01-12 06:05:27 +00:00
|
|
|
printf ( "\n" );
|
|
|
|
}
|
|
|
|
|
2007-01-12 07:22:20 +00:00
|
|
|
/**
|
|
|
|
* Free an image
|
|
|
|
*
|
|
|
|
* @v image Executable/loadable image
|
|
|
|
*/
|
|
|
|
void imgfree ( struct image *image ) {
|
|
|
|
unregister_image ( image );
|
|
|
|
}
|