base/win/startup_information.cc

   1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/win/startup_information.h"
   6
   7 #include "base/logging.h"
   8 #include "base/win/windows_version.h"
   9
  10 namespace {
  11
  12 typedef BOOL (WINAPI *InitializeProcThreadAttributeListFunction)(
  13     LPPROC_THREAD_ATTRIBUTE_LIST attribute_list,
  14     DWORD attribute_count,
  15     DWORD flags,
  16     PSIZE_T size);
  17 static InitializeProcThreadAttributeListFunction
  18     initialize_proc_thread_attribute_list;
  19
  20 typedef BOOL (WINAPI *UpdateProcThreadAttributeFunction)(
  21     LPPROC_THREAD_ATTRIBUTE_LIST attribute_list,
  22     DWORD flags,
  23     DWORD_PTR attribute,
  24     PVOID value,
  25     SIZE_T size,
  26     PVOID previous_value,
  27     PSIZE_T return_size);
  28 static UpdateProcThreadAttributeFunction update_proc_thread_attribute_list;
  29
  30 typedef VOID (WINAPI *DeleteProcThreadAttributeListFunction)(
  31     LPPROC_THREAD_ATTRIBUTE_LIST lpAttributeList);
  32 static DeleteProcThreadAttributeListFunction delete_proc_thread_attribute_list;
  33
  34 }  // namespace
  35
  36 namespace base {
  37 namespace win {
  38
  39 StartupInformation::StartupInformation() {
  40   memset(&startup_info_, 0, sizeof(startup_info_));
  41
  42   // Pre Windows Vista doesn't support STARTUPINFOEX.
  43   if (base::win::GetVersion() < base::win::VERSION_VISTA) {
  44     startup_info_.StartupInfo.cb = sizeof(STARTUPINFO);
  45     return;
  46   }
  47
  48   startup_info_.StartupInfo.cb = sizeof(startup_info_);
  49
  50   // Load the attribute API functions.
  51   if (!initialize_proc_thread_attribute_list ||
  52       !update_proc_thread_attribute_list ||
  53       !delete_proc_thread_attribute_list) {
  54     HMODULE module = ::GetModuleHandleW(L"kernel32.dll");
  55     initialize_proc_thread_attribute_list =
  56         reinterpret_cast<InitializeProcThreadAttributeListFunction>(
  57             ::GetProcAddress(module, "InitializeProcThreadAttributeList"));
  58     update_proc_thread_attribute_list =
  59         reinterpret_cast<UpdateProcThreadAttributeFunction>(
  60             ::GetProcAddress(module, "UpdateProcThreadAttribute"));
  61     delete_proc_thread_attribute_list =
  62         reinterpret_cast<DeleteProcThreadAttributeListFunction>(
  63             ::GetProcAddress(module, "DeleteProcThreadAttributeList"));
  64   }
  65 }
  66
  67 StartupInformation::~StartupInformation() {
  68   if (startup_info_.lpAttributeList) {
  69     delete_proc_thread_attribute_list(startup_info_.lpAttributeList);
  70     delete [] reinterpret_cast<BYTE*>(startup_info_.lpAttributeList);
  71   }
  72 }
  73
  74 bool StartupInformation::InitializeProcThreadAttributeList(
  75     DWORD attribute_count) {
  76   if (startup_info_.StartupInfo.cb != sizeof(startup_info_) ||
  77       startup_info_.lpAttributeList)
  78     return false;
  79
  80   SIZE_T size = 0;
  81   initialize_proc_thread_attribute_list(NULL, attribute_count, 0, &size);
  82   if (size == 0)
  83     return false;
  84
  85   startup_info_.lpAttributeList =
  86       reinterpret_cast<LPPROC_THREAD_ATTRIBUTE_LIST>(new BYTE[size]);
  87   if (!initialize_proc_thread_attribute_list(startup_info_.lpAttributeList,
  88                                            attribute_count, 0, &size)) {
  89     delete [] reinterpret_cast<BYTE*>(startup_info_.lpAttributeList);
  90     startup_info_.lpAttributeList = NULL;
  91     return false;
  92   }
  93
  94   return true;
  95 }
  96
  97 bool StartupInformation::UpdateProcThreadAttribute(
  98     DWORD_PTR attribute,
  99     void* value,
 100     size_t size) {
 101   if (!startup_info_.lpAttributeList)
 102     return false;
 103   return !!update_proc_thread_attribute_list(startup_info_.lpAttributeList, 0,
 104                                        attribute, value, size, NULL, NULL);
 105 }
 106
 107 }  // namespace win
 108 }  // namespace base
 109