1 // libTorrent - BitTorrent library 2 // Copyright (C) 2005-2011, 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 #ifndef LIBTORRENT_PROTOCOL_ENCRYPTION_H 38 #define LIBTORRENT_PROTOCOL_ENCRYPTION_H 39 40 #include "utils/rc4.h" 41 42 namespace torrent { 43 44 class EncryptionInfo { 45 public: EncryptionInfo()46 EncryptionInfo() : m_encrypted(false), m_obfuscated(false), m_decryptValid(false) {}; 47 encrypt(const void * indata,void * outdata,unsigned int length)48 void encrypt(const void *indata, void *outdata, unsigned int length) { m_encrypt.crypt(indata, outdata, length); } encrypt(void * data,unsigned int length)49 void encrypt(void *data, unsigned int length) { m_encrypt.crypt(data, length); } decrypt(const void * indata,void * outdata,unsigned int length)50 void decrypt(const void *indata, void *outdata, unsigned int length) { m_decrypt.crypt(indata, outdata, length); } decrypt(void * data,unsigned int length)51 void decrypt(void *data, unsigned int length) { m_decrypt.crypt(data, length); } 52 is_encrypted()53 bool is_encrypted() const { return m_encrypted; } is_obfuscated()54 bool is_obfuscated() const { return m_obfuscated; } decrypt_valid()55 bool decrypt_valid() const { return m_decryptValid; } 56 set_obfuscated()57 void set_obfuscated() { m_obfuscated = true; m_encrypted = m_decryptValid = false; } set_encrypt(const RC4 & encrypt)58 void set_encrypt(const RC4& encrypt) { m_encrypt = encrypt; m_encrypted = m_obfuscated = true; } set_decrypt(const RC4 & decrypt)59 void set_decrypt(const RC4& decrypt) { m_decrypt = decrypt; m_decryptValid = true; } 60 61 private: 62 bool m_encrypted; 63 bool m_obfuscated; 64 bool m_decryptValid; 65 66 RC4 m_encrypt; 67 RC4 m_decrypt; 68 }; 69 70 } 71 72 #endif 73