/*
 * Copyright (C) 2018 The Android Open Source Project
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#define LOG_TAG "apexd"

#include "apexd_prepostinstall.h"

#include <algorithm>
#include <vector>

#include <fcntl.h>
#include <sys/mount.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <unistd.h>

#include <android-base/logging.h>
#include <android-base/macros.h>
#include <android-base/scopeguard.h>
#include <android-base/strings.h>

#include "apex_database.h"
#include "apex_file.h"
#include "apexd.h"
#include "apexd_private.h"
#include "apexd_utils.h"
#include "string_log.h"

using android::base::Error;
using android::base::Result;

namespace android {
namespace apex {

namespace {

using MountedApexData = MountedApexDatabase::MountedApexData;

void CloseSTDDescriptors() {
  // exec()d process will reopen STD* file descriptors as
  // /dev/null
  close(STDIN_FILENO);
  close(STDOUT_FILENO);
  close(STDERR_FILENO);
}

// Instead of temp mounting inside this fuction, we can make a caller do it.
// This will align with the plan of extending temp mounting to provide a
// way to run additional pre-reboot verification of an APEX.
// TODO(b/158470432): pass mount points instead of apex files.
template <typename Fn>
Result<void> StageFnInstall(const std::vector<ApexFile>& apexes, Fn fn,
                            const char* arg, const char* name) {
  // TODO(b/158470023): consider supporting a session with more than one
  //   pre-install hook.
  int hook_idx = -1;
  for (size_t i = 0; i < apexes.size(); i++) {
    if (!(apexes[i].GetManifest().*fn)().empty()) {
      if (hook_idx != -1) {
        return Error() << "Missing support for multiple " << name << " hooks";
      }
      hook_idx = i;
    }
  }
  CHECK(hook_idx != -1);
  LOG(VERBOSE) << name << " for " << apexes[hook_idx].GetPath();

  std::vector<MountedApexData> mounted_apexes;
  std::vector<std::string> activation_dirs;
  auto preinstall_guard = android::base::make_scope_guard([&]() {
    for (const auto& mount : mounted_apexes) {
      Result<void> st = apexd_private::Unmount(mount);
      if (!st.ok()) {
        LOG(ERROR) << "Failed to unmount " << mount.full_path << " from "
                   << mount.mount_point << " after " << name << ": "
                   << st.error();
      }
    }
    for (const std::string& active_point : activation_dirs) {
      if (0 != rmdir(active_point.c_str())) {
        PLOG(ERROR) << "Could not delete temporary active point "
                    << active_point;
      }
    }
  });

  for (const ApexFile& apex : apexes) {
    // 1) Mount the package.
    std::string mount_point =
        apexd_private::GetPackageTempMountPoint(apex.GetManifest());

    auto mount_data = apexd_private::TempMountPackage(apex, mount_point);
    if (!mount_data.ok()) {
      return mount_data.error();
    }
    mounted_apexes.push_back(std::move(*mount_data));

    // Given the fact, that we only allow updates of existing APEXes, all the
    // activation points will always be already created. Only scenario, when it
    // won't be the case might be apexservice_test. But even then, it might be
    // safer to move active_point creation logic to run after unshare.
    // TODO(b/158470432): maybe move creation of activation points inside
    //   RunFnInstall?
    // 2) Ensure there is an activation point, and we will clean it up.
    std::string active_point =
        apexd_private::GetActiveMountPoint(apex.GetManifest());
    if (0 == mkdir(active_point.c_str(), kMkdirMode)) {
      activation_dirs.emplace_back(std::move(active_point));
    } else {
      int saved_errno = errno;
      if (saved_errno != EEXIST) {
        return Error() << "Unable to create mount point" << active_point << ": "
                       << strerror(saved_errno);
      }
    }
  }

  // 3) Create invocation args.
  std::vector<std::string> args{
      "/system/bin/apexd", arg,
      mounted_apexes[hook_idx].mount_point,  // Make the APEX with hook first.
  };
  for (size_t i = 0; i < mounted_apexes.size(); i++) {
    if ((int)i != hook_idx) {
      args.push_back(mounted_apexes[i].mount_point);
    }
  }

  return ForkAndRun(args);
}

template <typename Fn>
int RunFnInstall(char** in_argv, Fn fn, const char* name) {
  // 1) Unshare.
  if (unshare(CLONE_NEWNS) != 0) {
    PLOG(ERROR) << "Failed to unshare() for apex " << name;
    _exit(200);
  }

  // 2) Make everything private, so that our (and hook's) changes do not
  //    propagate.
  if (mount(nullptr, "/", nullptr, MS_PRIVATE | MS_REC, nullptr) == -1) {
    PLOG(ERROR) << "Failed to mount private.";
    _exit(201);
  }

  std::string hook_path;
  {
    auto bind_fn = [&fn, name](const std::string& mount_point) {
      std::string hook;
      std::string active_point;
      {
        Result<ApexManifest> manifest_or =
            ReadManifest(mount_point + "/" + kManifestFilenamePb);
        if (!manifest_or.ok()) {
          LOG(ERROR) << "Could not read manifest from  " << mount_point << "/"
                     << kManifestFilenamePb << " for " << name << ": "
                     << manifest_or.error();
          // Fallback to Json manifest if present.
          LOG(ERROR) << "Trying to find a JSON manifest";
          manifest_or = ReadManifest(mount_point + "/" + kManifestFilenameJson);
          if (!manifest_or.ok()) {
            LOG(ERROR) << "Could not read manifest from  " << mount_point << "/"
                       << kManifestFilenameJson << " for " << name << ": "
                       << manifest_or.error();
            _exit(202);
          }
        }
        const auto& manifest = *manifest_or;
        hook = (manifest.*fn)();
        active_point = apexd_private::GetActiveMountPoint(manifest);
      }

      // 3) Activate the new apex.
      Result<void> bind_status =
          apexd_private::BindMount(active_point, mount_point);
      if (!bind_status.ok()) {
        LOG(ERROR) << "Failed to bind-mount " << mount_point << " to "
                   << active_point << ": " << bind_status.error();
        _exit(203);
      }

      return std::make_pair(active_point, hook);
    };

    // First/main APEX.
    auto [active_point, hook] = bind_fn(in_argv[2]);
    hook_path = active_point + "/" + hook;

    for (size_t i = 3;; ++i) {
      if (in_argv[i] == nullptr) {
        break;
      }
      bind_fn(in_argv[i]);  // Ignore result, hook will be empty.
    }
  }

  // 4) Run the hook.

  // For now, just run sh. But this probably needs to run the new linker.
  std::vector<std::string> args{
      hook_path,
  };
  std::vector<const char*> argv;
  argv.resize(args.size() + 1, nullptr);
  std::transform(args.begin(), args.end(), argv.begin(),
                 [](const std::string& in) { return in.c_str(); });

  LOG(ERROR) << "execv of " << android::base::Join(args, " ");

  // Close all file descriptors. They are coming from the caller, we do not
  // want to pass them on across our fork/exec into a different domain.
  CloseSTDDescriptors();

  execv(argv[0], const_cast<char**>(argv.data()));
  PLOG(ERROR) << "execv of " << android::base::Join(args, " ") << " failed";
  _exit(204);
}

}  // namespace

Result<void> StagePreInstall(const std::vector<ApexFile>& apexes) {
  return StageFnInstall(apexes, &ApexManifest::preinstallhook, "--pre-install",
                        "pre-install");
}

int RunPreInstall(char** in_argv) {
  return RunFnInstall(in_argv, &ApexManifest::preinstallhook, "pre-install");
}

Result<void> StagePostInstall(const std::vector<ApexFile>& apexes) {
  return StageFnInstall(apexes, &ApexManifest::postinstallhook,
                        "--post-install", "post-install");
}

int RunPostInstall(char** in_argv) {
  return RunFnInstall(in_argv, &ApexManifest::postinstallhook, "post-install");
}

}  // namespace apex
}  // namespace android