/*
|
* atexit.c --- Clean things up when we exit normally.
|
*
|
* Copyright Oracle, 2014
|
* Author Darrick J. Wong <darrick.wong@oracle.com>
|
*
|
* %Begin-Header%
|
* This file may be redistributed under the terms of the GNU Library
|
* General Public License, version 2.
|
* %End-Header%
|
*/
|
|
#ifndef _LARGEFILE_SOURCE
|
#define _LARGEFILE_SOURCE
|
#endif
|
#ifndef _LARGEFILE64_SOURCE
|
#define _LARGEFILE64_SOURCE
|
#endif
|
|
#include "config.h"
|
#include <stdlib.h>
|
|
#include "ext2_fs.h"
|
#include "ext2fs.h"
|
#include "ext2fsP.h"
|
|
struct exit_data {
|
ext2_exit_fn func;
|
void *data;
|
};
|
|
static struct exit_data *items;
|
static size_t nr_items;
|
|
static void handle_exit(void)
|
{
|
struct exit_data *ed;
|
|
for (ed = items + nr_items - 1; ed >= items; ed--) {
|
if (ed->func == NULL)
|
continue;
|
ed->func(ed->data);
|
}
|
|
ext2fs_free_mem(&items);
|
nr_items = 0;
|
}
|
|
/*
|
* Schedule a function to be called at (normal) program termination.
|
* If you want this to be called during a signal exit, you must capture
|
* the signal and call exit() yourself!
|
*/
|
errcode_t ext2fs_add_exit_fn(ext2_exit_fn func, void *data)
|
{
|
struct exit_data *ed, *free_ed = NULL;
|
size_t x;
|
errcode_t ret;
|
|
if (func == NULL)
|
return EXT2_ET_INVALID_ARGUMENT;
|
|
for (x = 0, ed = items; x < nr_items; x++, ed++) {
|
if (ed->func == func && ed->data == data)
|
return EXT2_ET_FILE_EXISTS;
|
if (ed->func == NULL)
|
free_ed = ed;
|
}
|
|
if (free_ed) {
|
free_ed->func = func;
|
free_ed->data = data;
|
return 0;
|
}
|
|
if (nr_items == 0) {
|
ret = atexit(handle_exit);
|
if (ret)
|
return ret;
|
}
|
|
ret = ext2fs_resize_mem(0, (nr_items + 1) * sizeof(struct exit_data),
|
&items);
|
if (ret)
|
return ret;
|
|
items[nr_items].func = func;
|
items[nr_items].data = data;
|
nr_items++;
|
|
return 0;
|
}
|
|
/* Remove a function from the exit cleanup list. */
|
errcode_t ext2fs_remove_exit_fn(ext2_exit_fn func, void *data)
|
{
|
struct exit_data *ed;
|
size_t x;
|
|
if (func == NULL)
|
return EXT2_ET_INVALID_ARGUMENT;
|
|
for (x = 0, ed = items; x < nr_items; x++, ed++) {
|
if (ed->func == NULL)
|
return 0;
|
if (ed->func == func && ed->data == data) {
|
size_t sz = (nr_items - (x + 1)) *
|
sizeof(struct exit_data);
|
memmove(ed, ed + 1, sz);
|
memset(items + nr_items - 1, 0,
|
sizeof(struct exit_data));
|
}
|
}
|
|
return 0;
|
}
|
