111
|
1 /* Managing temporary directories and their content within libgccjit.so
|
131
|
2 Copyright (C) 2014-2018 Free Software Foundation, Inc.
|
111
|
3 Contributed by David Malcolm <dmalcolm@redhat.com>.
|
|
4
|
|
5 This file is part of GCC.
|
|
6
|
|
7 GCC is free software; you can redistribute it and/or modify it
|
|
8 under the terms of the GNU General Public License as published by
|
|
9 the Free Software Foundation; either version 3, or (at your option)
|
|
10 any later version.
|
|
11
|
|
12 GCC is distributed in the hope that it will be useful, but
|
|
13 WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
15 General Public License for more details.
|
|
16
|
|
17 You should have received a copy of the GNU General Public License
|
|
18 along with GCC; see the file COPYING3. If not see
|
|
19 <http://www.gnu.org/licenses/>. */
|
|
20
|
|
21 #include "config.h"
|
|
22 #include "system.h"
|
|
23 #include "coretypes.h"
|
|
24
|
|
25 #include "jit-tempdir.h"
|
|
26
|
|
27
|
|
28 /* Construct a tempdir path template suitable for use by mkdtemp
|
|
29 e.g. "/tmp/libgccjit-XXXXXX", but respecting the rules in
|
|
30 libiberty's choose_tempdir rather than hardcoding "/tmp/".
|
|
31
|
|
32 The memory is allocated using malloc and must be freed.
|
|
33 Aborts the process if allocation fails. */
|
|
34
|
|
35 static char *
|
|
36 make_tempdir_path_template ()
|
|
37 {
|
|
38 const char *tmpdir_buf;
|
|
39 size_t tmpdir_len;
|
|
40 const char *file_template_buf;
|
|
41 size_t file_template_len;
|
|
42 char *result;
|
|
43
|
|
44 /* The result of choose_tmpdir is a cached buffer within libiberty, so
|
|
45 we must *not* free it. */
|
|
46 tmpdir_buf = choose_tmpdir ();
|
|
47
|
|
48 /* choose_tmpdir aborts on malloc failure. */
|
|
49 gcc_assert (tmpdir_buf);
|
|
50
|
|
51 tmpdir_len = strlen (tmpdir_buf);
|
|
52 /* tmpdir_buf should now have a dir separator as the final byte. */
|
|
53 gcc_assert (tmpdir_len > 0);
|
|
54 gcc_assert (tmpdir_buf[tmpdir_len - 1] == DIR_SEPARATOR);
|
|
55
|
|
56 file_template_buf = "libgccjit-XXXXXX";
|
|
57 file_template_len = strlen (file_template_buf);
|
|
58
|
|
59 result = XNEWVEC (char, tmpdir_len + file_template_len + 1);
|
|
60 strcpy (result, tmpdir_buf);
|
|
61 strcpy (result + tmpdir_len, file_template_buf);
|
|
62
|
|
63 return result;
|
|
64 }
|
|
65
|
|
66 /* The constructor for the jit::tempdir object.
|
|
67 The real work is done by the jit::tempdir::create method. */
|
|
68
|
|
69 gcc::jit::tempdir::tempdir (logger *logger, int keep_intermediates)
|
|
70 : log_user (logger),
|
|
71 m_keep_intermediates (keep_intermediates),
|
|
72 m_path_template (NULL),
|
|
73 m_path_tempdir (NULL),
|
|
74 m_path_c_file (NULL),
|
|
75 m_path_s_file (NULL),
|
|
76 m_path_so_file (NULL)
|
|
77 {
|
|
78 JIT_LOG_SCOPE (get_logger ());
|
|
79 }
|
|
80
|
|
81 /* Do the real work of creating the on-disk tempdir.
|
|
82 We do this here, rather than in the jit::tempdir constructor
|
|
83 so that we can handle failure without needing exceptions. */
|
|
84
|
|
85 bool
|
|
86 gcc::jit::tempdir::create ()
|
|
87 {
|
|
88 JIT_LOG_SCOPE (get_logger ());
|
|
89
|
|
90 m_path_template = make_tempdir_path_template ();
|
|
91 if (!m_path_template)
|
|
92 return false;
|
|
93
|
|
94 log ("m_path_template: %s", m_path_template);
|
|
95
|
|
96 /* Create tempdir using mkdtemp. This is created with 0700 perms and
|
|
97 is unique. Hence no other (non-root) users should have access to
|
|
98 the paths within it. */
|
|
99 m_path_tempdir = mkdtemp (m_path_template);
|
|
100 if (!m_path_tempdir)
|
|
101 return false;
|
|
102 log ("m_path_tempdir: %s", m_path_tempdir);
|
|
103
|
|
104 m_path_c_file = concat (m_path_tempdir, "/fake.c", NULL);
|
|
105 m_path_s_file = concat (m_path_tempdir, "/fake.s", NULL);
|
|
106 m_path_so_file = concat (m_path_tempdir, "/fake.so", NULL);
|
|
107
|
|
108 /* Success. */
|
|
109 return true;
|
|
110 }
|
|
111
|
|
112 /* The destructor for the jit::tempdir object, which
|
|
113 cleans up the filesystem directory and its contents
|
|
114 (unless keep_intermediates was set). */
|
|
115
|
|
116 gcc::jit::tempdir::~tempdir ()
|
|
117 {
|
|
118 JIT_LOG_SCOPE (get_logger ());
|
|
119
|
|
120 if (m_keep_intermediates)
|
|
121 fprintf (stderr, "intermediate files written to %s\n", m_path_tempdir);
|
|
122 else
|
|
123 {
|
|
124 /* Clean up .s/.so. */
|
|
125 if (m_path_s_file)
|
|
126 {
|
|
127 log ("unlinking .s file: %s", m_path_s_file);
|
|
128 unlink (m_path_s_file);
|
|
129 }
|
|
130 if (m_path_so_file)
|
|
131 {
|
|
132 log ("unlinking .so file: %s", m_path_so_file);
|
|
133 unlink (m_path_so_file);
|
|
134 }
|
|
135
|
|
136 /* Clean up any other tempfiles. */
|
|
137 int i;
|
|
138 char *tempfile;
|
|
139 FOR_EACH_VEC_ELT (m_tempfiles, i, tempfile)
|
|
140 {
|
|
141 log ("unlinking tempfile: %s", tempfile);
|
|
142 unlink (tempfile);
|
|
143 }
|
|
144
|
|
145 /* The tempdir should now be empty; remove it. */
|
|
146 if (m_path_tempdir)
|
|
147 {
|
|
148 log ("removing tempdir: %s", m_path_tempdir);
|
|
149 rmdir (m_path_tempdir);
|
|
150 }
|
|
151 }
|
|
152
|
|
153 free (m_path_template);
|
|
154 /* m_path_tempdir aliases m_path_template, or is NULL, so don't
|
|
155 attempt to free it . */
|
|
156 free (m_path_c_file);
|
|
157 free (m_path_s_file);
|
|
158 free (m_path_so_file);
|
|
159
|
|
160 int i;
|
|
161 char *tempfile;
|
|
162 FOR_EACH_VEC_ELT (m_tempfiles, i, tempfile)
|
|
163 free (tempfile);
|
|
164 }
|