1 // Copyright 2017 The Chromium OS Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "puffin/src/sample_generator.h"
6
7 #include <malloc.h>
8 #include <zlib.h>
9
10 #include <fstream>
11 #include <iomanip>
12 #include <iostream>
13 #include <string>
14
15 #include "puffin/src/bit_reader.h"
16 #include "puffin/src/puff_writer.h"
17 #include "puffin/src/set_errors.h"
18
19 namespace puffin {
20 namespace sample_generator {
21
22 using std::cerr;
23 using std::cout;
24 using std::endl;
25 using std::string;
26
CompressToDeflate(const Buffer & uncomp,Buffer * comp,int compression,int strategy)27 bool CompressToDeflate(const Buffer& uncomp,
28 Buffer* comp,
29 int compression,
30 int strategy) {
31 z_stream stream;
32 stream.next_in = (z_const Bytef*)uncomp.data();
33 stream.avail_in = static_cast<unsigned int>(uncomp.size());
34 stream.next_out = comp->data();
35 stream.avail_out = comp->size();
36
37 stream.zalloc = nullptr;
38 stream.zfree = nullptr;
39 stream.opaque = nullptr;
40
41 TEST_AND_RETURN_FALSE(
42 Z_OK == deflateInit2(&stream, compression, Z_DEFLATED, -15, 8, strategy));
43 // If there was not enough output available return error.
44 TEST_AND_RETURN_FALSE(Z_STREAM_END == deflate(&stream, Z_FINISH));
45 TEST_AND_RETURN_FALSE(Z_OK == deflateEnd(&stream));
46 comp->resize(stream.total_out);
47 return true;
48 }
49
50 // Print an array into hex-format to the output. This can be used to create
51 // static arrays for unit testing of the puffer/huffer.
PrintArray(const string & name,const Buffer & array)52 void PrintArray(const string& name, const Buffer& array) {
53 cout << "const Buffer " << name << " = {" << endl << " ";
54 for (size_t idx = 0; idx < array.size(); idx++) {
55 cout << " 0x" << std::hex << std::uppercase << std::setfill('0')
56 << std::setw(2) << uint(array[idx]);
57 if (idx == array.size() - 1) {
58 cout << std::dec << "};" << endl;
59 return;
60 }
61 cout << ",";
62 if ((idx + 1) % 12 == 0) {
63 cout << endl << " ";
64 }
65 }
66 }
67
PrintSample(Puffer * puffer,int compression,int strategy,const Buffer & original)68 bool PrintSample(Puffer* puffer,
69 int compression,
70 int strategy,
71 const Buffer& original) {
72 PrintArray("original", original);
73
74 Buffer comp(original.size() * 4 + 10);
75 TEST_AND_RETURN_FALSE(
76 CompressToDeflate(original, &comp, compression, strategy));
77 PrintArray("compressed", comp);
78
79 Buffer puff(original.size() * 3 + 10);
80 puffin::Error error;
81
82 BufferBitReader bit_reader(comp.data(), comp.size());
83 BufferPuffWriter puff_writer(puff.data(), puff.size());
84 TEST_AND_RETURN_FALSE(
85 puffer->PuffDeflate(&bit_reader, &puff_writer, nullptr, &error));
86 TEST_AND_RETURN_FALSE(comp.size() == bit_reader.Offset());
87
88 puff.resize(puff_writer.Size());
89 PrintArray("puffed", puff);
90 return true;
91 }
92
93 } // namespace sample_generator
94 } // namespace puffin
95