![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to imgmgmt.c CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [Qemu by Fabrice Bellard] / qemu / roms / ipxe / src / usr|
Return to imgmgmt.c CVS log | Up to [Qemu by Fabrice Bellard] / qemu / roms / ipxe / src / usr |
1.1 ! root 1: /* ! 2: * Copyright (C) 2007 Michael Brown <[email protected]>. ! 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: FILE_LICENCE ( GPL2_OR_LATER ); ! 20: ! 21: #include <stdint.h> ! 22: #include <stdlib.h> ! 23: #include <stdio.h> ! 24: #include <errno.h> ! 25: #include <ipxe/image.h> ! 26: #include <ipxe/downloader.h> ! 27: #include <ipxe/monojob.h> ! 28: #include <ipxe/open.h> ! 29: #include <ipxe/uri.h> ! 30: #include <usr/imgmgmt.h> ! 31: ! 32: /** @file ! 33: * ! 34: * Image management ! 35: * ! 36: */ ! 37: ! 38: /** ! 39: * Register an image and leave it registered ! 40: * ! 41: * @v image Executable image ! 42: * @ret rc Return status code ! 43: * ! 44: * This function assumes an ownership of the passed image. ! 45: */ ! 46: int register_and_put_image ( struct image *image ) { ! 47: int rc; ! 48: ! 49: rc = register_image ( image ); ! 50: image_put ( image ); ! 51: return rc; ! 52: } ! 53: ! 54: /** ! 55: * Register and probe an image ! 56: * ! 57: * @v image Executable image ! 58: * @ret rc Return status code ! 59: * ! 60: * This function assumes an ownership of the passed image. ! 61: */ ! 62: int register_and_probe_image ( struct image *image ) { ! 63: int rc; ! 64: ! 65: if ( ( rc = register_and_put_image ( image ) ) != 0 ) ! 66: return rc; ! 67: ! 68: if ( ( rc = image_probe ( image ) ) != 0 ) ! 69: return rc; ! 70: ! 71: return 0; ! 72: } ! 73: ! 74: /** ! 75: * Register and select an image ! 76: * ! 77: * @v image Executable image ! 78: * @ret rc Return status code ! 79: * ! 80: * This function assumes an ownership of the passed image. ! 81: */ ! 82: int register_and_select_image ( struct image *image ) { ! 83: int rc; ! 84: ! 85: if ( ( rc = register_and_probe_image ( image ) ) != 0 ) ! 86: return rc; ! 87: ! 88: if ( ( rc = image_select ( image ) ) != 0 ) ! 89: return rc; ! 90: ! 91: return 0; ! 92: } ! 93: ! 94: /** ! 95: * Register and boot an image ! 96: * ! 97: * @v image Image ! 98: * @ret rc Return status code ! 99: * ! 100: * This function assumes an ownership of the passed image. ! 101: */ ! 102: int register_and_boot_image ( struct image *image ) { ! 103: int rc; ! 104: ! 105: if ( ( rc = register_and_select_image ( image ) ) != 0 ) ! 106: return rc; ! 107: ! 108: if ( ( rc = image_exec ( image ) ) != 0 ) ! 109: return rc; ! 110: ! 111: return 0; ! 112: } ! 113: ! 114: /** ! 115: * Register and replace image ! 116: * ! 117: * @v image Image ! 118: * @ret rc Return status code ! 119: * ! 120: * This function assumes an ownership of the passed image. ! 121: */ ! 122: int register_and_replace_image ( struct image *image ) { ! 123: int rc; ! 124: ! 125: if ( ( rc = register_and_probe_image ( image ) ) != 0 ) ! 126: return rc; ! 127: ! 128: if ( ( rc = image_replace ( image ) ) != 0 ) ! 129: return rc; ! 130: ! 131: return 0; ! 132: } ! 133: ! 134: /** ! 135: * Download an image ! 136: * ! 137: * @v uri URI ! 138: * @v name Image name, or NULL to use default ! 139: * @v cmdline Command line, or NULL for no command line ! 140: * @v action Action to take upon a successful download ! 141: * @ret rc Return status code ! 142: */ ! 143: int imgdownload ( struct uri *uri, const char *name, const char *cmdline, ! 144: int ( * action ) ( struct image *image ) ) { ! 145: struct image *image; ! 146: size_t len = ( unparse_uri ( NULL, 0, uri, URI_ALL ) + 1 ); ! 147: char uri_string_redacted[len]; ! 148: const char *password; ! 149: int rc; ! 150: ! 151: /* Allocate image */ ! 152: image = alloc_image(); ! 153: if ( ! image ) ! 154: return -ENOMEM; ! 155: ! 156: /* Set image name */ ! 157: if ( name ) ! 158: image_set_name ( image, name ); ! 159: ! 160: /* Set image URI */ ! 161: image_set_uri ( image, uri ); ! 162: ! 163: /* Set image command line */ ! 164: image_set_cmdline ( image, cmdline ); ! 165: ! 166: /* Redact password portion of URI, if necessary */ ! 167: password = uri->password; ! 168: if ( password ) ! 169: uri->password = "***"; ! 170: unparse_uri ( uri_string_redacted, sizeof ( uri_string_redacted ), ! 171: uri, URI_ALL ); ! 172: uri->password = password; ! 173: ! 174: /* Create downloader */ ! 175: if ( ( rc = create_downloader ( &monojob, image, LOCATION_URI, ! 176: uri ) ) != 0 ) { ! 177: image_put ( image ); ! 178: return rc; ! 179: } ! 180: ! 181: /* Wait for download to complete */ ! 182: if ( ( rc = monojob_wait ( uri_string_redacted ) ) != 0 ) { ! 183: image_put ( image ); ! 184: return rc; ! 185: } ! 186: ! 187: /* Act upon downloaded image. This action assumes our ! 188: * ownership of the image. ! 189: */ ! 190: if ( ( rc = action ( image ) ) != 0 ) ! 191: return rc; ! 192: ! 193: return 0; ! 194: } ! 195: ! 196: /** ! 197: * Download an image ! 198: * ! 199: * @v uri_string URI as a string (e.g. "http://www.nowhere.com/vmlinuz") ! 200: * @v name Image name, or NULL to use default ! 201: * @v cmdline Command line, or NULL for no command line ! 202: * @v action Action to take upon a successful download ! 203: * @ret rc Return status code ! 204: */ ! 205: int imgdownload_string ( const char *uri_string, const char *name, ! 206: const char *cmdline, ! 207: int ( * action ) ( struct image *image ) ) { ! 208: struct uri *uri; ! 209: int rc; ! 210: ! 211: if ( ! ( uri = parse_uri ( uri_string ) ) ) ! 212: return -ENOMEM; ! 213: ! 214: rc = imgdownload ( uri, name, cmdline, action ); ! 215: ! 216: uri_put ( uri ); ! 217: return rc; ! 218: } ! 219: ! 220: /** ! 221: * Display status of an image ! 222: * ! 223: * @v image Executable/loadable image ! 224: */ ! 225: void imgstat ( struct image *image ) { ! 226: printf ( "%s : %zd bytes", image->name, image->len ); ! 227: if ( image->type ) ! 228: printf ( " [%s]", image->type->name ); ! 229: if ( image->flags & IMAGE_SELECTED ) ! 230: printf ( " [SELECTED]" ); ! 231: if ( image->cmdline ) ! 232: printf ( " \"%s\"", image->cmdline ); ! 233: printf ( "\n" ); ! 234: } ! 235: ! 236: /** ! 237: * Free an image ! 238: * ! 239: * @v image Executable/loadable image ! 240: */ ! 241: void imgfree ( struct image *image ) { ! 242: unregister_image ( image ); ! 243: }
This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.