1*ec63e07aSXin Li // Copyright 2019 Google LLC
2*ec63e07aSXin Li //
3*ec63e07aSXin Li // Licensed under the Apache License, Version 2.0 (the "License");
4*ec63e07aSXin Li // you may not use this file except in compliance with the License.
5*ec63e07aSXin Li // You may obtain a copy of the License at
6*ec63e07aSXin Li //
7*ec63e07aSXin Li // https://www.apache.org/licenses/LICENSE-2.0
8*ec63e07aSXin Li //
9*ec63e07aSXin Li // Unless required by applicable law or agreed to in writing, software
10*ec63e07aSXin Li // distributed under the License is distributed on an "AS IS" BASIS,
11*ec63e07aSXin Li // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12*ec63e07aSXin Li // See the License for the specific language governing permissions and
13*ec63e07aSXin Li // limitations under the License.
14*ec63e07aSXin Li
15*ec63e07aSXin Li #include "sandboxed_api/util/file_helpers.h"
16*ec63e07aSXin Li
17*ec63e07aSXin Li #include <fstream>
18*ec63e07aSXin Li #include <sstream>
19*ec63e07aSXin Li #include <string>
20*ec63e07aSXin Li
21*ec63e07aSXin Li #include "absl/status/status.h"
22*ec63e07aSXin Li #include "absl/strings/str_cat.h"
23*ec63e07aSXin Li #include "absl/strings/string_view.h"
24*ec63e07aSXin Li
25*ec63e07aSXin Li namespace sapi::file {
26*ec63e07aSXin Li
Defaults()27*ec63e07aSXin Li const Options& Defaults() {
28*ec63e07aSXin Li static auto* instance = new Options{};
29*ec63e07aSXin Li return *instance;
30*ec63e07aSXin Li }
31*ec63e07aSXin Li
GetContents(absl::string_view path,std::string * output,const file::Options & options)32*ec63e07aSXin Li absl::Status GetContents(absl::string_view path, std::string* output,
33*ec63e07aSXin Li const file::Options& options) {
34*ec63e07aSXin Li std::ifstream in_stream{std::string(path), std::ios_base::binary};
35*ec63e07aSXin Li std::ostringstream out_stream;
36*ec63e07aSXin Li out_stream << in_stream.rdbuf();
37*ec63e07aSXin Li if (!in_stream || !out_stream) {
38*ec63e07aSXin Li return absl::UnknownError(absl::StrCat("Error during read: ", path));
39*ec63e07aSXin Li }
40*ec63e07aSXin Li *output = out_stream.str();
41*ec63e07aSXin Li return absl::OkStatus();
42*ec63e07aSXin Li }
43*ec63e07aSXin Li
SetContents(absl::string_view path,absl::string_view content,const file::Options & options)44*ec63e07aSXin Li absl::Status SetContents(absl::string_view path, absl::string_view content,
45*ec63e07aSXin Li const file::Options& options) {
46*ec63e07aSXin Li std::ofstream out_stream(std::string(path),
47*ec63e07aSXin Li std::ios_base::trunc | std::ios_base::binary);
48*ec63e07aSXin Li if (!out_stream) {
49*ec63e07aSXin Li return absl::UnknownError(absl::StrCat("Failed to open file: ", path));
50*ec63e07aSXin Li }
51*ec63e07aSXin Li out_stream.write(content.data(), content.size());
52*ec63e07aSXin Li if (!out_stream) {
53*ec63e07aSXin Li return absl::UnknownError(absl::StrCat("Error during write: ", path));
54*ec63e07aSXin Li }
55*ec63e07aSXin Li return absl::OkStatus();
56*ec63e07aSXin Li }
57*ec63e07aSXin Li
58*ec63e07aSXin Li } // namespace sapi::file
59