src/usr/imgmgmt.c

   1 /*
   2  * Copyright (C) 2007 Michael Brown <mbrown@fensystems.co.uk>.
   3  *
   4  * This program is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU General Public License as
   6  * published by the Free Software Foundation; either version 2 of the
   7  * License, or any later version.
   8  *
   9  * This program is distributed in the hope that it will be useful, but
  10  * WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU General Public License
  15  * along with this program; if not, write to the Free Software
  16  * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
  17  */
  18
  19 #include <stdint.h>
  20 #include <stdlib.h>
  21 #include <stdio.h>
  22 #include <errno.h>
  23 #include <gpxe/image.h>
  24 #include <gpxe/umalloc.h>
  25 #include <gpxe/download.h>
  26 #include <usr/imgmgmt.h>
  27
  28 /** @file
  29  *
  30  * Image management
  31  *
  32  */
  33
  34 /**
  35  * Fetch an image
  36  *
  37  * @v uri_string        URI as a string (e.g. "http://www.nowhere.com/vmlinuz")
  38  * @v name              Name for image, or NULL
  39  * @ret new_image       Newly created image
  40  * @ret rc              Return status code
  41  */
  42 int imgfetch ( const char *uri_string, const char *name,
  43                struct image **new_image ) {
  44         struct image *image;
  45         struct async async;
  46         int rc;
  47
  48         /* Allocate new image */
  49         image = malloc ( sizeof ( *image ) );
  50         if ( ! image )
  51                 return -ENOMEM;
  52         memset ( image, 0, sizeof ( *image ) );
  53
  54         /* Fill in image name */
  55         if ( name )
  56                 strncpy ( image->name, name, ( sizeof ( image->name ) - 1 ) );
  57
  58         /* Download the file */
  59         if ( ( rc = async_block ( &async, start_download ( uri_string, &async,
  60                                                            &image->data,
  61                                                            &image->len ))) !=0)
  62                 goto err;
  63
  64         /* Register the image */
  65         if ( ( rc = register_image ( image ) ) != 0 )
  66                 goto err;
  67
  68         *new_image = image;
  69         return 0;
  70
  71  err:
  72         ufree ( image->data );
  73         free ( image );
  74         return rc;
  75 }
  76
  77 /**
  78  * Load an image
  79  *
  80  * @v image             Image
  81  * @ret rc              Return status code
  82  */
  83 int imgload ( struct image *image ) {
  84         int rc;
  85
  86         /* Try to load image */
  87         if ( ( rc = image_autoload ( image ) ) != 0 )
  88                 return rc;
  89
  90         /* If we succeed, move the image to the start of the list */
  91         promote_image ( image );
  92
  93         return 0;
  94 }
  95
  96 /**
  97  * Execute an image
  98  *
  99  * @v image             Image
 100  * @ret rc              Return status code
 101  */
 102 int imgexec ( struct image *image ) {
 103         return image_exec ( image );
 104 }
 105
 106 /**
 107  * Identify the first loaded image
 108  *
 109  * @ret image           Image, or NULL
 110  */
 111 struct image * imgautoselect ( void ) {
 112         struct image *image;
 113
 114         for_each_image ( image ) {
 115                 if ( image->flags & IMAGE_LOADED )
 116                         return image;
 117         }
 118
 119         return NULL;
 120 }
 121
 122 /**
 123  * Display status of an image
 124  *
 125  * @v image             Executable/loadable image
 126  */
 127 void imgstat ( struct image *image ) {
 128         printf ( "%s: %zd bytes", image->name, image->len );
 129         if ( image->type )
 130                 printf ( " [%s]", image->type->name );
 131         if ( image->flags & IMAGE_LOADED )
 132                 printf ( " [LOADED]" );
 133         if ( image->cmdline[0] )
 134                 printf ( " \"%s\"", image->cmdline );
 135         printf ( "\n" );
 136 }
 137
 138 /**
 139  * Free an image
 140  *
 141  * @v image             Executable/loadable image
 142  */
 143 void imgfree ( struct image *image ) {
 144         unregister_image ( image );
 145         ufree ( image->data );
 146         free ( image );
 147 }