mirror of
https://github.com/google/sandboxed-api.git
synced 2024-03-22 13:11:30 +08:00
dbaf95c724
This change should make it less confusing where utility code comes from. Having it in two places made sense when we were debating whether to publish Sandbox2 separately, but not any longer. Follow-up changes will move `sandbox2/util.h` and rename the remaining `sandbox2/util` folder. PiperOrigin-RevId: 351601640 Change-Id: I6256845261f610e590c25e2c59851cc51da2d778
117 lines
3.6 KiB
C++
117 lines
3.6 KiB
C++
// Copyright 2020 Google LLC
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#include <iostream>
|
|
|
|
#include <glog/logging.h>
|
|
#include "helpers.h" // NOLINT(build/include)
|
|
#include "lodepng.h" // NOLINT(build/include)
|
|
#include "sandboxed_api/util/fileops.h"
|
|
#include "sandboxed_api/util/path.h"
|
|
|
|
void EncodeDecodeOneStep(const std::string& images_path) {
|
|
// Generate the values.
|
|
std::vector<uint8_t> image = GenerateValues();
|
|
|
|
// Encode the image.
|
|
const std::string filename =
|
|
sandbox2::file::JoinPath(images_path, "/out_generated1.png");
|
|
unsigned int result =
|
|
lodepng_encode32_file(filename.c_str(), image.data(), kWidth, kHeight);
|
|
|
|
CHECK(!result) << "Unexpected result from encode32_file call";
|
|
|
|
// After the image has been encoded, decode it to check that the
|
|
// pixel values are the same.
|
|
unsigned int width, height;
|
|
uint8_t* image2 = 0;
|
|
|
|
result = lodepng_decode32_file(&image2, &width, &height, filename.c_str());
|
|
|
|
CHECK(!result) << "Unexpected result from decode32_file call";
|
|
|
|
CHECK(width == kWidth) << "Widths differ";
|
|
CHECK(height == kHeight) << "Heights differ";
|
|
|
|
// Now, we can compare the values.
|
|
CHECK(absl::equal(image.begin(), image.end(), image2, image2 + kImgLen))
|
|
<< "Values differ";
|
|
|
|
free(image2);
|
|
}
|
|
|
|
void EncodeDecodeTwoSteps(const std::string& images_path) {
|
|
// Generate the values.
|
|
std::vector<uint8_t> image = GenerateValues();
|
|
|
|
// Encode the image into memory first.
|
|
const std::string filename =
|
|
sandbox2::file::JoinPath(images_path, "/out_generated2.png");
|
|
uint8_t* png;
|
|
size_t pngsize;
|
|
|
|
unsigned int result =
|
|
lodepng_encode32(&png, &pngsize, image.data(), kWidth, kHeight);
|
|
|
|
CHECK(!result) << "Unexpected result from encode32 call";
|
|
|
|
// Write the image into the file (from memory).
|
|
result = lodepng_save_file(png, pngsize, filename.c_str());
|
|
|
|
CHECK(!result) << "Unexpected result from save_file call";
|
|
|
|
// Now, decode the image using the 2 steps in order to compare the values.
|
|
unsigned int width, height;
|
|
uint8_t* png2;
|
|
size_t pngsize2;
|
|
|
|
// Load the file in memory.
|
|
result = lodepng_load_file(&png2, &pngsize2, filename.c_str());
|
|
|
|
CHECK(!result) << "Unexpected result from load_file call";
|
|
CHECK(pngsize == pngsize2) << "Png sizes differ";
|
|
|
|
uint8_t* image2;
|
|
result = lodepng_decode32(&image2, &width, &height, png2, pngsize2);
|
|
|
|
CHECK(!result) << "Unexpected result from decode32 call";
|
|
CHECK(width == kWidth) << "Widths differ";
|
|
CHECK(height == kHeight) << "Heights differ";
|
|
|
|
// Compare the values.
|
|
CHECK(absl::equal(image.begin(), image.end(), image2, image2 + kImgLen))
|
|
<< "Values differ";
|
|
|
|
free(png);
|
|
free(png2);
|
|
free(image2);
|
|
}
|
|
|
|
int main(int argc, char* argv[]) {
|
|
google::InitGoogleLogging(argv[0]);
|
|
|
|
const std::string images_path = CreateTempDirAtCWD();
|
|
CHECK(sandbox2::file_util::fileops::Exists(images_path, false))
|
|
<< "Temporary directory does not exist";
|
|
|
|
EncodeDecodeOneStep(images_path);
|
|
EncodeDecodeTwoSteps(images_path);
|
|
|
|
if (sandbox2::file_util::fileops::DeleteRecursively(images_path)) {
|
|
LOG(WARNING) << "Temporary folder could not be deleted";
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|