xref: /aosp_15_r20/external/pdfium/core/fdrm/fx_crypt.h (revision 3ac0a46f773bac49fa9476ec2b1cf3f8da5ec3a4)
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