xxq250
/
homework-jianmu
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

add option 'autoDump' for memory leak detection.

This commit is contained in:
localvar 2019-11-14 10:45:00 +00:00
parent 32abecfbc0
commit fc6737ab04
5 changed files with 20 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/client/jni/com_taosdata_jdbc_TSDBJNIConnector.h
View File

@ -15,7 +15,7 @@ extern "C" {
* Signature: (Ljava/lang/String;)V
*/
JNIEXPORT void JNICALL Java_com_taosdata_jdbc_TSDBJNIConnector_detectMemoryLeakImp
(JNIEnv *, jclass, jstring);
(JNIEnv *, jclass, jstring, jboolean);
/*
* Class: com_taosdata_jdbc_TSDBJNIConnector

6
src/client/src/TSDBJNIConnector.c
View File

@ -111,13 +111,13 @@ void jniGetGlobalMethod(JNIEnv *env) {
jniTrace("native method register finished");
}
JNIEXPORT void JNICALL Java_com_taosdata_jdbc_TSDBJNIConnector_detectMemoryLeakImp(JNIEnv *env, jobject jobj, jstring jPath) {
JNIEXPORT void JNICALL Java_com_taosdata_jdbc_TSDBJNIConnector_detectMemoryLeakImp(JNIEnv *env, jobject jobj, jstring jPath, jboolean jAutoDump) {
if (jPath != NULL) {
const char *path = (*env)->GetStringUTFChars(env, jPath, NULL);
taos_detect_memory_leak(path);
taos_detect_memory_leak(path, !!jAutoDump);
(*env)->ReleaseStringUTFChars(env, jPath, path);
} else {
taos_detect_memory_leak(NULL);
taos_detect_memory_leak(NULL, !!jAutoDump);
}
jniGetGlobalMethod(env);

2
src/inc/tutil.h
View File

@ -187,7 +187,7 @@ static FORCE_INLINE void taosEncryptPass(uint8_t *inBuf, unsigned int inLen, cha
char *taosIpStr(uint32_t ipInt);
extern void taos_detect_memory_leak(const char* path);
extern void taos_detect_memory_leak(const char* path, bool autoDump);
extern void taos_dump_memory_leak();
#if TAOS_MEM_CHECK == 1

4
src/system/detail/src/dnodeService.c
View File

@ -64,9 +64,9 @@ int main(int argc, char *argv[]) {
#if TAOS_MEM_CHECK == 2
} else if (strcmp(argv[i], "--check-mem-leak") == 0) {
if ((i < argc - 1) && (argv[i+1][0] != '-')) {
taos_detect_memory_leak(argv[++i]);
taos_detect_memory_leak(argv[++i], true);
} else {
taos_detect_memory_leak(NULL);
taos_detect_memory_leak(NULL, true);
}
#endif
}

11
src/util/src/tmem.c
View File

@ -307,13 +307,15 @@ void taos_dump_memory_leak() {
static void dump_memory_leak_at_sig(int sig) {
fprintf(fpMemLeak, "signal %d received, exiting...\n", sig);
taos_dump_memory_leak();
struct sigaction act = {0};
act.sa_handler = SIG_DFL;
sigaction(sig, &act, NULL);
taos_dump_memory_leak();
}
void taos_detect_memory_leak(const char* path) {
void taos_detect_memory_leak(const char* path, bool autoDump) {
if (fpMemLeak != NULL) {
printf("memory leak detection already enabled.\n");
return;
@ -326,14 +328,15 @@ void taos_detect_memory_leak(const char* path) {
return;
}
if (autoDump) {
atexit(taos_dump_memory_leak);
struct sigaction act = {0};
act.sa_handler = dump_memory_leak_at_sig;
sigaction(SIGFPE, &act, NULL);
sigaction(SIGSEGV, &act, NULL);
sigaction(SIGILL, &act, NULL);
}
}
#endif
@ -342,7 +345,7 @@ void taos_dump_memory_leak() {
// do nothing
}
void taos_detect_memory_leak(const char* path) {
void taos_detect_memory_leak(const char* path, bool autoDump) {
printf("memory leak detection not enabled, please set 'TAOS_MEM_CHECK' to 2.");
}
#endif