1/*
2 * Copyright (C) 2013 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17#include <fcntl.h>
18#include <unistd.h>
19
20#include "private/bionic_macros.h"
21
22template <typename T = int (*)(char*)>
23class GenericTemporaryFile {
24 public:
25  GenericTemporaryFile(T mk_fn = mkstemp) : mk_fn(mk_fn) {
26    // Since we might be running on the host or the target, and if we're
27    // running on the host we might be running under bionic or glibc,
28    // let's just try both possible temporary directories and take the
29    // first one that works.
30    init("/data/local/tmp");
31    if (fd == -1) {
32      init("/tmp");
33    }
34  }
35
36  GenericTemporaryFile(const char* dirpath, T mk_fn = mkstemp) : mk_fn(mk_fn) {
37    init(dirpath);
38  }
39
40  ~GenericTemporaryFile() {
41    close(fd);
42    unlink(filename);
43  }
44
45  void reopen() {
46    close(fd);
47    fd = open(filename, O_RDWR);
48  }
49
50  int fd;
51  char filename[1024];
52
53 private:
54  T mk_fn;
55
56  void init(const char* tmp_dir) {
57    snprintf(filename, sizeof(filename), "%s/TemporaryFile-XXXXXX", tmp_dir);
58    fd = mk_fn(filename);
59  }
60
61  DISALLOW_COPY_AND_ASSIGN(GenericTemporaryFile);
62};
63
64typedef GenericTemporaryFile<> TemporaryFile;
65
66class TemporaryDir {
67 public:
68  TemporaryDir() {
69    if (!init("/data/local/tmp")) {
70      init("/tmp");
71    }
72  }
73
74  ~TemporaryDir() {
75    rmdir(dirname);
76  }
77
78  char dirname[1024];
79
80 private:
81  bool init(const char* tmp_dir) {
82    snprintf(dirname, sizeof(dirname), "%s/TemporaryDir-XXXXXX", tmp_dir);
83    return (mkdtemp(dirname) != NULL);
84  }
85
86  DISALLOW_COPY_AND_ASSIGN(TemporaryDir);
87};
88