/** @file
Parser: @b math parser class.
Copyright (c) 2001-2012 Art. Lebedev Studio (http://www.artlebedev.com)
Author: Alexandr Petrosian <paf@design.ru>(http://paf.design.ru)
portions from gen_uuid.c,
Copyright (C) 1996, 1997, 1998, 1999 Theodore Ts'o.
*/
#include "pa_vmethod_frame.h"
#include "pa_common.h"
#include "pa_vint.h"
#include "pa_vmath.h"
#include "pa_request.h"
#include "pa_md5.h"
#include "pa_random.h"
#ifdef WIN32
# define _WIN32_WINNT 0x400
# include <windows.h>
# include <wincrypt.h>
#endif
#ifdef HAVE_CRYPT
# ifdef HAVE_CRYPT_H
# include <crypt.h>
# endif
#else
extern char *crypt(const char* , const char* );
#endif
volatile const char * IDENT_MATH_C="$Id: math.C,v 1.61 2012/05/28 10:33:18 moko Exp $";
// defines
#define MAX_SALT 8
// class
class MMath: public Methoded {
public:
MMath();
public: // Methoded
bool used_directly() { return false; }
};
// global variables
DECLARE_CLASS_VAR(math, 0 /*fictive*/, new MMath);
// methods
static void _random(Request& r, MethodParams& params) {
double top=params.as_double(0, "range must be expression", r);
if(top<=0 || top>MAX_UINT)
throw Exception(PARSER_RUNTIME,
0,
"top(%g) must be [1..%u]", top, MAX_UINT);
r.write_no_lang(*new VInt(_random(uint(top))));
}
typedef double(*math1_func_ptr)(double);
static double frac(double param) { return param-trunc(param); }
static double degrees(double param) { return param /PI *180; }
static double radians(double param) { return param /180 *PI; }
static void math1(Request& r, MethodParams& params, math1_func_ptr func) {
double param=params.as_double(0, "parameter must be expression", r);
double result=func(param);
r.write_no_lang(*new VDouble(result));
}
#define MATH1(name) \
static void _##name(Request& r, MethodParams& params) {\
math1(r, params, &name);\
}
#define MATH1P(name_parser, name_c) \
static void _##name_parser(Request& r, MethodParams& params) {\
math1(r, params, &name_c);\
}
MATH1(round); MATH1(floor); MATH1P(ceiling, ceil);
MATH1(trunc); MATH1(frac);
MATH1P(abs, fabs); MATH1(sign);
MATH1(exp);
MATH1(log); MATH1(log10);
MATH1(sin); MATH1(asin);
MATH1(cos); MATH1(acos);
MATH1(tan); MATH1(atan);
MATH1(degrees); MATH1(radians);
MATH1(sqrt);
typedef double (*math2_func_ptr)(double, double);
static void math2(Request& r, MethodParams& params, math2_func_ptr func) {
double a=params.as_double(0, "parameter must be expression", r);
double b=params.as_double(1, "parameter must be expression", r);
double result=func(a, b);
r.write_no_lang(*new VDouble(result));
}
#define MATH2(name) \
static void _##name(Request& r, MethodParams& params) {\
math2(r, params, &name);\
}
MATH2(pow);
inline bool is_salt_body_char(int c) {
return isalnum(c) || c == '.' || c=='/';
}
static size_t calc_prefix_size(const char* salt) {
if(strlen(salt)) {
if(!is_salt_body_char((unsigned char)salt[0])) { // $... {...
const char* cur=salt+1; // skip
while(is_salt_body_char((unsigned char)*cur++)) // ...$ ...}
;
return cur-salt;
} else
return 0;
} else
return 0;
}
static void _crypt(Request& r, MethodParams& params) {
const char* password=params.as_string(0, "password must be string").cstr();
const char* maybe_bodyless_salt=params.as_string(1, "salt must be string").cstr();
size_t prefix_size=calc_prefix_size(maybe_bodyless_salt);
const char* normal_salt;
char normalize_buf[MAX_STRING];
if(prefix_size==strlen(maybe_bodyless_salt)) { // bodyless?
strncpy(normalize_buf, maybe_bodyless_salt, MAX_STRING-MAX_SALT-1);
char *cur=normalize_buf+strlen(normalize_buf);
// sould add up MAX_SALT random chars
static unsigned char itoa64[] = /* 0 ... 63 => ASCII - 64 */
"./0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
for(int i=0; i<MAX_SALT; i++)
*cur++=itoa64[_random(64)];
*cur=0;
normal_salt=normalize_buf;
} else
normal_salt=maybe_bodyless_salt;
/* FreeBSD style MD5 string
*/
if(strncmp(normal_salt, PA_MD5PW_ID, PA_MD5PW_IDLEN) == 0) {
const size_t sample_size=120;
char *sample_buf=new(PointerFreeGC) char[sample_size];
pa_MD5Encode((const unsigned char *)password,
(const unsigned char *)normal_salt, sample_buf, sample_size);
String sample(sample_buf);
r.write_pass_lang(sample);
} else {
#ifdef HAVE_CRYPT
const char* static_sample_buf=crypt(password, normal_salt);
if(!static_sample_buf // nothing generated
|| !static_sample_buf[0] // generated nothing
|| strncmp(static_sample_buf, normal_salt, prefix_size)!=0) // salt prefix not preserved
throw Exception(PARSER_RUNTIME,
0,
"crypt on this platform does not support '%.*s' salt prefix", prefix_size, normal_salt);
r.write_pass_lang(String(pa_strdup(static_sample_buf)));
#else
throw Exception(PARSER_RUNTIME,
0,
"salt must start with '" PA_MD5PW_ID "'");
#endif
}
}
static void _md5(Request& r, MethodParams& params) {
const char *string=params.as_string(0, PARAMETER_MUST_BE_STRING).cstr();
PA_MD5_CTX context;
unsigned char digest[16];
pa_MD5Init(&context);
pa_MD5Update(&context, (const unsigned char*)string, strlen(string));
pa_MD5Final(digest, &context);
r.write_pass_lang(*new String(hex_string(digest, sizeof(digest), false)));
}
//SHA-1:
struct SHA1Context {
unsigned Message_Digest[5], Length_Low, Length_High;
unsigned int Message_Block[64];
int Message_Block_Index, Computed, Corrupted;
};
#define SHA1CircularShift(bits,word) ((((word) << (bits)) & 0xFFFFFFFF)|((word) >> (32-(bits))))
void SHA1ProcessMessageBlock(SHA1Context *);
void SHA1PadMessage(SHA1Context *);
void SHA1Reset(SHA1Context *context) {
context->Length_Low = context->Length_High = context->Message_Block_Index = 0;
context->Message_Digest[0] = 0x67452301;
context->Message_Digest[1] = 0xEFCDAB89;
context->Message_Digest[2] = 0x98BADCFE;
context->Message_Digest[3] = 0x10325476;
context->Message_Digest[4] = 0xC3D2E1F0;
context->Computed = context->Corrupted = 0;
}
int SHA1Result(SHA1Context *context) {
if (context->Corrupted)
return 0;
if (!context->Computed) {
SHA1PadMessage(context);
context->Computed = 1;
}
return 1;
}
void SHA1Input(SHA1Context *context, const unsigned char *message_array, unsigned length) {
if (!length)
return;
if (context->Computed || context->Corrupted) {
context->Corrupted = 1;
return;
}
while(length-- && !context->Corrupted) {
context->Message_Block[context->Message_Block_Index++] = (*message_array & 0xFF);
context->Length_Low += 8;
context->Length_Low &= 0xFFFFFFFF;
if (!context->Length_Low && !(context->Length_High=((1+context->Length_High)&0xFFFFFFFF)))
context->Corrupted = 1; // too long message
if (context->Message_Block_Index == 64)
SHA1ProcessMessageBlock(context);
message_array++;
}
}
void SHA1ProcessMessageBlock(SHA1Context *context) {
const unsigned K[] = {0x5A827999, 0x6ED9EBA1, 0x8F1BBCDC, 0xCA62C1D6 };
int t;
unsigned temp, W[80], buf[5];
unsigned &A=buf[0], &B=buf[1], &C=buf[2], &D=buf[3], &E=buf[4];
for(t = 0; t < 16; t++)
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]);
for(t = 16; t < 80; t++)
W[t] = SHA1CircularShift(1,W[t-3] ^ W[t-8] ^ W[t-14] ^ W[t-16]);
memcpy (buf, context->Message_Digest, sizeof(buf));
for(t = 0; t < 20; t++) {
temp = (SHA1CircularShift(5,A) + ((B & C) | ((~B) & D)) + E + W[t] + K[0]) & 0xFFFFFFFF;
E = D; D = C;
C = SHA1CircularShift(30,B);
B = A; A = temp;
}
for(t = 20; t < 40; t++) {
temp = (SHA1CircularShift(5,A) + (B ^ C ^ D) + E + W[t] + K[1]) & 0xFFFFFFFF;
E = D; D = C;
C = SHA1CircularShift(30,B);
B = A; A = temp;
}
for(t = 40; t < 60; t++) {
temp = (SHA1CircularShift(5,A) + ((B & C) | (B & D) | (C & D)) + E + W[t] + K[2]) & 0xFFFFFFFF;
E = D; D = C;
C = SHA1CircularShift(30,B);
B = A; A = temp;
}
for(t = 60; t < 80; t++) {
temp = (SHA1CircularShift(5,A) + (B ^ C ^ D) + E + W[t] + K[3]) & 0xFFFFFFFF;
E = D; D = C;
C = SHA1CircularShift(30,B);
B = A; A = temp;
}
for (t = 0; t < 5; t++)
context->Message_Digest[t] = (context->Message_Digest[t] + buf[t]) & 0xFFFFFFFF;
context->Message_Block_Index = 0;
}
void SHA1PadMessage(SHA1Context *context) {
context->Message_Block[context->Message_Block_Index++] = 0x80;
if (context->Message_Block_Index > 56) {
//was 55, one shift
while(context->Message_Block_Index < 64)
context->Message_Block[context->Message_Block_Index++] = 0;
SHA1ProcessMessageBlock(context);
while(context->Message_Block_Index < 56)
context->Message_Block[context->Message_Block_Index++] = 0;
} else
while(context->Message_Block_Index < 56)
context->Message_Block[context->Message_Block_Index++] = 0;
context->Message_Block[56] = (context->Length_High >> 24) & 0xFF;
context->Message_Block[57] = (context->Length_High >> 16) & 0xFF;
context->Message_Block[58] = (context->Length_High >> 8) & 0xFF;
context->Message_Block[59] = (context->Length_High) & 0xFF;
context->Message_Block[60] = (context->Length_Low >> 24) & 0xFF;
context->Message_Block[61] = (context->Length_Low >> 16) & 0xFF;
context->Message_Block[62] = (context->Length_Low >> 8) & 0xFF;
context->Message_Block[63] = (context->Length_Low) & 0xFF;
SHA1ProcessMessageBlock(context);
}
static void _sha1(Request& r, MethodParams& params) {
const char *string = params.as_string(0, PARAMETER_MUST_BE_STRING).cstr();
SHA1Context c;
SHA1Reset (&c);
SHA1Input (&c, (const unsigned char*)string, strlen(string));
if(!SHA1Result(&c))
throw Exception (PARSER_RUNTIME, 0, "Can not compute SHA1");
const size_t bufsize=40+1/*zero-teminator*/+1/*for faulty snprintfs*/;
char* cstr=new(PointerFreeGC) char[bufsize];
snprintf(cstr, bufsize,
"%08x%08x%08x%08x%08x",
c.Message_Digest[0],
c.Message_Digest[1],
c.Message_Digest[2],
c.Message_Digest[3],
c.Message_Digest[4]);
r.write_pass_lang(*new String(cstr));
}
static void _uuid(Request& r, MethodParams& /*params*/) {
uuid uuid=get_uuid();
const size_t bufsize=36+1/*zero-teminator*/+1/*for faulty snprintfs*/;
char* cstr=new(PointerFreeGC) char[bufsize];
snprintf(cstr, bufsize,
"%08X-%04X-%04X-%02X%02X-%02X%02X%02X%02X%02X%02X",
uuid.time_low, uuid.time_mid, uuid.time_hi_and_version,
uuid.clock_seq >> 8, uuid.clock_seq & 0xFF,
uuid.node[0], uuid.node[1], uuid.node[2],
uuid.node[3], uuid.node[4], uuid.node[5]);
r.write_pass_lang(*new String(cstr));
}
static void _uid64(Request& r, MethodParams& /*params*/) {
unsigned char id[64/8];
random(&id, sizeof(id));
r.write_pass_lang(*new String(hex_string(id, sizeof(id), true)));
}
static void _crc32(Request& r, MethodParams& params) {
const char *string=params.as_string(0, PARAMETER_MUST_BE_STRING).cstr();
r.write_no_lang(*new VInt(pa_crc32(string, strlen(string))));
}
static void toBase(unsigned int value, unsigned int base, char*& ptr){
static const char* hex="0123456789ABCDEF";
int rest = value % base;
if(value >= base)
toBase( (value-rest)/base, base, ptr);
*ptr++=(char)hex[rest];
}
static void _convert(Request& r, MethodParams& params) {
const char *str=params.as_string(0, PARAMETER_MUST_BE_STRING).cstr();
int base_from=params.as_int(1, "base from must be integer", r);
if(base_from < 2 || base_from > 16)
throw Exception(PARSER_RUNTIME, 0, "base from must be an integer from 2 to 16");
int base_to=params.as_int(2, "base to must be integer", r);
if(base_to < 2 || base_to > 16)
throw Exception(PARSER_RUNTIME, 0, "base to must be an integer from 2 to 16");
while(isspace(*str))
str++;
if(!*str)
return;
bool negative=false;
if(str[0]=='-') {
negative=true;
str++;
} else if(str[0]=='+') {
str++;
}
unsigned int value=pa_atoui(str, base_from);
char result_cstr[sizeof(unsigned int)*8+1/*minus for negative number*/+1/*terminator*/];
char* ptr=result_cstr;
if(negative)
*ptr++='-';
toBase(value, base_to, ptr);
*ptr=0;
r.write_pass_lang(*new String(pa_strdup(result_cstr)));
}
// constructor
MMath::MMath(): Methoded("math") {
// ^FUNC(expr)
#define ADDX(name, X) \
add_native_method(#name, Method::CT_STATIC, _##name, X, X)
#define ADD0(name) ADDX(name, 0)
#define ADD1(name) ADDX(name, 1)
#define ADD2(name) ADDX(name, 2)
ADD1(round); ADD1(floor); ADD1(ceiling);
ADD1(trunc); ADD1(frac);
ADD1(abs); ADD1(sign);
ADD1(exp);
ADD1(log); ADD1(log10);
ADD1(sin); ADD1(asin);
ADD1(cos); ADD1(acos);
ADD1(tan); ADD1(atan);
ADD1(degrees); ADD1(radians);
ADD1(sqrt);
ADD1(random);
// ^math:pow(x;y)
ADD2(pow);
// ^math:crypt[password;salt]
ADD2(crypt);
// ^math:md5[string]
ADD1(md5);
// ^math:sha1[string]
ADD1(sha1);
// ^math:crc32[string]
ADD1(crc32);
// ^math:uuid[]
ADD0(uuid);
// ^math:uid64[]
ADD0(uid64);
// ^math:convert[number](base-from;base-to)
add_native_method("convert", Method::CT_STATIC, _convert, 3, 3);
}
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