// Copyright 2018 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_CHROME_CLEANER_IPC_MOJO_SANDBOX_HOOKS_H_
#define CHROME_CHROME_CLEANER_IPC_MOJO_SANDBOX_HOOKS_H_

#include "base/command_line.h"
#include "base/macros.h"
#include "base/process/process.h"
#include "base/win/scoped_handle.h"
#include "chrome/chrome_cleaner/ipc/sandbox.h"
#include "components/chrome_cleaner/public/constants/result_codes.h"
#include "mojo/public/cpp/platform/platform_channel.h"
#include "mojo/public/cpp/system/invitation.h"
#include "mojo/public/cpp/system/message_pipe.h"
#include "sandbox/win/src/sandbox_policy.h"

namespace chrome_cleaner {

class MojoSandboxSetupHooks : public SandboxSetupHooks {
 public:
  MojoSandboxSetupHooks();
  ~MojoSandboxSetupHooks() override;

 protected:
  mojo::ScopedMessagePipeHandle SetupSandboxMessagePipe(
      sandbox::TargetPolicy* policy,
      base::CommandLine* command_line);

  // SandboxSetupHooks

  ResultCode TargetSpawned(
      const base::Process& target_process,
      const base::win::ScopedHandle& target_thread) override;

 private:
  bool message_pipe_was_created_ = false;
  mojo::OutgoingInvitation outgoing_invitation_;
  mojo::PlatformChannel mojo_channel_;

  DISALLOW_COPY_AND_ASSIGN(MojoSandboxSetupHooks);
};

class MojoSandboxTargetHooks : public SandboxTargetHooks {
 public:
  MojoSandboxTargetHooks();
  ~MojoSandboxTargetHooks() override;

 protected:
  mojo::ScopedMessagePipeHandle ExtractSandboxMessagePipe(
      const base::CommandLine& command_line);

 private:
  DISALLOW_COPY_AND_ASSIGN(MojoSandboxTargetHooks);
};

}  // namespace chrome_cleaner

#endif  // CHROME_CHROME_CLEANER_IPC_MOJO_SANDBOX_HOOKS_H_