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 * 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 40 default_memfree(void *arg, void *ptr) { 41 UNUSED(arg); 42 free(ptr); 43 } 44 45 ATF_TC(isc_mem_total); 46 ATF_TC_HEAD(isc_mem_total, tc) { 47 atf_tc_set_md_var(tc, "descr", "test TotalUse calculation"); 48 } 49 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); 113 ATF_TC_HEAD(isc_mem_inuse, tc) { 114 atf_tc_set_md_var(tc, "descr", "test InUse calculation"); 115 } 116 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 */ 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