| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
|
|
| #include <stdlib.h> |
| #include <stdio.h> |
| #include <errno.h> |
| #include <string.h> |
| #include <stdarg.h> |
| #include <assert.h> |
| #include <ctype.h> |
|
|
| #include "sqlite3.h" |
|
|
| #if defined(_WIN32) || defined(WIN32) |
| # include <io.h> |
| # include <fcntl.h> |
| # ifndef access |
| # define access(f,m) _access((f),(m)) |
| # endif |
| #else |
| # include <unistd.h> |
| #endif |
|
|
| #ifndef CMPP_DEFAULT_DELIM |
| #define CMPP_DEFAULT_DELIM "##" |
| #endif |
|
|
| #ifndef CMPP_ATSIGN |
| #define CMPP_ATSIGN (unsigned char)'@' |
| #endif |
|
|
| #if 1 |
| # define CMPP_NORETURN __attribute__((noreturn)) |
| #else |
| # define CMPP_NORETURN |
| #endif |
|
|
| |
| static CMPP_NORETURN void fatalv__base(char const *zFile, int line, |
| char const *zFmt, va_list); |
| static CMPP_NORETURN void fatal__base(char const *zFile, int line, |
| char const *zFmt, ...); |
| #define fatalv(...) fatalv__base(__FILE__,__LINE__,__VA_ARGS__) |
| #define fatal(...) fatal__base(__FILE__,__LINE__,__VA_ARGS__) |
|
|
| |
| static void cmpp_free(void *p); |
| |
| static void * cmpp_realloc(void * p, unsigned n); |
| #if 0 |
| |
| static void * cmpp_malloc(unsigned n); |
| #endif |
|
|
| static void check__oom2(void const *p, char const *zFile, int line){ |
| if(!p) fatal("Alloc failed at %s:%d", zFile, line); |
| } |
| #define check__oom(P) check__oom2((P), __FILE__, __LINE__) |
|
|
| |
| |
| |
| |
| static void FILE_close(FILE *p); |
| |
| |
| |
| |
| |
| static FILE * FILE_open(char const *zName, const char * zMode); |
| |
| |
| |
| |
| |
| static void FILE_slurp(FILE *pFile, unsigned char **pOut, |
| unsigned * nOut); |
|
|
| |
| |
| |
| |
| |
| |
| static void db_affirm_rc(int rc, const char * zMsg); |
|
|
| |
| |
| |
| |
| static char * db_str_finish(sqlite3_str *s, int * n); |
| |
| |
| |
| |
| static sqlite3_str * db_str_new(void); |
|
|
| |
| |
| |
| |
| static int db_step(sqlite3_stmt *pStmt); |
| |
| |
| |
| static void db_bind_int(sqlite3_stmt *pStmt, int col, int val); |
| |
| |
| |
| static void db_bind_null(sqlite3_stmt *pStmt, int col); |
| |
| |
| |
| static void db_bind_text(sqlite3_stmt *pStmt, int col, const char * zStr); |
| |
| |
| |
| static void db_bind_textn(sqlite3_stmt *pStmt, int col, const char * zStr, int len); |
| #if 0 |
| |
| |
| |
| |
| static void db_bind_textv(sqlite3_stmt *pStmt, int col, const char * zFmt, ...); |
| #endif |
| |
| |
| |
| |
| static void db_free(void *m); |
|
|
| |
| |
| |
| |
| |
| |
| static int cmpp_is_legal_key(char const *zKey, int nKey, char const **zErrPos); |
|
|
| |
| |
| |
| static void cmpp_affirm_legal_key(char const *zKey, int nKey); |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| static void db_define_add(const char * zKey, char const *zVal); |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| static int db_define_has(const char * zName, int nName); |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| static int db_define_get_bool(const char * zName, int nName); |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| static int db_define_get(const char * zName, int nName, char **zVal, unsigned int *nVal); |
|
|
| |
| |
| |
| |
| static void db_define_rm(const char * zKey); |
| |
| |
| |
| |
| |
| static void db_including_add(const char * zKey, const char * zSrc, int srcLine); |
| |
| |
| |
| |
| static void db_include_dir_add(const char * zKey); |
| |
| |
| |
| |
| |
| |
| static char * db_include_search(const char * zKey); |
| |
| |
| |
| static void db_include_rm(const char * zKey); |
| |
| |
| |
| static void db_prepare(sqlite3_stmt **pStmt, const char * zSql, ...); |
|
|
| |
| |
| |
| |
| |
| |
| static void cmpp_process_file(const char * zName, int bRaw); |
|
|
| |
| |
| |
| enum cmpp_key_op_e { |
| |
| cmpp_key_op_none, |
| |
| cmpp_key_op_eq1 |
| }; |
| typedef enum cmpp_key_op_e cmpp_key_op_e; |
|
|
| |
| |
| |
| #define cmpp_kvp_op_map(E) \ |
| E(none,"") \ |
| E(eq1,"=") \ |
| E(eq2,"==") \ |
| E(lt,"<") \ |
| E(le,"<=") \ |
| E(gt,">") \ |
| E(ge,">=") |
|
|
| enum cmpp_kvp_op_e { |
| #define E(N,S) cmpp_kvp_op_ ## N, |
| cmpp_kvp_op_map(E) |
| #undef E |
| }; |
| typedef enum cmpp_kvp_op_e cmpp_kvp_op_e; |
|
|
| |
| |
| |
| struct cmpp_snippet { |
| char const *z; |
| unsigned int n; |
| }; |
| typedef struct cmpp_snippet cmpp_snippet; |
| #define cmpp_snippet_empty_m {0,0} |
|
|
| |
| |
| |
| struct cmpp_kvp { |
| cmpp_snippet k; |
| cmpp_snippet v; |
| cmpp_kvp_op_e op; |
| }; |
|
|
| typedef struct cmpp_kvp cmpp_kvp; |
| #define cmpp_kvp_empty_m \ |
| {cmpp_snippet_empty_m,cmpp_snippet_empty_m,cmpp_kvp_op_none} |
| static const cmpp_kvp cmpp_kvp_empty = cmpp_kvp_empty_m; |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| static void cmpp_kvp_parse(cmpp_kvp * p, |
| char const *zKey, int nKey, |
| cmpp_kvp_op_e opPolicy); |
|
|
| |
| |
| |
| typedef struct FileWrapper FileWrapper; |
| struct FileWrapper { |
| |
| char const *zName; |
| |
| FILE * pFile; |
| |
| unsigned char * zContent; |
| |
| unsigned nContent; |
| |
| FileWrapper * pTail; |
| }; |
| #define FileWrapper_empty_m {0,0,0,0,0} |
| static const FileWrapper FileWrapper_empty = FileWrapper_empty_m; |
|
|
| |
| |
| |
| |
| static void FileWrapper_close(FileWrapper * p); |
| |
| static void FileWrapper_open(FileWrapper * p, const char * zName, const char *zMode); |
| |
| static void FileWrapper_slurp(FileWrapper * p); |
| |
| |
| |
| |
| int FileWrapper_chomp(FileWrapper * p); |
|
|
| |
| |
| |
| static void g_stderr(char const *zFmt, ...); |
| |
| |
| |
| static void g_stderrv(char const *zFmt, va_list); |
| #define g_debug(lvl,pfexpr) \ |
| if(lvl<=g.flags.doDebug) g_stderr("%s @ %s():%d: ",g.zArgv0,__func__,__LINE__); \ |
| if(lvl<=g.flags.doDebug) g_stderr pfexpr |
|
|
| #define g_warn(zFmt,...) g_stderr("%s:%d %s() " zFmt "\n", __FILE__, __LINE__, __func__, __VA_ARGS__) |
| #define g_warn0(zMsg) g_stderr("%s:%d %s() %s\n", __FILE__, __LINE__, __func__, zMsg) |
|
|
| void cmpp_free(void *p){ |
| sqlite3_free(p); |
| } |
|
|
| void * cmpp_realloc(void * p, unsigned n){ |
| void * const rc = sqlite3_realloc(p, n); |
| if(!rc) fatal("realloc(P,%u) failed", n); |
| return rc; |
| } |
|
|
| #if 0 |
| void * cmpp_malloc(unsigned n){ |
| void * const rc = sqlite3_alloc(n); |
| if(!rc) fatal("malloc(%u) failed", n); |
| return rc; |
| } |
| #endif |
|
|
| FILE * FILE_open(char const *zName, const char * zMode){ |
| FILE * p; |
| if('-'==zName[0] && 0==zName[1]){ |
| p = strstr(zMode,"w") ? stdout : stdin; |
| }else{ |
| p = fopen(zName, zMode); |
| if(!p) fatal("Cannot open file [%s] with mode [%s]", zName, zMode); |
| } |
| return p; |
| } |
|
|
| void FILE_close(FILE *p){ |
| if(p && p!=stdout && p!=stderr){ |
| fclose(p); |
| } |
| } |
|
|
| void FILE_slurp(FILE *pFile, unsigned char **pOut, |
| unsigned * nOut){ |
| unsigned char zBuf[1024 * 8]; |
| unsigned char * pDest = 0; |
| unsigned nAlloc = 0; |
| unsigned nOff = 0; |
| |
| |
| |
| while( !feof(pFile) ){ |
| size_t const n = fread(zBuf, 1, sizeof(zBuf), pFile); |
| if(n>0){ |
| if(nAlloc < nOff + n + 1){ |
| nAlloc = nOff + n + 1; |
| pDest = cmpp_realloc(pDest, nAlloc); |
| } |
| memcpy(pDest + nOff, zBuf, n); |
| nOff += n; |
| } |
| } |
| if(pDest) pDest[nOff] = 0; |
| *pOut = pDest; |
| *nOut = nOff; |
| } |
|
|
| void FileWrapper_close(FileWrapper * p){ |
| if(p->pFile) FILE_close(p->pFile); |
| if(p->zContent) cmpp_free(p->zContent); |
| *p = FileWrapper_empty; |
| } |
|
|
| void FileWrapper_open(FileWrapper * p, const char * zName, |
| const char * zMode){ |
| FileWrapper_close(p); |
| p->pFile = FILE_open(zName, zMode); |
| p->zName = zName; |
| } |
|
|
| void FileWrapper_slurp(FileWrapper * p){ |
| assert(!p->zContent); |
| assert(p->pFile); |
| FILE_slurp(p->pFile, &p->zContent, &p->nContent); |
| } |
|
|
| int FileWrapper_chomp(FileWrapper * p){ |
| if( p->nContent && '\n'==p->zContent[p->nContent-1] ){ |
| p->zContent[--p->nContent] = 0; |
| if( p->nContent && '\r'==p->zContent[p->nContent-1] ){ |
| p->zContent[--p->nContent] = 0; |
| } |
| return 1; |
| } |
| return 0; |
| } |
|
|
| enum CmppParseState { |
| TS_Start = 1, |
| TS_If, |
| TS_IfPassed, |
| TS_Else, |
| TS_Error |
| }; |
| typedef enum CmppParseState CmppParseState; |
| enum CmppTokenType { |
|
|
| #define CmppToken_map(E) \ |
| E(Invalid,0) \ |
| E(Assert,"assert") \ |
| E(AtPolicy,"@policy") \ |
| E(Comment,"//") \ |
| E(Define,"define") \ |
| E(Elif,"elif") \ |
| E(Else,"else") \ |
| E(Endif,"endif") \ |
| E(Error,"error") \ |
| E(If,"if") \ |
| E(Include,"include") \ |
| E(Line,0) \ |
| E(Opaque,0) \ |
| E(Pragma,"pragma") \ |
| E(Savepoint,"savepoint") \ |
| E(Stderr,"stderr") \ |
| E(Undef,"undef") |
|
|
| #define E(N,TOK) TT_ ## N, |
| CmppToken_map(E) |
| #undef E |
| }; |
| typedef enum CmppTokenType CmppTokenType; |
|
|
| |
| |
| |
| static const struct { |
| #define E(N,TOK) struct cmpp_snippet N; |
| CmppToken_map(E) |
| #undef E |
| } DStrings = { |
| #define E(N,TOK) .N = {TOK,sizeof(TOK)-1}, |
| CmppToken_map(E) |
| #undef E |
| }; |
|
|
| |
| char const * TT_cstr(int tt){ |
| switch(tt){ |
| #define E(N,TOK) case TT_ ## N: return DStrings.N.z; |
| CmppToken_map(E) |
| #undef E |
| } |
| return NULL; |
| } |
|
|
| struct CmppToken { |
| CmppTokenType ttype; |
| |
| unsigned lineNo; |
| |
| unsigned char const * zBegin; |
| |
| unsigned char const * zEnd; |
| }; |
| typedef struct CmppToken CmppToken; |
| #define CmppToken_empty_m {TT_Invalid,0,0,0} |
| static const CmppToken CmppToken_empty = CmppToken_empty_m; |
|
|
| |
| |
| |
| |
| |
| |
| typedef struct CmppLevel CmppLevel; |
| struct CmppLevel { |
| unsigned short flags; |
| |
| |
| |
| |
| unsigned short skipLevel; |
| |
| CmppToken token; |
| CmppParseState pstate; |
| }; |
| #define CmppLevel_empty_m {0U,0U,CmppToken_empty_m,TS_Start} |
| static const CmppLevel CmppLevel_empty = CmppLevel_empty_m; |
| enum CmppLevel_Flags { |
| |
| CmppLevel_Max = 10, |
| |
| CmppArgs_Max = 15, |
| |
| CmppArgs_BufSize = 1024, |
| |
| CmppLevel_F_ELIDE = 0x01, |
| |
| |
| |
| |
| CmppLevel_F_INHERIT_MASK = CmppLevel_F_ELIDE |
| }; |
|
|
| typedef struct CmppTokenizer CmppTokenizer; |
| typedef struct CmppKeyword CmppKeyword; |
| typedef void (*cmpp_keyword_f)(CmppKeyword const * pKw, CmppTokenizer * t); |
| struct CmppKeyword { |
| const char *zName; |
| unsigned nName; |
| int bTokenize; |
| CmppTokenType ttype; |
| cmpp_keyword_f xCall; |
| }; |
|
|
| static CmppKeyword const * CmppKeyword_search(const char *zName); |
| static void cmpp_process_keyword(CmppTokenizer * const t); |
|
|
| |
| |
| |
| struct CmppTokenizer { |
| const char * zName; |
| unsigned const char * zBegin; |
| unsigned const char * zEnd; |
| unsigned const char * zPos; |
| unsigned int lineNo; |
| unsigned nSavepoint; |
| CmppParseState pstate; |
| CmppToken token; |
| struct { |
| unsigned ndx; |
| CmppLevel stack[CmppLevel_Max]; |
| } level; |
| |
| struct { |
| CmppKeyword const * pKw; |
| int argc; |
| const unsigned char * argv[CmppArgs_Max]; |
| unsigned char lineBuf[CmppArgs_BufSize]; |
| } args; |
| }; |
| #define CT_level(t) (t)->level.stack[(t)->level.ndx] |
| #define CT_pstate(t) CT_level(t).pstate |
| #define CT_skipLevel(t) CT_level(t).skipLevel |
| #define CLvl_skip(lvl) ((lvl)->skipLevel || ((lvl)->flags & CmppLevel_F_ELIDE)) |
| #define CT_skip(t) CLvl_skip(&CT_level(t)) |
| #define CmppTokenizer_empty_m { \ |
| .zName=0, .zBegin=0, .zEnd=0, \ |
| .zPos=0, \ |
| .lineNo=1U, \ |
| .pstate = TS_Start, \ |
| .token = CmppToken_empty_m, \ |
| .level = {0U,{CmppLevel_empty_m}}, \ |
| .args = {0,0,{0},{0}} \ |
| } |
| static const CmppTokenizer CmppTokenizer_empty = CmppTokenizer_empty_m; |
|
|
| static void CmppTokenizer_cleanup(CmppTokenizer * const t); |
|
|
| static void cmpp_t_out(CmppTokenizer * t, void const *z, unsigned int n); |
| |
|
|
| |
| |
| |
| |
| static void CmppLevel_push(CmppTokenizer * const t); |
| |
| |
| |
| |
| static void CmppLevel_pop(CmppTokenizer * const t); |
| |
| |
| |
| static CmppLevel * CmppLevel_get(CmppTokenizer * const t); |
|
|
| |
| |
| |
| |
| enum AtPolicy { |
| AT_invalid = -1, |
| |
| AT_OFF = 0, |
| |
| AT_RETAIN, |
| |
| AT_ELIDE, |
| |
| AT_ERROR, |
| AT_DEFAULT = AT_ERROR |
| }; |
| typedef enum AtPolicy AtPolicy; |
|
|
| static AtPolicy AtPolicy_fromStr(char const *z, int bEnforce){ |
| if( 0==strcmp(z, "retain") ) return AT_RETAIN; |
| if( 0==strcmp(z, "elide") ) return AT_ELIDE; |
| if( 0==strcmp(z, "error") ) return AT_ERROR; |
| if( 0==strcmp(z, "off") ) return AT_OFF; |
| if( bEnforce ){ |
| fatal("Invalid @ policy value: %s. " |
| "Try one of retain|elide|error|off.", z); |
| } |
| return AT_invalid; |
| } |
|
|
| |
| |
| |
| static struct Global { |
| |
| const char * zArgv0; |
| |
| sqlite3 * db; |
| |
| CmppTokenizer const * tok; |
| |
| |
| |
| |
| |
| FileWrapper * pFiles; |
| |
| FileWrapper out; |
| struct { |
| |
| |
| |
| |
| const char * z; |
| |
| unsigned short n; |
| |
| |
| |
| |
| |
| |
| const unsigned char chAt; |
| } delim; |
| struct { |
| #define CMPP_SAVEPOINT_NAME "_cmpp_" |
| #define GStmt_map(E) \ |
| E(defIns,"INSERT OR REPLACE INTO def(k,v) VALUES(?,?)") \ |
| E(defDel,"DELETE FROM def WHERE k GLOB ?") \ |
| E(defHas,"SELECT 1 FROM def WHERE k GLOB ?") \ |
| E(defGet,"SELECT k,v FROM def WHERE k GLOB ?") \ |
| E(defGetBool, \ |
| "SELECT 1 FROM def WHERE k = ?1" \ |
| " AND v IS NOT NULL" \ |
| " AND '0'!=v AND ''!=v") \ |
| E(defSelAll,"SELECT k,v FROM def ORDER BY k") \ |
| E(inclIns,"INSERT OR FAIL INTO incl(file,srcFile," \ |
| "srcLine) VALUES(?,?,?)") \ |
| E(inclDel,"DELETE FROM incl WHERE file=?") \ |
| E(inclHas,"SELECT 1 FROM incl WHERE file=?") \ |
| E(inclPathAdd,"INSERT OR FAIL INTO " \ |
| "inclpath(seq,dir) VALUES(?,?)") \ |
| E(inclSearch, \ |
| "SELECT ?1 fn WHERE fileExists(fn) " \ |
| "UNION ALL SELECT * FROM (" \ |
| "SELECT replace(dir||'/'||?1, '//','/') AS fn " \ |
| "FROM inclpath WHERE fileExists(fn) ORDER BY seq"\ |
| ")") \ |
| E(spBegin,"SAVEPOINT " CMPP_SAVEPOINT_NAME) \ |
| E(spRollback,"ROLLBACK TO SAVEPOINT " \ |
| CMPP_SAVEPOINT_NAME) \ |
| E(spRelease,"RELEASE SAVEPOINT " CMPP_SAVEPOINT_NAME) |
|
|
| #define E(N,S) sqlite3_stmt * N; |
| GStmt_map(E) |
| #undef E |
| } stmt; |
| struct { |
| FILE * pFile; |
| int expandSql; |
| } sqlTrace; |
| struct { |
| AtPolicy atPolicy; |
| |
| char doDebug; |
| |
| char chompF; |
| } flags; |
| } g = { |
| .zArgv0 = "?", |
| .db = 0, |
| .tok = 0, |
| .pFiles = 0, |
| .out = FileWrapper_empty_m, |
| .delim = { |
| .z = CMPP_DEFAULT_DELIM, |
| .n = (unsigned short) sizeof(CMPP_DEFAULT_DELIM)-1, |
| .chAt = '@' |
| }, |
| .stmt = { |
| .defIns = 0, |
| .defDel = 0, |
| .defHas = 0, |
| .defGet = 0, |
| .defGetBool = 0, |
| .inclIns = 0, |
| .inclDel = 0, |
| .inclHas = 0, |
| .inclPathAdd = 0, |
| .inclSearch = 0 |
| }, |
| .sqlTrace = { |
| .pFile = 0, |
| .expandSql = 0 |
| }, |
| .flags = { |
| .atPolicy = AT_OFF, |
| .doDebug = 0, |
| .chompF = 0 |
| } |
| }; |
|
|
| |
| enum GStmt_e { |
| GStmt_none = 0, |
| #define E(N,S) GStmt_ ## N, |
| GStmt_map(E) |
| #undef E |
| }; |
|
|
| |
| |
| |
| |
| static sqlite3_stmt * g_stmt(enum GStmt_e which){ |
| sqlite3_stmt ** q = 0; |
| char const * zSql = 0; |
| switch(which){ |
| case GStmt_none: |
| fatal("GStmt_none is not a valid statement handle"); |
| return NULL; |
| #define E(N,S) case GStmt_ ## N: zSql = S; q = &g.stmt.N; break; |
| GStmt_map(E) |
| #undef E |
| } |
| assert( q ); |
| assert( zSql && *zSql ); |
| if( !*q ){ |
| db_prepare(q, "%s", zSql); |
| assert( *q ); |
| } |
| return *q; |
| } |
| static void g_stmt_reset(sqlite3_stmt * const q){ |
| sqlite3_clear_bindings(q); |
| sqlite3_reset(q); |
| } |
|
|
| #if 0 |
| |
| |
| |
| |
| static void g_outf(char const *zFmt, ...); |
| void g_outf(char const *zFmt, ...){ |
| va_list va; |
| va_start(va, zFmt); |
| vfprintf(g.out.pFile, zFmt, va); |
| va_end(va); |
| } |
| #endif |
|
|
| |
| static void g_out(void const *z, unsigned int n); |
| void g_out(void const *z, unsigned int n){ |
| if(g.out.pFile && 1!=fwrite(z, n, 1, g.out.pFile)){ |
| int const err = errno; |
| fatal("fwrite() output failed with errno #%d", err); |
| } |
| } |
|
|
| void g_stderrv(char const *zFmt, va_list va){ |
| if( g.out.pFile==stdout ){ |
| fflush(g.out.pFile); |
| } |
| vfprintf(stderr, zFmt, va); |
| } |
|
|
| void g_stderr(char const *zFmt, ...){ |
| va_list va; |
| va_start(va, zFmt); |
| g_stderrv(zFmt, va); |
| va_end(va); |
| } |
|
|
| |
| |
| |
| void cmpp_t_out(CmppTokenizer * t, void const *z, unsigned int n){ |
| g_debug(3,("CT_skipLevel() ?= %d\n",CT_skipLevel(t))); |
| g_debug(3,("CT_skip() ?= %d\n",CT_skip(t))); |
| if(!CT_skip(t)) g_out(z, n); |
| } |
|
|
| void CmppLevel_push(CmppTokenizer * const t){ |
| CmppLevel * pPrev; |
| CmppLevel * p; |
| if(t->level.ndx+1 == (unsigned)CmppLevel_Max){ |
| fatal("%sif nesting level is too deep. Max=%d\n", |
| g.delim.z, CmppLevel_Max); |
| } |
| pPrev = &CT_level(t); |
| g_debug(3,("push from tokenizer level=%u flags=%04x\n", |
| t->level.ndx, pPrev->flags)); |
| p = &t->level.stack[++t->level.ndx]; |
| *p = CmppLevel_empty; |
| p->token = t->token; |
| p->flags = (CmppLevel_F_INHERIT_MASK & pPrev->flags); |
| if(CLvl_skip(pPrev)) p->flags |= CmppLevel_F_ELIDE; |
| g_debug(3,("push to tokenizer level=%u flags=%04x\n", |
| t->level.ndx, p->flags)); |
| } |
|
|
| void CmppLevel_pop(CmppTokenizer * const t){ |
| if(!t->level.ndx){ |
| fatal("Internal error: CmppLevel_pop() at the top of the stack"); |
| } |
| g_debug(3,("pop from tokenizer level=%u, flags=%04x skipLevel?=%d\n", |
| t->level.ndx, |
| t->level.stack[t->level.ndx].flags, CT_skipLevel(t))); |
| g_debug(3,("CT_skipLevel() ?= %d\n",CT_skipLevel(t))); |
| g_debug(3,("CT_skip() ?= %d\n",CT_skip(t))); |
| t->level.stack[t->level.ndx--] = CmppLevel_empty; |
| g_debug(3,("pop to tokenizer level=%u, flags=%04x\n", t->level.ndx, |
| t->level.stack[t->level.ndx].flags)); |
| g_debug(3,("CT_skipLevel() ?= %d\n",CT_skipLevel(t))); |
| g_debug(3,("CT_skip() ?= %d\n",CT_skip(t))); |
| } |
|
|
| CmppLevel * CmppLevel_get(CmppTokenizer * const t){ |
| return &t->level.stack[t->level.ndx]; |
| } |
|
|
|
|
| void db_affirm_rc(int rc, const char * zMsg){ |
| switch(rc){ |
| case 0: |
| case SQLITE_DONE: |
| case SQLITE_ROW: |
| break; |
| default: |
| assert( g.db ); |
| fatal("Db error #%d %s: %s", rc, zMsg, |
| sqlite3_errmsg(g.db)); |
| } |
| } |
|
|
| int db_step(sqlite3_stmt *pStmt){ |
| int const rc = sqlite3_step(pStmt); |
| switch( rc ){ |
| case SQLITE_ROW: |
| case SQLITE_DONE: |
| break; |
| default: |
| db_affirm_rc(rc, "from db_step()"); |
| } |
| return rc; |
| } |
|
|
| static sqlite3_str * db_str_new(void){ |
| sqlite3_str * rc = sqlite3_str_new(g.db); |
| if(!rc) fatal("Alloc failed for sqlite3_str_new()"); |
| return rc; |
| } |
|
|
| static char * db_str_finish(sqlite3_str *s, int * n){ |
| int const rc = sqlite3_str_errcode(s); |
| if(rc) fatal("Error #%d from sqlite3_str_errcode()", rc); |
| if(n) *n = sqlite3_str_length(s); |
| char * z = sqlite3_str_finish(s); |
| if(!z) fatal("Alloc failed for sqlite3_str_new()"); |
| return z; |
| } |
|
|
| void db_prepare(sqlite3_stmt **pStmt, const char * zSql, ...){ |
| int rc; |
| sqlite3_str * str = db_str_new(); |
| char * z = 0; |
| int n = 0; |
| va_list va; |
| va_start(va, zSql); |
| sqlite3_str_vappendf(str, zSql, va); |
| va_end(va); |
| rc = sqlite3_str_errcode(str); |
| if(rc) fatal("sqlite3_str_errcode() = %d", rc); |
| z = db_str_finish(str, &n); |
| rc = sqlite3_prepare_v2(g.db, z, n, pStmt, 0); |
| if(rc) fatal("Error #%d (%s) preparing: %s", |
| rc, sqlite3_errmsg(g.db), z); |
| sqlite3_free(z); |
| } |
|
|
| void db_bind_int(sqlite3_stmt *pStmt, int col, int val){ |
| db_affirm_rc(sqlite3_bind_int(pStmt, col, val), |
| "from db_bind_int()"); |
| } |
|
|
| void db_bind_null(sqlite3_stmt *pStmt, int col){ |
| db_affirm_rc(sqlite3_bind_null(pStmt, col), |
| "from db_bind_null()"); |
| } |
|
|
| void db_bind_textn(sqlite3_stmt *pStmt, int col, |
| const char * zStr, int n){ |
| db_affirm_rc( |
| (zStr && n) |
| ? sqlite3_bind_text(pStmt, col, zStr, n, SQLITE_TRANSIENT) |
| : sqlite3_bind_null(pStmt, col), |
| "from db_bind_textn()" |
| ); |
| } |
|
|
| void db_bind_text(sqlite3_stmt *pStmt, int col, |
| const char * zStr){ |
| db_bind_textn(pStmt, col, zStr, -1); |
| } |
|
|
| #if 0 |
| void db_bind_textv(sqlite3_stmt *pStmt, int col, |
| const char * zFmt, ...){ |
| int rc; |
| sqlite3_str * str = db_str_new(); |
| int n = 0; |
| char * z; |
| va_list va; |
| va_start(va,zFmt); |
| sqlite3_str_vappendf(str, zFmt, va); |
| va_end(va); |
| z = db_str_finish(str, &n); |
| rc = sqlite3_bind_text(pStmt, col, z, n, sqlite3_free); |
| db_affirm_rc(rc,"from db_bind_textv()"); |
| } |
| #endif |
|
|
| void db_free(void *m){ |
| sqlite3_free(m); |
| } |
|
|
| void db_define_add(const char * zKey, char const *zVal){ |
| cmpp_kvp kvp = cmpp_kvp_empty; |
| cmpp_kvp_parse(&kvp, zKey, -1, |
| zVal |
| ? cmpp_key_op_none |
| : cmpp_key_op_eq1 |
| ); |
| if( kvp.v.z ){ |
| if( zVal ){ |
| assert(!"cannot happen - cmpp_key_op_none will prevent it"); |
| fatal("Cannot assign two values to [%.*s] [%.*s] [%s]", |
| kvp.k.n, kvp.k.z, kvp.v.n, kvp.v.z, zVal); |
| } |
| }else{ |
| kvp.v.z = zVal; |
| kvp.v.n = zVal ? (int)strlen(zVal) : 0; |
| } |
| sqlite3_stmt * const q = g_stmt(GStmt_defIns); |
| |
| db_bind_textn(q, 1, kvp.k.z, kvp.k.n); |
| if( kvp.v.z ){ |
| if( kvp.v.n ){ |
| db_bind_textn(q, 2, kvp.v.z, (int)kvp.v.n); |
| }else{ |
| db_bind_null(q, 2); |
| } |
| }else{ |
| db_bind_int(q, 2, 1); |
| } |
| db_step(q); |
| g_debug(2,("define: %s%s%s\n", |
| zKey, |
| zVal ? " with value " : "", |
| zVal ? zVal : "")); |
| sqlite3_reset(q); |
| } |
|
|
| static void db_define_add_file(const char * zKey){ |
| cmpp_kvp kvp = cmpp_kvp_empty; |
| cmpp_kvp_parse(&kvp, zKey, -1, cmpp_kvp_op_eq1); |
| if( !kvp.v.z || !kvp.v.n ){ |
| fatal("Invalid filename: %s", zKey); |
| } |
| sqlite3_stmt * q = 0; |
| FileWrapper fw = FileWrapper_empty; |
| FileWrapper_open(&fw, kvp.v.z, "r"); |
| FileWrapper_slurp(&fw); |
| q = g_stmt(GStmt_defIns); |
| |
| db_bind_textn(q, 1, kvp.k.z, (int)kvp.k.n); |
| if( g.flags.chompF ){ |
| FileWrapper_chomp(&fw); |
| } |
| if( fw.nContent ){ |
| db_affirm_rc( |
| sqlite3_bind_text(q, 2, |
| (char const *)fw.zContent, |
| (int)fw.nContent, sqlite3_free), |
| "binding file content"); |
| fw.zContent = 0 ; |
| fw.nContent = 0; |
| }else{ |
| db_affirm_rc( sqlite3_bind_null(q, 2), |
| "binding empty file content"); |
| } |
| FileWrapper_close(&fw); |
| db_step(q); |
| g_stmt_reset(q); |
| g_debug(2,("define: %s%s%s\n", |
| kvp.k.z, |
| kvp.v.z ? " with value " : "", |
| kvp.v.z ? kvp.v.z : "")); |
| } |
|
|
| #define ustr_c(X) ((unsigned char const *)X) |
|
|
| static inline unsigned int cmpp_strlen(char const *z, int n){ |
| return n<0 ? (int)strlen(z) : (unsigned)n; |
| } |
|
|
|
|
| int db_define_has(const char * zName, int nName){ |
| int rc; |
| sqlite3_stmt * const q = g_stmt(GStmt_defHas); |
| nName = cmpp_strlen(zName, nName); |
| db_bind_textn(q, 1, zName, nName); |
| rc = db_step(q); |
| if(SQLITE_ROW == rc){ |
| rc = 1; |
| }else{ |
| assert(SQLITE_DONE==rc); |
| rc = 0; |
| } |
| g_debug(1,("defined [%s] ?= %d\n",zName, rc)); |
| g_stmt_reset(q); |
| return rc; |
| } |
|
|
| int db_define_get_bool(const char * zName, int nName){ |
| sqlite3_stmt * const q = g_stmt(GStmt_defGetBool); |
| int rc = 0; |
| nName = cmpp_strlen(zName, nName); |
| db_bind_textn(q, 1, zName, nName); |
| rc = db_step(q); |
| if(SQLITE_ROW == rc){ |
| if( SQLITE_ROW==sqlite3_step(q) ){ |
| fatal("Key is ambiguous: %s", zName); |
| } |
| rc = 1; |
| }else{ |
| assert(SQLITE_DONE==rc); |
| rc = 0; |
| } |
| g_stmt_reset(q); |
| return rc; |
| } |
|
|
| int db_define_get(const char * zName, int nName, |
| char **zVal, unsigned int *nVal){ |
| sqlite3_stmt * q = g_stmt(GStmt_defGet); |
| nName = cmpp_strlen(zName, nName); |
| db_bind_textn(q, 1, zName, nName); |
| int n = 0; |
| int rc = db_step(q); |
| if(SQLITE_ROW == rc){ |
| const unsigned char * z = sqlite3_column_text(q, 1); |
| n = sqlite3_column_bytes(q,1); |
| if( nVal ) *nVal = (unsigned)n; |
| *zVal = sqlite3_mprintf("%.*s", n, z); |
| if( n && z ) check__oom(*zVal); |
| if( SQLITE_ROW==sqlite3_step(q) ){ |
| db_free(*zVal); |
| *zVal = 0; |
| fatal("Key is ambiguous: %.*s\n", |
| nName, zName); |
| } |
| rc = 1; |
| }else{ |
| assert(SQLITE_DONE==rc); |
| rc = 0; |
| } |
| g_debug(1,("define [%.*s] ?= %d %.*s\n", |
| nName, zName, rc, |
| *zVal ? n : 0, |
| *zVal ? *zVal : "<NULL>")); |
| g_stmt_reset(q); |
| return rc; |
| } |
|
|
| void db_define_rm(const char * zKey){ |
| int rc; |
| int n = 0; |
| sqlite3_stmt * const q = g_stmt(GStmt_defDel); |
| db_bind_text(q, 1, zKey); |
| rc = db_step(q); |
| if(SQLITE_DONE != rc){ |
| db_affirm_rc(rc, "Stepping DELETE on def"); |
| } |
| g_debug(2,("undefine: %.*s\n",n, zKey)); |
| g_stmt_reset(q); |
| } |
|
|
| void db_including_add(const char * zKey, const char * zSrc, int srcLine){ |
| int rc; |
| sqlite3_stmt * const q = g_stmt(GStmt_inclIns); |
| db_bind_text(q, 1, zKey); |
| db_bind_text(q, 2, zSrc); |
| db_bind_int(q, 3, srcLine); |
| rc = db_step(q); |
| if(SQLITE_DONE != rc){ |
| db_affirm_rc(rc, "Stepping INSERT on incl"); |
| } |
| g_debug(2,("is-including-file add [%s] from [%s]:%d\n", zKey, zSrc, srcLine)); |
| g_stmt_reset(q); |
| } |
|
|
| void db_include_rm(const char * zKey){ |
| int rc; |
| sqlite3_stmt * const q = g_stmt(GStmt_inclDel); |
| db_bind_text(q, 1, zKey); |
| rc = db_step(q); |
| if(SQLITE_DONE != rc){ |
| db_affirm_rc(rc, "Stepping DELETE on incl"); |
| } |
| g_debug(2,("inclpath rm [%s]\n", zKey)); |
| g_stmt_reset(q); |
| } |
|
|
| char * db_include_search(const char * zKey){ |
| char * zName = 0; |
| sqlite3_stmt * const q = g_stmt(GStmt_inclSearch); |
| db_bind_text(q, 1, zKey); |
| if(SQLITE_ROW==db_step(q)){ |
| const unsigned char * z = sqlite3_column_text(q, 0); |
| zName = z ? sqlite3_mprintf("%s", z) : 0; |
| if(!zName) fatal("Alloc failed"); |
| } |
| g_stmt_reset(q); |
| return zName; |
| } |
|
|
| static int db_including_has(const char * zName){ |
| int rc; |
| sqlite3_stmt * const q = g_stmt(GStmt_inclHas); |
| db_bind_text(q, 1, zName); |
| rc = db_step(q); |
| if(SQLITE_ROW == rc){ |
| rc = 1; |
| }else{ |
| assert(SQLITE_DONE==rc); |
| rc = 0; |
| } |
| g_debug(2,("inclpath has [%s] = %d\n",zName, rc)); |
| g_stmt_reset(q); |
| return rc; |
| } |
|
|
| #if 0 |
| |
| |
| |
| static void db_including_check(const char * zKey); |
| void db_including_check(const char * zName){ |
| if(db_including_has(zName)){ |
| fatal("Recursive include detected: %s\n", zName); |
| } |
| } |
| #endif |
|
|
| void db_include_dir_add(const char * zDir){ |
| static int seq = 0; |
| int rc; |
| sqlite3_stmt * const q = g_stmt(GStmt_inclPathAdd); |
| db_bind_int(q, 1, ++seq); |
| db_bind_text(q, 2, zDir); |
| rc = db_step(q); |
| if(SQLITE_DONE != rc){ |
| db_affirm_rc(rc, "Stepping INSERT on inclpath"); |
| } |
| g_debug(2,("inclpath add #%d: %s\n",seq, zDir)); |
| g_stmt_reset(q); |
| } |
|
|
| void g_FileWrapper_link(FileWrapper *fp){ |
| assert(!fp->pTail); |
| fp->pTail = g.pFiles; |
| g.pFiles = fp; |
| } |
|
|
| void g_FileWrapper_close(FileWrapper *fp){ |
| assert(fp); |
| assert(fp->pTail || g.pFiles==fp); |
| g.pFiles = fp->pTail; |
| fp->pTail = 0; |
| FileWrapper_close(fp); |
| } |
|
|
| static void g_cleanup(int bCloseFileChain){ |
| if( g.db ){ |
| #define E(N,S) sqlite3_finalize(g.stmt.N); g.stmt.N = 0; |
| GStmt_map(E) |
| #undef E |
| } |
| if( bCloseFileChain ){ |
| FileWrapper * fpNext = 0; |
| for( FileWrapper * fp=g.pFiles; fp; fp=fpNext ){ |
| fpNext = fp->pTail; |
| fp->pTail = 0; |
| FileWrapper_close(fp); |
| } |
| } |
| FileWrapper_close(&g.out); |
| if(g.db){ |
| sqlite3_close(g.db); |
| g.db = 0; |
| } |
| } |
|
|
| static void cmpp_atexit(void){ |
| g_cleanup(1); |
| } |
|
|
| int cmpp_is_legal_key(char const *zKey, int nKey, char const **zAt){ |
| char const * z = zKey; |
| nKey = cmpp_strlen(zKey, nKey); |
| if( !nKey ){ |
| if( zAt ) *zAt = z; |
| return 0; |
| } |
| char const * const zEnd = z ? z + nKey : NULL; |
| for( ; z < zEnd; ++z ){ |
| switch( (0x80 & *z) ? 0 : *z ){ |
| case 0: |
| case '_': |
| continue; |
| case '-': |
| case '.': |
| case '/': |
| case ':': |
| case '=': |
| case '0': case '1': case '2': case '3': case '4': |
| case '5': case '6': case '7': case '8': case '9': |
| if( z==zKey ) break; |
| continue; |
| default: |
| if( isalpha((int)*z) ) continue; |
| } |
| if( zAt ) *zAt = z; |
| return 0; |
| } |
| assert( z==zEnd ); |
| return 1; |
| } |
|
|
| void cmpp_affirm_legal_key(char const *zKey, int nKey){ |
| char const *zAt = 0; |
| nKey = cmpp_strlen(zKey, nKey); |
| if( !cmpp_is_legal_key(zKey, nKey, &zAt) ){ |
| assert( zAt ); |
| fatal("Illegal character 0x%02x in key [%.*s]\n", |
| (int)*zAt, nKey, zKey); |
| } |
| } |
|
|
| |
| |
| |
| |
| static void udf_file_exists( |
| sqlite3_context *context, |
| int argc, |
| sqlite3_value **argv |
| ){ |
| const char *zName; |
| (void)(argc); |
| zName = (const char*)sqlite3_value_text(argv[0]); |
| if( zName==0 ) return; |
| sqlite3_result_int(context, 0==access(zName, 0)); |
| } |
|
|
| |
| |
| |
| |
| static int cmpp__db_sq3TraceV2(unsigned t,void*c,void*p,void*x){ |
| static unsigned int counter = 0; |
| switch(t){ |
| case SQLITE_TRACE_STMT:{ |
| FILE * const fp = g.sqlTrace.pFile; |
| if( fp ){ |
| char const * const zSql = (char const *)x; |
| char * const zExp = g.sqlTrace.expandSql |
| ? sqlite3_expanded_sql((sqlite3_stmt*)p) |
| : 0; |
| fprintf(fp, "SQL TRACE #%u: %s\n", |
| ++counter, zExp ? zExp : zSql); |
| sqlite3_free(zExp); |
| } |
| break; |
| } |
| } |
| return 0; |
| } |
|
|
| |
| static void cmpp_initdb(void){ |
| int rc; |
| char * zErr = 0; |
| const char * zSchema = |
| "CREATE TABLE def(" |
| |
| "k TEXT PRIMARY KEY NOT NULL," |
| "v TEXT DEFAULT NULL" |
| ") WITHOUT ROWID;" |
| "CREATE TABLE incl(" |
| |
| "file TEXT PRIMARY KEY NOT NULL," |
| "srcFile TEXT DEFAULT NULL," |
| "srcLine INTEGER DEFAULT 0" |
| ") WITHOUT ROWID;" |
| "CREATE TABLE inclpath(" |
| |
| "seq INTEGER UNIQUE ON CONFLICT IGNORE, " |
| "dir TEXT PRIMARY KEY NOT NULL ON CONFLICT IGNORE" |
| ");" |
| "BEGIN;" |
| ; |
| assert(0==g.db); |
| if(g.db) return; |
| rc = sqlite3_open_v2(":memory:", &g.db, SQLITE_OPEN_READWRITE, 0); |
| if(rc) fatal("Error opening :memory: db."); |
| sqlite3_trace_v2(g.db, SQLITE_TRACE_STMT, cmpp__db_sq3TraceV2, 0); |
| rc = sqlite3_exec(g.db, zSchema, 0, 0, &zErr); |
| if(rc) fatal("Error initializing database: %s", zErr); |
| rc = sqlite3_create_function(g.db, "fileExists", 1, |
| SQLITE_UTF8|SQLITE_DIRECTONLY, 0, |
| udf_file_exists, 0, 0); |
| db_affirm_rc(rc, "UDF registration failed."); |
| } |
|
|
| |
| |
| |
| |
| |
| |
| unsigned short cmpp_is_delim(unsigned char const *zBegin, |
| unsigned char const *zEnd, |
| unsigned char const *zPos){ |
| assert(zEnd>zBegin); |
| assert(zPos<zEnd); |
| assert(zPos>=zBegin); |
| if(zPos>zBegin && |
| ('\n'!=*(zPos - 1) |
| || ((unsigned)(zEnd - zPos) <= g.delim.n))){ |
| return 0; |
| }else if(0==memcmp(zPos, g.delim.z, g.delim.n)){ |
| return g.delim.n; |
| }else{ |
| return 0; |
| } |
| } |
|
|
| static void cmpp_t_out_expand(CmppTokenizer * const t, |
| unsigned char const * zFrom, |
| unsigned int n); |
|
|
| static inline int cmpp__isspace(int ch){ |
| return ' '==ch || '\t'==ch; |
| } |
|
|
| static inline unsigned cmpp__strlenu(unsigned char const *z, int n){ |
| return n<0 ? (unsigned)strlen((char const *)z) : (unsigned)n; |
| } |
|
|
| static inline void cmpp__skip_space_c( unsigned char const **p, |
| unsigned char const *zEnd ){ |
| unsigned char const * z = *p; |
| while( z<zEnd && cmpp__isspace(*z) ) ++z; |
| *p = z; |
| } |
|
|
| #define ustr_c(X) ((unsigned char const *)X) |
| |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| static |
| int CmppTokenizer__delim_search(CmppTokenizer * const t){ |
| if(!t->zPos) t->zPos = t->zBegin; |
| if( t->zPos>=t->zEnd ){ |
| return 0; |
| } |
| assert( (t->zPos==t->zBegin || t->zPos[-1]=='\n') |
| && "Else we've mismanaged something."); |
| char const * const zD = g.delim.z; |
| unsigned short const nD = g.delim.n; |
| unsigned char const * const zEnd = t->zEnd; |
| unsigned char const * zLeft = t->zPos; |
| unsigned char const * z = zLeft; |
|
|
| assert( 0==*zEnd && "Else we'll misinteract with strcspn()" ); |
| if( *zEnd ){ |
| fatal("Input must be NUL-terminated."); |
| return 0; |
| } |
| #define tflush \ |
| if(z>zEnd) z=zEnd; \ |
| if( z>zLeft ) { \ |
| cmpp_t_out_expand(t, zLeft, (unsigned)(z-zLeft)); \ |
| } zLeft = z |
| while(z < zEnd){ |
| size_t nNlTotal = 0; |
| unsigned char const * zNl; |
| size_t nNl2 = strcspn((char const *)z, "\n"); |
| zNl = (z + nNl2 >= zEnd ? zEnd : z + nNl2); |
| if( nNl2 >= CmppArgs_BufSize |
| |
| |
| |
| ){ |
| |
| |
| ++t->lineNo; |
| z = zNl + 1; |
| tflush; |
| continue; |
| } |
| nNlTotal += nNl2; |
| assert( '\n'==*zNl || !*zNl ); |
| assert( '\n'==*zNl || zNl==zEnd ); |
| |
| unsigned char const * const zBOL = z; |
| cmpp__skip_space_c(&z, zNl); |
| if( z+nD < zNl && 0==memcmp(z, zD, nD) ){ |
| |
| if( zBOL!=z ){ |
| |
| |
| zLeft = z; |
| } |
| while( zNl>z && zNl<zEnd |
| && '\n'==*zNl && '\\'==zNl[-1] ){ |
| |
| |
| ++t->lineNo; |
| ++zNl; |
| nNl2 = strcspn((char const *)zNl, "\n"); |
| if( !nNl2 ) break; |
| nNlTotal += nNl2; |
| zNl += nNl2; |
| } |
| assert( zNl<=zEnd && "Else our input was not NUL-terminated"); |
| if( nNlTotal >= CmppArgs_BufSize ){ |
| fatal("Directive line is too long (%u)", |
| (unsigned)(zNl-z)); |
| break; |
| } |
| tflush; |
| t->token.zBegin = z + nD; |
| t->token.zEnd = zNl; |
| cmpp__skip_space_c(&t->token.zBegin, t->token.zEnd); |
| t->token.ttype = TT_Line; |
| t->token.lineNo = t->lineNo++; |
| t->zPos = t->token.zEnd + 1; |
| if( 0 ){ |
| g_stderr("token=<<%.*s>>", (t->token.zEnd - t->token.zBegin), |
| t->token.zBegin); |
| } |
| return 1; |
| } |
| z = zNl+1; |
| ++t->lineNo; |
| tflush; |
| |
| } |
| tflush; |
| t->zPos = z; |
| return 0; |
| #undef tflush |
| } |
|
|
| void cmpp_kvp_parse(cmpp_kvp * p, char const *zKey, int nKey, |
| cmpp_kvp_op_e opPolicy){ |
| char chEq = 0; |
| char opLen = 0; |
| *p = cmpp_kvp_empty; |
| p->k.z = zKey; |
| p->k.n = cmpp_strlen(zKey, nKey); |
| switch( opPolicy ){ |
| case cmpp_kvp_op_none: break; |
| case cmpp_kvp_op_eq1: |
| chEq = '='; |
| opLen = 1; |
| break; |
| default: |
| assert(!"don't use these yet"); |
| |
| chEq = '='; |
| opLen = 1; |
| break; |
| } |
| assert( chEq ); |
| p->op = cmpp_kvp_op_none; |
| const char * const zEnd = p->k.z + p->k.n; |
| for(const char * zPos = p->k.z ; *zPos && zPos<zEnd ; ++zPos) { |
| if( chEq==*zPos ){ |
| if( cmpp_kvp_op_none==opPolicy ){ |
| fatal("Illegal operator in key: %s", zKey); |
| } |
| p->op = cmpp_kvp_op_eq1; |
| p->k.n = (unsigned)(zPos - zKey); |
| zPos += opLen; |
| assert( zPos <= zEnd ); |
| p->v.z = zPos; |
| p->v.n = (unsigned)(zEnd - zPos); |
| break; |
| } |
| } |
| cmpp_affirm_legal_key(p->k.z, p->k.n); |
| } |
|
|
| static void cmpp_t_out_expand(CmppTokenizer * const t, |
| unsigned char const * zFrom, |
| unsigned int n){ |
| unsigned char const *zLeft = zFrom; |
| unsigned char const * const zEnd = zFrom + n; |
| unsigned char const *z = AT_OFF==g.flags.atPolicy ? zEnd : zLeft; |
| unsigned char const chEol = (unsigned char)'\n'; |
| int state = 0 |
| ; |
| if( 0 ){ |
| g_warn("zLeft=%d %c", (int)*zLeft, *zLeft); |
| } |
| #define tflush \ |
| if(z>zEnd) z=zEnd; \ |
| if(zLeft<z){ cmpp_t_out(t, zLeft, (z-zLeft)); } zLeft = z |
| for( ; z<zEnd; ++z ){ |
| zLeft = z; |
| for( ;z<zEnd; ++z ){ |
| if( chEol==*z ){ |
| state = 0; |
| continue; |
| } |
| if( g.delim.chAt==*z ){ |
| if( 0==state ){ |
| tflush; |
| state = 1; |
| }else{ |
| char *zVal = 0; |
| unsigned int nVal = 0; |
| assert( 1==state ); |
| assert( g.delim.chAt==*zLeft ); |
| assert( zLeft<z ); |
| if( z==zLeft+1 ){ |
| tflush; |
| }else{ |
| char const *zKey = (char const*)zLeft+1; |
| int const nKey = (z-zLeft-1); |
| if( db_define_get(zKey, nKey, &zVal, &nVal) ){ |
| if( nVal ){ |
| cmpp_t_out(t, (unsigned char const*)zVal, nVal); |
| }else{ |
| |
| } |
| zLeft = z+1; |
| db_free(zVal); |
| }else{ |
| assert( !zVal ); |
| |
| switch( g.flags.atPolicy ){ |
| case AT_RETAIN: |
| tflush; |
| break; |
| case AT_ERROR: |
| fatal("Undefined key: %c%.*s%c", |
| g.delim.chAt, nKey, zKey, g.delim.chAt ); |
| break; |
| case AT_ELIDE: |
| zLeft = z+1; |
| break; |
| case AT_invalid: |
| case AT_OFF: |
| fatal("Unhandled g.flags.atPolicy #%d!", |
| g.flags.atPolicy); |
| break; |
| } |
| } |
| state = 0; |
| } |
| } |
| } |
| } |
| } |
| tflush; |
| #undef tflush |
| return; |
| } |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| static int cmpp_next_keyword_line(CmppTokenizer * const t){ |
| CmppToken * const tok = &t->token; |
|
|
| assert(t->zBegin); |
| assert(t->zEnd > t->zBegin); |
| if(!t->zPos) t->zPos = t->zBegin; |
| t->args.pKw = 0; |
| t->args.argc = 0; |
| *tok = CmppToken_empty; |
| if( !CmppTokenizer__delim_search(t) ){ |
| return 0; |
| } |
| |
| int i, argc = 0, prevChar = 0; |
| const unsigned tokLen = (unsigned)(tok->zEnd - tok->zBegin); |
| unsigned char * zKwd; |
| unsigned char * zEsc; |
| unsigned char * zz; |
|
|
| assert(TT_Line==tok->ttype); |
| g_debug(2,("token @ line %u len=%u [[[%.*s]]]\n", |
| tok->lineNo, tokLen, tokLen, tok->zBegin)); |
| zKwd = &t->args.lineBuf[0]; |
| memcpy(zKwd, tok->zBegin, tokLen); |
| memset(zKwd + tokLen, 0, sizeof(t->args.lineBuf) - tokLen); |
| for( zEsc = 0, zz = zKwd; *zz; ++zz ){ |
| |
| switch((int)*zz){ |
| case (int)'\\': |
| if(zEsc) zEsc = 0; |
| else zEsc = zz; |
| break; |
| case (int)'\n': |
| assert(zEsc && "Should not have an unescaped newline?"); |
| if(zEsc==zz-1){ |
| *zEsc = (unsigned char)' '; |
| |
| |
| |
| |
| |
| } |
| zEsc = 0; |
| *zz = (unsigned char)' '; |
| break; |
| default: |
| zEsc = 0; |
| break; |
| } |
| } |
| t->args.argv[argc++] = zKwd; |
| for( zz = zKwd; *zz; ++zz ){ |
| if(isspace(*zz)){ |
| *zz = 0; |
| break; |
| } |
| } |
| t->args.pKw = CmppKeyword_search((char const *)zKwd); |
| if(!t->args.pKw){ |
| fatal("Unknown keyword '%s' at line %u\n", (char const *)zKwd, |
| tok->lineNo); |
| } |
| for( ++zz ; *zz && isspace(*zz); ++zz ){} |
| if(t->args.pKw->bTokenize){ |
| for( ; *zz; prevChar = *zz, ++zz ){ |
| |
| |
| |
| if(isspace(*zz)){ |
| assert(zz!=zKwd && "Leading space was stripped earlier."); |
| *zz = 0; |
| }else{ |
| if(argc == (int)CmppArgs_Max){ |
| fatal("Too many arguments @ line %u: %.*s", |
| tok->lineNo, tokLen, tok->zBegin); |
| }else if(zz>zKwd && !prevChar){ |
| t->args.argv[argc++] = zz; |
| } |
| } |
| } |
| }else{ |
| |
| if(*zz) t->args.argv[argc++] = zz; |
| } |
| tok->ttype = t->args.pKw->ttype; |
| if(g.flags.doDebug>1){ |
| for(i = 0; i < argc; ++i){ |
| g_debug(0,("line %u arg #%d=%s\n", |
| tok->lineNo, i, |
| (char const *)t->args.argv[i])); |
| } |
| } |
| t->args.argc = argc; |
| return 1; |
| } |
|
|
| |
| static CMPP_NORETURN void cmpp_kwd__err_(char const *zFile, int line, |
| CmppKeyword const * pKw, |
| CmppTokenizer const *t, |
| char const *zFmt, ...){ |
| va_list va; |
| g_stderr("%s @ %s line %u:", |
| pKw->zName, t->zName, t->token.lineNo); |
| va_start(va, zFmt); |
| g.tok = 0 ; |
| fatalv__base(zFile, line, zFmt, va); |
| |
| va_end(va); |
| } |
| #define cmpp_kwd__err(...) cmpp_kwd__err_(__FILE__,__LINE__, __VA_ARGS__) |
| #define cmpp_t__err(T,...) cmpp_kwd__err_(__FILE__,__LINE__, (T)->args.pKw, (T), __VA_ARGS__) |
|
|
| |
| static void cmpp_kwd_noop(CmppKeyword const * pKw, CmppTokenizer *t){ |
| (void)pKw; |
| (void)t; |
| } |
|
|
| |
| static void cmpp_kwd_error(CmppKeyword const * pKw, CmppTokenizer *t){ |
| if(CT_skip(t)) return; |
| else{ |
| assert(t->args.argc < 3); |
| const char *zBegin = t->args.argc>1 |
| ? (const char *)t->args.argv[1] : 0; |
| cmpp_t__err(t, "%s", zBegin ? zBegin : "(no additional info)"); |
| } |
| } |
|
|
| |
| static void cmpp_kwd_define(CmppKeyword const * pKw, CmppTokenizer *t){ |
| if(CT_skip(t)) return; |
| if(t->args.argc<2){ |
| cmpp_kwd__err(pKw, t, "Expecting one or more arguments"); |
| }else{ |
| int i = 1; |
| for( ; i < t->args.argc; ++i){ |
| char const * const zArg = (char const *)t->args.argv[i]; |
| cmpp_affirm_legal_key(zArg, -1); |
| if( TT_Define==pKw->ttype ){ |
| db_define_add( zArg, NULL ); |
| }else{ |
| db_define_rm( zArg ); |
| } |
| } |
| } |
| } |
|
|
| static int cmpp_val_matches(char const *zGlob, char const *zRhs){ |
| return 0==sqlite3_strglob(zGlob, zRhs); |
| } |
|
|
| typedef int (*cmpp_vcmp_f)(char const *zLhs, char const *zRhs); |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| int cmpp_kvp_truth(CmppKeyword const * const pKw, |
| CmppTokenizer const * const t, |
| cmpp_kvp const * const kvp, |
| int bCheckDefined){ |
| int buul = 0; |
| if( kvp->v.z ){ |
| if( bCheckDefined ){ |
| cmpp_kwd__err(pKw, t, "Value part is not legal for " |
| "is-defined checks: %.s", |
| kvp->k.n, kvp->k.z); |
| } |
| char * zVal = 0; |
| unsigned int nVal = 0; |
| buul = db_define_get(kvp->k.z, (int)kvp->k.n, &zVal, &nVal); |
| |
| if( kvp->v.n && nVal ){ |
| |
| |
| |
| |
| buul = cmpp_val_matches(kvp->v.z, zVal); |
| |
| }else{ |
| assert( 0==kvp->v.n || 0==nVal ); |
| buul = kvp->v.n == nVal; |
| } |
| db_free(zVal); |
| }else{ |
| if( bCheckDefined ){ |
| buul = db_define_has(kvp->k.z, kvp->k.n); |
| }else{ |
| buul = db_define_get_bool(kvp->k.z, kvp->k.n); |
| } |
| } |
| return buul; |
| } |
|
|
| #if 0 |
| |
| |
| |
| static int cmpp_key_truth(CmppKeyword const * pKw, |
| CmppTokenizer const * t, |
| char const *zKey, int bCheckDefined){ |
| cmpp_kvp kvp = cmpp_kvp_empty; |
| cmpp_kvp_parse(&kvp, zKey, -1, cmpp_kvp_op_eq1); |
| return cmpp_kvp_truth(pKw, t, &kvp, bCheckDefined); |
| } |
| #endif |
|
|
| |
| cmpp_kvp_op_e cmpp_t_is_op(CmppTokenizer const * t, int arg){ |
| if( t->args.argc > arg ){ |
| char const * const z = (char const *)t->args.argv[arg]; |
| #define E(N,S) if( strcmp(S,z) ) return cmpp_kvp_op_ ## N; else |
| cmpp_kvp_op_map(E) |
| #undef E |
| if(0) {} |
| } |
| return cmpp_kvp_op_none; |
| } |
|
|
| |
| |
| |
| |
| |
| |
| struct CmppExprDef { |
| |
| cmpp_kvp kvp; |
| struct { |
| int ndx; |
| int next; |
| } arg; |
| CmppTokenizer const * tizer; |
| |
| unsigned char bNegated; |
| |
| unsigned char bCheckDefined; |
| }; |
| typedef struct CmppExprDef CmppExprDef; |
| #define CmppExprDef_empty_m {cmpp_kvp_empty_m,{0,0},0,0,0} |
| static const CmppExprDef CmppExprDef_empty = CmppExprDef_empty_m; |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| static int CmppExprDef_eval(CmppExprDef const * cep){ |
| int buul = 0; |
|
|
| if( cep->bCheckDefined ){ |
| assert( !cep->kvp.v.n ); |
| buul = db_define_has(cep->kvp.k.z, (int)cep->kvp.k.n); |
| }else if( cep->kvp.v.z ){ |
| unsigned nVal = 0; |
| char * zVal = 0; |
| buul = db_define_get(cep->kvp.k.z, cep->kvp.k.n, &zVal, &nVal); |
| if( nVal ){ |
| buul = cmpp_val_matches(cep->kvp.v.z, zVal); |
| } |
| db_free(zVal); |
| }else{ |
| buul = db_define_get_bool(cep->kvp.k.z, cep->kvp.k.n); |
| } |
| return cep->bNegated ? !buul : buul; |
| } |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| static int CmppExprDef_read_one(CmppKeyword const * pKw, |
| CmppTokenizer const * t, |
| int startArg, CmppExprDef * cep){ |
| char const *zKey = 0; |
| *cep = CmppExprDef_empty; |
| cep->arg.ndx = startArg; |
| assert( t->args.pKw ); |
| assert( t->args.pKw==pKw ); |
| cep->tizer = t; |
| for(int i = startArg; !zKey && i<t->args.argc; ++i ){ |
| char const * z = (char const *)t->args.argv[i]; |
| if( 0==strcmp(z, "not") ){ |
| cep->bNegated = !cep->bNegated; |
| }else if( 0==strcmp(z,"defined") ){ |
| if( cep->bCheckDefined ){ |
| cmpp_kwd__err(pKw, t, |
| "Cannot use 'defined' more than once"); |
| } |
| cep->bCheckDefined = 1; |
| }else{ |
| assert( !zKey ); |
| cmpp_kvp_parse(&cep->kvp, z, -1, cmpp_kvp_op_eq1); |
| if( cep->bCheckDefined && cep->kvp.v.z ){ |
| cmpp_kwd__err(pKw, t, "Cannot use X=Y keys with 'defined'"); |
| cep->arg.next = ++i; |
| } |
| return 1; |
| } |
| } |
| return 0; |
| } |
|
|
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| |
| static int CmppExprDef_parse_cond(CmppKeyword const *pKw, |
| CmppTokenizer *t, |
| CmppExprDef const * pStart){ |
| enum { Op_none = 0, Op_And, Op_Or }; |
| int lhs = CmppExprDef_eval(pStart); |
| int op = Op_none; |
| int i = pStart->arg.next; |
| for( ; i < t->args.argc; ++i ){ |
| CmppExprDef eNext = CmppExprDef_empty; |
| char const *z = (char const *)t->args.argv[i]; |
| if( 0==strcmp("and",z) ){ |
| if( Op_none!=op ) goto multiple_ops; |
| op = Op_And; |
| continue; |
| }else if( 0==strcmp("or",z) ){ |
| if( Op_none!=op ) goto multiple_ops; |
| op = Op_Or; |
| continue; |
| }else if( !CmppExprDef_read_one(pKw, t, i, &eNext) ){ |
| if( Op_none!=op ){ |
| cmpp_t__err(t, "Stray operator: %s",z); |
| } |
| } |
| assert( eNext.kvp.k.z ); |
| int const rhs = CmppExprDef_eval(&eNext); |
| switch( op ){ |
| case Op_none: break; |
| case Op_And: lhs = lhs && rhs; break; |
| case Op_Or: lhs = lhs || rhs; break; |
| default: |
| assert(!"cannot happen"); |
| fatal("this cannot happen"); |
| } |
| op = Op_none; |
| } |
| if( Op_none!=op ){ |
| cmpp_t__err(t, "Extra operator at end of expression"); |
| }else if( i < t->args.argc ){ |
| assert(!"cannot happen"); |
| cmpp_kwd__err(t->args.pKw, t, "Unhandled extra arguments"); |
| }else{ |
| return lhs; |
| } |
| assert(!"not reached"); |
| multiple_ops: |
| cmpp_t__err(t,"Cannot have multiple operators"); |
| return 0 ; |
| } |
|
|
| |
| static void cmpp_kwd_if(CmppKeyword const * pKw, CmppTokenizer *t){ |
| CmppParseState tmpState = TS_Start; |
| CmppExprDef cep = CmppExprDef_empty; |
| |
| assert( TT_If==pKw->ttype |
| || TT_Elif==pKw->ttype |
| || TT_Assert==pKw->ttype); |
| if(t->args.argc<2){ |
| cmpp_kwd__err(pKw, t, "Expecting an argument"); |
| } |
| CmppExprDef_read_one(pKw, t, 1, &cep); |
| if( !cep.kvp.k.z ){ |
| cmpp_kwd__err(pKw, t, "Missing key argument"); |
| } |
| |
| |
| |
| switch(pKw->ttype){ |
| case TT_Assert: |
| break; |
| case TT_Elif: |
| switch(CT_pstate(t)){ |
| case TS_If: break; |
| case TS_IfPassed: CT_level(t).flags |= CmppLevel_F_ELIDE; return; |
| default: |
| cmpp_kwd__err(pKw, t, "'%s' used out of context", |
| pKw->zName); |
| } |
| break; |
| case TT_If: |
| CmppLevel_push(t); |
| break; |
| default: |
| assert(!"cannot happen"); |
| cmpp_kwd__err(pKw, t, "Unexpected keyword token type"); |
| break; |
| } |
| if( CmppExprDef_parse_cond( pKw, t, &cep ) ){ |
| CT_pstate(t) = tmpState = TS_IfPassed; |
| CT_skipLevel(t) = 0; |
| }else{ |
| if( TT_Assert==pKw->ttype ){ |
| cmpp_kwd__err(pKw, t, "Assertion failed: %s", |
| |
| |
| cep.kvp.k.z); |
| } |
| CT_pstate(t) = TS_If ; |
| CT_skipLevel(t) = 1; |
| g_debug(3,("setting CT_skipLevel = 1 @ level %d\n", t->level.ndx)); |
| } |
| if( TT_If==pKw->ttype ){ |
| unsigned const lvlIf = t->level.ndx; |
| CmppToken const lvlToken = CT_level(t).token; |
| while(cmpp_next_keyword_line(t)){ |
| cmpp_process_keyword(t); |
| if(lvlIf > t->level.ndx){ |
| assert(TT_Endif == t->token.ttype); |
| break; |
| } |
| #if 0 |
| if(TS_IfPassed==tmpState){ |
| tmpState = TS_Start; |
| t->level.stack[lvlIf].flags |= CmppLevel_F_ELIDE; |
| g_debug(1,("Setting ELIDE for TS_IfPassed @ lv %d (lvlIf=%d)\n", t->level.ndx, lvlIf)); |
| } |
| #endif |
| } |
| if(lvlIf <= t->level.ndx){ |
| cmpp_kwd__err(pKw, t, |
| "Input ended inside an unterminated %sif " |
| "opened at [%s] line %u", |
| g.delim.z, t->zName, lvlToken.lineNo); |
| } |
| } |
| } |
|
|
| |
| static void cmpp_kwd_else(CmppKeyword const * pKw, CmppTokenizer *t){ |
| if(t->args.argc>1){ |
| cmpp_kwd__err(pKw, t, "Expecting no arguments"); |
| } |
| switch(CT_pstate(t)){ |
| case TS_IfPassed: CT_skipLevel(t) = 1; break; |
| case TS_If: CT_skipLevel(t) = 0; break; |
| default: |
| cmpp_kwd__err(pKw, t, "'%s' with no matching 'if'", |
| pKw->zName); |
| } |
| |
| |
| CT_pstate(t) = TS_Else; |
| } |
|
|
| |
| static void cmpp_kwd_endif(CmppKeyword const * pKw, CmppTokenizer *t){ |
| |
| |
| |
| |
| |
| |
| switch(CT_pstate(t)){ |
| case TS_Else: |
| case TS_If: |
| case TS_IfPassed: |
| break; |
| default: |
| cmpp_kwd__err(pKw, t, "'%s' with no matching 'if'", |
| pKw->zName); |
| } |
| CmppLevel_pop(t); |
| } |
|
|
| |
| static void cmpp_kwd_include(CmppKeyword const * pKw, CmppTokenizer *t){ |
| char const * zFile; |
| char * zResolved; |
| int bRaw = 0 ; |
| if(CT_skip(t)) return; |
| else if(t->args.argc<2 || t->args.argc>3){ |
| cmpp_kwd__err(pKw, t, "Expected args: ?-raw? filename"); |
| } |
| if(t->args.argc==2){ |
| zFile = (const char *)t->args.argv[1]; |
| }else{ |
| if( 0==strcmp("-raw", (char*)t->args.argv[1]) ){ |
| bRaw = 1; |
| }else{ |
| cmpp_kwd__err(pKw, t, "Unhandled argument: %s", |
| t->args.argv[1]); |
| } |
| zFile = (const char *)t->args.argv[2]; |
| } |
| if(!bRaw && db_including_has(zFile)){ |
| |
| |
| |
| |
| |
| |
| cmpp_t__err(t, "Recursive include of file: %s", zFile); |
| } |
| zResolved = db_include_search(zFile); |
| if(zResolved){ |
| if( bRaw ) db_including_add(zFile, t->zName, t->token.lineNo); |
| cmpp_process_file(zResolved, bRaw); |
| if( bRaw ) db_include_rm(zFile); |
| db_free(zResolved); |
| }else{ |
| cmpp_t__err(t, "file not found: %s", zFile); |
| } |
| } |
|
|
| static void cmpp_dump_defines( FILE * fp, int bIndent ){ |
| sqlite3_stmt * const q = g_stmt(GStmt_defSelAll); |
| while( SQLITE_ROW==sqlite3_step(q) ){ |
| unsigned char const * zK = sqlite3_column_text(q, 0); |
| unsigned char const * zV = sqlite3_column_text(q, 1); |
| int const nK = sqlite3_column_bytes(q, 0); |
| int const nV = sqlite3_column_bytes(q, 1); |
| fprintf(fp, "%s%.*s = %.*s\n", |
| bIndent ? "\t" : "", nK, zK, nV, zV); |
| } |
| g_stmt_reset(q); |
| } |
|
|
| |
| static void cmpp_kwd_pragma(CmppKeyword const * pKw, CmppTokenizer *t){ |
| const char * zArg; |
| if(CT_skip(t)) return; |
| else if(t->args.argc<2){ |
| cmpp_kwd__err(pKw, t, "Expecting an argument"); |
| } |
| zArg = (const char *)t->args.argv[1]; |
| #define M(X) 0==strcmp(zArg,X) |
| if(M("defines")){ |
| cmpp_dump_defines(stderr, 1); |
| } |
| else if(M("chomp-F")){ |
| g.flags.chompF = 1; |
| }else if(M("no-chomp-F")){ |
| g.flags.chompF = 0; |
| } |
| #if 0 |
| |
| else if(M("@")){ |
| if(t->args.argc>2){ |
| g.flags.atPolicy = |
| AtPolicy_fromStr((char const *)t->args.argv[2], 1); |
| }else{ |
| g.flags.atPolicy = AT_DEFAULT; |
| } |
| }else if(M("no-@")){ |
| g.flags.atPolicy = AT_OFF; |
| } |
| #endif |
| else{ |
| cmpp_kwd__err(pKw, t, "Unknown pragma: %s", zArg); |
| } |
| #undef M |
| } |
|
|
| static void db_step_reset(sqlite3_stmt * const q, char const * zErrTip){ |
| db_affirm_rc(sqlite3_step(q), zErrTip); |
| g_stmt_reset(q); |
| } |
|
|
| static void cmpp_sp_begin(CmppTokenizer * const t){ |
| db_step_reset(g_stmt(GStmt_spBegin), "Starting savepoint"); |
| ++t->nSavepoint; |
| } |
|
|
| static void cmpp_sp_rollback(CmppTokenizer * const t){ |
| if( !t->nSavepoint ){ |
| cmpp_t__err(t, "Cannot roll back: no active savepoint"); |
| } |
| db_step_reset(g_stmt(GStmt_spRollback), |
| "Rolling back savepoint"); |
| db_step_reset(g_stmt(GStmt_spRelease), |
| "Releasing rolled-back savepoint"); |
| --t->nSavepoint; |
| } |
|
|
| static void cmpp_sp_commit(CmppTokenizer * const t){ |
| if( !t->nSavepoint ){ |
| cmpp_t__err(t, "Cannot commit: no active savepoint"); |
| } |
| db_step_reset(g_stmt(GStmt_spRelease), "Rolling back savepoint"); |
| --t->nSavepoint; |
| } |
|
|
| void CmppTokenizer_cleanup(CmppTokenizer * const t){ |
| while( t->nSavepoint ){ |
| cmpp_sp_rollback(t); |
| } |
| } |
|
|
| |
| static void cmpp_kwd_savepoint(CmppKeyword const * pKw, CmppTokenizer *t){ |
| const char * zArg; |
| if(CT_skip(t)) return; |
| else if(t->args.argc!=2){ |
| cmpp_kwd__err(pKw, t, "Expecting one argument"); |
| } |
| zArg = (const char *)t->args.argv[1]; |
| #define M(X) 0==strcmp(zArg,X) |
| if(M("begin")){ |
| cmpp_sp_begin(t); |
| }else if(M("rollback")){ |
| cmpp_sp_rollback(t); |
| }else if(M("commit")){ |
| cmpp_sp_commit(t); |
| }else{ |
| cmpp_kwd__err(pKw, t, "Unknown savepoint option: %s", zArg); |
| } |
| #undef SP_NAME |
| #undef M |
| } |
|
|
| |
| static void cmpp_kwd_stderr(CmppKeyword const * pKw, CmppTokenizer *t){ |
| if(CT_skip(t)) return; |
| else{ |
| const char *zBegin = t->args.argc>1 |
| ? (const char *)t->args.argv[1] : 0; |
| if(zBegin){ |
| g_stderr("%s:%u: %s\n", t->zName, t->token.lineNo, zBegin); |
| }else{ |
| g_stderr("%s:%u: (no %.*s%s argument)\n", |
| t->zName, t->token.lineNo, |
| g.delim.n, g.delim.z, pKw->zName); |
| } |
| } |
| } |
|
|
| |
| static void cmpp_kwd_at_policy(CmppKeyword const * pKw, CmppTokenizer *t){ |
| if(CT_skip(t)) return; |
| else if(t->args.argc<2){ |
| g.flags.atPolicy = AT_DEFAULT; |
| }else{ |
| g.flags.atPolicy = AtPolicy_fromStr((char const*)t->args.argv[1], 1); |
| } |
| } |
|
|
|
|
| #if 0 |
| |
| static void cmpp_kwd_todo(CmppKeyword const * pKw, CmppTokenizer *t){ |
| (void)t; |
| g_debug(0,("TODO: keyword handler for %s\n", pKw->zName)); |
| } |
| #endif |
|
|
| CmppKeyword aKeywords[] = { |
| |
| #define S(NAME) DStrings.NAME.z, DStrings.NAME.n |
| {S(Comment), 0, TT_Comment, cmpp_kwd_noop}, |
| {S(AtPolicy), 1, TT_AtPolicy, cmpp_kwd_at_policy}, |
| {S(Assert),1, TT_Assert, cmpp_kwd_if}, |
| {S(Define), 1, TT_Define, cmpp_kwd_define}, |
| {S(Elif), 1, TT_Elif, cmpp_kwd_if}, |
| {S(Else), 1, TT_Else, cmpp_kwd_else}, |
| {S(Endif), 0, TT_Endif, cmpp_kwd_endif}, |
| {S(Error), 0, TT_Error, cmpp_kwd_error}, |
| {S(If), 1, TT_If, cmpp_kwd_if}, |
| {S(Include), 1, TT_Include, cmpp_kwd_include}, |
| {S(Pragma), 1, TT_Pragma, cmpp_kwd_pragma}, |
| {S(Savepoint), 1, TT_Savepoint, cmpp_kwd_savepoint}, |
| {S(Stderr), 0, TT_Stderr, cmpp_kwd_stderr}, |
| {S(Undef), 1, TT_Undef, cmpp_kwd_define}, |
| #undef S |
| {0,0,TT_Invalid, 0} |
| }; |
|
|
| static int cmpp_CmppKeyword(const void *p1, const void *p2){ |
| char const * zName = (const char *)p1; |
| CmppKeyword const * kw = (CmppKeyword const *)p2; |
| return strcmp(zName, kw->zName); |
| } |
|
|
| CmppKeyword const * CmppKeyword_search(const char *zName){ |
| return (CmppKeyword const *)bsearch(zName, &aKeywords[0], |
| sizeof(aKeywords)/sizeof(aKeywords[0]) - 1, |
| sizeof(aKeywords[0]), |
| cmpp_CmppKeyword); |
| } |
|
|
| void cmpp_process_keyword(CmppTokenizer * const t){ |
| assert(t->args.pKw); |
| assert(t->args.argc); |
| t->args.pKw->xCall(t->args.pKw, t); |
| t->args.pKw = 0; |
| t->args.argc = 0; |
| } |
|
|
| void cmpp_process_string(const char * zName, |
| unsigned char const * zIn, |
| int nIn){ |
| nIn = cmpp__strlenu(zIn, nIn); |
| if( !nIn ) return; |
| CmppTokenizer const * const oldTok = g.tok; |
| CmppTokenizer ct = CmppTokenizer_empty; |
| ct.zName = zName; |
| ct.zBegin = zIn; |
| ct.zEnd = zIn + nIn; |
| while(cmpp_next_keyword_line(&ct)){ |
| cmpp_process_keyword(&ct); |
| } |
| if(0!=ct.level.ndx){ |
| CmppLevel const * const lv = CmppLevel_get(&ct); |
| fatal("Input ended inside an unterminated nested construct " |
| "opened at [%s] line %u", zName, lv->token.lineNo); |
| } |
| CmppTokenizer_cleanup(&ct); |
| g.tok = oldTok; |
| } |
|
|
| void cmpp_process_file(const char * zName, int bRaw){ |
| FileWrapper fw = FileWrapper_empty; |
| FileWrapper_open(&fw, zName, "r"); |
| g_FileWrapper_link(&fw); |
| FileWrapper_slurp(&fw); |
| g_debug(1,("Read %u byte(s) from [%s]\n", fw.nContent, fw.zName)); |
| if( fw.zContent ){ |
| if( bRaw ){ |
| FileWrapper fw = FileWrapper_empty; |
| FileWrapper_open(&fw, zName, "rb"); |
| g_FileWrapper_link(&fw); |
| FileWrapper_slurp(&fw); |
| if( 1!=fwrite(fw.zContent, fw.nContent, 1, g.out.pFile) ){ |
| fatal("fwrite() failed with errno %d\n", errno); |
| } |
| g_FileWrapper_close(&fw); |
| }else{ |
| cmpp_process_string(zName, fw.zContent, fw.nContent); |
| } |
| } |
| g_FileWrapper_close(&fw); |
| } |
|
|
|
|
| void fatalv__base(char const *zFile, int line, |
| char const *zFmt, va_list va){ |
| FILE * const fp = stderr; |
| fflush(stdout); |
| fputc('\n', fp); |
| if( g.flags.doDebug ){ |
| fprintf(fp, "%s: ", g.zArgv0); |
| if( zFile ){ |
| fprintf(fp, "%s:%d ",zFile, line); |
| } |
| } |
| if( g.tok ){ |
| fprintf(fp,"@%s:%d: ", |
| (g.tok->zName && 0==strcmp("-",g.tok->zName)) |
| ? "<stdin>" |
| : g.tok->zName, |
| g.tok->lineNo); |
| } |
| if(zFmt && *zFmt){ |
| vfprintf(fp, zFmt, va); |
| } |
| fputc('\n', fp); |
| fflush(fp); |
| exit(1); |
| } |
|
|
| void fatal__base(char const *zFile, int line, |
| char const *zFmt, ...){ |
| va_list va; |
| va_start(va, zFmt); |
| fatalv__base(zFile, line, zFmt, va); |
| va_end(va); |
| } |
|
|
| #undef CT_level |
| #undef CT_pstate |
| #undef CT_skipLevel |
| #undef CT_skip |
| #undef CLvl_skip |
|
|
| static void usage(int isErr){ |
| FILE * const fOut = isErr ? stderr : stdout; |
| fprintf(fOut, "Usage: %s [flags] [infile...]\n", g.zArgv0); |
| fprintf(fOut, |
| "Flags and filenames may be in any order and " |
| "they are processed in that order.\n" |
| "\nFlags:\n"); |
| #define GAP " " |
| #define arg(F,D) fprintf(fOut,"\n %s\n" GAP "%s\n",F, D) |
| arg("-o|--outfile FILE","Send output to FILE (default=- (stdout)).\n" |
| GAP "Because arguments are processed in order, this should\n" |
| GAP "normally be given before -f."); |
| arg("-f|--file FILE","Process FILE (default=- (stdin)).\n" |
| GAP "All non-flag arguments are assumed to be the input files."); |
| arg("-DXYZ[=value]","Define XYZ to the given value (default=1)."); |
| arg("-UXYZ","Undefine all defines matching glob XYZ."); |
| arg("-IXYZ","Add dir XYZ to the " CMPP_DEFAULT_DELIM "include path."); |
| arg("-FXYZ=filename", |
| "Define XYZ to the raw contents of the given file.\n" |
| GAP "The file is not processed as by " CMPP_DEFAULT_DELIM"include\n" |
| GAP "Maybe it should be. Or maybe we need a new flag for that."); |
| arg("-d|--delimiter VALUE", "Set keyword delimiter to VALUE " |
| "(default=" CMPP_DEFAULT_DELIM ")."); |
| arg("--@policy retain|elide|error|off", |
| "Specifies how to handle @tokens@ (default=off).\n" |
| GAP "off = do not look for @tokens@\n" |
| GAP "retain = parse @tokens@ and retain any undefined ones\n" |
| GAP "elide = parse @tokens@ and elide any undefined ones\n" |
| GAP "error = parse @tokens@ and error out for any undefined ones" |
| ); |
| arg("-@", "Equivalent to --@policy=error."); |
| arg("-no-@", "Equivalent to --@policy=off (the default)."); |
| arg("--sql-trace", "Send a trace of all SQL to stderr."); |
| arg("--sql-trace-x", |
| "Like --sql-trace but expand all bound values in the SQL."); |
| arg("--no-sql-trace", "Disable SQL tracing (default)."); |
| arg("--chomp-F", "One trailing newline is trimmed from files " |
| "read via -FXYZ=filename."); |
| arg("--no-chomp-F", "Disable --chomp-F (default)."); |
| #undef arg |
| #undef GAP |
| fputs("\nFlags which require a value accept either " |
| "--flag=value or --flag value.\n\n",fOut); |
| } |
|
|
| |
| |
| |
| |
| |
| |
| |
| static void get_flag_val(int argc, char const * const * argv, int * ndx, |
| char const **zVal){ |
| char const * zEq = strchr(argv[*ndx], '='); |
| if( zEq ){ |
| *zVal = zEq+1; |
| return; |
| } |
| if(*ndx+1>=argc){ |
| fatal("Missing value for flag '%s'", argv[*ndx]); |
| } |
| *zVal = argv[++*ndx]; |
| } |
|
|
| static int arg_is_flag( char const *zFlag, char const *zArg, |
| char const **zValIfEqX ){ |
| *zValIfEqX = 0; |
| if( 0==strcmp(zFlag, zArg) ) return 1; |
| char const * z = strchr(zArg,'='); |
| if( z && z>zArg ){ |
| |
| if( 0==strncmp(zFlag, zArg, z-zArg) ){ |
| if( !zFlag[z-zArg] ){ |
| *zValIfEqX = z+1; |
| return 1; |
| } |
| |
| } |
| } |
| return 0; |
| } |
|
|
| static void define_argv(int argc, char const * const * argv){ |
| sqlite3_str * const s = sqlite3_str_new(g.db); |
| sqlite3_str_append(s, "c-pp::argv=", 11); |
| for( int i = 0; i < argc; ++i ){ |
| if( i ) sqlite3_str_appendchar(s, 1, ' '); |
| sqlite3_str_appendf(s, "%s", argv[i]); |
| } |
| char * const z = sqlite3_str_finish(s); |
| assert(z); |
| if(z){ |
| db_define_add(z, NULL); |
| sqlite3_free(z); |
| } |
| } |
|
|
| int main(int argc, char const * const * argv){ |
| int rc = 0; |
| int inclCount = 0; |
| int nFile = 0; |
| int ndxTrace = 0; |
| int expandMode = 0; |
| char const * zVal = 0; |
| #define ARGVAL if( !zVal ) get_flag_val(argc, argv, &i, &zVal) |
| #define M(X) arg_is_flag(X, zArg, &zVal) |
| #define ISFLAG(X) else if(M(X)) |
| #define ISFLAG2(X,Y) else if(M(X) || M(Y)) |
| #define NOVAL if( zVal ) fatal("Unexpected value for %s", zArg) |
| #define g_out_open \ |
| if(!g.out.pFile) FileWrapper_open(&g.out, "-", "w"); \ |
| if(!inclCount){ db_include_dir_add("."); ++inclCount; } (void)0 |
|
|
| g.zArgv0 = argv[0]; |
| #define DOIT if(doIt) |
| for(int doIt = 0; doIt<2; ++doIt){ |
| |
| |
| |
| |
| |
| |
| |
| |
| DOIT{ |
| atexit(cmpp_atexit); |
| if( 1==ndxTrace ){ |
| |
| |
| |
| g.sqlTrace.pFile = stderr; |
| g.sqlTrace.expandSql = expandMode; |
| } |
| cmpp_initdb(); |
| define_argv(argc, argv); |
| } |
| for(int i = 1; i < argc; ++i){ |
| int negate = 0; |
| char const * zArg = argv[i]; |
| |
| zVal = 0; |
| while('-'==*zArg) ++zArg; |
| if(zArg==argv[i]){ |
| zVal = zArg; |
| goto do_infile; |
| } |
| if( 0==strncmp(zArg,"no-",3) ){ |
| zArg += 3; |
| negate = 1; |
| } |
| ISFLAG2("?","help"){ |
| NOVAL; |
| usage(0); |
| goto end; |
| }else if('D'==*zArg){ |
| ++zArg; |
| if(!*zArg) fatal("Missing key for -D"); |
| DOIT { |
| db_define_add(zArg, 0); |
| } |
| }else if('F'==*zArg){ |
| ++zArg; |
| if(!*zArg) fatal("Missing key for -F"); |
| DOIT { |
| db_define_add_file(zArg); |
| } |
| }else if('U'==*zArg){ |
| ++zArg; |
| if(!*zArg) fatal("Missing key for -U"); |
| DOIT { |
| db_define_rm(zArg); |
| } |
| }else if('I'==*zArg){ |
| ++zArg; |
| if(!*zArg) fatal("Missing directory for -I"); |
| DOIT { |
| db_include_dir_add(zArg); |
| ++inclCount; |
| } |
| } |
| ISFLAG2("o","outfile"){ |
| ARGVAL; |
| DOIT { |
| FileWrapper_open(&g.out, zVal, "w"); |
| } |
| } |
| ISFLAG2("f","file"){ |
| ARGVAL; |
| do_infile: |
| DOIT { |
| ++nFile; |
| g_out_open; |
| cmpp_process_file(zVal, 0); |
| } |
| } |
| ISFLAG("e"){ |
| ARGVAL; |
| DOIT { |
| ++nFile; |
| g_out_open; |
| cmpp_process_string("-e script", ustr_c(zVal), -1); |
| } |
| } |
| ISFLAG("@"){ |
| NOVAL; |
| DOIT { |
| assert( AT_DEFAULT!=AT_OFF ); |
| g.flags.atPolicy = negate ? AT_OFF : AT_DEFAULT; |
| } |
| } |
| ISFLAG("@policy"){ |
| AtPolicy aup; |
| ARGVAL; |
| aup = AtPolicy_fromStr(zVal, 1); |
| DOIT { |
| g.flags.atPolicy = aup; |
| } |
| } |
| ISFLAG("debug"){ |
| NOVAL; |
| g.flags.doDebug += negate ? -1 : 1; |
| } |
| ISFLAG("sql-trace"){ |
| NOVAL; |
| |
| |
| g.sqlTrace.expandSql = 0; |
| DOIT { |
| g.sqlTrace.pFile = negate ? (FILE*)0 : stderr; |
| }else if( !ndxTrace && !negate ){ |
| ndxTrace = i; |
| expandMode = 0; |
| } |
| } |
| ISFLAG("sql-trace-x"){ |
| NOVAL; |
| g.sqlTrace.expandSql = 1; |
| DOIT { |
| g.sqlTrace.pFile = negate ? (FILE*)0 : stderr; |
| }else if( !ndxTrace && !negate ){ |
| ndxTrace = i; |
| expandMode = 1; |
| } |
| } |
| ISFLAG("chomp-F"){ |
| NOVAL; |
| DOIT g.flags.chompF = !negate; |
| } |
| ISFLAG2("d","delimiter"){ |
| ARGVAL; |
| if( !doIt ){ |
| g.delim.z = zVal; |
| g.delim.n = (unsigned short)strlen(zVal); |
| if(!g.delim.n) fatal("Keyword delimiter may not be empty."); |
| } |
| } |
| ISFLAG2("dd", "dump-defines"){ |
| DOIT { |
| FILE * const fp = stderr; |
| fprintf(fp, "All %sdefine entries:\n", g.delim.z); |
| cmpp_dump_defines(fp, 1); |
| } |
| } |
| else{ |
| fatal("Unhandled flag: %s", argv[i]); |
| } |
| } |
| DOIT { |
| if(!nFile){ |
| if(!g.out.zName) g.out.zName = "-"; |
| if(!inclCount){ |
| db_include_dir_add("."); |
| ++inclCount; |
| } |
| FileWrapper_open(&g.out, g.out.zName, "w"); |
| cmpp_process_file("-", 0); |
| } |
| } |
| } |
| end: |
| g_cleanup(1); |
| return rc ? EXIT_FAILURE : EXIT_SUCCESS; |
| } |
|
|