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 <dcp/raw_convert.h>
30 #include <libavformat/avio.h>
32 #include <boost/algorithm/string.hpp>
33 #include <boost/foreach.hpp>
34 #include <boost/regex.hpp>
35 #include <sys/sysctl.h>
36 #include <mach-o/dyld.h>
37 #include <IOKit/pwr_mgt/IOPMLib.h>
38 #include <IOKit/storage/IOMedia.h>
39 #include <DiskArbitration/DADisk.h>
40 #include <DiskArbitration/DiskArbitration.h>
41 #include <CoreFoundation/CFURL.h>
42 #include <sys/types.h>
44 #include <netinet/in.h>
45 #include <arpa/inet.h>
60 using std::runtime_error;
61 using boost::shared_ptr;
62 using boost::optional;
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.) */
84 size_t N = sizeof (buffer);
85 if (sysctlbyname ("machdep.cpu.brand_string", buffer, &N, 0, 0) == 0) {
92 /** @return Path of the Contents directory in the .app */
93 boost::filesystem::path
96 /* Could use boost::dll::program_location().parent_path().parent_path() but that
97 * boost library isn't in the 10.6 environment we're currently using and I don't
98 * really want to change it right now.
100 uint32_t size = 1024;
102 if (_NSGetExecutablePath (buffer, &size)) {
103 throw runtime_error ("_NSGetExecutablePath failed");
106 boost::filesystem::path path (buffer);
107 path = boost::filesystem::canonical (path);
108 path = path.parent_path ();
109 path = path.parent_path ();
113 boost::filesystem::path
116 return app_contents() / "Resources";
120 run_ffprobe (boost::filesystem::path content, boost::filesystem::path out)
122 boost::filesystem::path path = app_contents();
126 string ffprobe = "\"" + path.string() + "\" \"" + content.string() + "\" 2> \"" + out.string() + "\"";
127 LOG_GENERAL (N_("Probing with %1"), ffprobe);
128 system (ffprobe.c_str ());
131 list<pair<string, string> >
134 list<pair<string, string> > m;
138 boost::filesystem::path
141 boost::filesystem::path path = app_contents();
147 #ifdef DCPOMATIC_DISK
148 boost::filesystem::path
151 boost::filesystem::path path = app_contents();
153 path /= "dcpomatic2_disk_writer";
158 /* Apparently there is no way to create an ofstream using a UTF-8
159 filename under Windows. We are hence reduced to using fopen
163 fopen_boost (boost::filesystem::path p, string t)
165 return fopen (p.c_str(), t.c_str ());
169 dcpomatic_fseek (FILE* stream, int64_t offset, int whence)
171 return fseek (stream, offset, whence);
182 boost::mutex::scoped_lock lm (_mutex);
183 /* We should use this */
184 // IOPMAssertionCreateWithName (kIOPMAssertionTypeNoIdleSleep, kIOPMAssertionLevelOn, CFSTR ("Encoding DCP"), &_assertion_id);
185 /* but it's not available on 10.5, so we use this */
186 IOPMAssertionCreate (kIOPMAssertionTypeNoIdleSleep, kIOPMAssertionLevelOn, &_assertion_id);
191 boost::mutex::scoped_lock lm (_mutex);
192 IOPMAssertionRelease (_assertion_id);
196 start_tool (boost::filesystem::path dcpomatic, string executable, string app)
198 boost::filesystem::path batch = dcpomatic.parent_path ();
199 batch = batch.parent_path (); // MacOS
200 batch = batch.parent_path (); // Contents
201 batch = batch.parent_path (); // DCP-o-matic.app
202 batch = batch.parent_path (); // Applications
210 int const r = system (batch.string().c_str());
211 exit (WEXITSTATUS (r));
216 start_batch_converter (boost::filesystem::path dcpomatic)
218 start_tool (dcpomatic, "dcpomatic2_batch", "DCP-o-matic\\ 2\\ Batch\\ Converter.app");
222 start_player (boost::filesystem::path dcpomatic)
224 start_tool (dcpomatic, "dcpomatic2_player", "DCP-o-matic\\ 2\\ Player.app");
230 return (uint64_t) pthread_self ();
234 avio_open_boost (AVIOContext** s, boost::filesystem::path file, int flags)
236 return avio_open (s, file.c_str(), flags);
239 boost::filesystem::path
242 return getenv("HOME");
246 command_and_read (string cmd)
251 /** @return true if this process is a 32-bit one running on a 64-bit-capable OS */
255 /* I'm assuming nobody does this on OS X */
259 static optional<string>
260 get_vendor (CFDictionaryRef& description)
262 void const* str = CFDictionaryGetValue (description, kDADiskDescriptionDeviceVendorKey);
264 return optional<string>();
267 string s = CFStringGetCStringPtr ((CFStringRef) str, kCFStringEncodingUTF8);
268 boost::algorithm::trim (s);
272 static optional<string>
273 get_model (CFDictionaryRef& description)
275 void const* str = CFDictionaryGetValue (description, kDADiskDescriptionDeviceModelKey);
277 return optional<string>();
280 string s = CFStringGetCStringPtr ((CFStringRef) str, kCFStringEncodingUTF8);
281 boost::algorithm::trim (s);
287 bool real; ///< true for a "real" disk, false for a synthesized APFS one
288 std::string prt; ///< "PRT" entry from the media path
291 static optional<MediaPath>
292 analyse_media_path (CFDictionaryRef& description)
294 using namespace boost::algorithm;
296 void const* str = CFDictionaryGetValue (description, kDADiskDescriptionMediaPathKey);
298 return optional<MediaPath>();
301 string path(CFStringGetCStringPtr((CFStringRef) str, kCFStringEncodingUTF8));
303 if (starts_with(path, "IODeviceTree:")) {
305 } else if (starts_with(path, "IOService:")) {
308 return optional<MediaPath>();
312 split(bits, path, boost::is_any_of("/"));
313 BOOST_FOREACH (string i, bits) {
314 if (starts_with(i, "PRT")) {
323 is_whole_drive (DADiskRef& disk)
325 io_service_t service = DADiskCopyIOMedia (disk);
326 CFTypeRef whole_media_ref = IORegistryEntryCreateCFProperty (service, CFSTR(kIOMediaWholeKey), kCFAllocatorDefault, 0);
327 bool whole_media = false;
328 if (whole_media_ref) {
329 whole_media = CFBooleanGetValue((CFBooleanRef) whole_media_ref);
330 CFRelease (whole_media_ref);
332 IOObjectRelease (service);
337 is_mounted (CFDictionaryRef& description)
339 CFURLRef volume_path_key = (CFURLRef) CFDictionaryGetValue (description, kDADiskDescriptionVolumePathKey);
340 char mount_path_buffer[1024];
341 return CFURLGetFileSystemRepresentation(volume_path_key, false, (UInt8 *) mount_path_buffer, sizeof(mount_path_buffer));
344 /* Here follows some rather intricate and (probably) fragile code to find the list of available
345 * "real" drives on macOS that we might want to write a DCP to.
347 * We use the Disk Arbitration framework to give us a series of devices (/dev/disk0, /dev/disk1,
348 * /dev/disk1s1 and so on) and we use the API to gather useful information about these devices into
349 * a vector of Disk structs.
351 * Then we read the Disks that we found and try to derive a list of drives that we should offer to the
352 * user, with details of whether those drives are currently mounted or not.
354 * At the basic level we find the "disk"-level devices, looking at whether any of their partitions are mounted.
356 * This is complicated enormously by recent-ish macOS versions' habit of making `synthesized' volumes which
357 * reflect data in `real' partitions. So, for example, we might have a real (physical) drive /dev/disk2 with
358 * a partition /dev/disk2s2 whose content is made into a synthesized /dev/disk3, itself containing some partitions
359 * which are mounted. /dev/disk2s2 is not considered to be mounted, in this case. So we need to know that
360 * disk2s2 is related to disk3 so we can consider disk2s2 as mounted if any parts of disk3 are. In order to do
361 * this I am picking out what looks like a suitable identifier prefixed with PRT from the MediaContentKey.
362 * If disk2s2 and disk3 have the same PRT code I am assuming they are linked.
364 * Lots of this is guesswork and may be broken. In my defence the documentation that I have been able to
365 * unearth is, to put it impolitely, crap.
371 optional<string> vendor;
372 optional<string> model;
381 disk_appeared (DADiskRef disk, void* context)
383 const char* bsd_name = DADiskGetBSDName (disk);
387 LOG_DISK("%1 appeared", bsd_name);
391 this_disk.device = string("/dev/") + bsd_name;
393 CFDictionaryRef description = DADiskCopyDescription (disk);
395 this_disk.vendor = get_vendor (description);
396 this_disk.model = get_model (description);
397 LOG_DISK("Vendor/model: %1 %2", this_disk.vendor.get_value_or("[none]"), this_disk.model.get_value_or("[none]"));
399 optional<MediaPath> media_path = analyse_media_path (description);
401 LOG_DISK("Finding media path for %1 failed", bsd_name);
405 this_disk.real = media_path->real;
406 this_disk.prt = media_path->prt;
407 this_disk.whole = is_whole_drive (disk);
408 this_disk.mounted = is_mounted (description);
409 LOG_DISK("%1 prt %2 whole %3 mounted %4", this_disk.real ? "Real" : "Synth", this_disk.prt, this_disk.whole ? "whole" : "part", this_disk.mounted ? "mounted" : "unmounted");
411 CFNumberGetValue ((CFNumberRef) CFDictionaryGetValue (description, kDADiskDescriptionMediaSizeKey), kCFNumberLongType, &this_disk.size);
412 CFRelease (description);
414 reinterpret_cast<vector<Disk>*>(context)->push_back(this_disk);
420 using namespace boost::algorithm;
423 DASessionRef session = DASessionCreate(kCFAllocatorDefault);
425 return vector<Drive>();
428 DARegisterDiskAppearedCallback (session, NULL, disk_appeared, &disks);
429 CFRunLoopRef run_loop = CFRunLoopGetCurrent ();
430 DASessionScheduleWithRunLoop (session, run_loop, kCFRunLoopDefaultMode);
431 CFRunLoopStop (run_loop);
432 CFRunLoopRunInMode(kCFRunLoopDefaultMode, 0.05, 0);
433 DAUnregisterCallback(session, (void *) disk_appeared, &disks);
436 /* Mark disks containing mounted partitions as themselves mounted */
437 BOOST_FOREACH (Disk& i, disks) {
441 BOOST_FOREACH (Disk& j, disks) {
442 if (j.mounted && starts_with(j.device, i.device)) {
443 LOG_DISK("Marking %1 as mounted because %2 is", i.device, j.device);
449 /* Make a list of the PRT codes of mounted, synthesized disks */
450 vector<string> mounted_synths;
451 BOOST_FOREACH (Disk& i, disks) {
452 if (!i.real && i.mounted) {
453 LOG_DISK("Found a mounted synth %1 with %2", i.device, i.prt);
454 mounted_synths.push_back (i.prt);
458 /* Mark containers of those mounted synths as themselves mounted */
459 BOOST_FOREACH (Disk& i, disks) {
460 if (i.real && find(mounted_synths.begin(), mounted_synths.end(), i.prt) != mounted_synths.end()) {
461 LOG_DISK("Marking %1 (%2) as mounted because it contains a mounted synth", i.device, i.prt);
466 vector<Drive> drives;
467 BOOST_FOREACH (Disk& i, disks) {
469 /* A whole disk that is not a container for a mounted synth */
470 LOG_DISK("Adding drive: %1 %2 %3 %4 %5", i.device, i.size, i.mounted ? "mounted" : "unmounted", i.vendor.get_value_or("[none]"), i.model.get_value_or("[none]"));
471 drives.push_back(Drive(i.device, i.size, i.mounted, i.vendor, i.model));
477 boost::filesystem::path
480 boost::filesystem::path p;
481 p /= g_get_home_dir ();
484 p /= "com.dcpomatic";