X-Git-Url: https://git.m6w6.name/?p=m6w6%2Fext-psi;a=blobdiff_plain;f=src%2Fcpp.c;h=fc723641dcca1e363399a35b933e248143c998c0;hp=c436ef33bf33f46ef0e0aeaa65e9dd79d618b089;hb=0682ecb84dfe1de08b609c6c0ae641b06f3fa481;hpb=ef4e043826e92afd32000b7d945cdf86a6e9a223

diff --git a/src/cpp.c b/src/cpp.c
index c436ef3..fc72364 100644
--- a/src/cpp.c
+++ b/src/cpp.c
@@ -41,7 +41,57 @@
 #define PSI_CPP_PREDEF
 #include "php_psi_predef.h"
 
-HashTable psi_cpp_defaults;
+static HashTable psi_cpp_defaults;
+static HashTable psi_cpp_pragmas;
+
+typedef bool (*psi_cpp_pragma_func)(struct psi_cpp *cpp, struct psi_cpp_macro_decl *decl);
+
+static bool psi_cpp_pragma_once(struct psi_cpp *cpp, struct psi_cpp_macro_decl *decl)
+{
+	return NULL != zend_hash_add_empty_element(&cpp->once, decl->token->file);
+}
+
+static bool psi_cpp_pragma_lib(struct psi_cpp *cpp, struct psi_cpp_macro_decl *decl)
+{
+	struct psi_token *lib = NULL;
+	zend_string *libname;
+
+	if (!psi_plist_get(decl->tokens, 0, &lib)
+			|| !lib || lib->type != PSI_T_QUOTED_STRING) {
+		return false;
+	}
+
+	libname = zend_string_copy(lib->text);
+	cpp->parser->file.libnames = psi_plist_add(cpp->parser->file.libnames,
+			&libname);
+	return true;
+}
+
+static bool psi_cpp_pragma_blacklist_decl(struct psi_cpp *cpp, struct psi_cpp_macro_decl *decl)
+{
+	struct psi_token *name;
+
+	if (!psi_plist_get(decl->tokens, 0, &name)
+			|| !name || name->type != PSI_T_QUOTED_STRING) {
+		return false;
+	}
+
+	psi_blacklist_add_decl(name->text->val, name->text->len);
+	return true;
+}
+
+static bool psi_cpp_pragma_blacklist_var(struct psi_cpp *cpp, struct psi_cpp_macro_decl *decl)
+{
+	struct psi_token *name;
+
+	if (!psi_plist_get(decl->tokens, 0, &name)
+			|| !name || name->type != PSI_T_QUOTED_STRING) {
+		return false;
+	}
+
+	psi_blacklist_add_var(name->text->val, name->text->len);
+	return true;
+}
 
 PHP_MINIT_FUNCTION(psi_cpp);
 PHP_MINIT_FUNCTION(psi_cpp)
@@ -73,6 +123,14 @@ PHP_MINIT_FUNCTION(psi_cpp)
 
 	psi_parser_dtor(&parser);
 
+#define PSI_CPP_PRAGMA(name) \
+	zend_hash_str_add_ptr(&psi_cpp_pragmas, #name, strlen(#name), psi_cpp_pragma_ ## name)
+	zend_hash_init(&psi_cpp_pragmas, 0, NULL, NULL, 1);
+	PSI_CPP_PRAGMA(once);
+	PSI_CPP_PRAGMA(lib);
+	PSI_CPP_PRAGMA(blacklist_decl);
+	PSI_CPP_PRAGMA(blacklist_var);
+
 	return SUCCESS;
 }
 
@@ -457,17 +515,33 @@ bool psi_cpp_undef(struct psi_cpp *cpp, struct psi_token *tok)
 bool psi_cpp_if(struct psi_cpp *cpp, struct psi_cpp_exp *exp)
 {
 	struct psi_validate_scope scope = {0};
+	unsigned flags = cpp->parser->flags;
 
 	scope.cpp = cpp;
+	cpp->parser->flags |= PSI_SILENT;
 	if (!psi_num_exp_validate(PSI_DATA(cpp->parser), exp->data.num, &scope)) {
+		cpp->parser->flags = flags;
 		return false;
 	}
+	cpp->parser->flags = flags;
 	if (!psi_num_exp_get_long(exp->data.num, NULL, cpp)) {
 		return false;
 	}
 	return true;
 }
 
+bool psi_cpp_pragma(struct psi_cpp *cpp, struct psi_cpp_macro_decl *decl)
+{
+	psi_cpp_pragma_func fn;
+
+	fn = zend_hash_find_ptr(&psi_cpp_pragmas, decl->token->text);
+	if (!fn) {
+		return false;
+	}
+
+	return fn(cpp, decl);
+}
+
 bool psi_cpp_include(struct psi_cpp *cpp, const struct psi_token *file, unsigned flags)
 {
 	bool parsed = false;