523a88ad3a
.. the previous arrangement lead to compiler diagnostics when
building with 'scan-build make', such as:
libtvm/tvm_parser.c:218:12: warning: Potential leak of memory pointed to by 'args'
return -1;
^
142 lines
3 KiB
C
142 lines
3 KiB
C
#include <tvm/tvm_preprocessor.h>
|
|
#include <tvm/tvm_file.h>
|
|
#include <tvm/tvm_tokens.h>
|
|
|
|
#include <unistd.h>
|
|
#include <string.h>
|
|
|
|
static int process_includes(
|
|
char **src, int *src_len, struct tvm_htab_ctx *defines)
|
|
{
|
|
char *pp_directive_delimiter = strstr(*src, TOK_INCLUDE);
|
|
|
|
if (!pp_directive_delimiter)
|
|
return 0;
|
|
|
|
char *begin = pp_directive_delimiter,
|
|
*end = strchr(begin, '\n');
|
|
|
|
if (!end || !begin)
|
|
return 0;
|
|
|
|
int linelen = end - begin;
|
|
char *temp_str = calloc(linelen + 1, sizeof(char));
|
|
|
|
memcpy(temp_str, begin, linelen);
|
|
|
|
char *filename = (strchr(temp_str, ' ') + 1);
|
|
FILE *filp = tvm_fopen(filename, ".vm", "r");
|
|
|
|
if (!filp) {
|
|
printf("Unable to open file \"%s\"\n", filename);
|
|
free(temp_str);
|
|
return -1;
|
|
}
|
|
|
|
free(temp_str);
|
|
|
|
size_t addition_len = tvm_flength(filp);
|
|
char *addition_str = calloc(addition_len, sizeof(char));
|
|
|
|
tvm_fcopy(addition_str, addition_len, filp);
|
|
fclose(filp);
|
|
|
|
size_t first_block_len = (begin - *src);
|
|
size_t second_block_len = ((*src + *src_len) - end);
|
|
size_t new_src_len = (
|
|
first_block_len + addition_len + second_block_len);
|
|
|
|
char *new_src = (char *)calloc(sizeof(char), new_src_len);
|
|
strncpy(new_src, (*src), first_block_len);
|
|
strcat(new_src, addition_str);
|
|
strcat(new_src, end+1);
|
|
|
|
free(*src);
|
|
*src = new_src;
|
|
*src_len = strlen(*src);
|
|
return 1;
|
|
}
|
|
|
|
static int process_defines(
|
|
char **src, int *src_len, struct tvm_htab_ctx *defines)
|
|
{
|
|
char *pp_directive_delimiter = strstr(*src, TOK_DEFINE);
|
|
|
|
if (!pp_directive_delimiter)
|
|
return 0;
|
|
|
|
char *begin = pp_directive_delimiter;
|
|
char *end = strchr(begin, '\n');
|
|
|
|
if (!end)
|
|
return 0;
|
|
|
|
int offset = strlen(TOK_DEFINE) + 1;
|
|
|
|
if (begin + offset >= end) {
|
|
printf("Define missing arguments.\n");
|
|
return -1;
|
|
}
|
|
|
|
int length = (end - (begin + offset));
|
|
char tempstr[length + 1];
|
|
|
|
memset(tempstr, 0, length + 1);
|
|
memcpy(tempstr, begin + offset, length);
|
|
|
|
char *keystr = tempstr;
|
|
char *valstr = strchr(tempstr, ' ');
|
|
|
|
/* If there is a value, seperate the key and value
|
|
* with a null character.
|
|
*/
|
|
if (valstr) {
|
|
*valstr = 0;
|
|
valstr += 1;
|
|
}
|
|
|
|
if (!keystr || !valstr) {
|
|
printf("Define missing arguments.\n");
|
|
return -1;
|
|
}
|
|
|
|
if (tvm_htab_find(defines, keystr) < 0)
|
|
tvm_htab_add_ref(defines, keystr, valstr, strlen(valstr) + 1);
|
|
else {
|
|
printf("Multiple definitions for %s.\n", keystr);
|
|
return -1;
|
|
}
|
|
|
|
/* Remove the define line so it is not processed again. */
|
|
size_t new_src_len = *src_len - (end - begin);
|
|
size_t first_block_len = begin - *src;
|
|
size_t second_block_len = (*src + *src_len) - end;
|
|
|
|
memmove(&(*src)[first_block_len], end, second_block_len);
|
|
|
|
*src = realloc(*src, sizeof(char) * new_src_len);
|
|
*src_len = new_src_len;
|
|
|
|
return 1;
|
|
}
|
|
|
|
static int preprocess_pass(
|
|
char **src, int *src_len, struct tvm_htab_ctx *defines)
|
|
{
|
|
int ret = 0;
|
|
|
|
ret += process_includes(src, src_len, defines);
|
|
ret += process_defines(src, src_len, defines);
|
|
return ret;
|
|
}
|
|
|
|
|
|
int tvm_preprocess(char **src, int *src_len, struct tvm_htab_ctx *defines)
|
|
{
|
|
int ret = 1;
|
|
|
|
while (ret > 0)
|
|
ret = preprocess_pass(src, src_len, defines);
|
|
|
|
return ret;
|
|
}
|