create sha1_info
This commit is contained in:
93
src/states/state_send_start_sha1.cpp
Normal file
93
src/states/state_send_start_sha1.cpp
Normal file
@ -0,0 +1,93 @@
|
||||
#include "./state_send_start_sha1.hpp"
|
||||
|
||||
#include "../tox_client.hpp"
|
||||
#include "../tox_utils.hpp"
|
||||
#include "../hash_utils.hpp"
|
||||
#include "../ft_sha1_info.hpp"
|
||||
|
||||
#include <mio/mio.hpp>
|
||||
|
||||
#include <iostream>
|
||||
#include <cassert>
|
||||
|
||||
namespace States {
|
||||
|
||||
SendStartSHA1::SendStartSHA1(ToxClient& tcl, const CommandLine& cl) : StateI(tcl) {
|
||||
std::cout << "SendStartSHA1 start building sha1_info\n";
|
||||
std::error_code err;
|
||||
_file_map = mio::make_mmap_source(cl.send_path, 0, mio::map_entire_file, err);
|
||||
|
||||
// TODO: propper error checking
|
||||
|
||||
assert(!_file_map.empty());
|
||||
|
||||
// build info
|
||||
_sha1_info.file_name = "testfile.bin";
|
||||
_sha1_info.file_size = _file_map.length();
|
||||
|
||||
{ // build chunks
|
||||
size_t i = 0;
|
||||
for (; i + FTInfoSHA1::chunk_size < _file_map.length(); i += FTInfoSHA1::chunk_size) {
|
||||
_sha1_info.chunks.push_back(hash_sha1(_file_map.data()+i, FTInfoSHA1::chunk_size));
|
||||
}
|
||||
|
||||
if (i < _file_map.length()) {
|
||||
_sha1_info.chunks.push_back(hash_sha1(_file_map.data()+i, _file_map.length()-i));
|
||||
}
|
||||
}
|
||||
|
||||
{ // unload file and remap (for when the file was large)
|
||||
_file_map.unmap();
|
||||
_file_map = mio::make_mmap_source(cl.send_path, 0, mio::map_entire_file, err);
|
||||
}
|
||||
|
||||
std::cout << "SHA1Start chunks: " << _sha1_info.chunks.size() << "\n";
|
||||
|
||||
_sha1_info_data = _sha1_info.toBuffer();
|
||||
|
||||
std::cout << "SHA1Start sha1_info size: " << _sha1_info_data.size() << "\n";
|
||||
|
||||
_sha1_info_hash = hash_sha1(_sha1_info_data.data(), _sha1_info_data.size());
|
||||
|
||||
std::cout << "SHA1Start sha1_info_hash: " << bin2hex(_sha1_info_hash) << "\n";
|
||||
}
|
||||
|
||||
bool SendStartSHA1::iterate(void) {
|
||||
(void)_tcl._tox;
|
||||
return false;
|
||||
}
|
||||
|
||||
std::unique_ptr<StateI> SendStartSHA1::nextState(void) {
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
// sha1_info
|
||||
void SendStartSHA1::onFT1ReceiveRequestSHA1Info(uint32_t, uint32_t, const uint8_t*, size_t) {
|
||||
}
|
||||
|
||||
bool SendStartSHA1::onFT1ReceiveInitSHA1Info(uint32_t, uint32_t, const uint8_t*, size_t, const uint8_t, const size_t) {
|
||||
return false;
|
||||
}
|
||||
|
||||
void SendStartSHA1::onFT1ReceiveDataSHA1Info(uint32_t, uint32_t, uint8_t, size_t, const uint8_t*, size_t) {
|
||||
}
|
||||
|
||||
void SendStartSHA1::onFT1SendDataSHA1Info(uint32_t, uint32_t, uint8_t, size_t, uint8_t*, size_t) {
|
||||
}
|
||||
|
||||
// sha1_chunk
|
||||
void SendStartSHA1::onFT1ReceiveRequestSHA1Chunk(uint32_t, uint32_t, const uint8_t*, size_t) {
|
||||
}
|
||||
|
||||
bool SendStartSHA1::onFT1ReceiveInitSHA1Chunk(uint32_t, uint32_t, const uint8_t*, size_t, const uint8_t, const size_t) {
|
||||
return false;
|
||||
}
|
||||
|
||||
void SendStartSHA1::onFT1ReceiveDataSHA1Chunk(uint32_t, uint32_t, uint8_t, size_t, const uint8_t*, size_t) {
|
||||
}
|
||||
|
||||
void SendStartSHA1::onFT1SendDataSHA1Chunk(uint32_t, uint32_t, uint8_t, size_t, uint8_t*, size_t) {
|
||||
}
|
||||
|
||||
} // States
|
||||
|
43
src/states/state_send_start_sha1.hpp
Normal file
43
src/states/state_send_start_sha1.hpp
Normal file
@ -0,0 +1,43 @@
|
||||
#pragma once
|
||||
|
||||
#include "../state.hpp"
|
||||
|
||||
#include "../command_line.hpp"
|
||||
#include "../ft_sha1_info.hpp"
|
||||
|
||||
#include <mio/mio.hpp>
|
||||
|
||||
namespace States {
|
||||
|
||||
// we are hashing the given file
|
||||
// does not react to any ft events
|
||||
struct SendStartSHA1 final : public StateI {
|
||||
public: // general interface
|
||||
SendStartSHA1(ToxClient& tcl, const CommandLine& cl);
|
||||
~SendStartSHA1(void) override = default;
|
||||
|
||||
bool iterate(void) override;
|
||||
std::unique_ptr<StateI> nextState(void) override;
|
||||
|
||||
public: // callbacks
|
||||
// sha1_info
|
||||
void onFT1ReceiveRequestSHA1Info(uint32_t group_number, uint32_t peer_number, const uint8_t* file_id, size_t file_id_size) override;
|
||||
bool onFT1ReceiveInitSHA1Info(uint32_t group_number, uint32_t peer_number, const uint8_t* file_id, size_t file_id_size, const uint8_t transfer_id, const size_t file_size) override;
|
||||
void onFT1ReceiveDataSHA1Info(uint32_t group_number, uint32_t peer_number, uint8_t transfer_id, size_t data_offset, const uint8_t* data, size_t data_size) override;
|
||||
void onFT1SendDataSHA1Info(uint32_t group_number, uint32_t peer_number, uint8_t transfer_id, size_t data_offset, uint8_t* data, size_t data_size) override;
|
||||
|
||||
// sha1_chunk
|
||||
void onFT1ReceiveRequestSHA1Chunk(uint32_t group_number, uint32_t peer_number, const uint8_t* file_id, size_t file_id_size) override;
|
||||
bool onFT1ReceiveInitSHA1Chunk(uint32_t group_number, uint32_t peer_number, const uint8_t* file_id, size_t file_id_size, const uint8_t transfer_id, const size_t file_size) override;
|
||||
void onFT1ReceiveDataSHA1Chunk(uint32_t group_number, uint32_t peer_number, uint8_t transfer_id, size_t data_offset, const uint8_t* data, size_t data_size) override;
|
||||
void onFT1SendDataSHA1Chunk(uint32_t group_number, uint32_t peer_number, uint8_t transfer_id, size_t data_offset, uint8_t* data, size_t data_size) override;
|
||||
|
||||
private:
|
||||
mio::mmap_source _file_map;
|
||||
FTInfoSHA1 _sha1_info;
|
||||
std::vector<uint8_t> _sha1_info_data;
|
||||
std::vector<uint8_t> _sha1_info_hash;
|
||||
};
|
||||
|
||||
} // States
|
||||
|
Reference in New Issue
Block a user