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/temp_file.h"
16*ec63e07aSXin Li
17*ec63e07aSXin Li #include <unistd.h>
18*ec63e07aSXin Li
19*ec63e07aSXin Li #include <cerrno>
20*ec63e07aSXin Li #include <string>
21*ec63e07aSXin Li #include <utility>
22*ec63e07aSXin Li
23*ec63e07aSXin Li #include "absl/status/status.h"
24*ec63e07aSXin Li #include "absl/status/statusor.h"
25*ec63e07aSXin Li #include "absl/strings/str_cat.h"
26*ec63e07aSXin Li #include "absl/strings/string_view.h"
27*ec63e07aSXin Li #include "sandboxed_api/util/status_macros.h"
28*ec63e07aSXin Li
29*ec63e07aSXin Li namespace sapi {
30*ec63e07aSXin Li
31*ec63e07aSXin Li namespace {
32*ec63e07aSXin Li constexpr absl::string_view kMktempSuffix = "XXXXXX";
33*ec63e07aSXin Li } // namespace
34*ec63e07aSXin Li
CreateNamedTempFile(absl::string_view prefix)35*ec63e07aSXin Li absl::StatusOr<std::pair<std::string, int>> CreateNamedTempFile(
36*ec63e07aSXin Li absl::string_view prefix) {
37*ec63e07aSXin Li std::string name_template = absl::StrCat(prefix, kMktempSuffix);
38*ec63e07aSXin Li int fd = mkstemp(&name_template[0]);
39*ec63e07aSXin Li if (fd < 0) {
40*ec63e07aSXin Li return absl::ErrnoToStatus(errno, "mkstemp()");
41*ec63e07aSXin Li }
42*ec63e07aSXin Li return std::pair<std::string, int>{std::move(name_template), fd};
43*ec63e07aSXin Li }
44*ec63e07aSXin Li
CreateNamedTempFileAndClose(absl::string_view prefix)45*ec63e07aSXin Li absl::StatusOr<std::string> CreateNamedTempFileAndClose(
46*ec63e07aSXin Li absl::string_view prefix) {
47*ec63e07aSXin Li SAPI_ASSIGN_OR_RETURN(auto result, CreateNamedTempFile(prefix));
48*ec63e07aSXin Li close(result.second);
49*ec63e07aSXin Li return std::move(result.first);
50*ec63e07aSXin Li }
51*ec63e07aSXin Li
CreateTempDir(absl::string_view prefix)52*ec63e07aSXin Li absl::StatusOr<std::string> CreateTempDir(absl::string_view prefix) {
53*ec63e07aSXin Li std::string name_template = absl::StrCat(prefix, kMktempSuffix);
54*ec63e07aSXin Li if (mkdtemp(&name_template[0]) == nullptr) {
55*ec63e07aSXin Li return absl::ErrnoToStatus(errno, "mkdtemp()");
56*ec63e07aSXin Li }
57*ec63e07aSXin Li return name_template;
58*ec63e07aSXin Li }
59*ec63e07aSXin Li
60*ec63e07aSXin Li } // namespace sapi
61