1 files changed, 117 insertions, 0 deletions
diff --git a/core/fdrm/fx_crypt.h b/core/fdrm/fx_crypt.h
new file mode 100644
index 0000000000..4e8a7132a8
--- /dev/null
+++ b/core/fdrm/fx_crypt.h
@@ -0,0 +1,117 @@
+// Copyright 2014 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
+
+#ifndef CORE_FDRM_FX_CRYPT_H_
+#define CORE_FDRM_FX_CRYPT_H_
+
+#include "core/fxcrt/fx_system.h"
+
+constexpr int32_t kRC4ContextPermutationLength = 256;
+struct CRYPT_rc4_context {
+  int32_t x;
+  int32_t y;
+  int32_t m[kRC4ContextPermutationLength];
+};
+
+#define MAX_NR 14
+#define MAX_NB 8
+struct CRYPT_aes_context {
+  void (*encrypt)(CRYPT_aes_context* ctx, unsigned int* block);
+  void (*decrypt)(CRYPT_aes_context* ctx, unsigned int* block);
+  int Nb;
+  int Nr;
+  unsigned int keysched[(MAX_NR + 1) * MAX_NB];
+  unsigned int invkeysched[(MAX_NR + 1) * MAX_NB];
+  unsigned int iv[MAX_NB];
+};
+
+struct CRYPT_md5_context {
+  uint32_t total[2];
+  uint32_t state[4];
+  uint8_t buffer[64];
+};
+
+struct CRYPT_sha1_context {
+  unsigned int h[5];
+  unsigned char block[64];
+  int blkused;
+  unsigned int lenhi;
+  unsigned int lenlo;
+};
+
+struct CRYPT_sha2_context {
+  uint64_t total[2];
+  uint64_t state[8];
+  uint8_t buffer[128];
+};
+
+void CRYPT_ArcFourCryptBlock(uint8_t* data,
+                             uint32_t size,
+                             const uint8_t* key,
+                             uint32_t keylen);
+void CRYPT_ArcFourSetup(CRYPT_rc4_context* context,
+                        const uint8_t* key,
+                        uint32_t length);
+void CRYPT_ArcFourCrypt(CRYPT_rc4_context* context,
+                        uint8_t* data,
+                        uint32_t size);
+
+void CRYPT_AESSetKey(CRYPT_aes_context* context,
+                     const uint8_t* key,
+                     uint32_t keylen,
+                     bool bEncrypt);
+void CRYPT_AESSetIV(CRYPT_aes_context* context, const uint8_t* iv);
+void CRYPT_AESDecrypt(CRYPT_aes_context* context,
+                      uint8_t* dest,
+                      const uint8_t* src,
+                      uint32_t size);
+void CRYPT_AESEncrypt(CRYPT_aes_context* context,
+                      uint8_t* dest,
+                      const uint8_t* src,
+                      uint32_t size);
+
+void CRYPT_MD5Start(CRYPT_md5_context* context);
+void CRYPT_MD5Update(CRYPT_md5_context* context,
+                     const uint8_t* data,
+                     uint32_t size);
+void CRYPT_MD5Finish(CRYPT_md5_context* context, uint8_t digest[16]);
+void CRYPT_MD5Generate(const uint8_t* data, uint32_t size, uint8_t digest[16]);
+
+void CRYPT_SHA1Start(CRYPT_sha1_context* context);
+void CRYPT_SHA1Update(CRYPT_sha1_context* context,
+                      const uint8_t* data,
+                      uint32_t size);
+void CRYPT_SHA1Finish(CRYPT_sha1_context* context, uint8_t digest[20]);
+void CRYPT_SHA1Generate(const uint8_t* data, uint32_t size, uint8_t digest[20]);
+
+void CRYPT_SHA256Start(CRYPT_sha2_context* context);
+void CRYPT_SHA256Update(CRYPT_sha2_context* context,
+                        const uint8_t* data,
+                        uint32_t size);
+void CRYPT_SHA256Finish(CRYPT_sha2_context* context, uint8_t digest[32]);
+void CRYPT_SHA256Generate(const uint8_t* data,
+                          uint32_t size,
+                          uint8_t digest[32]);
+
+void CRYPT_SHA384Start(CRYPT_sha2_context* context);
+void CRYPT_SHA384Update(CRYPT_sha2_context* context,
+                        const uint8_t* data,
+                        uint32_t size);
+void CRYPT_SHA384Finish(CRYPT_sha2_context* context, uint8_t digest[48]);
+void CRYPT_SHA384Generate(const uint8_t* data,
+                          uint32_t size,
+                          uint8_t digest[48]);
+
+void CRYPT_SHA512Start(CRYPT_sha2_context* context);
+void CRYPT_SHA512Update(CRYPT_sha2_context* context,
+                        const uint8_t* data,
+                        uint32_t size);
+void CRYPT_SHA512Finish(CRYPT_sha2_context* context, uint8_t digest[64]);
+void CRYPT_SHA512Generate(const uint8_t* data,
+                          uint32_t size,
+                          uint8_t digest[64]);
+
+#endif  // CORE_FDRM_FX_CRYPT_H_