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

1.1       paf         1: /** @file
1.9       paf         2:        Parser: parser @b operators.
1.1       paf         3: 
1.70      paf         4:        Copyright (c) 2001, 2002 ArtLebedev Group (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.118   ! paf         8: static const char* IDENT_OP_C="$Date: 2002/10/15 10:09:51 $";
1.1       paf         9: 
1.13      paf        10: #include "classes.h"
1.1       paf        11: #include "pa_common.h"
                     12: #include "pa_request.h"
                     13: #include "pa_vint.h"
                     14: #include "pa_sql_connection.h"
1.79      paf        15: #include "pa_vdate.h"
1.105     paf        16: #include "pa_vmethod_frame.h"
1.1       paf        17: 
1.18      parser     18: // limits
                     19: 
                     20: #define MAX_LOOPS 10000
                     21: 
1.10      paf        22: // defines
                     23: 
1.82      paf        24: #define CASE_DEFAULT_VALUE "DEFAULT"
1.11      paf        25: 
1.10      paf        26: // class
                     27: 
1.113     paf        28: class VClassMAIN: public VClass {
1.10      paf        29: public:
1.113     paf        30:        VClassMAIN(Pool& apool);
1.10      paf        31: };
                     32: 
                     33: // methods
                     34: 
1.5       paf        35: static void _if(Request& r, const String&, MethodParams *params) {
1.48      parser     36:        Value& condition_code=params->as_junction(0, "condition must be expression");
1.1       paf        37: 
1.81      paf        38:        bool condition=r.process_to_value(condition_code, 
1.83      paf        39:                /*0/*no name* /,*/
1.1       paf        40:                false/*don't intercept string*/).as_bool();
1.6       paf        41:        if(condition)
1.85      paf        42:                r.write_pass_lang(r.process(params->as_junction(1, "'then' parameter must be code")));
1.50      parser     43:        else if(params->size()>2)
1.85      paf        44:                r.write_pass_lang(r.process(params->as_junction(2, "'else' parameter must be code")));
1.1       paf        45: }
                     46: 
1.5       paf        47: static void _untaint(Request& r, const String& method_name, MethodParams *params) {
1.1       paf        48:        Pool& pool=r.pool();
                     49: 
1.60      paf        50:        uchar lang;
1.59      paf        51:        if(params->size()==1)
                     52:                lang=String::UL_AS_IS; // mark as simply 'tainted'. useful in html from sql 
                     53:        else {
                     54:                const String& lang_name=params->as_string(0, "lang must be string");
1.60      paf        55:                lang=untaint_lang_name2enum->get_int(lang_name);
1.59      paf        56:                if(!lang)
1.78      paf        57:                        throw Exception(0,
1.59      paf        58:                                &lang_name,
                     59:                                "invalid taint language");
                     60:        }
1.1       paf        61: 
                     62:        {
1.59      paf        63:                Value& vbody=params->as_junction(params->size()-1, "body must be code");
1.1       paf        64:                
                     65:                Temp_lang temp_lang(r, lang); // set temporarily specified ^untaint[language;
1.86      paf        66:                r.write_pass_lang(r.process(vbody)); // process marking tainted with that lang
1.1       paf        67:        }
                     68: }
                     69: 
1.5       paf        70: static void _taint(Request& r, const String&, MethodParams *params) {
1.1       paf        71:        Pool& pool=r.pool();
                     72: 
1.60      paf        73:        uchar lang;
1.1       paf        74:        if(params->size()==1)
                     75:                lang=String::UL_TAINTED; // mark as simply 'tainted'. useful in table:set
                     76:        else {
1.48      parser     77:                const String& lang_name=params->as_string(0, "lang must be string");
1.60      paf        78:                lang=untaint_lang_name2enum->get_int(lang_name);
1.1       paf        79:                if(!lang)
1.78      paf        80:                        throw Exception(0,
1.3       paf        81:                                &lang_name,
                     82:                                "invalid taint language");
1.1       paf        83:        }
                     84: 
                     85:        {
1.34      parser     86:                Value& vbody=params->as_no_junction(params->size()-1, "body must not be code");
1.1       paf        87:                
1.90      paf        88:                String& result=*new(pool) String(pool);
1.1       paf        89:                result.append(
                     90:                        vbody.as_string(),  // process marking tainted with that lang
                     91:                        lang, true);  // force result language to specified
                     92:                r.write_pass_lang(result);
                     93:        }
                     94: }
                     95: 
1.5       paf        96: static void _process(Request& r, const String& method_name, MethodParams *params) {
1.97      paf        97:        Pool& pool=r.pool();
1.73      paf        98:        const Method *main_method;
1.118   ! paf        99:        VStateless_class *target_class;
1.1       paf       100:        {
1.116     paf       101:                Value& vjunction=params->as_junction(params->size()-1, "body must be code");
1.114     paf       102: 
1.116     paf       103:                Value *vtarget;
                    104:                vtarget=params->size()>1?
                    105:                        &params->as_no_junction(0, "target must not be code")
                    106:                        :r.get_method_frame()->caller();
                    107:                if(!vtarget)
                    108:                        throw Exception("parser.runtime",
1.114     paf       109:                                &method_name,
1.116     paf       110:                                "no caller"); // can't be, somebody really called ^process
1.114     paf       111: 
1.118   ! paf       112:                target_class=vtarget->get_class();
1.116     paf       113:                if(!target_class)
                    114:                        throw Exception("parser.runtime",
                    115:                                &method_name,
                    116:                                "no target class");
                    117:                
1.114     paf       118:                // evaluate source to process
                    119:                const String& source=r.process_to_string(vjunction);
                    120: 
1.73      paf       121:                // temporary remove language change
                    122:                Temp_lang temp_lang(r, String::UL_PASS_APPENDED);
1.1       paf       123:                // temporary zero @main so to maybe-replace it in processed code
1.116     paf       124:                Temp_method temp_method_main(*target_class, r.main_method_name, 0);
1.1       paf       125:                // temporary zero @auto so it wouldn't be auto-called in Request::use_buf
1.116     paf       126:                Temp_method temp_method_auto(*target_class, *auto_method_name, 0);
1.1       paf       127:                
1.101     paf       128:                // calculate pseudo file name of processed chars
                    129:                // would be something like "/some/file(4) process"
                    130:                char local_place[MAX_STRING];
                    131: #ifndef NO_STRING_ORIGIN
                    132:                const Origin& source_origin=source.origin();
                    133:                const Origin& method_origin=method_name.origin();
                    134:                size_t place_size;
                    135:                if(source_origin.file==method_origin.file)
                    136:                        place_size=snprintf(local_place, MAX_STRING, "%s(%d) %s",  // same file
                    137:                                source_origin.file?source_origin.file:"unknown_file", 1+source_origin.line, 
                    138:                                method_name.cstr())+1;
                    139:                else // different files ^process{external__file_text__or__sql}
                    140:                        place_size=snprintf(local_place, MAX_STRING, "%s", 
                    141:                                source_origin.file?source_origin.file:"unknown_file")+1;
                    142: #else
                    143:                strncpy(local_place, method_name.cstr(), MAX_STRING-1); place[MAX_STRING-1]=0;
                    144:                size_t place_size=strlen(local_place)+1;
                    145: #endif
                    146:                char *heap_place=(char *)r.malloc(place_size);
                    147:                memcpy(heap_place, local_place, place_size);
1.1       paf       148: 
                    149:                // process source code, append processed methods to 'self' class
                    150:                // maybe-define new @main
1.116     paf       151:                r.use_buf(*target_class,
1.67      paf       152:                        source.cstr(String::UL_UNSPECIFIED, r.connection(0)), 
1.97      paf       153:                        *new(pool) String(pool, heap_place, place_size, true /*tainted*/),
1.113     paf       154:                        heap_place);
1.1       paf       155:                
1.73      paf       156:                // main_method
1.116     paf       157:                main_method=target_class->get_method(r.main_method_name);
1.73      paf       158:        }
                    159:        // after restoring current-request-lang
                    160:        // maybe-execute @main[]
                    161:        if(main_method) {
1.118   ! paf       162:                // temporarily set request's self to target_class
        !           163:                Temp_request_self trs(r, *target_class);
        !           164:                // temporarily set method_frame's self to target_class
        !           165:                Temp_method_frame_self tmfs(*r.get_method_frame(), *target_class);
1.73      paf       166:                // execute!     
                    167:                r.execute(*main_method->parser_code);
1.1       paf       168:        }
                    169: }
                    170:        
1.5       paf       171: static void _rem(Request& r, const String&, MethodParams *params) {
1.34      parser    172:        params->as_junction(0, "body must be code");
1.1       paf       173: }
                    174: 
1.5       paf       175: static void _while(Request& r, const String& method_name, MethodParams *params) {
1.1       paf       176:        Pool& pool=r.pool();
                    177: 
1.34      parser    178:        Value& vcondition=params->as_junction(0, "condition must be expression");
                    179:        Value& body=params->as_junction(1, "body must be code");
1.1       paf       180: 
                    181:        // while...
                    182:        int endless_loop_count=0;
                    183:        while(true) {
1.18      parser    184:                if(++endless_loop_count>=MAX_LOOPS) // endless loop?
1.78      paf       185:                        throw Exception("parser.runtime",
1.1       paf       186:                                &method_name,
                    187:                                "endless loop detected");
                    188: 
1.86      paf       189:                bool condition=r.process_to_value(vcondition, 
1.83      paf       190:                                /*0/*no name* /,*/
1.1       paf       191:                                false/*don't intercept string*/).as_bool();
                    192:                if(!condition) // ...condition is true
                    193:                        break;
                    194: 
                    195:                // write processed body
1.86      paf       196:                r.write_pass_lang(r.process(body));
1.1       paf       197:        }
                    198: }
                    199: 
1.5       paf       200: static void _use(Request& r, const String& method_name, MethodParams *params) {
1.34      parser    201:        Value& vfile=params->as_no_junction(0, "file name must not be code");
1.113     paf       202:        r.use_file(r.main_class, vfile.as_string());
1.1       paf       203: }
                    204: 
1.5       paf       205: static void _for(Request& r, const String& method_name, MethodParams *params) {
1.1       paf       206:        Pool& pool=r.pool();
1.48      parser    207:        const String& var_name=params->as_string(0, "var name must be string");
                    208:        int from=params->as_int(1, "from must be int", r);
                    209:        int to=params->as_int(2, "to must be int", r);
1.34      parser    210:        Value& body_code=params->as_junction(3, "body must be code");
1.50      parser    211:        Value *delim_maybe_code=params->size()>4?&params->get(4):0;
1.1       paf       212: 
1.57      paf       213:        if(to-from>=MAX_LOOPS) // too long loop?
1.78      paf       214:                throw Exception("parser.runtime",
1.57      paf       215:                        &method_name,
                    216:                        "endless loop detected");
                    217: 
1.1       paf       218:        bool need_delim=false;
                    219:        VInt *vint=new(pool) VInt(pool, 0);
1.117     paf       220:        r.get_method_frame()->caller()->put_element(var_name, vint, false);
1.1       paf       221:        for(int i=from; i<=to; i++) {
                    222:                vint->set_int(i);
                    223: 
1.108     paf       224:                StringOrValue sv_processed=r.process(body_code);
                    225:                const String *s_processed=sv_processed.get_string();
                    226:                if(delim_maybe_code && s_processed && s_processed->size()) { // delimiter set and we have body
                    227:                        if(need_delim) // need delim & iteration produced string?
1.86      paf       228:                                r.write_pass_lang(r.process(*delim_maybe_code));
1.1       paf       229:                        need_delim=true;
                    230:                }
1.108     paf       231:                r.write_pass_lang(sv_processed);
1.1       paf       232:        }
                    233: }
                    234: 
1.15      paf       235: static void _eval(Request& r, const String& method_name, MethodParams *params) {
1.34      parser    236:        Value& expr=params->as_junction(0, "need expression");
1.1       paf       237:        // evaluate expresion
1.91      paf       238:        Value *value_result=r.process_to_value(expr, 
1.83      paf       239:                /*0/*no name YET* /,*/
1.1       paf       240:                true/*don't intercept string*/).as_expr_result();
1.50      parser    241:        if(params->size()>1) {
1.34      parser    242:                Value& fmt=params->as_no_junction(1, "fmt must not be code");
1.1       paf       243: 
                    244:                Pool& pool=r.pool();
                    245:                String& string=*new(pool) String(pool);
1.91      paf       246:                string.APPEND_CONST(format(pool, value_result->as_double(), fmt.as_string().cstr()));
                    247:                r.write_no_lang(string);
                    248:        } else
                    249:                r.write_no_lang(*value_result);
1.1       paf       250: }
                    251: 
1.42      parser    252: static void _connect(Request& r, const String& method_name, MethodParams *params) {
1.1       paf       253:        Pool& pool=r.pool();
1.63      paf       254: #ifdef RESOURCES_DEBUG
                    255: struct timeval mt[2];
                    256: #endif
1.34      parser    257:        Value& url=params->as_no_junction(0, "url must not be code");
                    258:        Value& body_code=params->as_junction(1, "body must be code");
1.1       paf       259: 
1.113     paf       260:        Table *protocol2driver_and_client=0;
                    261:        if(Value *sql=r.main_class.get_element(*new(pool) String(pool, MAIN_SQL_NAME), &r.main_class, false)) {
                    262:                if(Value *element=sql->get_element(*new(pool) String(pool, MAIN_SQL_DRIVERS_NAME), sql, false)) {
                    263:                        protocol2driver_and_client=element->get_table();
                    264:                }
                    265:        }
1.11      paf       266: 
1.63      paf       267: #ifdef RESOURCES_DEBUG
                    268: //measure:before
                    269: gettimeofday(&mt[0],NULL);
                    270: #endif
1.1       paf       271:        // connect
1.67      paf       272:        SQL_Connection_ptr connection=SQL_driver_manager->get_connection(
1.42      parser    273:                url.as_string(), method_name, protocol2driver_and_client);
1.1       paf       274: 
1.63      paf       275: #ifdef RESOURCES_DEBUG
                    276: //measure:after connect
                    277: gettimeofday(&mt[1],NULL);
                    278: 
                    279: double t[2];
                    280: for(int i=0;i<2;i++)
                    281:     t[i]=mt[i].tv_sec+mt[i].tv_usec/1000000.0;
                    282: 
                    283: r.sql_connect_time+=t[1]-t[0];
                    284: #endif
1.67      paf       285:        Temp_connection temp_connection(r, connection.get());
1.1       paf       286:        // execute body
1.53      parser    287:        try {
1.86      paf       288:                r.write_assign_lang(r.process(body_code));
1.75      paf       289:        } catch(...) { // process problem
1.67      paf       290:                connection->mark_to_rollback();
                    291:                /*re*/throw; 
1.1       paf       292:        }
                    293: }
                    294: 
1.41      parser    295: #ifndef DOXYGEN
1.28      parser    296: struct Switch_data {
1.82      paf       297:        Request *r;
1.28      parser    298:        Value *searching;
                    299:        Value *found;
                    300:        Value *_default;
                    301: };
1.41      parser    302: #endif
1.28      parser    303: static void _switch(Request& r, const String&, MethodParams *params) {
1.82      paf       304:        Switch_data data={&r, &r.process_to_value(params->get(0))};     
1.79      paf       305:        Temp_hash_value switch_data_setter(r.classes_conf, *switch_data_name, &data);
1.28      parser    306: 
1.83      paf       307:        Value& cases_code=params->as_junction(1, "switch cases must be code");
1.82      paf       308:        // execution of found ^case[...]{code} must be in context of ^switch[...]{code}
                    309:        // because of stacked WWrapper used there as wcontext
1.84      paf       310:        r.process(cases_code, true/*intercept_string*/);
1.83      paf       311:        if(Value *selected_code=data.found ? data.found : data._default) {
                    312:                // setting code context, would execute in ^switch[...]{>>context<<}
1.105     paf       313:                //selected_code->get_junction()->change_context(cases_code.get_junction());
1.107     paf       314:                r.write_pass_lang(r.process(*selected_code));
1.83      paf       315:        }
1.28      parser    316: }
                    317: 
1.38      parser    318: static void _case(Request& r, const String& method_name, MethodParams *params) {
                    319:        Pool& pool=r.pool();
                    320: 
                    321:        Switch_data *data=static_cast<Switch_data *>(r.classes_conf.get(*switch_data_name));
                    322:        if(!data)
1.78      paf       323:                throw Exception("parser.runtime",
1.38      parser    324:                        &method_name,
                    325:                        "without switch");
1.28      parser    326: 
                    327:        int count=params->size();
1.34      parser    328:        Value *code=&params->as_junction(--count, "case result must be code");
1.83      paf       329:        
                    330:        // killing context for safety, would execute in ^switch[...]{>>context<<}
                    331:        // reason: context is stacked, and it would become invalid afterwards
1.105     paf       332:        //code->get_junction()->change_context(0);
1.83      paf       333: 
1.28      parser    334:        for(int i=0; i<count; i++) {
1.81      paf       335:                Value& value=r.process_to_value(params->get(i));
1.28      parser    336: 
1.82      paf       337:                if(value.as_string() == CASE_DEFAULT_VALUE) {
1.38      parser    338:                        data->_default=code;
1.28      parser    339:                        break;
                    340:                }
                    341: 
                    342:                bool matches;
1.38      parser    343:                if(data->searching->is_string())
                    344:                        matches=data->searching->as_string() == value.as_string();
1.28      parser    345:                else
1.38      parser    346:                        matches=data->searching->as_double() == value.as_double();
1.28      parser    347: 
                    348:                if(matches) {
1.82      paf       349:                        if(data->found)
                    350:                                throw Exception("parser.runtime",
                    351:                                        &method_name,
                    352:                                        "duplicate found");
                    353: 
1.38      parser    354:                        data->found=code;
1.28      parser    355:                        break;
                    356:                }
                    357:        }
                    358: }
                    359: 
1.63      paf       360: // cache--
                    361: 
                    362: // consts
                    363: 
1.79      paf       364: const int DATA_STRING_SERIALIZED_VERSION=0x0002;
1.63      paf       365: 
                    366: // helper types
                    367: 
                    368: #ifndef DOXYGEN
                    369: struct Data_string_serialized_prolog {
                    370:        int version;
1.79      paf       371:        time_t expires;
1.63      paf       372: };
                    373: #endif
                    374: 
1.68      paf       375: void cache_delete(const String& file_spec) {
                    376:        file_delete(file_spec, false/*fail_on_read_problem*/);
1.63      paf       377: }
1.69      paf       378: 
                    379: #ifndef DOXYGEN
1.79      paf       380: struct Cache_data {
                    381:        time_t expires;
                    382: };
1.69      paf       383: struct Locked_process_and_cache_put_action_info {
                    384:        Request *r;
1.79      paf       385:        Cache_data *data;
1.81      paf       386:        Value *body_code; const String *evaluated_body;
1.69      paf       387: };
                    388: #endif
                    389: static void locked_process_and_cache_put_action(int f, void *context) {
                    390:        Locked_process_and_cache_put_action_info& info=
                    391:                *static_cast<Locked_process_and_cache_put_action_info *>(context);
                    392:        
                    393:        // body->process 
1.81      paf       394:        info.evaluated_body=&info.r->process_to_string(*info.body_code);
1.69      paf       395: 
1.79      paf       396:        // expiration time not spoiled by ^cache(0) or something?
                    397:        if(info.data->expires > time(0)) {
                    398:                // string -serialize> buffer
                    399:                void *data; size_t data_size;
1.81      paf       400:                info.evaluated_body->serialize(
1.79      paf       401:                        sizeof(Data_string_serialized_prolog), 
                    402:                        data, data_size);
                    403:                Data_string_serialized_prolog& prolog=
                    404:                        *static_cast<Data_string_serialized_prolog *>(data);
                    405:                prolog.version=DATA_STRING_SERIALIZED_VERSION;
                    406:                prolog.expires=info.data->expires;
                    407:                
                    408:                // buffer -write> file
                    409:                write(f, data, data_size);
                    410:        } else // expired!
                    411:                info.data->expires=0; // flag it so that could be easily checked by caller
1.69      paf       412: }
1.81      paf       413: const String *locked_process_and_cache_put(Request& r, 
1.69      paf       414:                                                                        Value& body_code,
1.79      paf       415:                                                                        Cache_data& data,
1.69      paf       416:                                                                        const String& file_spec) {
                    417:        Locked_process_and_cache_put_action_info info={
                    418:                &r,
1.81      paf       419:                &data,
                    420:                &body_code
1.69      paf       421:        };
                    422: 
1.81      paf       423:        const String *result=file_write_action_under_lock(
1.69      paf       424:                file_spec, 
                    425:                "cache_put", locked_process_and_cache_put_action, &info,
                    426:                false/*as_text*/,
                    427:                false/*do_append*/,
1.96      paf       428:                false/*block*/,
                    429:                false/*fail on lock problem*/) ? info.evaluated_body: 0;
1.100     paf       430:        time_t now=time(0);
                    431:        if(data.expires<=now)
1.79      paf       432:                cache_delete(file_spec);
                    433:        return result;
1.63      paf       434: }
1.79      paf       435: String *cache_get(Pool& pool, const String& file_spec, time_t now) {
1.63      paf       436:        void* data; size_t data_size;
1.72      paf       437:        if(file_read(pool, file_spec, 
1.63      paf       438:                           data, data_size, 
                    439:                           false/*as_text*/, 
1.69      paf       440:                           false/*fail_on_read_problem*/)
1.72      paf       441:            && data_size/* ignore reads which are empty due to 
                    442:                        non-unary open+lockEX conflict with lockSH */) {
1.63      paf       443:        
1.72      paf       444:                Data_string_serialized_prolog& prolog=
                    445:                        *static_cast<Data_string_serialized_prolog *>(data);
1.63      paf       446: 
1.72      paf       447:                String *result=new(pool) String(pool);
                    448:                if(
                    449:                        data_size>=sizeof(Data_string_serialized_prolog)
                    450:                        && prolog.version==DATA_STRING_SERIALIZED_VERSION
1.79      paf       451:                        && prolog.expires > now
1.72      paf       452:                        && result->deserialize(
                    453:                                sizeof(Data_string_serialized_prolog),  data, data_size, file_spec.cstr()))
                    454:                        return result;
                    455:        }
1.63      paf       456: 
1.72      paf       457:        return 0;
1.63      paf       458: }
1.79      paf       459: static time_t as_expires(Request& r, const String& method_name, MethodParams *params, 
                    460:                                                int index, time_t now) {
                    461:        time_t result;
                    462:        Value& vlifespan_or_expires=params->get(index);
1.104     paf       463:        if(Value *vdate=vlifespan_or_expires.as(VDATE_TYPE, false))
                    464:                result=static_cast<VDate *>(vdate)->get_time();
1.79      paf       465:        else
                    466:                result=now+(time_t)params->as_double(index, "lifespan must be date or number", r);
                    467:        
                    468:        return result;
                    469: }
                    470: static const String as_file_spec(Request&r, MethodParams *params, int index) {
                    471:        return r.absolute(params->as_string(index, "filespec must be string"));
                    472: }
1.63      paf       473: static void _cache(Request& r, const String& method_name, MethodParams *params) {
                    474:        Pool& pool=r.pool();
1.79      paf       475:        time_t now=time(0);
                    476: 
                    477:        // ^cache[filename] ^cache(seconds) ^cache[expires date]
                    478:        if(params->size()==1) {
                    479:                if(params->get(0).is_string()) { // filename?
                    480:                        cache_delete(as_file_spec(r, params, 0));
                    481:                        return;
                    482:                }
                    483: 
                    484:                // secods|expires date
                    485:                Cache_data *data=static_cast<Cache_data *>(r.classes_conf.get(*cache_data_name));
                    486:                if(!data)
                    487:                        throw Exception("parser.runtime",
                    488:                                &method_name,
                    489:                                "expire-time reducing instruction without cache");
                    490:                
                    491:                time_t expires=as_expires(r, method_name, params, 0, now);
                    492:                if(expires < data->expires)
                    493:                        data->expires=expires;
                    494: 
                    495:                return;
                    496:        }
1.63      paf       497:        
                    498:        // file_spec, expires, body code
1.65      paf       499:        const String &file_spec=r.absolute(params->as_string(0, "filespec must be string"));
                    500: 
1.79      paf       501:        Cache_data data;
                    502:        Temp_hash_value cache_data_setter(r.classes_conf, *cache_data_name, &data);
                    503:        data.expires=as_expires(r, method_name, params, 1, now);
1.63      paf       504:        Value& body_code=params->as_junction(2, "body must be code");
                    505: 
1.79      paf       506:        if(data.expires>now) { // valid 'expires' specified? try cached copy...
1.69      paf       507:                // hence we don't hope to have unary create/lockEX
                    508:                // we need some plan to live in a life like that, so... 
                    509:                // worst races plan:
                    510:                // A        B
                    511:                // open
                    512:                //          |open
                    513:                // lockSH
                    514:                //          |nonblocking-lockEX fails
                    515:                // unlockSH
                    516:                // close, cache_get returns 0
                    517:                // open
                    518:                // nonblocking-lockEX succeeds; process, write, close
                    519:                //          |retry1: open
                    520:                // ...
                    521:                //          |lockSH succeeds; ...
                    522: 
                    523:                for(int retry=0; retry<2; retry++) {
1.79      paf       524:                        if(String *cached_body=cache_get(pool, file_spec, now)) { // have cached copy?
                    525:                                // write it out 
                    526:                                r.write_assign_lang(*cached_body);
                    527:                                // happy with it
                    528:                                return;
                    529:                        }
1.69      paf       530: 
                    531:                        // non-blocked lock; process; cache it
1.81      paf       532:                        if(const String*processed_body=
                    533:                                locked_process_and_cache_put(r, body_code, data, file_spec)) {
1.63      paf       534:                                // write it out 
1.69      paf       535:                                r.write_assign_lang(*processed_body);
1.63      paf       536:                                // happy with it
                    537:                                return;
1.69      paf       538:                        } else { // somebody writing result right now
                    539:                                pa_sleep(0, 500000); // waiting half a second
                    540:                                retry=0; // prolonging our wait, than could cache_get it, without processing body_code
1.63      paf       541:                        }
1.69      paf       542:                }
1.78      paf       543:                throw Exception(0,
1.69      paf       544:                        &file_spec,
                    545:                        "locking problem");
                    546:        } else { 
1.79      paf       547:                // instructed not to cache; forget cached copy
1.68      paf       548:                cache_delete(file_spec);
1.69      paf       549:                // process
1.81      paf       550:                const String& processed_body=r.process_to_string(body_code);
1.69      paf       551:                // write it out 
                    552:                r.write_assign_lang(processed_body);
                    553:                // happy with it
                    554:                return;
                    555:        }
                    556:        // never reached
1.63      paf       557: }
                    558: 
1.74      paf       559: // also used in pa_request.C to pass param to @unhandled_exception
                    560: VHash& exception2vhash(Pool& pool, const Exception& e) {
                    561:        VHash& result=*new(pool) VHash(pool);
                    562:        Hash& hash=result.hash(0);
1.94      paf       563:        if(const char *type=e.type(true))
1.78      paf       564:                hash.put(*exception_type_part_name, new(pool) VString(*new(pool) String(pool, type)));
1.76      paf       565:        if(const String *asource=e.problem_source()) {
                    566:                String& source=*new(pool) String(pool); 
                    567:                source.append(*asource, String::UL_TAINTED, true/*forced*/);
1.74      paf       568: 
1.76      paf       569:                hash.put(*exception_source_part_name, new(pool) VString(source));
1.74      paf       570: #ifndef NO_STRING_ORIGIN
1.76      paf       571:                const Origin& origin=source.origin();
1.89      paf       572:                hash.put(*new(pool) String(pool, "file", 0, true),
1.74      paf       573:                        new(pool) VString(*new(pool) String(pool, origin.file)));
                    574:                hash.put(*new(pool) String(pool, "lineno"),
                    575:                        new(pool) VInt(pool, 1+origin.line));
                    576: #endif
                    577:        }
1.94      paf       578:        if(const char *ecomment=e.comment(true)) {
1.74      paf       579:                int comment_size=strlen(ecomment);
                    580:                char *pcomment=(char *)pool.malloc(comment_size);
                    581:                memcpy(pcomment, ecomment, comment_size);
                    582:                hash.put(*exception_comment_part_name, 
1.76      paf       583:                        new(pool) VString(*new(pool) String(pool, pcomment, comment_size, true/*tainted*/)));
1.74      paf       584:        }
                    585:        hash.put(*exception_handled_part_name, 
                    586:                new(pool) VBool(pool, false));
                    587: 
                    588:        return result;
                    589: }
                    590: 
                    591: static void _try_operator(Request& r, const String& method_name, MethodParams *params) {
                    592:        Pool& pool=r.pool();
                    593: 
                    594:        Value& body_code=params->as_junction(0, "body_code must be code");
                    595:        Value& catch_code=params->as_junction(1, "catch_code must be code");
                    596: 
1.86      paf       597:        StringOrValue result;
1.74      paf       598:        try {
1.112     paf       599:                Request_context_saver context_saved(r); // taking snapshot of request processing status, restoring it at }
1.86      paf       600:                result=r.process(body_code);
1.112     paf       601:                context_saved.restore(); // manually restoring context
1.74      paf       602:        } catch(const Exception& e) {
                    603:                VHash& vhash=exception2vhash(pool, e);
                    604: 
                    605:                Junction *junction=catch_code.get_junction();
1.105     paf       606:                Value *method_frame=junction->method_frame;
                    607:                Value *saved_exception_var_value=method_frame->get_element(*exception_var_name, method_frame, false);
                    608:                junction->method_frame->put_element(*exception_var_name, &vhash, false);
1.86      paf       609:                result=r.process(catch_code);
1.74      paf       610:                bool handled=false;
                    611:                if(Value *value=static_cast<Value *>(vhash.hash(0).get(*exception_handled_part_name)))
                    612:                        handled=value->as_bool();               
1.105     paf       613:                junction->method_frame->put_element(*exception_var_name, saved_exception_var_value, false);
1.74      paf       614: 
                    615:                if(!handled)
                    616:                        throw(e); // rethrow
                    617:        }
                    618:        // write it out 
1.86      paf       619:        r.write_pass_lang(result);
1.74      paf       620: }
                    621: 
                    622: static void _throw_operator(Request& r, const String& method_name, MethodParams *params) {
                    623:        Pool& pool=r.pool();
                    624: 
                    625:        if(params->size()==1) {
                    626:                Value& param0=params->get(0);
                    627:                if(Hash *hash=param0.get_hash(&method_name)) {
1.78      paf       628:                        const char *type=0;
1.74      paf       629:                        if(Value *value=static_cast<Value *>(hash->get(*exception_type_part_name)))
1.78      paf       630:                                type=value->as_string().cstr();
1.74      paf       631:                        const String *source=0;
                    632:                        if(Value *value=static_cast<Value *>(hash->get(*exception_source_part_name)))
                    633:                                source=&value->as_string();
                    634:                        const char *comment=0;
                    635:                        if(Value *value=
                    636:                                static_cast<Value *>(hash->get(*exception_comment_part_name)))
                    637:                                comment=value->as_string().cstr();
                    638: 
1.78      paf       639:                        throw Exception(type,
1.74      paf       640:                                source?source:&method_name,
                    641:                                comment);
                    642:                } else
1.78      paf       643:                        throw Exception("parser.runtime",
1.74      paf       644:                                &method_name,
                    645:                                "one-param version has hash param");
                    646:        } else {
1.78      paf       647:                const char *type=params->as_string(0, "type must be string").cstr();
1.74      paf       648:                const String& source=params->as_string(1, "source must be string");
1.92      paf       649:                const char *comment=params->size()>2?params->as_string(2, "comment must be string").cstr():0;
1.97      paf       650:                throw Exception(type, &source, "%s", comment?comment:"");
1.74      paf       651:        }
                    652: }
                    653:        
1.10      paf       654: // constructor
                    655: 
1.113     paf       656: VClassMAIN::VClassMAIN(Pool& apool): VClass(apool) {
1.1       paf       657:        // ^if(condition){code-when-true}
                    658:        // ^if(condition){code-when-true}{code-when-false}
1.10      paf       659:        add_native_method("if", Method::CT_ANY, _if, 2, 3);
1.1       paf       660: 
                    661:        // ^untaint[as-is|uri|sql|js|html|html-typo]{code}
1.59      paf       662:        add_native_method("untaint", Method::CT_ANY, _untaint, 1, 2);
1.1       paf       663: 
                    664:        // ^taint[as-is|uri|sql|js|html|html-typo]{code}
1.10      paf       665:        add_native_method("taint", Method::CT_ANY, _taint, 1, 2);
1.1       paf       666: 
                    667:        // ^process[code]
1.116     paf       668:        add_native_method("process", Method::CT_ANY, _process, 1, 2);
1.1       paf       669: 
                    670:        // ^rem{code}
1.51      parser    671:        add_native_method("rem", Method::CT_ANY, _rem, 1, 10000);
1.1       paf       672: 
                    673:        // ^while(condition){code}
1.10      paf       674:        add_native_method("while", Method::CT_ANY, _while, 2, 2);
1.1       paf       675: 
                    676:        // ^use[file]
1.10      paf       677:        add_native_method("use", Method::CT_ANY, _use, 1, 1);
1.1       paf       678: 
1.54      paf       679:        // ^for[i](from-number;to-number-inclusive){code}[delim]
1.10      paf       680:        add_native_method("for", Method::CT_ANY, _for, 3+1, 3+1+1);
1.1       paf       681: 
                    682:        // ^eval(expr)
                    683:        // ^eval(expr)[format]
1.10      paf       684:        add_native_method("eval", Method::CT_ANY, _eval, 1, 2);
1.52      parser    685: 
1.1       paf       686:        // ^connect[protocol://user:pass@host[:port]/database]{code with ^sql-s}
1.10      paf       687:        add_native_method("connect", Method::CT_ANY, _connect, 2, 2);
                    688: 
1.63      paf       689: 
1.65      paf       690:        // ^cache[file_spec] delete cache
1.63      paf       691:        // ^cache[file_spec](time){code} time=0 no cache
1.65      paf       692:        add_native_method("cache", Method::CT_ANY, _cache, 1, 3);
1.63      paf       693:        
1.28      parser    694:        // switch
                    695: 
                    696:        // ^switch[value]{cases}
                    697:        add_native_method("switch", Method::CT_ANY, _switch, 2, 2);
                    698: 
                    699:        // ^case[value]{code}
1.51      parser    700:        add_native_method("case", Method::CT_ANY, _case, 2, 10000);
1.74      paf       701: 
                    702:        // try-catch
                    703: 
                    704:        // ^try{code}{catch code}
                    705:        add_native_method("try", Method::CT_ANY, _try_operator, 2, 2);
                    706:        // ^throw[$exception hash]
                    707:        // ^throw[type;source;comment]
                    708:        add_native_method("throw", Method::CT_ANY, _throw_operator, 1, 3);
                    709: 
1.10      paf       710: }
1.11      paf       711: 
                    712: // constructor & configurator
1.1       paf       713: 
1.113     paf       714: VClass& VClassMAIN_create(Pool& pool) {
                    715:        return *new(pool) VClassMAIN(pool);
1.1       paf       716: }

E-mail: