From fffd213c8cba2135afda493d797c41c10354770e Mon Sep 17 00:00:00 2001
From: Othmar Gsenger <otti@anytun.org>
Date: Sat, 12 Apr 2008 11:38:42 +0000
Subject: big svn cleanup

---
 src/mpi.cpp | 201 ++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 201 insertions(+)
 create mode 100644 src/mpi.cpp

(limited to 'src/mpi.cpp')

diff --git a/src/mpi.cpp b/src/mpi.cpp
new file mode 100644
index 0000000..6ca2d83
--- /dev/null
+++ b/src/mpi.cpp
@@ -0,0 +1,201 @@
+/*
+ *  anytun
+ *
+ *  The secure anycast tunneling protocol (satp) defines a protocol used
+ *  for communication between any combination of unicast and anycast
+ *  tunnel endpoints.  It has less protocol overhead than IPSec in Tunnel
+ *  mode and allows tunneling of every ETHER TYPE protocol (e.g.
+ *  ethernet, ip, arp ...). satp directly includes cryptography and
+ *  message authentication based on the methodes used by SRTP.  It is
+ *  intended to deliver a generic, scaleable and secure solution for
+ *  tunneling and relaying of packets of any protocol.
+ *
+ *
+ *  Copyright (C) 2007 anytun.org <satp@wirdorange.org>
+ *
+ *  This program is free software; you can redistribute it and/or modify
+ *  it under the terms of the GNU General Public License version 2
+ *  as published by the Free Software Foundation.
+ *
+ *  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 more details.
+ *
+ *  You should have received a copy of the GNU General Public License
+ *  along with this program (see the file COPYING included with this
+ *  distribution); if not, write to the Free Software Foundation, Inc.,
+ *  59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
+ */
+
+#include "mpi.h"
+
+#include "datatypes.h"
+#include "cipher.h"
+
+#include <stdexcept>
+#include <gcrypt.h>
+
+#include <iostream>
+
+Mpi::Mpi() : val_(NULL)
+{
+  val_ = gcry_mpi_set_ui(NULL, 0);
+  if(!val_)
+    throw std::bad_alloc();
+}
+
+Mpi::Mpi(u_int8_t length) : val_(NULL)
+{
+  val_ = gcry_mpi_new(length);
+  if(!val_)
+    throw std::bad_alloc();
+}
+
+Mpi::Mpi(const Mpi &src) : val_(NULL)
+{
+  val_ = gcry_mpi_copy(src.val_);
+  if(!val_)
+    throw std::bad_alloc();
+}
+
+Mpi::Mpi(const u_int8_t* src, u_int32_t len) : val_(NULL)
+{
+  u_int8_t* src_cpy = new u_int8_t[len+1];
+  if(!src_cpy)
+    throw std::bad_alloc();
+
+  u_int8_t* buf = src_cpy;
+  u_int32_t buf_len = len;
+  if(src[0] & 0x80) // this would be a negative number, scan can't handle this :(
+  {
+    src_cpy[0] = 0;
+    buf++;
+    buf_len++;
+  }
+  std::memcpy(buf, src, len);
+
+  gcry_mpi_scan( &val_, GCRYMPI_FMT_STD, src_cpy, buf_len, NULL );
+  delete[] src_cpy;
+  if(!val_)
+    throw std::bad_alloc();
+}
+
+Mpi::~Mpi()
+{
+  gcry_mpi_release( val_ ); 
+}
+
+
+void Mpi::operator=(const Mpi &src)
+{
+  gcry_mpi_release( val_ ); 
+  val_ = gcry_mpi_copy(src.val_);
+  if(!val_)
+    throw std::bad_alloc();
+}
+
+void Mpi::operator=(const u_int32_t src)
+{
+  gcry_mpi_release( val_ ); 
+  val_ = gcry_mpi_set_ui(NULL, src);
+  if(!val_)
+    throw std::bad_alloc();
+}
+
+Mpi Mpi::operator+(const Mpi &b) const
+{
+  Mpi res;
+  gcry_mpi_add(res.val_, val_, b.val_);
+  return res;
+}
+
+Mpi Mpi::operator+(const u_int32_t &b) const
+{
+  Mpi res;
+  gcry_mpi_add_ui(res.val_, val_, b);
+  return res;
+}
+
+Mpi Mpi::operator*(const u_int32_t n) const
+{
+  Mpi res;
+  gcry_mpi_mul_ui(res.val_, val_, n);
+  return res;
+}
+
+Mpi Mpi::operator/(const Mpi &b) const
+{
+  Mpi res;
+  gcry_mpi_div(res.val_, NULL, val_, b.val_, 0);
+  return res;
+}
+
+//TODO: this is outstandingly ugly!!!!!!!!
+Mpi Mpi::operator^(const Mpi &b) const
+{
+  u_int32_t a_len = gcry_mpi_get_nbits(val_);
+  u_int32_t b_len = gcry_mpi_get_nbits(b.val_);
+
+  Mpi res = (a_len >= b_len) ? Mpi(*this) : Mpi(b);
+
+  for(u_int32_t i=0; i<a_len && i<b_len; i++) {
+    if(gcry_mpi_test_bit(val_, i) ^ gcry_mpi_test_bit(b.val_, i))
+      gcry_mpi_set_bit(res.val_, i);
+    else
+      gcry_mpi_clear_bit(res.val_, i);
+  }
+  return res;
+}
+
+Mpi Mpi::mul2exp(u_int32_t e) const
+{
+  Mpi res;
+  gcry_mpi_mul_2exp( res.val_, val_, e );
+  return res;
+}
+
+//TODO: problem, seems as gcry_mpi_(a)print doesn't work for mpi values of '0'
+u_int8_t* Mpi::getNewBuf(size_t* written) const
+{
+  u_int8_t* res_cpy;
+  gcry_mpi_aprint( GCRYMPI_FMT_STD, &res_cpy, written, val_ );
+  if(!res_cpy)
+    throw std::bad_alloc();
+    
+  u_int8_t* buf = res_cpy;
+  if(*written > 1 && ! (res_cpy[0])) // positive number with highestBit set
+  {
+    buf++;
+    (*written)--;
+  }
+
+  u_int8_t* res = new u_int8_t[*written];
+  if(!res)
+    throw std::bad_alloc();
+
+  std::memcpy(res, buf, *written);
+
+  gcry_free(res_cpy);
+
+  return res;
+}
+
+//TODO: why does this not work ?????
+std::string Mpi::getHexDump() const
+{
+//   u_int8_t *buf;
+//   u_int32_t len;
+//   gcry_mpi_aprint( GCRYMPI_FMT_HEX, &buf, &len, val_ );
+//   std::string res(buf, len);
+//   delete[] buf;
+
+  gcry_mpi_dump( val_ );
+  std::string res("\n");
+  return res;
+}
+
+u_int32_t Mpi::getLength() const
+{
+  return gcry_mpi_get_nbits( val_ );
+}
-- 
cgit v1.2.3