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