#include "CryptTask.h"
#include "crypt.h"
#include <chrono>
#include <iostream>
#include <memory>
#include <thread>

//初始化加解密秘钥
void CryptTask::Init(std::string passwd){
    enc_ = std::make_shared<Crypt>();
    enc_->Init(passwd);
}

void CryptTask::Main(){
    std::cout<<"begin CryptTask::Main()"<<std::endl;
    while(!is_exit_){
        auto data = PopFront();
        if(!data){
            std::this_thread::sleep_for(std::chrono::milliseconds(10));
            continue;        
        }
        auto out = Data::Make(mem_pool_);
        int outsize = data->size() + enc_->getPadding(data->size());
        int data_size = 0;
        bool is_end = data->end();
        out->New(outsize);
        if(is_encrypt_){
            data_size = enc_->Encrypt((char*)data->data(), data->size(), (char*)out->data(),is_end);
        }else{
            data_size = enc_->Encrypt((char*)data->data(), data->size(), (char*)out->data(),is_end);
        }
        out->set_size(data_size);
        std::cout<<"<"<<out->size()<<">"<<std::flush;
        out->set_end(data->end());
        if(next_){
            next_->PushBack(out);
        }
        if(data->end()){
            break;
        }
    }
    std::cout<<"begin CryptTask::End()"<<std::endl;
    
}