<?php

/*

 * Enano - an open-source CMS capable of wiki functions, Drupal-like sidebar blocks, and everything in between

 * Copyright (C) 2006-2009 Dan Fuhry

 * diffiehellman.php - Diffie Hellman key exchange and supporting functions

 *

 * This program is Free Software; you can redistribute and/or modify it under the terms of the GNU General Public License

 * as published by the Free Software Foundation; either version 2 of the License, or (at your option) any later version.

 *

 * This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied

 * warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for details.

 */

/**

 * EnanoMath GMP backend

 */

class EnanoMath_GMP

{

  var $api = 'GMP';

  /**

   * Initializes a number to a GMP integer.

   * @param string String representation of the number

   * @param int Base the number is currently in, defaults to 10

   * @return resource

   */

  function init($int, $base = 10)

  {

    return ( is_resource($int) ) ? $int : gmp_init($int, $base);

  }

  /**

   * Converts a number from a GMP integer to a string

   * @param resource

   * @param int Base, default is 10

   * @return string

   */

  function str($int, $base = 10)

  {

    return ( is_string($int) ) ? $int : gmp_strval($int, $base);

  }

  /**

   * Converts a number between bases.

   * @param resource BigInt to convert

   * @param int Base to convert from

   * @param int Base to convert to

   */

  function basecon($int, $from, $to)

  {

    return $this->init(gmp_strval(gmp_init($this->str($int), $from), $to));

  }

  /**

   * Generates a random integer.

   * @param int Length

   * @return resource

   */

  function random($len)

  {

    return gmp_random($len / 8);

  }

  /**

   * Powmod operation (calculates (a ^ b) mod m)

   * @param resource a

   * @param resource b

   * @param resource m

   * @return resource

   */

  function powmod($a, $b, $m)

  {

    $a = $this->init($a);

    $b = $this->init($b);

    $m = $this->init($m);

    return ( function_exists('gmp_powm') ) ? gmp_powm($a, $b, $m) : gmp_mod(gmp_pow($a, $b), $m);

  }

}

/**

 * EnanoMath big_int backend

 */

class EnanoMath_BigInt

{

  var $api = 'big_int';

  /**

   * Initializes a number to a BigInt integer.

   * @param string String representation of the number

   * @param int Base the number is in, defaults to 10

   * @return resource

   */

  function init($int, $base = 10)

  {

    return (is_resource($int)) ? $int : bi_from_str($int, $base);

  }

  /**

   * Converts a number from a BigInt integer to a string

   * @param resource

   * @param int Base, default is 10

   * @return string

   */

  function str($int, $base = 10)

  {

    return ( is_string($int) ) ? $int : bi_to_str($int, $base);

  }

  /**

   * Generates a random integer

   * @param int Length (bits)

   * @return resource

   */

  function random($len)

  {

    return bi_rand($len);

  }

  /**

   * Converts a number between bases.

   * @param resource BigInt to convert

   * @param int Base to convert from

   * @param int Base to convert to

   */

  function basecon($int, $from, $to)

  {

    return bi_base_convert($this->str($int, $from), $from, $to);

  }

  /**

   * Powmod operation (calculates (a ^ b) mod m)

   * @param resource a

   * @param resource b

   * @param resource m

   * @return resource

   */

  function powmod($a, $b, $m)

  {

    $a = $this->init($a);

    $b = $this->init($b);

    $m = $this->init($m);

    return bi_powmod($a, $b, $m);

  }

}

/**

 * EnanoMath BCMath backend

 */

class EnanoMath_BCMath

{

  var $api = 'BCMath';

  /**

   * Initializes a number to a BCMath integer.

   * @param string String representation of the number

   * @param int Base the number is in, defaults to 10

   * @return resource

   */

  function init($int, $base = 10)

  {

    return $this->basecon($int, $base, 10);

  }

  /**

   * Converts a number from a BCMath integer to a string

   * @param resource

   * @param int Base, default is 10

   * @return string

   */

  function str($res)

  {

    return ( is_string($res) ) ? $res : strval($this->basecon($res, 10, $base));

  }

  /**

   * Generates a random integer

   * @param int Length in bits

   * @return resource

   */

  function random($len)

  {

    $len = 4 * $len;

    $chars = '0123456789abcdef';

    $ret = '';

    for ( $i = 0; $i < $len; $i++ )

    {

      $chid = mt_rand ( 0, strlen($chars) - 1 );

      $chr = $chars{$chid};

      $ret .= $chr;

    }

    return $this->basecon($this->init($ret), 16, 10);

  }

  /**

   * Converts a number between bases.

   * @param resource BigInt to convert

   * @param int Base to convert from

   * @param int Base to convert to

   */

  function basecon($int, $from, $to)

  {

    if ( $from != 10 )

      $int = $this->_bcmath_base2dec($int, $from);

    if ( $to != 10 )

      $int = $this->_bcmath_dec2base($int, $to);

    return $int;

  }

  /**

   * Powmod operation (calculates (a ^ b) mod m)

   * @param resource a

   * @param resource b

   * @param resource m

   * @return resource

   */

  function powmod($a, $b, $m)

  {

    $a = $this->init($a);

    $b = $this->init($b);

    $m = $this->init($m);

    return ( function_exists('bcpowmod') ) ? bcpowmod($a, $b, $m) : bcmod( bcpow($a, $b), $m );

  }

  // from us.php.net/bc:

  // convert a decimal value to any other base value

  function _bcmath_dec2base($dec,$base,$digits=FALSE) {

      if($base<2 or $base>256) die("Invalid Base: ".$base);

      bcscale(0);

      $value="";

      if(!$digits) $digits=$this->_bcmath_digits($base);

      while($dec>$base-1) {

          $rest=bcmod($dec,$base);

          $dec=bcdiv($dec,$base);

          $value=$digits[$rest].$value;

      }

      $value=$digits[intval($dec)].$value;

      return (string) $value;

  }

  // convert another base value to its decimal value

  function _bcmath_base2dec($value,$base,$digits=FALSE) {

      if($base<2 or $base>256) die("Invalid Base: ".$base);

      bcscale(0);

      if($base<37) $value=strtolower($value);

      if(!$digits) $digits=$this->_bcmath_digits($base);

      $size=strlen($value);

      $dec="0";

      for($loop=0;$loop<$size;$loop++) {

          $element=strpos($digits,$value[$loop]);

          $power=bcpow($base,$size-$loop-1);

          $dec=bcadd($dec,bcmul($element,$power));

      }

      return (string) $dec;

  }

  function _bcmath_digits($base) {

      if($base>64) {

          $digits="";

          for($loop=0;$loop<256;$loop++) {

              $digits.=chr($loop);

          }

      } else {

          $digits ="0123456789abcdefghijklmnopqrstuvwxyz";

          $digits.="ABCDEFGHIJKLMNOPQRSTUVWXYZ-_";

      }

      $digits=substr($digits,0,$base);

      return (string) $digits;

  }

}

/**

 * Creates a new math object based on what libraries are available.

 * @return object

 */

function enanomath_create()

{

  if ( function_exists('gmp_init') )

    return new EnanoMath_GMP();

  else if ( function_exists('bi_from_str') )

    return new EnanoMath_BigInt();

  else if ( function_exists('bcadd') )

    return new EnanoMath_BCMath();

  else

    throw new Exception('dh_err_not_supported');

}

?>