1 /* $NetBSD: mem_test.c,v 1.1.1.1 2015/07/08 15:38:05 christos Exp $ */
2
3 /*
4 * Copyright (C) 2015 Internet Systems Consortium, Inc. ("ISC")
5 *
6 * Permission to use, copy, modify, and/or distribute this software for any
7 * purpose with or without fee is hereby granted, provided that the above
8 * copyright notice and this permission notice appear in all copies.
9 *
10 * THE SOFTWARE IS PROVIDED "AS IS" AND ISC DISCLAIMS ALL WARRANTIES WITH
11 * REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF MERCHANTABILITY
12 * AND FITNESS. IN NO EVENT SHALL ISC BE LIABLE FOR ANY SPECIAL, DIRECT,
13 * INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES WHATSOEVER RESULTING FROM
14 * LOSS OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE
15 * OR OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
16 * PERFORMANCE OF THIS SOFTWARE.
17 */
18
19 #include <config.h>
20 #include <stdlib.h>
21 #include <unistd.h>
22 #include <fcntl.h>
23
24 #include <atf-c.h>
25
26 #include "isctest.h"
27
28 #include <isc/mem.h>
29 #include <isc/result.h>
30
31 static void *
default_memalloc(void * arg,size_t size)32 default_memalloc(void *arg, size_t size) {
33 UNUSED(arg);
34 if (size == 0U)
35 size = 1;
36 return (malloc(size));
37 }
38
39 static void
default_memfree(void * arg,void * ptr)40 default_memfree(void *arg, void *ptr) {
41 UNUSED(arg);
42 free(ptr);
43 }
44
45 ATF_TC(isc_mem_total);
ATF_TC_HEAD(isc_mem_total,tc)46 ATF_TC_HEAD(isc_mem_total, tc) {
47 atf_tc_set_md_var(tc, "descr", "test TotalUse calculation");
48 }
49
ATF_TC_BODY(isc_mem_total,tc)50 ATF_TC_BODY(isc_mem_total, tc) {
51 isc_result_t result;
52 isc_mem_t *mctx2 = NULL;
53 size_t before, after;
54 ssize_t diff;
55 int i;
56
57 result = isc_test_begin(NULL, ISC_TRUE);
58 ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
59
60 /* Local alloc, free */
61 mctx2 = NULL;
62 result = isc_mem_createx2(0, 0, default_memalloc, default_memfree,
63 NULL, &mctx2, 0);
64 if (result != ISC_R_SUCCESS)
65 goto out;
66
67 before = isc_mem_total(mctx2);
68
69 for (i = 0; i < 100000; i++) {
70 void *ptr;
71
72 ptr = isc_mem_allocate(mctx2, 2048);
73 isc_mem_free(mctx2, ptr);
74 }
75
76 after = isc_mem_total(mctx2);
77 diff = after - before;
78
79 printf("total_before=%lu, total_after=%lu, total_diff=%lu\n",
80 (unsigned long)before, (unsigned long)after,
81 (unsigned long)diff);
82 /* 2048 +8 bytes extra for size_info */
83 ATF_CHECK_EQ(diff, (2048 + 8) * 100000);
84
85 /* ISC_MEMFLAG_INTERNAL */
86
87 before = isc_mem_total(mctx);
88
89 for (i = 0; i < 100000; i++) {
90 void *ptr;
91
92 ptr = isc_mem_allocate(mctx, 2048);
93 isc_mem_free(mctx, ptr);
94 }
95
96 after = isc_mem_total(mctx);
97 diff = after - before;
98
99 printf("total_before=%lu, total_after=%lu, total_diff=%lu\n",
100 (unsigned long)before, (unsigned long)after,
101 (unsigned long)diff);
102 /* 2048 +8 bytes extra for size_info */
103 ATF_CHECK_EQ(diff, (2048 + 8) * 100000);
104
105 out:
106 if (mctx2 != NULL)
107 isc_mem_destroy(&mctx2);
108
109 isc_test_end();
110 }
111
112 ATF_TC(isc_mem_inuse);
ATF_TC_HEAD(isc_mem_inuse,tc)113 ATF_TC_HEAD(isc_mem_inuse, tc) {
114 atf_tc_set_md_var(tc, "descr", "test InUse calculation");
115 }
116
ATF_TC_BODY(isc_mem_inuse,tc)117 ATF_TC_BODY(isc_mem_inuse, tc) {
118 isc_result_t result;
119 isc_mem_t *mctx2 = NULL;
120 size_t before, during, after;
121 ssize_t diff;
122 void *ptr;
123
124 result = isc_test_begin(NULL, ISC_TRUE);
125 ATF_REQUIRE_EQ(result, ISC_R_SUCCESS);
126
127 mctx2 = NULL;
128 result = isc_mem_createx2(0, 0, default_memalloc, default_memfree,
129 NULL, &mctx2, 0);
130 if (result != ISC_R_SUCCESS)
131 goto out;
132
133 before = isc_mem_inuse(mctx2);
134 ptr = isc_mem_allocate(mctx2, 1024000);
135 during = isc_mem_inuse(mctx2);
136 isc_mem_free(mctx2, ptr);
137 after = isc_mem_inuse(mctx2);
138
139 diff = after - before;
140
141 printf("inuse_before=%lu, inuse_during=%lu, inuse_after=%lu\n",
142 (unsigned long)before, (unsigned long)during,
143 (unsigned long)after);
144 ATF_REQUIRE_EQ(diff, 0);
145
146 out:
147 if (mctx2 != NULL)
148 isc_mem_destroy(&mctx2);
149
150 isc_test_end();
151 }
152
153 /*
154 * Main
155 */
ATF_TP_ADD_TCS(tp)156 ATF_TP_ADD_TCS(tp) {
157 ATF_TP_ADD_TC(tp, isc_mem_total);
158 ATF_TP_ADD_TC(tp, isc_mem_inuse);
159
160 return (atf_no_error());
161 }
162