mirror of
https://github.com/ClickHouse/ClickHouse.git
synced 2024-12-12 01:12:12 +00:00
62 lines
1.4 KiB
C++
62 lines
1.4 KiB
C++
#include <IO/WriteBufferFromEncryptedFile.h>
|
|
|
|
#if USE_SSL
|
|
|
|
namespace DB
|
|
{
|
|
|
|
WriteBufferFromEncryptedFile::WriteBufferFromEncryptedFile(
|
|
size_t buffer_size_,
|
|
std::unique_ptr<WriteBufferFromFileBase> out_,
|
|
const String & key_,
|
|
const FileEncryption::Header & header_,
|
|
size_t old_file_size)
|
|
: WriteBufferDecorator<WriteBufferFromFileBase>(std::move(out_), buffer_size_, nullptr, 0)
|
|
, header(header_)
|
|
, flush_header(!old_file_size)
|
|
, encryptor(header.algorithm, key_, header.init_vector)
|
|
{
|
|
encryptor.setOffset(old_file_size);
|
|
}
|
|
|
|
WriteBufferFromEncryptedFile::~WriteBufferFromEncryptedFile()
|
|
{
|
|
finalize();
|
|
}
|
|
|
|
void WriteBufferFromEncryptedFile::finalizeBefore()
|
|
{
|
|
/// If buffer has pending data - write it.
|
|
next();
|
|
|
|
/// Note that if there is no data to write an empty file will be written, even without the initialization vector
|
|
/// (see nextImpl(): it writes the initialization vector only if there is some data ready to write).
|
|
/// That's fine because DiskEncrypted allows files without initialization vectors when they're empty.
|
|
}
|
|
|
|
void WriteBufferFromEncryptedFile::sync()
|
|
{
|
|
/// If buffer has pending data - write it.
|
|
next();
|
|
|
|
out->sync();
|
|
}
|
|
|
|
void WriteBufferFromEncryptedFile::nextImpl()
|
|
{
|
|
if (!offset())
|
|
return;
|
|
|
|
if (flush_header)
|
|
{
|
|
header.write(*out);
|
|
flush_header = false;
|
|
}
|
|
|
|
encryptor.encrypt(working_buffer.begin(), offset(), *out);
|
|
}
|
|
|
|
}
|
|
|
|
#endif
|