2 Copyright (C) 2012-2020 Carl Hetherington <cth@carlh.net>
4 This file is part of DCP-o-matic.
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.
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.
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/>.
22 #include "compose.hpp"
24 #include "dcpomatic_log.h"
26 #include "exceptions.h"
27 #include "dcpomatic_log.h"
28 #include <dcp/raw_convert.h>
31 #include <libavformat/avio.h>
33 #include <boost/algorithm/string.hpp>
34 #include <boost/foreach.hpp>
35 #include <boost/function.hpp>
36 #if BOOST_VERSION >= 106100
37 #include <boost/dll/runtime_symbol_info.hpp>
41 #include <sys/types.h>
42 #include <sys/mount.h>
44 #include <netinet/in.h>
45 #include <arpa/inet.h>
59 using std::runtime_error;
60 using boost::shared_ptr;
61 using boost::optional;
62 using boost::function;
64 /** @param s Number of seconds to sleep for */
66 dcpomatic_sleep_seconds (int s)
72 dcpomatic_sleep_milliseconds (int ms)
77 /** @return A string of CPU information (model name etc.) */
83 /* This use of ifstream is ok; the filename can never
86 ifstream f ("/proc/cpuinfo");
90 if (boost::algorithm::starts_with (l, "model name")) {
91 string::size_type const c = l.find (':');
92 if (c != string::npos) {
93 info = l.substr (c + 2);
101 boost::filesystem::path
104 char const * p = getenv ("DCPOMATIC_LINUX_SHARE_PREFIX");
106 return boost::filesystem::path(p) / "dcpomatic2";
108 return boost::filesystem::canonical(LINUX_SHARE_PREFIX) / "dcpomatic2";
112 boost::filesystem::path
115 char const * p = getenv ("DCPOMATIC_LINUX_SHARE_PREFIX");
117 return boost::filesystem::path(p) / "libdcp" / "xsd";
119 return boost::filesystem::canonical(LINUX_SHARE_PREFIX) / "libdcp" / "xsd";
124 run_ffprobe (boost::filesystem::path content, boost::filesystem::path out)
126 string ffprobe = "ffprobe \"" + content.string() + "\" 2> \"" + out.string() + "\"";
127 LOG_GENERAL (N_("Probing with %1"), ffprobe);
128 int const r = system (ffprobe.c_str());
129 if (r == -1 || (WIFEXITED(r) && WEXITSTATUS(r) != 0)) {
130 LOG_GENERAL (N_("Could not run ffprobe (system returned %1"), r);
134 list<pair<string, string> >
137 list<pair<string, string> > m;
139 FILE* f = setmntent ("/etc/mtab", "r");
145 struct mntent* mnt = getmntent (f);
150 m.push_back (make_pair (mnt->mnt_dir, mnt->mnt_type));
159 boost::filesystem::path
160 directory_containing_executable ()
162 #if BOOST_VERSION >= 106100
163 return boost::dll::program_location().parent_path();
165 char buffer[PATH_MAX];
166 ssize_t N = readlink ("/proc/self/exe", buffer, PATH_MAX);
167 return boost::filesystem::path(string(buffer, N)).parent_path();
172 boost::filesystem::path
175 boost::filesystem::path p = directory_containing_executable() / "dcpomatic2_openssl";
176 if (boost::filesystem::is_regular_file(p)) {
180 return "dcpomatic2_openssl";
184 #ifdef DCPOMATIC_DISK
185 boost::filesystem::path
188 return directory_containing_executable() / "dcpomatic2_disk_writer";
193 /* Apparently there is no way to create an ofstream using a UTF-8
194 filename under Windows. We are hence reduced to using fopen
198 fopen_boost (boost::filesystem::path p, string t)
200 return fopen (p.c_str(), t.c_str ());
204 dcpomatic_fseek (FILE* stream, int64_t offset, int whence)
206 return fseek (stream, offset, whence);
227 start_tool (string executable)
229 boost::filesystem::path batch = directory_containing_executable() / executable;
233 int const r = system (batch.string().c_str());
234 exit (WEXITSTATUS (r));
240 start_batch_converter ()
242 start_tool ("dcpomatic2_batch");
249 start_tool ("dcpomatic2_player");
256 return (uint64_t) pthread_self ();
260 avio_open_boost (AVIOContext** s, boost::filesystem::path file, int flags)
262 return avio_open (s, file.c_str(), flags);
266 boost::filesystem::path
269 return getenv("HOME");
273 command_and_read (string cmd)
275 FILE* pipe = popen (cmd.c_str(), "r");
277 throw runtime_error ("popen failed");
283 while (fgets(buffer, sizeof(buffer), pipe)) {
295 /** @return true if this process is a 32-bit one running on a 64-bit-capable OS */
299 /* I'm assuming nobody does this on Linux */
305 vector<pair<string, string> >
306 get_mounts (string prefix)
308 vector<pair<string, string> > mounts;
310 std::ifstream f("/proc/mounts");
315 boost::algorithm::split (bits, line, boost::is_any_of(" "));
316 if (bits.size() > 1 && boost::algorithm::starts_with(bits[0], prefix)) {
317 boost::algorithm::replace_all (bits[1], "\\040", " ");
318 mounts.push_back(make_pair(bits[0], bits[1]));
319 LOG_DISK("Found mounted device %1 from prefix %2", bits[0], prefix);
330 vector<Drive> drives;
332 using namespace boost::filesystem;
333 vector<pair<string, string> > mounted_devices = get_mounts("/dev/");
335 for (directory_iterator i = directory_iterator("/sys/block"); i != directory_iterator(); ++i) {
336 string const name = i->path().filename().string();
337 path device_type_file("/sys/block/" + name + "/device/type");
338 optional<string> device_type;
339 if (exists(device_type_file)) {
340 device_type = dcp::file_to_string (device_type_file);
341 boost::trim(*device_type);
343 /* Device type 5 is "SCSI_TYPE_ROM" in blkdev.h; seems usually to be a CD/DVD drive */
344 if (!boost::algorithm::starts_with(name, "loop") && (!device_type || *device_type != "5")) {
345 uint64_t const size = dcp::raw_convert<uint64_t>(dcp::file_to_string(*i / "size")) * 512;
349 optional<string> vendor;
351 vendor = dcp::file_to_string("/sys/block/" + name + "/device/vendor");
352 boost::trim(*vendor);
354 optional<string> model;
356 model = dcp::file_to_string("/sys/block/" + name + "/device/model");
359 vector<boost::filesystem::path> mount_points;
360 for (vector<pair<string, string> >::const_iterator j = mounted_devices.begin(); j != mounted_devices.end(); ++j) {
361 if (boost::algorithm::starts_with(j->first, "/dev/" + name)) {
362 mount_points.push_back (j->second);
365 drives.push_back(Drive("/dev/" + name, mount_points, size, vendor, model));
366 LOG_DISK_NC(drives.back().log_summary());
377 BOOST_FOREACH (boost::filesystem::path i, _mount_points) {
378 int const r = umount(i.string().c_str());
379 LOG_DISK("Tried to unmount %1 and got %2 and %3", i.string(), r, errno);
391 uid_t ruid, euid, suid;
392 if (getresuid(&ruid, &euid, &suid) == -1) {
393 cerr << "getresuid() failed.\n";
395 if (seteuid(ruid) == -1) {
396 cerr << "seteuid() failed.\n";
400 PrivilegeEscalator::~PrivilegeEscalator ()
405 PrivilegeEscalator::PrivilegeEscalator ()
407 int const r = seteuid(0);
409 throw PrivilegeError (String::compose("seteuid() call failed with %1", errno));
413 boost::filesystem::path
416 boost::filesystem::path p;
417 p /= g_get_user_config_dir ();
424 disk_write_finished ()