Fix potential memory leak. #73

Chernov Dmitriy

2016-03-02 3344d1a728d37f0fb3e82a6a945eee0c780eb734

 libscpi/src/parser.c

@@ -53,7 +53,11 @@
 * @return number of bytes written
 */
static size_t writeData(scpi_t * context, const char * data, size_t len) {
    return context->interface->write(context, data, len);
    if (len > 0) {
        return context->interface->write(context, data, len);
    } else {
        return 0;
    }
}

/**
@@ -215,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;
            }
        }
@@ -242,22 +250,31 @@
 * @param buffer
 * @param interface
 */
void SCPI_Init(scpi_t * context) {
    if (context->idn[0] == NULL) {
        context->idn[0] = SCPI_DEFAULT_1_MANUFACTURE;
    }
    if (context->idn[1] == NULL) {
        context->idn[1] = SCPI_DEFAULT_2_MODEL;
    }
    if (context->idn[2] == NULL) {
        context->idn[2] = SCPI_DEFAULT_3;
    }
    if (context->idn[3] == NULL) {
        context->idn[3] = SCPI_DEFAULT_4_REVISION;
    }

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);
}

/**
@@ -477,13 +494,94 @@
 */
size_t SCPI_ResultText(scpi_t * context, const char * data) {
    size_t result = 0;
    size_t len = strlen(data);
    const char * quote;
    result += writeDelimiter(context);
    result += writeData(context, "\"", 1);
    // TODO: convert " to ""
    result += writeData(context, data, strlen(data));
    while ((quote = strnpbrk(data, len, "\""))) {
        result += writeData(context, data, quote - data + 1);
        result += writeData(context, "\"", 1);
        len -= quote - data + 1;
        data = quote + 1;
    }
    result += writeData(context, data, len);
    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;
}

/**
@@ -1433,7 +1531,7 @@
 * @param count
 * @param item_size
 * @param format
 * @return 
 * @return
 */
static size_t parserResultArrayBinary(scpi_t * context, const void * array, size_t count, size_t item_size, scpi_array_format_t format) {

@@ -1457,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;
@@ -1468,19 +1567,19 @@
                break;
            case 2:
                for (i = 0; i < count; i++) {
                    uint16_t val = SCPI_Swap16(((uint16_t*)array)[i]);
                    uint16_t val = SCPI_Swap16(((uint16_t*) array)[i]);
                    result += SCPI_ResultArbitraryBlockData(context, &val, item_size);
                }
                break;
            case 4:
                for (i = 0; i < count; i++) {
                    uint32_t val = SCPI_Swap32(((uint32_t*)array)[i]);
                    uint32_t val = SCPI_Swap32(((uint32_t*) array)[i]);
                    result += SCPI_ResultArbitraryBlockData(context, &val, item_size);
                }
                break;
            case 8:
                for (i = 0; i < count; i++) {
                    uint64_t val = SCPI_Swap64(((uint64_t*)array)[i]);
                    uint64_t val = SCPI_Swap64(((uint64_t*) array)[i]);
                    result += SCPI_ResultArbitraryBlockData(context, &val, item_size);
                }
                break;
@@ -1497,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);\
@@ -1537,7 +1633,7 @@
 * @param array
 * @param count
 * @param format
 * @return 
 * @return
 */
size_t SCPI_ResultArrayInt16(scpi_t * context, const int16_t * array, size_t count, scpi_array_format_t format) {
    RESULT_ARRAY(SCPI_ResultInt16);
@@ -1549,7 +1645,7 @@
 * @param array
 * @param count
 * @param format
 * @return 
 * @return
 */
size_t SCPI_ResultArrayUInt16(scpi_t * context, const uint16_t * array, size_t count, scpi_array_format_t format) {
    RESULT_ARRAY(SCPI_ResultUInt16);
@@ -1561,7 +1657,7 @@
 * @param array
 * @param count
 * @param format
 * @return 
 * @return
 */
size_t SCPI_ResultArrayInt32(scpi_t * context, const int32_t * array, size_t count, scpi_array_format_t format) {
    RESULT_ARRAY(SCPI_ResultInt32);
@@ -1573,7 +1669,7 @@
 * @param array
 * @param count
 * @param format
 * @return 
 * @return
 */
size_t SCPI_ResultArrayUInt32(scpi_t * context, const uint32_t * array, size_t count, scpi_array_format_t format) {
    RESULT_ARRAY(SCPI_ResultUInt32);
@@ -1585,7 +1681,7 @@
 * @param array
 * @param count
 * @param format
 * @return 
 * @return
 */
size_t SCPI_ResultArrayInt64(scpi_t * context, const int64_t * array, size_t count, scpi_array_format_t format) {
    RESULT_ARRAY(SCPI_ResultInt64);
@@ -1597,7 +1693,7 @@
 * @param array
 * @param count
 * @param format
 * @return 
 * @return
 */
size_t SCPI_ResultArrayUInt64(scpi_t * context, const uint64_t * array, size_t count, scpi_array_format_t format) {
    RESULT_ARRAY(SCPI_ResultUInt64);
@@ -1609,7 +1705,7 @@
 * @param array
 * @param count
 * @param format
 * @return 
 * @return
 */
size_t SCPI_ResultArrayFloat(scpi_t * context, const float * array, size_t count, scpi_array_format_t format) {
    RESULT_ARRAY(SCPI_ResultFloat);
@@ -1621,7 +1717,7 @@
 * @param array
 * @param count
 * @param format
 * @return 
 * @return
 */
size_t SCPI_ResultArrayDouble(scpi_t * context, const double * array, size_t count, scpi_array_format_t format) {
    RESULT_ARRAY(SCPI_ResultDouble);