sys/sys/malloc.h

   1 /*
   2  * Copyright (c) 1987, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *      This product includes software developed by the University of
  16  *      California, Berkeley and its contributors.
  17  * 4. Neither the name of the University nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  *
  33  *      @(#)malloc.h    8.5 (Berkeley) 5/3/95
  34  * $FreeBSD$
  35  */
  36
  37 #ifndef _SYS_MALLOC_H_
  38 #define _SYS_MALLOC_H_
  39
  40 #define splmem splhigh
  41
  42 #define KMEMSTATS
  43
  44 /*
  45  * flags to malloc.
  46  */
  47
  48 #define M_WAITOK        0x0000
  49 #define M_NOWAIT        0x0001          /* do not block                 */
  50 #define M_USE_RESERVE   0x0002          /* can alloc out of reserve memory  */
  51 #define M_ASLEEP        0x0004          /* async sleep on failure       */
  52
  53 #define M_MAGIC         877983977       /* time when first defined :-) */
  54
  55 struct malloc_type {
  56         struct malloc_type *ks_next;    /* next in list */
  57         long    ks_memuse;      /* total memory held in bytes */
  58         long    ks_limit;       /* most that are allowed to exist */
  59         long    ks_size;        /* sizes of this thing that are allocated */
  60         long    ks_inuse;       /* # of packets of this type currently in use */
  61         long    ks_calls;       /* total packets of this type ever allocated */
  62         long    ks_maxused;     /* maximum number ever used */
  63         u_long  ks_magic;       /* if it's not magic, don't touch it */
  64         const char *ks_shortdesc;       /* short description */
  65         u_short ks_limblocks;   /* number of times blocked for hitting limit */
  66         u_short ks_mapblocks;   /* number of times blocked for kernel map */
  67 };
  68
  69 #ifdef KERNEL
  70
  71 void    malloc_init __P((void *));
  72 void    malloc_uninit __P((void *));
  73
  74 #define MALLOC_DEFINE(type, shortdesc, longdesc) \
  75         struct malloc_type type[1] = { \
  76                 { NULL, 0, 0, 0, 0, 0, 0, M_MAGIC, shortdesc, 0, 0 } \
  77         }; \
  78         SYSINIT(type##_init, SI_SUB_KMEM, SI_ORDER_ANY, malloc_init, type); \
  79         SYSUNINIT(type##_uninit, SI_SUB_KMEM, SI_ORDER_ANY, malloc_uninit, type)
  80
  81 #define MALLOC_DECLARE(type) \
  82         extern struct malloc_type type[1]
  83
  84 MALLOC_DECLARE(M_CACHE);
  85 MALLOC_DECLARE(M_DEVBUF);
  86 MALLOC_DECLARE(M_TEMP);
  87 #endif  /* KERNEL */
  88
  89 /*
  90  * Array of descriptors that describe the contents of each page
  91  */
  92 struct kmemusage {
  93         short ku_indx;          /* bucket index */
  94         union {
  95                 u_short freecnt;/* for small allocations, free pieces in page */
  96                 u_short pagecnt;/* for large allocations, pages alloced */
  97         } ku_un;
  98 };
  99 #define ku_freecnt ku_un.freecnt
 100 #define ku_pagecnt ku_un.pagecnt
 101
 102 /*
 103  * Set of buckets for each size of memory block that is retained
 104  */
 105 struct kmembuckets {
 106         caddr_t kb_next;        /* list of free blocks */
 107         caddr_t kb_last;        /* last free block */
 108         long    kb_total;       /* total number of blocks allocated */
 109         long    kb_elmpercl;    /* # of elements in this sized allocation */
 110         long    kb_totalfree;   /* # of free elements in this bucket */
 111         long    kb_calls;       /* total calls to allocate this size */
 112         long    kb_highwat;     /* high water mark */
 113         long    kb_couldfree;   /* over high water mark and could free */
 114 };
 115
 116 #ifdef KERNEL
 117
 118 #define MINALLOCSIZE    (1 << MINBUCKET)
 119 #define BUCKETINDX(size) \
 120         ((size) <= (MINALLOCSIZE * 128) \
 121                 ? (size) <= (MINALLOCSIZE * 8) \
 122                         ? (size) <= (MINALLOCSIZE * 2) \
 123                                 ? (size) <= (MINALLOCSIZE * 1) \
 124                                         ? (MINBUCKET + 0) \
 125                                         : (MINBUCKET + 1) \
 126                                 : (size) <= (MINALLOCSIZE * 4) \
 127                                         ? (MINBUCKET + 2) \
 128                                         : (MINBUCKET + 3) \
 129                         : (size) <= (MINALLOCSIZE* 32) \
 130                                 ? (size) <= (MINALLOCSIZE * 16) \
 131                                         ? (MINBUCKET + 4) \
 132                                         : (MINBUCKET + 5) \
 133                                 : (size) <= (MINALLOCSIZE * 64) \
 134                                         ? (MINBUCKET + 6) \
 135                                         : (MINBUCKET + 7) \
 136                 : (size) <= (MINALLOCSIZE * 2048) \
 137                         ? (size) <= (MINALLOCSIZE * 512) \
 138                                 ? (size) <= (MINALLOCSIZE * 256) \
 139                                         ? (MINBUCKET + 8) \
 140                                         : (MINBUCKET + 9) \
 141                                 : (size) <= (MINALLOCSIZE * 1024) \
 142                                         ? (MINBUCKET + 10) \
 143                                         : (MINBUCKET + 11) \
 144                         : (size) <= (MINALLOCSIZE * 8192) \
 145                                 ? (size) <= (MINALLOCSIZE * 4096) \
 146                                         ? (MINBUCKET + 12) \
 147                                         : (MINBUCKET + 13) \
 148                                 : (size) <= (MINALLOCSIZE * 16384) \
 149                                         ? (MINBUCKET + 14) \
 150                                         : (MINBUCKET + 15))
 151
 152 /*
 153  * Turn virtual addresses into kmem map indices
 154  */
 155 #define kmemxtob(alloc) (kmembase + (alloc) * PAGE_SIZE)
 156 #define btokmemx(addr)  (((caddr_t)(addr) - kmembase) / PAGE_SIZE)
 157 #define btokup(addr)    (&kmemusage[((caddr_t)(addr) - kmembase) >> PAGE_SHIFT])
 158
 159 /*
 160  * Macro versions for the usual cases of malloc/free
 161  */
 162 #if defined(KMEMSTATS) || defined(DIAGNOSTIC)
 163 #define MALLOC(space, cast, size, type, flags) \
 164         (space) = (cast)malloc((u_long)(size), type, flags)
 165 #define FREE(addr, type) free((addr), type)
 166
 167 #else /* do not collect statistics */
 168 #define MALLOC(space, cast, size, type, flags) do { \
 169         register struct kmembuckets *kbp = &bucket[BUCKETINDX(size)]; \
 170         long s = splmem(); \
 171         if (kbp->kb_next == NULL) { \
 172                 (space) = (cast)malloc((u_long)(size), type, flags); \
 173         } else { \
 174                 (space) = (cast)kbp->kb_next; \
 175                 kbp->kb_next = *(caddr_t *)(space); \
 176         } \
 177         splx(s); \
 178 } while (0)
 179
 180 #define FREE(addr, type) do { \
 181         register struct kmembuckets *kbp; \
 182         register struct kmemusage *kup = btokup(addr); \
 183         long s = splmem(); \
 184         if (1 << kup->ku_indx > MAXALLOCSAVE) { \
 185                 free((addr), type); \
 186         } else { \
 187                 kbp = &bucket[kup->ku_indx]; \
 188                 if (kbp->kb_next == NULL) \
 189                         kbp->kb_next = (caddr_t)(addr); \
 190                 else \
 191                         *(caddr_t *)(kbp->kb_last) = (caddr_t)(addr); \
 192                 *(caddr_t *)(addr) = NULL; \
 193                 kbp->kb_last = (caddr_t)(addr); \
 194         } \
 195         splx(s); \
 196 } while (0)
 197
 198 extern struct kmemusage *kmemusage;
 199 extern char *kmembase;
 200 extern struct kmembuckets bucket[];
 201 #endif /* do not collect statistics */
 202
 203 /*
 204  * XXX this should be declared in <sys/uio.h>, but that tends to fail
 205  * because <sys/uio.h> is included in a header before the source file
 206  * has a chance to include <sys/malloc.h> to get MALLOC_DECLARE() defined.
 207  */
 208 MALLOC_DECLARE(M_IOV);
 209
 210 /* XXX struct malloc_type is unused for contig*(). */
 211 void    *contigmalloc __P((unsigned long size, struct malloc_type *type,
 212                            int flags,
 213                            unsigned long low, unsigned long high,
 214                            unsigned long alignment, unsigned long boundary));
 215 void    contigfree __P((void *addr, unsigned long size,
 216                            struct malloc_type *type));
 217 void    free __P((void *addr, struct malloc_type *type));
 218 void    *malloc __P((unsigned long size, struct malloc_type *type, int flags));
 219 #endif /* KERNEL */
 220
 221 #endif /* !_SYS_MALLOC_H_ */