From 3344d1a728d37f0fb3e82a6a945eee0c780eb734 Mon Sep 17 00:00:00 2001 From: Chernov Dmitriy <cd_work@mail.ru> Date: 周三, 02 3月 2016 21:29:32 +0800 Subject: [PATCH] Fix potential memory leak. #73 --- libscpi/src/error.c | 87 ++++++++++++++++++++++++++++--------------- 1 files changed, 57 insertions(+), 30 deletions(-) diff --git a/libscpi/src/error.c b/libscpi/src/error.c index 9d8e756..d1307a4 100644 --- a/libscpi/src/error.c +++ b/libscpi/src/error.c @@ -45,7 +45,7 @@ * Initialize error queue * @param context - scpi context */ -void SCPI_ErrorInit(scpi_t * context, int16_t * data, int16_t size) { +void SCPI_ErrorInit(scpi_t * context, scpi_error_t * data, int16_t size) { fifo_init(&context->error_queue, data, size); } @@ -81,6 +81,12 @@ * @param context - scpi context */ void SCPI_ErrorClear(scpi_t * context) { +#if USE_DEVICE_DEPENDENT_ERROR_INFORMATION + scpi_error_t error; + while(fifo_remove(&context->error_queue, &error)){ + SCPIDEFINE_free(&context->error_info_heap, error.device_dependent_info, false); + } +#endif fifo_clear(&context->error_queue); SCPI_ErrorEmitEmpty(context); @@ -89,16 +95,18 @@ /** * Pop error from queue * @param context - scpi context - * @return error number + * @param error + * @return */ -int16_t SCPI_ErrorPop(scpi_t * context) { - int16_t result = 0; +scpi_bool_t SCPI_ErrorPop(scpi_t * context, scpi_error_t * error) { + if(!error || !context) return FALSE; + error->error_code = 0; + error->device_dependent_info = NULL; + fifo_remove(&context->error_queue, error); - fifo_remove(&context->error_queue, &result); + SCPI_ErrorEmitEmpty(context); - SCPI_ErrorEmitEmpty(context); - - return result; + return TRUE; } /** @@ -114,10 +122,10 @@ return result; } -static scpi_bool_t SCPI_ErrorAddInternal(scpi_t * context, int16_t err) { - if (!fifo_add(&context->error_queue, err)) { +static scpi_bool_t SCPI_ErrorAddInternal(scpi_t * context, int16_t err, char * info) { + if (!fifo_add(&context->error_queue, err, info)) { fifo_remove_last(&context->error_queue, NULL); - fifo_add(&context->error_queue, SCPI_ERROR_QUEUE_OVERFLOW); + fifo_add(&context->error_queue, SCPI_ERROR_QUEUE_OVERFLOW, NULL); return FALSE; } return TRUE; @@ -148,26 +156,45 @@ * @param context - scpi context * @param err - error number */ +void SCPI_ErrorPushEx(scpi_t * context, int16_t err, char * info) { + int i; + char * info_ptr = NULL; + +#if USE_DEVICE_DEPENDENT_ERROR_INFORMATION + if (info){ + info_ptr = SCPIDEFINE_strdup(&context->error_info_heap, info); + } +#endif + + scpi_bool_t queue_overflow = !SCPI_ErrorAddInternal(context, err, info_ptr); + + for (i = 0; i < ERROR_DEFS_N; i++) { + if ((err <= errs[i].from) && (err >= errs[i].to)) { + SCPI_RegSetBits(context, SCPI_REG_ESR, errs[i].bit); + } + } + + SCPI_ErrorEmit(context, err); + if (queue_overflow) { + SCPI_ErrorEmit(context, SCPI_ERROR_QUEUE_OVERFLOW); +#if USE_DEVICE_DEPENDENT_ERROR_INFORMATION + SCPIDEFINE_free(&context->error_info_heap, info_ptr, true); +#endif + } + + if (context) { + context->cmd_error = TRUE; + } +} + +/** + * Push error to queue + * @param context - scpi context + * @param err - error number + */ void SCPI_ErrorPush(scpi_t * context, int16_t err) { - - int i; - - scpi_bool_t queue_overflow = !SCPI_ErrorAddInternal(context, err); - - for (i = 0; i < ERROR_DEFS_N; i++) { - if ((err <= errs[i].from) && (err >= errs[i].to)) { - SCPI_RegSetBits(context, SCPI_REG_ESR, errs[i].bit); - } - } - - SCPI_ErrorEmit(context, err); - if (queue_overflow) { - SCPI_ErrorEmit(context, SCPI_ERROR_QUEUE_OVERFLOW); - } - - if (context) { - context->cmd_error = TRUE; - } + SCPI_ErrorPushEx(context, err, NULL); + return; } /** -- Gitblit v1.9.1