1
0
mirror of https://git.FreeBSD.org/src.git synced 2024-10-19 02:29:40 +00:00

buf_ring: Make buf_ring.h amenable to userspace compilation

This will be useful for adding test cases.

Reviewed by:	andrew
MFC after:	1 week
Differential Revision:	https://reviews.freebsd.org/D45869

(cherry picked from commit a161269b24)
This commit is contained in:
Mark Johnston 2024-07-11 14:55:11 -04:00 committed by Andrew Turner
parent 556739a53c
commit 292b2f9c5d

View File

@ -30,11 +30,18 @@
#ifndef _SYS_BUF_RING_H_ #ifndef _SYS_BUF_RING_H_
#define _SYS_BUF_RING_H_ #define _SYS_BUF_RING_H_
#include <sys/param.h>
#include <sys/kassert.h>
#include <machine/atomic.h>
#include <machine/cpu.h> #include <machine/cpu.h>
#ifdef DEBUG_BUFRING #ifdef DEBUG_BUFRING
#ifdef _KERNEL
#include <sys/lock.h> #include <sys/lock.h>
#include <sys/mutex.h> #include <sys/mutex.h>
#else
#error "DEBUG_BUFRING is only supported in kernel"
#endif
#endif #endif
struct buf_ring { struct buf_ring {
@ -361,8 +368,36 @@ buf_ring_count(struct buf_ring *br)
& br->br_prod_mask); & br->br_prod_mask);
} }
#ifdef _KERNEL
struct buf_ring *buf_ring_alloc(int count, struct malloc_type *type, int flags, struct buf_ring *buf_ring_alloc(int count, struct malloc_type *type, int flags,
struct mtx *); struct mtx *);
void buf_ring_free(struct buf_ring *br, struct malloc_type *type); void buf_ring_free(struct buf_ring *br, struct malloc_type *type);
#else
#endif #include <stdlib.h>
static inline struct buf_ring *
buf_ring_alloc(int count)
{
struct buf_ring *br;
KASSERT(powerof2(count), ("buf ring must be size power of 2"));
br = calloc(1, sizeof(struct buf_ring) + count * sizeof(void *));
if (br == NULL)
return (NULL);
br->br_prod_size = br->br_cons_size = count;
br->br_prod_mask = br->br_cons_mask = count - 1;
br->br_prod_head = br->br_cons_head = 0;
br->br_prod_tail = br->br_cons_tail = 0;
return (br);
}
static inline void
buf_ring_free(struct buf_ring *br)
{
free(br);
}
#endif /* !_KERNEL */
#endif /* _SYS_BUF_RING_H_ */