1 /*-
2  * SPDX-License-Identifier: BSD-2-Clause-FreeBSD
3  *
4  * Copyright (c) 2000-2008 Poul-Henning Kamp
5  * Copyright (c) 2000-2008 Dag-Erling Coïdan Smørgrav
6  * All rights reserved.
7  *
8  * Redistribution and use in source and binary forms, with or without
9  * modification, are permitted provided that the following conditions
10  * are met:
11  * 1. Redistributions of source code must retain the above copyright
12  *    notice, this list of conditions and the following disclaimer
13  *    in this position and unchanged.
14  * 2. Redistributions in binary form must reproduce the above copyright
15  *    notice, this list of conditions and the following disclaimer in the
16  *    documentation and/or other materials provided with the distribution.
17  *
18  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
19  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
20  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
21  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
22  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
23  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
24  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
25  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
26  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
27  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
28  * SUCH DAMAGE.
29  *
30  *      $FreeBSD: stable/12/sys/sys/sbuf.h 370145 2021-07-16 17:47:40Z imp $
31  */
32 
33 #ifndef _SYS_SBUF_H_
34 #define	_SYS_SBUF_H_
35 
36 #include <sys/_types.h>
37 
38 struct sbuf;
39 typedef int (sbuf_drain_func)(void *, const char *, int);
40 
41 /*
42  * Structure definition
43  */
44 struct sbuf {
45 	char		*s_buf;		/* storage buffer */
46 	sbuf_drain_func	*s_drain_func;	/* drain function */
47 	void		*s_drain_arg;	/* user-supplied drain argument */
48 	int		 s_error;	/* current error code */
49 	ssize_t		 s_size;	/* size of storage buffer */
50 	ssize_t		 s_len;		/* current length of string */
51 #define	SBUF_FIXEDLEN	0x00000000	/* fixed length buffer (default) */
52 #define	SBUF_AUTOEXTEND	0x00000001	/* automatically extend buffer */
53 #define	SBUF_INCLUDENUL	0x00000002	/* nulterm byte is counted in len */
54 #define	SBUF_DRAINTOEOR	0x00000004	/* use section 0 as drain EOR marker */
55 #define	SBUF_NOWAIT	0x00000008	/* Extend with non-blocking malloc */
56 #define	SBUF_USRFLAGMSK	0x0000ffff	/* mask of flags the user may specify */
57 #define	SBUF_DYNAMIC	0x00010000	/* s_buf must be freed */
58 #define	SBUF_FINISHED	0x00020000	/* set by sbuf_finish() */
59 #define	SBUF_DYNSTRUCT	0x00080000	/* sbuf must be freed */
60 #define	SBUF_INSECTION	0x00100000	/* set by sbuf_start_section() */
61 	int		 s_flags;	/* flags */
62 	ssize_t		 s_sect_len;	/* current length of section */
63 	ssize_t		 s_rec_off;	/* current record start offset */
64 };
65 
66 #ifndef HD_COLUMN_MASK
67 #define	HD_COLUMN_MASK	0xff
68 #define	HD_DELIM_MASK	0xff00
69 #define	HD_OMIT_COUNT	(1 << 16)
70 #define	HD_OMIT_HEX	(1 << 17)
71 #define	HD_OMIT_CHARS	(1 << 18)
72 #endif /* HD_COLUMN_MASK */
73 
74 __BEGIN_DECLS
75 /*
76  * API functions
77  */
78 struct sbuf	*sbuf_new(struct sbuf *, char *, int, int);
79 #define		 sbuf_new_auto()				\
80 	sbuf_new(NULL, NULL, 0, SBUF_AUTOEXTEND)
81 int		 sbuf_get_flags(struct sbuf *);
82 void		 sbuf_clear_flags(struct sbuf *, int);
83 void		 sbuf_set_flags(struct sbuf *, int);
84 void		 sbuf_clear(struct sbuf *);
85 int		 sbuf_setpos(struct sbuf *, ssize_t);
86 int		 sbuf_bcat(struct sbuf *, const void *, size_t);
87 int		 sbuf_bcpy(struct sbuf *, const void *, size_t);
88 int		 sbuf_cat(struct sbuf *, const char *);
89 int		 sbuf_cpy(struct sbuf *, const char *);
90 int		 sbuf_printf(struct sbuf *, const char *, ...)
91 	__printflike(2, 3);
92 int		 sbuf_vprintf(struct sbuf *, const char *, __va_list)
93 	__printflike(2, 0);
94 int		 sbuf_putc(struct sbuf *, int);
95 void		 sbuf_set_drain(struct sbuf *, sbuf_drain_func *, void *);
96 int		 sbuf_drain(struct sbuf *);
97 int		 sbuf_trim(struct sbuf *);
98 int		 sbuf_error(const struct sbuf *);
99 int		 sbuf_finish(struct sbuf *);
100 char		*sbuf_data(struct sbuf *);
101 ssize_t		 sbuf_len(struct sbuf *);
102 int		 sbuf_done(const struct sbuf *);
103 void		 sbuf_delete(struct sbuf *);
104 void		 sbuf_start_section(struct sbuf *, ssize_t *);
105 ssize_t		 sbuf_end_section(struct sbuf *, ssize_t, size_t, int);
106 void		 sbuf_hexdump(struct sbuf *, const void *, int, const char *,
107 		     int);
108 int		 sbuf_count_drain(void *arg, const char *data, int len);
109 void		 sbuf_putbuf(struct sbuf *);
110 
111 #ifdef _KERNEL
112 struct uio;
113 struct sbuf	*sbuf_uionew(struct sbuf *, struct uio *, int *);
114 int		 sbuf_bcopyin(struct sbuf *, const void *, size_t);
115 int		 sbuf_copyin(struct sbuf *, const void *, size_t);
116 #endif
117 __END_DECLS
118 
119 #endif
120