15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved.
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file.
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// MeasurePageLoadTime.cpp : Implementation of DLL Exports.
65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "stdafx.h"
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "resource.h"
95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "MeasurePageLoadTime.h"
105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class CMeasurePageLoadTimeModule : public CAtlDllModuleT< CMeasurePageLoadTimeModule >
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles){
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)public :
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)	DECLARE_LIBID(LIBID_MeasurePageLoadTimeLib)
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)	DECLARE_REGISTRY_APPID_RESOURCEID(IDR_MEASUREPAGELOADTIME, "{56C6D9F9-643C-4F6E-906C-5F7CECB23C24}")
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)};
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)CMeasurePageLoadTimeModule _AtlModule;
205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifdef _MANAGED
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#pragma managed(push, off)
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// DLL Entry Point
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)extern "C" BOOL WINAPI DllMain(HINSTANCE hInstance, DWORD dwReason, LPVOID lpReserved)
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles){
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    if (dwReason == DLL_PROCESS_ATTACH)
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    {
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)        DisableThreadLibraryCalls(hInstance);
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    }
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return _AtlModule.DllMain(dwReason, lpReserved);
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifdef _MANAGED
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#pragma managed(pop)
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Used to determine whether the DLL can be unloaded by OLE
445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)STDAPI DllCanUnloadNow(void)
455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles){
465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return _AtlModule.DllCanUnloadNow();
475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Returns a class factory to create an object of the requested type
515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)STDAPI DllGetClassObject(REFCLSID rclsid, REFIID riid, LPVOID* ppv)
525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles){
535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    return _AtlModule.DllGetClassObject(rclsid, riid, ppv);
545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// DllRegisterServer - Adds entries to the system registry
585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)STDAPI DllRegisterServer(void)
595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles){
605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    // registers object, typelib and all interfaces in typelib
615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)    HRESULT hr = _AtlModule.DllRegisterServer();
625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)	return hr;
635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// DllUnregisterServer - Removes entries from the system registry
675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)STDAPI DllUnregisterServer(void)
685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles){
695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)	HRESULT hr = _AtlModule.DllUnregisterServer();
705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)	return hr;
715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
73