gdipp_client/gdipp_client.cpp

   1 #include "stdafx.h"
   2 #include "gdipp_client/gamma.h"
   3 #include "gdipp_client/global.h"
   4 #include "gdipp_config/constant_client.h"
   5 #include "gdipp_config/exclude_config.h"
   6 #include "gdipp_lib/helper.h"
   7 #include "gdipp_lib/scoped_rw_lock.h"
   8
   9 namespace gdipp
  10 {
  11
  12 HANDLE process_heap = GetProcessHeap();
  13
  14 bool init_rpc_client()
  15 {
  16         if (process_heap == NULL)
  17                 return false;
  18
  19         RPC_WSTR binding_str;
  20         RPC_STATUS rpc_status;
  21
  22         rpc_status = RpcStringBindingCompose(NULL, reinterpret_cast<RPC_WSTR>(L"ncalrpc"), NULL, reinterpret_cast<RPC_WSTR>(L"gdipp"), NULL, &binding_str);
  23         if (rpc_status != RPC_S_OK)
  24                 return false;
  25
  26         rpc_status = RpcBindingFromStringBinding(binding_str, &h_gdipp_rpc);
  27         if (rpc_status != RPC_S_OK)
  28                 return false;
  29
  30         rpc_status = RpcStringFree(&binding_str);
  31         if (rpc_status != RPC_S_OK)
  32                 return false;
  33
  34         return true;
  35 }
  36
  37 }
  38
  39 void __RPC_FAR *__RPC_USER MIDL_user_allocate(size_t size)
  40 {
  41         return HeapAlloc(gdipp::process_heap, HEAP_GENERATE_EXCEPTIONS, size);
  42 }
  43
  44 void __RPC_USER MIDL_user_free(void __RPC_FAR *ptr)
  45 {
  46         HeapFree(gdipp::process_heap, 0, ptr);
  47 }
  48
  49 BOOL APIENTRY DllMain(HMODULE hModule, DWORD ul_reason_for_call, LPVOID lpReserved)
  50 {
  51         BOOL b_ret;
  52
  53         switch (ul_reason_for_call)
  54         {
  55                 case DLL_PROCESS_ATTACH:
  56                 {
  57                         gdipp::h_self = hModule;
  58
  59                         wchar_t this_proc_name[MAX_PATH];
  60                         DWORD dw_ret = GetModuleBaseNameW(GetCurrentProcess(), NULL, this_proc_name, MAX_PATH);
  61                         assert(dw_ret != 0);
  62
  63                         if (gdipp::exclude_config::is_process_excluded(gdipp::config_instance, this_proc_name))
  64                                 return FALSE;
  65
  66                         OSVERSIONINFO ver_info = {sizeof(OSVERSIONINFO)};
  67                         b_ret = GetVersionEx(&ver_info);
  68                         if (!b_ret)
  69                                 return FALSE;
  70                         gdipp::os_support_directwrite = (ver_info.dwMajorVersion >= 6);
  71
  72                         gdipp::scoped_rw_lock::initialize();
  73
  74                         if (!gdipp::init_rpc_client())
  75                                 return FALSE;
  76
  77                         gdipp::client_config_instance.parse(gdipp::config_instance);
  78
  79                         if (!gdipp::hook_instance.start())
  80                                 return FALSE;
  81
  82                         break;
  83                 }
  84                 case DLL_PROCESS_DETACH:
  85                         gdipp::hook_instance.stop();
  86                         break;
  87         }
  88
  89         return TRUE;
  90 }