summaryrefslogtreecommitdiff
path: root/util/nvmutil/lib/num.c
blob: dd6e9901f059fe3e6abe58fc787e6d471f96ada6 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
/* SPDX-License-Identifier: MIT
 *
 * Copyright (c) 2026 Leah Rowe <leah@libreboot.org>
 *
 * Numerical functions.
 */

#ifdef __OpenBSD__
#include <sys/param.h>
#endif
#include <sys/types.h>
#include <sys/stat.h>

#include <errno.h>
#include <fcntl.h>
#include <limits.h>
#include <stdarg.h>
#include <stddef.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
#include <unistd.h>

#include "../include/common.h"

unsigned short
hextonum(char ch_s)
{
	unsigned char ch;

	ch = (unsigned char)ch_s;

	if ((unsigned int)(ch - '0') <= 9)
		return ch - '0';

	ch |= 0x20;

	if ((unsigned int)(ch - 'a') <= 5)
		return ch - 'a' + 10;

	if (ch == '?' || ch == 'x')
		return (unsigned short)rlong() & 0xf;

	return 16; /* invalid character */
}

/*
 * Portable random
 * number generator
 */
unsigned long
rlong(void)
{
#if (defined(__OpenBSD__) && (OpenBSD) >= 201) || \
    defined(__FreeBSD__) || \
    defined(__NetBSD__) || defined(__APPLE__)

	unsigned long rval;
	arc4random_buf(&rval, sizeof(unsigned long));

	return rval;
#else
	int fd;

	long nr;

	unsigned long rval;

	fd = open("/dev/urandom", O_RDONLY | O_BINARY);

#ifdef __OpenBSD__
	if (fd < 0) /* old openbsd */
		fd = open("/dev/arandom", O_RDONLY | O_BINARY);
#endif

	if (fd < 0)
		fd = open("/dev/random", O_RDONLY | O_BINARY);

	if (fd < 0)
		err(errno, "can't open random device");

	nr = rw_file_exact(fd, (unsigned char *)&rval,
	    sizeof(unsigned long), 0, IO_READ, LOOP_EAGAIN,
	    LOOP_EINTR, MAX_ZERO_RW_RETRY, OFF_ERR);

	if (x_i_close(fd) < 0)
		err(errno, "Can't close randomness fd");

	if (nr != sizeof(unsigned long))
		err(errno, "Incomplete read from random device");

	return rval;
#endif
}

void
check_bin(unsigned long a, const char *a_name)
{
	if (a > 1)
		err(EINVAL, "%s must be 0 or 1, but is %lu",
		    a_name, (unsigned long)a);
}