diff options
author | Michael Brown | 2007-06-09 18:32:27 +0200 |
---|---|---|
committer | Michael Brown | 2007-06-09 18:32:27 +0200 |
commit | 08e286714fc8a73960525d0f8d1f1c4b04a752e7 (patch) | |
tree | dd2f53bf1595520e05480ee0cc8367285f3e8bd1 /src/core/image.c | |
parent | Merge branch 'master' into mcb-tcp-xfer (diff) | |
download | ipxe-08e286714fc8a73960525d0f8d1f1c4b04a752e7.tar.gz ipxe-08e286714fc8a73960525d0f8d1f1c4b04a752e7.tar.xz ipxe-08e286714fc8a73960525d0f8d1f1c4b04a752e7.zip |
Add reference counting to register/unregister procedure.
Diffstat (limited to 'src/core/image.c')
-rw-r--r-- | src/core/image.c | 32 |
1 files changed, 32 insertions, 0 deletions
diff --git a/src/core/image.c b/src/core/image.c index 5a2ee824..e270540a 100644 --- a/src/core/image.c +++ b/src/core/image.c @@ -23,6 +23,7 @@ #include <errno.h> #include <assert.h> #include <gpxe/list.h> +#include <gpxe/umalloc.h> #include <gpxe/image.h> /** @file @@ -41,6 +42,35 @@ static struct image_type image_types_end[0] __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 * * @v image Executable/loadable image @@ -56,6 +86,7 @@ int register_image ( struct image *image ) { } /* Add to image list */ + image_get ( image ); list_add_tail ( &image->list, &images ); DBGC ( image, "IMAGE %p at [%lx,%lx) registered as %s\n", image, user_to_phys ( image->data, 0 ), @@ -71,6 +102,7 @@ int register_image ( struct image *image ) { */ void unregister_image ( struct image *image ) { list_del ( &image->list ); + image_put ( image ); DBGC ( image, "IMAGE %p unregistered\n", image ); } |