nasm/nasmlib/numstr.c
H. Peter Anvin (Intel) 6251fc5908 nasmlib/numstr.c: check for buflen == 0
buflen == 0 is an error, but be paranoid about it; the code already
checks for a valid base so it is better to check for a valid buffer
size as well.

Signed-off-by: H. Peter Anvin (Intel) <hpa@zytor.com>
2026-04-07 23:09:21 -07:00

52 lines
1.2 KiB
C

/* SPDX-License-Identifier: BSD-2-Clause */
/* Copyright 2023-2025 The NASM Authors - All Rights Reserved */
#include "nasmlib.h"
const char * const nasmlib_digit_chars[2] = {
/* Lower case version */
"0123456789"
"abcdefghijklmnopqrstuvwxyz"
"ABCDEFGHIJKLMNOPQRSTUVWXYZ"
"@_",
/* Upper case version */
"0123456789"
"ABCDEFGHIJKLMNOPQRSTUVWXYZ"
"abcdefghijklmnopqrstuvwxyz"
"@_"
};
/*
* Produce an unsigned integer string from a number with a specified
* base, digits and signedness.
*/
int numstr(char *buf, size_t buflen, uint64_t n,
int digits, unsigned int base, bool ucase)
{
const char * const dchars = nasm_digit_chars(ucase);
bool moredigits = digits <= 0;
char *p;
int len;
if (unlikely(base < 2) || unlikely(base > NUMSTR_MAXBASE) ||
unlikely(!buflen))
return -1;
if (moredigits)
digits = -digits;
p = buf + buflen;
*--p = '\0';
while (p > buf && (digits-- > 0 || (moredigits && n))) {
*--p = dchars[n % base];
n /= base;
}
len = buflen - (p - buf); /* Including final null */
if (p != buf)
memmove(buf, p, len);
return len - 1;
}