// Copyright 2017 PDFium Authors. All rights reserved. // Use of this source code is governed by a BSD-style license that can be // found in the LICENSE file. // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com #include "core/fpdfapi/edit/cpdf_encryptor.h" #include "core/fpdfapi/parser/cpdf_crypto_handler.h" CPDF_Encryptor::CPDF_Encryptor(CPDF_CryptoHandler* pHandler, int objnum) : m_pHandler(pHandler), m_ObjNum(objnum) { ASSERT(m_pHandler); } std::vector<uint8_t> CPDF_Encryptor::Encrypt( pdfium::span<const uint8_t> src_data) const { if (src_data.empty()) return std::vector<uint8_t>(); std::vector<uint8_t> result; uint32_t buf_size = m_pHandler->EncryptGetSize(src_data); result.resize(buf_size); m_pHandler->EncryptContent(m_ObjNum, 0, src_data, result.data(), buf_size); // Updates |buf_size| with actual. result.resize(buf_size); return result; } CPDF_Encryptor::~CPDF_Encryptor() {}