crashpad/util/stream/file_encoder.cc
Mark Mentovai 6278690abe Update copyright boilerplate, 2022 edition (Crashpad)
sed -i '' -E -e 's/Copyright (.+) The Crashpad Authors\. All rights reserved\.$/Copyright \1 The Crashpad Authors/' $(git grep -El 'Copyright (.+) The Crashpad Authors\. All rights reserved\.$')

Bug: chromium:1098010
Change-Id: I8d6138469ddbe3d281a5d83f64cf918ec2491611
Reviewed-on: https://chromium-review.googlesource.com/c/crashpad/crashpad/+/3878262
Reviewed-by: Joshua Peraza <jperaza@chromium.org>
Commit-Queue: Mark Mentovai <mark@chromium.org>
2022-09-06 23:54:07 +00:00

85 lines
2.6 KiB
C++

// Copyright 2019 The Crashpad Authors
//
// 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 "util/stream/file_encoder.h"
#include <memory>
#include <tuple>
#include "util/file/file_io.h"
#include "util/file/file_reader.h"
#include "util/file/scoped_remove_file.h"
#include "util/stream/base94_output_stream.h"
#include "util/stream/file_output_stream.h"
#include "util/stream/output_stream_interface.h"
#include "util/stream/zlib_output_stream.h"
namespace crashpad {
FileEncoder::FileEncoder(Mode mode,
const base::FilePath& input_path,
const base::FilePath& output_path)
: mode_(mode), input_path_(input_path), output_path_(output_path) {}
FileEncoder::~FileEncoder() {}
bool FileEncoder::Process() {
ScopedRemoveFile file_remover;
ScopedFileHandle write_handle(LoggingOpenFileForWrite(
output_path_, FileWriteMode::kCreateOrFail, FilePermissions::kOwnerOnly));
if (!write_handle.is_valid())
return false;
// Remove the output file on failure.
file_remover.reset(output_path_);
std::unique_ptr<OutputStreamInterface> output;
if (mode_ == Mode::kEncode) {
output = std::make_unique<ZlibOutputStream>(
ZlibOutputStream::Mode::kCompress,
std::make_unique<Base94OutputStream>(
Base94OutputStream::Mode::kEncode,
std::make_unique<FileOutputStream>(write_handle.get())));
} else {
output = std::make_unique<Base94OutputStream>(
Base94OutputStream::Mode::kDecode,
std::make_unique<ZlibOutputStream>(
ZlibOutputStream::Mode::kDecompress,
std::make_unique<FileOutputStream>(write_handle.get())));
}
FileReader file_reader;
if (!file_reader.Open(input_path_))
return false;
FileOperationResult read_result;
do {
uint8_t buffer[4096];
read_result = file_reader.Read(buffer, sizeof(buffer));
if (read_result < 0)
return false;
if (read_result > 0 && (!output->Write(buffer, read_result)))
return false;
} while (read_result > 0);
if (!output->Flush())
return false;
std::ignore = file_remover.release();
return true;
}
} // namespace crashpad