Annotation of parser3/src/classes/math.C, revision 1.107
1.1 parser 1: /** @file
2: Parser: @b math parser class.
3:
1.101 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.28 paf 6:
7: portions from gen_uuid.c,
8: Copyright (C) 1996, 1997, 1998, 1999 Theodore Ts'o.
1.22 paf 9: */
1.1 parser 10:
1.34 paf 11: #include "pa_vmethod_frame.h"
1.1 parser 12: #include "pa_common.h"
1.90 moko 13: #include "pa_base64.h"
1.1 parser 14: #include "pa_vint.h"
1.2 parser 15: #include "pa_vmath.h"
1.69 moko 16: #include "pa_vfile.h"
1.1 parser 17: #include "pa_request.h"
1.19 paf 18: #include "pa_md5.h"
1.65 moko 19: #include "pa_sha2.h"
1.55 misha 20: #include "pa_random.h"
1.1 parser 21:
1.34 paf 22: #ifdef HAVE_CRYPT
1.76 moko 23: extern "C" char *crypt(const char* , const char* );
1.20 paf 24: #endif
25:
1.107 ! moko 26: volatile const char * IDENT_MATH_C="$Id: math.C,v 1.106 2024/09/28 14:37:53 moko Exp $";
1.59 moko 27:
1.1 parser 28: // defines
29:
1.20 paf 30: #define MAX_SALT 8
1.1 parser 31:
32: // class
33:
1.34 paf 34: class MMath: public Methoded {
1.81 moko 35: public: // Methoded
36: bool used_directly() { return false; }
37:
1.1 parser 38: public:
1.34 paf 39: MMath();
1.1 parser 40: };
41:
1.34 paf 42: // global variables
43:
1.81 moko 44: DECLARE_CLASS_VAR(math, new MMath);
1.34 paf 45:
1.1 parser 46: // methods
1.34 paf 47:
48: static void _random(Request& r, MethodParams& params) {
1.41 paf 49: double top=params.as_double(0, "range must be expression", r);
1.103 moko 50: if(top<1 || top>INT_MAX)
51: throw Exception(PARSER_RUNTIME, 0, "top(%.15g) must be [1..%u]", top, INT_MAX);
1.83 moko 52: r.write(*new VInt(_random(uint(top))));
1.1 parser 53: }
54:
55:
1.20 paf 56: typedef double(*math1_func_ptr)(double);
1.3 parser 57: static double frac(double param) { return param-trunc(param); }
58: static double degrees(double param) { return param /PI *180; }
59: static double radians(double param) { return param /180 *PI; }
1.1 parser 60:
1.34 paf 61: static void math1(Request& r, MethodParams& params, math1_func_ptr func) {
1.41 paf 62: double param=params.as_double(0, "parameter must be expression", r);
63: double result=func(param);
1.83 moko 64: r.write(*new VDouble(result));
1.1 parser 65: }
66:
67: #define MATH1(name) \
1.34 paf 68: static void _##name(Request& r, MethodParams& params) {\
69: math1(r, params, &name);\
1.1 parser 70: }
1.74 moko 71:
1.1 parser 72: #define MATH1P(name_parser, name_c) \
1.34 paf 73: static void _##name_parser(Request& r, MethodParams& params) {\
74: math1(r, params, &name_c);\
1.1 parser 75: }
1.74 moko 76:
77: MATH1(round)
78: MATH1(floor)
79: MATH1P(ceiling, ceil)
80: MATH1(trunc)
81: MATH1(frac)
82: MATH1P(abs, fabs)
83: MATH1(sign)
84: MATH1(exp)
85: MATH1(log)
86: MATH1(log10)
87: MATH1(sin)
88: MATH1(asin)
89: MATH1(cos)
90: MATH1(acos)
91: MATH1(tan)
92: MATH1(atan)
93: MATH1(degrees)
94: MATH1(radians)
95: MATH1(sqrt)
1.1 parser 96:
97:
98: typedef double (*math2_func_ptr)(double, double);
1.34 paf 99: static void math2(Request& r, MethodParams& params, math2_func_ptr func) {
1.41 paf 100: double a=params.as_double(0, "parameter must be expression", r);
101: double b=params.as_double(1, "parameter must be expression", r);
102: double result=func(a, b);
1.83 moko 103: r.write(*new VDouble(result));
1.1 parser 104: }
105:
106: #define MATH2(name) \
1.34 paf 107: static void _##name(Request& r, MethodParams& params) {\
108: math2(r, params, &name);\
1.1 parser 109: }
1.74 moko 110:
111: MATH2(pow)
1.105 moko 112: MATH2(atan2)
1.1 parser 113:
1.77 moko 114: inline bool is_salt_body_char(unsigned char c) {
1.73 moko 115: return pa_isalnum(c) || c == '.' || c=='/';
1.20 paf 116: }
1.77 moko 117:
1.34 paf 118: static size_t calc_prefix_size(const char* salt) {
1.37 paf 119: if(strlen(salt)) {
1.39 paf 120: if(!is_salt_body_char((unsigned char)salt[0])) { // $... {...
1.34 paf 121: const char* cur=salt+1; // skip
1.39 paf 122: while(is_salt_body_char((unsigned char)*cur++)) // ...$ ...}
1.20 paf 123: ;
124: return cur-salt;
125: } else
126: return 0;
127: } else
128: return 0;
129: }
1.34 paf 130: static void _crypt(Request& r, MethodParams& params) {
131: const char* password=params.as_string(0, "password must be string").cstr();
132: const char* maybe_bodyless_salt=params.as_string(1, "salt must be string").cstr();
1.20 paf 133:
134: size_t prefix_size=calc_prefix_size(maybe_bodyless_salt);
1.34 paf 135: const char* normal_salt;
1.20 paf 136: char normalize_buf[MAX_STRING];
137: if(prefix_size==strlen(maybe_bodyless_salt)) { // bodyless?
1.99 moko 138: pa_strncpy(normalize_buf, maybe_bodyless_salt, MAX_STRING-MAX_SALT);
1.20 paf 139: char *cur=normalize_buf+strlen(normalize_buf);
140: // sould add up MAX_SALT random chars
141: static unsigned char itoa64[] = /* 0 ... 63 => ASCII - 64 */
142: "./0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
143: for(int i=0; i<MAX_SALT; i++)
144: *cur++=itoa64[_random(64)];
145: *cur=0;
146: normal_salt=normalize_buf;
147: } else
148: normal_salt=maybe_bodyless_salt;
1.19 paf 149:
1.54 misha 150: /* FreeBSD style MD5 string
151: */
152: if(strncmp(normal_salt, PA_MD5PW_ID, PA_MD5PW_IDLEN) == 0) {
1.19 paf 153: const size_t sample_size=120;
1.34 paf 154: char *sample_buf=new(PointerFreeGC) char[sample_size];
155: pa_MD5Encode((const unsigned char *)password,
156: (const unsigned char *)normal_salt, sample_buf, sample_size);
157: String sample(sample_buf);
1.83 moko 158: r.write(sample);
1.54 misha 159: } else {
1.20 paf 160: #ifdef HAVE_CRYPT
1.34 paf 161: const char* static_sample_buf=crypt(password, normal_salt);
162: if(!static_sample_buf // nothing generated
163: || !static_sample_buf[0] // generated nothing
164: || strncmp(static_sample_buf, normal_salt, prefix_size)!=0) // salt prefix not preserved
1.45 misha 165: throw Exception(PARSER_RUNTIME,
1.34 paf 166: 0,
167: "crypt on this platform does not support '%.*s' salt prefix", prefix_size, normal_salt);
1.20 paf 168:
1.83 moko 169: r.write(String(pa_strdup(static_sample_buf)));
1.20 paf 170: #else
1.45 misha 171: throw Exception(PARSER_RUNTIME,
1.34 paf 172: 0,
1.19 paf 173: "salt must start with '" PA_MD5PW_ID "'");
1.20 paf 174: #endif
175: }
1.19 paf 176: }
177:
1.34 paf 178: static void _md5(Request& r, MethodParams& params) {
1.72 moko 179: const char *string=params.as_string(0, PARAMETER_MUST_BE_STRING).cstr_to_string_body_untaint(String::L_AS_IS, r.connection(false), &r.charsets).cstr();
1.27 paf 180:
181: PA_MD5_CTX context;
182: unsigned char digest[16];
1.34 paf 183: pa_MD5Init(&context);
184: pa_MD5Update(&context, (const unsigned char*)string, strlen(string));
185: pa_MD5Final(digest, &context);
1.27 paf 186:
1.83 moko 187: r.write(*new String(hex_string(digest, sizeof(digest), false)));
1.26 paf 188: }
189:
1.49 misha 190:
191: //SHA-1:
192:
1.50 misha 193: struct SHA1Context {
1.54 misha 194: unsigned Message_Digest[5], Length_Low, Length_High;
195: unsigned int Message_Block[64];
196: int Message_Block_Index, Computed, Corrupted;
1.50 misha 197: };
1.49 misha 198:
199: #define SHA1CircularShift(bits,word) ((((word) << (bits)) & 0xFFFFFFFF)|((word) >> (32-(bits))))
200: void SHA1ProcessMessageBlock(SHA1Context *);
201: void SHA1PadMessage(SHA1Context *);
1.50 misha 202: void SHA1Reset(SHA1Context *context) {
1.54 misha 203: context->Length_Low = context->Length_High = context->Message_Block_Index = 0;
204: context->Message_Digest[0] = 0x67452301;
205: context->Message_Digest[1] = 0xEFCDAB89;
206: context->Message_Digest[2] = 0x98BADCFE;
207: context->Message_Digest[3] = 0x10325476;
208: context->Message_Digest[4] = 0xC3D2E1F0;
209: context->Computed = context->Corrupted = 0;
1.50 misha 210: }
1.49 misha 211:
1.50 misha 212: int SHA1Result(SHA1Context *context) {
1.54 misha 213: if (context->Corrupted)
214: return 0;
215: if (!context->Computed) {
216: SHA1PadMessage(context);
217: context->Computed = 1;
218: }
219: return 1;
220: }
221:
222: void SHA1Input(SHA1Context *context, const unsigned char *message_array, unsigned length) {
223: if (!length)
224: return;
225: if (context->Computed || context->Corrupted) {
226: context->Corrupted = 1;
227: return;
228: }
229:
230: while(length-- && !context->Corrupted) {
231: context->Message_Block[context->Message_Block_Index++] = (*message_array & 0xFF);
232: context->Length_Low += 8;
233: context->Length_Low &= 0xFFFFFFFF;
234: if (!context->Length_Low && !(context->Length_High=((1+context->Length_High)&0xFFFFFFFF)))
235: context->Corrupted = 1; // too long message
236: if (context->Message_Block_Index == 64)
237: SHA1ProcessMessageBlock(context);
238: message_array++;
1.49 misha 239: }
1.50 misha 240: }
1.49 misha 241:
1.50 misha 242: void SHA1ProcessMessageBlock(SHA1Context *context) {
1.54 misha 243: const unsigned K[] = {0x5A827999, 0x6ED9EBA1, 0x8F1BBCDC, 0xCA62C1D6 };
244: int t;
245: unsigned temp, W[80], buf[5];
246: unsigned &A=buf[0], &B=buf[1], &C=buf[2], &D=buf[3], &E=buf[4];
1.50 misha 247:
1.54 misha 248: for(t = 0; t < 16; t++)
249: W[t] = (((unsigned) context->Message_Block[t * 4]) << 24) | (((unsigned) context->Message_Block[t * 4 + 1]) << 16) | (((unsigned) context->Message_Block[t * 4 + 2]) << 8) | ((unsigned) context->Message_Block[t * 4 + 3]);
1.50 misha 250:
1.54 misha 251: for(t = 16; t < 80; t++)
252: W[t] = SHA1CircularShift(1,W[t-3] ^ W[t-8] ^ W[t-14] ^ W[t-16]);
1.49 misha 253:
1.54 misha 254: memcpy (buf, context->Message_Digest, sizeof(buf));
255: for(t = 0; t < 20; t++) {
256: temp = (SHA1CircularShift(5,A) + ((B & C) | ((~B) & D)) + E + W[t] + K[0]) & 0xFFFFFFFF;
257: E = D; D = C;
258: C = SHA1CircularShift(30,B);
259: B = A; A = temp;
1.49 misha 260: }
261:
1.54 misha 262: for(t = 20; t < 40; t++) {
263: temp = (SHA1CircularShift(5,A) + (B ^ C ^ D) + E + W[t] + K[1]) & 0xFFFFFFFF;
264: E = D; D = C;
265: C = SHA1CircularShift(30,B);
266: B = A; A = temp;
1.49 misha 267: }
268:
1.54 misha 269: for(t = 40; t < 60; t++) {
270: temp = (SHA1CircularShift(5,A) + ((B & C) | (B & D) | (C & D)) + E + W[t] + K[2]) & 0xFFFFFFFF;
271: E = D; D = C;
272: C = SHA1CircularShift(30,B);
273: B = A; A = temp;
1.49 misha 274: }
275:
1.54 misha 276: for(t = 60; t < 80; t++) {
277: temp = (SHA1CircularShift(5,A) + (B ^ C ^ D) + E + W[t] + K[3]) & 0xFFFFFFFF;
278: E = D; D = C;
279: C = SHA1CircularShift(30,B);
280: B = A; A = temp;
1.49 misha 281: }
282:
1.54 misha 283: for (t = 0; t < 5; t++)
284: context->Message_Digest[t] = (context->Message_Digest[t] + buf[t]) & 0xFFFFFFFF;
1.50 misha 285:
1.54 misha 286: context->Message_Block_Index = 0;
1.50 misha 287: }
1.49 misha 288:
1.50 misha 289: void SHA1PadMessage(SHA1Context *context) {
1.54 misha 290: context->Message_Block[context->Message_Block_Index++] = 0x80;
1.50 misha 291: if (context->Message_Block_Index > 56) {
292: //was 55, one shift
1.54 misha 293: while(context->Message_Block_Index < 64)
294: context->Message_Block[context->Message_Block_Index++] = 0;
295: SHA1ProcessMessageBlock(context);
296: while(context->Message_Block_Index < 56)
297: context->Message_Block[context->Message_Block_Index++] = 0;
1.50 misha 298: } else
1.54 misha 299: while(context->Message_Block_Index < 56)
300: context->Message_Block[context->Message_Block_Index++] = 0;
301: context->Message_Block[56] = (context->Length_High >> 24) & 0xFF;
302: context->Message_Block[57] = (context->Length_High >> 16) & 0xFF;
303: context->Message_Block[58] = (context->Length_High >> 8) & 0xFF;
304: context->Message_Block[59] = (context->Length_High) & 0xFF;
305: context->Message_Block[60] = (context->Length_Low >> 24) & 0xFF;
306: context->Message_Block[61] = (context->Length_Low >> 16) & 0xFF;
307: context->Message_Block[62] = (context->Length_Low >> 8) & 0xFF;
308: context->Message_Block[63] = (context->Length_Low) & 0xFF;
309: SHA1ProcessMessageBlock(context);
1.50 misha 310: }
1.49 misha 311:
1.62 moko 312: #ifdef PA_BIG_ENDIAN
313: #define SWAP(n) (n)
314: #else
315: #define SWAP(n) (((n) << 24) | (((n) & 0xff00) << 8) | (((n) >> 8) & 0xff00) | ((n) >> 24))
316: #endif
317:
1.64 moko 318: void SHA1ReadDigest(void *buf, SHA1Context *c)
1.62 moko 319: {
1.64 moko 320: if(!SHA1Result(c))
1.106 moko 321: throw Exception (PARSER_RUNTIME, 0, "Cannot compute SHA1");
1.62 moko 322:
1.64 moko 323: ((uint32_t *)buf)[0] = SWAP(c->Message_Digest[0]);
324: ((uint32_t *)buf)[1] = SWAP(c->Message_Digest[1]);
325: ((uint32_t *)buf)[2] = SWAP(c->Message_Digest[2]);
326: ((uint32_t *)buf)[3] = SWAP(c->Message_Digest[3]);
327: ((uint32_t *)buf)[4] = SWAP(c->Message_Digest[4]);
1.62 moko 328: }
329:
1.54 misha 330: static void _sha1(Request& r, MethodParams& params) {
1.72 moko 331: const char *string = params.as_string(0, PARAMETER_MUST_BE_STRING).cstr_to_string_body_untaint(String::L_AS_IS, r.connection(false), &r.charsets).cstr();
1.49 misha 332:
1.54 misha 333: SHA1Context c;
1.64 moko 334: unsigned char digest[20];
1.54 misha 335: SHA1Reset (&c);
336: SHA1Input (&c, (const unsigned char*)string, strlen(string));
1.64 moko 337: SHA1ReadDigest(digest, &c);
1.54 misha 338:
1.83 moko 339: r.write(*new String(hex_string(digest, sizeof(digest), false)));
1.62 moko 340: }
341:
1.87 moko 342: String::C getData(Value& vdata, Request& r){
343: if(const String* sdata=vdata.get_string()){
344: String::Body body=sdata->cstr_to_string_body_untaint(String::L_AS_IS, r.connection(false), &r.charsets); // explode content, honor tainting changes
345: return String::C(body.cstr(), body.length());
346: } else {
1.98 moko 347: VFile *file=vdata.as_vfile();
1.87 moko 348: return String::C(file->value_ptr(),file->value_size());
349: }
350: }
351:
1.64 moko 352: void memxor(char *dest, const char *src, size_t n){
353: for (;n>0;n--) *dest++ ^= *src++;
354: }
355:
1.62 moko 356: #define IPAD 0x36
357: #define OPAD 0x5c
358:
1.65 moko 359: #define HMAC(key,init,update,final,blocklen,digestlen){ \
1.64 moko 360: unsigned char tempdigest[digestlen], keydigest[digestlen]; \
361: size_t keylen=strlen(key); \
1.65 moko 362: /* Reduce the key's size, so that it becomes <= blocklen bytes. */ \
363: if (keylen > blocklen){ \
1.64 moko 364: init(&c); \
365: update(&c,(const unsigned char*)hmac, keylen); \
366: final(keydigest, &c); \
367: key = (char *)keydigest; \
368: keylen = digestlen; \
369: } \
370: /* Compute TEMP from KEY and STRING. */ \
1.65 moko 371: char block[blocklen]; \
372: memset (block, IPAD, blocklen); \
1.64 moko 373: memxor (block, key, keylen); \
374: init(&c); \
1.65 moko 375: update(&c, (const unsigned char*)block, blocklen); \
1.69 moko 376: update(&c, (const unsigned char*)data.str, data.length); \
1.64 moko 377: final(tempdigest, &c); \
378: /* Compute result from KEY and TEMP. */ \
1.65 moko 379: memset (block, OPAD, blocklen); \
1.64 moko 380: memxor (block, key, keylen); \
381: init(&c); \
1.65 moko 382: update(&c, (const unsigned char*)block, blocklen); \
1.64 moko 383: update(&c, (const unsigned char*)tempdigest, digestlen); \
1.62 moko 384: }
1.55 misha 385:
1.62 moko 386: static void _digest(Request& r, MethodParams& params) {
1.63 moko 387: const String &smethod = params.as_string(0, PARAMETER_MUST_BE_STRING);
1.69 moko 388:
1.87 moko 389: String::C data=getData(params.as_no_junction(1, "parameter must be string or file"), r);
1.62 moko 390:
1.65 moko 391: enum Method { M_MD5, M_SHA1, M_SHA256, M_SHA512 } method;
1.62 moko 392:
1.63 moko 393: if (smethod == "md5") method = M_MD5;
394: else if (smethod == "sha1" ) method = M_SHA1;
1.65 moko 395: else if (smethod == "sha256" ) method = M_SHA256;
396: else if (smethod == "sha512" ) method = M_SHA512;
1.86 moko 397: else throw Exception(PARSER_RUNTIME, &smethod, "must be 'md5' or 'sha1' or 'sha256' or 'sha512'");
1.62 moko 398:
399: const char *hmac=0;
1.63 moko 400: enum Format { F_HEX, F_BASE64 } format = F_HEX;
1.62 moko 401:
402: if(params.count() == 3)
403: if(HashStringValue* options=params.as_hash(2)) {
404: int valid_options=0;
405: if(Value* value=options->get("hmac")) {
406: hmac=value->as_string().cstr();
407: valid_options++;
408: }
1.63 moko 409: if(Value* value=options->get("format")) {
410: const String& sformat=value->as_string();
411: if (sformat == "hex") format = F_HEX;
412: else if (sformat == "base64" ) format = F_BASE64;
413: else throw Exception(PARSER_RUNTIME, &sformat, "must be 'hex' or 'base64'");
1.62 moko 414: valid_options++;
415: }
416: if(valid_options!=options->count())
417: throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
418: }
419:
420: String::C digest;
421:
1.63 moko 422: if(method == M_MD5){
1.64 moko 423: PA_MD5_CTX c;
424: if(hmac){
1.65 moko 425: HMAC(hmac, pa_MD5Init, pa_MD5Update, pa_MD5Final, 64, 16);
1.64 moko 426: } else {
427: pa_MD5Init(&c);
1.69 moko 428: pa_MD5Update(&c, (const unsigned char*)data.str, data.length);
1.64 moko 429: }
1.62 moko 430: char *str=(char *)pa_malloc(16);
1.64 moko 431: pa_MD5Final((unsigned char *)str, &c);
1.62 moko 432: digest = String::C(str, 16);
433: }
434:
1.63 moko 435: if(method == M_SHA1){
1.62 moko 436: SHA1Context c;
437: if(hmac){
1.65 moko 438: HMAC(hmac, SHA1Reset, SHA1Input, SHA1ReadDigest, 64, 20);
1.62 moko 439: } else {
1.64 moko 440: SHA1Reset(&c);
1.69 moko 441: SHA1Input(&c, (const unsigned char*)data.str, data.length);
1.62 moko 442: }
443: char *str=(char *)pa_malloc(20);
1.64 moko 444: SHA1ReadDigest(str, &c);
1.62 moko 445: digest = String::C(str, 20);
446: }
1.49 misha 447:
1.65 moko 448: if(method == M_SHA256){
449: SHA256_CTX c;
450: if(hmac){
451: HMAC(hmac, pa_SHA256_Init, pa_SHA256_Update, pa_SHA256_Final, 64, SHA256_DIGEST_LENGTH);
452: } else {
453: pa_SHA256_Init(&c);
1.69 moko 454: pa_SHA256_Update(&c, (const unsigned char*)data.str, data.length);
1.65 moko 455: }
456: char *str=(char *)pa_malloc(SHA256_DIGEST_LENGTH);
457: pa_SHA256_Final((unsigned char *)str, &c);
458: digest = String::C(str, SHA256_DIGEST_LENGTH);
459: }
460:
461: if(method == M_SHA512){
462: SHA512_CTX c;
463: if(hmac){
464: HMAC(hmac, pa_SHA512_Init, pa_SHA512_Update, pa_SHA512_Final, 128, SHA512_DIGEST_LENGTH);
465: } else {
466: pa_SHA512_Init(&c);
1.69 moko 467: pa_SHA512_Update(&c, (const unsigned char*)data.str, data.length);
1.65 moko 468: }
469: char *str=(char *)pa_malloc(SHA512_DIGEST_LENGTH);
470: pa_SHA512_Final((unsigned char *)str, &c);
471: digest = String::C(str, SHA512_DIGEST_LENGTH);
472: }
473:
1.63 moko 474: if(format == F_HEX){
1.83 moko 475: r.write(*new String(hex_string((unsigned char *)digest.str, digest.length, false)));
1.62 moko 476: }
1.63 moko 477: if(format == F_BASE64){
1.94 moko 478: r.write(*new String(pa_base64_encode(digest.str, digest.length, Base64Options(false /*no wrap*/))));
1.62 moko 479: }
1.54 misha 480: }
1.49 misha 481:
1.97 moko 482: static void _uuid(Request& r, MethodParams& params) {
483: bool lower=false;
484: bool solid=false;
485:
486: if (params.count() == 1)
487: if (HashStringValue* options = params.as_hash(0)) {
488: int valid_options = 0;
489: if (Value* vlower = options->get("lower")) {
490: lower = r.process(*vlower).as_bool();
491: valid_options++;
492: }
493: if (Value* vsolid = options->get("solid")) {
494: solid = r.process(*vsolid).as_bool();
495: valid_options++;
496: }
497: if (valid_options != options->count())
498: throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
499: }
500:
501: r.write(*new String(get_uuid_cstr(lower, solid)));
1.28 paf 502: }
503:
1.100 moko 504: static void _uuid7(Request& r, MethodParams& params) {
505: bool lower=false;
506: bool solid=false;
507:
508: if (params.count() == 1)
509: if (HashStringValue* options = params.as_hash(0)) {
510: int valid_options = 0;
511: if (Value* vlower = options->get("lower")) {
512: lower = r.process(*vlower).as_bool();
513: valid_options++;
514: }
515: if (Value* vsolid = options->get("solid")) {
516: solid = r.process(*vsolid).as_bool();
517: valid_options++;
518: }
519: if (valid_options != options->count())
520: throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
521: }
522:
523: r.write(*new String(get_uuid7_cstr(lower, solid)));
524: }
525:
1.97 moko 526: static void _uid64(Request& r, MethodParams& params) {
527: bool lower = false;
528:
529: if (params.count() == 1)
530: if (HashStringValue* options = params.as_hash(0)) {
531: int valid_options = 0;
532: if (Value* vlower = options->get("lower")) {
533: lower = r.process(*vlower).as_bool();
534: valid_options++;
535: }
536: if (valid_options != options->count())
537: throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
538: }
539:
1.30 paf 540: unsigned char id[64/8];
541: random(&id, sizeof(id));
542:
1.97 moko 543: r.write(*new String(hex_string(id, sizeof(id), !lower)));
1.30 paf 544: }
545:
1.43 misha 546: static void _crc32(Request& r, MethodParams& params) {
1.51 misha 547: const char *string=params.as_string(0, PARAMETER_MUST_BE_STRING).cstr();
1.102 moko 548: r.write(*new VDouble((uint)pa_crc32(string, strlen(string))));
1.43 misha 549: }
550:
1.89 moko 551: static const char* abc_hex = "0123456789ABCDEF";
1.87 moko 552:
1.89 moko 553: static unsigned char hex_lookup[256] = {
1.87 moko 554: 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
555: 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
556: 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
557: 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 0, 0, 0, 0, 0, 0,
558: 0,10,11,12,13,14,15, 0, 0, 0, 0, 0, 0, 0, 0, 0,
559: 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0,
560: 0,10,11,12,13,14,15, 0, 0, 0, 0, 0, 0, 0, 0, 0
561: };
562:
1.89 moko 563: static unsigned char abc_lookup[256] = {};
564: static unsigned char abc_256_lookup[256] = {};
1.87 moko 565:
1.89 moko 566: inline unsigned char *init_abc_256() {
567: if(!abc_256_lookup[255])
568: for(int i=0; i<256; i++) abc_256_lookup[i] = (unsigned char)i;
1.87 moko 569: return abc_256_lookup;
1.57 misha 570: }
571:
572: static void _convert(Request& r, MethodParams& params) {
1.87 moko 573: String::C data=getData(params.as_no_junction(0, "parameter must be string or file"), r);
1.60 moko 574:
1.87 moko 575: bool abc_mode = true;
576: unsigned char *lookup;
577: const char *abc_from;
578: int base_from;
579:
1.89 moko 580: if(params[1].is_string()) {
1.87 moko 581: abc_from = params[1].get_string()->cstr();
582: base_from = strlen(abc_from);
583: if(base_from < 2)
584: throw Exception(PARSER_RUNTIME, 0, "alphabet 'from' must contain at least 2 characters");
585: lookup = abc_lookup;
586: memset(abc_lookup,0,sizeof(abc_lookup));
587: for(int i=0; i<base_from; i++) abc_lookup[(unsigned char)abc_from[i]] = (unsigned char)i;
588: } else {
589: base_from=params.as_int(1, "base 'from' must be integer or string", r);
590: if(base_from < 2 || base_from > 16 && base_from != 256)
591: throw Exception(PARSER_RUNTIME, 0, "base 'from' must be an integer from 2 to 16 or 256");
1.89 moko 592: if (base_from == 256) {
1.87 moko 593: abc_from = "";
594: lookup = init_abc_256();
595: } else {
596: abc_mode = false;
597: abc_from = abc_hex;
598: lookup = hex_lookup;
599: }
600: }
1.57 misha 601:
1.87 moko 602: const char *abc_to;
603: int base_to;
1.60 moko 604:
1.89 moko 605: if(params[2].is_string()) {
1.87 moko 606: abc_to=params[2].get_string()->cstr();
607: base_to=strlen(abc_to);
608: if(base_to < 2)
609: throw Exception(PARSER_RUNTIME, 0, "alphabet 'to' must contain at least 2 characters");
610: } else {
611: base_to=params.as_int(2, "base 'to' must be integer or string", r);
612: if(base_to < 2 || base_to > 16 && base_to != 256)
613: throw Exception(PARSER_RUNTIME, 0, "base 'to' must be an integer from 2 to 16 or 256");
1.89 moko 614: if (base_to == 256) {
1.87 moko 615: abc_to = (char *)init_abc_256();
616: } else {
617: abc_to = abc_hex;
618: }
619: }
620:
621: VFile* result_file = 0;
1.60 moko 622:
1.87 moko 623: if(params.count() == 4)
624: if(HashStringValue* options=params.as_hash(3)) {
625: int valid_options=0;
626: if(Value* value=options->get("format")) {
627: const String& sformat=value->as_string();
628: if (sformat == "file" ) result_file = new VFile;
629: else if (sformat != "string") throw Exception(PARSER_RUNTIME, &sformat, "must be 'string' or 'file'");
630: valid_options++;
631: }
632: if(valid_options!=options->count())
633: throw Exception(PARSER_RUNTIME, 0, CALLED_WITH_INVALID_OPTION);
634: }
1.60 moko 635:
636: bool negative=false;
1.87 moko 637:
638: // converting digits to their numeric values
639:
640: unsigned char *src=(unsigned char *)pa_strdup(data.str, data.length);
641: const unsigned char *src_end = src + data.length;
642:
643: unsigned char *c;
644:
645: if(abc_mode){
646:
1.89 moko 647: for(c=src;c<src_end;c++) {
1.87 moko 648: unsigned char digit=lookup[*c];
649: if(!digit && *c != abc_from[0])
1.107 ! moko 650: throw Exception("number.format", 0, "'%c' is an invalid digit", *c);
1.87 moko 651: *c=digit;
652: }
653:
654: } else {
655: // numbers mode, allow whitespace and sign
656:
657: while(isspace(*src))
658: src++;
659:
660: if(src[0]=='-') {
661: negative=true;
662: src++;
1.104 moko 663: if(!*src || isspace(*src))
1.107 ! moko 664: throw Exception("number.format", 0, "'-' is an invalid number");
1.87 moko 665: } else if(src[0]=='+') {
666: src++;
1.104 moko 667: if(!*src || isspace(*src))
1.107 ! moko 668: throw Exception("number.format", 0, "'+' is an invalid number");
1.87 moko 669: }
670:
671: for(c=src;c<src_end;c++) {
672: unsigned char digit=lookup[*c];
673: if(!digit && *c != abc_from[0] || digit>=base_from) {
674: for(unsigned char *s=c;s<src_end;s++)
675: if(!isspace(*s))
1.107 ! moko 676: throw Exception("number.format", 0, "'%c' is an invalid digit", *s);
1.87 moko 677: src_end=c;
678: break;
679: }
680: *c=digit;
681: }
682:
1.60 moko 683: }
684:
1.89 moko 685: if(src==src_end) {
1.87 moko 686: if(result_file)
687: r.write(*result_file);
688: return;
689: }
690:
1.92 moko 691: // core, using log since log2 is not present in FreeBSD < 8.X
1.87 moko 692:
1.95 moko 693: Array<char> remainders((size_t)round(data.length * log((double)base_from) / log((double)base_to)) + 1);
1.57 misha 694:
1.87 moko 695: do {
696: int carry = 0;
697: unsigned char *dst = src;
698: for (c=src; c<src_end; c++) {
699: carry = carry * base_from + *c;
700: if (carry >= base_to) {
1.95 moko 701: *(dst++) = (unsigned char)(carry / base_to);
1.87 moko 702: carry %= base_to;
703: } else if (dst > src) {
704: *(dst++) = 0;
705: }
706: }
707: src_end = dst;
708: remainders += abc_to[carry];
709: } while (src_end > src);
710:
711: // result processing
712:
713: size_t result_length = negative + remainders.count();
714: char *result_str = (char *)pa_malloc_atomic(result_length+1);
1.60 moko 715: if(negative)
1.87 moko 716: result_str[0] = '-';
1.95 moko 717: for(size_t i=0; i<remainders.count(); i++)
1.87 moko 718: result_str[result_length - 1 - i] = remainders[i];
719: result_str[result_length]='\0';
720:
1.89 moko 721: if(result_file) {
1.88 moko 722: result_file->set(true /*tainted*/, 0 /*binary*/, result_str, result_length, 0, 0, &r);
1.87 moko 723: r.write(*result_file);
724: } else {
1.91 moko 725: if(memchr(result_str, 0, result_length))
726: throw Exception(PARSER_RUNTIME, 0, "Invalid \\x00 character found while converting to string. Convert to file instead.");
727:
728: fix_line_breaks(result_str, result_length);
729:
730: if(result_length)
731: r.write(*new String(result_str, String::L_TAINTED));
1.87 moko 732: }
1.57 misha 733: }
734:
1.1 parser 735: // constructor
736:
1.34 paf 737: MMath::MMath(): Methoded("math") {
1.1 parser 738: // ^FUNC(expr)
1.105 moko 739: #define ADDN(name, N) \
740: add_native_method(#name, Method::CT_STATIC, _##name, N, N)
741: #define ADD1(name) ADDN(name, 1)
1.1 parser 742:
743: ADD1(round); ADD1(floor); ADD1(ceiling);
1.3 parser 744: ADD1(trunc); ADD1(frac);
1.1 parser 745: ADD1(abs); ADD1(sign);
1.46 misha 746: ADD1(exp);
1.48 misha 747: ADD1(log); ADD1(log10);
1.87 moko 748: ADD1(sin); ADD1(asin);
749: ADD1(cos); ADD1(acos);
1.105 moko 750: ADD1(tan); ADD1(atan); ADDN(atan2, 2);
1.3 parser 751: ADD1(degrees); ADD1(radians);
1.1 parser 752: ADD1(sqrt);
1.3 parser 753: ADD1(random);
1.1 parser 754:
1.49 misha 755: // ^math:pow(x;y)
1.105 moko 756: ADDN(pow, 2);
1.1 parser 757:
1.49 misha 758: // ^math:crypt[password;salt]
1.105 moko 759: ADDN(crypt, 2);
1.26 paf 760:
1.49 misha 761: // ^math:md5[string]
1.26 paf 762: ADD1(md5);
1.5 parser 763:
1.49 misha 764: // ^math:sha1[string]
765: ADD1(sha1);
766:
1.71 moko 767: // ^math:digest[method;string|file;options]
1.62 moko 768: add_native_method("digest", Method::CT_STATIC, _digest, 2, 3);
769:
1.49 misha 770: // ^math:crc32[string]
1.43 misha 771: ADD1(crc32);
772:
1.49 misha 773: // ^math:uuid[]
1.97 moko 774: // ^math:uuid[options hash]
775: add_native_method("uuid", Method::CT_STATIC, _uuid, 0, 1);
1.29 paf 776:
1.100 moko 777: // ^math:uuid7[]
778: // ^math:uuid7[options hash]
779: add_native_method("uuid7", Method::CT_STATIC, _uuid7, 0, 1);
780:
1.49 misha 781: // ^math:uid64[]
1.97 moko 782: // ^math:uid64[options hash]
783: add_native_method("uid64", Method::CT_STATIC, _uid64, 0, 1);
1.57 misha 784:
1.87 moko 785: // ^math:convert[number|file](base-from)|[abc_from](base-to)|[abc_to][options]
786: add_native_method("convert", Method::CT_STATIC, _convert, 3, 4);
1.1 parser 787: }
E-mail:
![[BACK]](/cvsweb/icons/back.gif){kind=icon}
Return to math.C CVS log ![[TXT]](/cvsweb/icons/text.gif){kind=icon}
![[DIR]](/cvsweb/icons/dir.gif){kind=icon}
Up to [parser3project] / parser3 / src / classes