diff options
Diffstat (limited to 'src/usb-manager.cpp')
| -rw-r--r-- | src/usb-manager.cpp | 180 |
1 files changed, 180 insertions, 0 deletions
diff --git a/src/usb-manager.cpp b/src/usb-manager.cpp new file mode 100644 index 0000000..4d750c0 --- /dev/null +++ b/src/usb-manager.cpp @@ -0,0 +1,180 @@ +/* + * Copyright 2016 Canonical Ltd. + * + * This program is free software: you can redistribute it and/or modify it + * under the terms of the GNU General Public License version 3, as published + * by the Free Software Foundation. + * + * This program is distributed in the hope that it will be useful, but + * WITHOUT ANY WARRANTY; without even the implied warranties of + * MERCHANTABILITY, SATISFACTORY QUALITY, or FITNESS FOR A PARTICULAR + * PURPOSE. See the GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License along + * with this program. If not, see <http://www.gnu.org/licenses/>. + * + * Authors: + * Charles Kerr <charles.kerr@canonical.com> + */ + +#include <src/adbd-client.h> +#include <src/usb-manager.h> +#include <src/usb-snap.h> + +#include <glib.h> + +#include <sys/types.h> +#include <sys/stat.h> +#include <fcntl.h> +#include <unistd.h> + +#include <set> + +class UsbManager::Impl +{ +public: + + explicit Impl( + const std::string& socket_path, + const std::string& public_keys_filename, + const std::shared_ptr<UsbMonitor>& usb_monitor, + const std::shared_ptr<Greeter>& greeter + ): + m_socket_path{socket_path}, + m_public_keys_filename{public_keys_filename}, + m_usb_monitor{usb_monitor}, + m_greeter{greeter} + { + m_usb_monitor->on_usb_disconnected().connect([this](const std::string& /*usb_name*/) { + restart(); + }); + + m_greeter->is_active().changed().connect([this](bool /*is_active*/) { + maybe_snap(); + }); + + restart(); + } + + ~Impl() + { + if (m_restart_idle_tag) + g_source_remove(m_restart_idle_tag); + + clear(); + } + +private: + + void clear() + { + // clear out old state + m_snap_connections.clear(); + m_snap.reset(); + m_req = decltype(m_req)(); + m_adbd_client.reset(); + } + + void restart() + { + clear(); + + // set a new client + m_adbd_client.reset(new GAdbdClient{m_socket_path}); + m_adbd_client->on_pk_request().connect( + [this](const AdbdClient::PKRequest& req) { + g_debug("%s got pk request: %s", G_STRLOC, req.fingerprint.c_str()); + m_req = req; + maybe_snap(); + } + ); + } + + void maybe_snap() + { + // don't prompt in the greeter! + if (!m_req.public_key.empty() && !m_greeter->is_active().get()) + snap(); + } + + void snap() + { + m_snap = std::make_shared<UsbSnap>(m_req.fingerprint); + m_snap_connections.insert((*m_snap).on_user_response().connect( + [this](AdbdClient::PKResponse response, bool remember_choice){ + g_debug("%s user responded! response %d, remember %d", G_STRLOC, int(response), int(remember_choice)); + m_req.respond(response); + if (remember_choice && (response == AdbdClient::PKResponse::ALLOW)) + write_public_key(m_req.public_key); + m_restart_idle_tag = g_idle_add([](gpointer gself){ + auto self = static_cast<Impl*>(gself); + self->m_restart_idle_tag = 0; + self->restart(); + return G_SOURCE_REMOVE; + }, this); + } + )); + } + + void write_public_key(const std::string& public_key) + { + g_debug("%s writing public key '%s' to '%s'", G_STRLOC, public_key.c_str(), m_public_keys_filename.c_str()); + + // confirm the directory exists + auto dirname = g_path_get_dirname(m_public_keys_filename.c_str()); + const auto dir_exists = g_file_test(dirname, G_FILE_TEST_IS_DIR); + if (!dir_exists) + g_warning("ADB data directory '%s' does not exist", dirname); + g_clear_pointer(&dirname, g_free); + if (!dir_exists) + return; + + // open the file in append mode, with user rw and group r permissions + const auto fd = open( + m_public_keys_filename.c_str(), + O_APPEND|O_CREAT|O_WRONLY, + S_IRUSR|S_IWUSR|S_IRGRP + ); + if (fd == -1) { + g_warning("Error opening ADB datafile: %s", g_strerror(errno)); + return; + } + + // write the new public key on its own line + std::string buf {public_key + '\n'}; + if (write(fd, buf.c_str(), buf.size()) == -1) + g_warning("Error writing ADB datafile: %d %s", errno, g_strerror(errno)); + close(fd); + } + + const std::string m_socket_path; + const std::string m_public_keys_filename; + const std::shared_ptr<UsbMonitor> m_usb_monitor; + const std::shared_ptr<Greeter> m_greeter; + + unsigned int m_restart_idle_tag {}; + + std::shared_ptr<GAdbdClient> m_adbd_client; + AdbdClient::PKRequest m_req; + std::shared_ptr<UsbSnap> m_snap; + std::set<core::ScopedConnection> m_snap_connections; +}; + +/*** +**** +***/ + +UsbManager::UsbManager( + const std::string& socket_path, + const std::string& public_keys_filename, + const std::shared_ptr<UsbMonitor>& usb_monitor, + const std::shared_ptr<Greeter>& greeter +): + impl{new Impl{socket_path, public_keys_filename, usb_monitor, greeter}} +{ +} + +UsbManager::~UsbManager() +{ +} + |