transmission/libtransmission/crypto.c

200 lines
4.8 KiB
C
Raw Normal View History

/*
* This file Copyright (C) 2007-2014 Mnemosyne LLC
*
* It may be used under the GNU GPL versions 2 or 3
* or any future license endorsed by Mnemosyne LLC.
*
*/
2007-11-09 20:07:52 +00:00
#include <assert.h>
#include <string.h> /* memcpy (), memmove (), memset () */
2007-10-25 13:59:46 +00:00
#include "transmission.h"
#include "crypto.h"
#include "crypto-utils.h"
#include "utils.h"
/**
***
**/
#define PRIME_LEN 96
#define DH_PRIVKEY_LEN 20
static const uint8_t dh_P[PRIME_LEN] =
{
0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xFF, 0xC9, 0x0F, 0xDA, 0xA2,
0x21, 0x68, 0xC2, 0x34, 0xC4, 0xC6, 0x62, 0x8B, 0x80, 0xDC, 0x1C, 0xD1,
0x29, 0x02, 0x4E, 0x08, 0x8A, 0x67, 0xCC, 0x74, 0x02, 0x0B, 0xBE, 0xA6,
0x3B, 0x13, 0x9B, 0x22, 0x51, 0x4A, 0x08, 0x79, 0x8E, 0x34, 0x04, 0xDD,
0xEF, 0x95, 0x19, 0xB3, 0xCD, 0x3A, 0x43, 0x1B, 0x30, 0x2B, 0x0A, 0x6D,
0xF2, 0x5F, 0x14, 0x37, 0x4F, 0xE1, 0x35, 0x6D, 0x6D, 0x51, 0xC2, 0x45,
0xE4, 0x85, 0xB5, 0x76, 0x62, 0x5E, 0x7E, 0xC6, 0xF4, 0x4C, 0x42, 0xE9,
0xA6, 0x3A, 0x36, 0x21, 0x00, 0x00, 0x00, 0x00, 0x00, 0x09, 0x05, 0x63,
};
static const uint8_t dh_G[] = { 2 };
/**
***
**/
static void ensureKeyExists(tr_crypto* crypto)
{
if (crypto->dh == NULL)
{
size_t public_key_length;
crypto->dh = tr_dh_new(dh_P, sizeof(dh_P), dh_G, sizeof(dh_G));
tr_dh_make_key(crypto->dh, DH_PRIVKEY_LEN, crypto->myPublicKey, &public_key_length);
assert(public_key_length == KEY_LEN);
}
}
void tr_cryptoConstruct(tr_crypto* crypto, const uint8_t* torrentHash, bool isIncoming)
{
memset(crypto, 0, sizeof(tr_crypto));
crypto->isIncoming = isIncoming;
tr_cryptoSetTorrentHash(crypto, torrentHash);
}
void tr_cryptoDestruct(tr_crypto* crypto)
{
tr_dh_secret_free(crypto->mySecret);
tr_dh_free(crypto->dh);
tr_rc4_free(crypto->enc_key);
tr_rc4_free(crypto->dec_key);
}
/**
***
**/
bool tr_cryptoComputeSecret(tr_crypto* crypto, const uint8_t* peerPublicKey)
{
ensureKeyExists(crypto);
crypto->mySecret = tr_dh_agree(crypto->dh, peerPublicKey, KEY_LEN);
return crypto->mySecret != NULL;
}
const uint8_t* tr_cryptoGetMyPublicKey(const tr_crypto* crypto, int* setme_len)
{
ensureKeyExists((tr_crypto*)crypto);
*setme_len = KEY_LEN;
return crypto->myPublicKey;
}
/**
***
**/
static void initRC4(tr_crypto* crypto, tr_rc4_ctx_t* setme, const char* key)
{
uint8_t buf[SHA_DIGEST_LENGTH];
assert(crypto->torrentHashIsSet);
if (*setme == NULL)
{
*setme = tr_rc4_new();
}
if (tr_cryptoSecretKeySha1(crypto, key, 4, crypto->torrentHash, SHA_DIGEST_LENGTH, buf))
{
tr_rc4_set_key(*setme, buf, SHA_DIGEST_LENGTH);
}
}
void tr_cryptoDecryptInit(tr_crypto* crypto)
{
uint8_t discard[1024];
const char* txt = crypto->isIncoming ? "keyA" : "keyB";
initRC4(crypto, &crypto->dec_key, txt);
tr_rc4_process(crypto->dec_key, discard, discard, sizeof(discard));
}
void tr_cryptoDecrypt(tr_crypto* crypto, size_t buf_len, const void* buf_in, void* buf_out)
{
/* FIXME: someone calls this function with uninitialized key */
if (crypto->dec_key == NULL)
{
if (buf_in != buf_out)
{
memmove(buf_out, buf_in, buf_len);
}
return;
}
tr_rc4_process(crypto->dec_key, buf_in, buf_out, buf_len);
}
void tr_cryptoEncryptInit(tr_crypto* crypto)
{
uint8_t discard[1024];
const char* txt = crypto->isIncoming ? "keyB" : "keyA";
initRC4(crypto, &crypto->enc_key, txt);
tr_rc4_process(crypto->enc_key, discard, discard, sizeof(discard));
}
void tr_cryptoEncrypt(tr_crypto* crypto, size_t buf_len, const void* buf_in, void* buf_out)
{
/* FIXME: someone calls this function with uninitialized key */
if (crypto->enc_key == NULL)
{
if (buf_in != buf_out)
{
memmove(buf_out, buf_in, buf_len);
}
return;
}
tr_rc4_process(crypto->enc_key, buf_in, buf_out, buf_len);
}
bool tr_cryptoSecretKeySha1(const tr_crypto* crypto, const void* prepend_data, size_t prepend_data_size,
const void* append_data, size_t append_data_size, uint8_t* hash)
{
assert(crypto != NULL);
assert(crypto->mySecret != NULL);
return tr_dh_secret_derive(crypto->mySecret, prepend_data, prepend_data_size, append_data, append_data_size, hash);
}
/**
***
**/
void tr_cryptoSetTorrentHash(tr_crypto* crypto, const uint8_t* hash)
{
crypto->torrentHashIsSet = hash != NULL;
if (hash)
{
memcpy(crypto->torrentHash, hash, SHA_DIGEST_LENGTH);
}
else
{
memset(crypto->torrentHash, 0, SHA_DIGEST_LENGTH);
}
}
const uint8_t* tr_cryptoGetTorrentHash(const tr_crypto* crypto)
{
assert(crypto);
return crypto->torrentHashIsSet ? crypto->torrentHash : NULL;
}
bool tr_cryptoHasTorrentHash(const tr_crypto* crypto)
{
assert(crypto);
return crypto->torrentHashIsSet;
}