mirror of
https://github.com/google/sandboxed-api.git
synced 2024-03-22 13:11:30 +08:00
Codestyle update
This commit is contained in:
parent
6f7110dd4f
commit
47fd491e20
|
@ -1,3 +1,17 @@
|
||||||
|
# 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.
|
||||||
|
|
||||||
load(
|
load(
|
||||||
"@com_google_sandboxed_api//sandboxed_api/bazel:sapi.bzl",
|
"@com_google_sandboxed_api//sandboxed_api/bazel:sapi.bzl",
|
||||||
"sapi_library",
|
"sapi_library",
|
||||||
|
|
0
oss-internship-2020/guetzli/README.md
Normal file
0
oss-internship-2020/guetzli/README.md
Normal file
|
@ -1,3 +1,25 @@
|
||||||
|
// 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 <sys/stat.h>
|
||||||
|
|
||||||
|
#include <algorithm>
|
||||||
|
#include <cstdio>
|
||||||
|
#include <iostream>
|
||||||
|
#include <string>
|
||||||
|
#include <vector>
|
||||||
|
|
||||||
#include "guetzli/jpeg_data_reader.h"
|
#include "guetzli/jpeg_data_reader.h"
|
||||||
#include "guetzli/quality.h"
|
#include "guetzli/quality.h"
|
||||||
#include "guetzli_entry_points.h"
|
#include "guetzli_entry_points.h"
|
||||||
|
@ -5,17 +27,10 @@
|
||||||
#include "sandboxed_api/sandbox2/util/fileops.h"
|
#include "sandboxed_api/sandbox2/util/fileops.h"
|
||||||
#include "sandboxed_api/util/statusor.h"
|
#include "sandboxed_api/util/statusor.h"
|
||||||
|
|
||||||
#include <algorithm>
|
|
||||||
#include <iostream>
|
|
||||||
#include <string>
|
|
||||||
#include <stdio.h>
|
|
||||||
#include <sys/stat.h>
|
|
||||||
#include <vector>
|
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
|
|
||||||
constexpr int kBytesPerPixel = 350;
|
constexpr int kBytesPerPixel = 350;
|
||||||
constexpr int kLowestMemusageMB = 100; // in MB
|
constexpr int kLowestMemusageMB = 100;
|
||||||
|
|
||||||
struct GuetzliInitData {
|
struct GuetzliInitData {
|
||||||
std::string in_data;
|
std::string in_data;
|
||||||
|
@ -49,7 +64,6 @@ sapi::StatusOr<std::string> ReadFromFd(int fd) {
|
||||||
status = read(fd, buf.get(), fsize);
|
status = read(fd, buf.get(), fsize);
|
||||||
|
|
||||||
if (status < 0) {
|
if (status < 0) {
|
||||||
lseek(fd, 0, SEEK_SET);
|
|
||||||
return absl::FailedPreconditionError(
|
return absl::FailedPreconditionError(
|
||||||
"Error reading input from fd"
|
"Error reading input from fd"
|
||||||
);
|
);
|
||||||
|
|
|
@ -1,4 +1,19 @@
|
||||||
#pragma once
|
// 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.
|
||||||
|
|
||||||
|
#ifndef GUETZLI_SANDBOXED_GUETZLI_ENTRY_POINTS_H_
|
||||||
|
#define GUETZLI_SANDBOXED_GUETZLI_ENTRY_POINTS_H_
|
||||||
|
|
||||||
#include "guetzli/processor.h"
|
#include "guetzli/processor.h"
|
||||||
#include "sandboxed_api/lenval_core.h"
|
#include "sandboxed_api/lenval_core.h"
|
||||||
|
@ -16,3 +31,5 @@ extern "C" bool ProcessJpeg(const ProcessingParams* processing_params,
|
||||||
extern "C" bool ProcessRgb(const ProcessingParams* processing_params,
|
extern "C" bool ProcessRgb(const ProcessingParams* processing_params,
|
||||||
sapi::LenValStruct* output);
|
sapi::LenValStruct* output);
|
||||||
extern "C" bool WriteDataToFd(int fd, sapi::LenValStruct* data);
|
extern "C" bool WriteDataToFd(int fd, sapi::LenValStruct* data);
|
||||||
|
|
||||||
|
#endif // GUETZLI_SANDBOXED_GUETZLI_ENTRY_POINTS_H_
|
||||||
|
|
|
@ -1,13 +1,29 @@
|
||||||
#pragma once
|
// 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.
|
||||||
|
|
||||||
|
#ifndef GUETZLI_SANDBOXED_GUETZLI_SANDBOX_H_
|
||||||
|
#define GUETZLI_SANDBOXED_GUETZLI_SANDBOX_H_
|
||||||
|
|
||||||
#include <libgen.h>
|
#include <libgen.h>
|
||||||
#include <syscall.h>
|
#include <syscall.h>
|
||||||
|
|
||||||
#include "guetzli_sapi.sapi.h"
|
|
||||||
#include "sandboxed_api/sandbox2/policy.h"
|
#include "sandboxed_api/sandbox2/policy.h"
|
||||||
#include "sandboxed_api/sandbox2/policybuilder.h"
|
#include "sandboxed_api/sandbox2/policybuilder.h"
|
||||||
#include "sandboxed_api/util/flag.h"
|
#include "sandboxed_api/util/flag.h"
|
||||||
|
|
||||||
|
#include "guetzli_sapi.sapi.h"
|
||||||
|
|
||||||
namespace guetzli {
|
namespace guetzli {
|
||||||
namespace sandbox {
|
namespace sandbox {
|
||||||
|
|
||||||
|
@ -34,3 +50,5 @@ class GuetzliSapiSandbox : public GuetzliSandbox {
|
||||||
|
|
||||||
} // namespace sandbox
|
} // namespace sandbox
|
||||||
} // namespace guetzli
|
} // namespace guetzli
|
||||||
|
|
||||||
|
#endif // GUETZLI_SANDBOXED_GUETZLI_SANDBOX_H_
|
||||||
|
|
|
@ -1,19 +1,34 @@
|
||||||
|
// 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 <errno.h>
|
||||||
#include <fcntl.h>
|
#include <fcntl.h>
|
||||||
#include <sys/stat.h>
|
#include <sys/stat.h>
|
||||||
#include <sys/types.h>
|
#include <sys/types.h>
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
#include <errno.h>
|
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
|
||||||
#include "guetzli_transaction.h"
|
|
||||||
#include "sandboxed_api/sandbox2/util/fileops.h"
|
#include "sandboxed_api/sandbox2/util/fileops.h"
|
||||||
#include "sandboxed_api/util/statusor.h"
|
#include "sandboxed_api/util/statusor.h"
|
||||||
|
|
||||||
|
#include "guetzli_transaction.h"
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
|
|
||||||
constexpr int kDefaultJPEGQuality = 95;
|
constexpr int kDefaultJPEGQuality = 95;
|
||||||
constexpr int kDefaultMemlimitMB = 6000; // in MB
|
constexpr int kDefaultMemlimitMB = 6000;
|
||||||
|
|
||||||
void TerminateHandler() {
|
void TerminateHandler() {
|
||||||
fprintf(stderr, "Unhandled exception. Most likely insufficient memory available.\n"
|
fprintf(stderr, "Unhandled exception. Most likely insufficient memory available.\n"
|
||||||
|
|
|
@ -1,3 +1,17 @@
|
||||||
|
// 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 "guetzli_transaction.h"
|
#include "guetzli_transaction.h"
|
||||||
|
|
||||||
#include <iostream>
|
#include <iostream>
|
||||||
|
@ -62,14 +76,14 @@ absl::Status GuetzliTransaction::Main() {
|
||||||
params_.memlimit_mb
|
params_.memlimit_mb
|
||||||
};
|
};
|
||||||
|
|
||||||
auto result_status = image_type_ == ImageType::JPEG ?
|
auto result_status = image_type_ == ImageType::kJpeg ?
|
||||||
api.ProcessJpeg(processing_params.PtrBefore(), output.PtrBoth()) :
|
api.ProcessJpeg(processing_params.PtrBefore(), output.PtrBoth()) :
|
||||||
api.ProcessRgb(processing_params.PtrBefore(), output.PtrBoth());
|
api.ProcessRgb(processing_params.PtrBefore(), output.PtrBoth());
|
||||||
|
|
||||||
if (!result_status.value_or(false)) {
|
if (!result_status.value_or(false)) {
|
||||||
std::stringstream error_stream;
|
std::stringstream error_stream;
|
||||||
error_stream << "Error processing "
|
error_stream << "Error processing "
|
||||||
<< (image_type_ == ImageType::JPEG ? "jpeg" : "rgb") << " data"
|
<< (image_type_ == ImageType::kJpeg ? "jpeg" : "rgb") << " data"
|
||||||
<< std::endl;
|
<< std::endl;
|
||||||
|
|
||||||
return absl::FailedPreconditionError(
|
return absl::FailedPreconditionError(
|
||||||
|
@ -113,7 +127,7 @@ sapi::StatusOr<ImageType> GuetzliTransaction::GetImageTypeFromFd(int fd) const {
|
||||||
}
|
}
|
||||||
|
|
||||||
return memcmp(read_buf, kPNGMagicBytes, sizeof(kPNGMagicBytes)) == 0 ?
|
return memcmp(read_buf, kPNGMagicBytes, sizeof(kPNGMagicBytes)) == 0 ?
|
||||||
ImageType::PNG : ImageType::JPEG;
|
ImageType::kPng : ImageType::kJpeg;
|
||||||
}
|
}
|
||||||
|
|
||||||
} // namespace sandbox
|
} // namespace sandbox
|
||||||
|
|
|
@ -1,21 +1,34 @@
|
||||||
#pragma once
|
// 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.
|
||||||
|
|
||||||
|
#ifndef GUETZLI_SANDBOXED_GUETZLI_TRANSACTION_H_
|
||||||
|
#define GUETZLI_SANDBOXED_GUETZLI_TRANSACTION_H_
|
||||||
|
|
||||||
#include <libgen.h>
|
#include <libgen.h>
|
||||||
#include <syscall.h>
|
#include <syscall.h>
|
||||||
|
|
||||||
#include "guetzli_sandbox.h"
|
|
||||||
#include "sandboxed_api/transaction.h"
|
#include "sandboxed_api/transaction.h"
|
||||||
#include "sandboxed_api/vars.h"
|
#include "sandboxed_api/vars.h"
|
||||||
|
|
||||||
|
#include "guetzli_sandbox.h"
|
||||||
|
|
||||||
namespace guetzli {
|
namespace guetzli {
|
||||||
namespace sandbox {
|
namespace sandbox {
|
||||||
|
|
||||||
constexpr int kDefaultTransactionRetryCount = 0;
|
|
||||||
constexpr uint64_t kMpixPixels = 1'000'000;
|
|
||||||
|
|
||||||
enum class ImageType {
|
enum class ImageType {
|
||||||
JPEG,
|
kJpeg,
|
||||||
PNG
|
kPng
|
||||||
};
|
};
|
||||||
|
|
||||||
struct TransactionParams {
|
struct TransactionParams {
|
||||||
|
@ -55,8 +68,13 @@ class GuetzliTransaction : public sapi::Transaction {
|
||||||
const TransactionParams params_;
|
const TransactionParams params_;
|
||||||
sapi::v::Fd in_fd_;
|
sapi::v::Fd in_fd_;
|
||||||
sapi::v::Fd out_fd_;
|
sapi::v::Fd out_fd_;
|
||||||
ImageType image_type_ = ImageType::JPEG;
|
ImageType image_type_ = ImageType::kJpeg;
|
||||||
|
|
||||||
|
static const int kDefaultTransactionRetryCount = 0;
|
||||||
|
static const uint64_t kMpixPixels = 1'000'000;
|
||||||
};
|
};
|
||||||
|
|
||||||
} // namespace sandbox
|
} // namespace sandbox
|
||||||
} // namespace guetzli
|
} // namespace guetzli
|
||||||
|
|
||||||
|
#endif // GUETZLI_SANDBOXED_GUETZLI_TRANSACTION_H_
|
||||||
|
|
|
@ -1,3 +1,17 @@
|
||||||
|
# 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.
|
||||||
|
|
||||||
cc_test(
|
cc_test(
|
||||||
name = "transaction_tests",
|
name = "transaction_tests",
|
||||||
srcs = ["guetzli_transaction_test.cc"],
|
srcs = ["guetzli_transaction_test.cc"],
|
||||||
|
|
|
@ -1,16 +1,32 @@
|
||||||
|
// 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 <fcntl.h>
|
||||||
|
#include <sys/stat.h>
|
||||||
|
#include <sys/types.h>
|
||||||
|
#include <syscall.h>
|
||||||
|
|
||||||
|
#include <algorithm>
|
||||||
|
#include <fstream>
|
||||||
|
#include <memory>
|
||||||
|
#include <sstream>
|
||||||
|
|
||||||
#include "gtest/gtest.h"
|
#include "gtest/gtest.h"
|
||||||
#include "guetzli_sandbox.h"
|
|
||||||
#include "sandboxed_api/sandbox2/util/fileops.h"
|
#include "sandboxed_api/sandbox2/util/fileops.h"
|
||||||
#include "sandboxed_api/vars.h"
|
#include "sandboxed_api/vars.h"
|
||||||
|
|
||||||
#include <fstream>
|
#include "guetzli_sandbox.h"
|
||||||
#include <memory>
|
|
||||||
#include <syscall.h>
|
|
||||||
#include <sys/types.h>
|
|
||||||
#include <sys/stat.h>
|
|
||||||
#include <fcntl.h>
|
|
||||||
#include <sstream>
|
|
||||||
#include <algorithm>
|
|
||||||
|
|
||||||
namespace guetzli {
|
namespace guetzli {
|
||||||
namespace sandbox {
|
namespace sandbox {
|
||||||
|
@ -18,23 +34,23 @@ namespace tests {
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
|
|
||||||
constexpr const char* IN_PNG_FILENAME = "bees.png";
|
constexpr const char* kInPngFilename = "bees.png";
|
||||||
constexpr const char* IN_JPG_FILENAME = "nature.jpg";
|
constexpr const char* kInJpegFilename = "nature.jpg";
|
||||||
constexpr const char* PNG_REFERENCE_FILENAME = "bees_reference.jpg";
|
constexpr const char* kPngReferenceFilename = "bees_reference.jpg";
|
||||||
constexpr const char* JPG_REFERENCE_FILENAME = "nature_reference.jpg";
|
constexpr const char* kJpegReferenceFIlename = "nature_reference.jpg";
|
||||||
|
|
||||||
constexpr int PNG_EXPECTED_SIZE = 38'625;
|
constexpr int kPngExpectedSize = 38'625;
|
||||||
constexpr int JPG_EXPECTED_SIZE = 10'816;
|
constexpr int kJpegExpectedSize = 10'816;
|
||||||
|
|
||||||
constexpr int DEFAULT_QUALITY_TARGET = 95;
|
constexpr int kDefaultQualityTarget = 95;
|
||||||
constexpr int DEFAULT_MEMLIMIT_MB = 6000;
|
constexpr int kDefaultMemlimitMb = 6000;
|
||||||
|
|
||||||
constexpr const char* RELATIVE_PATH_TO_TESTDATA =
|
constexpr const char* kRelativePathToTestdata =
|
||||||
"/guetzli_sandboxed/tests/testdata/";
|
"/guetzli_sandboxed/tests/testdata/";
|
||||||
|
|
||||||
std::string GetPathToInputFile(const char* filename) {
|
std::string GetPathToInputFile(const char* filename) {
|
||||||
return std::string(getenv("TEST_SRCDIR"))
|
return std::string(getenv("TEST_SRCDIR"))
|
||||||
+ std::string(RELATIVE_PATH_TO_TESTDATA)
|
+ std::string(kRelativePathToTestdata)
|
||||||
+ std::string(filename);
|
+ std::string(filename);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -78,7 +94,7 @@ protected:
|
||||||
|
|
||||||
// This test can take up to few minutes depending on your hardware
|
// This test can take up to few minutes depending on your hardware
|
||||||
TEST_F(GuetzliSapiTest, ProcessRGB) {
|
TEST_F(GuetzliSapiTest, ProcessRGB) {
|
||||||
sapi::v::Fd in_fd(open(GetPathToInputFile(IN_PNG_FILENAME).c_str(),
|
sapi::v::Fd in_fd(open(GetPathToInputFile(kInPngFilename).c_str(),
|
||||||
O_RDONLY));
|
O_RDONLY));
|
||||||
ASSERT_TRUE(in_fd.GetValue() != -1) << "Error opening input file";
|
ASSERT_TRUE(in_fd.GetValue() != -1) << "Error opening input file";
|
||||||
ASSERT_EQ(api_->sandbox()->TransferToSandboxee(&in_fd), absl::OkStatus())
|
ASSERT_EQ(api_->sandbox()->TransferToSandboxee(&in_fd), absl::OkStatus())
|
||||||
|
@ -87,17 +103,17 @@ TEST_F(GuetzliSapiTest, ProcessRGB) {
|
||||||
sapi::v::Struct<ProcessingParams> processing_params;
|
sapi::v::Struct<ProcessingParams> processing_params;
|
||||||
*processing_params.mutable_data() = {in_fd.GetRemoteFd(),
|
*processing_params.mutable_data() = {in_fd.GetRemoteFd(),
|
||||||
0,
|
0,
|
||||||
DEFAULT_QUALITY_TARGET,
|
kDefaultQualityTarget,
|
||||||
DEFAULT_MEMLIMIT_MB
|
kDefaultMemlimitMb
|
||||||
};
|
};
|
||||||
sapi::v::LenVal output(0);
|
sapi::v::LenVal output(0);
|
||||||
auto processing_result = api_->ProcessRgb(processing_params.PtrBefore(),
|
auto processing_result = api_->ProcessRgb(processing_params.PtrBefore(),
|
||||||
output.PtrBoth());
|
output.PtrBoth());
|
||||||
ASSERT_TRUE(processing_result.value_or(false)) << "Error processing rgb data";
|
ASSERT_TRUE(processing_result.value_or(false)) << "Error processing rgb data";
|
||||||
ASSERT_EQ(output.GetDataSize(), PNG_EXPECTED_SIZE)
|
ASSERT_EQ(output.GetDataSize(), kPngExpectedSize)
|
||||||
<< "Incorrect result data size";
|
<< "Incorrect result data size";
|
||||||
std::string reference_data =
|
std::string reference_data =
|
||||||
ReadFromFile(GetPathToInputFile(PNG_REFERENCE_FILENAME));
|
ReadFromFile(GetPathToInputFile(kPngReferenceFilename));
|
||||||
ASSERT_EQ(output.GetDataSize(), reference_data.size())
|
ASSERT_EQ(output.GetDataSize(), reference_data.size())
|
||||||
<< "Incorrect result data size";
|
<< "Incorrect result data size";
|
||||||
ASSERT_TRUE(CompareBytesInLenValAndContainer(output, reference_data))
|
ASSERT_TRUE(CompareBytesInLenValAndContainer(output, reference_data))
|
||||||
|
@ -106,7 +122,7 @@ TEST_F(GuetzliSapiTest, ProcessRGB) {
|
||||||
|
|
||||||
// This test can take up to few minutes depending on your hardware
|
// This test can take up to few minutes depending on your hardware
|
||||||
TEST_F(GuetzliSapiTest, ProcessJpeg) {
|
TEST_F(GuetzliSapiTest, ProcessJpeg) {
|
||||||
sapi::v::Fd in_fd(open(GetPathToInputFile(IN_JPG_FILENAME).c_str(),
|
sapi::v::Fd in_fd(open(GetPathToInputFile(kInJpegFilename).c_str(),
|
||||||
O_RDONLY));
|
O_RDONLY));
|
||||||
ASSERT_TRUE(in_fd.GetValue() != -1) << "Error opening input file";
|
ASSERT_TRUE(in_fd.GetValue() != -1) << "Error opening input file";
|
||||||
ASSERT_EQ(api_->sandbox()->TransferToSandboxee(&in_fd), absl::OkStatus())
|
ASSERT_EQ(api_->sandbox()->TransferToSandboxee(&in_fd), absl::OkStatus())
|
||||||
|
@ -115,27 +131,23 @@ TEST_F(GuetzliSapiTest, ProcessJpeg) {
|
||||||
sapi::v::Struct<ProcessingParams> processing_params;
|
sapi::v::Struct<ProcessingParams> processing_params;
|
||||||
*processing_params.mutable_data() = {in_fd.GetRemoteFd(),
|
*processing_params.mutable_data() = {in_fd.GetRemoteFd(),
|
||||||
0,
|
0,
|
||||||
DEFAULT_QUALITY_TARGET,
|
kDefaultQualityTarget,
|
||||||
DEFAULT_MEMLIMIT_MB
|
kDefaultMemlimitMb
|
||||||
};
|
};
|
||||||
sapi::v::LenVal output(0);
|
sapi::v::LenVal output(0);
|
||||||
auto processing_result = api_->ProcessJpeg(processing_params.PtrBefore(),
|
auto processing_result = api_->ProcessJpeg(processing_params.PtrBefore(),
|
||||||
output.PtrBoth());
|
output.PtrBoth());
|
||||||
ASSERT_TRUE(processing_result.value_or(false)) << "Error processing jpg data";
|
ASSERT_TRUE(processing_result.value_or(false)) << "Error processing jpg data";
|
||||||
ASSERT_EQ(output.GetDataSize(), JPG_EXPECTED_SIZE)
|
ASSERT_EQ(output.GetDataSize(), kJpegExpectedSize)
|
||||||
<< "Incorrect result data size";
|
<< "Incorrect result data size";
|
||||||
std::string reference_data =
|
std::string reference_data =
|
||||||
ReadFromFile(GetPathToInputFile(JPG_REFERENCE_FILENAME));
|
ReadFromFile(GetPathToInputFile(kJpegReferenceFIlename));
|
||||||
ASSERT_EQ(output.GetDataSize(), reference_data.size())
|
ASSERT_EQ(output.GetDataSize(), reference_data.size())
|
||||||
<< "Incorrect result data size";
|
<< "Incorrect result data size";
|
||||||
ASSERT_TRUE(CompareBytesInLenValAndContainer(output, reference_data))
|
ASSERT_TRUE(CompareBytesInLenValAndContainer(output, reference_data))
|
||||||
<< "Processed data doesn't match reference output";
|
<< "Processed data doesn't match reference output";
|
||||||
}
|
}
|
||||||
|
|
||||||
// TEST_F(GuetzliSapiTest, WriteDataToFd) {
|
|
||||||
// sapi::v::Fd fd(open(".", O_TMPFILE | O_RDWR, S_IRUSR | S_IWUSR));
|
|
||||||
// }
|
|
||||||
|
|
||||||
} // namespace tests
|
} // namespace tests
|
||||||
} // namespace sandbox
|
} // namespace sandbox
|
||||||
} // namespace guetzli
|
} // namespace guetzli
|
|
@ -1,14 +1,30 @@
|
||||||
#include "gtest/gtest.h"
|
// Copyright 2020 Google LLC
|
||||||
#include "guetzli_transaction.h"
|
//
|
||||||
#include "sandboxed_api/sandbox2/util/fileops.h"
|
// 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 <fcntl.h>
|
#include <fcntl.h>
|
||||||
#include <sys/stat.h>
|
#include <sys/stat.h>
|
||||||
#include <sys/types.h>
|
#include <sys/types.h>
|
||||||
#include <unistd.h>
|
#include <unistd.h>
|
||||||
|
|
||||||
#include <fstream>
|
#include <fstream>
|
||||||
#include <sstream>
|
|
||||||
#include <memory>
|
#include <memory>
|
||||||
|
#include <sstream>
|
||||||
|
|
||||||
|
#include "gtest/gtest.h"
|
||||||
|
#include "sandboxed_api/sandbox2/util/fileops.h"
|
||||||
|
|
||||||
|
#include "guetzli_transaction.h"
|
||||||
|
|
||||||
namespace guetzli {
|
namespace guetzli {
|
||||||
namespace sandbox {
|
namespace sandbox {
|
||||||
|
@ -16,23 +32,23 @@ namespace tests {
|
||||||
|
|
||||||
namespace {
|
namespace {
|
||||||
|
|
||||||
constexpr const char* IN_PNG_FILENAME = "bees.png";
|
constexpr const char* kInPngFilename = "bees.png";
|
||||||
constexpr const char* IN_JPG_FILENAME = "nature.jpg";
|
constexpr const char* kInJpegFilename = "nature.jpg";
|
||||||
constexpr const char* PNG_REFERENCE_FILENAME = "bees_reference.jpg";
|
constexpr const char* kPngReferenceFilename = "bees_reference.jpg";
|
||||||
constexpr const char* JPG_REFERENCE_FILENAME = "nature_reference.jpg";
|
constexpr const char* kJpegReferenceFIlename = "nature_reference.jpg";
|
||||||
|
|
||||||
constexpr int PNG_EXPECTED_SIZE = 38'625;
|
constexpr int kPngExpectedSize = 38'625;
|
||||||
constexpr int JPG_EXPECTED_SIZE = 10'816;
|
constexpr int kJpegExpectedSize = 10'816;
|
||||||
|
|
||||||
constexpr int DEFAULT_QUALITY_TARGET = 95;
|
constexpr int kDefaultQualityTarget = 95;
|
||||||
constexpr int DEFAULT_MEMLIMIT_MB = 6000;
|
constexpr int kDefaultMemlimitMb = 6000;
|
||||||
|
|
||||||
constexpr const char* RELATIVE_PATH_TO_TESTDATA =
|
constexpr const char* kRelativePathToTestdata =
|
||||||
"/guetzli_sandboxed/tests/testdata/";
|
"/guetzli_sandboxed/tests/testdata/";
|
||||||
|
|
||||||
std::string GetPathToInputFile(const char* filename) {
|
std::string GetPathToInputFile(const char* filename) {
|
||||||
return std::string(getenv("TEST_SRCDIR"))
|
return std::string(getenv("TEST_SRCDIR"))
|
||||||
+ std::string(RELATIVE_PATH_TO_TESTDATA)
|
+ std::string(kRelativePathToTestdata)
|
||||||
+ std::string(filename);
|
+ std::string(filename);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -52,7 +68,7 @@ std::string ReadFromFile(const std::string& filename) {
|
||||||
|
|
||||||
TEST(GuetzliTransactionTest, TestTransactionJpg) {
|
TEST(GuetzliTransactionTest, TestTransactionJpg) {
|
||||||
sandbox2::file_util::fileops::FDCloser in_fd_closer(
|
sandbox2::file_util::fileops::FDCloser in_fd_closer(
|
||||||
open(GetPathToInputFile(IN_JPG_FILENAME).c_str(), O_RDONLY));
|
open(GetPathToInputFile(kInJpegFilename).c_str(), O_RDONLY));
|
||||||
ASSERT_TRUE(in_fd_closer.get() != -1) << "Error opening input jpg file";
|
ASSERT_TRUE(in_fd_closer.get() != -1) << "Error opening input jpg file";
|
||||||
sandbox2::file_util::fileops::FDCloser out_fd_closer(
|
sandbox2::file_util::fileops::FDCloser out_fd_closer(
|
||||||
open(".", O_TMPFILE | O_RDWR, S_IRUSR | S_IWUSR));
|
open(".", O_TMPFILE | O_RDWR, S_IRUSR | S_IWUSR));
|
||||||
|
@ -61,8 +77,8 @@ TEST(GuetzliTransactionTest, TestTransactionJpg) {
|
||||||
in_fd_closer.get(),
|
in_fd_closer.get(),
|
||||||
out_fd_closer.get(),
|
out_fd_closer.get(),
|
||||||
0,
|
0,
|
||||||
DEFAULT_QUALITY_TARGET,
|
kDefaultQualityTarget,
|
||||||
DEFAULT_MEMLIMIT_MB
|
kDefaultMemlimitMb
|
||||||
};
|
};
|
||||||
{
|
{
|
||||||
GuetzliTransaction transaction(std::move(params));
|
GuetzliTransaction transaction(std::move(params));
|
||||||
|
@ -72,7 +88,7 @@ TEST(GuetzliTransactionTest, TestTransactionJpg) {
|
||||||
}
|
}
|
||||||
ASSERT_TRUE(fcntl(out_fd_closer.get(), F_GETFD) != -1 || errno != EBADF)
|
ASSERT_TRUE(fcntl(out_fd_closer.get(), F_GETFD) != -1 || errno != EBADF)
|
||||||
<< "Local output fd closed";
|
<< "Local output fd closed";
|
||||||
auto reference_data = ReadFromFile(GetPathToInputFile(JPG_REFERENCE_FILENAME));
|
auto reference_data = ReadFromFile(GetPathToInputFile(kJpegReferenceFIlename));
|
||||||
auto output_size = lseek(out_fd_closer.get(), 0, SEEK_END);
|
auto output_size = lseek(out_fd_closer.get(), 0, SEEK_END);
|
||||||
ASSERT_EQ(reference_data.size(), output_size)
|
ASSERT_EQ(reference_data.size(), output_size)
|
||||||
<< "Different sizes of reference and returned data";
|
<< "Different sizes of reference and returned data";
|
||||||
|
@ -90,7 +106,7 @@ TEST(GuetzliTransactionTest, TestTransactionJpg) {
|
||||||
|
|
||||||
TEST(GuetzliTransactionTest, TestTransactionPng) {
|
TEST(GuetzliTransactionTest, TestTransactionPng) {
|
||||||
sandbox2::file_util::fileops::FDCloser in_fd_closer(
|
sandbox2::file_util::fileops::FDCloser in_fd_closer(
|
||||||
open(GetPathToInputFile(IN_PNG_FILENAME).c_str(), O_RDONLY));
|
open(GetPathToInputFile(kInPngFilename).c_str(), O_RDONLY));
|
||||||
ASSERT_TRUE(in_fd_closer.get() != -1) << "Error opening input png file";
|
ASSERT_TRUE(in_fd_closer.get() != -1) << "Error opening input png file";
|
||||||
sandbox2::file_util::fileops::FDCloser out_fd_closer(
|
sandbox2::file_util::fileops::FDCloser out_fd_closer(
|
||||||
open(".", O_TMPFILE | O_RDWR, S_IRUSR | S_IWUSR));
|
open(".", O_TMPFILE | O_RDWR, S_IRUSR | S_IWUSR));
|
||||||
|
@ -99,8 +115,8 @@ TEST(GuetzliTransactionTest, TestTransactionPng) {
|
||||||
in_fd_closer.get(),
|
in_fd_closer.get(),
|
||||||
out_fd_closer.get(),
|
out_fd_closer.get(),
|
||||||
0,
|
0,
|
||||||
DEFAULT_QUALITY_TARGET,
|
kDefaultQualityTarget,
|
||||||
DEFAULT_MEMLIMIT_MB
|
kDefaultMemlimitMb
|
||||||
};
|
};
|
||||||
{
|
{
|
||||||
GuetzliTransaction transaction(std::move(params));
|
GuetzliTransaction transaction(std::move(params));
|
||||||
|
@ -110,7 +126,7 @@ TEST(GuetzliTransactionTest, TestTransactionPng) {
|
||||||
}
|
}
|
||||||
ASSERT_TRUE(fcntl(out_fd_closer.get(), F_GETFD) != -1 || errno != EBADF)
|
ASSERT_TRUE(fcntl(out_fd_closer.get(), F_GETFD) != -1 || errno != EBADF)
|
||||||
<< "Local output fd closed";
|
<< "Local output fd closed";
|
||||||
auto reference_data = ReadFromFile(GetPathToInputFile(PNG_REFERENCE_FILENAME));
|
auto reference_data = ReadFromFile(GetPathToInputFile(kPngReferenceFilename));
|
||||||
auto output_size = lseek(out_fd_closer.get(), 0, SEEK_END);
|
auto output_size = lseek(out_fd_closer.get(), 0, SEEK_END);
|
||||||
ASSERT_EQ(reference_data.size(), output_size)
|
ASSERT_EQ(reference_data.size(), output_size)
|
||||||
<< "Different sizes of reference and returned data";
|
<< "Different sizes of reference and returned data";
|
||||||
|
|
Loading…
Reference in New Issue
Block a user