Logo Search packages:      
Sourcecode: parser version File versions  Download package

math.C

Go to the documentation of this file.
/** @file
      Parser: @b math parser class.

      Copyright(c) 2001-2005 ArtLebedev Group(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.
*/

static const char * const IDENT_MATH_C="$Date: 2008-06-16 12:38:41 $";

#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_threads.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

// 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);

#ifdef WIN32
class Random_provider {
      HCRYPTPROV fhProv;
      
      void acquire() {
            SYNCHRONIZED;

            if(fhProv)
                  return;

            if(!CryptAcquireContext(&fhProv, NULL, NULL, PROV_RSA_FULL, CRYPT_VERIFYCONTEXT))
                  throw Exception(0,
                        0,
                        "CryptAcquireContext failed");
      }
      void release() {
            if(fhProv)
                  CryptReleaseContext(fhProv, 0);
      }
      
public:
      Random_provider(): fhProv(0) {}
      ~Random_provider() { release(); }
      void generate(void *buffer, size_t size) {
            acquire();

            if(!CryptGenRandom(fhProv, size, (BYTE*)buffer))
                  throw Exception(0,
                        0,
                        "CryptGenRandom failed");
      }
}
      random_provider;

#else

/// from gen_uuid.c
00090 static int get_random_fd(void)
{
        struct timeval  tv;
        static int      fd = -2;
        int             i;

        if (fd == -2) {
                gettimeofday(&tv, 0);
                fd = open("/dev/urandom", O_RDONLY);
                if (fd == -1)
                        fd = open("/dev/random", O_RDONLY | O_NONBLOCK);
                srand((getpid() << 16) ^ getuid() ^ tv.tv_sec ^ tv.tv_usec);
        }
        /* Crank the random number generator a few times */
        gettimeofday(&tv, 0);
        for (i = (tv.tv_sec ^ tv.tv_usec) & 0x1F; i > 0; i--)
                rand();
        return fd;
}


/*
 * Generate a series of random bytes.  Use /dev/urandom if possible,
 * and if not, use srandom/random.
 */
static void get_random_bytes(void *buf, int nbytes)
{
        int i, fd = get_random_fd();
        int lose_counter = 0;
        char *cp = (char *) buf;

        if (fd >= 0) {
                while (nbytes > 0) {
                        i = read(fd, cp, nbytes);
                        if (i <= 0) {
                                if (lose_counter++ > 16)
                                        break;
                                continue;
                        }
                        nbytes -= i;
                        cp += i;
                        lose_counter = 0;
                }
        }

        /* XXX put something better here if no /dev/random! */
        for (i = 0; i < nbytes; i++)
                *cp++ = rand() & 0xFF;
        return;
}


#endif


// helpers

static void random(void *buffer, size_t size) {
#ifdef WIN32
      random_provider.generate(buffer, size);
#else
      get_random_bytes(buffer, size);
#endif
}


// methods

#define MAX_UINT 0xFFFFFFFFu

static inline int _random(uint top) {
      uint raw;
      random(&raw, sizeof(raw));
      return int(double(raw) / MAX_UINT * top );
}

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");
    
    char digest[128];
    sprintf(digest, "%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]);
    
    char *ret = new(PointerFreeGC) char[strlen(digest)+1];
    strcpy(ret, digest);
    r.write_pass_lang(*new String(ret, 0, false));
    }


/// to hell with extra bytes on 64bit platforms
00439 struct uuid {
        unsigned int   time_low;
        unsigned short   time_mid;
        unsigned short   time_hi_and_version;
        unsigned short   clock_seq;
        unsigned char    node[6];
};
static void _uuid(Request& r, MethodParams& /*params*/) {

      // random
      struct uuid uuid;
      random(&uuid, sizeof(uuid));

      // http://www.opengroup.org/onlinepubs/9629399/apdxa.htm#tagtcjh_35
      // ~
      // version = DCE Security version, with embedded POSIX UIDs.  
      // variant = DCE
      //
      // DCE=Distributed Computing Environment
      // http://www.opengroup.org/dce/
      //
      // they say this influences comparison&such,
      // but could not figure out how, hence structure layout specified strictly
      // anyhow, uuidgen on Win32 yield those values
      // 
      // xxxxxxxx-xxxx-4xxx-{8,9,A,B}xxx-xxxxxxxxxxxx
      uuid.clock_seq = (uuid.clock_seq & 0x3FFF) | 0x8000;
        uuid.time_hi_and_version = (uuid.time_hi_and_version & 0x0FFF) | 0x4000;
 
      // format 
      const int uuid_cstr_bufsize=36+1/*for zero-teminator*/+1/*for faulty snprintfs*/;
      char *uuid_cstr=new(PointerFreeGC) char[uuid_cstr_bufsize];
        snprintf(uuid_cstr, uuid_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(uuid_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))));
}

// 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);
}

Generated by  Doxygen 1.6.0   Back to index