BOOST_FOREACH.
[dcpomatic.git] / src / lib / cross_linux.cc
1 /*
2     Copyright (C) 2012-2020 Carl Hetherington <cth@carlh.net>
3
4     This file is part of DCP-o-matic.
5
6     DCP-o-matic is free software; you can redistribute it and/or modify
7     it under the terms of the GNU General Public License as published by
8     the Free Software Foundation; either version 2 of the License, or
9     (at your option) any later version.
10
11     DCP-o-matic is distributed in the hope that it will be useful,
12     but WITHOUT ANY WARRANTY; without even the implied warranty of
13     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
14     GNU General Public License for more details.
15
16     You should have received a copy of the GNU General Public License
17     along with DCP-o-matic.  If not, see <http://www.gnu.org/licenses/>.
18
19 */
20
21 #include "cross.h"
22 #include "compose.hpp"
23 #include "log.h"
24 #include "dcpomatic_log.h"
25 #include "config.h"
26 #include "exceptions.h"
27 #include "dcpomatic_log.h"
28 #include <dcp/raw_convert.h>
29 #include <glib.h>
30 extern "C" {
31 #include <libavformat/avio.h>
32 }
33 #include <boost/algorithm/string.hpp>
34 #include <boost/function.hpp>
35 #if BOOST_VERSION >= 106100
36 #include <boost/dll/runtime_symbol_info.hpp>
37 #endif
38 #include <unistd.h>
39 #include <mntent.h>
40 #include <sys/types.h>
41 #include <sys/mount.h>
42 #include <ifaddrs.h>
43 #include <netinet/in.h>
44 #include <arpa/inet.h>
45 #include <fstream>
46
47 #include "i18n.h"
48
49 using std::pair;
50 using std::list;
51 using std::ifstream;
52 using std::string;
53 using std::wstring;
54 using std::make_pair;
55 using std::vector;
56 using std::cerr;
57 using std::cout;
58 using std::runtime_error;
59 using std::shared_ptr;
60 using boost::optional;
61 using boost::function;
62
63 /** @param s Number of seconds to sleep for */
64 void
65 dcpomatic_sleep_seconds (int s)
66 {
67         sleep (s);
68 }
69
70 void
71 dcpomatic_sleep_milliseconds (int ms)
72 {
73         usleep (ms * 1000);
74 }
75
76 /** @return A string of CPU information (model name etc.) */
77 string
78 cpu_info ()
79 {
80         string info;
81
82         /* This use of ifstream is ok; the filename can never
83            be non-Latin
84         */
85         ifstream f ("/proc/cpuinfo");
86         while (f.good ()) {
87                 string l;
88                 getline (f, l);
89                 if (boost::algorithm::starts_with (l, "model name")) {
90                         string::size_type const c = l.find (':');
91                         if (c != string::npos) {
92                                 info = l.substr (c + 2);
93                         }
94                 }
95         }
96
97         return info;
98 }
99
100 boost::filesystem::path
101 resources_path ()
102 {
103         return directory_containing_executable().parent_path() / "share" / "dcpomatic2";
104 }
105
106
107 boost::filesystem::path
108 xsd_path ()
109 {
110         return boost::filesystem::canonical(LINUX_SHARE_PREFIX) / "libdcp" / "xsd";
111 }
112
113
114 boost::filesystem::path
115 tags_path ()
116 {
117         return boost::filesystem::canonical(LINUX_SHARE_PREFIX) / "libdcp" / "tags";
118 }
119
120
121 void
122 run_ffprobe (boost::filesystem::path content, boost::filesystem::path out)
123 {
124         string ffprobe = "ffprobe \"" + content.string() + "\" 2> \"" + out.string() + "\"";
125         LOG_GENERAL (N_("Probing with %1"), ffprobe);
126         int const r = system (ffprobe.c_str());
127         if (r == -1 || (WIFEXITED(r) && WEXITSTATUS(r) != 0)) {
128                 LOG_GENERAL (N_("Could not run ffprobe (system returned %1"), r);
129         }
130 }
131
132 list<pair<string, string> >
133 mount_info ()
134 {
135         list<pair<string, string> > m;
136
137         FILE* f = setmntent ("/etc/mtab", "r");
138         if (!f) {
139                 return m;
140         }
141
142         while (true) {
143                 struct mntent* mnt = getmntent (f);
144                 if (!mnt) {
145                         break;
146                 }
147
148                 m.push_back (make_pair (mnt->mnt_dir, mnt->mnt_type));
149         }
150
151         endmntent (f);
152
153         return m;
154 }
155
156
157 boost::filesystem::path
158 directory_containing_executable ()
159 {
160 #if BOOST_VERSION >= 106100
161         return boost::dll::program_location().parent_path();
162 #else
163         char buffer[PATH_MAX];
164         ssize_t N = readlink ("/proc/self/exe", buffer, PATH_MAX);
165         return boost::filesystem::path(string(buffer, N)).parent_path();
166 #endif
167 }
168
169
170 boost::filesystem::path
171 openssl_path ()
172 {
173         boost::filesystem::path p = directory_containing_executable() / "dcpomatic2_openssl";
174         if (boost::filesystem::is_regular_file(p)) {
175                 return p;
176         }
177
178         return "dcpomatic2_openssl";
179 }
180
181
182 #ifdef DCPOMATIC_DISK
183 boost::filesystem::path
184 disk_writer_path ()
185 {
186         return directory_containing_executable() / "dcpomatic2_disk_writer";
187 }
188 #endif
189
190
191 /* Apparently there is no way to create an ofstream using a UTF-8
192    filename under Windows.  We are hence reduced to using fopen
193    with this wrapper.
194 */
195 FILE *
196 fopen_boost (boost::filesystem::path p, string t)
197 {
198         return fopen (p.c_str(), t.c_str ());
199 }
200
201 int
202 dcpomatic_fseek (FILE* stream, int64_t offset, int whence)
203 {
204         return fseek (stream, offset, whence);
205 }
206
207 void
208 Waker::nudge ()
209 {
210
211 }
212
213 Waker::Waker ()
214 {
215
216 }
217
218 Waker::~Waker ()
219 {
220
221 }
222
223
224 void
225 start_tool (string executable)
226 {
227         boost::filesystem::path batch = directory_containing_executable() / executable;
228
229         pid_t pid = fork ();
230         if (pid == 0) {
231                 int const r = system (batch.string().c_str());
232                 exit (WEXITSTATUS (r));
233         }
234 }
235
236
237 void
238 start_batch_converter ()
239 {
240         start_tool ("dcpomatic2_batch");
241 }
242
243
244 void
245 start_player ()
246 {
247         start_tool ("dcpomatic2_player");
248 }
249
250
251 uint64_t
252 thread_id ()
253 {
254         return (uint64_t) pthread_self ();
255 }
256
257 int
258 avio_open_boost (AVIOContext** s, boost::filesystem::path file, int flags)
259 {
260         return avio_open (s, file.c_str(), flags);
261 }
262
263
264 boost::filesystem::path
265 home_directory ()
266 {
267         return getenv("HOME");
268 }
269
270 /** @return true if this process is a 32-bit one running on a 64-bit-capable OS */
271 bool
272 running_32_on_64 ()
273 {
274         /* I'm assuming nobody does this on Linux */
275         return false;
276 }
277
278
279 static
280 vector<pair<string, string> >
281 get_mounts (string prefix)
282 {
283         vector<pair<string, string> > mounts;
284
285         std::ifstream f("/proc/mounts");
286         string line;
287         while (f.good()) {
288                 getline(f, line);
289                 vector<string> bits;
290                 boost::algorithm::split (bits, line, boost::is_any_of(" "));
291                 if (bits.size() > 1 && boost::algorithm::starts_with(bits[0], prefix)) {
292                         boost::algorithm::replace_all (bits[1], "\\040", " ");
293                         mounts.push_back(make_pair(bits[0], bits[1]));
294                         LOG_DISK("Found mounted device %1 from prefix %2", bits[0], prefix);
295                 }
296         }
297
298         return mounts;
299 }
300
301
302 vector<Drive>
303 Drive::get ()
304 {
305         vector<Drive> drives;
306
307         using namespace boost::filesystem;
308         vector<pair<string, string> > mounted_devices = get_mounts("/dev/");
309
310         for (directory_iterator i = directory_iterator("/sys/block"); i != directory_iterator(); ++i) {
311                 string const name = i->path().filename().string();
312                 path device_type_file("/sys/block/" + name + "/device/type");
313                 optional<string> device_type;
314                 if (exists(device_type_file)) {
315                         device_type = dcp::file_to_string (device_type_file);
316                         boost::trim(*device_type);
317                 }
318                 /* Device type 5 is "SCSI_TYPE_ROM" in blkdev.h; seems usually to be a CD/DVD drive */
319                 if (!boost::algorithm::starts_with(name, "loop") && (!device_type || *device_type != "5")) {
320                         uint64_t const size = dcp::raw_convert<uint64_t>(dcp::file_to_string(*i / "size")) * 512;
321                         if (size == 0) {
322                                 continue;
323                         }
324                         optional<string> vendor;
325                         try {
326                                 vendor = dcp::file_to_string("/sys/block/" + name + "/device/vendor");
327                                 boost::trim(*vendor);
328                         } catch (...) {}
329                         optional<string> model;
330                         try {
331                                 model = dcp::file_to_string("/sys/block/" + name + "/device/model");
332                                 boost::trim(*model);
333                         } catch (...) {}
334                         vector<boost::filesystem::path> mount_points;
335                         for (vector<pair<string, string> >::const_iterator j = mounted_devices.begin(); j != mounted_devices.end(); ++j) {
336                                 if (boost::algorithm::starts_with(j->first, "/dev/" + name)) {
337                                         mount_points.push_back (j->second);
338                                 }
339                         }
340                         drives.push_back(Drive("/dev/" + name, mount_points, size, vendor, model));
341                         LOG_DISK_NC(drives.back().log_summary());
342                 }
343         }
344
345         return drives;
346 }
347
348
349 bool
350 Drive::unmount ()
351 {
352         for (auto i: _mount_points) {
353                 int const r = umount(i.string().c_str());
354                 LOG_DISK("Tried to unmount %1 and got %2 and %3", i.string(), r, errno);
355                 if (r == -1) {
356                         return false;
357                 }
358         }
359         return true;
360 }
361
362
363 void
364 unprivileged ()
365 {
366         uid_t ruid, euid, suid;
367         if (getresuid(&ruid, &euid, &suid) == -1) {
368                 cerr << "getresuid() failed.\n";
369         }
370         if (seteuid(ruid) == -1) {
371                 cerr << "seteuid() failed.\n";
372         }
373 }
374
375
376 bool PrivilegeEscalator::test = false;
377
378 PrivilegeEscalator::~PrivilegeEscalator ()
379 {
380         if (!test) {
381                 unprivileged ();
382         }
383 }
384
385 PrivilegeEscalator::PrivilegeEscalator ()
386 {
387         if (!test) {
388                 int const r = seteuid(0);
389                 if (r < 0) {
390                         throw PrivilegeError (String::compose("seteuid() call failed with %1", errno));
391                 }
392         }
393 }
394
395 boost::filesystem::path
396 config_path ()
397 {
398         boost::filesystem::path p;
399         p /= g_get_user_config_dir ();
400         p /= "dcpomatic2";
401         return p;
402 }
403
404
405 void
406 disk_write_finished ()
407 {
408
409 }
410
411 string
412 dcpomatic::get_process_id ()
413 {
414         return dcp::raw_convert<string>(getpid());
415 }