X-Git-Url: https://git.carlh.net/gitweb/?a=blobdiff_plain;f=src%2Ftools%2Fdcpomatic_disk_writer.cc;h=ef3bf2f776c86936b6ad443e26b3d717eca1d26b;hb=25eb7aa723af004162fc7f95995df250c4e4571b;hp=a31a7673f977e1242282300cbc0092b13408ebc3;hpb=b2c1eb46888c3a606e751c037b6482306a461d5f;p=dcpomatic.git diff --git a/src/tools/dcpomatic_disk_writer.cc b/src/tools/dcpomatic_disk_writer.cc index a31a7673f..ef3bf2f77 100644 --- a/src/tools/dcpomatic_disk_writer.cc +++ b/src/tools/dcpomatic_disk_writer.cc @@ -27,9 +27,11 @@ #include "lib/exceptions.h" #include "lib/ext.h" #include "lib/file_log.h" +#include "lib/state.h" #include "lib/nanomsg.h" +#include "lib/util.h" #include "lib/version.h" -#include "lib/warnings.h" +#include #ifdef DCPOMATIC_POSIX #include @@ -43,6 +45,7 @@ extern "C" { #include } +#include #include #endif @@ -57,9 +60,9 @@ extern "C" { } #endif -DCPOMATIC_DISABLE_WARNINGS +LIBDCP_DISABLE_WARNINGS #include -DCPOMATIC_ENABLE_WARNINGS +LIBDCP_ENABLE_WARNINGS #include #include @@ -87,20 +90,11 @@ static PolkitAuthority* polkit_authority = nullptr; static Nanomsg* nanomsg = nullptr; -struct Parameters -{ - boost::filesystem::path dcp_path; - std::string device; - std::string posix_partition; -}; - - #ifdef DCPOMATIC_LINUX -static void polkit_callback (GObject *, GAsyncResult* res, gpointer data) { - auto parameters = reinterpret_cast (data); + auto parameters = reinterpret_cast, std::function>*> (data); GError* error = nullptr; auto result = polkit_authority_check_authorization_finish (polkit_authority, res, &error); bool failed = false; @@ -110,7 +104,7 @@ polkit_callback (GObject *, GAsyncResult* res, gpointer data) failed = true; } else { if (polkit_authorization_result_get_is_authorized(result)) { - dcpomatic::write (parameters->dcp_path, parameters->device, parameters->posix_partition, nanomsg); + parameters->first(); } else { failed = true; if (polkit_authorization_result_get_is_challenge(result)) { @@ -121,8 +115,8 @@ polkit_callback (GObject *, GAsyncResult* res, gpointer data) } } - if (failed && nanomsg) { - nanomsg->send(DISK_WRITER_ERROR "\nCould not obtain authorization to write to the drive\n", LONG_TIMEOUT); + if (failed) { + parameters->second(); } delete parameters; @@ -134,6 +128,26 @@ polkit_callback (GObject *, GAsyncResult* res, gpointer data) #endif +#ifdef DCPOMATIC_LINUX +void request_privileges (string action, std::function granted, std::function denied) +#else +void request_privileges (string, std::function granted, std::function) +#endif +{ +#ifdef DCPOMATIC_LINUX + polkit_authority = polkit_authority_get_sync (0, 0); + auto subject = polkit_unix_process_new_for_owner (getppid(), 0, -1); + + auto parameters = new std::pair, std::function>(granted, denied); + polkit_authority_check_authorization ( + polkit_authority, subject, action.c_str(), 0, POLKIT_CHECK_AUTHORIZATION_FLAGS_ALLOW_USER_INTERACTION, 0, polkit_callback, parameters + ); +#else + granted (); +#endif +} + + bool idle () try @@ -152,45 +166,75 @@ try } else if (*s == DISK_WRITER_PING) { nanomsg->send(DISK_WRITER_PONG "\n", LONG_TIMEOUT); } else if (*s == DISK_WRITER_UNMOUNT) { - /* XXX: should do Linux polkit stuff here */ auto xml_head = nanomsg->receive (LONG_TIMEOUT); auto xml_body = nanomsg->receive (LONG_TIMEOUT); if (!xml_head || !xml_body) { LOG_DISK_NC("Failed to receive unmount request"); throw CommunicationFailedError (); } - bool const success = Drive(*xml_head + *xml_body).unmount(); - if (!nanomsg->send (success ? (DISK_WRITER_OK "\n") : (DISK_WRITER_ERROR "\n"), LONG_TIMEOUT)) { - LOG_DISK_NC("CommunicationFailedError in unmount_finished"); - throw CommunicationFailedError (); - } + auto xml = *xml_head + *xml_body; + request_privileges ( + "com.dcpomatic.write-drive", + [xml]() { + bool const success = Drive(xml).unmount(); + bool sent_reply = false; + if (success) { + sent_reply = nanomsg->send(DISK_WRITER_OK "\n", LONG_TIMEOUT); + } else { + sent_reply = nanomsg->send(DISK_WRITER_ERROR "\nCould not unmount drive\n1\n", LONG_TIMEOUT); + } + if (!sent_reply) { + LOG_DISK_NC("CommunicationFailedError in unmount_finished"); + throw CommunicationFailedError (); + } + }, + []() { + if (!nanomsg->send(DISK_WRITER_ERROR "\nCould not get permission to unmount drive\n1\n", LONG_TIMEOUT)) { + LOG_DISK_NC("CommunicationFailedError in unmount_finished"); + throw CommunicationFailedError (); + } + }); } else if (*s == DISK_WRITER_WRITE) { - auto dcp_path = nanomsg->receive (LONG_TIMEOUT); - auto device = nanomsg->receive (LONG_TIMEOUT); - if (!dcp_path || !device) { + auto device_opt = nanomsg->receive (LONG_TIMEOUT); + if (!device_opt) { LOG_DISK_NC("Failed to receive write request"); throw CommunicationFailedError(); } + auto device = *device_opt; + + vector dcp_paths; + while (true) { + auto dcp_path_opt = nanomsg->receive (LONG_TIMEOUT); + if (!dcp_path_opt) { + LOG_DISK_NC("Failed to receive write request"); + throw CommunicationFailedError(); + } + if (*dcp_path_opt != "") { + dcp_paths.push_back(*dcp_path_opt); + } else { + break; + } + } /* Do some basic sanity checks; this is a bit belt-and-braces but it can't hurt... */ #ifdef DCPOMATIC_OSX - if (!starts_with(*device, "/dev/disk")) { - LOG_DISK ("Will not write to %1", *device); + if (!starts_with(device, "/dev/disk")) { + LOG_DISK ("Will not write to %1", device); nanomsg->send(DISK_WRITER_ERROR "\nRefusing to write to this drive\n1\n", LONG_TIMEOUT); return true; } #endif #ifdef DCPOMATIC_LINUX - if (!starts_with(*device, "/dev/sd") && !starts_with(*device, "/dev/hd")) { - LOG_DISK ("Will not write to %1", *device); + if (!starts_with(device, "/dev/sd") && !starts_with(device, "/dev/hd")) { + LOG_DISK ("Will not write to %1", device); nanomsg->send(DISK_WRITER_ERROR "\nRefusing to write to this drive\n1\n", LONG_TIMEOUT); return true; } #endif #ifdef DCPOMATIC_WINDOWS - if (!starts_with(*device, "\\\\.\\PHYSICALDRIVE")) { - LOG_DISK ("Will not write to %1", *device); + if (!starts_with(device, "\\\\.\\PHYSICALDRIVE")) { + LOG_DISK ("Will not write to %1", device); nanomsg->send(DISK_WRITER_ERROR "\nRefusing to write to this drive\n1\n", LONG_TIMEOUT); return true; } @@ -199,47 +243,52 @@ try bool on_drive_list = false; bool mounted = false; for (auto const& i: Drive::get()) { - if (i.device() == *device) { + if (i.device() == device) { on_drive_list = true; mounted = i.mounted(); } } if (!on_drive_list) { - LOG_DISK ("Will not write to %1 as it's not recognised as a drive", *device); + LOG_DISK ("Will not write to %1 as it's not recognised as a drive", device); nanomsg->send(DISK_WRITER_ERROR "\nRefusing to write to this drive\n1\n", LONG_TIMEOUT); return true; } if (mounted) { - LOG_DISK ("Will not write to %1 as it's mounted", *device); + LOG_DISK ("Will not write to %1 as it's mounted", device); nanomsg->send(DISK_WRITER_ERROR "\nRefusing to write to this drive\n1\n", LONG_TIMEOUT); return true; } - LOG_DISK ("Here we go writing %1 to %2", *dcp_path, *device); + LOG_DISK("Here we go writing these to %1", device); + for (auto dcp: dcp_paths) { + LOG_DISK(" %1", dcp); + } + request_privileges ( + "com.dcpomatic.write-drive", + [dcp_paths, device]() { #if defined(DCPOMATIC_LINUX) - polkit_authority = polkit_authority_get_sync (0, 0); - auto subject = polkit_unix_process_new_for_owner (getppid(), 0, -1); - auto parameters = new Parameters; - parameters->dcp_path = *dcp_path; - parameters->device = *device; - parameters->posix_partition = *device; - /* XXX: don't know if this logic is sensible */ - if (parameters->posix_partition.size() > 0 && isdigit(parameters->posix_partition[parameters->posix_partition.length() - 1])) { - parameters->posix_partition += "p1"; - } else { - parameters->posix_partition += "1"; - } - polkit_authority_check_authorization ( - polkit_authority, subject, "com.dcpomatic.write-drive", 0, POLKIT_CHECK_AUTHORIZATION_FLAGS_ALLOW_USER_INTERACTION, 0, polkit_callback, parameters - ); + auto posix_partition = device; + /* XXX: don't know if this logic is sensible */ + if (posix_partition.size() > 0 && isdigit(posix_partition[posix_partition.length() - 1])) { + posix_partition += "p1"; + } else { + posix_partition += "1"; + } + dcpomatic::write (dcp_paths, device, posix_partition, nanomsg); #elif defined(DCPOMATIC_OSX) - auto fast_device = boost::algorithm::replace_first_copy (*device, "/dev/disk", "/dev/rdisk"); - dcpomatic::write (*dcp_path, fast_device, fast_device + "s1", nanomsg); + auto fast_device = boost::algorithm::replace_first_copy (device, "/dev/disk", "/dev/rdisk"); + dcpomatic::write (dcp_paths, fast_device, fast_device + "s1", nanomsg); #elif defined(DCPOMATIC_WINDOWS) - dcpomatic::write (*dcp_path, *device, "", nanomsg); + dcpomatic::write (dcp_paths, device, "", nanomsg); #endif + }, + []() { + if (nanomsg) { + nanomsg->send(DISK_WRITER_ERROR "\nCould not obtain authorization to write to the drive\n1\n", LONG_TIMEOUT); + } + }); } return true; @@ -257,12 +306,12 @@ main () * redirect this to a file in /var/log */ dcpomatic_log.reset(new StdoutLog(LogEntry::TYPE_DISK)); - LOG_DISK("dcpomatic_disk_writer %1 started", dcpomatic_git_commit); + LOG_DISK("dcpomatic_disk_writer %1 started uid=%2 euid=%3", dcpomatic_git_commit, getuid(), geteuid()); #else /* XXX: this is a hack, but I expect we'll need logs and I'm not sure if there's * a better place to put them. */ - dcpomatic_log.reset(new FileLog(config_path() / "disk_writer.log", LogEntry::TYPE_DISK)); + dcpomatic_log.reset(new FileLog(State::write_path("disk_writer.log"), LogEntry::TYPE_DISK)); LOG_DISK_NC("dcpomatic_disk_writer started"); #endif @@ -280,6 +329,7 @@ main () exit (EXIT_FAILURE); } + LOG_DISK_NC("Entering main loop"); auto ml = Glib::MainLoop::create (); Glib::signal_timeout().connect(sigc::ptr_fun(&idle), 500); ml->run ();