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
|
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "types.h"
#include "util.h"
bool isbinary(struct filedata *f) {
// guess if printable
// from https://github.com/sharkdp/content_inspector/blob/master/src/lib.rs
int testlen = f->buflen >= 64 ? 64 : f->buflen;
char *testbuf[testlen];
memcpy(testbuf, f->buf, testlen);
char *result = memchr(testbuf, 0x00, testlen);
if (result) {
return true;
} else {
return false;
}
}
struct filedata readfile(FILE *fp, bool isstdin) {
struct filedata f;
f.lc = 0;
f.buflen = 0;
f.binary = 0;
f.buf = NULL;
f.lines = NULL;
if (isstdin) {
size_t bufsize = 1024;
f.buf = malloc(bufsize);
if (f.buf == NULL)
die("malloc");
char c;
while (fread(&c, 1, 1, fp) > 0) {
if (f.buflen == bufsize - 1) {
bufsize *= 2;
char *new_buf = realloc(f.buf, bufsize);
if (new_buf == NULL)
die("realloc");
f.buf = new_buf;
}
f.buf[f.buflen++] = c;
}
if (f.buflen < bufsize - 1) {
char *new_buf = realloc(f.buf, f.buflen + 1);
if (new_buf == NULL)
die("realloc");
f.buf = new_buf;
}
f.buf[f.buflen] = '\0';
f.binary = isbinary(&f);
return f;
}
// expects to be at beginning of file
fseek(fp, 0, SEEK_END);
f.buflen = ftell(fp);
fseek(fp, 0, SEEK_SET);
f.buf = malloc(f.buflen);
if (f.buf == NULL)
die("malloc");
if (fread(f.buf, f.buflen, 1, fp) < 0) {
die("fread");
}
f.binary = isbinary(&f);
return f;
}
|
