1 // Copyright 2014 The PDFium Authors 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 #ifndef CORE_FDRM_FX_CRYPT_H_ 8 #define CORE_FDRM_FX_CRYPT_H_ 9 10 #include <stdint.h> 11 12 #include "third_party/base/containers/span.h" 13 14 struct CRYPT_rc4_context { 15 static constexpr int32_t kPermutationLength = 256; 16 17 int32_t x; 18 int32_t y; 19 int32_t m[kPermutationLength]; 20 }; 21 22 struct CRYPT_aes_context { 23 static constexpr int kMaxNb = 8; 24 static constexpr int kMaxNr = 14; 25 static constexpr int kSchedSize = (kMaxNr + 1) * kMaxNb; 26 27 int Nb; 28 int Nr; 29 unsigned int keysched[kSchedSize]; 30 unsigned int invkeysched[kSchedSize]; 31 unsigned int iv[kMaxNb]; 32 }; 33 34 struct CRYPT_md5_context { 35 uint32_t total[2]; 36 uint32_t state[4]; 37 uint8_t buffer[64]; 38 }; 39 40 struct CRYPT_sha1_context { 41 uint64_t total_bytes; 42 uint32_t blkused; // Constrained to [0, 64). 43 uint32_t h[5]; 44 uint8_t block[64]; 45 }; 46 47 struct CRYPT_sha2_context { 48 uint64_t total_bytes; 49 uint64_t state[8]; 50 uint8_t buffer[128]; 51 }; 52 53 void CRYPT_ArcFourCryptBlock(pdfium::span<uint8_t> data, 54 pdfium::span<const uint8_t> key); 55 void CRYPT_ArcFourSetup(CRYPT_rc4_context* context, 56 pdfium::span<const uint8_t> key); 57 void CRYPT_ArcFourCrypt(CRYPT_rc4_context* context, pdfium::span<uint8_t> data); 58 59 void CRYPT_AESSetKey(CRYPT_aes_context* context, 60 const uint8_t* key, 61 uint32_t keylen); 62 void CRYPT_AESSetIV(CRYPT_aes_context* context, const uint8_t* iv); 63 void CRYPT_AESDecrypt(CRYPT_aes_context* context, 64 uint8_t* dest, 65 const uint8_t* src, 66 uint32_t size); 67 void CRYPT_AESEncrypt(CRYPT_aes_context* context, 68 uint8_t* dest, 69 const uint8_t* src, 70 uint32_t size); 71 72 CRYPT_md5_context CRYPT_MD5Start(); 73 void CRYPT_MD5Update(CRYPT_md5_context* context, 74 pdfium::span<const uint8_t> data); 75 void CRYPT_MD5Finish(CRYPT_md5_context* context, uint8_t digest[16]); 76 void CRYPT_MD5Generate(pdfium::span<const uint8_t> data, uint8_t digest[16]); 77 78 void CRYPT_SHA1Start(CRYPT_sha1_context* context); 79 void CRYPT_SHA1Update(CRYPT_sha1_context* context, 80 const uint8_t* data, 81 uint32_t size); 82 void CRYPT_SHA1Finish(CRYPT_sha1_context* context, uint8_t digest[20]); 83 void CRYPT_SHA1Generate(const uint8_t* data, uint32_t size, uint8_t digest[20]); 84 85 void CRYPT_SHA256Start(CRYPT_sha2_context* context); 86 void CRYPT_SHA256Update(CRYPT_sha2_context* context, 87 const uint8_t* data, 88 uint32_t size); 89 void CRYPT_SHA256Finish(CRYPT_sha2_context* context, uint8_t digest[32]); 90 void CRYPT_SHA256Generate(const uint8_t* data, 91 uint32_t size, 92 uint8_t digest[32]); 93 94 void CRYPT_SHA384Start(CRYPT_sha2_context* context); 95 void CRYPT_SHA384Update(CRYPT_sha2_context* context, 96 const uint8_t* data, 97 uint32_t size); 98 void CRYPT_SHA384Finish(CRYPT_sha2_context* context, uint8_t digest[48]); 99 void CRYPT_SHA384Generate(const uint8_t* data, 100 uint32_t size, 101 uint8_t digest[48]); 102 103 void CRYPT_SHA512Start(CRYPT_sha2_context* context); 104 void CRYPT_SHA512Update(CRYPT_sha2_context* context, 105 const uint8_t* data, 106 uint32_t size); 107 void CRYPT_SHA512Finish(CRYPT_sha2_context* context, uint8_t digest[64]); 108 void CRYPT_SHA512Generate(const uint8_t* data, 109 uint32_t size, 110 uint8_t digest[64]); 111 112 #endif // CORE_FDRM_FX_CRYPT_H_ 113