Annotation of parser3/src/classes/string.C, revision 1.255
1.24 paf 1: /** @file
2: Parser: @b string parser class.
3:
1.251 moko 4: Copyright (c) 2001-2023 Art. Lebedev Studio (http://www.artlebedev.com)
5: Authors: Konstantin Morshnev <moko@design.ru>, Alexandr Petrosian <paf@design.ru>
1.113 paf 6: */
1.24 paf 7:
1.43 paf 8: #include "classes.h"
1.126 paf 9: #include "pa_vmethod_frame.h"
1.241 moko 10: #include "pa_base64.h"
1.126 paf 11:
1.1 paf 12: #include "pa_request.h"
13: #include "pa_vdouble.h"
14: #include "pa_vint.h"
1.17 paf 15: #include "pa_vtable.h"
1.25 paf 16: #include "pa_vbool.h"
1.27 paf 17: #include "pa_string.h"
1.53 parser 18: #include "pa_sql_connection.h"
1.71 parser 19: #include "pa_dictionary.h"
1.119 paf 20: #include "pa_vmethod_frame.h"
1.174 misha 21: #include "pa_vregex.h"
1.189 misha 22: #include "pa_charsets.h"
1.1 paf 23:
1.255 ! moko 24: volatile const char * IDENT_STRING_C="$Id: string.C,v 1.254 2024/09/13 04:01:22 moko Exp $";
1.203 moko 25:
1.41 paf 26: // class
27:
1.126 paf 28: class MString: public Methoded {
1.41 paf 29: public:
1.126 paf 30: MString();
1.41 paf 31: };
1.1 paf 32:
1.126 paf 33: // global variable
34:
1.224 moko 35: DECLARE_CLASS_VAR(string, new MString);
1.126 paf 36:
1.196 moko 37: // void class, inherited from string and thus should be inited afterwards
38:
39: class MVoid: public Methoded {
40: public:
41: MVoid();
42: };
43:
44: // void global variable should be after string global variable
45:
1.224 moko 46: DECLARE_CLASS_VAR(void, new MVoid);
1.196 moko 47:
1.126 paf 48: // defines for statics
49:
50: #define MATCH_VAR_NAME "match"
1.162 misha 51: #define TRIM_START_OPTION "left"
52: #define TRIM_END_OPTION "right"
1.133 paf 53: #define TRIM_BOTH_OPTION "both"
1.126 paf 54:
1.189 misha 55: #define MODE_APPEND "append"
56:
1.220 moko 57: #define UNESCAPE_MODE_JS "js"
58: #define UNESCAPE_MODE_URI "uri"
59:
1.126 paf 60: // statics
61:
62: static const String match_var_name(MATCH_VAR_NAME);
63:
1.1 paf 64: // methods
65:
1.126 paf 66: static void _length(Request& r, MethodParams&) {
1.163 misha 67: double result=GET_SELF(r, VString).string().length(r.charsets.source());
1.233 moko 68: r.write(*new VDouble(result));
1.1 paf 69: }
70:
1.126 paf 71: static void _int(Request& r, MethodParams& params) {
72: const String& self_string=GET_SELF(r, VString).string();
1.72 parser 73: int converted;
1.206 moko 74:
75: if(self_string.is_empty()) {
1.144 paf 76: if(params.count()>0)
77: converted=params.as_int(0, "default must be int", r); // (default)
1.84 parser 78: else
1.206 moko 79: throw Exception(PARSER_RUNTIME, 0, "unable to convert empty string without default specified");
80: } else {
81: try {
82: converted=self_string.as_int();
83: } catch(...) { // convert problem
84: if(params.count()>0)
85: converted=params.as_int(0, "default must be int", r); // (default)
86: else
87: rethrow; // we have a problem when no default
88: }
1.72 parser 89: }
1.206 moko 90:
1.233 moko 91: r.write(*new VInt(converted));
1.1 paf 92: }
93:
1.126 paf 94: static void _double(Request& r, MethodParams& params) {
95: const String& self_string=GET_SELF(r, VString).string();
1.206 moko 96:
97: if(self_string.is_empty()) {
1.144 paf 98: if(params.count()>0)
1.233 moko 99: r.write(*new VDouble(params.as_double(0, "default must be double", r))); // (default)
1.84 parser 100: else
1.206 moko 101: throw Exception(PARSER_RUNTIME, 0, "unable to convert empty string without default specified");
102: } else {
103: try {
1.233 moko 104: r.write(*new VDouble(self_string.as_double()));
1.206 moko 105: } catch(...) { // convert problem
106: if(params.count()>0)
1.233 moko 107: r.write(*new VDouble(params.as_double(0, "default must be double", r))); // (default)
1.206 moko 108: else
109: rethrow; // we have a problem when no default
110: }
1.72 parser 111: }
1.1 paf 112: }
113:
1.151 misha 114: static void _bool(Request& r, MethodParams& params) {
115: const String& self_string=GET_SELF(r, VString).string();
116: bool converted;
1.206 moko 117: const char *str=self_string.cstr();
118:
119: if(self_string.is_empty()) {
120: if(params.count()>0)
121: converted=params.as_bool(0, "default must be bool", r); // (default)
122: else
123: throw Exception(PARSER_RUNTIME, 0, "unable to convert empty string without default specified");
1.207 moko 124: } else if( (str[0]=='T' || str[0]=='t') && (str[1]=='R' || str[1]=='r') && (str[2]=='U' || str[2]=='u') &&
1.206 moko 125: (str[3]=='E' || str[3]=='e') && str[4]==0 ) { // "true"
126: converted=true;
1.207 moko 127: } else if( (str[0]=='F' || str[0]=='f') && (str[1]=='A' || str[1]=='a') && (str[2]=='L' || str[2]=='l') &&
1.206 moko 128: (str[3]=='S' || str[3]=='s') && (str[4]=='E' || str[4]=='e') && str[5]==0 ) { // "false"
129: converted=false;
130: } else {
1.158 misha 131: try {
132: converted=self_string.as_bool();
1.206 moko 133: } catch(...) { // convert problem
134: if(params.count()>0)
135: converted=params.as_bool(0, "default must be bool", r); // (default)
136: else
137: rethrow; // we have a problem when no default
1.158 misha 138: }
1.151 misha 139: }
140:
1.233 moko 141: r.write(VBool::get(converted));
1.151 misha 142: }
143:
1.126 paf 144: /*not static*/void _string_format(Request& r, MethodParams& params) {
1.9 paf 145:
1.126 paf 146: Value& fmt_maybe_code=params[0];
1.95 paf 147: // for some time due to stupid {} in original design
1.99 paf 148: const String& fmt=r.process_to_string(fmt_maybe_code);
1.9 paf 149:
1.159 misha 150: const char* buf=format(r.get_self().as_double(), fmt.trim().cstrm());
1.63 parser 151:
1.233 moko 152: r.write(String(buf));
1.9 paf 153: }
1.11 paf 154:
1.126 paf 155: static void _left(Request& r, MethodParams& params) {
1.244 moko 156: int sn=params.as_int(0, "n must be int", r);
1.126 paf 157: const String& string=GET_SELF(r, VString).string();
1.244 moko 158: r.write(sn<0 ? string : string.mid(r.charsets.source(), 0, sn));
1.15 paf 159: }
160:
1.126 paf 161: static void _right(Request& r, MethodParams& params) {
1.244 moko 162: int sn=params.as_int(0, "n must be int", r);
1.216 moko 163: if(sn>0){
1.244 moko 164: size_t n=sn;
1.216 moko 165: const String& string=GET_SELF(r, VString).string();
166: size_t length=string.length(r.charsets.source());
1.233 moko 167: r.write(n<length ? string.mid(r.charsets.source(), length-n, length, length) : string);
1.216 moko 168: }
1.15 paf 169: }
170:
1.126 paf 171: static void _mid(Request& r, MethodParams& params) {
172: const String& string=GET_SELF(r, VString).string();
1.83 parser 173:
1.245 moko 174: int begin=params.as_int(0, "p must be int", r);
175: if(begin<0)
176: throw Exception(PARSER_RUNTIME, 0, "p(%d) must be >=0", begin);
1.138 paf 177:
178: size_t end;
1.164 misha 179: size_t length=0;
1.245 moko 180:
1.138 paf 181: if(params.count()>1) {
1.245 moko 182: int n=params.as_int(1, "n must be int", r);
183: if(n<0)
184: throw Exception(PARSER_RUNTIME, 0, "n(%d) must be >=0", n);
185: end=begin+n;
1.164 misha 186: } else {
187: length=string.length(r.charsets.source());
188: end=length;
189: }
1.163 misha 190:
1.233 moko 191: r.write(string.mid(r.charsets.source(), begin, end, length));
1.15 paf 192: }
193:
1.126 paf 194: static void _pos(Request& r, MethodParams& params) {
195: Value& substr=params.as_no_junction(0, "substr must not be code");
1.16 paf 196:
1.126 paf 197: const String& string=GET_SELF(r, VString).string();
1.245 moko 198: int offset=0;
1.165 misha 199: if(params.count()>1){
200: offset=params.as_int(1, "n must be int", r);
201: if(offset<0)
1.227 moko 202: throw Exception(PARSER_RUNTIME, 0, "n(%d) must be >=0", offset);
1.165 misha 203: }
204:
1.245 moko 205: r.write(*new VInt((int)string.pos(r.charsets.source(), substr.as_string(), offset)));
1.16 paf 206: }
207:
1.250 moko 208: struct Split_action_info {
209: const String& src;
210: ArrayString &result;
211: };
212:
213: static void split_action(Table& , ArrayString* row, int prestart, int prefinish, int poststart, int postfinish, void *info) {
214: Split_action_info& ai=*static_cast<Split_action_info *>(info);
215: if(row) { // begin&middle
216: // piece from last match['prestart'] to beginning of this match['prefinish']
217: ai.result += &ai.src.mid(prestart, prefinish);
218: } else // end
219: if(poststart != postfinish)
220: ai.result += &ai.src.mid(poststart, postfinish);
221: }
222:
1.249 moko 223: static void split_list(Value& delim_value, const String& string, ArrayString& result) {
1.254 moko 224: if(VRegex *vregex=dynamic_cast<VRegex*>(&delim_value)){
1.250 moko 225: vregex->study();
226:
227: int matches_count=0;
228: Split_action_info ai = { string, result };
229:
230: string.match(vregex, split_action, &ai, matches_count);
231: } else
232: string.split(result, 0, delim_value.as_string());
1.19 paf 233: }
234:
1.118 paf 235: #define SPLIT_LEFT 0x0001
236: #define SPLIT_RIGHT 0x0010
237: #define SPLIT_HORIZONTAL 0x0100
238: #define SPLIT_VERTICAL 0x1000
239:
1.126 paf 240: static int split_options(const String* options) {
1.118 paf 241: struct Split_option {
1.126 paf 242: const char* keyL;
243: const char* keyU;
1.118 paf 244: int setBit;
245: int checkBit;
246: } split_option[]={
247: {"l", "L", SPLIT_LEFT, SPLIT_RIGHT}, // 0xVHRL
248: {"r", "R", SPLIT_RIGHT, SPLIT_LEFT},
249: {"h", "H", SPLIT_HORIZONTAL, SPLIT_VERTICAL},
250: {"v", "V", SPLIT_VERTICAL, SPLIT_HORIZONTAL},
1.130 paf 251: {0, 0, 0, 0}
1.118 paf 252: };
253:
254: int result=0;
1.126 paf 255: if(options) {
1.118 paf 256: for(Split_option *o=split_option; o->keyL; o++)
1.126 paf 257: if(options->pos(o->keyL)!=STRING_NOT_FOUND
258: || (o->keyU && options->pos(o->keyU)!=STRING_NOT_FOUND)) {
1.118 paf 259: if(result & o->checkBit)
1.238 moko 260: throw Exception(PARSER_RUNTIME, options, "conflicting split options");
1.118 paf 261: result |= o->setBit;
262: }
263: }
264:
265: return result;
266: }
267:
1.156 misha 268: static Table& split_vertical(ArrayString& pieces, bool right, const String* column_name) {
1.126 paf 269: Table::columns_type columns(new ArrayString);
1.156 misha 270: *columns+=column_name;
1.19 paf 271:
1.126 paf 272: Table& table=*new Table(columns, pieces.count());
1.118 paf 273: if(right) { // right
1.126 paf 274: for(int i=pieces.count(); --i>=0; ) {
275: Table::element_type row(new ArrayString);
276: *row+=pieces[i];
277: table+=row;
1.118 paf 278: }
279: } else { // left
1.253 moko 280: for(ArrayString::Iterator i(pieces); i; ) {
1.126 paf 281: Table::element_type row(new ArrayString);
282: *row+=i.next();
283: table+=row;
1.118 paf 284: }
1.61 parser 285: }
1.118 paf 286:
1.126 paf 287: return table;
1.19 paf 288: }
289:
1.128 paf 290: static Table& split_horizontal(ArrayString& pieces, bool right) {
1.126 paf 291: Table& table=*new Table(Table::columns_type(0) /* nameless */);
292: Table::element_type row(new ArrayString(pieces.count()));
1.118 paf 293: if(right) { // right
1.131 paf 294: for(int i=pieces.count(); --i>=0; )
1.126 paf 295: *row+=pieces[i];
1.118 paf 296: } else { // left
1.253 moko 297: for(ArrayString::Iterator i(pieces); i; )
1.126 paf 298: *row+=i.next();
1.118 paf 299: }
1.126 paf 300: table+=row;
1.118 paf 301:
1.126 paf 302: return table;
1.118 paf 303: }
304:
1.238 moko 305: static void split_with_options(Request& r, MethodParams& params, int bits) {
1.126 paf 306: const String& string=GET_SELF(r, VString).string();
1.176 misha 307: size_t params_count=params.count();
1.19 paf 308:
1.126 paf 309: ArrayString pieces;
1.249 moko 310: split_list(params.as_no_junction(0, "delimiter must not be code"), string, pieces);
1.19 paf 311:
1.118 paf 312: if(!bits) {
1.126 paf 313: const String* options=0;
1.176 misha 314: if(params_count>1)
1.187 misha 315: options=¶ms.as_string(1, OPTIONS_MUST_NOT_BE_CODE);
1.118 paf 316: bits=split_options(options);
317: }
1.21 paf 318:
1.118 paf 319: bool right=(bits & SPLIT_RIGHT) != 0;
320: bool horizontal=(bits & SPLIT_HORIZONTAL) !=0;
1.156 misha 321:
322: const String* column_name=0;
1.176 misha 323: if(params_count>2){
1.156 misha 324: column_name=¶ms.as_string(2, COLUMN_NAME_MUST_BE_STRING);
1.178 misha 325: if (horizontal && !column_name->is_empty())
1.238 moko 326: throw Exception(PARSER_RUNTIME, column_name, "column name can't be specified with horisontal split");
1.156 misha 327: }
1.178 misha 328: if(!column_name || column_name->is_empty())
1.156 misha 329: column_name=new String("piece");
330:
331: Table& table=horizontal?split_horizontal(pieces, right):split_vertical(pieces, right, column_name);
1.17 paf 332:
1.233 moko 333: r.write(*new VTable(&table));
1.118 paf 334: }
1.126 paf 335: static void _split(Request& r, MethodParams& params) {
336: split_with_options(r, params, 0 /* maybe-determine from param #2 */);
1.118 paf 337: }
1.126 paf 338: static void _lsplit(Request& r, MethodParams& params) {
339: split_with_options(r, params, SPLIT_LEFT);
1.118 paf 340: }
1.126 paf 341: static void _rsplit(Request& r, MethodParams& params) {
342: split_with_options(r, params, SPLIT_RIGHT);
1.17 paf 343: }
344:
1.126 paf 345: static void search_action(Table& table, Table::element_type row, int, int, int, int, void *) {
1.28 paf 346: if(row)
347: table+=row;
1.27 paf 348: }
349:
1.74 parser 350: #ifndef DOXYGEN
1.27 paf 351: struct Replace_action_info {
1.181 misha 352: Request* request;
353: const String* src;
354: String* dest;
1.126 paf 355: VTable* vtable;
356: Value* replacement_code;
1.27 paf 357: };
1.74 parser 358: #endif
1.105 paf 359: /// @todo they can do $global[$result] there, getting pointer to later-invalid local var, kill this
1.238 moko 360: static void replace_action(Table& table, ArrayString* row, int prestart, int prefinish, int poststart, int postfinish, void *info) {
1.27 paf 361: Replace_action_info& ai=*static_cast<Replace_action_info *>(info);
1.31 paf 362: if(row) { // begin&middle
1.104 paf 363: // piece from last match['prestart'] to beginning of this match['prefinish']
364: if(prestart!=prefinish)
365: *ai.dest << ai.src->mid(prestart, prefinish);//ai.dest->APPEND_CONST("-");
1.51 parser 366: // store found parts in one-record VTable
1.126 paf 367: if(table.count()) // middle
1.32 paf 368: table.put(0, row);
369: else // begin
1.30 paf 370: table+=row;
1.181 misha 371:
1.30 paf 372: { // execute 'replacement_code' in 'table' context
1.181 misha 373: if(ai.replacement_code){
374: ai.vtable->set_table(table);
375: *ai.dest << ai.request->process_to_string(*ai.replacement_code);
376: }
1.29 paf 377: }
378: } else // end
1.104 paf 379: *ai.dest << ai.src->mid(poststart, postfinish);
1.27 paf 380: }
381:
1.126 paf 382: static void _match(Request& r, MethodParams& params) {
1.174 misha 383: size_t params_count=params.count();
384:
1.126 paf 385: Value& regexp=params.as_no_junction(0, "regexp must not be code");
1.187 misha 386: Value* options=(params_count>1)?¶ms.as_no_junction(1, OPTIONS_MUST_NOT_BE_CODE):0;
1.174 misha 387:
1.254 moko 388: VRegex* vregex=dynamic_cast<VRegex*>(®exp);
1.174 misha 389: VRegexCleaner vrcleaner;
1.126 paf 390:
1.254 moko 391: if(vregex){
1.174 misha 392: if(options && options->is_defined())
1.255 ! moko 393: throw Exception(PARSER_RUNTIME, 0, "you cannot specify regex-object and options together");
1.174 misha 394: } else {
1.237 moko 395: vregex=new VRegex(r.charsets.source(), ®exp.as_string(), (options) ? (&options->as_string()) : 0);
1.175 misha 396: vregex->study();
1.174 misha 397: vrcleaner.vregex=vregex;
398: }
1.126 paf 399:
400: const String& src=GET_SELF(r, VString).string();
1.152 misha 401: int matches_count=0;
1.174 misha 402:
403: if(params_count<3) { // search
1.237 moko 404: Table* table=src.match(vregex, search_action, 0, matches_count);
1.174 misha 405:
1.152 misha 406: if(table){
1.233 moko 407: r.write(*new VTable(table));
1.152 misha 408: } else {
1.233 moko 409: r.write(*new VInt(matches_count));
1.152 misha 410: }
411:
1.27 paf 412: } else { // replace
1.181 misha 413:
414: Value* replacement_code=0;
415: bool is_junction=false;
416:
417: Value* replacement=¶ms[2];
418: if(replacement->get_junction()){
419: replacement_code=replacement;
420: is_junction=true;
421: } else if(replacement->is_string()){
422: if(replacement->is_defined())
423: replacement_code=replacement;
424: } else if(!replacement->is_void())
1.238 moko 425: throw Exception(PARSER_RUNTIME, 0, "replacement option should be junction or string");
1.106 paf 426:
1.238 moko 427: Value* default_code=(params_count==4) ? ¶ms.as_junction(3, "default value must be code") : 0;
1.192 misha 428:
1.126 paf 429: String result;
430: VTable* vtable=new VTable;
1.130 paf 431: Replace_action_info info={
432: &r,
433: &src,
434: &result,
435: vtable,
1.181 misha 436: replacement_code
1.130 paf 437: };
1.181 misha 438:
1.191 misha 439: if(is_junction){
1.211 moko 440: Temp_value_element temp(r, *replacement_code->get_junction()->method_frame, match_var_name, vtable);
1.191 misha 441: src.match(vregex, replace_action, &info, matches_count);
442: } else {
443: src.match(vregex, replace_action, &info, matches_count);
444: }
1.181 misha 445:
1.192 misha 446: if(!matches_count && default_code)
447: r.process_write(*default_code);
448: else
1.233 moko 449: r.write(result);
1.27 paf 450: }
1.24 paf 451: }
452:
1.238 moko 453: static void change_case(Request& r, MethodParams&, String::Change_case_kind kind) {
1.126 paf 454: const String& src=GET_SELF(r, VString).string();
1.49 parser 455:
1.233 moko 456: r.write(src.change_case(r.charsets.source(), kind));
1.49 parser 457: }
1.126 paf 458: static void _upper(Request& r, MethodParams& params) {
459: change_case(r, params, String::CC_UPPER);
1.49 parser 460: }
1.126 paf 461: static void _lower(Request& r, MethodParams& params) {
462: change_case(r, params, String::CC_LOWER);
1.49 parser 463: }
464:
1.65 parser 465: #ifndef DOXYGEN
1.126 paf 466: class String_sql_event_handlers: public SQL_Driver_query_event_handlers {
467: bool got_column;
1.65 parser 468: public:
1.126 paf 469: bool got_cell;
1.208 moko 470: const String* result;
1.126 paf 471: public:
1.240 moko 472: String_sql_event_handlers():
1.126 paf 473: got_column(false),
474: got_cell(false),
1.208 moko 475: result(&String::Empty) {}
1.65 parser 476:
1.128 paf 477: bool add_column(SQL_Error& error, const char* /*str*/, size_t /*length*/) {
1.124 paf 478: if(got_column) {
1.240 moko 479: error=SQL_Error("result must contain exactly one column");
1.124 paf 480: return true;
481: }
1.65 parser 482: got_column=true;
1.124 paf 483: return false;
1.65 parser 484: }
1.124 paf 485: bool before_rows(SQL_Error& /*error*/ ) { /* ignore */ return false; }
486: bool add_row(SQL_Error& /*error*/) { /* ignore */ return false; }
1.210 moko 487: bool add_row_cell(SQL_Error& error, const char* str, size_t) {
1.124 paf 488: if(got_cell) {
1.255 ! moko 489: error=SQL_Error("result must contain no more than one row");
1.124 paf 490: return true;
491: }
492: try {
493: got_cell=true;
1.208 moko 494: result=new String(str, String::L_TAINTED /* no length as 0x00 can be inside */ );
1.124 paf 495: return false;
496: } catch(...) {
1.235 moko 497: error=SQL_Error("exception occurred in String_sql_event_handlers::add_row_cell");
1.124 paf 498: return true;
499: }
1.65 parser 500: }
501: };
502: #endif
1.239 moko 503:
1.141 paf 504: extern int marshal_binds(HashStringValue& hash, SQL_Driver::Placeholder*& placeholders);
505: extern void unmarshal_bind_updates(HashStringValue& hash, int placeholder_count, SQL_Driver::Placeholder* placeholders);
506:
1.204 moko 507: const String* sql_result_string(Request& r, MethodParams& params, Value*& default_code) {
1.126 paf 508: Value& statement=params.as_junction(0, "statement must be code");
1.53 parser 509:
1.141 paf 510: HashStringValue* bind=0;
1.160 misha 511: ulong limit=SQL_NO_LIMIT;
1.70 parser 512: ulong offset=0;
1.81 parser 513: default_code=0;
1.199 misha 514: if(params.count()>1)
1.205 misha 515: if(HashStringValue* options=params.as_hash(1, "sql options")) {
1.199 misha 516: int valid_options=0;
517: if(Value* vbind=options->get(sql_bind_name)) {
518: valid_options++;
519: bind=vbind->get_hash();
520: }
521: if(Value* vlimit=options->get(sql_limit_name)) {
522: valid_options++;
1.230 moko 523: limit=(ulong)r.process(*vlimit).as_double();
1.199 misha 524: }
525: if(Value* voffset=options->get(sql_offset_name)) {
526: valid_options++;
1.230 moko 527: offset=(ulong)r.process(*voffset).as_double();
1.199 misha 528: }
529: if((default_code=options->get(sql_default_name))) {
530: valid_options++;
531: }
532: if(valid_options!=options->count())
533: throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
1.200 misha 534: }
1.70 parser 535:
1.141 paf 536: SQL_Driver::Placeholder* placeholders=0;
537: uint placeholders_count=0;
538: if(bind)
539: placeholders_count=marshal_binds(*bind, placeholders);
540:
1.99 paf 541: const String& statement_string=r.process_to_string(statement);
1.232 moko 542: const char* statement_cstr=statement_string.untaint_cstr(String::L_SQL, r.connection());
1.183 misha 543:
1.240 moko 544: String_sql_event_handlers handlers;
1.160 misha 545:
1.240 moko 546: r.connection()->query(statement_cstr, placeholders_count, placeholders, offset, limit, handlers, statement_string);
1.53 parser 547:
1.141 paf 548: if(bind)
549: unmarshal_bind_updates(*bind, placeholders_count, placeholders);
550:
1.65 parser 551: if(!handlers.got_cell)
552: return 0; // no lines, caller should return second param[default value]
1.62 parser 553:
1.208 moko 554: return handlers.result;
1.53 parser 555: }
556:
1.126 paf 557: static void _sql(Request& r, MethodParams& params) {
1.53 parser 558:
1.126 paf 559: Value* default_code;
1.204 moko 560: const String* string=sql_result_string(r, params, default_code);
1.62 parser 561: if(!string) {
1.81 parser 562: if(default_code) {
1.99 paf 563: string=&r.process_to_string(*default_code);
1.68 parser 564: } else
1.238 moko 565: throw Exception(PARSER_RUNTIME, 0, "produced no result, but no default option specified");
1.62 parser 566: }
1.100 paf 567:
1.233 moko 568: r.write(*string);
1.53 parser 569: }
570:
1.126 paf 571: static void _replace(Request& r, MethodParams& params) {
572: const String& src=GET_SELF(r, VString).string();
1.71 parser 573:
1.201 misha 574: if(params.count()==1) {
575: // ^string.replace[table]
1.248 moko 576: if(Table* table=params.as_table(0, "param")){
577: Dictionary dict(*table);
578: r.write(src.replace(dict));
579: } else
580: r.write(src);
1.201 misha 581: } else {
582: // ^string.replace[from-string;to-string]
1.238 moko 583: Dictionary dict(params.as_string(0, "from must be string"), params.as_string(1, "to must be string"));
1.233 moko 584: r.write(src.replace(dict));
1.201 misha 585: }
1.71 parser 586:
587: }
1.79 parser 588:
1.126 paf 589: static void _save(Request& r, MethodParams& params) {
1.189 misha 590: bool do_append=false;
591: Charset* asked_charset=0;
592:
593: size_t file_name_index=0;
1.214 moko 594: if(params.count()>1) {
1.189 misha 595: if(HashStringValue* options=params.as_no_junction(1, "second parameter should be string or hash").get_hash()){
1.205 misha 596: // ^file.save[filespec;$.charset[] $.append(true)]
1.190 misha 597: int valid_options=0;
1.189 misha 598: if(Value* vcharset_name=options->get(PA_CHARSET_NAME)){
1.229 moko 599: asked_charset=&pa_charsets.get(vcharset_name->as_string());
1.189 misha 600: valid_options++;
601: }
602: if(Value* vappend=options->get(MODE_APPEND)){
603: do_append=vappend->as_bool();
604: valid_options++;
605: }
606: if(valid_options != options->count())
1.194 misha 607: throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
1.189 misha 608: } else {
1.205 misha 609: // ^file.save[append;filespec]
1.189 misha 610: const String& mode=params.as_string(0, "mode must be string");
611: if(mode==MODE_APPEND){
612: do_append=true;
613: file_name_index++;
614: } else
615: throw Exception(PARSER_RUNTIME,
616: &mode,
617: "unknown mode, must be 'append'");
618: }
1.214 moko 619: }
1.79 parser 620:
1.189 misha 621: const String& file_name=params.as_string(file_name_index, FILE_NAME_MUST_BE_STRING);
1.126 paf 622: const String& src=GET_SELF(r, VString).string();
1.79 parser 623:
1.209 moko 624: String::Body sbody=src.cstr_to_string_body_untaint(String::L_AS_IS, r.connection(false), &r.charsets);
1.87 paf 625:
1.79 parser 626: // write
1.246 moko 627: file_write(r.charsets, r.full_disk_path(file_name), sbody.cstr(), sbody.length(), true, do_append, asked_charset);
1.79 parser 628: }
629:
1.126 paf 630: static void _normalize(Request& r, MethodParams&) {
631: const String& src=GET_SELF(r, VString).string();
632:
1.233 moko 633: r.write(src);
1.109 paf 634: }
635:
1.133 paf 636: static void _trim(Request& r, MethodParams& params) {
637: const String& src=GET_SELF(r, VString).string();
638:
639: String::Trim_kind kind=String::TRIM_BOTH;
1.176 misha 640: size_t params_count=params.count();
1.133 paf 641: const char* chars=0;
1.176 misha 642: if(params_count>0) {
643: const String& skind=params.as_string(0, "'where' must be string");
1.214 moko 644: if(!skind.is_empty()) {
1.162 misha 645: if(skind==TRIM_BOTH_OPTION)
646: kind=String::TRIM_BOTH;
647: else if(skind==TRIM_START_OPTION || skind=="start")
1.137 paf 648: kind=String::TRIM_START;
1.162 misha 649: else if(skind==TRIM_END_OPTION || skind=="end")
1.137 paf 650: kind=String::TRIM_END;
1.218 moko 651: else if(params_count==1)
652: chars=skind.cstr();
1.137 paf 653: else
1.238 moko 654: throw Exception(PARSER_RUNTIME, &skind, "'kind' must be one of " TRIM_START_OPTION ", " TRIM_BOTH_OPTION ", " TRIM_END_OPTION);
1.214 moko 655: }
1.133 paf 656:
1.176 misha 657: if(params_count>1) {
1.136 paf 658: const String& schars=params.as_string(1, "'chars' must be string");
1.178 misha 659: if(!schars.is_empty())
1.137 paf 660: chars=schars.cstr();
1.136 paf 661: }
1.133 paf 662: }
663:
1.233 moko 664: r.write(src.trim(kind, chars, &r.charsets.source()));
1.133 paf 665: }
666:
1.241 moko 667: Base64Options base64_encode_options(Request& r, HashStringValue* options) {
668: Base64Options result;
669: if(options) {
670: int valid_options=0;
671: for(HashStringValue::Iterator i(*options); i; i.next()) {
672: String::Body key=i.key();
673: Value* value=i.value();
674: if(key == "pad") {
675: result.pad=r.process(*value).as_bool();
676: valid_options++;
677: } else if(key == "wrap") {
678: result.wrap=r.process(*value).as_bool();
679: valid_options++;
680: } else if(key == "url-safe") {
681: if(r.process(*value).as_bool())
682: result.set_url_safe_abc();
683: valid_options++;
684: }
685: }
686:
687: if(valid_options != options->count())
688: throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
689: }
690: return result;
691: }
692:
693: Base64Options base64_decode_options(Request& r, HashStringValue* options) {
694: Base64Options result;
695: if(options) {
696: int valid_options=0;
697: for(HashStringValue::Iterator i(*options); i; i.next() ) {
698: String::Body key=i.key();
699: Value* value=i.value();
700: if(key == "pad") {
701: result.pad=r.process(*value).as_bool();
702: valid_options++;
703: } else if(key == "strict") {
704: result.strict=r.process(*value).as_bool();
705: valid_options++;
706: } else if(key == "url-safe") {
707: if(r.process(*value).as_bool())
708: result.set_url_safe_abc();
709: valid_options++;
710: }
711: }
712:
713: if(valid_options != options->count())
714: throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
715: }
716: return result;
717: }
718:
1.146 paf 719: static void _base64(Request& r, MethodParams& params) {
1.219 moko 720: if(&r.get_self() == string_class) {
1.241 moko 721: // decode: ^string:base64[encoded[;options]]
1.219 moko 722: const char* cstr=params.count() ? params.as_string(0, PARAMETER_MUST_BE_STRING).cstr() : "";
1.241 moko 723: Base64Options options = base64_decode_options(r, params.count() > 1 ? params.as_hash(1) : NULL);
724:
1.169 misha 725: char* decoded=0;
1.243 moko 726: size_t length=pa_base64_decode(cstr, strlen(cstr), decoded, options);
1.202 misha 727:
1.169 misha 728: if(decoded && length){
1.242 moko 729: if(memchr(decoded, 0, length))
730: throw Exception(PARSER_RUNTIME, 0, "Invalid \\x00 character found while decoding to string. Decode to file instead.");
1.169 misha 731:
732: fix_line_breaks(decoded, length);
1.242 moko 733:
1.179 misha 734: if(length)
1.233 moko 735: r.write(*new String(decoded, String::L_TAINTED));
1.169 misha 736: }
1.147 paf 737: } else {
1.241 moko 738: // encode: ^str.base64[options]
1.148 paf 739: VString& self=GET_SELF(r, VString);
740: const char* cstr=self.string().cstr();
1.241 moko 741: Base64Options options = base64_encode_options(r, params.count() ? params.as_hash(0) : NULL);
742: const char* encoded=pa_base64_encode(cstr, strlen(cstr), options);
743: r.write(*new String(encoded, String::L_TAINTED /*once ?param=base64(something) was needed*/ ));
1.146 paf 744: }
745: }
746:
1.215 moko 747: static void _idna(Request& r, MethodParams& params) {
1.219 moko 748: if(&r.get_self() == string_class) {
1.215 moko 749: // decode: ^string:idna[encoded]
1.219 moko 750: const char* cstr=params.count() ? params.as_string(0, PARAMETER_MUST_BE_STRING).cstr() : "";
1.233 moko 751: r.write(*new String(pa_idna_decode(cstr, r.charsets.source()), String::L_TAINTED));
1.215 moko 752: } else {
753: // encode: ^str.idna[]
754: VString& self=GET_SELF(r, VString);
755: const char* cstr=self.string().cstr();
1.233 moko 756: r.write(*new String(pa_idna_encode(cstr, r.charsets.source()), String::L_TAINTED));
1.215 moko 757: }
758: }
759:
1.220 moko 760: static void _js_escape(Request& r, MethodParams&){
1.167 misha 761: const String& src=GET_SELF(r, VString).string();
1.233 moko 762: r.write(src.escape(r.charsets.source()));
1.167 misha 763: }
764:
1.220 moko 765: static void _js_unescape(Request& r, MethodParams& params){
1.167 misha 766: const String& src=params.as_string(0, PARAMETER_MUST_BE_STRING);
1.193 misha 767: if(const char* result=unescape_chars(src.cstr(), src.length(), &r.charsets.source(), true))
1.233 moko 768: r.write(*new String(result, String::L_TAINTED));
1.167 misha 769: }
770:
1.220 moko 771: static void _unescape(Request& r, MethodParams& params){
772: const String& mode=params.as_string(0, MODE_MUST_NOT_BE_CODE);
773: const String& src=params.as_string(1, PARAMETER_MUST_BE_STRING);
774:
775: Charset* from_charset=&r.charsets.client();
776:
777: if(params.count() > 2)
778: if(HashStringValue* options=params.as_hash(2)) {
779: int valid_options=0;
780: if(Value* vcharset_name=options->get(PA_CHARSET_NAME)){
1.229 moko 781: from_charset=&pa_charsets.get(vcharset_name->as_string());
1.220 moko 782: valid_options++;
783: }
784: if(valid_options!=options->count())
785: throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
786: }
787:
788: bool mode_js;
789: if(mode==UNESCAPE_MODE_JS){
790: mode_js=true;
791: } else if(mode==UNESCAPE_MODE_URI){
792: mode_js=false;
793: } else {
1.228 moko 794: throw Exception(PARSER_RUNTIME, &mode, "is invalid mode, must be either '" UNESCAPE_MODE_JS "' or '" UNESCAPE_MODE_URI "'");
1.220 moko 795: }
796:
797: const char* unescaped=unescape_chars(src.cstr(), src.length(), from_charset, mode_js);
798: if(*unescaped){
1.221 moko 799: const String* result=new String(Charset::transcode(unescaped, *from_charset, r.charsets.source()), String::L_TAINTED);
1.233 moko 800: r.write(*result);
1.220 moko 801: }
802: }
803:
1.225 moko 804: static void _contains(Request& r, MethodParams& params) {
805: // empty or whitespace string is hash compatible
806: GET_SELF(r, VString).get_element(params.as_string(0, "key must be string"));
807: // ignoring result as it allways null
1.233 moko 808: r.write(VBool::get(false));
1.225 moko 809: }
810:
1.41 paf 811: // constructor
812:
1.126 paf 813: MString::MString(): Methoded("string") {
1.1 paf 814: // ^string.length[]
1.41 paf 815: add_native_method("length", Method::CT_DYNAMIC, _length, 0, 0);
1.6 paf 816:
1.1 paf 817: // ^string.int[]
1.72 parser 818: // ^string.int(default)
819: add_native_method("int", Method::CT_DYNAMIC, _int, 0, 1);
1.1 paf 820: // ^string.double[]
1.72 parser 821: // ^string.double(default)
822: add_native_method("double", Method::CT_DYNAMIC, _double, 0, 1);
1.151 misha 823: // ^void.bool[]
824: // ^void.bool(default)
825: add_native_method("bool", Method::CT_DYNAMIC, _bool, 0, 1);
1.9 paf 826:
1.165 misha 827: // ^string.format[format]
1.41 paf 828: add_native_method("format", Method::CT_DYNAMIC, _string_format, 1, 1);
1.14 paf 829:
1.15 paf 830: // ^string.left(n)
1.41 paf 831: add_native_method("left", Method::CT_DYNAMIC, _left, 1, 1);
1.15 paf 832: // ^string.right(n)
1.41 paf 833: add_native_method("right", Method::CT_DYNAMIC, _right, 1, 1);
1.165 misha 834: // ^string.mid(p)
1.15 paf 835: // ^string.mid(p;n)
1.82 parser 836: add_native_method("mid", Method::CT_DYNAMIC, _mid, 1, 2);
1.16 paf 837:
838: // ^string.pos[substr]
1.165 misha 839: // ^string.pos[substr](n)
840: add_native_method("pos", Method::CT_DYNAMIC, _pos, 1, 2);
1.17 paf 841:
1.118 paf 842: // ^string.split[delim]
843: // ^string.split[delim][options]
1.156 misha 844: // ^string.split[delim][options][column name]
845: add_native_method("split", Method::CT_DYNAMIC, _split, 1, 3);
1.236 moko 846: // old names for backward compatibility
847: // ^string.lsplit[delim]
848: add_native_method("lsplit", Method::CT_DYNAMIC, _lsplit, 1, 1);
849: // ^string.rsplit[delim]
850: add_native_method("rsplit", Method::CT_DYNAMIC, _rsplit, 1, 1);
851:
1.32 paf 852: // ^string.match[regexp][options]
853: // ^string.match[regexp][options]{replacement-code}
1.192 misha 854: // ^string.match[regexp][options]{replacement-code}{code-if-nothing-is-found}
855: add_native_method("match", Method::CT_DYNAMIC, _match, 1, 4);
1.49 parser 856:
1.165 misha 857: // ^string.upper[]
1.49 parser 858: add_native_method("upper", Method::CT_DYNAMIC, _upper, 0, 0);
1.165 misha 859: // ^string.lower[]
1.49 parser 860: add_native_method("lower", Method::CT_DYNAMIC, _lower, 0, 0);
1.53 parser 861:
1.189 misha 862: // ^string:sql{query}
863: // ^string:sql{query}[options hash]
1.67 parser 864: add_native_method("sql", Method::CT_STATIC, _sql, 1, 2);
1.71 parser 865:
866: // ^string.replace[table]
1.201 misha 867: add_native_method("replace", Method::CT_DYNAMIC, _replace, 1, 2);
1.79 parser 868:
1.189 misha 869: // ^string.save[append][file]
870: // ^string.save[file]
871: // ^string.save[file][$.append(true) $.charset[...]]
1.87 paf 872: add_native_method("save", Method::CT_DYNAMIC, _save, 1, 2);
1.109 paf 873:
1.112 paf 874: // ^string.normalize[]
875: add_native_method("normalize", Method::CT_DYNAMIC, _normalize, 0, 0);
1.133 paf 876:
877: // ^string.trim[[start|both|end][;chars]]
878: add_native_method("trim", Method::CT_DYNAMIC, _trim, 0, 2);
1.139 paf 879:
1.146 paf 880: // ^string.base64[] << encode
1.215 moko 881: // ^string:base64[encoded string] << decode
1.202 misha 882: add_native_method("base64", Method::CT_ANY, _base64, 0, 2);
1.167 misha 883:
1.215 moko 884: // ^string.idna[] << encode
885: // ^string:idna[encoded string] << decode
886: add_native_method("idna", Method::CT_ANY, _idna, 0, 1);
887:
1.168 misha 888: // ^string.js-escape[]
1.220 moko 889: add_native_method("js-escape", Method::CT_DYNAMIC, _js_escape, 0, 0);
1.189 misha 890:
1.168 misha 891: // ^string:js-unescape[escaped%uXXXXstring]
1.220 moko 892: add_native_method("js-unescape", Method::CT_STATIC, _js_unescape, 1, 1);
893:
894: // ^string:unescape[js|uri;escaped;$.charset[...]]
895: add_native_method("unescape", Method::CT_STATIC, _unescape, 2, 3);
1.225 moko 896:
897: // ^string.contains[key] for hash compatibility
898: add_native_method("contains", Method::CT_DYNAMIC, _contains, 1, 1);
1.220 moko 899: }
E-mail: