Microsoft MVP성태의 닷넷 이야기
.NET Framework: 576. 기본적인 CLR Profiler 소스 코드 설명 [링크 복사], [링크+제목 복사],
조회: 26191
글쓴 사람
정성태 (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)
13793정성태10/28/20245135C/C++: 183. C++ - 윈도우에서 한글(및 유니코드)을 포함한 콘솔 프로그램을 컴파일 및 실행하는 방법
13792정성태10/27/20244633Linux: 99. Linux - 프로세스의 실행 파일 경로 확인
13791정성태10/27/20244901Windows: 267. Win32 API의 A(ANSI) 버전은 DBCS를 사용할까요?파일 다운로드1
13790정성태10/27/20244607Linux: 98. Ubuntu 22.04 - 리눅스 커널 빌드 및 업그레이드
13789정성태10/27/20244909Linux: 97. menuconfig에 CONFIG_DEBUG_INFO_BTF, CONFIG_DEBUG_INFO_BTF_MODULES 옵션이 없는 경우
13788정성태10/26/20244458Linux: 96. eBPF (bpf2go) - fentry, fexit를 이용한 트레이스
13787정성태10/26/20244951개발 환경 구성: 730. github - Linux 커널 repo를 윈도우 환경에서 git clone하는 방법 [1]
13786정성태10/26/20245203Windows: 266. Windows - 대소문자 구분이 가능한 파일 시스템
13785정성태10/23/20244983C/C++: 182. 윈도우가 운영하는 2개의 Code Page파일 다운로드1
13784정성태10/23/20245247Linux: 95. eBPF - kprobe를 이용한 트레이스
13783정성태10/23/20244859Linux: 94. eBPF - vmlinux.h 헤더 포함하는 방법 (bpf2go에서 사용)
13782정성태10/23/20244605Linux: 93. Ubuntu 22.04 - 커널 이미지로부터 커널 함수 역어셈블
13781정성태10/22/20244785오류 유형: 930. WSL + eBPF: modprobe: FATAL: Module kheaders not found in directory
13780정성태10/22/20245547Linux: 92. WSL 2 - 커널 이미지로부터 커널 함수 역어셈블
13779정성태10/22/20244840개발 환경 구성: 729. WSL 2 - Mariner VM 커널 이미지 업데이트 방법
13778정성태10/21/20245669C/C++: 181. C/C++ - 소스코드 파일의 인코딩, 바이너리 모듈 상태의 인코딩
13777정성태10/20/20244952Windows: 265. Win32 API의 W(유니코드) 버전은 UCS-2일까요? UTF-16 인코딩일까요?
13776정성태10/19/20245250C/C++: 180. C++ - 고수준 FILE I/O 함수에서의 Unicode stream 모드(_O_WTEXT, _O_U16TEXT, _O_U8TEXT)파일 다운로드1
13775정성태10/19/20245478개발 환경 구성: 728. 윈도우 환경의 개발자를 위한 UTF-8 환경 설정
13774정성태10/18/20245182Linux: 91. Container 환경에서 출력하는 eBPF bpf_get_current_pid_tgid의 pid가 존재하지 않는 이유
13773정성태10/18/20244871Linux: 90. pid 네임스페이스 구성으로 본 WSL 2 + docker-desktop
13772정성태10/17/20245145Linux: 89. pid 네임스페이스 구성으로 본 WSL 2 배포본의 계층 관계
13771정성태10/17/20245049Linux: 88. WSL 2 리눅스 배포본 내에서의 pid 네임스페이스 구성
13770정성태10/17/20245328Linux: 87. ps + grep 조합에서 grep 명령어를 사용한 프로세스를 출력에서 제거하는 방법
13769정성태10/15/20246114Linux: 86. Golang + bpf2go를 사용한 eBPF 기본 예제파일 다운로드1
13768정성태10/15/20245382C/C++: 179. C++ - _O_WTEXT, _O_U16TEXT, _O_U8TEXT의 Unicode stream 모드파일 다운로드2
1  2  3  4  5  [6]  7  8  9  10  11  12  13  14  15  ...