/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef __SECURITY_SANDBOX_SANDBOXTARGET_H__
#define __SECURITY_SANDBOX_SANDBOXTARGET_H__

#include <windows.h>

#include "mozilla/Assertions.h"
#include "base/MissingBasicTypes.h"
#include "sandbox/win/src/sandbox.h"
#include "sandbox/win/src/target_services.h"

#ifdef TARGET_SANDBOX_EXPORTS
#define TARGET_SANDBOX_EXPORT __declspec(dllexport)
#else
#define TARGET_SANDBOX_EXPORT __declspec(dllimport)
#endif
namespace mozilla {


class TARGET_SANDBOX_EXPORT SandboxTarget
{
public:
  /**
   * Obtains a pointer to the singleton instance
   */
  static SandboxTarget* Instance();

  /**
   * Used by the application to pass in the target services that provide certain
   * functions to the sandboxed code.
   * The target services must already be initialized.
   *
   * @param aTargetServices The target services that will be used
   */
  void SetTargetServices(sandbox::TargetServices* aTargetServices)
  {
    MOZ_ASSERT(aTargetServices);
    MOZ_ASSERT(!mTargetServices,
               "Sandbox TargetServices must only be set once.");

    mTargetServices = aTargetServices;
  }

  /**
   * Called by the library that wants to "start" the sandbox, i.e. change to the
   * more secure delayed / lockdown policy.
   */
  void StartSandbox()
  {
    if (mTargetServices) {
      mTargetServices->LowerToken();
    }
  }

  /**
   * Used to duplicate handles via the broker process. The permission for the
   * handle type and target process has to have been set on the sandbox policy.
   */
  bool BrokerDuplicateHandle(HANDLE aSourceHandle, DWORD aTargetProcessId,
                             HANDLE* aTargetHandle, DWORD aDesiredAccess,
                             DWORD aOptions)
  {
    if (!mTargetServices) {
      return false;
    }

    sandbox::ResultCode result =
      mTargetServices->DuplicateHandle(aSourceHandle, aTargetProcessId,
                                       aTargetHandle, aDesiredAccess, aOptions);
    return (sandbox::SBOX_ALL_OK == result);
  }

protected:
  SandboxTarget() :
    mTargetServices(nullptr)
  {
  }

  sandbox::TargetServices* mTargetServices;
};


} // mozilla

#endif
