Keep HTTP progress updated, and display a progress indicator during the
[people/dverkamp/gpxe.git] / 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         promote_image ( image );
93
94         return 0;
95 }
96
97 /**
98  * Execute an image
99  *
100  * @v image             Image
101  * @ret rc              Return status code
102  */
103 int imgexec ( struct image *image ) {
104         return image_exec ( image );
105 }
106
107 /**
108  * Identify the first loaded image
109  *
110  * @ret image           Image, or NULL
111  */
112 struct image * imgautoselect ( void ) {
113         struct image *image;
114
115         for_each_image ( image ) {
116                 if ( image->flags & IMAGE_LOADED )
117                         return image;
118         }
119
120         return NULL;
121 }
122
123 /**
124  * Display status of an image
125  *
126  * @v image             Executable/loadable image
127  */
128 void imgstat ( struct image *image ) {
129         printf ( "%s: %zd bytes", image->name, image->len );
130         if ( image->type )
131                 printf ( " [%s]", image->type->name );
132         if ( image->flags & IMAGE_LOADED )
133                 printf ( " [LOADED]" );
134         if ( image->cmdline[0] )
135                 printf ( " \"%s\"", image->cmdline );
136         printf ( "\n" );
137 }
138
139 /**
140  * Free an image
141  *
142  * @v image             Executable/loadable image
143  */
144 void imgfree ( struct image *image ) {
145         unregister_image ( image );
146         ufree ( image->data );
147         free ( image );
148 }