mirror of
https://github.com/kidfromjupiter/nearby.git
synced 2026-09-15 07:06:11 -04:00
hai007
80ddcbe0fc
Suspect the file was corrupted when sending to receiver side. Check if it is good first, then peek. By this way, the crash maybe avoid. For further protection, add exception protection here. PiperOrigin-RevId: 542884209
120 lines
3.3 KiB
C++
120 lines
3.3 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
|
|
//
|
|
// https://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 "internal/platform/implementation/windows/file.h"
|
|
|
|
#include <algorithm>
|
|
#include <cstddef>
|
|
#include <ios>
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
#include "absl/memory/memory.h"
|
|
#include "absl/strings/string_view.h"
|
|
#include "internal/platform/exception.h"
|
|
#include "internal/platform/implementation/windows/utils.h"
|
|
#include "internal/platform/logging.h"
|
|
|
|
namespace nearby {
|
|
namespace windows {
|
|
|
|
// InputFile
|
|
std::unique_ptr<IOFile> IOFile::CreateInputFile(
|
|
const absl::string_view file_path, size_t size) {
|
|
return absl::WrapUnique(new IOFile(file_path, size));
|
|
}
|
|
|
|
IOFile::IOFile(const absl::string_view file_path, size_t size)
|
|
: path_(file_path) {
|
|
// Always open input file path as wide string on Windows platform.
|
|
std::wstring wide_path = string_to_wstring(std::string(file_path));
|
|
file_.open(wide_path, std::ios::binary | std::ios::in | std::ios::ate);
|
|
|
|
total_size_ = file_.tellg();
|
|
file_.seekg(0);
|
|
}
|
|
|
|
std::unique_ptr<IOFile> IOFile::CreateOutputFile(const absl::string_view path) {
|
|
return std::unique_ptr<IOFile>(new IOFile(path));
|
|
}
|
|
|
|
IOFile::IOFile(const absl::string_view file_path)
|
|
: file_(), path_(file_path), total_size_(0) {
|
|
// Always open input file path as wide string on Windows platform.
|
|
std::wstring wide_path = string_to_wstring(path_);
|
|
file_.open(wide_path, std::ios::binary | std::ios::out);
|
|
}
|
|
|
|
ExceptionOr<ByteArray> IOFile::Read(std::int64_t size) {
|
|
try {
|
|
if (!file_.is_open()) {
|
|
return ExceptionOr<ByteArray>{Exception::kIo};
|
|
}
|
|
|
|
if (!file_.good()) {
|
|
return ExceptionOr<ByteArray>{Exception::kIo};
|
|
}
|
|
|
|
if (file_.peek() == EOF) {
|
|
return ExceptionOr<ByteArray>{ByteArray{}};
|
|
}
|
|
|
|
ByteArray bytes(size);
|
|
std::unique_ptr<char[]> read_bytes{new char[size]};
|
|
file_.read(read_bytes.get(), static_cast<ptrdiff_t>(size));
|
|
auto num_bytes_read = file_.gcount();
|
|
if (num_bytes_read == 0) {
|
|
return ExceptionOr<ByteArray>{Exception::kIo};
|
|
}
|
|
return ExceptionOr<ByteArray>(ByteArray(read_bytes.get(), num_bytes_read));
|
|
} catch (...) {
|
|
NEARBY_LOGS(ERROR) << "Fail to read";
|
|
return ExceptionOr<ByteArray>{Exception::kIo};
|
|
}
|
|
}
|
|
|
|
Exception IOFile::Close() {
|
|
if (file_.is_open()) {
|
|
file_.close();
|
|
}
|
|
return {Exception::kSuccess};
|
|
}
|
|
|
|
Exception IOFile::Write(const ByteArray& data) {
|
|
try {
|
|
if (!file_.is_open()) {
|
|
return {Exception::kIo};
|
|
}
|
|
|
|
if (!file_.good()) {
|
|
return {Exception::kIo};
|
|
}
|
|
|
|
file_.write(data.data(), data.size());
|
|
file_.flush();
|
|
return {file_.good() ? Exception::kSuccess : Exception::kIo};
|
|
} catch (...) {
|
|
NEARBY_LOGS(ERROR) << "Fail to write";
|
|
return {Exception::kIo};
|
|
}
|
|
}
|
|
|
|
Exception IOFile::Flush() {
|
|
file_.flush();
|
|
return {file_.good() ? Exception::kSuccess : Exception::kIo};
|
|
}
|
|
|
|
} // namespace windows
|
|
} // namespace nearby
|