mirror of https://github.com/PurpleI2P/i2pd.git
orignal
4 years ago
5 changed files with 123 additions and 181 deletions
@ -1,88 +0,0 @@
@@ -1,88 +0,0 @@
|
||||
/*
|
||||
* Copyright (c) 2013-2020, The PurpleI2P Project |
||||
* |
||||
* This file is part of Purple i2pd project and licensed under BSD3 |
||||
* |
||||
* See full license text in LICENSE file at top of project tree |
||||
*/ |
||||
|
||||
#ifndef CRYPTO_WORKER_H_ |
||||
#define CRYPTO_WORKER_H_ |
||||
|
||||
#include <condition_variable> |
||||
#include <mutex> |
||||
#include <deque> |
||||
#include <thread> |
||||
#include <vector> |
||||
#include <memory> |
||||
|
||||
namespace i2p |
||||
{ |
||||
namespace worker |
||||
{ |
||||
template<typename Caller> |
||||
struct ThreadPool |
||||
{ |
||||
typedef std::function<void(void)> ResultFunc; |
||||
typedef std::function<ResultFunc(void)> WorkFunc; |
||||
typedef std::pair<std::shared_ptr<Caller>, WorkFunc> Job; |
||||
typedef std::mutex mtx_t; |
||||
typedef std::unique_lock<mtx_t> lock_t; |
||||
typedef std::condition_variable cond_t; |
||||
ThreadPool(int workers) |
||||
{ |
||||
stop = false; |
||||
if(workers > 0) |
||||
{ |
||||
while(workers--) |
||||
{ |
||||
threads.emplace_back([this] { |
||||
for (;;) |
||||
{ |
||||
Job job; |
||||
{ |
||||
lock_t lock(this->queue_mutex); |
||||
this->condition.wait( |
||||
lock, [this] { return this->stop || !this->jobs.empty(); }); |
||||
if (this->stop && this->jobs.empty()) return; |
||||
job = std::move(this->jobs.front()); |
||||
this->jobs.pop_front(); |
||||
} |
||||
ResultFunc result = job.second(); |
||||
job.first->GetService().post(result); |
||||
} |
||||
}); |
||||
} |
||||
} |
||||
}; |
||||
|
||||
void Offer(const Job & job) |
||||
{ |
||||
{ |
||||
lock_t lock(queue_mutex); |
||||
if (stop) return; |
||||
jobs.emplace_back(job); |
||||
} |
||||
condition.notify_one(); |
||||
} |
||||
|
||||
~ThreadPool() |
||||
{ |
||||
{ |
||||
lock_t lock(queue_mutex); |
||||
stop = true; |
||||
} |
||||
condition.notify_all(); |
||||
for(auto &t: threads) t.join(); |
||||
} |
||||
|
||||
std::vector<std::thread> threads; |
||||
std::deque<Job> jobs; |
||||
mtx_t queue_mutex; |
||||
cond_t condition; |
||||
bool stop; |
||||
}; |
||||
} |
||||
} |
||||
|
||||
#endif |
||||
@ -0,0 +1,114 @@
@@ -0,0 +1,114 @@
|
||||
/*
|
||||
* Copyright (c) 2013-2020, The PurpleI2P Project |
||||
* |
||||
* This file is part of Purple i2pd project and licensed under BSD3 |
||||
* |
||||
* See full license text in LICENSE file at top of project tree |
||||
* |
||||
*/ |
||||
|
||||
#include <memory> |
||||
#include <openssl/rand.h> |
||||
#include "Transports.h" |
||||
#include "Timestamp.h" |
||||
#include "I2PEndian.h" |
||||
#include "I2NPProtocol.h" |
||||
#include "TunnelConfig.h" |
||||
|
||||
namespace i2p |
||||
{ |
||||
namespace tunnel |
||||
{ |
||||
TunnelHopConfig::TunnelHopConfig (std::shared_ptr<const i2p::data::IdentityEx> r) |
||||
{ |
||||
RAND_bytes (layerKey, 32); |
||||
RAND_bytes (ivKey, 32); |
||||
RAND_bytes (replyKey, 32); |
||||
RAND_bytes (replyIV, 16); |
||||
RAND_bytes ((uint8_t *)&tunnelID, 4); |
||||
if (!tunnelID) tunnelID = 1; // tunnelID can't be zero
|
||||
isGateway = true; |
||||
isEndpoint = true; |
||||
ident = r; |
||||
//nextRouter = nullptr;
|
||||
nextTunnelID = 0; |
||||
|
||||
next = nullptr; |
||||
prev = nullptr; |
||||
} |
||||
|
||||
void TunnelHopConfig::SetNextIdent (const i2p::data::IdentHash& ident) |
||||
{ |
||||
nextIdent = ident; |
||||
isEndpoint = false; |
||||
RAND_bytes ((uint8_t *)&nextTunnelID, 4); |
||||
if (!nextTunnelID) nextTunnelID = 1; // tunnelID can't be zero
|
||||
} |
||||
|
||||
void TunnelHopConfig::SetReplyHop (uint32_t replyTunnelID, const i2p::data::IdentHash& replyIdent) |
||||
{ |
||||
nextIdent = replyIdent; |
||||
nextTunnelID = replyTunnelID; |
||||
isEndpoint = true; |
||||
} |
||||
|
||||
void TunnelHopConfig::SetNext (TunnelHopConfig * n) |
||||
{ |
||||
next = n; |
||||
if (next) |
||||
{ |
||||
next->prev = this; |
||||
next->isGateway = false; |
||||
isEndpoint = false; |
||||
nextIdent = next->ident->GetIdentHash (); |
||||
nextTunnelID = next->tunnelID; |
||||
} |
||||
} |
||||
|
||||
void TunnelHopConfig::SetPrev (TunnelHopConfig * p) |
||||
{ |
||||
prev = p; |
||||
if (prev) |
||||
{ |
||||
prev->next = this; |
||||
prev->isEndpoint = false; |
||||
isGateway = false; |
||||
} |
||||
} |
||||
|
||||
void TunnelHopConfig::CreateBuildRequestRecord (uint8_t * record, uint32_t replyMsgID, BN_CTX * ctx) const |
||||
{ |
||||
uint8_t clearText[BUILD_REQUEST_RECORD_CLEAR_TEXT_SIZE]; |
||||
htobe32buf (clearText + BUILD_REQUEST_RECORD_RECEIVE_TUNNEL_OFFSET, tunnelID); |
||||
memcpy (clearText + BUILD_REQUEST_RECORD_OUR_IDENT_OFFSET, ident->GetIdentHash (), 32); |
||||
htobe32buf (clearText + BUILD_REQUEST_RECORD_NEXT_TUNNEL_OFFSET, nextTunnelID); |
||||
memcpy (clearText + BUILD_REQUEST_RECORD_NEXT_IDENT_OFFSET, nextIdent, 32); |
||||
memcpy (clearText + BUILD_REQUEST_RECORD_LAYER_KEY_OFFSET, layerKey, 32); |
||||
memcpy (clearText + BUILD_REQUEST_RECORD_IV_KEY_OFFSET, ivKey, 32); |
||||
memcpy (clearText + BUILD_REQUEST_RECORD_REPLY_KEY_OFFSET, replyKey, 32); |
||||
memcpy (clearText + BUILD_REQUEST_RECORD_REPLY_IV_OFFSET, replyIV, 16); |
||||
uint8_t flag = 0; |
||||
if (isGateway) flag |= 0x80; |
||||
if (isEndpoint) flag |= 0x40; |
||||
clearText[BUILD_REQUEST_RECORD_FLAG_OFFSET] = flag; |
||||
htobe32buf (clearText + BUILD_REQUEST_RECORD_REQUEST_TIME_OFFSET, i2p::util::GetHoursSinceEpoch ()); |
||||
htobe32buf (clearText + BUILD_REQUEST_RECORD_SEND_MSG_ID_OFFSET, replyMsgID); |
||||
RAND_bytes (clearText + BUILD_REQUEST_RECORD_PADDING_OFFSET, BUILD_REQUEST_RECORD_CLEAR_TEXT_SIZE - BUILD_REQUEST_RECORD_PADDING_OFFSET); |
||||
auto encryptor = ident->CreateEncryptor (nullptr); |
||||
if (encryptor) |
||||
{ |
||||
if (ident->GetCryptoKeyType () == i2p::data::CRYPTO_KEY_TYPE_ECIES_X25519_AEAD_RATCHET) |
||||
EncryptECIES (encryptor, clearText, record + BUILD_REQUEST_RECORD_ENCRYPTED_OFFSET, ctx); |
||||
else |
||||
encryptor->Encrypt (clearText, record + BUILD_REQUEST_RECORD_ENCRYPTED_OFFSET, ctx, false); |
||||
} |
||||
memcpy (record + BUILD_REQUEST_RECORD_TO_PEER_OFFSET, (const uint8_t *)ident->GetIdentHash (), 16); |
||||
} |
||||
|
||||
void TunnelHopConfig::EncryptECIES (std::shared_ptr<i2p::crypto::CryptoKeyEncryptor>& encryptor, |
||||
const uint8_t * clearText, uint8_t * encrypted, BN_CTX * ctx) const |
||||
{ |
||||
memset (encrypted, 0, 512); // TODO: implement
|
||||
} |
||||
} |
||||
} |
||||
Loading…
Reference in new issue