ae63e45c65
The htab functions find_str and add_str have been renamed to include _ref in the places they previously noted _str. These are intended to manage an htab containing references to any data; string or not. The htab_find function has been divided up. htab_find_core now handles actually finding the correct node. htab_find and htab_find_ref are just outward facing functions for retrieving a specific kind of data from the node, depending on what the htab is used for.
111 lines
2.6 KiB
C
111 lines
2.6 KiB
C
#include <tvm/tvm_preprocessor.h>
|
|
#include <tvm/tvm_file.h>
|
|
|
|
#include <string.h>
|
|
|
|
int tvm_preprocess(char *src, int *src_len, tvm_htab_t *defines)
|
|
{
|
|
char* pp_directive_delimiter = NULL;
|
|
if((pp_directive_delimiter = strstr(src, "%include")))
|
|
{
|
|
char *strbegin = pp_directive_delimiter, *strend = strchr(strbegin, '\n');
|
|
|
|
if(!strend || !strbegin) return 0;
|
|
|
|
int linelen = strend - strbegin;
|
|
char* temp_str = calloc(linelen + 1, sizeof(char));
|
|
memcpy(temp_str, strbegin, linelen);
|
|
|
|
char *filename = (strchr(temp_str, ' ') + 1);
|
|
|
|
FILE* pFile = tvm_fopen(filename, ".vm", "r");
|
|
if(!pFile)
|
|
{
|
|
printf("Unable to open file \"%s\"\n", filename);
|
|
return -1;
|
|
}
|
|
|
|
free(temp_str);
|
|
|
|
size_t addition_len = tvm_flength(pFile);
|
|
char *addition_str = calloc(addition_len, sizeof(char));
|
|
tvm_fcopy(addition_str, addition_len, pFile);
|
|
fclose(pFile);
|
|
|
|
size_t first_block_len = (strbegin - src);
|
|
size_t second_block_len = ((src + *src_len) - strend);
|
|
size_t new_src_len = (first_block_len + addition_len + second_block_len);
|
|
|
|
src = (char *)realloc((char *)src, sizeof(char) * new_src_len);
|
|
src[new_src_len] = 0;
|
|
|
|
memmove(&src[first_block_len + addition_len], strend, second_block_len);
|
|
memcpy(&src[first_block_len], addition_str, addition_len);
|
|
|
|
// Fuckin' hack
|
|
for(int i = 0; i < new_src_len; i++) if(src[i] == 0) src[i] = ' ';
|
|
|
|
*src_len = strlen(src);
|
|
return 1;
|
|
}
|
|
else if((pp_directive_delimiter = strstr(src, "%define ")))
|
|
{
|
|
char *begin = pp_directive_delimiter;
|
|
char *end = strchr(begin, '\n');
|
|
|
|
if(!end) return 0;
|
|
|
|
int offset = strlen("%define ");
|
|
|
|
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(htab_find(defines, keystr) < 0)
|
|
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_length = *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_length);
|
|
*src_len = new_length;
|
|
|
|
return 1;
|
|
}
|
|
|
|
return 0;
|
|
|
|
}
|