src/lib/file.c


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

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

#include "file.h"
#include "util.h"

struct filedata readfile(FILE *fp) {
  struct filedata f;

  f.lc = 0;
  f.len = 0;
  f.binary = 0;

  size_t bufsize = 4;
  f.buf = malloc(bufsize);
  if (f.buf == NULL)
    die("malloc");

  char c;
  while (fread(&c, sizeof(char), 1, fp) > 0) {
    if (f.len == bufsize - 1) {
      bufsize *= 2;

      char *new_buf = realloc(f.buf, bufsize);
      if (f.buf == NULL) {
        free(f.buf);
        die("realloc");
      }

      f.buf = new_buf;
    }

    if (c == '\n') {
      f.lc++;
    }

    f.buf[f.len++] = c;
  }

  // guess if printable
  // from https://github.com/sharkdp/content_inspector/blob/master/src/lib.rs
  int testlen = f.len >= 64 ? 64 : f.len;
  char *testbuf[testlen];
  memcpy(testbuf, f.buf, testlen);

  char *result = memchr(testbuf, 0x00, testlen);

  if (result) {
    f.binary = 1;
  } else {
    f.binary = 0;
  }

  return f;
}