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 <dcp/raw_convert.h>
  28 #include <glib.h>
  29 extern "C" {
  30 #include <libavformat/avio.h>
  31 }
  32 #include <boost/algorithm/string.hpp>
  33 #include <boost/foreach.hpp>
  34 #include <unistd.h>
  35 #include <mntent.h>
  36 #include <sys/types.h>
  37 #include <ifaddrs.h>
  38 #include <netinet/in.h>
  39 #include <arpa/inet.h>
  40 #include <fstream>
  41
  42 #include "i18n.h"
  43
  44 using std::pair;
  45 using std::list;
  46 using std::ifstream;
  47 using std::string;
  48 using std::wstring;
  49 using std::make_pair;
  50 using std::vector;
  51 using std::cerr;
  52 using std::cout;
  53 using std::runtime_error;
  54 using boost::shared_ptr;
  55 using boost::optional;
  56
  57 /** @param s Number of seconds to sleep for */
  58 void
  59 dcpomatic_sleep_seconds (int s)
  60 {
  61         sleep (s);
  62 }
  63
  64 void
  65 dcpomatic_sleep_milliseconds (int ms)
  66 {
  67         usleep (ms * 1000);
  68 }
  69
  70 /** @return A string of CPU information (model name etc.) */
  71 string
  72 cpu_info ()
  73 {
  74         string info;
  75
  76         /* This use of ifstream is ok; the filename can never
  77            be non-Latin
  78         */
  79         ifstream f ("/proc/cpuinfo");
  80         while (f.good ()) {
  81                 string l;
  82                 getline (f, l);
  83                 if (boost::algorithm::starts_with (l, "model name")) {
  84                         string::size_type const c = l.find (':');
  85                         if (c != string::npos) {
  86                                 info = l.substr (c + 2);
  87                         }
  88                 }
  89         }
  90
  91         return info;
  92 }
  93
  94 boost::filesystem::path
  95 shared_path ()
  96 {
  97         char const * p = getenv ("DCPOMATIC_LINUX_SHARE_PREFIX");
  98         if (p) {
  99                 return p;
 100         }
 101         return boost::filesystem::canonical (LINUX_SHARE_PREFIX);
 102 }
 103
 104 void
 105 run_ffprobe (boost::filesystem::path content, boost::filesystem::path out)
 106 {
 107         string ffprobe = "ffprobe \"" + content.string() + "\" 2> \"" + out.string() + "\"";
 108         LOG_GENERAL (N_("Probing with %1"), ffprobe);
 109         system (ffprobe.c_str ());
 110 }
 111
 112 list<pair<string, string> >
 113 mount_info ()
 114 {
 115         list<pair<string, string> > m;
 116
 117         FILE* f = setmntent ("/etc/mtab", "r");
 118         if (!f) {
 119                 return m;
 120         }
 121
 122         while (true) {
 123                 struct mntent* mnt = getmntent (f);
 124                 if (!mnt) {
 125                         break;
 126                 }
 127
 128                 m.push_back (make_pair (mnt->mnt_dir, mnt->mnt_type));
 129         }
 130
 131         endmntent (f);
 132
 133         return m;
 134 }
 135
 136 boost::filesystem::path
 137 openssl_path ()
 138 {
 139         return "dcpomatic2_openssl";
 140 }
 141
 142 boost::filesystem::path
 143 dist_writer_path ()
 144 {
 145         return "dcpomatic2_dist_writer";
 146 }
 147
 148 /* Apparently there is no way to create an ofstream using a UTF-8
 149    filename under Windows.  We are hence reduced to using fopen
 150    with this wrapper.
 151 */
 152 FILE *
 153 fopen_boost (boost::filesystem::path p, string t)
 154 {
 155         return fopen (p.c_str(), t.c_str ());
 156 }
 157
 158 int
 159 dcpomatic_fseek (FILE* stream, int64_t offset, int whence)
 160 {
 161         return fseek (stream, offset, whence);
 162 }
 163
 164 void
 165 Waker::nudge ()
 166 {
 167
 168 }
 169
 170 Waker::Waker ()
 171 {
 172
 173 }
 174
 175 Waker::~Waker ()
 176 {
 177
 178 }
 179
 180 void
 181 start_tool (boost::filesystem::path dcpomatic, string executable, string)
 182 {
 183         boost::filesystem::path batch = dcpomatic.parent_path() / executable;
 184
 185         pid_t pid = fork ();
 186         if (pid == 0) {
 187                 int const r = system (batch.string().c_str());
 188                 exit (WEXITSTATUS (r));
 189         }
 190 }
 191
 192 void
 193 start_batch_converter (boost::filesystem::path dcpomatic)
 194 {
 195         start_tool (dcpomatic, "dcpomatic2_batch", "DCP-o-matic\\ 2\\ Batch\\ Converter.app");
 196 }
 197
 198 void
 199 start_player (boost::filesystem::path dcpomatic)
 200 {
 201         start_tool (dcpomatic, "dcpomatic2_player", "DCP-o-matic\\ 2\\ Player.app");
 202 }
 203
 204 uint64_t
 205 thread_id ()
 206 {
 207         return (uint64_t) pthread_self ();
 208 }
 209
 210 int
 211 avio_open_boost (AVIOContext** s, boost::filesystem::path file, int flags)
 212 {
 213         return avio_open (s, file.c_str(), flags);
 214 }
 215
 216
 217 boost::filesystem::path
 218 home_directory ()
 219 {
 220                 return getenv("HOME");
 221 }
 222
 223 string
 224 command_and_read (string cmd)
 225 {
 226         FILE* pipe = popen (cmd.c_str(), "r");
 227         if (!pipe) {
 228                 throw runtime_error ("popen failed");
 229         }
 230
 231         string result;
 232         char buffer[128];
 233         try {
 234                 while (fgets(buffer, sizeof(buffer), pipe)) {
 235                         result += buffer;
 236                 }
 237         } catch (...) {
 238                 pclose (pipe);
 239                 throw;
 240         }
 241
 242         pclose (pipe);
 243         return result;
 244 }
 245
 246 /** @return true if this process is a 32-bit one running on a 64-bit-capable OS */
 247 bool
 248 running_32_on_64 ()
 249 {
 250         /* I'm assuming nobody does this on Linux */
 251         return false;
 252 }
 253
 254 vector<Drive>
 255 get_drives ()
 256 {
 257         vector<Drive> drives;
 258
 259         using namespace boost::filesystem;
 260         list<string> mounted_devices;
 261         std::ifstream f("/proc/mounts");
 262         string line;
 263         while (f.good()) {
 264                 getline(f, line);
 265                 vector<string> bits;
 266                 boost::algorithm::split (bits, line, boost::is_any_of(" "));
 267                 if (bits.size() > 0 && boost::algorithm::starts_with(bits[0], "/dev/")) {
 268                         mounted_devices.push_back(bits[0]);
 269                         LOG_DIST("Mounted device %1", bits[0]);
 270                 }
 271         }
 272
 273         for (directory_iterator i = directory_iterator("/sys/block"); i != directory_iterator(); ++i) {
 274                 string const name = i->path().filename().string();
 275                 path device_type_file("/sys/block/" + name + "/device/type");
 276                 optional<string> device_type;
 277                 if (exists(device_type_file)) {
 278                         device_type = dcp::file_to_string (device_type_file);
 279                         boost::trim(*device_type);
 280                 }
 281                 /* Device type 5 is "SCSI_TYPE_ROM" in blkdev.h; seems usually to be a CD/DVD drive */
 282                 if (!boost::algorithm::starts_with(name, "loop") && (!device_type || *device_type != "5")) {
 283                         uint64_t const size = dcp::raw_convert<uint64_t>(dcp::file_to_string(*i / "size")) * 512;
 284                         if (size == 0) {
 285                                 continue;
 286                         }
 287                         bool mounted = false;
 288                         optional<string> vendor;
 289                         try {
 290                                 vendor = dcp::file_to_string("/sys/block/" + name + "/device/vendor");
 291                                 boost::trim(*vendor);
 292                         } catch (...) {}
 293                         optional<string> model;
 294                         try {
 295                                 model = dcp::file_to_string("/sys/block/" + name + "/device/model");
 296                                 boost::trim(*model);
 297                         } catch (...) {}
 298                         BOOST_FOREACH (string j, mounted_devices) {
 299                                 if (boost::algorithm::starts_with(j, "/dev/" + name)) {
 300                                         mounted = true;
 301                                 }
 302                         }
 303                         drives.push_back(Drive(i->path().filename().string(), size, mounted, vendor, model));
 304                         LOG_DIST("Block device %1 size %2 %3 vendor %4 model %5", name, size, mounted ? "mounted" : "not mounted", vendor.get_value_or("[none]"), model.get_value_or("[none]"));
 305                 }
 306         }
 307
 308         return drives;
 309 }
 310
 311 string
 312 Drive::description () const
 313 {
 314         char gb[64];
 315         snprintf(gb, 64, "%.1f", _size / 1000000000.0);
 316
 317         string name;
 318         if (_vendor) {
 319                 name += *_vendor;
 320         }
 321         if (_model) {
 322                 if (name.size() > 0) {
 323                         name += " " + *_model;
 324                 }
 325         }
 326         if (name.size() == 0) {
 327                 name = _("Unknown");
 328         }
 329
 330         return String::compose("%1 (%2 GB) [%3]", name, gb, _internal_name);
 331 }
 332
 333 void
 334 unprivileged ()
 335 {
 336         uid_t ruid, euid, suid;
 337         if (getresuid(&ruid, &euid, &suid) == -1) {
 338                 cerr << "getresuid() failed.\n";
 339                 exit (EXIT_FAILURE);
 340         }
 341         seteuid (ruid);
 342 }
 343
 344 PrivilegeEscalator::~PrivilegeEscalator ()
 345 {
 346         unprivileged ();
 347 }
 348
 349 PrivilegeEscalator::PrivilegeEscalator ()
 350 {
 351         seteuid (0);
 352 }
 353
 354 boost::filesystem::path
 355 config_path ()
 356 {
 357         boost::filesystem::path p;
 358         p /= g_get_user_config_dir ();
 359         p /= "dcpomatic2";
 360         return p;
 361 }