server/internal-encryption.cpp

   1 /*
   2  *  Copyright (c) 2015-2017 Samsung Electronics Co., Ltd All Rights Reserved
   3  *
   4  *  Licensed under the Apache License, Version 2.0 (the "License");
   5  *  you may not use this file except in compliance with the License.
   6  *  You may obtain a copy of the License at
   7  *
   8  *      http://www.apache.org/licenses/LICENSE-2.0
   9  *
  10  *  Unless required by applicable law or agreed to in writing, software
  11  *  distributed under the License is distributed on an "AS IS" BASIS,
  12  *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  13  *  See the License for the specific language governing permissions and
  14  *  limitations under the License
  15  */
  16 #include <set>
  17 #include <algorithm>
  18 #include <memory>
  19
  20 #include <fcntl.h>
  21 #include <signal.h>
  22 #include <unistd.h>
  23 #include <sys/mount.h>
  24 #include <sys/reboot.h>
  25 #include <limits.h>
  26 #include <stdlib.h>
  27 #include <string.h>
  28
  29 #include <vconf.h>
  30 #include <tzplatform_config.h>
  31 #include <klay/process.h>
  32 #include <klay/file-user.h>
  33 #include <klay/filesystem.h>
  34 #include <klay/dbus/connection.h>
  35 #include <klay/error.h>
  36
  37 #include "misc.h"
  38 #include "logger.h"
  39 #include "progress-bar.h"
  40 #include "rmi/common.h"
  41
  42 #include "internal-encryption.h"
  43
  44 namespace ode {
  45
  46 namespace {
  47
  48 const char *INTERNAL_DEV_PATH   = "/dev/disk/by-partlabel";
  49 const char *INTERNAL_DEV_NAME   = "USER";
  50 const char *INTERNAL_PATH               = "/opt/usr";
  51
  52 const char *PRIVILEGE_PLATFORM  = "http://tizen.org/privilege/internal/default/platform";
  53
  54 // TODO: see recovery()
  55 const std::string PROG_FACTORY_RESET = "/usr/bin/dbus-send";
  56 const std::vector<std::string> wipeCommand = {
  57     PROG_FACTORY_RESET,
  58     "--system",
  59     "--type=signal",
  60     "--print-reply",
  61     "--dest=com.samsung.factoryreset",
  62     "/com/samsung/factoryreset",
  63     "com.samsung.factoryreset.start.setting"
  64 };
  65
  66 std::string findDevPath()
  67 {
  68         std::string source = INTERNAL_DEV_PATH + std::string("/") + INTERNAL_DEV_NAME;
  69         try {
  70                 runtime::DirectoryIterator iter(INTERNAL_DEV_PATH), end;
  71
  72                 while (iter != end) {
  73                         const std::string& path = (*iter).getPath();
  74                         std::string name = path.substr(path.rfind('/') + 1);
  75                         std::string upper;
  76                         upper.reserve(name.size());
  77                         for (char c : name) {
  78                                 upper += std::toupper(c);
  79                         }
  80                         if (upper.compare(0, strlen(INTERNAL_DEV_NAME), INTERNAL_DEV_NAME) == 0) {
  81                                 source = path;
  82                                 break;
  83                         }
  84                         ++iter;
  85                 }
  86         } catch (runtime::Exception &e) {}
  87
  88         char *dev = ::realpath(source.c_str(), NULL);
  89         if (dev == NULL) {
  90                 ERROR(SINK, "Failed to get device path.");
  91                 return "";
  92         }
  93
  94         std::string devPath(dev);
  95         free(dev);
  96
  97         return devPath;
  98 }
  99
 100 void stopKnownSystemdServices()
 101 {
 102         std::vector<std::string> knownSystemdServices;
 103         dbus::Connection& systemDBus = dbus::Connection::getSystem();
 104         dbus::VariantIterator iter;
 105
 106         systemDBus.methodcall("org.freedesktop.systemd1",
 107                                                         "/org/freedesktop/systemd1",
 108                                                         "org.freedesktop.systemd1.Manager",
 109                                                         "ListUnits",
 110                                                         -1, "(a(ssssssouso))", "")
 111                                                                 .get("(a(ssssssouso))", &iter);
 112
 113         while (1) {
 114                 unsigned int dataUint;
 115                 char *dataStr[9];
 116                 int ret;
 117
 118                 ret = iter.get("(ssssssouso)", dataStr, dataStr + 1, dataStr + 2,
 119                                                 dataStr + 3, dataStr + 4, dataStr + 5,
 120                                                 dataStr + 6, &dataUint, dataStr + 7,
 121                                                 dataStr + 8);
 122
 123                 if (!ret) {
 124                         break;
 125                 }
 126
 127                 std::string service(dataStr[0]);
 128                 if (service.compare(0, 5, "user@") == 0 ||
 129                         service == "tlm.service" ||
 130                         service == "resourced.service") {
 131                         knownSystemdServices.push_back(service);
 132                 }
 133         }
 134
 135         for (const std::string& service : knownSystemdServices) {
 136                 INFO(SINK, "Stopping service: " + service);
 137                 systemDBus.methodcall("org.freedesktop.systemd1",
 138                                                                 "/org/freedesktop/systemd1",
 139                                                                 "org.freedesktop.systemd1.Manager",
 140                                                                 "StopUnit",
 141                                                                 -1, "", "(ss)", service.c_str(), "flush");
 142         }
 143
 144         sleep(1);
 145 }
 146
 147 void stopDependedSystemdServices()
 148 {
 149         dbus::Connection& systemDBus = dbus::Connection::getSystem();
 150         std::set<std::string> servicesToStop;
 151
 152         for (pid_t pid : runtime::FileUser::getList(INTERNAL_PATH, true)) {
 153                 try {
 154                         char *service;
 155                         systemDBus.methodcall("org.freedesktop.systemd1",
 156                                                                         "/org/freedesktop/systemd1",
 157                                                                         "org.freedesktop.systemd1.Manager",
 158                                                                         "GetUnitByPID",
 159                                                                         -1, "(o)", "(u)", (unsigned int)pid)
 160                                                                                 .get("(o)", &service);
 161                         servicesToStop.insert(service);
 162                 } catch (runtime::Exception &e) {
 163                         INFO(SINK, "Killing process: " + std::to_string(pid));
 164                         ::kill(pid, SIGKILL);
 165                 }
 166         }
 167
 168         for (const std::string& service : servicesToStop) {
 169                 INFO(SINK, "Stopping service: " + service);
 170                 systemDBus.methodcall("org.freedesktop.systemd1",
 171                                                                 service,
 172                                                                 "org.freedesktop.systemd1.Unit",
 173                                                                 "Stop",
 174                                                                 -1, "", "(s)", "flush");
 175         }
 176 }
 177
 178 void showProgressUI(const std::string type)
 179 {
 180         ::tzplatform_set_user(::tzplatform_getuid(TZ_SYS_DEFAULT_USER));
 181         std::string defaultUserHome(::tzplatform_getenv(TZ_USER_HOME));
 182         ::tzplatform_reset_user();
 183
 184         try {
 185                 runtime::File shareDirectory("/opt/home/root/share");
 186                 if (!shareDirectory.exists()) {
 187                         shareDirectory.makeDirectory(true);
 188                 }
 189
 190                 runtime::File elmConfigDir(shareDirectory.getPath() + "/.elementary");
 191                 if (!elmConfigDir.exists()) {
 192                         runtime::File defaultElmConfigDir(defaultUserHome + "/share/.elementary");
 193                         defaultElmConfigDir.copyTo(shareDirectory.getPath());
 194                 }
 195         } catch (runtime::Exception &e) {
 196                 ERROR(SINK, "Failed to set up elm configuration: " + std::string(e.what()));
 197         }
 198
 199         std::vector<std::string> args = {
 200                 "ode", "progress", type, "Internal"
 201         };
 202
 203         runtime::Process proc("/usr/bin/ode", args);
 204         proc.execute();
 205 }
 206
 207 unsigned int getOptions()
 208 {
 209         unsigned int result = 0;
 210         int value;
 211
 212         value = 0;
 213         ::vconf_get_bool(VCONFKEY_ODE_FAST_ENCRYPTION, &value);
 214         if (value) {
 215                 result |= InternalEncryption::Option::IncludeUnusedRegion;
 216         }
 217
 218         return result;
 219 }
 220
 221 void setOptions(unsigned int options)
 222 {
 223         bool value;
 224
 225         if (options & InternalEncryption::Option::IncludeUnusedRegion) {
 226                 value = true;
 227         } else {
 228                 value = false;
 229         }
 230         ::vconf_set_bool(VCONFKEY_ODE_FAST_ENCRYPTION, value);
 231 }
 232
 233 }
 234
 235 InternalEncryptionServer::InternalEncryptionServer(ServerContext& srv,
 236                                                                                                    KeyServer& key) :
 237         server(srv),
 238         keyServer(key)
 239 {
 240         server.expose(this, PRIVILEGE_PLATFORM, (int)(InternalEncryptionServer::setMountPassword)(std::string));
 241         server.expose(this, PRIVILEGE_PLATFORM, (int)(InternalEncryptionServer::mount)());
 242         server.expose(this, PRIVILEGE_PLATFORM, (int)(InternalEncryptionServer::umount)());
 243         server.expose(this, PRIVILEGE_PLATFORM, (int)(InternalEncryptionServer::encrypt)(std::string, unsigned int));
 244         server.expose(this, PRIVILEGE_PLATFORM, (int)(InternalEncryptionServer::decrypt)(std::string));
 245         server.expose(this, "", (int)(InternalEncryptionServer::isPasswordInitialized)());
 246         server.expose(this, PRIVILEGE_PLATFORM, (int)(InternalEncryptionServer::initPassword)(std::string));
 247         server.expose(this, PRIVILEGE_PLATFORM, (int)(InternalEncryptionServer::cleanPassword)(std::string));
 248         server.expose(this, PRIVILEGE_PLATFORM, (int)(InternalEncryptionServer::changePassword)(std::string, std::string));
 249         server.expose(this, PRIVILEGE_PLATFORM, (int)(InternalEncryptionServer::verifyPassword)(std::string));
 250         server.expose(this, "", (int)(InternalEncryptionServer::getState)());
 251         server.expose(this, "", (unsigned int)(InternalEncryptionServer::getSupportedOptions)());
 252         server.expose(this, "", (std::string)(InternalEncryptionServer::getDevicePath)());
 253
 254         server.createNotification("InternalEncryptionServer::mount");
 255
 256         std::string source = findDevPath();
 257
 258         engine.reset(new INTERNAL_ENGINE(
 259                 source, INTERNAL_PATH,
 260                 ProgressBar([](int v) {
 261                         ::vconf_set_str(VCONFKEY_ODE_ENCRYPT_PROGRESS,
 262                                                         std::to_string(v).c_str());
 263                 })
 264         ));
 265 }
 266
 267 InternalEncryptionServer::~InternalEncryptionServer()
 268 {
 269 }
 270
 271 int InternalEncryptionServer::setMountPassword(const std::string& password)
 272 {
 273         return keyServer.get(engine->getSource(), password, mountKey);
 274 }
 275
 276 int InternalEncryptionServer::mount()
 277 {
 278         if (mountKey.empty()) {
 279                 ERROR(SINK, "You need to call set_mount_password() first.");
 280                 return error::NoData;
 281         }
 282
 283         BinaryData key = mountKey;
 284         mountKey.clear();
 285
 286         if (getState() != State::Encrypted) {
 287                 INFO(SINK, "Cannot mount, SD partition's state incorrect.");
 288                 return error::NoSuchDevice;
 289         }
 290
 291         if (engine->isMounted()) {
 292                 INFO(SINK, "Partition already mounted.");
 293                 return error::None;
 294         }
 295
 296         INFO(SINK, "Mounting internal storage.");
 297         try {
 298                 engine->mount(key, getOptions());
 299
 300                 server.notify("InternalEncryptionServer::mount");
 301
 302                 runtime::File("/tmp/.lazy_mount").create(O_WRONLY);
 303                 runtime::File("/tmp/.unlock_mnt").create(O_WRONLY);
 304         } catch (runtime::Exception &e) {
 305                 ERROR(SINK, "Mount failed: " + std::string(e.what()));
 306                 return error::Unknown;
 307         }
 308
 309         return error::None;
 310 }
 311
 312 int InternalEncryptionServer::umount()
 313 {
 314         if (getState() != State::Encrypted) {
 315                 ERROR(SINK, "Cannot umount, partition's state incorrect.");
 316                 return error::NoSuchDevice;
 317         }
 318
 319         if (!engine->isMounted()) {
 320                 INFO(SINK, "Partition already umounted.");
 321                 return error::None;
 322         }
 323
 324         INFO(SINK, "Closing all processes using internal storage.");
 325         try {
 326                 stopDependedSystemdServices();
 327                 INFO(SINK, "Umounting internal storage.");
 328                 engine->umount();
 329         } catch (runtime::Exception &e) {
 330                 ERROR(SINK, "Umount failed: " + std::string(e.what()));
 331                 return error::Unknown;
 332         }
 333
 334         return error::None;
 335 }
 336
 337 int InternalEncryptionServer::encrypt(const std::string& password, unsigned int options)
 338 {
 339         if (getState() != State::Unencrypted) {
 340                 ERROR(SINK, "Cannot encrypt, partition's state incorrect.");
 341                 return error::NoSuchDevice;
 342         }
 343
 344         BinaryData masterKey;
 345         int ret = keyServer.get(engine->getSource(), password, masterKey);
 346         if (ret != error::None)
 347                 return ret;
 348
 349         auto encryptWorker = [masterKey, options, this]() {
 350                 try {
 351                         std::string source = engine->getSource();
 352                         std::string mntPath = findMountPointByDevice(source);
 353
 354                         if (!mntPath.empty()) {
 355                                 INFO(SINK, "Closing all known systemd services that might be using internal storage.");
 356                                 stopKnownSystemdServices();
 357                                 INFO(SINK, "Closing all processes using internal storage.");
 358                                 stopDependedSystemdServices();
 359                         }
 360
 361                         while (!mntPath.empty()) {
 362                                 INFO(SINK, "Umounting internal storage.");
 363                                 while (::umount(mntPath.c_str()) == -1) {
 364                                         if (errno != EBUSY) {
 365                                                 throw runtime::Exception("Umount error: " + runtime::GetSystemErrorMessage());
 366                                         }
 367                                         stopDependedSystemdServices();
 368                                 }
 369                                 mntPath = findMountPointByDevice(source);
 370                         }
 371
 372                         showProgressUI("Encrypting");
 373
 374                         INFO(SINK, "Encryption started.");
 375                         ::vconf_set_str(VCONFKEY_ODE_CRYPTO_STATE, "error_partially_encrypted");
 376                         engine->encrypt(masterKey, options);
 377                         setOptions(options & getSupportedOptions());
 378
 379                         INFO(SINK, "Encryption completed.");
 380                         ::vconf_set_str(VCONFKEY_ODE_CRYPTO_STATE, "encrypted");
 381                         server.notify("InternalEncryptionServer::mount");
 382
 383                         INFO(SINK, "Syncing disk and rebooting.");
 384                         ::sync();
 385                         ::reboot(RB_AUTOBOOT);
 386                 } catch (runtime::Exception &e) {
 387                         ERROR(SINK, "Encryption failed: " + std::string(e.what()));
 388                 }
 389         };
 390
 391         std::thread asyncWork(encryptWorker);
 392         asyncWork.detach();
 393
 394         return error::None;
 395 }
 396
 397 int InternalEncryptionServer::decrypt(const std::string& password)
 398 {
 399         if (getState() != State::Encrypted) {
 400                 ERROR(SINK, "Cannot decrypt, partition's state incorrect.");
 401                 return error::NoSuchDevice;
 402         }
 403
 404         BinaryData masterKey;
 405         int ret = keyServer.get(engine->getSource(), password, masterKey);
 406         if (ret != error::None)
 407                 return ret;
 408
 409         auto decryptWorker = [masterKey, this]() {
 410                 try {
 411                         if (engine->isMounted()) {
 412                                 INFO(SINK, "Closing all known systemd services that might be using internal storage.");
 413                                 stopKnownSystemdServices();
 414                                 INFO(SINK, "Closing all processes using internal storage.");
 415                                 stopDependedSystemdServices();
 416
 417                                 INFO(SINK, "Umounting internal storage.");
 418                                 while (1) {
 419                                         try {
 420                                                 engine->umount();
 421                                                 break;
 422                                         } catch (runtime::Exception& e) {
 423                                                 stopDependedSystemdServices();
 424                                         }
 425                                 }
 426                         }
 427
 428                         showProgressUI("Decrypting");
 429
 430                         INFO(SINK, "Decryption started.");
 431                         ::vconf_set_str(VCONFKEY_ODE_CRYPTO_STATE, "error_partially_encrypted");
 432                         engine->decrypt(masterKey, getOptions());
 433
 434                         INFO(SINK, "Decryption complete.");
 435                         ::vconf_set_str(VCONFKEY_ODE_CRYPTO_STATE, "unencrypted");
 436
 437                         INFO(SINK, "Syncing disk and rebooting.");
 438                         ::sync();
 439                         ::reboot(RB_AUTOBOOT);
 440                 } catch (runtime::Exception &e) {
 441                         ERROR(SINK, "Decryption failed: " + std::string(e.what()));
 442                 }
 443         };
 444
 445         std::thread asyncWork(decryptWorker);
 446         asyncWork.detach();
 447
 448         return error::None;
 449 }
 450
 451 int InternalEncryptionServer::recovery()
 452 {
 453         if (getState() == State::Unencrypted) {
 454                 return error::NoSuchDevice;
 455         }
 456
 457         //TODO
 458         runtime::Process proc(PROG_FACTORY_RESET, wipeCommand);
 459         if (proc.execute() == -1) {
 460                 ERROR(SINK, "Failed to launch factory-reset");
 461                 return error::WrongPassword;
 462         }
 463
 464         return error::None;
 465 }
 466
 467 int InternalEncryptionServer::isPasswordInitialized()
 468 {
 469         return keyServer.isInitialized(engine->getSource());
 470 }
 471
 472 int InternalEncryptionServer::initPassword(const std::string& password)
 473 {
 474         return keyServer.init(engine->getSource(), password, Key::DEFAULT_256BIT);
 475 }
 476
 477 int InternalEncryptionServer::cleanPassword(const std::string& password)
 478 {
 479         return keyServer.remove(engine->getSource(), password);
 480 }
 481
 482 int InternalEncryptionServer::changePassword(const std::string& oldPassword,
 483                                                                                 const std::string& newPassword)
 484 {
 485         return keyServer.changePassword(engine->getSource(), oldPassword, newPassword);
 486 }
 487
 488 int InternalEncryptionServer::verifyPassword(const std::string& password)
 489 {
 490         return keyServer.verifyPassword(engine->getSource(), password);
 491 }
 492
 493 int InternalEncryptionServer::getState()
 494 {
 495         char *value = ::vconf_get_str(VCONFKEY_ODE_CRYPTO_STATE);
 496         if (value == NULL) {
 497                 throw runtime::Exception("Failed to get vconf value.");
 498         }
 499
 500         std::string valueStr(value);
 501         free(value);
 502
 503         if (valueStr == "encrypted") {
 504                 return State::Encrypted;
 505         } else if (valueStr == "unencrypted") {
 506                 return State::Unencrypted;
 507         } else {
 508                 return State::Corrupted;
 509         }
 510 }
 511
 512 unsigned int InternalEncryptionServer::getSupportedOptions()
 513 {
 514         return engine->getSupportedOptions();
 515 }
 516
 517 std::string InternalEncryptionServer::getDevicePath() const
 518 {
 519         return engine->getSource();
 520 }
 521
 522 } // namespace ode