Microsoft MVP성태의 닷넷 이야기
.NET Framework: 576. 기본적인 CLR Profiler 소스 코드 설명 [링크 복사], [링크+제목 복사]
조회: 16968
글쓴 사람
정성태 (techsharer at outlook.com)
홈페이지
첨부 파일
(연관된 글이 4개 있습니다.)
(시리즈 글이 9개 있습니다.)
.NET Framework: 216. 라이선스까지도 뛰어넘는 .NET Profiler
; https://www.sysnet.pe.kr/2/0/1046

.NET Framework: 336. .NET Profiler가 COM 개체일까?
; https://www.sysnet.pe.kr/2/0/1352

.NET Framework: 576. 기본적인 CLR Profiler 소스 코드 설명
; https://www.sysnet.pe.kr/2/0/10950

.NET Framework: 582. CLR Profiler - 별도 정의한 .NET 코드를 호출하도록 IL 코드 변경
; https://www.sysnet.pe.kr/2/0/10959

.NET Framework: 808. .NET Profiler - GAC 모듈에서 GAC 비-등록 모듈을 참조하는 경우의 문제
; https://www.sysnet.pe.kr/2/0/11810

오류 유형: 672. AllowPartiallyTrustedCallers 특성이 적용된 어셈블리의 struct 멤버 메서드를 재정의하면 System.Security.VerificationException 예외 발생
; https://www.sysnet.pe.kr/2/0/12384

.NET Framework: 987. .NET Profiler - FunctionID와 연관된 ClassID를 구할 수 없는 문제
; https://www.sysnet.pe.kr/2/0/12465

.NET Framework: 1041. C# - AssemblyID, ModuleID를 관리 코드에서 구하는 방법
; https://www.sysnet.pe.kr/2/0/12605

닷넷: 2228. .NET Profiler - IMetaDataEmit2::DefineMethodSpec 사용법
; https://www.sysnet.pe.kr/2/0/13576




기본적인 CLR Profiler 소스 코드 설명

사실, 기본 뼈대 만드는 방법을 다음의 글에 이미 소개했었는데요.

라이선스까지도 뛰어넘는 .NET Profiler
; https://www.sysnet.pe.kr/2/0/1046

소스 코드를 포함시켜 두지 않았군요. ^^;;; 그래서, 가장 기본적인 코드만 만족하는 CLR Profiler 예제 소스 코드를 만들어 봤습니다.




우선, "Visual C++" / "ATL" 범주의 "ATLProject" 유형으로 프로젝트를 생성해 줍니다.

그다음 솔루션 탐색기에서 생성된 프로젝트에 대해 마우스 우클릭을 해 "Add" / "Class" 메뉴를 선택하고 뜨는 마법사 창에서 "Visual C++" / "ATL" 범주의 "ATL Simple Object"를 추가합니다. (이 글에서는 추가된 클래스 명이 CBasicClrProfiler)

기본적인 COM 객체가 만들어졌으니, 이제 CLR Profiler의 규약에 맞는 작업을 추가합니다. 먼저, ICorProfilerCabllback 인터페이스를 구현해야 하는데, 이를 위해 다음과 같은 기본 뼈대를 이미 갖춘 Impl 헤더 파일을 마련해 프로젝트에 추가합니다.

// ICorProfilerCallbackImpl.h
#pragma once

#include <cor.h>
#include <corprof.h>

#include <atlbase.h>

#pragma comment (lib, "corguids.lib")

template<class T>
class ATL_NO_VTABLE ICorProfilerCallback3Impl : 
    public ICorProfilerCallback3
{
public:

    ICorProfilerCallback3Impl() {};
    virtual ~ICorProfilerCallback3Impl() {};

    STDMETHOD(QueryInterface)(REFIID riid, void** ppvObject) = 0;
    _ATL_DEBUG_ADDREF_RELEASE_IMPL(ICorProfilerCallback3Impl)
        
    STDMETHOD(Initialize)(IUnknown *pICorProfilerInfoUnk)
    {
        UNREFERENCED_PARAMETER(pICorProfilerInfoUnk);
        return E_NOTIMPL;
    }

    STDMETHOD(Shutdown)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(AppDomainCreationStarted)(AppDomainID appDomainId)
    {
        UNREFERENCED_PARAMETER(appDomainId);
        return E_NOTIMPL;
    }

    STDMETHOD(AppDomainCreationFinished)(AppDomainID appDomainId, HRESULT hrStatus)
    {
        UNREFERENCED_PARAMETER(appDomainId);
        UNREFERENCED_PARAMETER(hrStatus);
        return E_NOTIMPL;
    }

    STDMETHOD(AppDomainShutdownStarted)(AppDomainID appDomainId)
    {
        UNREFERENCED_PARAMETER(appDomainId);
        return E_NOTIMPL;
    }

    STDMETHOD(AppDomainShutdownFinished)(AppDomainID appDomainId, HRESULT hrStatus)
    {
        UNREFERENCED_PARAMETER(appDomainId);
        UNREFERENCED_PARAMETER(hrStatus);
        return E_NOTIMPL;
    }

    STDMETHOD(AssemblyLoadStarted)(AssemblyID assemblyId)
    {
        UNREFERENCED_PARAMETER(assemblyId);
        return S_OK;
    }

    STDMETHOD(AssemblyLoadFinished)(AssemblyID assemblyId, HRESULT hrStatus)
    {
        UNREFERENCED_PARAMETER(assemblyId);
        UNREFERENCED_PARAMETER(hrStatus);
        return S_OK;
    }

    STDMETHOD(AssemblyUnloadStarted)(AssemblyID assemblyId)
    {
        UNREFERENCED_PARAMETER(assemblyId);
        return E_NOTIMPL;
    }

    STDMETHOD(AssemblyUnloadFinished)(AssemblyID assemblyId, HRESULT hrStatus)
    {
        UNREFERENCED_PARAMETER(assemblyId);
        UNREFERENCED_PARAMETER(hrStatus);
        return E_NOTIMPL;
    }

    STDMETHOD(ModuleLoadStarted)(ModuleID moduleId)
    {
        UNREFERENCED_PARAMETER(moduleId);
        return S_OK;
    }

    STDMETHOD(ModuleLoadFinished)(ModuleID moduleId, HRESULT hrStatus)
    {
        UNREFERENCED_PARAMETER(moduleId);
        UNREFERENCED_PARAMETER(hrStatus);
        return S_OK;
    }

    STDMETHOD(ModuleUnloadStarted)(ModuleID moduleId)
    {
        UNREFERENCED_PARAMETER(moduleId);
        return E_NOTIMPL;
    }

    STDMETHOD(ModuleUnloadFinished)(ModuleID moduleId, HRESULT hrStatus)
    {
        UNREFERENCED_PARAMETER(moduleId);
        UNREFERENCED_PARAMETER(hrStatus);
        return E_NOTIMPL;
    }

    STDMETHOD(ModuleAttachedToAssembly)(ModuleID moduleId, AssemblyID assemblyId)
    {
        UNREFERENCED_PARAMETER(moduleId);
        UNREFERENCED_PARAMETER(assemblyId);
        return S_OK;
    }

    STDMETHOD(ClassLoadStarted)(ClassID classId)
    {
        UNREFERENCED_PARAMETER(classId);
        return E_NOTIMPL;
    }

    STDMETHOD(ClassLoadFinished)(ClassID classId, HRESULT hrStatus)
    {
        UNREFERENCED_PARAMETER(classId);
        UNREFERENCED_PARAMETER(hrStatus);
        return E_NOTIMPL;
    }

    STDMETHOD(ClassUnloadStarted)(ClassID classId)
    {
        UNREFERENCED_PARAMETER(classId);
        return E_NOTIMPL;
    }

    STDMETHOD(ClassUnloadFinished)(ClassID classId, HRESULT hrStatus)
    {
        UNREFERENCED_PARAMETER(classId);
        UNREFERENCED_PARAMETER(hrStatus);
        return E_NOTIMPL;
    }

    STDMETHOD(FunctionUnloadStarted)(FunctionID functionId)
    {
        UNREFERENCED_PARAMETER(functionId);
        return E_NOTIMPL;
    }

    STDMETHOD(JITCompilationStarted)(FunctionID functionId, BOOL fIsSafeToBlock)
    {
        UNREFERENCED_PARAMETER(functionId);
        UNREFERENCED_PARAMETER(fIsSafeToBlock);
        return S_OK;
    }

    STDMETHOD(JITCompilationFinished)(FunctionID functionId, HRESULT hrStatus,
                                        BOOL fIsSafeToBlock)
    {
        UNREFERENCED_PARAMETER(functionId);
        UNREFERENCED_PARAMETER(hrStatus);
        UNREFERENCED_PARAMETER(fIsSafeToBlock);
        return S_OK;
    }

    STDMETHOD(JITCachedFunctionSearchStarted)(FunctionID functionId,
                                                BOOL * pbUseCachedFunction)
    {
        UNREFERENCED_PARAMETER(functionId);
        UNREFERENCED_PARAMETER(pbUseCachedFunction);
        return E_NOTIMPL;
    }

    STDMETHOD(JITCachedFunctionSearchFinished)(FunctionID functionId,
                                                COR_PRF_JIT_CACHE result)
    {
        UNREFERENCED_PARAMETER(functionId);
        UNREFERENCED_PARAMETER(result);
        return E_NOTIMPL;
    }

    STDMETHOD(JITFunctionPitched)(FunctionID functionId)
    {
        UNREFERENCED_PARAMETER(functionId);
        return E_NOTIMPL;
    }

    STDMETHOD(JITInlining)(FunctionID callerId, FunctionID calleeId,
                            BOOL * pfShouldInline)
    {
        UNREFERENCED_PARAMETER(callerId);
        UNREFERENCED_PARAMETER(calleeId);
        UNREFERENCED_PARAMETER(pfShouldInline);
        return E_NOTIMPL;
    }

    STDMETHOD(ThreadCreated)(ThreadID threadId)
    {
        UNREFERENCED_PARAMETER(threadId);
        return E_NOTIMPL;
    }

    STDMETHOD(ThreadDestroyed)(ThreadID threadId)
    {
        UNREFERENCED_PARAMETER(threadId);
        return E_NOTIMPL;
    }

    STDMETHOD(ThreadAssignedToOSThread)(ThreadID managedThreadId,
                                            ULONG osThreadId)
    {
        UNREFERENCED_PARAMETER(managedThreadId);
        UNREFERENCED_PARAMETER(osThreadId);
        return E_NOTIMPL;
    }

    STDMETHOD(RemotingClientInvocationStarted)()
    {
        return E_NOTIMPL;
    }
    
    STDMETHOD(RemotingClientSendingMessage)(GUID * pCookie, BOOL fIsAsync)
    {
        UNREFERENCED_PARAMETER(pCookie);
        UNREFERENCED_PARAMETER(fIsAsync);
        return E_NOTIMPL;
    }

    STDMETHOD(RemotingClientReceivingReply)(GUID * pCookie, BOOL fIsAsync)
    {
        UNREFERENCED_PARAMETER(pCookie);
        UNREFERENCED_PARAMETER(fIsAsync);
        return E_NOTIMPL;
    }

    STDMETHOD(RemotingClientInvocationFinished)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(RemotingServerReceivingMessage)(GUID * pCookie, BOOL fIsAsync)
    {
        UNREFERENCED_PARAMETER(pCookie);
        UNREFERENCED_PARAMETER(fIsAsync);
        return E_NOTIMPL;
    }

    STDMETHOD(RemotingServerInvocationStarted)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(RemotingServerInvocationReturned)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(RemotingServerSendingReply)(GUID * pCookie, BOOL fIsAsync)
    {
        UNREFERENCED_PARAMETER(pCookie);
        UNREFERENCED_PARAMETER(fIsAsync);
        return E_NOTIMPL;
    }

    STDMETHOD(UnmanagedToManagedTransition)(FunctionID functionId,
                                            COR_PRF_TRANSITION_REASON reason)
    {
        UNREFERENCED_PARAMETER(functionId);
        UNREFERENCED_PARAMETER(reason);
        return E_NOTIMPL;
    }

    STDMETHOD(ManagedToUnmanagedTransition)(FunctionID functionId,
                                            COR_PRF_TRANSITION_REASON reason)
    {
        UNREFERENCED_PARAMETER(functionId);
        UNREFERENCED_PARAMETER(reason);
        return E_NOTIMPL;
    }

    STDMETHOD(RuntimeSuspendStarted)(COR_PRF_SUSPEND_REASON suspendReason)
    {
        UNREFERENCED_PARAMETER(suspendReason);
        return E_NOTIMPL;
    }

    STDMETHOD(RuntimeSuspendFinished)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(RuntimeSuspendAborted)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(RuntimeResumeStarted)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(RuntimeResumeFinished)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(RuntimeThreadSuspended)(ThreadID threadId)
    {
        UNREFERENCED_PARAMETER(threadId);
        return E_NOTIMPL;
    }

    STDMETHOD(RuntimeThreadResumed)(ThreadID threadId)
    {
        UNREFERENCED_PARAMETER(threadId);
        return E_NOTIMPL;
    }

    STDMETHOD(MovedReferences)(ULONG cMovedObjectIDRanges,
                                ObjectID oldObjectIDRangeStart[],
                                ObjectID newObjectIDRangeStart[],
                                ULONG cObjectIDRangeLength[])
    {
        UNREFERENCED_PARAMETER(cMovedObjectIDRanges);
        UNREFERENCED_PARAMETER(oldObjectIDRangeStart);
        UNREFERENCED_PARAMETER(newObjectIDRangeStart);
        UNREFERENCED_PARAMETER(cObjectIDRangeLength);
        return E_NOTIMPL;
    }

    STDMETHOD(ObjectAllocated)(ObjectID objectId, ClassID classId)
    {
        UNREFERENCED_PARAMETER(objectId);
        UNREFERENCED_PARAMETER(classId);
        return E_NOTIMPL;
    }

    STDMETHOD(ObjectsAllocatedByClass)(ULONG cClassCount, ClassID classIds[],
                                        ULONG cObjects[])
    {
        UNREFERENCED_PARAMETER(cClassCount);
        UNREFERENCED_PARAMETER(classIds);
        UNREFERENCED_PARAMETER(cObjects);
        return E_NOTIMPL;
    }

    STDMETHOD(ObjectReferences)(ObjectID objectId, ClassID classId,
                                    ULONG cObjectRefs, ObjectID objectRefIds[])
    {
        UNREFERENCED_PARAMETER(objectId);
        UNREFERENCED_PARAMETER(classId);
        UNREFERENCED_PARAMETER(cObjectRefs);
        UNREFERENCED_PARAMETER(objectRefIds);
        return E_NOTIMPL;
    }

    STDMETHOD(RootReferences)(ULONG cRootRefs, ObjectID rootRefIds[])
    {
        UNREFERENCED_PARAMETER(cRootRefs);
        UNREFERENCED_PARAMETER(rootRefIds);
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionThrown)(ObjectID thrownObjectId)
    {
        UNREFERENCED_PARAMETER(thrownObjectId);
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionSearchFunctionEnter)(FunctionID functionId)
    {
        UNREFERENCED_PARAMETER(functionId);
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionSearchFunctionLeave)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionSearchFilterEnter)(FunctionID functionId)
    {
        UNREFERENCED_PARAMETER(functionId);
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionSearchFilterLeave)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionSearchCatcherFound)(FunctionID functionId)
    {
        UNREFERENCED_PARAMETER(functionId);
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionOSHandlerEnter)(FunctionID functionId)
    {
        UNREFERENCED_PARAMETER(functionId);
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionOSHandlerLeave)(FunctionID functionId)
    {
        UNREFERENCED_PARAMETER(functionId);
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionUnwindFunctionEnter)(FunctionID functionId)
    {
        UNREFERENCED_PARAMETER(functionId);
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionUnwindFunctionLeave)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionUnwindFinallyEnter)(FunctionID functionId)
    {
        UNREFERENCED_PARAMETER(functionId);
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionUnwindFinallyLeave)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionCatcherEnter)(FunctionID functionId, ObjectID objectId)
    {
        UNREFERENCED_PARAMETER(functionId);
        UNREFERENCED_PARAMETER(objectId);
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionCatcherLeave)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(COMClassicVTableCreated)(ClassID wrappedClassId,
                        REFGUID implementedIID, void *pVTable, ULONG cSlots)
    {
        UNREFERENCED_PARAMETER(wrappedClassId);
        UNREFERENCED_PARAMETER(implementedIID);
        UNREFERENCED_PARAMETER(pVTable);
        UNREFERENCED_PARAMETER(cSlots);
        return E_NOTIMPL;
    }

    STDMETHOD(COMClassicVTableDestroyed)(ClassID wrappedClassId,
                        REFGUID implementedIID, void *pVTable)
    {
        UNREFERENCED_PARAMETER(wrappedClassId);
        UNREFERENCED_PARAMETER(implementedIID);
        UNREFERENCED_PARAMETER(pVTable);
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionCLRCatcherFound)(void)
    {
        return E_NOTIMPL;
    }

    STDMETHOD(ExceptionCLRCatcherExecute)(void)
    {
        return E_NOTIMPL;
    }

    STDMETHOD(ThreadNameChanged)(ThreadID threadId, ULONG cchName, WCHAR* name)
    {
        UNREFERENCED_PARAMETER(threadId);
        UNREFERENCED_PARAMETER(cchName);
        UNREFERENCED_PARAMETER(name);
        return E_NOTIMPL;
    }
    
    STDMETHOD(GarbageCollectionStarted)( 
            /* [in] */ int cGenerations,
            /* [length_is][size_is][in] */ BOOL generationCollected[  ],
            /* [in] */ COR_PRF_GC_REASON reason)
    {
        UNREFERENCED_PARAMETER(cGenerations);
        UNREFERENCED_PARAMETER(generationCollected);
        UNREFERENCED_PARAMETER(reason);
        return E_NOTIMPL;
    }

    STDMETHOD(SurvivingReferences)( 
            /* [in] */ ULONG cSurvivingObjectIDRanges,
            /* [size_is][in] */ ObjectID objectIDRangeStart[  ],
            /* [size_is][in] */ ULONG cObjectIDRangeLength[  ])
    {
        UNREFERENCED_PARAMETER(cSurvivingObjectIDRanges);
        UNREFERENCED_PARAMETER(objectIDRangeStart);
        UNREFERENCED_PARAMETER(cObjectIDRangeLength);
        return E_NOTIMPL;
    }

    STDMETHOD(GarbageCollectionFinished)()
    {
        return E_NOTIMPL;
    }
        
    STDMETHOD(FinalizeableObjectQueued)( 
            /* [in] */ DWORD finalizerFlags,
            /* [in] */ ObjectID objectID)
    {
        UNREFERENCED_PARAMETER(finalizerFlags);
        UNREFERENCED_PARAMETER(objectID);
        return E_NOTIMPL;
    }
        
    STDMETHOD(RootReferences2)( 
            /* [in] */ ULONG cRootRefs,
            /* [size_is][in] */ ObjectID rootRefIds[  ],
            /* [size_is][in] */ COR_PRF_GC_ROOT_KIND rootKinds[  ],
            /* [size_is][in] */ COR_PRF_GC_ROOT_FLAGS rootFlags[  ],
            /* [size_is][in] */ UINT_PTR rootIds[  ])
    {
        UNREFERENCED_PARAMETER(cRootRefs);
        UNREFERENCED_PARAMETER(rootRefIds);
        UNREFERENCED_PARAMETER(rootKinds);
        UNREFERENCED_PARAMETER(rootFlags);
        UNREFERENCED_PARAMETER(rootIds);
        return E_NOTIMPL;
    }
        
    STDMETHOD(HandleCreated)( 
            /* [in] */ GCHandleID handleId,
            /* [in] */ ObjectID initialObjectId)
    {
        UNREFERENCED_PARAMETER(handleId);
        UNREFERENCED_PARAMETER(initialObjectId);
        return E_NOTIMPL;
    }
        
    STDMETHOD(HandleDestroyed)( 
            /* [in] */ GCHandleID handleId)
    {
        UNREFERENCED_PARAMETER(handleId);
        return E_NOTIMPL;
    }
    
    STDMETHOD(InitializeForAttach)(/* [in] */ IUnknown *pCorProfilerInfoUnk,
        /* [in] */ void *pvClientData,
        /* [in] */ UINT cbClientData)
    {
        UNREFERENCED_PARAMETER(pCorProfilerInfoUnk);
        UNREFERENCED_PARAMETER(pvClientData);
        UNREFERENCED_PARAMETER(cbClientData);
        return E_NOTIMPL;
    }

    STDMETHOD(ProfilerAttachComplete)()
    {
        return E_NOTIMPL;
    }

    STDMETHOD(ProfilerDetachSucceeded)()
    {
        return E_NOTIMPL;
    }
};

우리가 만든 클래스(CBasicClrProfiler)에서는 이를 상속받도록 만들고, COM_INTERFACE_ENTRY를 추가해 줍니다.

#pragma once
#include "resource.h"       // main symbols

#include "SampleProfiler_i.h"

#include "ICorProfilerCallback3Impl.h"

// ..[생략]...

class ATL_NO_VTABLE CBasicClrProfiler :
    // ..[생략]...
    public ICorProfilerCallback3Impl<CBasicClrProfiler>
{
public:
    CBasicClrProfiler()
    {
    }

DECLARE_REGISTRY_RESOURCEID(IDR_BASICCLRPROFILER)

BEGIN_COM_MAP(CBasicClrProfiler)
    COM_INTERFACE_ENTRY(IBasicClrProfiler)
    COM_INTERFACE_ENTRY(IDispatch)
    COM_INTERFACE_ENTRY(ICorProfilerCallback)
    COM_INTERFACE_ENTRY(ICorProfilerCallback2)
    COM_INTERFACE_ENTRY(ICorProfilerCallback3)
END_COM_MAP()

// ..[생략]...
};

OBJECT_ENTRY_AUTO(__uuidof(BasicClrProfiler), CBasicClrProfiler)

여기까지가 CLR이 요구하는 프로파일러 요건의 전부입니다. CLR은 이 클래스를 로드해서 ICorProfilerCallback::Initialize 메서드를 호출하는데, 이 메서드에서 우리는 CLR에게 관심있는 콜백이 무엇인지를 등록해 주면 됩니다.

일단, Initialize 메서드까지 불리는 것을 확인해 보고 다음으로 넘어가겠습니다.

테스트를 위해 C# EXE 프로젝트를 하나 만들고,

using System;

namespace ConsoleApplication1
{
    class Program
    {
        static void Main(string[] args)
        {
            Console.WriteLine("TEST");
            Console.ReadLine();
        }
    }
}

C++ 프로파일러 프로젝트의 디버깅 시작 옵션을 C# EXE로 맞춘 후 CLR에게 프로파일러를 로드하도록 환경 변수를 설정해줍니다.

sample_clr_profiler_1.png

Command: $(SolutionDir)ConsoleApplication1\bin\Debug\ConsoleApplication1.exe

Debugger Type: Native Only

Environment: COR_ENABLE_PROFILING=1
             COR_PROFILER={F89DDF6B-9451-4EA9-8E02-7902E61C0BCC}
             COR_PROFILER_PATH=$(TargetPath)

CLR이 로드 후 단 한번 불러 줄 Initialize 메서드에서 닷넷 모듈이 로드되는 시점의 콜백을 받고 싶다면 다음과 같이 구현해 주면 됩니다.

// BasicClrProfiler.h

#pragma once
#include "resource.h"       // main symbols

#include "SampleProfiler_i.h"

#include "ICorProfilerCallback3Impl.h"

// ..[생략]...

class ATL_NO_VTABLE CBasicClrProfiler :
    // ..[생략]...
    public ICorProfilerCallback3Impl<CBasicClrProfiler>
{
public:

// ..[생략]...
    STDMETHOD(Initialize)(IUnknown *pICorProfilerInfoUnk);

    CComQIPtr m_pICorProfilerInfo2;

// ..[생략]...
};

OBJECT_ENTRY_AUTO(__uuidof(BasicClrProfiler), CBasicClrProfiler)

// BasicClrProfiler.cpp : Implementation of CBasicClrProfiler

#include "stdafx.h"
#include "BasicClrProfiler.h"

HRESULT CBasicClrProfiler::Initialize(IUnknown *pICorProfilerInfoUnk)
{
    if (pICorProfilerInfoUnk == NULL)
    {
	    return S_OK;
    }

    m_pICorProfilerInfo2 = pICorProfilerInfoUnk;

    DWORD dwEventMask = COR_PRF_MONITOR_MODULE_LOADS;
    m_pICorProfilerInfo2->SetEventMask(dwEventMask);

    return S_OK;
}

이제 Initialize에 BP(BreakPoint)를 걸고 "F5" 키를 눌러 디버깅을 시작하면 끝!




COR_PRF_MONITOR_MODULE_LOADS 관련 이벤트를 받겠다고 명시했으니 이제 CLR은 모듈의 로드/언로드에 따라 다음과 같은 콜백을 호출해 줍니다.

  • ModuleLoadStarted
  • ModuleLoadFinished
  • ModuleUnloadStarted
  • ModuleUnloadFinished

여기서는 ModuleLoadFinished 콜백을 받아 어떤 모듈이 로드되었는지에 대한 정보를 출력하겠습니다.

이를 위해 헤더 파일과 소스 코드에 다음과 같은 코드를 추가합니다.

// BasicClrProfiler.h

#pragma once
#include "resource.h"       // main symbols

#include "SampleProfiler_i.h"

#include "ICorProfilerCallback3Impl.h"

// ..[생략]...

class ATL_NO_VTABLE CBasicClrProfiler :
    // ..[생략]...
    public ICorProfilerCallback3Impl<CBasicClrProfiler>
{
public:

// ..[생략]...
    STDMETHOD(ModuleLoadFinished)(ModuleID moduleId, HRESULT hrStatus);
    void OutputDebugText(const wchar_t* format, ...);

// ..[생략]...
};

OBJECT_ENTRY_AUTO(__uuidof(BasicClrProfiler), CBasicClrProfiler)

// BasicClrProfiler.cpp : Implementation of CBasicClrProfiler

#include "stdafx.h"
#include "BasicClrProfiler.h"

#include <Strsafe.h>

// ...[생략]...

HRESULT CBasicClrProfiler::ModuleLoadFinished(ModuleID moduleId, HRESULT hrStatus)
{
    ULONG cchModule = _MAX_PATH;
    ULONG rCchModule = 0;
    AssemblyID assemblyId = 0;
    LPCBYTE pModuleBaseLoadAddress;
    wchar_t szModule[_MAX_PATH];

    HRESULT hr = m_pICorProfilerInfo2->GetModuleInfo(moduleId,
        &pModuleBaseLoadAddress, cchModule, &rCchModule, szModule, &assemblyId);

    OutputDebugText(L"=============== %s loaded at 0x%x, ModuleID = 0x%x\n", szModule, pModuleBaseLoadAddress, moduleId);

    return S_OK;
}

void CBasicClrProfiler::OutputDebugText(const wchar_t* format, ...)
{
    wchar_t buffer[4096];

    va_list vaList;
    va_start(vaList, format);
    StringCchVPrintfW(buffer, 4096, format, vaList);

    va_end(vaList);
    OutputDebugString(buffer);
}

이렇게 하고 실행하면, Debug 창에 다음과 같은 출력을 확인할 수 있습니다.

C:\WINDOWS\Microsoft.Net\assembly\GAC_32\mscorlib\v4.0_4.0.0.0__b77a5c561934e089\mscorlib.dll loaded at 0x6a6b0000, ModuleID = 0x6a6b1000
C:\...\ConsoleApplication1\bin\Debug\ConsoleApplication1.exe loaded at 0x5d0000, ModuleID = 0x973fbc

이 때 출력되는 "loaded at" 주소 값은 windbg.exe에서 "lm" 명령으로 확인했을 때의 모듈 로드 주소입니다. 그리고 ModuleID는 다음과 같이 !name2ee로 출력했을 때의 Module 값에 대응합니다.

0:007> !name2ee *!ConsoleApplication1
Module:      6a6b1000
Assembly:    mscorlib.dll
--------------------------------------
Module:      00973fbc
Assembly:    ConsoleApplication1.exe

이 정도면, 대충 프로파일러가 돌아가는 구조가 눈에 보이실 것입니다.

(첨부한 파일은 이 글의 예제 코드를 포함합니다.)




[이 글에 대해서 여러분들과 의견을 공유하고 싶습니다. 틀리거나 미흡한 부분 또는 의문 사항이 있으시면 언제든 댓글 남겨주십시오.]

[연관 글]






[최초 등록일: ]
[최종 수정일: 7/9/2022]

Creative Commons License
이 저작물은 크리에이티브 커먼즈 코리아 저작자표시-비영리-변경금지 2.0 대한민국 라이센스에 따라 이용하실 수 있습니다.
by SeongTae Jeong, mailto:techsharer at outlook.com

비밀번호

댓글 작성자
 



2022-07-09 11시50분
Phrack: .NET Instrumentation via MSIL bytecode injection (Antonio 's4tan' Parata)
; https://www.exploit-db.com/papers/43524
; https://web.archive.org/web/20221206170326/https://www.exploit-db.com/papers/43524
정성태
2023-01-13 08시14분
정성태

1  2  3  [4]  5  6  7  8  9  10  11  12  13  14  15  ...
NoWriterDateCnt.TitleFile(s)
13532정성태1/17/20242188닷넷: 2205. C# - SuperSimpleTcp 사용 시 주의할 점파일 다운로드1
13531정성태1/16/20242233닷넷: 2204. C# - TCP KeepAlive에 새로 추가된 Retry 옵션파일 다운로드1
13530정성태1/15/20242188닷넷: 2203. C# - Python과의 AES 암호화 연동파일 다운로드1
13529정성태1/15/20242048닷넷: 2202. C# - PublishAot의 glibc에 대한 정적 링킹하는 방법
13528정성태1/14/20242160Linux: 68. busybox 컨테이너에서 실행 가능한 C++, Go 프로그램 빌드
13527정성태1/14/20242105오류 유형: 892. Visual Studio - Failed to launch debug adapter. Additional information may be available in the output window.
13526정성태1/14/20242202닷넷: 2201. C# - Facebook 연동 / 사용자 탈퇴 처리 방법
13525정성태1/13/20242156오류 유형: 891. Visual Studio - Web Application을 실행하지 못하는 IISExpress
13524정성태1/12/20242198오류 유형: 890. 한국투자증권 KIS Developers OpenAPI - GW라우팅 중 오류가 발생했습니다.
13523정성태1/12/20242026오류 유형: 889. Visual Studio - error : A project with that name is already opened in the solution.
13522정성태1/11/20242187닷넷: 2200. C# - HttpClient.PostAsJsonAsync 호출 시 "Transfer-Encoding: chunked" 대신 "Content-Length" 헤더 처리
13521정성태1/11/20242253닷넷: 2199. C# - 한국투자증권 KIS Developers OpenAPI의 WebSocket Ping, Pong 처리
13520정성태1/10/20241985오류 유형: 888. C# - Unable to resolve service for type 'Microsoft.Extensions.ObjectPool.ObjectPool`....'
13519정성태1/10/20242078닷넷: 2198. C# - Reflection을 이용한 ClientWebSocket의 Ping 호출파일 다운로드1
13518정성태1/9/20242349닷넷: 2197. C# - ClientWebSocket의 Ping, Pong 처리
13517정성태1/8/20242196스크립트: 63. Python - 공개 패키지를 이용한 위성 이미지 생성 (pystac_client, odc.stac)
13516정성태1/7/20242300닷넷: 2196. IIS - AppPool의 "Disable Overlapped Recycle" 옵션의 부작용
13515정성태1/6/20242579닷넷: 2195. async 메서드 내에서 C# 7의 discard 구문 활용 사례 [1]
13514정성태1/5/20242237개발 환경 구성: 702. IIS - AppPool의 "Disable Overlapped Recycle" 옵션
13513정성태1/5/20242161닷넷: 2194. C# - WebActivatorEx / System.Web의 PreApplicationStartMethod 특성
13512정성태1/4/20242125개발 환경 구성: 701. IIS - w3wp.exe 프로세스의 ASP.NET 런타임을 항상 Warmup 모드로 유지하는 preload Enabled 설정
13511정성태1/4/20242144닷넷: 2193. C# - ASP.NET Web Application + OpenAPI(Swashbuckle) 스펙 제공
13510정성태1/3/20242079닷넷: 2192. C# - 특정 실행 파일이 있는지 확인하는 방법 (Linux)
13509정성태1/3/20242103오류 유형: 887. .NET Core 2 이하의 프로젝트에서 System.Runtime.CompilerServices.Unsafe doesn't support netcoreapp2.0.
13508정성태1/3/20242109오류 유형: 886. ORA-28000: the account is locked
13507정성태1/2/20242785닷넷: 2191. C# - IPGlobalProperties를 이용해 netstat처럼 사용 중인 Socket 목록 구하는 방법파일 다운로드1
1  2  3  [4]  5  6  7  8  9  10  11  12  13  14  15  ...