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_progress ( &async,
  60                                            start_download ( uri_string, &async,
  61                                                             &image->data,
  62                                                             &image->len )))!=0)
  63                 goto err;
  64
  65         /* Register the image */
  66         if ( ( rc = register_image ( image ) ) != 0 )
  67                 goto err;
  68
  69         *new_image = image;
  70         return 0;
  71
  72  err:
  73         ufree ( image->data );
  74         free ( image );
  75         return rc;
  76 }
  77
  78 /**
  79  * Load an image
  80  *
  81  * @v image             Image
  82  * @ret rc              Return status code
  83  */
  84 int imgload ( struct image *image ) {
  85         int rc;
  86
  87         /* Try to load image */
  88         if ( ( rc = image_autoload ( image ) ) != 0 )
  89                 return rc;
  90
  91         /* If we succeed, move the image to the start of the list */
  92 #warning "No longer exists"
  93         //      promote_image ( image );
  94
  95         return 0;
  96 }
  97
  98 /**
  99  * Execute an image
 100  *
 101  * @v image             Image
 102  * @ret rc              Return status code
 103  */
 104 int imgexec ( struct image *image ) {
 105         return image_exec ( image );
 106 }
 107
 108 /**
 109  * Identify the first loaded image
 110  *
 111  * @ret image           Image, or NULL
 112  */
 113 struct image * imgautoselect ( void ) {
 114         struct image *image;
 115
 116         for_each_image ( image ) {
 117                 if ( image->flags & IMAGE_LOADED )
 118                         return image;
 119         }
 120
 121         return NULL;
 122 }
 123
 124 /**
 125  * Display status of an image
 126  *
 127  * @v image             Executable/loadable image
 128  */
 129 void imgstat ( struct image *image ) {
 130         printf ( "%s: %zd bytes", image->name, image->len );
 131         if ( image->type )
 132                 printf ( " [%s]", image->type->name );
 133         if ( image->flags & IMAGE_LOADED )
 134                 printf ( " [LOADED]" );
 135         if ( image->cmdline[0] )
 136                 printf ( " \"%s\"", image->cmdline );
 137         printf ( "\n" );
 138 }
 139
 140 /**
 141  * Free an image
 142  *
 143  * @v image             Executable/loadable image
 144  */
 145 void imgfree ( struct image *image ) {
 146         unregister_image ( image );
 147         ufree ( image->data );
 148         free ( image );
 149 }