1// Copyright 2017 PDFium Authors. All rights reserved. 2// Use of this source code is governed by a BSD-style license that can be 3// found in the LICENSE file. 4 5// Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com 6 7#include "core/fpdfapi/edit/cpdf_encryptor.h" 8#include "core/fpdfapi/parser/cpdf_crypto_handler.h" 9 10CPDF_Encryptor::CPDF_Encryptor(CPDF_CryptoHandler* pHandler, 11 int objnum, 12 uint8_t* src_data, 13 uint32_t src_size) 14 : m_pData(nullptr), m_dwSize(0), m_bNewBuf(false) { 15 if (src_size == 0) 16 return; 17 18 if (!pHandler) { 19 m_pData = (uint8_t*)src_data; 20 m_dwSize = src_size; 21 return; 22 } 23 m_dwSize = pHandler->EncryptGetSize(objnum, 0, src_data, src_size); 24 m_pData = FX_Alloc(uint8_t, m_dwSize); 25 pHandler->EncryptContent(objnum, 0, src_data, src_size, m_pData, m_dwSize); 26 m_bNewBuf = true; 27} 28 29CPDF_Encryptor::~CPDF_Encryptor() { 30 if (m_bNewBuf) 31 FX_Free(m_pData); 32} 33