about summary refs log tree commit diff
path: root/holes.c
blob: 41fc50cf10ed8c6de9a29c998a3e731daba31f43 (plain) (blame)
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
104
105
106
107
108
109
110
111
/* holes - find holes (= runs of zero bytes) in input files */

#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>

ssize_t minlen = 64;
char byte = 0;
int ret = 0;
char *argv0;

void
holes(FILE *input, char *filename)
{
	off_t offset = 0;
	off_t run = 0;	
	int ch;

	while ((ch = getc_unlocked(input)) != EOF) {
		if (ch == byte) {
			run++;
		} else {
			if (run >= minlen) {
				if (filename)
					printf("%s: ", filename);
				printf("%08lx %ld\n", offset - run, run);
			}
			run = 0;
		}
		offset++;
	}
	if (ferror(input)) {
		fprintf(stderr, "%s: can't read '%s': %s\n",
		    argv0, filename ? filename : "-", strerror(errno));
		ret = 1;
		return;
	}

	if (offset == 0 ||	// empty file
	    run >= minlen) {
		if (filename)
			printf("%s: ", filename);
		printf("%08lx %ld\n", offset - run, run);
	}
}

int
main(int argc, char *argv[])
{
	int c, i;
	long b;
	char *e;

	argv0 = argv[0];

	while ((c = getopt(argc, argv, "b:n:")) != -1)
		switch(c) {
		case 'b':
			errno = 0;
			b = strtol(optarg, &e, 0);
			if (errno != 0 || *e || b < 0 || b > 256) {
				fprintf(stderr,
				    "%s: can't parse byte '%s'.\n",
				    argv0, optarg);
				exit(2);
			}
			byte = b;
			break;
                case 'n':
			errno = 0;
			minlen = strtoll(optarg, &e, 0);
			if (errno != 0 || *e) {
				fprintf(stderr,
				    "%s: can't parse length '%s'.\n",
				    argv0, optarg);
				exit(2);
			}
			if (minlen < 1) {
				fprintf(stderr,
				    "%s: MINLEN must not be smaller than 1.\n",
				    argv0);
				exit(2);
			}
			break;
                default:
                        fprintf(stderr,
			    "Usage: %s [-b BYTE] [-n MINLEN] [FILES...]\n",
			    argv0);
			exit(2);
		}

	if (optind == argc)
		holes(stdin, 0);
	else
		for (i = optind; i < argc; i++) {
			FILE *input = (strcmp(argv[i], "-") == 0) ?
			    stdin : fopen(argv[i], "rb");
			
			if (!input) {
				fprintf(stderr, "%s: can't open file '%s': %s\n",
				    argv0, argv[i], strerror(errno));
				ret = 1;
			} else {
				holes(input, argc - optind > 1 ? argv[i] : 0);
			}
		}

	return ret;
}