Annotation of parser3/src/classes/op.C, revision 1.210

1.1       paf         1: /** @file
1.9       paf         2:        Parser: parser @b operators.
1.1       paf         3: 
1.209     moko        4:        Copyright (c) 2001-2012 Art. Lebedev Studio (http://www.artlebedev.com)
1.71      paf         5:        Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
1.98      paf         6: */
1.1       paf         7: 
1.13      paf         8: #include "classes.h"
1.129     paf         9: #include "pa_vmethod_frame.h"
                     10: 
1.1       paf        11: #include "pa_common.h"
1.134     paf        12: #include "pa_os.h"
1.1       paf        13: #include "pa_request.h"
                     14: #include "pa_vint.h"
                     15: #include "pa_sql_connection.h"
1.79      paf        16: #include "pa_vdate.h"
1.105     paf        17: #include "pa_vmethod_frame.h"
1.129     paf        18: #include "pa_vclass.h"
1.144     paf        19: #include "pa_charset.h"
1.1       paf        20: 
1.210   ! moko       21: volatile const char * IDENT_OP_C="$Id: op.C,v 1.209 2012-03-16 09:24:07 moko Exp $";
1.209     moko       22: 
1.18      parser     23: // limits
                     24: 
1.162     paf        25: #define MAX_LOOPS 20000
1.18      parser     26: 
1.10      paf        27: // defines
                     28: 
1.82      paf        29: #define CASE_DEFAULT_VALUE "DEFAULT"
1.146     paf        30: #define PROCESS_MAIN_OPTION_NAME "main"
                     31: #define PROCESS_FILE_OPTION_NAME "file"
                     32: #define PROCESS_LINENO_OPTION_NAME "lineno"
1.11      paf        33: 
1.10      paf        34: // class
                     35: 
1.113     paf        36: class VClassMAIN: public VClass {
1.10      paf        37: public:
1.129     paf        38:        VClassMAIN();
1.10      paf        39: };
                     40: 
1.129     paf        41: // defines for statics
                     42: 
                     43: #define SWITCH_DATA_NAME "SWITCH-DATA"
                     44: #define CACHE_DATA_NAME "CACHE-DATA"
1.164     paf        45: 
1.129     paf        46: #define EXCEPTION_VAR_NAME "exception"
                     47: 
                     48: // statics
                     49: 
                     50: //^switch ^case
                     51: static const String switch_data_name(SWITCH_DATA_NAME);
                     52: //^cache
                     53: static const String cache_data_name(CACHE_DATA_NAME);
                     54: 
                     55: static const String exception_var_name(EXCEPTION_VAR_NAME);
                     56: 
1.164     paf        57: 
1.136     paf        58: // local defines
                     59: 
                     60: #define CACHE_EXCEPTION_HANDLED_CACHE_NAME "cache"
                     61: 
1.129     paf        62: // helpers
                     63: 
1.187     misha      64: class Untaint_lang_name2enum: public HashString<String::Language> {
1.129     paf        65: public:
                     66:        Untaint_lang_name2enum() {
                     67:                #define ULN(name, LANG) \
1.130     paf        68:                        put(String::Body(name), (value_type)(String::L_##LANG));
1.198     misha      69:                ULN("clean", CLEAN);
1.129     paf        70:                ULN("as-is", AS_IS);
                     71:                ULN("optimized-as-is", AS_IS|String::L_OPTIMIZE_BIT);
                     72:                ULN("file-spec", FILE_SPEC);
                     73:                ULN("http-header", HTTP_HEADER);
                     74:                ULN("mail-header", MAIL_HEADER);
                     75:                ULN("uri", URI);
                     76:                ULN("sql", SQL);
                     77:                ULN("js", JS);
                     78:                ULN("xml", XML);
                     79:                ULN("optimized-xml", XML|String::L_OPTIMIZE_BIT);
                     80:                ULN("html", HTML);
                     81:                ULN("optimized-html", HTML|String::L_OPTIMIZE_BIT);
1.159     paf        82:                ULN("regex", REGEX);
1.193     misha      83:                ULN("parser-code", PARSER_CODE);
1.204     misha      84:                ULN("json", JSON);
1.129     paf        85:                #undef ULN
                     86:        }
                     87: } untaint_lang_name2enum;
                     88: 
1.10      paf        89: // methods
                     90: 
1.129     paf        91: static void _if(Request& r, MethodParams& params) {
1.200     moko       92:        size_t max_param=params.count()-1;
                     93:        size_t i=0;
                     94:        do {
                     95:                bool condition=params.as_bool(i, "condition must be expression", r);
                     96:                if(condition) {
                     97:                        r.process_write(*params.get(i+1));
                     98:                        return;
                     99:                }
                    100:                i+=2;
                    101:        } while (i < max_param);
                    102: 
                    103:        if(i == max_param)
                    104:                r.process_write(*params.get(i));
1.1       paf       105: }
                    106: 
1.210   ! moko      107: String::Language get_untaint_lang(const String& lang_name){
1.175     misha     108:        String::Language lang=untaint_lang_name2enum.get(lang_name);
                    109:        if(!lang)
1.210   ! moko      110:                throw Exception(PARSER_RUNTIME, &lang_name, "invalid taint language");
1.175     misha     111:        return lang;
                    112: }
                    113: 
1.129     paf       114: static void _untaint(Request& r, MethodParams& params) {
                    115:        String::Language lang;
                    116:        if(params.count()==1)
1.175     misha     117:                lang=String::L_AS_IS; // mark as simply 'as-is'. useful in html from sql 
                    118:        else
1.210   ! moko      119:                lang=get_untaint_lang(params.as_string(0, "lang must be string"));
1.1       paf       120: 
                    121:        {
1.129     paf       122:                Value& vbody=params.as_junction(params.count()-1, "body must be code");
1.1       paf       123:                
1.191     misha     124:                Temp_lang temp_lang(r, lang); // set temporarily specified ^untaint[language;
                    125:                StringOrValue result=r.process(vbody); // process marking tainted with that lang
1.188     misha     126:                r.write_assign_lang(result);
1.1       paf       127:        }
                    128: }
                    129: 
1.129     paf       130: static void _taint(Request& r, MethodParams& params) {
                    131:        String::Language lang;
                    132:        if(params.count()==1)
1.175     misha     133:                lang=String::L_TAINTED; // mark as simply 'tainted'. useful in table:create
                    134:        else
1.210   ! moko      135:                lang=get_untaint_lang(params.as_string(0, "lang must be string"));
1.1       paf       136: 
                    137:        {
1.129     paf       138:                Value& vbody=params.as_no_junction(params.count()-1, "body must not be code");
1.1       paf       139:                
1.191     misha     140:                String result(vbody.as_string(), lang); // force result language to specified
1.188     misha     141:                r.write_assign_lang(result);
1.1       paf       142:        }
                    143: }
                    144: 
1.205     moko      145: static void _apply_taint(Request& r, MethodParams& params) {
1.210   ! moko      146:        String::Language lang=params.count()==1 ? String::L_AS_IS : get_untaint_lang(params.as_string(0, "lang must be string"));
1.205     moko      147:        const String &sbody=params.as_string(params.count()-1, "body must be string");
                    148:        String::Body result_body=sbody.cstr_to_string_body_untaint(lang, 0, &r.charsets);
                    149:        r.write_pass_lang(*new String(result_body, String::L_AS_IS));
                    150: }
                    151: 
1.129     paf       152: static void _process(Request& r, MethodParams& params) {
                    153:        Method* main_method;
                    154: 
                    155:        size_t index=0;
                    156:        Value* target_self;
                    157:        Value& maybe_target_self=params[index];
                    158:        if(maybe_target_self.get_string() || maybe_target_self.get_junction())
                    159:                target_self=&r.get_method_frame()->caller()->self();
                    160:        else {
                    161:                target_self=&maybe_target_self;  index++;
                    162:        }
                    163: 
1.1       paf       164:        {
1.197     misha     165:                VStateless_class *target_class=target_self->get_class();
1.116     paf       166:                if(!target_class)
1.166     misha     167:                        throw Exception(PARSER_RUNTIME,
1.129     paf       168:                                0,
1.116     paf       169:                                "no target class");
1.114     paf       170: 
1.73      paf       171:                // temporary remove language change
1.192     misha     172:                Temp_lang temp_lang(r, String::L_PARSER_CODE);
1.1       paf       173:                // temporary zero @main so to maybe-replace it in processed code
1.129     paf       174:                Temp_method temp_method_main(*target_class, main_method_name, 0);
1.1       paf       175:                // temporary zero @auto so it wouldn't be auto-called in Request::use_buf
1.129     paf       176:                Temp_method temp_method_auto(*target_class, auto_method_name, 0);
                    177: 
1.146     paf       178:                size_t options_index=index+1;
                    179:                HashStringValue* options=0;
                    180:                if(options_index<params.count()) {
1.195     misha     181:                        Value& voptions=params.as_no_junction(options_index, OPTIONS_MUST_NOT_BE_CODE);
1.146     paf       182:                        options=voptions.get_hash();
                    183:                        if(!options)
1.201     misha     184:                                throw Exception(PARSER_RUNTIME, 0, OPTIONS_MUST_BE_HASH);
1.146     paf       185:                }
                    186: 
1.129     paf       187:                const String* main_alias=0;
1.146     paf       188:                const String* file_alias=0;
                    189:                int line_no_alias_offset=0;
                    190:                if(options) {
                    191:                        int valid_options=0;
                    192:                        if(Value* vmain_alias=options->get(PROCESS_MAIN_OPTION_NAME)) {
                    193:                                valid_options++;
                    194:                                main_alias=&vmain_alias->as_string();
                    195:                        }
                    196:                        if(Value* vfile_alias=options->get(PROCESS_FILE_OPTION_NAME)) {
                    197:                                valid_options++;
                    198:                                file_alias=&vfile_alias->as_string();
                    199:                        }
                    200:                        if(Value* vline_no_alias_offset=options->get(PROCESS_LINENO_OPTION_NAME)) {
                    201:                                valid_options++;
                    202:                                line_no_alias_offset=vline_no_alias_offset->as_int();
                    203:                        }
                    204:        
                    205:                        if(valid_options!=options->count())
1.201     misha     206:                                throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
1.146     paf       207:                }
1.129     paf       208: 
1.146     paf       209:                uint processe_file_no=file_alias?
                    210:                        r.register_file(r.absolute(*file_alias))
                    211:                        : pseudo_file_no__process;
                    212:                // process...{string}
                    213:                Value& vjunction=params.as_junction(index, "body must be code");
                    214:                // evaluate source to process
                    215:                const String& source=r.process_to_string(vjunction);
                    216:                r.use_buf(*target_class,
1.190     misha     217:                        source.untaint_cstr(String::L_AS_IS, r.connection(false)),
1.146     paf       218:                        main_alias,
                    219:                        processe_file_no,
                    220:                        line_no_alias_offset);
1.1       paf       221: 
1.73      paf       222:                // main_method
1.129     paf       223:                main_method=target_class->get_method(main_method_name);
1.73      paf       224:        }
                    225:        // after restoring current-request-lang
                    226:        // maybe-execute @main[]
                    227:        if(main_method) {
1.202     moko      228:                VMethodFrame frame(*main_method, r.get_method_frame()->caller(), *target_self);
                    229:                frame.empty_params();
1.203     moko      230:                r.op_call(frame);
                    231:                r.write_pass_lang(frame.result());
1.1       paf       232:        }
                    233: }
                    234:        
1.138     paf       235: static void _rem(Request&, MethodParams& params) {
1.129     paf       236:        params.as_junction(0, "body must be code");
1.1       paf       237: }
                    238: 
1.129     paf       239: static void _while(Request& r, MethodParams& params) {
1.196     misha     240:        InCycle temp(r);
1.164     paf       241: 
1.170     misha     242:        Value& vcondition=params.as_expression(0, "condition must be number, bool or expression");
                    243: 
1.162     paf       244:        Value& body_code=params.as_junction(1, "body must be code");
                    245:        Value* delim_maybe_code=params.count()>2?&params[2]:0;
1.1       paf       246: 
                    247:        // while...
                    248:        int endless_loop_count=0;
1.186     misha     249:        if(delim_maybe_code){ // delimiter set
1.185     misha     250:                bool need_delim=false;
                    251:                while(true) {
                    252:                        if(++endless_loop_count>=MAX_LOOPS) // endless loop?
                    253:                                throw Exception(PARSER_RUNTIME,
                    254:                                        0,
                    255:                                        "endless loop detected");
                    256: 
                    257:                        if(!r.process_to_value(vcondition, false/*don't intercept string*/).as_bool())
                    258:                                break;
1.1       paf       259: 
1.185     misha     260:                        StringOrValue sv_processed=r.process(body_code);
                    261:                        Request::Skip lskip=r.get_skip(); r.set_skip(Request::SKIP_NOTHING);
1.186     misha     262: 
1.185     misha     263:                        const String* s_processed=sv_processed.get_string();
1.186     misha     264:                        if(s_processed && !s_processed->is_empty()) { // we have body
1.185     misha     265:                                if(need_delim) // need delim & iteration produced string?
                    266:                                        r.write_pass_lang(r.process(*delim_maybe_code));
                    267:                                else
                    268:                                        need_delim=true;
                    269:                        }
                    270:                        r.write_pass_lang(sv_processed);
1.1       paf       271: 
1.185     misha     272:                        if(lskip==Request::SKIP_BREAK)
                    273:                                break;
1.162     paf       274:                }
1.185     misha     275:        } else {
                    276:                while(true) {
                    277:                        if(++endless_loop_count>=MAX_LOOPS) // endless loop?
                    278:                                throw Exception(PARSER_RUNTIME,
                    279:                                        0,
                    280:                                        "endless loop detected");
                    281: 
                    282:                        if(!r.process_to_value(vcondition, false/*don't intercept string*/).as_bool())
                    283:                                break;
1.164     paf       284: 
1.185     misha     285:                        r.process_write(body_code);
                    286:                        Request::Skip lskip=r.get_skip(); r.set_skip(Request::SKIP_NOTHING);
                    287: 
                    288:                        if(lskip==Request::SKIP_BREAK)
                    289:                                break;
                    290:                }
1.1       paf       291:        }
                    292: }
                    293: 
1.129     paf       294: static void _use(Request& r, MethodParams& params) {
1.171     misha     295:        Value& vfile=params.as_no_junction(0, FILE_NAME_MUST_NOT_BE_CODE);
1.199     misha     296: 
                    297:        // _use could be called from the parser3 method only, so caller is always defined
1.202     moko      298:        r.use_file(r.main_class, vfile.as_string(), r.get_method_filename(&r.get_method_frame()->caller()->method));
1.1       paf       299: }
                    300: 
1.164     paf       301: static void set_skip(Request& r, Request::Skip askip) {
1.196     misha     302:        if(!r.get_in_cycle())
1.208     misha     303:                throw Exception(askip==Request::SKIP_BREAK?"parser.break":"parser.continue",
1.164     paf       304:                        0,
                    305:                        "without cycle");
                    306: 
                    307:        r.set_skip(askip);
                    308: }
                    309: 
                    310: static void _break(Request& r, MethodParams&) {
                    311:        set_skip(r, Request::SKIP_BREAK);
                    312: }
                    313: 
                    314: static void _continue(Request& r, MethodParams&) {
                    315:        set_skip(r, Request::SKIP_CONTINUE);
                    316: }
                    317: 
1.129     paf       318: static void _for(Request& r, MethodParams& params) {
1.196     misha     319:        InCycle temp(r);
1.164     paf       320: 
1.129     paf       321:        const String& var_name=params.as_string(0, "var name must be string");
                    322:        int from=params.as_int(1, "from must be int", r);
                    323:        int to=params.as_int(2, "to must be int", r);
1.186     misha     324:        Value& body_code=params.as_junction(3, "body must be code");
1.129     paf       325:        Value* delim_maybe_code=params.count()>4?&params[4]:0;
1.1       paf       326: 
1.57      paf       327:        if(to-from>=MAX_LOOPS) // too long loop?
1.166     misha     328:                throw Exception(PARSER_RUNTIME,
1.129     paf       329:                        0,
1.57      paf       330:                        "endless loop detected");
                    331: 
1.129     paf       332:        VInt* vint=new VInt(0);
1.154     paf       333: 
                    334:        VMethodFrame& caller=*r.get_method_frame()->caller();
1.197     misha     335:        caller.put_element(var_name, vint, false);
1.186     misha     336:        if(delim_maybe_code){ // delimiter set 
1.185     misha     337:                bool need_delim=false;
                    338: 
                    339:                for(int i=from; i<=to; i++) {
                    340:                        vint->set_int(i);
                    341: 
                    342:                        StringOrValue sv_processed=r.process(body_code);
                    343:                        Request::Skip lskip=r.get_skip(); r.set_skip(Request::SKIP_NOTHING);
1.186     misha     344: 
1.185     misha     345:                        const String* s_processed=sv_processed.get_string();
1.186     misha     346:                        if(s_processed && !s_processed->is_empty()) { // we have body
1.185     misha     347:                                if(need_delim) // need delim & iteration produced string?
                    348:                                        r.write_pass_lang(r.process(*delim_maybe_code));
                    349:                                else
                    350:                                        need_delim=true;
                    351:                        }
                    352:                        r.write_pass_lang(sv_processed);
1.1       paf       353: 
1.185     misha     354:                        if(lskip==Request::SKIP_BREAK)
                    355:                                break;
1.1       paf       356:                }
1.185     misha     357:        } else {
                    358:                for(int i=from; i<=to; i++) {
                    359:                        vint->set_int(i);
                    360:  
                    361:                        r.process_write(body_code);
                    362:                        Request::Skip lskip=r.get_skip(); r.set_skip(Request::SKIP_NOTHING);
1.164     paf       363: 
1.185     misha     364:                        if(lskip==Request::SKIP_BREAK)
                    365:                                break;
                    366:                }
1.1       paf       367:        }
                    368: }
                    369: 
1.129     paf       370: static void _eval(Request& r, MethodParams& params) {
                    371:        Value& expr=params.as_junction(0, "need expression");
1.1       paf       372:        // evaluate expresion
1.129     paf       373:        Value& value_result=r.process_to_value(expr, 
1.165     misha     374:                false/*don't intercept string*/).as_expr_result();
1.129     paf       375:        if(params.count()>1) {
1.177     misha     376:                const String& fmt=params.as_string(1, "fmt must be string").trim();
                    377:                if(fmt.is_empty()){
                    378:                        r.write_no_lang(value_result);
                    379:                } else {
                    380:                        r.write_no_lang(String(format(value_result.as_double(), fmt.cstrm())));
                    381:                }
1.91      paf       382:        } else
1.129     paf       383:                r.write_no_lang(value_result);
1.1       paf       384: }
                    385: 
1.129     paf       386: static void _connect(Request& r, MethodParams& params) {
1.63      paf       387: #ifdef RESOURCES_DEBUG
                    388: struct timeval mt[2];
                    389: #endif
1.129     paf       390:        Value& url=params.as_no_junction(0, "url must not be code");
                    391:        Value& body_code=params.as_junction(1, "body must be code");
1.1       paf       392: 
1.129     paf       393:        Table* protocol2driver_and_client=0;
1.197     misha     394:        if(Value* sql=r.main_class.get_element(String(MAIN_SQL_NAME))) {
                    395:                if(Value* element=sql->get_element(String(MAIN_SQL_DRIVERS_NAME))) {
1.113     paf       396:                        protocol2driver_and_client=element->get_table();
                    397:                }
                    398:        }
1.11      paf       399: 
1.63      paf       400: #ifdef RESOURCES_DEBUG
                    401: //measure:before
                    402: gettimeofday(&mt[0],NULL);
                    403: #endif
1.1       paf       404:        // connect
1.142     paf       405:        SQL_Connection* connection=SQL_driver_manager->get_connection(url.as_string(), 
1.144     paf       406:                protocol2driver_and_client,
1.180     misha     407:                r.charsets.source().NAME().cstr(),
                    408:                r.request_info.document_root);
1.1       paf       409: 
1.63      paf       410: #ifdef RESOURCES_DEBUG
                    411: //measure:after connect
                    412: gettimeofday(&mt[1],NULL);
                    413: 
                    414: double t[2];
                    415: for(int i=0;i<2;i++)
                    416:     t[i]=mt[i].tv_sec+mt[i].tv_usec/1000000.0;
                    417: 
                    418: r.sql_connect_time+=t[1]-t[0];
                    419: #endif
1.129     paf       420:        Temp_connection temp_connection(r, connection);
1.1       paf       421:        // execute body
1.53      parser    422:        try {
1.189     misha     423:                r.process_write(body_code);
1.129     paf       424:                connection->commit();
                    425:                connection->close();
1.75      paf       426:        } catch(...) { // process problem
1.129     paf       427:                connection->rollback();
                    428:                connection->close();
                    429:                rethrow;
1.1       paf       430:        }
                    431: }
                    432: 
1.41      parser    433: #ifndef DOXYGEN
1.129     paf       434: class Switch_data: public PA_Object {
                    435: public:
                    436:        Request& r;
1.181     misha     437:        const String* searching_string;
                    438:        double searching_double;
1.207     misha     439:        bool searching_bool;
1.129     paf       440:        Value* found;
                    441:        Value* _default;
                    442: public:
                    443:        Switch_data(Request& ar, Value& asearching): 
1.181     misha     444:                r(ar)
                    445:        {
                    446:                if(asearching.is_string() || asearching.is_void()){
                    447:                        searching_string=&asearching.as_string();
                    448:                        searching_double=0;
1.207     misha     449:                        searching_bool=false;
1.181     misha     450:                } else {
                    451:                        searching_string=0;
                    452:                        searching_double=asearching.as_double();
1.207     misha     453:                        searching_bool=asearching.is_bool();
1.181     misha     454:                }
                    455:        }
1.28      parser    456: };
1.41      parser    457: #endif
1.129     paf       458: static void _switch(Request& r, MethodParams& params) {
                    459:        Switch_data* data=new Switch_data(r, r.process_to_value(params[0]));
1.135     paf       460:        Temp_hash_value<const String::Body, void*> 
1.129     paf       461:                switch_data_setter(r.classes_conf, switch_data_name, data);
1.28      parser    462: 
1.129     paf       463:        Value& cases_code=params.as_junction(1, "switch cases must be code");
1.82      paf       464:        // execution of found ^case[...]{code} must be in context of ^switch[...]{code}
                    465:        // because of stacked WWrapper used there as wcontext
1.84      paf       466:        r.process(cases_code, true/*intercept_string*/);
1.147     paf       467:        if(Value* selected_code=data->found? data->found: data->_default)
1.107     paf       468:                r.write_pass_lang(r.process(*selected_code));
1.28      parser    469: }
                    470: 
1.129     paf       471: static void _case(Request& r, MethodParams& params) {
                    472:        Switch_data* data=static_cast<Switch_data*>(r.classes_conf.get(switch_data_name));
1.38      parser    473:        if(!data)
1.166     misha     474:                throw Exception(PARSER_RUNTIME,
1.129     paf       475:                        0,
1.38      parser    476:                        "without switch");
1.28      parser    477: 
1.181     misha     478:        if(data->found) // matches already was found
                    479:                return;
                    480: 
1.129     paf       481:        int count=params.count();
1.184     misha     482:        Value* code=&params.as_expression(--count, "case result must be code");
                    483: 
                    484: #ifdef USE_DESTRUCTORS
                    485:        Junction *j=code->get_junction();
                    486:        if (j){
                    487:                code=new VJunction(j->self,j->method,j->method_frame,j->rcontext,j->wcontext,j->code);
                    488:                if (j->wcontext) j->wcontext->attach_junction((VJunction *)code);
                    489:        }
                    490: #endif
1.83      paf       491:        
1.181     misha     492:        for(int i=0; i<count; i++){
1.129     paf       493:                Value& value=r.process_to_value(params[i]);
1.28      parser    494: 
1.181     misha     495:                if(value.is_string() && value.as_string() == CASE_DEFAULT_VALUE){
1.184     misha     496:                        data->_default=code;
1.181     misha     497:                        continue;
1.28      parser    498:                }
                    499: 
                    500:                bool matches;
1.181     misha     501:                if(data->searching_string)
                    502:                        matches=(*data->searching_string) == value.as_string();
1.207     misha     503:                else if(data->searching_bool || value.is_bool())
                    504:                        matches=(data->searching_double != 0) == value.as_bool();
1.28      parser    505:                else
1.181     misha     506:                        matches=data->searching_double == value.as_double();
1.82      paf       507: 
1.181     misha     508:                if(matches){
1.184     misha     509:                        data->found=code;
1.28      parser    510:                        break;
                    511:                }
                    512:        }
                    513: }
1.136     paf       514: #ifndef DOXYGEN
                    515: struct Try_catch_result {
                    516:        StringOrValue processed_code;
                    517:        const String* exception_should_be_handled;
                    518: 
                    519:        Try_catch_result(): exception_should_be_handled(0) {}
                    520: };
                    521: #endif
                    522: 
                    523: /// used by ^try and ^cache, @returns $exception.handled[string] if any
                    524: template<class I>
                    525: static Try_catch_result try_catch(Request& r, 
                    526:                  StringOrValue body_code(Request&, I), I info, 
1.179     misha     527:                  Value* catch_code,
                    528:                  bool could_be_handled_by_caller=false) 
1.136     paf       529: {
                    530:        Try_catch_result result;
1.179     misha     531: 
1.136     paf       532:        if(!catch_code) {
                    533:                result.processed_code=body_code(r, info);
                    534:                return result;
                    535:        }
                    536: 
                    537:        // taking snapshot of try-context
                    538:        Request_context_saver try_context(r);
                    539:        try {
                    540:                result.processed_code=body_code(r, info);
                    541:        } catch(const Exception& e) {
                    542:                Request_context_saver throw_context(r); // taking snapshot of throw-context [stack trace contains error]
                    543:                Request::Exception_details details=r.get_details(e);
                    544:                try_context.restore(); // restoring try-context to perform catch-code
                    545: 
                    546:                Junction* junction=catch_code->get_junction();
                    547:                Value* method_frame=junction->method_frame;
1.197     misha     548:                Value* saved_exception_var_value=method_frame->get_element(exception_var_name);
1.154     paf       549:                VMethodFrame& frame=*junction->method_frame;
1.197     misha     550:                frame.put_element(exception_var_name, &details.vhash, false);
1.179     misha     551: 
1.136     paf       552:                result.processed_code=r.process(*catch_code);
                    553:                
                    554:                // retriving $exception.handled, restoring $exception var
                    555:                Value* vhandled=details.vhash.hash().get(exception_handled_part_name);
1.197     misha     556:                frame.put_element(exception_var_name, saved_exception_var_value, false);
1.136     paf       557: 
                    558:                bool bhandled=false;
                    559:                if(vhandled) {
                    560:                        if(vhandled->is_string()) { // not simple $exception.handled(1/0)?
1.160     paf       561:                                if(could_be_handled_by_caller) { // and we can possibly handle it
                    562:                                        result.exception_should_be_handled=vhandled->get_string(); // considering 'recovered' and let the caller recover
                    563:                                        return result;
                    564:                                }
                    565:                                
                    566:                                bhandled=false;
                    567:                        } else
1.179     misha     568:                                bhandled=vhandled->as_bool();
1.136     paf       569:                }
                    570: 
                    571:                if(!bhandled) {
                    572:                        throw_context.restore(); // restoring throw-context [exception were not handled]
                    573:                        rethrow;
                    574:                }
                    575:        }
1.179     misha     576: 
1.136     paf       577:        return result;
                    578: }
1.28      parser    579: 
1.63      paf       580: // cache--
                    581: 
                    582: // consts
                    583: 
1.152     paf       584: const int DATA_STRING_SERIALIZED_VERSION=0x0006;
1.63      paf       585: 
                    586: // helper types
                    587: 
                    588: #ifndef DOXYGEN
                    589: struct Data_string_serialized_prolog {
                    590:        int version;
1.79      paf       591:        time_t expires;
1.63      paf       592: };
                    593: #endif
                    594: 
1.68      paf       595: void cache_delete(const String& file_spec) {
1.167     misha     596:        file_delete(file_spec, false/*fail_on_problem*/);
1.63      paf       597: }
1.69      paf       598: 
                    599: #ifndef DOXYGEN
1.135     paf       600: struct Cache_scope {
1.129     paf       601: public:
1.79      paf       602:        time_t expires;
1.135     paf       603:        const String* body_from_disk;
1.79      paf       604: };
1.69      paf       605: struct Locked_process_and_cache_put_action_info {
                    606:        Request *r;
1.136     paf       607:        Cache_scope *scope;
1.167     misha     608:        Value* body_code;
                    609:        Value* catch_code; 
1.136     paf       610:        const String* processed_code;
1.69      paf       611: };
                    612: #endif
1.136     paf       613: 
                    614: 
                    615: 
                    616: static StringOrValue process_cache_body_code(Request& r, Value* body_code) {
1.153     paf       617:        return StringOrValue(r.process_to_string(*body_code));
1.136     paf       618: }
                    619: 
1.129     paf       620: /* @todo maybe network order worth spending some effort?
                    621:        don't bothering myself with network byte order,
                    622:        am not planning to be able to move resulting file across platforms
                    623: */
1.69      paf       624: static void locked_process_and_cache_put_action(int f, void *context) {
                    625:        Locked_process_and_cache_put_action_info& info=
                    626:                *static_cast<Locked_process_and_cache_put_action_info *>(context);
1.129     paf       627: 
1.160     paf       628:        const String* body_from_disk=info.scope->body_from_disk;
1.69      paf       629:        // body->process 
1.136     paf       630:        Try_catch_result result=try_catch(*info.r, 
                    631:                process_cache_body_code, info.body_code,
1.160     paf       632:                info.catch_code, body_from_disk!=0 /*we have something old=we can handle=recover later*/);
1.136     paf       633: 
                    634:        if(result.exception_should_be_handled) {
                    635:                if(*result.exception_should_be_handled==CACHE_EXCEPTION_HANDLED_CACHE_NAME) {
1.160     paf       636:                        assert(body_from_disk);
                    637:                        info.processed_code=body_from_disk;
1.136     paf       638:                } else
1.166     misha     639:                        throw Exception(PARSER_RUNTIME,
1.136     paf       640:                                result.exception_should_be_handled,
                    641:                                "$"EXCEPTION_VAR_NAME"."EXCEPTION_HANDLED_PART_NAME" value must be "
                    642:                                "either boolean or string '"CACHE_EXCEPTION_HANDLED_CACHE_NAME"'");
                    643:        } else
                    644:                info.processed_code=&result.processed_code.as_string();
1.69      paf       645: 
1.79      paf       646:        // expiration time not spoiled by ^cache(0) or something?
1.136     paf       647:        if(info.scope->expires > time(0)) {
1.79      paf       648:                // string -serialize> buffer
1.136     paf       649:                String::Cm serialized=info.processed_code->serialize(
1.129     paf       650:                        sizeof(Data_string_serialized_prolog));
1.79      paf       651:                Data_string_serialized_prolog& prolog=
1.129     paf       652:                        *reinterpret_cast<Data_string_serialized_prolog *>(serialized.str);
1.79      paf       653:                prolog.version=DATA_STRING_SERIALIZED_VERSION;
1.136     paf       654:                prolog.expires=info.scope->expires;
1.79      paf       655:                
                    656:                // buffer -write> file
1.129     paf       657:                write(f, serialized.str, serialized.length);
1.79      paf       658:        } else // expired!
1.136     paf       659:                info.scope->expires=0; // flag it so that could be easily checked by caller
1.69      paf       660: }
1.129     paf       661: const String* locked_process_and_cache_put(Request& r, 
1.128     paf       662:                                           Value& body_code,
1.136     paf       663:                                           Value* catch_code,
                    664:                                           Cache_scope& scope,
                    665:                                           const String& file_spec) 
                    666: {
1.140     paf       667:        Locked_process_and_cache_put_action_info info={&r, &scope, &body_code, catch_code, 0};
1.69      paf       668: 
1.129     paf       669:        const String* result=file_write_action_under_lock(
1.69      paf       670:                file_spec, 
1.168     misha     671:                "cache_put",
                    672:                locked_process_and_cache_put_action,
                    673:                &info,
1.69      paf       674:                false/*as_text*/,
                    675:                false/*do_append*/,
1.173     misha     676:                false/*block == don't wait till other thread release lock*/,
                    677:                false/*dun throw exception if lock failed*/) ? info.processed_code: 0;
1.168     misha     678: 
1.100     paf       679:        time_t now=time(0);
1.136     paf       680:        if(scope.expires<=now)
1.79      paf       681:                cache_delete(file_spec);
                    682:        return result;
1.63      paf       683: }
1.137     paf       684: #ifndef DOXYGEN
1.135     paf       685: struct Cache_get_result {
                    686:        const String* body;
                    687:        bool expired;
1.137     paf       688: };
                    689: #endif
                    690: static Cache_get_result cache_get(Request_charsets& charsets, const String& file_spec, time_t now) {
1.140     paf       691:        Cache_get_result result={0, false};
1.135     paf       692: 
1.129     paf       693:        File_read_result file=file_read(charsets, file_spec, 
1.63      paf       694:                           false/*as_text*/, 
1.129     paf       695:                           0, //no params
                    696:                           false/*fail_on_read_problem*/);
                    697:        if(file.success && file.length/* ignore reads which are empty due to 
1.72      paf       698:                        non-unary open+lockEX conflict with lockSH */) {
1.129     paf       699:                        
1.72      paf       700:                Data_string_serialized_prolog& prolog=
1.129     paf       701:                        *reinterpret_cast<Data_string_serialized_prolog *>(file.str);
1.63      paf       702: 
1.135     paf       703:                String* body=new String;
1.72      paf       704:                if(
1.129     paf       705:                        file.length>=sizeof(Data_string_serialized_prolog)
1.135     paf       706:                        && prolog.version==DATA_STRING_SERIALIZED_VERSION) {
                    707:                        if(body->deserialize(sizeof(Data_string_serialized_prolog),  file.str, file.length)) {
                    708:                                result.body=body;
                    709:                                result.expired=prolog.expires <= now;
                    710:                        }
                    711:                }
1.72      paf       712:        }
1.63      paf       713: 
1.135     paf       714:        return result;
1.63      paf       715: }
1.129     paf       716: static time_t as_expires(Request& r, MethodParams& params, 
1.79      paf       717:                                                int index, time_t now) {
                    718:        time_t result;
1.197     misha     719:        if(Value* vdate=params[index].as(VDATE_TYPE))
1.129     paf       720:                result=static_cast<VDate*>(vdate)->get_time();
1.79      paf       721:        else
1.129     paf       722:                result=now+(time_t)params.as_double(index, "lifespan must be date or number", r);
1.79      paf       723:        
                    724:        return result;
                    725: }
1.129     paf       726: static const String& as_file_spec(Request& r, MethodParams& params, int index) {
                    727:        return r.absolute(params.as_string(index, "filespec must be string"));
1.79      paf       728: }
1.129     paf       729: static void _cache(Request& r, MethodParams& params) {
1.173     misha     730:        if(params.count()==0) {
                    731:                // ^cache[] -- return current expiration time
1.158     paf       732:                Cache_scope* scope=static_cast<Cache_scope*>(r.classes_conf.get(cache_data_name));
                    733:                if(!scope)
1.166     misha     734:                        throw Exception(PARSER_RUNTIME,
1.158     paf       735:                                0,
                    736:                                "expire-time get without cache");
                    737:                r.write_no_lang(*new VDate(scope->expires));
                    738:                return;
                    739:        }
                    740: 
1.79      paf       741:        time_t now=time(0);
                    742: 
1.129     paf       743:        if(params.count()==1) {
1.173     misha     744:                // ^cache[filename] ^cache(seconds) ^cache[expires date]
1.129     paf       745:                if(params[0].is_string()) { // filename?
1.79      paf       746:                        cache_delete(as_file_spec(r, params, 0));
                    747:                        return;
                    748:                }
                    749: 
                    750:                // secods|expires date
1.135     paf       751:                Cache_scope* scope=static_cast<Cache_scope*>(r.classes_conf.get(cache_data_name));
                    752:                if(!scope)
1.166     misha     753:                        throw Exception(PARSER_RUNTIME,
1.129     paf       754:                                0,
1.79      paf       755:                                "expire-time reducing instruction without cache");
                    756:                
1.129     paf       757:                time_t expires=as_expires(r, params, 0, now);
1.135     paf       758:                if(expires < scope->expires)
                    759:                        scope->expires=expires;
1.79      paf       760: 
                    761:                return;
1.149     paf       762:        } else if(params.count()<3)
1.166     misha     763:                throw Exception(PARSER_RUNTIME,
1.149     paf       764:                        0,
                    765:                        "invalid number of parameters"); 
1.63      paf       766:        
                    767:        // file_spec, expires, body code
1.172     misha     768:        const String& file_spec=as_file_spec(r, params, 0);
1.65      paf       769: 
1.141     paf       770:        Cache_scope scope={as_expires(r, params, 1, now), 0};
1.135     paf       771: 
                    772:        Temp_hash_value<const String::Body, void*> 
                    773:                cache_scope_setter(r.classes_conf, cache_data_name, &scope);
1.136     paf       774:        Value& body_code=params.as_junction(2, "body_code must be code");
                    775:        Value* catch_code=0;
                    776:        if(params.count()>3)
                    777:                catch_code=&params.as_junction(3, "catch_code must be code");
1.63      paf       778: 
1.168     misha     779:        if(scope.expires>now) {
                    780:                Cache_get_result cached=cache_get(r.charsets, file_spec, now);
1.69      paf       781: 
1.168     misha     782:                if(cached.body) { // have cached copy
1.174     misha     783:                        if(cached.expired) {
                    784:                                scope.body_from_disk=cached.body; // storing for user to retrive it with ^cache[]
                    785:                        } else {
                    786:                                // and it's not expired yet write it out 
1.168     misha     787:                                r.write_assign_lang(*cached.body);
                    788:                                // happy with it
                    789:                                return;
                    790:                        }
                    791:                }
                    792: 
                    793:                // no cached info or it's already expired
1.173     misha     794: 
                    795:                // trying to process it under lock and store result in file
1.174     misha     796:                const String* processed_body=locked_process_and_cache_put(r, body_code, catch_code, scope, file_spec);
1.173     misha     797:                if(processed_body){
1.174     misha     798:                        // write it out 
                    799:                        r.write_assign_lang(*processed_body);
                    800:                        // happy with it
                    801:                        return;
1.173     misha     802:                } else {
                    803:                        // we fail while get exclusive lock. nvm, we just execute body_code a bit later
1.174     misha     804:                }
1.69      paf       805:        } else { 
1.79      paf       806:                // instructed not to cache; forget cached copy
1.68      paf       807:                cache_delete(file_spec);
1.69      paf       808:        }
1.168     misha     809:        
                    810:        // process without cacheing
                    811:        const String& processed_body=r.process_to_string(body_code);
                    812:        // write it out 
                    813:        r.write_assign_lang(processed_body);
1.63      paf       814: }
                    815: 
1.136     paf       816: static StringOrValue process_try_body_code(Request& r, Value* body_code) {
                    817:        return r.process(*body_code);
                    818: }
1.129     paf       819: static void _try_operator(Request& r, MethodParams& params) {
                    820:        Value& body_code=params.as_junction(0, "body_code must be code");
                    821:        Value& catch_code=params.as_junction(1, "catch_code must be code");
1.179     misha     822:        Value* finally_code=(params.count()==3) ? &params.as_junction(2, "finally_code must be code") : 0;
                    823: 
                    824:        Try_catch_result result;
                    825:        StringOrValue finally_result;
                    826:        try{
                    827:                result=try_catch(r, 
                    828:                        process_try_body_code, &body_code,
                    829:                        &catch_code);
                    830:                if(result.exception_should_be_handled)
                    831:                        throw Exception(PARSER_RUNTIME,
                    832:                                result.exception_should_be_handled,
                    833:                                "catch block must set $exception.handled to some boolean value, not string");
                    834:        } catch(...){
                    835:                if(finally_code)
                    836:                        finally_result=r.process(*finally_code);
                    837:                rethrow;
                    838:        }
1.74      paf       839: 
1.179     misha     840:        if(finally_code)
                    841:                finally_result=r.process(*finally_code);
1.123     paf       842: 
1.136     paf       843:        // write out processed body_code or catch_code
                    844:        r.write_pass_lang(result.processed_code);
1.179     misha     845: 
                    846:        // write out processed finally code
                    847:        if(finally_code)
                    848:                r.write_pass_lang(finally_result);
1.74      paf       849: }
                    850: 
1.138     paf       851: static void _throw_operator(Request&, MethodParams& params) {
1.178     misha     852:        if(params.count()==1 && !params[0].is_string()) {
1.129     paf       853:                if(HashStringValue *hash=params[0].get_hash()) {
                    854:                        const char* type=0;
                    855:                        if(Value* value=hash->get(exception_type_part_name))
1.78      paf       856:                                type=value->as_string().cstr();
1.129     paf       857:                        const String* source=0;
                    858:                        if(Value* value=hash->get(exception_source_part_name))
1.74      paf       859:                                source=&value->as_string();
1.138     paf       860:                        const char* comment=0;
1.129     paf       861:                        if(Value* value=hash->get(exception_comment_part_name))
1.74      paf       862:                                comment=value->as_string().cstr();
                    863: 
1.78      paf       864:                        throw Exception(type,
1.129     paf       865:                                source?source:0,
                    866:                                "%s", comment?comment:"");
1.74      paf       867:                } else
1.166     misha     868:                        throw Exception(PARSER_RUNTIME,
1.129     paf       869:                                0,
1.178     misha     870:                                "one-param version has hash or string param");
1.74      paf       871:        } else {
1.129     paf       872:                const char* type=params.as_string(0, "type must be string").cstr();
1.178     misha     873:                const String* source=params.count()>1? &params.as_string(1, "source must be string"):0;
                    874:                const char* comment=params.count()>2? params.as_string(2, "comment must be string").cstr():0;
                    875:                throw Exception(type, source, "%s", comment?comment:"");
1.74      paf       876:        }
1.132     paf       877:  }
1.129     paf       878: 
1.150     paf       879: static void _sleep_operator(Request& r, MethodParams& params) {
                    880:        double seconds=params.as_double(0, "seconds must be double", r);
1.206     moko      881:        if(seconds>0)
                    882:                pa_sleep((int)trunc(seconds), (int)trunc((seconds-trunc(seconds))*1000000));
1.150     paf       883:  }
                    884: 
1.129     paf       885: #if defined(WIN32) && defined(_DEBUG)
                    886: #      define PA_BPT
1.138     paf       887: static void _bpt(Request&, MethodParams&) {
1.129     paf       888:        _asm int 3;
                    889: }
                    890: #endif
                    891: 
1.10      paf       892: // constructor
                    893: 
1.129     paf       894: VClassMAIN::VClassMAIN(): VClass() {
                    895:        set_name(*new String(MAIN_CLASS_NAME));
                    896: 
                    897: #ifdef PA_BPT
                    898:        // ^bpt[]
                    899:        add_native_method("bpt", Method::CT_ANY, _bpt, 0, 0);
                    900: #endif
1.121     paf       901: 
1.1       paf       902:        // ^if(condition){code-when-true}
                    903:        // ^if(condition){code-when-true}{code-when-false}
1.200     moko      904:        // ^if(condition){code-when-true} (another condition){code-when-true} ... {code-when-false}
                    905:        add_native_method("if", Method::CT_ANY, _if, 2, 10000, Method::CO_WITHOUT_FRAME);
1.1       paf       906: 
1.194     misha     907:        // ^untaint[as-is|uri|sql|js|html|html-typo|regex|parser-code]{code}
                    908:        add_native_method("untaint", Method::CT_ANY, _untaint, 1, 2, Method::CO_WITHOUT_FRAME);
1.1       paf       909: 
1.194     misha     910:        // ^taint[as-is|uri|sql|js|html|html-typo|regex|parser-code]{code}
                    911:        add_native_method("taint", Method::CT_ANY, _taint, 1, 2, Method::CO_WITHOUT_FRAME);
1.1       paf       912: 
1.205     moko      913:        // ^apply-taint[untaint lang][string]
                    914:        add_native_method("apply-taint", Method::CT_ANY, _apply_taint, 1, 2, Method::CO_WITHOUT_FRAME);
                    915: 
1.1       paf       916:        // ^process[code]
1.146     paf       917:        add_native_method("process", Method::CT_ANY, _process, 1, 3);
1.1       paf       918: 
                    919:        // ^rem{code}
1.185     misha     920:        add_native_method("rem", Method::CT_ANY, _rem, 1, 10000, Method::CO_WITHOUT_FRAME);
1.1       paf       921: 
                    922:        // ^while(condition){code}
1.185     misha     923:        add_native_method("while", Method::CT_ANY, _while, 2, 3, Method::CO_WITHOUT_FRAME);
1.1       paf       924: 
                    925:        // ^use[file]
1.10      paf       926:        add_native_method("use", Method::CT_ANY, _use, 1, 1);
1.1       paf       927: 
1.164     paf       928:        // ^break[]
1.185     misha     929:        add_native_method("break", Method::CT_ANY, _break, 0, 0, Method::CO_WITHOUT_FRAME);
                    930: 
1.164     paf       931:        // ^continue[]
1.185     misha     932:        add_native_method("continue", Method::CT_ANY, _continue, 0, 0, Method::CO_WITHOUT_FRAME);
                    933: 
1.54      paf       934:        // ^for[i](from-number;to-number-inclusive){code}[delim]
1.185     misha     935:        add_native_method("for", Method::CT_ANY, _for, 3+1, 3+1+1, Method::CO_WITHOUT_WCONTEXT);
1.1       paf       936: 
                    937:        // ^eval(expr)
                    938:        // ^eval(expr)[format]
1.185     misha     939:        add_native_method("eval", Method::CT_ANY, _eval, 1, 2, Method::CO_WITHOUT_FRAME);
1.52      parser    940: 
1.1       paf       941:        // ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s}
1.10      paf       942:        add_native_method("connect", Method::CT_ANY, _connect, 2, 2);
                    943: 
1.63      paf       944: 
1.136     paf       945:        // ^cache[file_spec](time){code}[{catch code}] time=0 no cache
1.65      paf       946:        // ^cache[file_spec] delete cache
1.161     paf       947:        // ^cache[] get current expiration time
                    948:        add_native_method("cache", Method::CT_ANY, _cache, 0, 4);
1.63      paf       949:        
1.28      parser    950:        // switch
                    951: 
                    952:        // ^switch[value]{cases}
1.185     misha     953:        add_native_method("switch", Method::CT_ANY, _switch, 2, 2, Method::CO_WITHOUT_FRAME);
1.28      parser    954: 
                    955:        // ^case[value]{code}
1.185     misha     956:        add_native_method("case", Method::CT_ANY, _case, 2, 10000, Method::CO_WITHOUT_FRAME);
1.74      paf       957: 
                    958:        // try-catch
                    959: 
                    960:        // ^try{code}{catch code}
1.185     misha     961:        add_native_method("try", Method::CT_ANY, _try_operator, 2, 3, Method::CO_WITHOUT_FRAME);
1.74      paf       962:        // ^throw[$exception hash]
                    963:        // ^throw[type;source;comment]
                    964:        add_native_method("throw", Method::CT_ANY, _throw_operator, 1, 3);
                    965: 
1.150     paf       966:        add_native_method("sleep", Method::CT_ANY, _sleep_operator, 1, 1);
1.10      paf       967: }
1.11      paf       968: 
                    969: // constructor & configurator
1.1       paf       970: 
1.129     paf       971: VStateless_class& VClassMAIN_create() {
                    972:        return *new VClassMAIN;
1.1       paf       973: }

E-mail: