summaryrefslogtreecommitdiff
path: root/util/sbase/libutil/parseoffset.c
blob: 362a7829491459286de4625bc8d23edf472339e6 (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
/* See LICENSE file for copyright and license details. */
#include <ctype.h>
#include <errno.h>
#include <inttypes.h>
#include <stdlib.h>
#include <string.h>

#include "../util.h"

off_t
parseoffset(const char *str)
{
	off_t res, scale = 1;
	char *end;

	/* strictly check what strtol() usually would let pass */
	if (!str || !*str || isspace(*str) || *str == '+' || *str == '-') {
		weprintf("parseoffset %s: invalid value\n", str);
		return -1;
	}

	errno = 0;
	res = strtol(str, &end, 0);
	if (errno) {
		weprintf("parseoffset %s: invalid value\n", str);
		return -1;
	}
	if (res < 0) {
		weprintf("parseoffset %s: negative value\n", str);
		return -1;
	}

	/* suffix */
	if (*end) {
		switch (toupper((int)*end)) {
		case 'B':
			scale = 512L;
			break;
		case 'K':
			scale = 1024L;
			break;
		case 'M':
			scale = 1024L * 1024L;
			break;
		case 'G':
			scale = 1024L * 1024L * 1024L;
			break;
		default:
			weprintf("parseoffset %s: invalid suffix '%s'\n", str, end);
			return -1;
		}
	}

	/* prevent overflow */
	if (res > (SSIZE_MAX / scale)) {
		weprintf("parseoffset %s: out of range\n", str);
		return -1;
	}

	return res * scale;
}