Use stdio.h instead of vsprintf.h
[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 ( &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 }