narenas
/
opengnsys_ipxe
mirror of https://github.com/ipxe/ipxe.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add reference counting to register/unregister procedure.

pull/1/head
Michael Brown 2007-06-09 17:32:27 +01:00
parent 53bd022235
commit 08e286714f
1 changed files with 32 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

32
src/core/image.c
View File

@ -23,6 +23,7 @@
#include <errno.h> #include <errno.h>
#include <assert.h> #include <assert.h>
#include <gpxe/list.h> #include <gpxe/list.h>
#include <gpxe/umalloc.h>
#include <gpxe/image.h> #include <gpxe/image.h>
/** @file /** @file
@ -40,6 +41,35 @@ static struct image_type image_types[0]
static struct image_type image_types_end[0] static struct image_type image_types_end[0]
__table_end ( struct image_type, image_types ); __table_end ( struct image_type, image_types );
/**
* Free executable/loadable image
*
* @v refcnt Reference counter
*/
static void free_image ( struct refcnt *refcnt ) {
struct image *image = container_of ( refcnt, struct image, refcnt );
ufree ( image->data );
free ( image );
DBGC ( image, "IMAGE %p freed\n", image );
}
/**
* Allocate executable/loadable image
*
* @ret image Executable/loadable image
*/
struct image * alloc_image ( void ) {
struct image *image;
image = malloc ( sizeof ( *image ) );
if ( image ) {
memset ( image, 0, sizeof ( *image ) );
image->refcnt.free = free_image;
}
return image;
}
/** /**
* Register executable/loadable image * Register executable/loadable image
* *
@ -56,6 +86,7 @@ int register_image ( struct image *image ) {
} }
/* Add to image list */ /* Add to image list */
image_get ( image );
list_add_tail ( &image->list, &images ); list_add_tail ( &image->list, &images );
DBGC ( image, "IMAGE %p at [%lx,%lx) registered as %s\n", DBGC ( image, "IMAGE %p at [%lx,%lx) registered as %s\n",
image, user_to_phys ( image->data, 0 ), image, user_to_phys ( image->data, 0 ),
@ -71,6 +102,7 @@ int register_image ( struct image *image ) {
*/ */
void unregister_image ( struct image *image ) { void unregister_image ( struct image *image ) {
list_del ( &image->list ); list_del ( &image->list );
image_put ( image );
DBGC ( image, "IMAGE %p unregistered\n", image ); DBGC ( image, "IMAGE %p unregistered\n", image );
} }