From 5c912c70e9333298ff48f7ea15424f72ca977b99 Mon Sep 17 00:00:00 2001 From: Fu Juntang <StrongTiger_001@163.com> Date: 星期五, 17 九月 2021 09:43:55 +0800 Subject: [PATCH] Add the heartbeat logic feature. --- src/bus_error.cpp | 46 ++++++++++++++++++++++++++++++++++++++++------ 1 files changed, 40 insertions(+), 6 deletions(-) diff --git a/src/bus_error.cpp b/src/bus_error.cpp index 29d5683..3836ebf 100644 --- a/src/bus_error.cpp +++ b/src/bus_error.cpp @@ -49,15 +49,11 @@ } char * -bus_strerror(int err) +bus_strerror(int err, int flag) { int s; char *buf; /* Make first caller allocate key for thread-specific data */ - - if (err == 0) { - err = EBUS_BASE; - } s = pthread_once(&once, createKey); if (s != 0) @@ -68,15 +64,23 @@ { /* If first call from this thread, allocate buffer for thread, and save its location */ - buf = (char *)malloc(MAX_ERROR_LEN); + buf = (char *)malloc(MAX_ERROR_LEN + sizeof(int)); if (buf == NULL) err_exit(errno, "malloc"); + memset(buf, 0x00, MAX_ERROR_LEN + sizeof(int)); s = pthread_setspecific(strerrorKey, buf); if (s != 0) err_exit(s, "pthread_setspecific"); } + if (flag != 0) { + err = *(int *)(buf + MAX_ERROR_LEN); + } + + if (err == 0) { + err = EBUS_BASE; + } if(err < EBUS_BASE) { // libc閿欒 @@ -106,3 +110,33 @@ return buf; } + +void bus_errorset(int err) +{ + int s; + char *buf; + /* Make first caller allocate key for thread-specific data */ + s = pthread_once(&once, createKey); + if (s != 0) + err_exit(s, "pthread_once"); + + buf = (char *)pthread_getspecific(strerrorKey); + if (buf == NULL) + { + /* If first call from this thread, allocate + buffer for thread, and save its location */ + buf = (char *)malloc(MAX_ERROR_LEN + sizeof(int)); + if (buf == NULL) + err_exit(errno, "malloc"); + + s = pthread_setspecific(strerrorKey, buf); + if (s != 0) + err_exit(s, "pthread_setspecific"); + } + + *(int *)(buf + MAX_ERROR_LEN) = err; + +} + + + -- Gitblit v1.8.0