<commit_before>#include <stddef.h>
#include <stdlib.h>
#include <limits.h>
#include "libc.h"

/* Ensure that at least 32 atexit handlers can be registered without malloc */
#define COUNT 32

static struct fl
{
	struct fl *next;
	void (*f[COUNT])(void);
} builtin, *head;

void __funcs_on_exit()
{
	int i;
	for (; head; head=head->next) {
		for (i=COUNT-1; i>=0 && !head->f[i]; i--);
		for (; i>=0; i--) head->f[i]();
	}
}

int atexit(void (*func)(void))
{
	static int lock;
	int i;

	/* Hook for atexit extensions */
	if (libc.atexit) return libc.atexit(func);

	LOCK(&lock);

	/* Defer initialization of head so it can be in BSS */
	if (!head) head = &builtin;

	/* If the current function list is full, add a new one */
	if (head->f[COUNT-1]) {
		struct fl *new_fl = calloc(sizeof(struct fl), 1);
		if (!new_fl) {
			UNLOCK(&lock);
			return -1;
		}
		new_fl->next = head;
		head = new_fl;
	}

	/* Append function to the list. */
	for (i=0; i<COUNT && head->f[i]; i++);
	head->f[i] = func;

	UNLOCK(&lock);
	return 0;
}
<commit_msg>support __cxa_atexit, and registering atexit functions from atexit handlers<commit_after>#include <stddef.h>
#include <stdlib.h>
#include <stdint.h>
#include <limits.h>
#include "libc.h"

/* Ensure that at least 32 atexit handlers can be registered without malloc */
#define COUNT 32

static struct fl
{
	struct fl *next;
	void (*f[COUNT])(void *);
	void *a[COUNT];
} builtin, *head;

static int lock;

void __funcs_on_exit()
{
	int i;
	void (*func)(void *), *arg;
	LOCK(&lock);
	for (; head; head=head->next) {
		for (i=COUNT-1; i>=0 && !head->f[i]; i--);
		if (i<0) continue;
		func = head->f[i];
		arg = head->a[i];
		head->f[i] = 0;
		UNLOCK(&lock);
		func(arg);
		LOCK(&lock);
	}
}

int __cxa_atexit(void (*func)(void *), void *arg, void *dso)
{
	int i;

	LOCK(&lock);

	/* Defer initialization of head so it can be in BSS */
	if (!head) head = &builtin;

	/* If the current function list is full, add a new one */
	if (head->f[COUNT-1]) {
		struct fl *new_fl = calloc(sizeof(struct fl), 1);
		if (!new_fl) {
			UNLOCK(&lock);
			return -1;
		}
		new_fl->next = head;
		head = new_fl;
	}

	/* Append function to the list. */
	for (i=0; i<COUNT && head->f[i]; i++);
	head->f[i] = func;
	head->a[i] = arg;

	UNLOCK(&lock);
	return 0;
}

static void call(void *p)
{
	((void (*)(void))(uintptr_t)p)();
}

int atexit(void (*func)(void))
{
	return __cxa_atexit(call, (void *)(uintptr_t)func, 0);
}

{'fraction_non_alphanumeric': 0.1904566022213081, 'fraction_numerical': 0.0106951871657754, 'mean_word_length': 3.5886792452830187, 'pattern_counts': {'":': 0, '<': 13, '<?xml version=': 0, '>': 30, 'https://': 0, 'lorem ipsum': 0, 'www.': 0, 'xml': 0}, 'pii_count': 0, 'substrings_counts': 0, 'word_list_counts': {'cursed_substrings.json': 0, 'profanity_word_list.json': 0, 'sexual_word_list.json': 0, 'zh_pornsignals.json': 0}}
{'dir': 'git-commits-cleaned', 'id': '4778347', 'n_tokens_mistral': 1058, 'n_tokens_neox': 986, 'n_words': 282}