xref: /src/common/lib/libutil/snprintb.c

/*	$NetBSD: snprintb.c,v 1.26 2024/02/16 01:57:50 rillig Exp $	*/

/*-
 * Copyright (c) 2002 The NetBSD Foundation, Inc.
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
 * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
 * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
 * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 */

/*
 * snprintb: print an interpreted bitmask to a buffer
 *
 * => returns the length of the buffer that would be required to print the
 *    string minus the terminating NUL.
 */
#ifndef _STANDALONE
# ifndef _KERNEL

#  if HAVE_NBTOOL_CONFIG_H
#   include "nbtool_config.h"
#  endif

#  include <sys/cdefs.h>
#  if defined(LIBC_SCCS) && !defined(lint)
__RCSID("$NetBSD: snprintb.c,v 1.26 2024/02/16 01:57:50 rillig Exp $");
#  endif

#  include <sys/types.h>
#  include <inttypes.h>
#  include <stdio.h>
#  include <util.h>
#  include <errno.h>
# else /* ! _KERNEL */
#  include <sys/cdefs.h>
__KERNEL_RCSID(0, "$NetBSD: snprintb.c,v 1.26 2024/02/16 01:57:50 rillig Exp $");
#  include <sys/param.h>
#  include <sys/inttypes.h>
#  include <sys/systm.h>
#  include <lib/libkern/libkern.h>
# endif /* ! _KERNEL */

# ifndef HAVE_SNPRINTB_M
int
snprintb_m(char *buf, size_t bufsize, const char *bitfmt, uint64_t val,
	   size_t l_max)
{
	char *bp = buf, *s_bp = NULL;
	const char *c_fmt, *s_fmt = NULL, *cur_fmt;
	const char *sbase;
	int bit, ch, t_len, s_len = 0, l_len, f_len, v_len, sep;
	int restart = 0, matched = 1;
	uint64_t field;

#ifdef _KERNEL
	/*
	 * For safety; no other *s*printf() do this, but in the kernel
	 * we don't usually check the return value
	 */
	(void)memset(buf, 0, bufsize);
#endif /* _KERNEL */

	ch = *bitfmt++;
	switch (ch != '\177' ? ch : *bitfmt++) {
	case 8:
		sbase = "%#jo";
		break;
	case 10:
		sbase = "%ju";
		break;
	case 16:
		sbase = "%#jx";
		break;
	default:
		goto internal;
	}

	/* Reserve space for trailing blank line if needed */
	if (l_max > 0)
		bufsize--;

	t_len = snprintf(bp, bufsize, sbase, (uintmax_t)val);
	if (t_len < 0)
		goto internal;

	v_len = l_len = t_len;

	if ((size_t)t_len < bufsize)
		bp += t_len;
	else
		bp += bufsize - 1;

	/*
	 * If the value we printed was 0 and we're using the old-style format,
	 * we're done.
	 */
	if (val == 0 && ch != '\177')
		goto terminate;

#define	STORE(c) do {							\
		l_len++;						\
		if ((size_t)(++t_len) < bufsize)			\
			*bp++ = (c);					\
	} while (0)

#define	BACKUP() do {							\
		if (s_bp != NULL) {					\
			bp = s_bp;					\
			s_bp = NULL;					\
			t_len -= l_len - s_len;				\
			restart = 1;					\
			bitfmt = s_fmt;					\
		}							\
		STORE('>');						\
		STORE('\0');						\
		if ((size_t)t_len < bufsize)				\
			snprintf(bp, bufsize - t_len, sbase, (uintmax_t)val);\
		t_len += v_len;						\
		l_len = v_len;						\
		bp += v_len;						\
	} while (0)

#define	PUTSEP() do {							\
		if (l_max > 0 && (size_t)l_len >= l_max) {		\
			BACKUP();					\
			STORE('<');					\
		} else {						\
			/* Remember separator location */		\
			if (l_max > 0 && sep != '<') {			\
				s_len = l_len;				\
				s_bp  = bp;				\
				s_fmt = cur_fmt;			\
			}						\
			STORE(sep);					\
			restart = 0;					\
		}							\
	} while (0)

#define	PUTCHR(c) do {							\
		if (l_max > 0 && (size_t)l_len >= l_max - 1) {		\
			BACKUP();					\
			if (restart == 0)				\
				STORE(c);				\
			else						\
				sep = '<';				\
		} else {						\
			STORE(c);					\
			restart = 0;					\
		}							\
	} while (0)

#define	PUTS(s) do {							\
		while ((ch = *(s)++) != 0) {				\
			PUTCHR(ch);					\
			if (restart)					\
				break;					\
		}							\
	} while (0)

#define	FMTSTR(sb, f) do {						\
		f_len = snprintf(bp, bufsize - t_len, sb, (uintmax_t)f); \
		if (f_len < 0)						\
			goto internal;					\
		t_len += f_len;						\
		l_len += f_len;						\
		if ((size_t)t_len < bufsize)				\
			bp += f_len;					\
	} while (0)

	/*
	 * Chris Torek's new bitmask format is identified by a leading \177
	 */
	sep = '<';
	if (ch != '\177') {
		/* old (standard) format. */
		while ((bit = *bitfmt) != 0) {
			cur_fmt = bitfmt++;
			if (val & (1U << (bit - 1))) {
				PUTSEP();
				if (restart)
					continue;
				sep = ',';
				for (; (ch = *bitfmt) > ' '; ++bitfmt) {
					PUTCHR(ch);
					if (restart)
						break;
				}
			} else
				for (; *bitfmt > ' '; ++bitfmt)
					continue;
		}
	} else {
		/* new quad-capable format; also does fields. */
		field = val;
		while (c_fmt = bitfmt, (ch = *bitfmt++) != '\0') {
			bit = *bitfmt++;	/* now 0-origin */
			switch (ch) {
			case 'b':
				if (((val >> bit) & 1) == 0)
					goto skip;
				cur_fmt = c_fmt;
				PUTSEP();
				if (restart)
					break;
				PUTS(bitfmt);
				if (restart == 0)
					sep = ',';
				break;
			case 'f':
			case 'F':
				matched = 0;
				cur_fmt = c_fmt;
				f_len = *bitfmt++;	/* field length */
				field = (val >> bit) &
				    (((uint64_t)1 << f_len) - 1);
				PUTSEP();
				if (restart == 0)
					sep = ',';
				if (ch == 'F') {	/* just extract */
					/* duplicate PUTS() effect on bitfmt */
					while (*bitfmt++ != '\0')
						continue;
					break;
				}
				if (restart == 0)
					PUTS(bitfmt);
				if (restart == 0)
					PUTCHR('=');
				if (restart == 0) {
					FMTSTR(sbase, field);
					if (l_max > 0 && (size_t)l_len > l_max)
						PUTCHR('#');
				}
				break;
			case '=':
			case ':':
				/*
				 * Here "bit" is actually a value instead,
				 * to be compared against the last field.
				 * This only works for values in [0..255],
				 * of course.
				 */
				if ((int)field != bit)
					goto skip;
				matched = 1;
				if (ch == '=')
					PUTCHR('=');
				PUTS(bitfmt);
				break;
			case '*':
				bitfmt--;
				if (!matched) {
					matched = 1;
					FMTSTR(bitfmt, field);
				}
				/*FALLTHROUGH*/
			default:
			skip:
				while (*bitfmt++ != '\0')
					continue;
				break;
			}
		}
	}
	if (sep != '<')
		STORE('>');
terminate:
	if (l_max > 0) {
		bufsize++;
		STORE('\0');
		if ((size_t)t_len >= bufsize && bufsize > 1)
			buf[bufsize - 2] = '\0';
	}
	STORE('\0');
	if ((size_t)t_len >= bufsize && bufsize > 0)
		buf[bufsize - 1] = '\0';
	return t_len - 1;
internal:
#ifndef _KERNEL
	errno = EINVAL;
#endif
	return -1;
}

int
snprintb(char *buf, size_t bufsize, const char *bitfmt, uint64_t val)
{
	return snprintb_m(buf, bufsize, bitfmt, val, 0);
}
# endif /* ! HAVE_SNPRINTB_M */
#endif /* ! _STANDALONE */