| //------------------------------------------------------------------------------------------------------- |
| // Copyright (C) Microsoft Corporation and contributors. All rights reserved. |
| // Licensed under the MIT license. See LICENSE.txt file in the project root for full license information. |
| //------------------------------------------------------------------------------------------------------- |
| #include "RuntimeBasePch.h" |
| |
| #ifdef VTUNE_PROFILING |
| |
| #include "ittnotify_config.h" |
| |
| #if ITT_PLATFORM==ITT_PLATFORM_WIN |
| #include <windows.h> |
| #endif /* ITT_PLATFORM==ITT_PLATFORM_WIN */ |
| #if ITT_PLATFORM != ITT_PLATFORM_MAC && ITT_PLATFORM != ITT_PLATFORM_FREEBSD |
| #include <malloc.h> |
| #endif |
| #include <stdlib.h> |
| |
| #include "jitprofiling.h" |
| |
| static const char rcsid[] = "\n@(#) $Revision: 463960 $\n"; |
| |
| #define DLL_ENVIRONMENT_VAR "VS_PROFILER" |
| |
| #ifndef NEW_DLL_ENVIRONMENT_VAR |
| #if ITT_ARCH==ITT_ARCH_IA32 |
| #define NEW_DLL_ENVIRONMENT_VAR "INTEL_JIT_PROFILER32" |
| #else |
| #define NEW_DLL_ENVIRONMENT_VAR "INTEL_JIT_PROFILER64" |
| #endif |
| #endif /* NEW_DLL_ENVIRONMENT_VAR */ |
| |
| #if ITT_PLATFORM==ITT_PLATFORM_WIN |
| #define DEFAULT_DLLNAME "JitPI.dll" |
| HINSTANCE m_libHandle = NULL; |
| #elif ITT_PLATFORM==ITT_PLATFORM_MAC |
| #define DEFAULT_DLLNAME "libJitPI.dylib" |
| void* m_libHandle = NULL; |
| #else |
| #define DEFAULT_DLLNAME "libJitPI.so" |
| void* m_libHandle = NULL; |
| #endif /* ITT_PLATFORM==ITT_PLATFORM_WIN */ |
| |
| /* default location of JIT profiling agent on Android */ |
| #define ANDROID_JIT_AGENT_PATH "/data/intel/libittnotify.so" |
| |
| /* the function pointers */ |
| typedef unsigned int(JITAPI *TPInitialize)(void); |
| static TPInitialize FUNC_Initialize=NULL; |
| |
| typedef unsigned int(JITAPI *TPNotify)(unsigned int, void*); |
| static TPNotify FUNC_NotifyEvent=NULL; |
| |
| static iJIT_IsProfilingActiveFlags executionMode = iJIT_NOTHING_RUNNING; |
| |
| /* end collector dll part. */ |
| |
| /* loadiJIT_Funcs() : this function is called just in the beginning |
| * and is responsible to load the functions from BistroJavaCollector.dll |
| * result: |
| * on success: the functions loads, iJIT_DLL_is_missing=0, return value = 1 |
| * on failure: the functions are NULL, iJIT_DLL_is_missing=1, return value = 0 |
| */ |
| static int loadiJIT_Funcs(void); |
| |
| /* global representing whether the collector can't be loaded */ |
| static int iJIT_DLL_is_missing = 0; |
| |
| ITT_EXTERN_C int JITAPI |
| iJIT_NotifyEvent(iJIT_JVM_EVENT event_type, void *EventSpecificData) |
| { |
| int ReturnValue = 0; |
| |
| /* initialization part - the collector has not been loaded yet. */ |
| if (!FUNC_NotifyEvent) |
| { |
| if (iJIT_DLL_is_missing) |
| return 0; |
| |
| if (!loadiJIT_Funcs()) |
| return 0; |
| } |
| |
| if (event_type == iJVM_EVENT_TYPE_METHOD_LOAD_FINISHED || |
| event_type == iJVM_EVENT_TYPE_METHOD_UPDATE) |
| { |
| if (((piJIT_Method_Load)EventSpecificData)->method_id == 0) |
| return 0; |
| } |
| else if (event_type == iJVM_EVENT_TYPE_METHOD_LOAD_FINISHED_V2) |
| { |
| if (((piJIT_Method_Load_V2)EventSpecificData)->method_id == 0) |
| return 0; |
| } |
| else if (event_type == iJVM_EVENT_TYPE_METHOD_LOAD_FINISHED_V3) |
| { |
| if (((piJIT_Method_Load_V3)EventSpecificData)->method_id == 0) |
| return 0; |
| } |
| else if (event_type == iJVM_EVENT_TYPE_METHOD_INLINE_LOAD_FINISHED) |
| { |
| if (((piJIT_Method_Inline_Load)EventSpecificData)->method_id == 0 || |
| ((piJIT_Method_Inline_Load)EventSpecificData)->parent_method_id == 0) |
| return 0; |
| } |
| |
| if (FUNC_NotifyEvent) |
| { |
| ReturnValue = (int)FUNC_NotifyEvent(event_type, EventSpecificData); |
| } |
| |
| return ReturnValue; |
| } |
| |
| ITT_EXTERN_C iJIT_IsProfilingActiveFlags JITAPI iJIT_IsProfilingActive() |
| { |
| if (!iJIT_DLL_is_missing) |
| { |
| loadiJIT_Funcs(); |
| } |
| |
| return executionMode; |
| } |
| |
| /* This function loads the collector dll and the relevant functions. |
| * on success: all functions load, iJIT_DLL_is_missing = 0, return value = 1 |
| * on failure: all functions are NULL, iJIT_DLL_is_missing = 1, return value = 0 |
| */ |
| #pragma prefast(push) |
| #pragma prefast(disable:38020, "Not used in release build. VTune is sensitive to changes in this file.") |
| static int loadiJIT_Funcs() |
| { |
| static int bDllWasLoaded = 0; |
| char *dllName = (char*)rcsid; /* !! Just to avoid unused code elimination */ |
| #if ITT_PLATFORM==ITT_PLATFORM_WIN |
| DWORD dNameLength = 0; |
| #endif /* ITT_PLATFORM==ITT_PLATFORM_WIN */ |
| |
| if(bDllWasLoaded) |
| { |
| /* dll was already loaded, no need to do it for the second time */ |
| return 1; |
| } |
| |
| /* Assumes that the DLL will not be found */ |
| iJIT_DLL_is_missing = 1; |
| FUNC_NotifyEvent = NULL; |
| |
| if (m_libHandle) |
| { |
| #if ITT_PLATFORM==ITT_PLATFORM_WIN |
| FreeLibrary(m_libHandle); |
| #else /* ITT_PLATFORM==ITT_PLATFORM_WIN */ |
| dlclose(m_libHandle); |
| #endif /* ITT_PLATFORM==ITT_PLATFORM_WIN */ |
| m_libHandle = NULL; |
| } |
| |
| /* Try to get the dll name from the environment */ |
| #if ITT_PLATFORM==ITT_PLATFORM_WIN |
| dNameLength = GetEnvironmentVariableA(NEW_DLL_ENVIRONMENT_VAR, NULL, 0); |
| if (dNameLength) |
| { |
| DWORD envret = 0; |
| dllName = (char*)malloc(sizeof(char) * (dNameLength + 1)); |
| if(dllName != NULL) |
| { |
| envret = GetEnvironmentVariableA(NEW_DLL_ENVIRONMENT_VAR, |
| dllName, dNameLength); |
| if (envret) |
| { |
| /* Try to load the dll from the PATH... */ |
| m_libHandle = LoadLibraryExA(dllName, |
| NULL, LOAD_WITH_ALTERED_SEARCH_PATH); |
| } |
| free(dllName); |
| } |
| } else { |
| /* Try to use old VS_PROFILER variable */ |
| dNameLength = GetEnvironmentVariableA(DLL_ENVIRONMENT_VAR, NULL, 0); |
| if (dNameLength) |
| { |
| DWORD envret = 0; |
| dllName = (char*)malloc(sizeof(char) * (dNameLength + 1)); |
| if(dllName != NULL) |
| { |
| envret = GetEnvironmentVariableA(DLL_ENVIRONMENT_VAR, |
| dllName, dNameLength); |
| if (envret) |
| { |
| /* Try to load the dll from the PATH... */ |
| m_libHandle = LoadLibraryA(dllName); |
| } |
| free(dllName); |
| } |
| } |
| } |
| #else /* ITT_PLATFORM==ITT_PLATFORM_WIN */ |
| dllName = getenv(NEW_DLL_ENVIRONMENT_VAR); |
| if (!dllName) |
| dllName = getenv(DLL_ENVIRONMENT_VAR); |
| #if defined(__ANDROID__) || defined(ANDROID) |
| if (!dllName) |
| dllName = ANDROID_JIT_AGENT_PATH; |
| #endif |
| if (dllName) |
| { |
| /* Try to load the dll from the PATH... */ |
| m_libHandle = dlopen(dllName, RTLD_LAZY); |
| } |
| #endif /* ITT_PLATFORM==ITT_PLATFORM_WIN */ |
| |
| if (!m_libHandle) |
| { |
| #if ITT_PLATFORM==ITT_PLATFORM_WIN |
| m_libHandle = LoadLibraryA(DEFAULT_DLLNAME); |
| #else /* ITT_PLATFORM==ITT_PLATFORM_WIN */ |
| m_libHandle = dlopen(DEFAULT_DLLNAME, RTLD_LAZY); |
| #endif /* ITT_PLATFORM==ITT_PLATFORM_WIN */ |
| } |
| |
| /* if the dll wasn't loaded - exit. */ |
| if (!m_libHandle) |
| { |
| iJIT_DLL_is_missing = 1; /* don't try to initialize |
| * JIT agent the second time |
| */ |
| return 0; |
| } |
| |
| #if ITT_PLATFORM==ITT_PLATFORM_WIN |
| FUNC_NotifyEvent = (TPNotify)GetProcAddress(m_libHandle, "NotifyEvent"); |
| #else /* ITT_PLATFORM==ITT_PLATFORM_WIN */ |
| FUNC_NotifyEvent = (TPNotify)dlsym(m_libHandle, "NotifyEvent"); |
| #endif /* ITT_PLATFORM==ITT_PLATFORM_WIN */ |
| if (!FUNC_NotifyEvent) |
| { |
| FUNC_Initialize = NULL; |
| return 0; |
| } |
| |
| #if ITT_PLATFORM==ITT_PLATFORM_WIN |
| FUNC_Initialize = (TPInitialize)GetProcAddress(m_libHandle, "Initialize"); |
| #else /* ITT_PLATFORM==ITT_PLATFORM_WIN */ |
| FUNC_Initialize = (TPInitialize)dlsym(m_libHandle, "Initialize"); |
| #endif /* ITT_PLATFORM==ITT_PLATFORM_WIN */ |
| if (!FUNC_Initialize) |
| { |
| FUNC_NotifyEvent = NULL; |
| return 0; |
| } |
| |
| executionMode = (iJIT_IsProfilingActiveFlags)FUNC_Initialize(); |
| |
| bDllWasLoaded = 1; |
| iJIT_DLL_is_missing = 0; /* DLL is ok. */ |
| |
| return 1; |
| } |
| |
| ITT_EXTERN_C unsigned int JITAPI iJIT_GetNewMethodID() |
| { |
| static unsigned int methodID = 1; |
| |
| if (methodID == 0) |
| return 0; /* ERROR : this is not a valid value */ |
| |
| return methodID++; |
| } |
| #pragma prefast(pop) |
| #endif /* VTUNE_PROFILING */ |
