JVMPIについて

/* メソッド開始イベント */

void MethodEntryHandler(jmethodID method_id, JNIEnv *thread_id)

{

  fprintf(stderr, "aaa\n");

  int *p_ctr = (int *)CALL(GetThreadLocalStorage)(thread_id);

  if (!p_ctr) {

    p_ctr = (int *)malloc(sizeof(int));

    *p_ctr = 0;

    CALL(SetThreadLocalStorage)(thread_id, p_ctr);

  }

  (*p_ctr)++;

}

JDK_HOME = c:\progra~1\j2sdk1.4.2
LIB_DIR = $(JDK_HOME)\lib

INCLUDES = -I. -I$(JDK_HOME)\include -I$(JDK_HOME)\include\win32

SRC_DIR = .

all: myprofiler.dll

myprofiler.dll: $(SRC_DIR)\myprofiler.cpp
        cl -LDd -Zi -Zp1 $(INCLUDES) -Tp$(SRC_DIR)\myprofiler.cpp -o $@ /link /libpath:$(LIB_DIR) jvm.lib

all : a.cpp
	echo "compile"

#include <jvmpi.h>

#include <jni.h>

#include <string.h>

#include <stdlib.h>



/****************** スレッドの情報を保持する構造体 *******************/

typedef struct thread_storage

{

  jint call; /* メソッドが呼ばれた回数 */

  jlong time; /* スレッドの実行時間 */

  jlong start; /* スレッドの開始時間 */



} ThreadStorage;



/***************** jvmpiインタフェース型のポインタ *******************/

JVMPI_Interface *jvmpi_interface;

#define CALL(f) (jvmpi_interface->f)



/********************* データアクセスロック用 ************************/

JVMPI_RawMonitor data_access_lock;



/*********************** グローバル変数 ******************************/

FILE *fp;

JNIEnv *ThreadID[1000];

int thread_count = 0;



/********************** スレッド開始イベント *************************/

void notifyThreadStart(JNIEnv *env_id, jobjectID thread_id, char *thread_name)

{

  int i;

  ThreadStorage *info;

  

  /* 同じIDのスレッドの時は終了 */

  for(i=0; i<thread_count; i++) {

    if(ThreadID[i] == env_id)

      return;

  }

  

  fprintf(fp, "thread start %x(env_id), %s(thread_name)\n", env_id, thread_name);

  

  info = (ThreadStorage *)malloc(sizeof(ThreadStorage));

  info->call = 0;

  info->time = (jlong)0;

  info->start = CALL(GetCurrentThreadCpuTime)();

  

  CALL(SetThreadLocalStorage)(env_id, info);

}



/********************* メソッド開始イベント ***************************/

void notifyMethodEntry(jmethodID method_id, JNIEnv *env_id)

{

  int i;

  ThreadStorage *info;

  

  /* 同じIDのスレッドの時は終了 */

  for(i=0; i<thread_count; i++) {

    if(ThreadID[i] == env_id)

      return;

  }

  

  info = (ThreadStorage *)CALL(GetThreadLocalStorage)(env_id);

  info->call++;

  info->time = (CALL(GetCurrentThreadCpuTime)() - info->start);

  

  

}



/********************** スレッド終了イベント **************************/

void notifyThreadEnd(JNIEnv *env_id)

{

  int i;

  ThreadStorage *info;

  

  /* 同じIDのスレッドの時は終了 */

  for(i=0; i<thread_count; i++) {

    if(ThreadID[i] == env_id)

      return;

  }

  

  info= (ThreadStorage *)CALL(GetThreadLocalStorage)(env_id);

  

  fprintf(fp, "Thread %x executed %d method\n", env_id, info->call);

  fprintf(fp, "Thread %x execution time %f\n", env_id, (float)info->time/1000000000.0);

  

  /* スレッドIDの登録 */

  ThreadID[thread_count++] = env_id;

  

  free(info);

}





/********************** イベントハンドリング関数 **************************/

void notifyEvent(JVMPI_Event *event)

{

  switch(event->event_type) {

  case JVMPI_EVENT_THREAD_START:

    CALL(RawMonitorEnter)(data_access_lock);

    notifyThreadStart(event->u.thread_start.thread_env_id, event->u.thread_start.thread_id, event->u.thread_start.thread_name);

    CALL(RawMonitorExit)(data_access_lock);

    break;

    

  case JVMPI_EVENT_METHOD_ENTRY:

    CALL(RawMonitorEnter)(data_access_lock);

    notifyMethodEntry(event->u.method.method_id, event->env_id);

    CALL(RawMonitorEnter)(data_access_lock);

    break;

    

  case JVMPI_EVENT_THREAD_END:

    CALL(RawMonitorEnter)(data_access_lock);

    notifyThreadEnd(event->env_id);

    CALL(RawMonitorExit)(data_access_lock);

    break;

  }

}



extern "C"

{



/*************** プロファイラエージェントエントリーポイント *****************/

JNIEXPORT jint JNICALL JVM_OnLoad(JavaVM *jvm, char *options, void *reserved)

  {

    

    /* jvmpiのポインタの取得 */

    if ((jvm->GetEnv((void **)&jvmpi_interface, JVMPI_VERSION_1)) < 0) {

      fprintf(stderr, "error in obtaining jvmpi interface pointer\n");

      return JNI_ERR;

    }



    fp = fopen("C:\\jvmpi\\myprofiler.log", "w");

    

    /* イベントコールバック関数の登録 */

    jvmpi_interface->NotifyEvent = notifyEvent;

    

    /* イベントの設定 */

    jvmpi_interface->EnableEvent(JVMPI_EVENT_THREAD_START, NULL);

    jvmpi_interface->EnableEvent(JVMPI_EVENT_METHOD_ENTRY, NULL);

    jvmpi_interface->EnableEvent(JVMPI_EVENT_THREAD_END, NULL);

    

    return JNI_OK;

  }

}