1 files changed, 32 insertions, 0 deletions
diff --git a/core/fpdfapi/edit/cpdf_encryptor.cpp b/core/fpdfapi/edit/cpdf_encryptor.cpp
new file mode 100644
index 0000000000..858b5b6369
--- /dev/null
+++ b/core/fpdfapi/edit/cpdf_encryptor.cpp
@@ -0,0 +1,32 @@
+// 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,
+                               uint8_t* src_data,
+                               uint32_t src_size)
+    : m_pData(nullptr), m_dwSize(0), m_bNewBuf(false) {
+  if (src_size == 0)
+    return;
+
+  if (!pHandler) {
+    m_pData = (uint8_t*)src_data;
+    m_dwSize = src_size;
+    return;
+  }
+  m_dwSize = pHandler->EncryptGetSize(objnum, 0, src_data, src_size);
+  m_pData = FX_Alloc(uint8_t, m_dwSize);
+  pHandler->EncryptContent(objnum, 0, src_data, src_size, m_pData, m_dwSize);
+  m_bNewBuf = true;
+}
+
+CPDF_Encryptor::~CPDF_Encryptor() {
+  if (m_bNewBuf)
+    FX_Free(m_pData);
+}