• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
1 /*
2  * parse_num.c		- Parse the number of blocks
3  *
4  * Copyright (C) 2004,2005  Theodore Ts'o <tytso@mit.edu>
5  *
6  * %Begin-Header%
7  * This file may be redistributed under the terms of the GNU Library
8  * General Public License, version 2.
9  * %End-Header%
10  */
11 
12 #include "config.h"
13 #include "e2p.h"
14 
15 #include <stdlib.h>
16 
parse_num_blocks2(const char * arg,int log_block_size)17 unsigned long long parse_num_blocks2(const char *arg, int log_block_size)
18 {
19 	char *p;
20 	unsigned long long num;
21 
22 	num = strtoull(arg, &p, 0);
23 
24 	if (p[0] && p[1])
25 		return 0;
26 
27 	switch (*p) {		/* Using fall-through logic */
28 	case 'T': case 't':
29 		num <<= 10;
30 		/* fallthrough */
31 	case 'G': case 'g':
32 		num <<= 10;
33 		/* fallthrough */
34 	case 'M': case 'm':
35 		num <<= 10;
36 		/* fallthrough */
37 	case 'K': case 'k':
38 		if (log_block_size < 0)
39 			num <<= 10;
40 		else
41 			num >>= log_block_size;
42 		break;
43 	case 's':
44 		if (log_block_size < 0)
45 			num <<= 9;
46 		else
47 			num >>= (1+log_block_size);
48 		break;
49 	case '\0':
50 		break;
51 	default:
52 		return 0;
53 	}
54 	return num;
55 }
56 
parse_num_blocks(const char * arg,int log_block_size)57 unsigned long parse_num_blocks(const char *arg, int log_block_size)
58 {
59 	return parse_num_blocks2(arg, log_block_size);
60 }
61 
62 #ifdef DEBUG
63 #include <unistd.h>
64 #include <stdio.h>
65 
main(int argc,char ** argv)66 main(int argc, char **argv)
67 {
68 	unsigned long num;
69 	int log_block_size = 0;
70 
71 	if (argc != 2 && argc != 3) {
72 		fprintf(stderr, "Usage: %s arg [log_block_size]\n", argv[0]);
73 		exit(1);
74 	}
75 
76 	if (argc == 3) {
77 		char *p;
78 
79 		log_block_size = strtol(argv[2], &p, 0);
80 		if (*p) {
81 			fprintf(stderr, "Bad log_block_size: %s\n", argv[2]);
82 			exit(1);
83 		}
84 	}
85 
86 	num = parse_num_blocks(argv[1], log_block_size);
87 
88 	printf("Parsed number: %lu\n", num);
89 	exit(0);
90 }
91 #endif
92