Microsoft MVP성태의 닷넷 이야기
.NET Framework: 576. 기본적인 CLR Profiler 소스 코드 설명 [링크 복사], [링크+제목 복사],
조회: 26183
글쓴 사람
정성태 (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)
13768정성태10/15/20245379C/C++: 179. C++ - _O_WTEXT, _O_U16TEXT, _O_U8TEXT의 Unicode stream 모드파일 다운로드2
13767정성태10/14/20244764오류 유형: 929. bpftrace 수행 시 "ERROR: Could not resolve symbol: /proc/self/exe:BEGIN_trigger"
13766정성태10/14/20244549C/C++: 178. C++ - 파일에 대한 Text 모드의 "translated" 동작파일 다운로드1
13765정성태10/12/20245258오류 유형: 928. go build 시 "package maps is not in GOROOT" 오류
13764정성태10/11/20245624Linux: 85. Ubuntu - 원하는 golang 버전 설치
13763정성태10/11/20244979Linux: 84. WSL / Ubuntu 20.04 - bpftool 설치
13762정성태10/11/20245004Linux: 83. WSL / Ubuntu 22.04 - bpftool 설치
13761정성태10/11/20244911오류 유형: 927. WSL / Ubuntu - /usr/include/linux/types.h:5:10: fatal error: 'asm/types.h' file not found
13760정성태10/11/20245438Linux: 82. Ubuntu - clang 최신(stable) 버전 설치
13759정성태10/10/20246359C/C++: 177. C++ - 자유 함수(free function) 및 주소 지정 가능한 함수(addressable function) [6]
13758정성태10/8/20245570오류 유형: 926. dotnet tools를 sudo로 실행하는 경우 command not found
13757정성태10/8/20245508닷넷: 2306. Linux - dotnet tool의 설치 디렉터리가 PATH 환경변수에 자동 등록이 되는 이유
13756정성태10/8/20245617오류 유형: 925. ssh로 docker 접근을 할 때 "... malformed HTTP status code ..." 오류 발생
13755정성태10/7/20246009닷넷: 2305. C# 13 - (9) 메서드 바인딩의 우선순위를 지정하는 OverloadResolutionPriority 특성 도입 (Overload resolution priority)파일 다운로드1
13754정성태10/4/20245570닷넷: 2304. C# 13 - (8) 부분 메서드 정의를 속성 및 인덱서에도 확대파일 다운로드1
13753정성태10/4/20245584Linux: 81. Linux - PATH 환경변수의 적용 규칙
13752정성태10/2/20246266닷넷: 2303. C# 13 - (7) ref struct의 interface 상속 및 제네릭 제약으로 사용 가능 [6]파일 다운로드1
13751정성태10/2/20245400C/C++: 176. C/C++ - ARM64로 포팅할 때 유의할 점
13750정성태10/1/20245280C/C++: 175. C++ - WinMain/wWinMain 호출 전의 CRT 초기화 단계
13749정성태9/30/20245534닷넷: 2302. C# - ssh-keygen으로 생성한 Private Key와 Public Key 연동파일 다운로드1
13748정성태9/29/20245742닷넷: 2301. C# - BigInteger 타입이 byte 배열로 직렬화하는 방식
13747정성태9/28/20245574닷넷: 2300. C# - OpenSSH의 공개키 파일에 대한 "BEGIN OPENSSH PUBLIC KEY" / "END OPENSSH PUBLIC KEY" PEM 포맷파일 다운로드1
13746정성태9/28/20245682오류 유형: 924. Python - LocalProtocolError("Illegal header value ...")
13745정성태9/28/20245544Linux: 80. 리눅스 - 실행 중인 프로세스 내부의 환경변수 설정을 구하는 방법 (lldb)
13744정성태9/27/20245972닷넷: 2299. C# - Windows Hello 사용자 인증 다이얼로그 표시하기파일 다운로드1
13743정성태9/26/20246416닷넷: 2298. C# - Console 프로젝트에서의 await 대상으로 Main 스레드 활용하는 방법 [1]
1  2  3  4  5  6  [7]  8  9  10  11  12  13  14  15  ...