--- /dev/null
+/* $OpenBSD: dma_alloc.c,v 1.1 2010/07/13 16:47:03 deraadt Exp $ */
+/*
+ * Copyright (c) 2010 Theo de Raadt <deraadt@openbsd.org>
+ *
+ * Permission to use, copy, modify, and distribute this software for any
+ * purpose with or without fee is hereby granted, provided that the above
+ * copyright notice and this permission notice appear in all copies.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
+ * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
+ * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
+ * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
+ * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
+ * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
+ * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
+ */
+
+#include <sys/param.h>
+#include <sys/pool.h>
+#include <uvm/uvm.h>
+
+static __inline int dma_alloc_index(size_t size);
+
+#define DMA_BUCKET_OFFSET 4
+static char dmanames[PAGE_SHIFT - DMA_BUCKET_OFFSET][8];
+struct pool dmapools[PAGE_SHIFT - DMA_BUCKET_OFFSET];
+
+void
+dma_alloc_init(void)
+{
+ int i;
+
+ for (i = 0; i < nitems(dmapools); i++) {
+ snprintf(dmanames[i], sizeof(dmanames[0]), "dma%d",
+ 1 << (i + DMA_BUCKET_OFFSET));
+ pool_init(&dmapools[i], 1 << (i + DMA_BUCKET_OFFSET), 0, 0, 0,
+ dmanames[i], NULL);
+ pool_set_constraints(&dmapools[i], &dma_constraint, 1);
+ /* XXX need pool_setlowat(&dmapools[i], dmalowat); */
+ }
+}
+
+static __inline int
+dma_alloc_index(size_t sz)
+{
+ int b;
+
+ for (b = 0; b < nitems(dmapools); b++)
+ if (sz <= (1 << (b + DMA_BUCKET_OFFSET)))
+ return (b);
+#ifdef DEBUG
+ printf("dma_alloc/free: object %d too large\n", sz)
+#endif
+ return (-1);
+}
+
+void *
+dma_alloc(size_t size, int prflags)
+{
+ int pi = dma_alloc_index(size);
+
+ if (pi == -1)
+ return (NULL);
+ return pool_get(&dmapools[pi], prflags);
+}
+
+
+void
+dma_free(void *m, size_t size)
+{
+ int pi = dma_alloc_index(size);
+
+ if (pi == -1)
+ return;
+ pool_put(&dmapools[pi], m);
+}
-/* $OpenBSD: pool.h,v 1.34 2010/06/27 03:03:48 thib Exp $ */
+/* $OpenBSD: pool.h,v 1.35 2010/07/13 16:47:02 deraadt Exp $ */
/* $NetBSD: pool.h,v 1.27 2001/06/06 22:00:17 rafal Exp $ */
/*-
void pool_walk(struct pool *, int, int (*)(const char *, ...),
void (*)(void *, int, int (*)(const char *, ...)));
#endif
+
+/* the allocator for dma-able memory is a thin layer on top of pool */
+void dma_alloc_init(void);
+void *dma_alloc(size_t size, int flags);
+void dma_free(void *m, size_t size);
#endif /* _KERNEL */
#endif /* _SYS_POOL_H_ */