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

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

E-mail: