summaryrefslogtreecommitdiffstats
path: root/tests/context/context_test.c
blob: e9a55593d7ea69682a3dab42982ffc9f062f38c9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
#include <stdint.h>
#include <stdlib.h>
#include <unistd.h>
#include <stdio.h>

#include <sys/eventfd.h>

#include <osmocom/core/logging.h>
#include <osmocom/core/application.h>
#include <osmocom/core/utils.h>
#include <osmocom/core/select.h>
#include <osmocom/core/talloc.h>

static struct osmo_fd g_evfd;

static void *alloc_res_select;
static void *alloc_res_global;

static int destructor_called;

static int talloc_destructor(void *ptr)
{
	printf("destructor was called automatically\n");
	/* ensure the destructor is only called for the chunk allocated from the
	 * volatile select context */
	OSMO_ASSERT(ptr == alloc_res_select);
	destructor_called += 1;
	return 0;
}

static int evfd_cb(struct osmo_fd *ofd, unsigned int what)
{
	uint64_t rval;
	int rc;

	rc = read(ofd->fd, &rval, sizeof(rval));
	OSMO_ASSERT(rc == sizeof(rval));

	printf("allocating from select context\n");
	alloc_res_select = talloc_named_const(OTC_SELECT, 23, "alloc_select");
	OSMO_ASSERT(alloc_res_select);
	talloc_set_destructor(alloc_res_select, talloc_destructor);

	printf("allocating from global context\n");
	alloc_res_global = talloc_named_const(OTC_GLOBAL, 42, "alloc_global");
	OSMO_ASSERT(alloc_res_global);
	talloc_set_destructor(alloc_res_global, talloc_destructor);
	return 0;
}

const struct log_info_cat default_categories[] = {
};

static struct log_info info = {
	.cat = default_categories,
	.num_cat = ARRAY_SIZE(default_categories),
};

int main(int argc, char **argv)
{
	int rc;

	osmo_init_logging2(OTC_GLOBAL, &info);

	rc = eventfd(0, 0);
	OSMO_ASSERT(rc >= 0);
	osmo_fd_setup(&g_evfd, rc, OSMO_FD_READ, evfd_cb, NULL, 0);
	osmo_fd_register(&g_evfd);

	/* make sure the select loop will immediately call the callback */
	uint64_t val = 1;
	rc = write(g_evfd.fd, &val, sizeof(val));
	OSMO_ASSERT(rc == sizeof(val));

	/* enter osmo_select_main_ctx() once */
	printf("entering osmo_select_main\n");
	osmo_select_main_ctx(1);

	/* the allocation must have happened, and the destructor must have been called
	 * automatically exactly once */
	OSMO_ASSERT(destructor_called == 1);

	exit(0);
}