1 /* $NetBSD: chfs_pool.c,v 1.3 2018/01/29 15:48:50 sevan Exp $ */ 2 3 /*- 4 * Copyright (c) 2010 Department of Software Engineering, 5 * University of Szeged, Hungary 6 * All rights reserved. 7 * 8 * This code is derived from software contributed to The NetBSD Foundation 9 * by the Department of Software Engineering, University of Szeged, Hungary 10 * 11 * Redistribution and use in source and binary forms, with or without 12 * modification, are permitted provided that the following conditions 13 * are met: 14 * 1. Redistributions of source code must retain the above copyright 15 * notice, this list of conditions and the following disclaimer. 16 * 2. Redistributions in binary form must reproduce the above copyright 17 * notice, this list of conditions and the following disclaimer in the 18 * documentation and/or other materials provided with the distribution. 19 * 20 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR 21 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 22 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 23 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT, 24 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, 25 * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 26 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED 27 * AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, 28 * OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY 29 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF 30 * SUCH DAMAGE. 31 */ 32 33 /* 34 * Pool allocator and convenience routines for chfs. 35 */ 36 37 #include <sys/cdefs.h> 38 39 #include <sys/param.h> 40 #include <sys/pool.h> 41 #include <sys/atomic.h> 42 43 #include <uvm/uvm.h> 44 45 #include "chfs.h" 46 47 /* --------------------------------------------------------------------- */ 48 49 void * chfs_pool_page_alloc(struct pool *, int); 50 void chfs_pool_page_free(struct pool *, void *); 51 52 /* --------------------------------------------------------------------- */ 53 54 struct pool_allocator chfs_pool_allocator = { 55 .pa_alloc = chfs_pool_page_alloc, 56 .pa_free = chfs_pool_page_free, 57 }; 58 59 /* --------------------------------------------------------------------- */ 60 61 void 62 chfs_pool_init(struct chfs_pool *chpp, size_t size, const char *what, 63 struct chfs_mount *chmp) 64 { 65 int cnt; 66 67 cnt = snprintf(chpp->chp_name, sizeof(chpp->chp_name), 68 "%s_chfs_%p", what, chmp); 69 KASSERT(cnt < sizeof(chpp->chp_name)); 70 71 pool_init(&chpp->chp_pool, size, 0, 0, 0, chpp->chp_name, 72 &chfs_pool_allocator, IPL_NONE); 73 chpp->chp_mount = chmp; 74 } 75 76 /* --------------------------------------------------------------------- */ 77 78 void 79 chfs_pool_destroy(struct chfs_pool *chpp) 80 { 81 pool_destroy((struct pool *)chpp); 82 } 83 84 /* --------------------------------------------------------------------- */ 85 86 void * 87 chfs_pool_page_alloc(struct pool *pp, int flags) 88 { 89 struct chfs_pool *chpp; 90 struct chfs_mount *chmp; 91 unsigned int pages; 92 void *page; 93 dbg("CHFS: pool_page_alloc()\n"); 94 95 chpp = (struct chfs_pool *)pp; 96 chmp = chpp->chp_mount; 97 98 pages = atomic_inc_uint_nv(&chmp->chm_pages_used); 99 if (pages >= CHFS_PAGES_MAX(chmp)) { 100 atomic_dec_uint(&chmp->chm_pages_used); 101 return NULL; 102 } 103 page = pool_get(pp, flags | PR_WAITOK); 104 if (page == NULL) { 105 atomic_dec_uint(&chmp->chm_pages_used); 106 } 107 108 return page; 109 } 110 111 /* --------------------------------------------------------------------- */ 112 113 void 114 chfs_pool_page_free(struct pool *pp, void *v) 115 { 116 struct chfs_pool *chpp; 117 struct chfs_mount *chmp; 118 dbg("CHFS: pool_page_free()\n"); 119 120 chpp = (struct chfs_pool *)pp; 121 chmp = chpp->chp_mount; 122 123 atomic_dec_uint(&chmp->chm_pages_used); 124 pool_put(pp,v); 125 } 126 127 /* --------------------------------------------------------------------- */ 128 129 void 130 chfs_str_pool_init(struct chfs_str_pool *chsp, struct chfs_mount *chmp) 131 { 132 dbg("CHFS: str_pool_init()\n"); 133 134 chfs_pool_init(&chsp->chsp_pool_16, 16, "str", chmp); 135 chfs_pool_init(&chsp->chsp_pool_32, 32, "str", chmp); 136 chfs_pool_init(&chsp->chsp_pool_64, 64, "str", chmp); 137 chfs_pool_init(&chsp->chsp_pool_128, 128, "str", chmp); 138 chfs_pool_init(&chsp->chsp_pool_256, 256, "str", chmp); 139 chfs_pool_init(&chsp->chsp_pool_512, 512, "str", chmp); 140 chfs_pool_init(&chsp->chsp_pool_1024, 1024, "str", chmp); 141 } 142 143 /* --------------------------------------------------------------------- */ 144 145 void 146 chfs_str_pool_destroy(struct chfs_str_pool *chsp) 147 { 148 dbg("CHFS: str_pool_destroy()\n"); 149 150 chfs_pool_destroy(&chsp->chsp_pool_16); 151 chfs_pool_destroy(&chsp->chsp_pool_32); 152 chfs_pool_destroy(&chsp->chsp_pool_64); 153 chfs_pool_destroy(&chsp->chsp_pool_128); 154 chfs_pool_destroy(&chsp->chsp_pool_256); 155 chfs_pool_destroy(&chsp->chsp_pool_512); 156 chfs_pool_destroy(&chsp->chsp_pool_1024); 157 } 158 159 /* --------------------------------------------------------------------- */ 160 161 char * 162 chfs_str_pool_get(struct chfs_str_pool *chsp, size_t len, int flags) 163 { 164 struct chfs_pool *p; 165 dbg("CHFS: str_pool_get()\n"); 166 167 KASSERT(len <= 1024); 168 169 if (len <= 16) p = &chsp->chsp_pool_16; 170 else if (len <= 32) p = &chsp->chsp_pool_32; 171 else if (len <= 64) p = &chsp->chsp_pool_64; 172 else if (len <= 128) p = &chsp->chsp_pool_128; 173 else if (len <= 256) p = &chsp->chsp_pool_256; 174 else if (len <= 512) p = &chsp->chsp_pool_512; 175 else if (len <= 1024) p = &chsp->chsp_pool_1024; 176 else { 177 KASSERT(0); 178 p = NULL; /* Silence compiler warnings */ 179 } 180 181 return (char *)CHFS_POOL_GET(p, flags); 182 } 183 184 /* --------------------------------------------------------------------- */ 185 186 void 187 chfs_str_pool_put(struct chfs_str_pool *chsp, char *str, size_t len) 188 { 189 struct chfs_pool *p; 190 dbg("CHFS: str_pool_put()\n"); 191 192 KASSERT(len <= 1024); 193 194 if (len <= 16) p = &chsp->chsp_pool_16; 195 else if (len <= 32) p = &chsp->chsp_pool_32; 196 else if (len <= 64) p = &chsp->chsp_pool_64; 197 else if (len <= 128) p = &chsp->chsp_pool_128; 198 else if (len <= 256) p = &chsp->chsp_pool_256; 199 else if (len <= 512) p = &chsp->chsp_pool_512; 200 else if (len <= 1024) p = &chsp->chsp_pool_1024; 201 else { 202 KASSERT(0); 203 p = NULL; /* Silence compiler warnings */ 204 } 205 206 CHFS_POOL_PUT(p, str); 207 } 208