1097 lines
30 KiB
C++
1097 lines
30 KiB
C++
//---------------------------------------------------------------------------
|
|
#include "tokenize.h"
|
|
#include "CommonCheck.h" // <- IsName
|
|
//---------------------------------------------------------------------------
|
|
|
|
#include <locale>
|
|
#include <fstream>
|
|
|
|
#include <map>
|
|
#include <string>
|
|
|
|
#include <stdlib.h> // <- strtoul
|
|
#include <stdio.h>
|
|
|
|
#ifdef __BORLANDC__
|
|
#include <mem.h>
|
|
#endif
|
|
|
|
//---------------------------------------------------------------------------
|
|
|
|
// Helper functions..
|
|
|
|
static void Define(const char Name[], const char Value[]);
|
|
|
|
static void addtoken(const char str[], const unsigned int lineno, const unsigned int fileno);
|
|
|
|
static void combine_2tokens(TOKEN *tok, const char str1[], const char str2[]);
|
|
|
|
static void DeleteNextToken(TOKEN *tok);
|
|
|
|
static TOKEN *_gettok(TOKEN *tok, int index)
|
|
{
|
|
while (tok && index>0)
|
|
{
|
|
tok = tok->next;
|
|
index--;
|
|
}
|
|
return tok;
|
|
}
|
|
//---------------------------------------------------------------------------
|
|
|
|
|
|
//---------------------------------------------------------------------------
|
|
|
|
std::vector<std::string> Files;
|
|
struct TOKEN *tokens, *tokens_back;
|
|
|
|
//---------------------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
//---------------------------------------------------------------------------
|
|
// Defined symbols.
|
|
// "#define abc 123" will create a defined symbol "abc" with the value 123
|
|
//---------------------------------------------------------------------------
|
|
|
|
struct DefineSymbol
|
|
{
|
|
char *name;
|
|
char *value;
|
|
struct DefineSymbol *next;
|
|
};
|
|
static struct DefineSymbol * dsymlist;
|
|
|
|
static void Define(const char Name[], const char Value[])
|
|
{
|
|
if (!(Name && Name[0]))
|
|
return;
|
|
|
|
if (!(Value && Value[0]))
|
|
return;
|
|
|
|
// Is 'Value' a decimal value..
|
|
bool dec = true, hex = true;
|
|
for (int i = 0; Value[i]; i++)
|
|
{
|
|
if ( ! std::isdigit(Value[i]) )
|
|
dec = false;
|
|
|
|
if ( ! std::isxdigit(Value[i]) && (!(i==1 && Value[i]=='x')))
|
|
hex = false;
|
|
}
|
|
|
|
if (!dec && !hex)
|
|
return;
|
|
|
|
char *strValue = strdup(Value);
|
|
|
|
if (!dec && hex)
|
|
{
|
|
char str[50];
|
|
unsigned long value = strtoul(Value+2, NULL, 16);
|
|
free(strValue);
|
|
sprintf(str, "%lu", value);
|
|
strValue = strdup(str);
|
|
}
|
|
|
|
DefineSymbol *NewSym = new DefineSymbol;
|
|
memset(NewSym, 0, sizeof(DefineSymbol));
|
|
NewSym->name = strdup(Name);
|
|
NewSym->value = strValue;
|
|
NewSym->next = dsymlist;
|
|
dsymlist = NewSym;
|
|
}
|
|
//---------------------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
//---------------------------------------------------------------------------
|
|
// addtoken
|
|
// add a token. Used by 'Tokenizer'
|
|
//---------------------------------------------------------------------------
|
|
|
|
static void addtoken(const char str[], const unsigned int lineno, const unsigned int fileno)
|
|
{
|
|
if (str[0] == 0)
|
|
return;
|
|
|
|
// Replace hexadecimal value with decimal
|
|
char str2[50];
|
|
memset(str2, 0, sizeof(str2));
|
|
if (strncmp(str,"0x",2)==0)
|
|
{
|
|
unsigned int value = strtoul(str+2, NULL, 16);
|
|
sprintf( str2, "%d", value );
|
|
}
|
|
|
|
TOKEN *newtoken = new TOKEN;
|
|
memset(newtoken, 0, sizeof(TOKEN));
|
|
newtoken->str = strdup(str2[0] ? str2 : str);
|
|
newtoken->linenr = lineno;
|
|
newtoken->FileIndex = fileno;
|
|
if (tokens_back)
|
|
{
|
|
tokens_back->next = newtoken;
|
|
tokens_back = newtoken;
|
|
}
|
|
else
|
|
{
|
|
tokens = tokens_back = newtoken;
|
|
}
|
|
|
|
// Check if str is defined..
|
|
for (DefineSymbol *sym = dsymlist; sym; sym = sym->next)
|
|
{
|
|
if (strcmp(str,sym->name)==0)
|
|
{
|
|
free(newtoken->str);
|
|
newtoken->str = strdup(sym->value);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
//---------------------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
//---------------------------------------------------------------------------
|
|
// combine_2tokens
|
|
// Combine two tokens that belong to each other. Ex: "<" and "=" may become "<="
|
|
//---------------------------------------------------------------------------
|
|
|
|
static void combine_2tokens(TOKEN *tok, const char str1[], const char str2[])
|
|
{
|
|
if (!(tok && tok->next))
|
|
return;
|
|
if (strcmp(tok->str,str1) || strcmp(tok->next->str,str2))
|
|
return;
|
|
|
|
free(tok->str);
|
|
tok->str = (char *)malloc(strlen(str1)+strlen(str2)+1);
|
|
strcpy(tok->str, str1);
|
|
strcat(tok->str, str2);
|
|
|
|
DeleteNextToken(tok);
|
|
}
|
|
//---------------------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
//---------------------------------------------------------------------------
|
|
// SizeOfType - gives the size of a type
|
|
//---------------------------------------------------------------------------
|
|
|
|
std::map<std::string, unsigned int> TypeSize;
|
|
|
|
int SizeOfType(const char type[])
|
|
{
|
|
if (!type)
|
|
return 0;
|
|
|
|
return TypeSize[type];
|
|
}
|
|
//---------------------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
//---------------------------------------------------------------------------
|
|
// DeleteNextToken. Unlink and delete next token.
|
|
//---------------------------------------------------------------------------
|
|
|
|
static void DeleteNextToken(TOKEN *tok)
|
|
{
|
|
TOKEN *next = tok->next;
|
|
tok->next = next->next;
|
|
free(next->str);
|
|
delete next;
|
|
}
|
|
//---------------------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
|
|
//---------------------------------------------------------------------------
|
|
// InsertTokens - Copy and insert tokens
|
|
//---------------------------------------------------------------------------
|
|
|
|
static void InsertTokens(TOKEN *dest, TOKEN *src, unsigned int n)
|
|
{
|
|
while (n > 0)
|
|
{
|
|
TOKEN *NewToken = new TOKEN;
|
|
NewToken->FileIndex = src->FileIndex;
|
|
NewToken->linenr = src->linenr;
|
|
NewToken->str = strdup(src->str);
|
|
|
|
NewToken->next = dest->next;
|
|
dest->next = NewToken;
|
|
|
|
dest = dest->next;
|
|
src = src->next;
|
|
n--;
|
|
}
|
|
}
|
|
//---------------------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
//---------------------------------------------------------------------------
|
|
// Tokenize - tokenizes a given file.
|
|
//---------------------------------------------------------------------------
|
|
|
|
void Tokenize(const char FileName[])
|
|
{
|
|
// Has this file been tokenized already?
|
|
for (unsigned int i = 0; i < Files.size(); i++)
|
|
{
|
|
if ( SameFileName( Files[i].c_str(), FileName ) )
|
|
return;
|
|
}
|
|
|
|
// Open file..
|
|
std::ifstream fin(FileName);
|
|
if (!fin.is_open())
|
|
return;
|
|
|
|
// The "Files" vector remembers what files have been tokenized..
|
|
Files.push_back(FileName);
|
|
|
|
// Tokenize the file..
|
|
TokenizeCode( fin, Files.size() - 1 );
|
|
}
|
|
//---------------------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
//---------------------------------------------------------------------------
|
|
// Tokenize - tokenizes input stream
|
|
//---------------------------------------------------------------------------
|
|
|
|
void TokenizeCode(std::istream &code, const unsigned int FileIndex)
|
|
{
|
|
// Tokenize the file.
|
|
unsigned int lineno = 1;
|
|
char CurrentToken[1000] = {0};
|
|
char *pToken = CurrentToken;
|
|
for (char ch = (char)code.get(); !code.eof(); ch = (char)code.get())
|
|
{
|
|
// Preprocessor stuff?
|
|
if (ch == '#' && !CurrentToken[0])
|
|
{
|
|
std::string line;
|
|
getline(code,line);
|
|
line = "#" + line;
|
|
if (strncmp(line.c_str(),"#include",8)==0 &&
|
|
line.find("\"") != std::string::npos)
|
|
{
|
|
// Extract the filename
|
|
line.erase(0, line.find("\"")+1);
|
|
line.erase(line.find("\""));
|
|
|
|
// Relative path..
|
|
if (Files.back().find_first_of("\\/") != std::string::npos)
|
|
{
|
|
std::string path = Files.back();
|
|
path.erase( 1 + path.find_last_of("\\/") );
|
|
line = path + line;
|
|
}
|
|
|
|
addtoken("#include", lineno, FileIndex);
|
|
addtoken(line.c_str(), lineno, FileIndex);
|
|
|
|
Tokenize(line.c_str());
|
|
}
|
|
|
|
else if (strncmp(line.c_str(), "#define", 7) == 0)
|
|
{
|
|
char *strId = NULL;
|
|
enum {Space1, Id, Space2, Value} State;
|
|
State = Space1;
|
|
for (unsigned int i = 8; i < line.length(); i++)
|
|
{
|
|
if (State==Space1 || State==Space2)
|
|
{
|
|
if (std::isspace(line[i]))
|
|
continue;
|
|
State = (State==Space1) ? Id : Value;
|
|
}
|
|
|
|
else if (State==Id)
|
|
{
|
|
if ( std::isspace( line[i] ) )
|
|
{
|
|
strId = strdup(CurrentToken);
|
|
memset(CurrentToken, 0, sizeof(CurrentToken));
|
|
pToken = CurrentToken;
|
|
State = Space2;
|
|
continue;
|
|
}
|
|
else if ( ! std::isalnum(line[i]) )
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
|
|
*pToken = line[i];
|
|
pToken++;
|
|
}
|
|
|
|
if (State==Value)
|
|
{
|
|
addtoken("def", lineno, FileIndex);
|
|
addtoken(strId, lineno, FileIndex);
|
|
addtoken(";", lineno, FileIndex);
|
|
Define(strId, CurrentToken);
|
|
}
|
|
|
|
pToken = CurrentToken;
|
|
memset(CurrentToken, 0, sizeof(CurrentToken));
|
|
free(strId);
|
|
}
|
|
|
|
else
|
|
{
|
|
addtoken("#", lineno, FileIndex);
|
|
addtoken(";", lineno, FileIndex);
|
|
}
|
|
|
|
lineno++;
|
|
continue;
|
|
}
|
|
|
|
if (ch == '\n')
|
|
{
|
|
// Add current token..
|
|
addtoken(CurrentToken, lineno++, FileIndex);
|
|
memset(CurrentToken, 0, sizeof(CurrentToken));
|
|
pToken = CurrentToken;
|
|
continue;
|
|
}
|
|
|
|
// Comments..
|
|
if (ch == '/' && !code.eof())
|
|
{
|
|
// Add current token..
|
|
addtoken(CurrentToken, lineno, FileIndex);
|
|
memset(CurrentToken, 0, sizeof(CurrentToken));
|
|
pToken = CurrentToken;
|
|
|
|
// Read next character..
|
|
ch = (char)code.get();
|
|
|
|
// If '//'..
|
|
if (ch == '/')
|
|
{
|
|
while (!code.eof() && (char)code.get()!='\n');
|
|
lineno++;
|
|
continue;
|
|
}
|
|
|
|
// If '/*'..
|
|
if (ch == '*')
|
|
{
|
|
char chPrev;
|
|
ch = chPrev = 'A';
|
|
while (!code.eof() && (chPrev!='*' || ch!='/'))
|
|
{
|
|
chPrev = ch;
|
|
ch = (char)code.get();
|
|
if (ch == '\n')
|
|
lineno++;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
// Not a comment.. add token..
|
|
addtoken("/", lineno, FileIndex);
|
|
}
|
|
|
|
// char..
|
|
if (ch == '\'')
|
|
{
|
|
// Add previous token
|
|
addtoken(CurrentToken, lineno, FileIndex);
|
|
memset(CurrentToken, 0, sizeof(CurrentToken));
|
|
|
|
// Read this ..
|
|
CurrentToken[0] = ch;
|
|
CurrentToken[1] = (char)code.get();
|
|
CurrentToken[2] = (char)code.get();
|
|
if (CurrentToken[1] == '\\')
|
|
CurrentToken[3] = (char)code.get();
|
|
|
|
// Add token and start on next..
|
|
addtoken(CurrentToken, lineno, FileIndex);
|
|
memset(CurrentToken, 0, sizeof(CurrentToken));
|
|
pToken = CurrentToken;
|
|
|
|
continue;
|
|
}
|
|
|
|
// String..
|
|
if (ch == '\"')
|
|
{
|
|
addtoken(CurrentToken, lineno, FileIndex);
|
|
memset(CurrentToken, 0, sizeof(CurrentToken));
|
|
pToken = CurrentToken;
|
|
bool special = false;
|
|
char c = ch;
|
|
do
|
|
{
|
|
// Append token..
|
|
if ( pToken < &CurrentToken[sizeof(CurrentToken)-10] )
|
|
{
|
|
*pToken = c;
|
|
pToken++;
|
|
}
|
|
|
|
// Special sequence '\.'
|
|
if (special)
|
|
special = false;
|
|
else
|
|
special = (c == '\\');
|
|
|
|
// Get next character
|
|
c = (char)code.get();
|
|
}
|
|
while (!code.eof() && (special || c != '\"'));
|
|
*pToken = '\"';
|
|
addtoken(CurrentToken, lineno, FileIndex);
|
|
memset(CurrentToken, 0, sizeof(CurrentToken));
|
|
pToken = CurrentToken;
|
|
continue;
|
|
}
|
|
|
|
if (strchr("+-*/%&|^?!=<>[](){};:,.",ch))
|
|
{
|
|
addtoken(CurrentToken, lineno, FileIndex);
|
|
memset(CurrentToken, 0, sizeof(CurrentToken));
|
|
CurrentToken[0] = ch;
|
|
addtoken(CurrentToken, lineno, FileIndex);
|
|
memset(CurrentToken, 0, sizeof(CurrentToken));
|
|
pToken = CurrentToken;
|
|
continue;
|
|
}
|
|
|
|
|
|
if (std::isspace(ch) || std::iscntrl(ch))
|
|
{
|
|
addtoken(CurrentToken, lineno, FileIndex);
|
|
pToken = CurrentToken;
|
|
memset(CurrentToken, 0, sizeof(CurrentToken));
|
|
continue;
|
|
}
|
|
|
|
*pToken = ch;
|
|
pToken++;
|
|
}
|
|
|
|
// Combine tokens..
|
|
for (TOKEN *tok = tokens; tok && tok->next; tok = tok->next)
|
|
{
|
|
combine_2tokens(tok, "<", "<");
|
|
combine_2tokens(tok, ">", ">");
|
|
|
|
combine_2tokens(tok, "&", "&");
|
|
combine_2tokens(tok, "|", "|");
|
|
|
|
combine_2tokens(tok, "+", "=");
|
|
combine_2tokens(tok, "-", "=");
|
|
combine_2tokens(tok, "*", "=");
|
|
combine_2tokens(tok, "/", "=");
|
|
combine_2tokens(tok, "&", "=");
|
|
combine_2tokens(tok, "|", "=");
|
|
|
|
combine_2tokens(tok, "=", "=");
|
|
combine_2tokens(tok, "!", "=");
|
|
combine_2tokens(tok, "<", "=");
|
|
combine_2tokens(tok, ">", "=");
|
|
|
|
combine_2tokens(tok, ":", ":");
|
|
combine_2tokens(tok, "-", ">");
|
|
|
|
combine_2tokens(tok, "private", ":");
|
|
combine_2tokens(tok, "protected", ":");
|
|
combine_2tokens(tok, "public", ":");
|
|
}
|
|
|
|
// Replace "->" with "."
|
|
for ( TOKEN *tok = tokens; tok; tok = tok->next )
|
|
{
|
|
if ( strcmp(tok->str, "->") == 0 )
|
|
{
|
|
strcpy( tok->str, "." );
|
|
}
|
|
}
|
|
}
|
|
//---------------------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
//---------------------------------------------------------------------------
|
|
// Simplify token list
|
|
//---------------------------------------------------------------------------
|
|
|
|
void SimplifyTokenList()
|
|
{
|
|
|
|
// Remove the keyword 'unsigned'
|
|
for ( TOKEN *tok = tokens; tok; tok = tok->next )
|
|
{
|
|
if (tok->next && strcmp(tok->next->str,"unsigned")==0)
|
|
{
|
|
DeleteNextToken( tok );
|
|
}
|
|
}
|
|
|
|
// Replace constants..
|
|
for (TOKEN *tok = tokens; tok; tok = tok->next)
|
|
{
|
|
if (Match(tok,"const %type% %var% = %num% ;"))
|
|
{
|
|
const char *sym = getstr(tok,2);
|
|
const char *num = getstr(tok,4);
|
|
|
|
for (TOKEN *tok2 = _gettok(tok,6); tok2; tok2 = tok2->next)
|
|
{
|
|
if (strcmp(tok2->str,sym) == 0)
|
|
{
|
|
free(tok2->str);
|
|
tok2->str = strdup(num);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
/*
|
|
// typedefs..
|
|
TOKEN *prev = NULL;
|
|
for (TOKEN *tok = tokens; tok; tok = tok->next)
|
|
{
|
|
if (!prev)
|
|
tok = tokens;
|
|
|
|
if ( strcmp(tok->str, "typedef") == 0 )
|
|
{
|
|
TOKEN *type0 = tok->next;
|
|
int len = 0, parlevel = 0;
|
|
for ( TOKEN *tok2 = gettok(type0,1); tok2; tok2 = tok2->next)
|
|
{
|
|
if (strchr("{(", tok2->str[0]))
|
|
parlevel++;
|
|
else if (strchr("})", tok2->str[0]))
|
|
parlevel--;
|
|
else if (parlevel==0 && tok2->str[0]==';')
|
|
break;
|
|
len++;
|
|
}
|
|
|
|
const char *typestr = getstr(type0, len);
|
|
if (typestr[0] != ')') // Function pointer
|
|
{
|
|
// Replace tokens..
|
|
for ( TOKEN *tok2 = gettok(type0, len+1); tok2; tok2 = tok2->next )
|
|
{
|
|
if (strcmp(tok2->str,"typedef")==0)
|
|
{
|
|
int parlevel = 0;
|
|
while (parlevel > 0 || tok2->next->str[0] != ';')
|
|
{
|
|
if (tok2->str[0] == '{')
|
|
parlevel++;
|
|
else if (tok2->str[0] == '}')
|
|
parlevel--;
|
|
|
|
tok2 = tok2->next;
|
|
if (!tok2->next)
|
|
break;
|
|
}
|
|
continue;
|
|
}
|
|
|
|
TOKEN *next = tok2->next;
|
|
if (next && strcmp(next->str, typestr) == 0)
|
|
{
|
|
DeleteNextToken(tok2);
|
|
InsertTokens( tok2, type0, len );
|
|
tok2 = gettok(tok2, len);
|
|
}
|
|
}
|
|
}
|
|
|
|
// Delete typedef..
|
|
if (!prev)
|
|
{
|
|
int parlevel = 0;
|
|
while ( parlevel > 0 || tokens->str[0] != ';' )
|
|
{
|
|
if ( strchr( "({", tokens->str[0] ) )
|
|
parlevel++;
|
|
else if ( strchr( ")}", tokens->str[0] ) )
|
|
parlevel--;
|
|
|
|
// Delete the first element in the tokens list.
|
|
TOKEN *next = tokens->next;
|
|
free(tokens->str);
|
|
delete tokens;
|
|
tokens = next;
|
|
|
|
|
|
if (!tokens)
|
|
break;
|
|
}
|
|
tok = tokens;
|
|
prev = NULL;
|
|
continue;
|
|
}
|
|
else
|
|
{
|
|
int parlevel = 0;
|
|
while ( parlevel > 0 || prev->next->str[0] != ';' )
|
|
{
|
|
if ( strchr( "({", prev->next->str[0] ) )
|
|
parlevel++;
|
|
else if ( strchr( ")}", prev->next->str[0] ) )
|
|
parlevel--;
|
|
DeleteNextToken(prev);
|
|
if (!prev->next)
|
|
break;
|
|
}
|
|
tok = prev;
|
|
}
|
|
}
|
|
prev = tok;
|
|
}
|
|
*/
|
|
|
|
|
|
// Fill the map TypeSize..
|
|
TypeSize.clear();
|
|
TypeSize["char"] = sizeof(char);
|
|
TypeSize["short"] = sizeof(short);
|
|
TypeSize["int"] = sizeof(int);
|
|
TypeSize["long"] = sizeof(long);
|
|
TypeSize["float"] = sizeof(float);
|
|
TypeSize["double"] = sizeof(double);
|
|
for (TOKEN *tok = tokens; tok; tok = tok->next)
|
|
{
|
|
if (Match(tok,"class %var%"))
|
|
{
|
|
TypeSize[getstr(tok,1)] = 11;
|
|
}
|
|
|
|
else if (Match(tok, "struct %var%"))
|
|
{
|
|
TypeSize[getstr(tok,1)] = 13;
|
|
}
|
|
}
|
|
|
|
|
|
// Replace 'sizeof(type)'..
|
|
for (TOKEN *tok = tokens; tok; tok = tok->next)
|
|
{
|
|
if (strcmp(tok->str,"sizeof") != 0)
|
|
continue;
|
|
|
|
if (Match(tok, "sizeof ( %type% * )"))
|
|
{
|
|
free(tok->str);
|
|
char str[10];
|
|
// 'sizeof(type *)' has the same size as 'sizeof(char *)'
|
|
sprintf( str, "%d", sizeof(char *));
|
|
tok->str = strdup( str );
|
|
|
|
for (int i = 0; i < 4; i++)
|
|
{
|
|
DeleteNextToken(tok);
|
|
}
|
|
}
|
|
|
|
else if (Match(tok, "sizeof ( %type% )"))
|
|
{
|
|
const char *type = getstr(tok, 2);
|
|
int size = SizeOfType(type);
|
|
if (size > 0)
|
|
{
|
|
free(tok->str);
|
|
char str[10];
|
|
sprintf( str, "%d", size );
|
|
tok->str = strdup( str );
|
|
for (int i = 0; i < 3; i++)
|
|
{
|
|
DeleteNextToken(tok);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
// Replace 'sizeof(var)'
|
|
for (TOKEN *tok = tokens; tok; tok = tok->next)
|
|
{
|
|
// type array [ num ] ;
|
|
if ( ! Match(tok, "%type% %var% [ %num% ] ;") )
|
|
continue;
|
|
|
|
int size = SizeOfType(tok->str);
|
|
if (size <= 0)
|
|
continue;
|
|
|
|
const char *varname = getstr(tok, 1);
|
|
int total_size = size * atoi( getstr(tok, 3) );
|
|
|
|
// Replace 'sizeof(var)' with number
|
|
int indentlevel = 0;
|
|
for ( TOKEN *tok2 = _gettok(tok,5); tok2; tok2 = tok2->next )
|
|
{
|
|
if (tok2->str[0] == '{')
|
|
{
|
|
indentlevel++;
|
|
}
|
|
|
|
else if (tok2->str[0] == '}')
|
|
{
|
|
indentlevel--;
|
|
if (indentlevel < 0)
|
|
break;
|
|
}
|
|
|
|
// Todo: Match varname directly
|
|
else if (Match(tok2, "sizeof ( %var% )"))
|
|
{
|
|
if (strcmp(getstr(tok2,2), varname) == 0)
|
|
{
|
|
free(tok2->str);
|
|
char str[20];
|
|
sprintf( str, "%d", total_size);
|
|
tok2->str = strdup(str );
|
|
// Delete the other tokens..
|
|
for (int i = 0; i < 3; i++)
|
|
{
|
|
DeleteNextToken(tok2);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
// Simple calculations..
|
|
|
|
bool done = false;
|
|
while (!done)
|
|
{
|
|
done = true;
|
|
|
|
for (TOKEN *tok = tokens; tok; tok = tok->next)
|
|
{
|
|
if (Match(tok->next, "* 1") || Match(tok->next, "1 *"))
|
|
{
|
|
for (int i = 0; i < 2; i++)
|
|
DeleteNextToken(tok);
|
|
done = false;
|
|
}
|
|
|
|
// (1-2)
|
|
if (strchr("[,(=<>",tok->str[0]) &&
|
|
IsNumber(getstr(tok,1)) &&
|
|
strchr("+-*/",*(getstr(tok,2))) &&
|
|
IsNumber(getstr(tok,3)) &&
|
|
strchr("],);=<>",*(getstr(tok,4))) )
|
|
{
|
|
int i1 = atoi(getstr(tok,1));
|
|
int i2 = atoi(getstr(tok,3));
|
|
if ( i2 == 0 && *(getstr(tok,2)) == '/' )
|
|
{
|
|
continue;
|
|
}
|
|
|
|
switch (*(getstr(tok,2)))
|
|
{
|
|
case '+': i1 += i2; break;
|
|
case '-': i1 -= i2; break;
|
|
case '*': i1 *= i2; break;
|
|
case '/': i1 /= i2; break;
|
|
}
|
|
tok = tok->next;
|
|
free(tok->str);
|
|
char str[10];
|
|
sprintf(str,"%d", i1);
|
|
tok->str = strdup(str);
|
|
for (int i = 0; i < 2; i++)
|
|
{
|
|
DeleteNextToken(tok);
|
|
}
|
|
|
|
done = false;
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
// Replace "*(str + num)" => "str[num]"
|
|
for (TOKEN *tok = tokens; tok; tok = tok->next)
|
|
{
|
|
if ( ! strchr(";{}(=<>", tok->str[0]) )
|
|
continue;
|
|
|
|
TOKEN *next = tok->next;
|
|
if ( ! next )
|
|
break;
|
|
|
|
if (Match(next, "* ( %var% + %num% )"))
|
|
{
|
|
const char *str[4] = {"var","[","num","]"};
|
|
str[0] = getstr(tok,3);
|
|
str[2] = getstr(tok,5);
|
|
|
|
for (int i = 0; i < 4; i++)
|
|
{
|
|
tok = tok->next;
|
|
free(tok->str);
|
|
tok->str = strdup(str[i]);
|
|
}
|
|
|
|
DeleteNextToken(tok);
|
|
DeleteNextToken(tok);
|
|
}
|
|
}
|
|
|
|
|
|
|
|
// Split up variable declarations if possible..
|
|
for (TOKEN *tok = tokens; tok; tok = tok->next)
|
|
{
|
|
if ( ! strchr("{};", tok->str[0]) )
|
|
continue;
|
|
|
|
TOKEN *type0 = tok->next;
|
|
if (!type0)
|
|
break;
|
|
if (strcmp(type0->str, "else") == 0)
|
|
continue;
|
|
|
|
TOKEN *tok2 = NULL;
|
|
unsigned int typelen = 0;
|
|
|
|
if ( Match(type0, "%type% %var% ,") )
|
|
{
|
|
tok2 = _gettok(type0, 2); // The ',' token
|
|
typelen = 1;
|
|
}
|
|
|
|
else if ( Match(type0, "%type% * %var% ,") )
|
|
{
|
|
tok2 = _gettok(type0, 3); // The ',' token
|
|
typelen = 1;
|
|
}
|
|
|
|
else if ( Match(type0, "%type% %var% [ %num% ] ,") )
|
|
{
|
|
tok2 = _gettok(type0, 5); // The ',' token
|
|
typelen = 1;
|
|
}
|
|
|
|
else if ( Match(type0, "%type% * %var% [ %num% ] ,") )
|
|
{
|
|
tok2 = _gettok(type0, 6); // The ',' token
|
|
typelen = 1;
|
|
}
|
|
|
|
else if ( Match(type0, "struct %type% %var% ,") )
|
|
{
|
|
tok2 = _gettok(type0, 3);
|
|
typelen = 2;
|
|
}
|
|
|
|
else if ( Match(type0, "struct %type% * %var% ,") )
|
|
{
|
|
tok2 = _gettok(type0, 4);
|
|
typelen = 2;
|
|
}
|
|
|
|
|
|
else if ( Match(type0, "%type% %var% =") )
|
|
{
|
|
tok2 = _gettok(type0, 2);
|
|
typelen = 1;
|
|
}
|
|
|
|
else if ( Match(type0, "%type% * %var% =") )
|
|
{
|
|
tok2 = _gettok(type0, 3);
|
|
typelen = 1;
|
|
}
|
|
|
|
else if ( Match(type0, "struct %type% * %var% =") )
|
|
{
|
|
tok2 = _gettok(type0, 4);
|
|
typelen = 2;
|
|
}
|
|
|
|
if (tok2)
|
|
{
|
|
if (tok2->str[0] == ',')
|
|
{
|
|
free(tok2->str);
|
|
tok2->str = strdup(";");
|
|
InsertTokens(tok2, type0, typelen);
|
|
}
|
|
|
|
else
|
|
{
|
|
TOKEN *eq = tok2;
|
|
|
|
int parlevel = 0;
|
|
while (tok2)
|
|
{
|
|
if ( strchr("{(", tok2->str[0]) )
|
|
{
|
|
parlevel++;
|
|
}
|
|
|
|
else if ( strchr("})", tok2->str[0]) )
|
|
{
|
|
if (parlevel<0)
|
|
break;
|
|
parlevel--;
|
|
}
|
|
|
|
else if ( parlevel==0 && strchr(";,",tok2->str[0]) )
|
|
{
|
|
// "type var =" => "type var; var ="
|
|
TOKEN *VarTok = _gettok(type0,typelen);
|
|
if (VarTok->str[0]=='*')
|
|
VarTok = VarTok->next;
|
|
InsertTokens(eq, VarTok, 2);
|
|
free(eq->str);
|
|
eq->str = strdup(";");
|
|
|
|
// "= x, " => "= x; type "
|
|
if (tok2->str[0] == ',')
|
|
{
|
|
free(tok2->str);
|
|
tok2->str = strdup(";");
|
|
InsertTokens( tok2, type0, typelen );
|
|
}
|
|
break;
|
|
}
|
|
|
|
tok2 = tok2->next;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
//---------------------------------------------------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
//---------------------------------------------------------------------------
|
|
// Helper functions for handling the tokens list
|
|
//---------------------------------------------------------------------------
|
|
|
|
const TOKEN *findtoken(const TOKEN *tok1, const char *tokenstr[])
|
|
{
|
|
for (const TOKEN *ret = tok1; ret; ret = ret->next)
|
|
{
|
|
unsigned int i = 0;
|
|
const TOKEN *tok = ret;
|
|
while (tokenstr[i])
|
|
{
|
|
if (!tok)
|
|
return NULL;
|
|
if (*(tokenstr[i]) && strcmp(tokenstr[i],tok->str))
|
|
break;
|
|
tok = tok->next;
|
|
i++;
|
|
}
|
|
if (!tokenstr[i])
|
|
return ret;
|
|
}
|
|
return NULL;
|
|
}
|
|
//---------------------------------------------------------------------------
|
|
|
|
const TOKEN *gettok(const TOKEN *tok, int index)
|
|
{
|
|
while (tok && index>0)
|
|
{
|
|
tok = tok->next;
|
|
index--;
|
|
}
|
|
return tok;
|
|
}
|
|
//---------------------------------------------------------------------------
|
|
|
|
const char *getstr(const TOKEN *tok, int index)
|
|
{
|
|
tok = gettok(tok, index);
|
|
return tok ? tok->str : "";
|
|
}
|
|
//---------------------------------------------------------------------------
|
|
|
|
|
|
|
|
// Deallocate lists..
|
|
void DeallocateTokens()
|
|
{
|
|
while (tokens)
|
|
{
|
|
TOKEN *next = tokens->next;
|
|
free(tokens->str);
|
|
delete tokens;
|
|
tokens = next;
|
|
}
|
|
tokens_back = tokens;
|
|
|
|
while (dsymlist)
|
|
{
|
|
struct DefineSymbol *next = dsymlist->next;
|
|
free(dsymlist->name);
|
|
free(dsymlist->value);
|
|
delete dsymlist;
|
|
dsymlist = next;
|
|
}
|
|
}
|
|
|
|
|
|
|
|
|
|
|