~ubuntu-branches/ubuntu/oneiric/libtorrent/oneiric

« back to all changes in this revision

Viewing changes to src/utils/diffie_hellman.cc

  • Committer: Bazaar Package Importer
  • Author(s): Rogério Brito
  • Date: 2011-03-20 01:06:18 UTC
  • mfrom: (1.1.13 upstream) (4.1.9 sid)
  • Revision ID: james.westby@ubuntu.com-20110320010618-g3wyylccqzqko73c
Tags: 0.12.7-5
* Use Steinar's "real" patch for IPv6. Addresses #490277, #618275,
  and Closes: #617791.
* Adapt libtorrent-0.12.6-ipv6-07.patch. It FTBFS otherwise.
* Add proper attibution to the IPv6 patch.

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
// libTorrent - BitTorrent library
 
2
// Copyright (C) 2005-2007, Jari Sundell
 
3
//
 
4
// This program is free software; you can redistribute it and/or modify
 
5
// it under the terms of the GNU General Public License as published by
 
6
// the Free Software Foundation; either version 2 of the License, or
 
7
// (at your option) any later version.
 
8
// 
 
9
// This program is distributed in the hope that it will be useful,
 
10
// but WITHOUT ANY WARRANTY; without even the implied warranty of
 
11
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
12
// GNU General Public License for more details.
 
13
// 
 
14
// You should have received a copy of the GNU General Public License
 
15
// along with this program; if not, write to the Free Software
 
16
// Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 
17
//
 
18
// In addition, as a special exception, the copyright holders give
 
19
// permission to link the code of portions of this program with the
 
20
// OpenSSL library under certain conditions as described in each
 
21
// individual source file, and distribute linked combinations
 
22
// including the two.
 
23
//
 
24
// You must obey the GNU General Public License in all respects for
 
25
// all of the code used other than OpenSSL.  If you modify file(s)
 
26
// with this exception, you may extend this exception to your version
 
27
// of the file(s), but you are not obligated to do so.  If you do not
 
28
// wish to do so, delete this exception statement from your version.
 
29
// If you delete this exception statement from all source files in the
 
30
// program, then also delete it here.
 
31
//
 
32
// Contact:  Jari Sundell <jaris@ifi.uio.no>
 
33
//
 
34
//           Skomakerveien 33
 
35
//           3185 Skoppum, NORWAY
 
36
 
 
37
#include "config.h"
 
38
 
 
39
#include <cstring>
 
40
#include <string>
 
41
 
 
42
#ifdef USE_OPENSSL
 
43
#include <openssl/bn.h>
 
44
#endif
 
45
 
 
46
#include "diffie_hellman.h"
 
47
#include "torrent/exceptions.h"
 
48
 
 
49
namespace torrent {
 
50
 
 
51
DiffieHellman::DiffieHellman(const unsigned char *prime, int primeLength,
 
52
                             const unsigned char *generator, int generatorLength) :
 
53
  m_secret(NULL) {
 
54
 
 
55
#ifdef USE_OPENSSL
 
56
  m_dh = DH_new();
 
57
  m_dh->p = BN_bin2bn(prime, primeLength, NULL);
 
58
  m_dh->g = BN_bin2bn(generator, generatorLength, NULL);
 
59
 
 
60
  DH_generate_key(m_dh);
 
61
#else
 
62
  throw internal_error("Compiled without encryption support.");
 
63
#endif
 
64
};
 
65
 
 
66
DiffieHellman::~DiffieHellman() {
 
67
  delete [] m_secret;
 
68
#ifdef USE_OPENSSL
 
69
  DH_free(m_dh);
 
70
#endif
 
71
};
 
72
 
 
73
void
 
74
DiffieHellman::compute_secret(const unsigned char *pubkey, unsigned int length) {
 
75
#ifdef USE_OPENSSL
 
76
  BIGNUM* k = BN_bin2bn(pubkey, length, NULL);
 
77
 
 
78
  delete [] m_secret;
 
79
  m_secret = new char[DH_size(m_dh)];
 
80
 
 
81
  m_size = DH_compute_key((unsigned char*)m_secret, k, m_dh);
 
82
  
 
83
  BN_free(k);
 
84
#endif
 
85
};
 
86
 
 
87
void
 
88
DiffieHellman::store_pub_key(unsigned char* dest, unsigned int length) {
 
89
#ifdef USE_OPENSSL
 
90
  std::memset(dest, 0, length);
 
91
 
 
92
  if ((int)length >= BN_num_bytes(m_dh->pub_key))
 
93
    BN_bn2bin(m_dh->pub_key, dest + length - BN_num_bytes(m_dh->pub_key));
 
94
#endif
 
95
}
 
96
 
 
97
};