CPP
/* $Id$ */
/* Copyright (c) 2008-2015 Pierre Pronchery <khorben@defora.org> */
/* This file is part of DeforaOS Devel cpp */
/* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Lesser General Public License as published by
* the Free Software Foundation, version 3 of the License.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Lesser General Public License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>. */
/* FIXME:
* - potential memory leak with tokens' data
* - apparently not checking that scopes are properly closed upon exit */
#include <assert.h>
#include <unistd.h>
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <ctype.h>
#include <errno.h>
#include "common.h"
#ifdef DEBUG
# define DEBUG_SCOPE() fprintf(stderr, "DEBUG: %s(%p) %zu scope=%d\n", \
__func__, cpp, _cpp_scope_get_count(cpp), _cpp_scope_get(cpp))
#else
# define DEBUG_SCOPE()
#endif
/* private */
/* prototypes */
/* scope */
static int _cpp_scope_push(Cpp * cpp, CppScope scope);
static CppScope _cpp_scope_get(Cpp * cpp);
static size_t _cpp_scope_get_count(Cpp * cpp);
static void _cpp_scope_set(Cpp * cpp, CppScope scope);
static int _cpp_scope_pop(Cpp * cpp);
/* functions */
/* scope */
/* cpp_scope_push */
static int _cpp_scope_push(Cpp * cpp, CppScope scope)
{
CppScope * p;
if(_cpp_scope_get(cpp) != CPP_SCOPE_TAKING)
scope = CPP_SCOPE_TAKEN;
if((p = realloc(cpp->scopes, sizeof(*p) * (cpp->scopes_cnt + 1)))
== NULL)
return error_set_code(1, "%s", strerror(errno));
cpp->scopes = p;
p[cpp->scopes_cnt++] = scope;
return 0;
}
/* cpp_scope_get */
static CppScope _cpp_scope_get(Cpp * cpp)
{
return (cpp->scopes_cnt == 0) ? CPP_SCOPE_TAKING
: cpp->scopes[cpp->scopes_cnt - 1];
}
/* cpp_scope_get_count */
static size_t _cpp_scope_get_count(Cpp * cpp)
{
return cpp->scopes_cnt;
}
/* cpp_scope_set */
static void _cpp_scope_set(Cpp * cpp, CppScope scope)
{
assert(cpp->scopes_cnt > 0);
cpp->scopes[cpp->scopes_cnt - 1] = scope;
}
/* cpp_scope_pop */
static int _cpp_scope_pop(Cpp * cpp)
{
CppScope * p;
assert(cpp->scopes_cnt > 0);
if(cpp->scopes_cnt == 1)
{
free(cpp->scopes);
p = NULL;
}
else if((p = realloc(cpp->scopes, sizeof(*p) * (cpp->scopes_cnt - 1)))
== NULL)
return error_set_code(1, "%s", strerror(errno));
cpp->scopes = p;
cpp->scopes_cnt--;
return 0;
}
/* public */
/* cpp_scan */
static void _scan_meta_line(Token * token);
static int _scan_ifdef(Cpp * cpp, Token ** token);
static int _scan_ifndef(Cpp * cpp, Token ** token);
static int _scan_if(Cpp * cpp, Token ** token);
static CppScope _if_do(Cpp * cpp, char const * str, int level);
static int _scan_elif(Cpp * cpp, Token ** token);
static int _scan_else(Cpp * cpp, Token ** token);
static int _scan_endif(Cpp * cpp, Token ** token);
static int _scan_define(Cpp * cpp, Token * token);
static int _scan_include(Cpp * cpp, Token * token);
static int _scan_undef(Cpp * cpp, Token * token);
int cpp_scan(Cpp * cpp, Token ** token)
{
int ret;
TokenCode code;
char const * str;
char const * s;
for(; (ret = cppparser_scan(cpp->parser, token)) == 0;
token_delete(*token))
{
if(*token == NULL) /* end of file */
break;
if((code = token_get_code(*token)) >= CPP_CODE_META_FIRST
&& code <= CPP_CODE_META_LAST)
_scan_meta_line(*token);
if(cpp->options & CPP_OPTION_IGNORE_META)
{
if(code == CPP_CODE_META_ERROR
|| code == CPP_CODE_META_WARNING)
continue;
return 0;
}
switch(code)
{
case CPP_CODE_META_IFDEF:
return _scan_ifdef(cpp, token);
case CPP_CODE_META_IFNDEF:
return _scan_ifndef(cpp, token);
case CPP_CODE_META_IF:
return _scan_if(cpp, token);
case CPP_CODE_META_ELIF:
return _scan_elif(cpp, token);
case CPP_CODE_META_ELSE:
return _scan_else(cpp, token);
case CPP_CODE_META_ENDIF:
return _scan_endif(cpp, token);
default:
break;
}
if(_cpp_scope_get(cpp) != CPP_SCOPE_TAKING) /* not in scope */
continue;
switch(code)
{
case CPP_CODE_META_DEFINE:
return _scan_define(cpp, *token);
case CPP_CODE_META_INCLUDE:
return _scan_include(cpp, *token);
case CPP_CODE_META_UNDEF:
return _scan_undef(cpp, *token);
case CPP_CODE_WORD:
str = token_get_string(*token);
if((s = cpp_define_get(cpp, str)) != NULL)
{
if(strcmp(str, s) == 0) /* XXX hack */
return 0;
if(cppparser_inject(cpp->parser, s)
!= 0)
return -1;
continue;
}
return 0;
default:
return 0;
}
}
return ret;
}
static void _scan_meta_line(Token * token)
{
/* decrement the line number, not so elegant but works */
token_set_line(token, token_get_line(token) - 1);
}
static int _scan_ifdef(Cpp * cpp, Token ** token)
{
char * name;
int take;
DEBUG_SCOPE();
name = token_get_data(*token);
take = (cpp_define_get(cpp, name) != NULL) ? 1 : 0;
token_set_data(*token, NULL);
free(name);
_cpp_scope_push(cpp, take ? CPP_SCOPE_TAKING : CPP_SCOPE_NOTYET);
return 0;
}
static int _scan_ifndef(Cpp * cpp, Token ** token)
{
char * name;
int take;
DEBUG_SCOPE();
name = token_get_data(*token);
take = (cpp_define_get(cpp, name) == NULL) ? 1 : 0;
token_set_data(*token, NULL);
free(name);
_cpp_scope_push(cpp, take ? CPP_SCOPE_TAKING : CPP_SCOPE_NOTYET);
return 0;
}
static int _scan_if(Cpp * cpp, Token ** token)
{
char * str;
DEBUG_SCOPE();
str = token_get_data(*token);
_cpp_scope_push(cpp, _if_do(cpp, str, 0));
token_set_data(*token, NULL);
free(str);
return 0;
}
static CppScope _if_do(Cpp * cpp, char const * str, int level)
{
char * p;
char const * q;
long l;
if(str == NULL || level >= 10)
/* FIXME it's probably an error case instead */
return CPP_SCOPE_NOTYET;
if(str[0] == '!')
return (_if_do(cpp, &str[1], level) == CPP_SCOPE_TAKING)
? CPP_SCOPE_NOTYET : CPP_SCOPE_TAKING;
l = strtol(str, &p, 0);
if(str[0] != '\0' && p != str)
return (l != 0) ? CPP_SCOPE_TAKING : CPP_SCOPE_NOTYET;
if(strncmp(str, "defined(", 8) == 0 && (p = strchr(str, ')')) != NULL)
{
*p = '\0';
return (cpp_define_get(cpp, &str[8]) != NULL)
? CPP_SCOPE_TAKING : CPP_SCOPE_NOTYET;
}
else if((q = cpp_define_get(cpp, str)) != NULL)
return _if_do(cpp, q, level + 1);
/* FIXME really check the condition */
return CPP_SCOPE_NOTYET;
}
static int _scan_elif(Cpp * cpp, Token ** token)
{
int ret = 0;
CppScope scope;
char * str;
DEBUG_SCOPE();
str = token_get_data(*token);
if(_cpp_scope_get_count(cpp) == 0)
{
token_set_code(*token, CPP_CODE_META_ERROR);
token_set_string(*token, "#elif without #if or #ifdef"
" or #ifndef");
ret = 0;
}
else
{
scope = _cpp_scope_get(cpp);
if(scope == CPP_SCOPE_TAKING)
_cpp_scope_set(cpp, CPP_SCOPE_TAKEN);
else if(scope == CPP_SCOPE_NOTYET)
_cpp_scope_set(cpp, _if_do(cpp, str, 0));
}
if(str != NULL)
{
token_set_data(*token, NULL);
free(str);
}
return ret;
}
static int _scan_else(Cpp * cpp, Token ** token)
{
CppScope scope;
DEBUG_SCOPE();
if(_cpp_scope_get_count(cpp) == 0)
{
token_set_code(*token, CPP_CODE_META_ERROR);
token_set_string(*token, "#else without #if or #ifdef"
" or #ifndef");
return 0;
}
scope = _cpp_scope_get(cpp);
if(scope == CPP_SCOPE_TAKING)
_cpp_scope_set(cpp, CPP_SCOPE_TAKEN);
else if(scope == CPP_SCOPE_NOTYET)
_cpp_scope_set(cpp, CPP_SCOPE_TAKING);
return 0;
}
static int _scan_endif(Cpp * cpp, Token ** token)
{
DEBUG_SCOPE();
if(_cpp_scope_get_count(cpp) == 0)
{
token_set_code(*token, CPP_CODE_META_ERROR);
token_set_string(*token, "#endif without #if or #ifdef"
" or #ifndef");
}
_cpp_scope_pop(cpp);
return 0;
}
static int _scan_define(Cpp * cpp, Token * token)
{
char * str;
int tmp;
size_t i;
size_t j;
size_t k = 0;
char * var;
char const * val;
str = token_get_data(token);
/* fetch variable name */
for(i = 1; (tmp = str[i]) != '\0' && !isspace(tmp); i++)
{
/* FIXME actually implement macros */
if(str[i] != '(')
continue;
for(k = i; str[i] != '\0' && str[i] != ')'; i++);
if(str[i] == ')')
i++;
break;
}
/* skip white-spaces and fetch value */
for(j = i; (tmp = str[j]) != '\0' && isspace(tmp); j++);
val = (str[j] != '\0') ? &str[j] : NULL;
if((var = strdup(str)) == NULL)
{
token_set_code(token, CPP_CODE_META_ERROR);
token_set_string(token, strerror(errno));
token_set_data(token, NULL);
free(str);
return 0;
}
var[k != 0 ? k : i] = '\0';
if(cpp_define_add(cpp, var, val) != 0)
{
token_set_code(token, CPP_CODE_META_ERROR);
token_set_string(token, error_get(NULL));
}
free(var);
token_set_data(token, NULL);
free(str);
return 0;
}
static int _scan_include(Cpp * cpp, Token * token)
{
if(cppparser_include(cpp->parser, token_get_data(token)) == 0)
return 0;
token_set_code(token, CPP_CODE_META_ERROR);
token_set_string(token, error_get(NULL));
return 0;
}
static int _scan_undef(Cpp * cpp, Token * token)
/* FIXME ignores what's after the spaces after the variable name */
{
char * str;
int tmp;
size_t i;
char * var;
str = token_get_data(token);
/* fetch variable name */
for(i = 1; (tmp = str[i]) != '\0' && !isspace(tmp); i++);
if((var = strdup(str)) == NULL)
{
token_set_code(token, CPP_CODE_META_ERROR);
token_set_string(token, strerror(errno));
free(str);
return 0;
}
var[i] = '\0';
if(cpp_define_remove(cpp, var) != 0)
{
token_set_code(token, CPP_CODE_META_ERROR);
token_set_string(token, error_get(NULL));
}
free(var);
free(str);
return 0;
}