1*01826a49SYabin Cui /*
2*01826a49SYabin Cui * Copyright (c) Meta Platforms, Inc. and affiliates.
3*01826a49SYabin Cui * All rights reserved.
4*01826a49SYabin Cui *
5*01826a49SYabin Cui * This source code is licensed under both the BSD-style license (found in the
6*01826a49SYabin Cui * LICENSE file in the root directory of this source tree) and the GPLv2 (found
7*01826a49SYabin Cui * in the COPYING file in the root directory of this source tree).
8*01826a49SYabin Cui */
9*01826a49SYabin Cui #pragma once
10*01826a49SYabin Cui
11*01826a49SYabin Cui #include "Options.h"
12*01826a49SYabin Cui #include "Pzstd.h"
13*01826a49SYabin Cui #include "utils/ScopeGuard.h"
14*01826a49SYabin Cui
15*01826a49SYabin Cui #include <cstdio>
16*01826a49SYabin Cui #include <string>
17*01826a49SYabin Cui #include <cstdint>
18*01826a49SYabin Cui #include <memory>
19*01826a49SYabin Cui
20*01826a49SYabin Cui namespace pzstd {
21*01826a49SYabin Cui
check(std::string source,std::string decompressed)22*01826a49SYabin Cui inline bool check(std::string source, std::string decompressed) {
23*01826a49SYabin Cui std::unique_ptr<std::uint8_t[]> sBuf(new std::uint8_t[1024]);
24*01826a49SYabin Cui std::unique_ptr<std::uint8_t[]> dBuf(new std::uint8_t[1024]);
25*01826a49SYabin Cui
26*01826a49SYabin Cui auto sFd = std::fopen(source.c_str(), "rb");
27*01826a49SYabin Cui auto dFd = std::fopen(decompressed.c_str(), "rb");
28*01826a49SYabin Cui auto guard = makeScopeGuard([&] {
29*01826a49SYabin Cui std::fclose(sFd);
30*01826a49SYabin Cui std::fclose(dFd);
31*01826a49SYabin Cui });
32*01826a49SYabin Cui
33*01826a49SYabin Cui size_t sRead, dRead;
34*01826a49SYabin Cui
35*01826a49SYabin Cui do {
36*01826a49SYabin Cui sRead = std::fread(sBuf.get(), 1, 1024, sFd);
37*01826a49SYabin Cui dRead = std::fread(dBuf.get(), 1, 1024, dFd);
38*01826a49SYabin Cui if (std::ferror(sFd) || std::ferror(dFd)) {
39*01826a49SYabin Cui return false;
40*01826a49SYabin Cui }
41*01826a49SYabin Cui if (sRead != dRead) {
42*01826a49SYabin Cui return false;
43*01826a49SYabin Cui }
44*01826a49SYabin Cui
45*01826a49SYabin Cui for (size_t i = 0; i < sRead; ++i) {
46*01826a49SYabin Cui if (sBuf.get()[i] != dBuf.get()[i]) {
47*01826a49SYabin Cui return false;
48*01826a49SYabin Cui }
49*01826a49SYabin Cui }
50*01826a49SYabin Cui } while (sRead == 1024);
51*01826a49SYabin Cui if (!std::feof(sFd) || !std::feof(dFd)) {
52*01826a49SYabin Cui return false;
53*01826a49SYabin Cui }
54*01826a49SYabin Cui return true;
55*01826a49SYabin Cui }
56*01826a49SYabin Cui
roundTrip(Options & options)57*01826a49SYabin Cui inline bool roundTrip(Options& options) {
58*01826a49SYabin Cui if (options.inputFiles.size() != 1) {
59*01826a49SYabin Cui return false;
60*01826a49SYabin Cui }
61*01826a49SYabin Cui std::string source = options.inputFiles.front();
62*01826a49SYabin Cui std::string compressedFile = std::tmpnam(nullptr);
63*01826a49SYabin Cui std::string decompressedFile = std::tmpnam(nullptr);
64*01826a49SYabin Cui auto guard = makeScopeGuard([&] {
65*01826a49SYabin Cui std::remove(compressedFile.c_str());
66*01826a49SYabin Cui std::remove(decompressedFile.c_str());
67*01826a49SYabin Cui });
68*01826a49SYabin Cui
69*01826a49SYabin Cui {
70*01826a49SYabin Cui options.outputFile = compressedFile;
71*01826a49SYabin Cui options.decompress = false;
72*01826a49SYabin Cui if (pzstdMain(options) != 0) {
73*01826a49SYabin Cui return false;
74*01826a49SYabin Cui }
75*01826a49SYabin Cui }
76*01826a49SYabin Cui {
77*01826a49SYabin Cui options.decompress = true;
78*01826a49SYabin Cui options.inputFiles.front() = compressedFile;
79*01826a49SYabin Cui options.outputFile = decompressedFile;
80*01826a49SYabin Cui if (pzstdMain(options) != 0) {
81*01826a49SYabin Cui return false;
82*01826a49SYabin Cui }
83*01826a49SYabin Cui }
84*01826a49SYabin Cui return check(source, decompressedFile);
85*01826a49SYabin Cui }
86*01826a49SYabin Cui }
87