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/parser.c | 110 ++++++++++++++++++++++++++++++++++++++++++++++++++++--- 1 files changed, 104 insertions(+), 6 deletions(-) diff --git a/libscpi/src/parser.c b/libscpi/src/parser.c index 70187f3..00cba78 100644 --- a/libscpi/src/parser.c +++ b/libscpi/src/parser.c @@ -219,7 +219,11 @@ result &= processCommand(context); cmd_prev = state->programHeader; } else { - SCPI_ErrorPush(context, SCPI_ERROR_UNDEFINED_HEADER); + /* test */ + /* place undefined header with error */ + data[r?(r-1):r]=0; + SCPI_ErrorPushEx(context, SCPI_ERROR_UNDEFINED_HEADER, data); + //SCPI_ErrorPush(context, SCPI_ERROR_UNDEFINED_HEADER); result = FALSE; } } @@ -246,9 +250,31 @@ * @param buffer * @param interface */ -void SCPI_Init(scpi_t * context) { +void SCPI_Init(scpi_t * context, + const scpi_command_t * commands, + scpi_interface_t * interface, + const scpi_unit_def_t * units, + const char * idn1, const char * idn2, const char * idn3, const char * idn4, + char * input_buffer, size_t input_buffer_length, + scpi_error_t * error_queue_data, int16_t error_queue_size, + char * error_info_heap, size_t error_info_heap_length) { + memset(context, 0, sizeof(*context)); + context->cmdlist = commands; + context->interface = interface; + context->units = units; + context->idn[0] = idn1; + context->idn[1] = idn2; + context->idn[2] = idn3; + context->idn[3] = idn4; + context->buffer.data = input_buffer; + context->buffer.length = input_buffer_length; context->buffer.position = 0; - SCPI_ErrorInit(context); + context->error_info_heap.data = error_info_heap; + context->error_info_heap.wr = 0; + context->error_info_heap.size = error_info_heap_length; + context->error_info_heap.count = context->error_info_heap.size; + memset(context->error_info_heap.data,0,context->error_info_heap.size); + SCPI_ErrorInit(context, error_queue_data, error_queue_size); } /** @@ -482,6 +508,80 @@ result += writeData(context, "\"", 1); context->output_count++; return result; +} + + +/** + * SCPI-99:21.8 Device-dependent error information. + * Write error information with the following syntax: + * <Error/event_number>,"<Error/event_description>[;<Device-dependent_info>]" + * The maximum string length of <Error/event_description> plus <Device-dependent_info> + * is SCPI_STD_ERROR_DESC_MAX_STRING_LENGTH (255) characters. + * + * @param context + * @param error + * @return + */ +size_t SCPI_ResultError(scpi_t * context, scpi_error_t * error) { + size_t result = 0; + size_t outputlimit = SCPI_STD_ERROR_DESC_MAX_STRING_LENGTH; + size_t step = 0; + const char * quote; + + char * data[SCPIDEFINE_DESCRIPTION_MAX_PARTS]; + size_t len[SCPIDEFINE_DESCRIPTION_MAX_PARTS]; + + data[0] = SCPI_ErrorTranslate(error->error_code); + len[0] = strlen(data[0]); + +#if USE_DEVICE_DEPENDENT_ERROR_INFORMATION + data[1] = error->device_dependent_info; +#if USE_MEMORY_ALLOCATION_FREE + len[1] = error->device_dependent_info ? strlen(data[1]) : 0; +#else + SCPIDEFINE_get_parts(&context->error_info_heap, data[1], &len[1], &data[2], &len[2]); +#endif +#endif + + result += SCPI_ResultInt32(context, error->error_code); + result += writeDelimiter(context); + result += writeData(context, "\"", 1); + + for(size_t i = 0; data[i] && outputlimit && (i < SCPIDEFINE_DESCRIPTION_MAX_PARTS); i++){ + if(i==1){ + result += writeSemicolon(context); + outputlimit -= 1; + } + if(len[i] > outputlimit) { + len[i] = outputlimit; + } + + while ((quote = strnpbrk(data[i], len[i], "\""))) { + if((step = quote - data[i] + 1) >= outputlimit){ + len[i] -= 1; + outputlimit -= 1; + break; + } + result += writeData(context, data[i], step); + result += writeData(context, "\"", 1); + len[i] -= step ; + outputlimit -= step + 1; + data[i] = quote + 1; + if(len[i] > outputlimit) { + len[i] = outputlimit; + } + } + + result += writeData(context, data[i], len[i]); + outputlimit-=len[i]; + } + result += writeData(context, "\"", 1); + + #if USE_DEVICE_DEPENDENT_ERROR_INFORMATION + SCPIDEFINE_free(&context->error_info_heap, error->device_dependent_info, false); + #endif + + return result; } /** @@ -1455,6 +1555,7 @@ case 4: case 8: result += SCPI_ResultArbitraryBlockHeader(context, count * item_size); + break; default: SCPI_ErrorPush(context, SCPI_ERROR_SYSTEM_ERROR); return 0; @@ -1495,9 +1596,6 @@ size_t i;\ for (i = 0; i < count; i++) {\ result += func(context, array[i]);\ - }\ - if (count > 0) {\ - result += count - 1; /* add length of commas */\ }\ } else {\ result = parserResultArrayBinary(context, array, count, sizeof(*array), format);\ -- Gitblit v1.9.1