// Copyright (c) 2011 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.

#include "chrome_frame/update_launcher.h"

#include <windows.h>
#include <Shellapi.h>

#include "google_update_idl.h"  // NOLINT

namespace {

const wchar_t kChromeFrameGuid[] = L"{8BA986DA-5100-405E-AA35-86F34A02ACBF}";

const DWORD kLaunchFailureExitCode = 0xFF;

const wchar_t kUpdateCommandFlag[] = L"--update-cmd";

// Waits indefinitely for the provided process to exit. Returns the process's
// exit code, or kLaunchFailureExitCode if an error occurs in the waiting.
DWORD WaitForProcessExitCode(HANDLE handle) {
  DWORD exit_code = 0;

  DWORD wait_result = ::WaitForSingleObject(handle, INFINITE);

  if (wait_result == WAIT_OBJECT_0 && ::GetExitCodeProcess(handle, &exit_code))
    return exit_code;

  return kLaunchFailureExitCode;
}

}  // namespace

namespace update_launcher {

std::wstring GetUpdateCommandFromArguments(const wchar_t* command_line) {
  std::wstring command;

  if (command_line != NULL) {
    int num_args = 0;
    wchar_t** args = NULL;
    args = ::CommandLineToArgvW(command_line, &num_args);

    if (args) {
      if (num_args == 3 && _wcsicmp(args[1], kUpdateCommandFlag) == 0)
        command = args[2];
      ::LocalFree(args);
    }
  }

  return command;
}

// Because we do not have 'base' and all of its pretty RAII helpers, please
// ensure that this function only ever contains a single 'return', in order
// to reduce the chance of introducing a leak.
DWORD LaunchUpdateCommand(const std::wstring& command) {
  DWORD exit_code = kLaunchFailureExitCode;

  HRESULT hr = ::CoInitialize(NULL);

  if (SUCCEEDED(hr)) {
    IProcessLauncher* ipl = NULL;
    HANDLE process = NULL;

    hr = ::CoCreateInstance(__uuidof(ProcessLauncherClass), NULL,
                            CLSCTX_ALL, __uuidof(IProcessLauncher),
                            reinterpret_cast<void**>(&ipl));

    if (SUCCEEDED(hr)) {
      ULONG_PTR phandle = NULL;
      DWORD id = ::GetCurrentProcessId();

      hr = ipl->LaunchCmdElevated(kChromeFrameGuid,
                                  command.c_str(), id, &phandle);
      if (SUCCEEDED(hr)) {
        process = reinterpret_cast<HANDLE>(phandle);
        exit_code = WaitForProcessExitCode(process);
      }
    }

    if (process)
      ::CloseHandle(process);
    if (ipl)
      ipl->Release();

    ::CoUninitialize();
  }

  return exit_code;
}

}  // namespace process_launcher