summaryrefslogtreecommitdiff
path: root/klm/util/cat_compressed_main.cc
blob: 2b4d729277bec28554414e258cc73fc448e2c1c6 (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
// Like cat but interprets compressed files.
#include "util/file.hh"
#include "util/read_compressed.hh"

#include <string.h>
#include <iostream>

namespace {
const std::size_t kBufSize = 16384;
void Copy(util::ReadCompressed &from, int to) {
  util::scoped_malloc buffer(util::MallocOrThrow(kBufSize));
  while (std::size_t amount = from.Read(buffer.get(), kBufSize)) {
    util::WriteOrThrow(to, buffer.get(), amount);
  }
}
} // namespace

int main(int argc, char *argv[]) {
  // Lane Schwartz likes -h and --help
  for (int i = 1; i < argc; ++i) {
    char *arg = argv[i];
    if (!strcmp(arg, "--")) break;
    if (!strcmp(arg, "-h") || !strcmp(arg, "--help")) {
      std::cerr << 
        "A cat implementation that interprets compressed files.\n"
        "Usage: " << argv[0] << " [file1] [file2] ...\n"
        "If no file is provided, then stdin is read.\n";
      return 1;
    }
  }

  try {
    if (argc == 1) {
      util::ReadCompressed in(0);
      Copy(in, 1);
    } else {
      for (int i = 1; i < argc; ++i) {
        util::ReadCompressed in(util::OpenReadOrThrow(argv[i]));
        Copy(in, 1);
      }
    }
  } catch (const std::exception &e) {
    std::cerr << e.what() << std::endl;
    return 2;
  }
  return 0;
}