| OLD | NEW |
| 1 // Copyright 2014 PDFium Authors. All rights reserved. | 1 // Copyright 2014 PDFium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com | 5 // Original code copyright 2014 Foxit Software Inc. http://www.foxitsoftware.com |
| 6 | 6 |
| 7 #include "core/fdrm/crypto/include/fx_crypt.h" | 7 #include "core/fdrm/crypto/include/fx_crypt.h" |
| 8 | 8 |
| 9 #ifdef __cplusplus | 9 #ifdef __cplusplus |
| 10 extern "C" { | 10 extern "C" { |
| (...skipping 44 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 55 uint32_t keylen) { | 55 uint32_t keylen) { |
| 56 rc4_state s; | 56 rc4_state s; |
| 57 CRYPT_ArcFourSetup(&s, key, keylen); | 57 CRYPT_ArcFourSetup(&s, key, keylen); |
| 58 CRYPT_ArcFourCrypt(&s, pData, size); | 58 CRYPT_ArcFourCrypt(&s, pData, size); |
| 59 } | 59 } |
| 60 struct md5_context { | 60 struct md5_context { |
| 61 uint32_t total[2]; | 61 uint32_t total[2]; |
| 62 uint32_t state[4]; | 62 uint32_t state[4]; |
| 63 uint8_t buffer[64]; | 63 uint8_t buffer[64]; |
| 64 }; | 64 }; |
| 65 #define GET_FX_DWORD(n, b, i) \ | 65 #define GET_UINT32(n, b, i) \ |
| 66 { \ | 66 { \ |
| 67 (n) = (uint32_t)((uint8_t*)b)[(i)] | \ | 67 (n) = (uint32_t)((uint8_t*)b)[(i)] | \ |
| 68 (((uint32_t)((uint8_t*)b)[(i) + 1]) << 8) | \ | 68 (((uint32_t)((uint8_t*)b)[(i) + 1]) << 8) | \ |
| 69 (((uint32_t)((uint8_t*)b)[(i) + 2]) << 16) | \ | 69 (((uint32_t)((uint8_t*)b)[(i) + 2]) << 16) | \ |
| 70 (((uint32_t)((uint8_t*)b)[(i) + 3]) << 24); \ | 70 (((uint32_t)((uint8_t*)b)[(i) + 3]) << 24); \ |
| 71 } | 71 } |
| 72 #define PUT_FX_DWORD(n, b, i) \ | 72 #define PUT_UINT32(n, b, i) \ |
| 73 { \ | 73 { \ |
| 74 (((uint8_t*)b)[(i)]) = (uint8_t)(((n)) & 0xFF); \ | 74 (((uint8_t*)b)[(i)]) = (uint8_t)(((n)) & 0xFF); \ |
| 75 (((uint8_t*)b)[(i) + 1]) = (uint8_t)(((n) >> 8) & 0xFF); \ | 75 (((uint8_t*)b)[(i) + 1]) = (uint8_t)(((n) >> 8) & 0xFF); \ |
| 76 (((uint8_t*)b)[(i) + 2]) = (uint8_t)(((n) >> 16) & 0xFF); \ | 76 (((uint8_t*)b)[(i) + 2]) = (uint8_t)(((n) >> 16) & 0xFF); \ |
| 77 (((uint8_t*)b)[(i) + 3]) = (uint8_t)(((n) >> 24) & 0xFF); \ | 77 (((uint8_t*)b)[(i) + 3]) = (uint8_t)(((n) >> 24) & 0xFF); \ |
| 78 } | 78 } |
| 79 void md5_process(struct md5_context* ctx, const uint8_t data[64]) { | 79 void md5_process(struct md5_context* ctx, const uint8_t data[64]) { |
| 80 uint32_t A, B, C, D, X[16]; | 80 uint32_t A, B, C, D, X[16]; |
| 81 GET_FX_DWORD(X[0], data, 0); | 81 GET_UINT32(X[0], data, 0); |
| 82 GET_FX_DWORD(X[1], data, 4); | 82 GET_UINT32(X[1], data, 4); |
| 83 GET_FX_DWORD(X[2], data, 8); | 83 GET_UINT32(X[2], data, 8); |
| 84 GET_FX_DWORD(X[3], data, 12); | 84 GET_UINT32(X[3], data, 12); |
| 85 GET_FX_DWORD(X[4], data, 16); | 85 GET_UINT32(X[4], data, 16); |
| 86 GET_FX_DWORD(X[5], data, 20); | 86 GET_UINT32(X[5], data, 20); |
| 87 GET_FX_DWORD(X[6], data, 24); | 87 GET_UINT32(X[6], data, 24); |
| 88 GET_FX_DWORD(X[7], data, 28); | 88 GET_UINT32(X[7], data, 28); |
| 89 GET_FX_DWORD(X[8], data, 32); | 89 GET_UINT32(X[8], data, 32); |
| 90 GET_FX_DWORD(X[9], data, 36); | 90 GET_UINT32(X[9], data, 36); |
| 91 GET_FX_DWORD(X[10], data, 40); | 91 GET_UINT32(X[10], data, 40); |
| 92 GET_FX_DWORD(X[11], data, 44); | 92 GET_UINT32(X[11], data, 44); |
| 93 GET_FX_DWORD(X[12], data, 48); | 93 GET_UINT32(X[12], data, 48); |
| 94 GET_FX_DWORD(X[13], data, 52); | 94 GET_UINT32(X[13], data, 52); |
| 95 GET_FX_DWORD(X[14], data, 56); | 95 GET_UINT32(X[14], data, 56); |
| 96 GET_FX_DWORD(X[15], data, 60); | 96 GET_UINT32(X[15], data, 60); |
| 97 #define S(x, n) ((x << n) | ((x & 0xFFFFFFFF) >> (32 - n))) | 97 #define S(x, n) ((x << n) | ((x & 0xFFFFFFFF) >> (32 - n))) |
| 98 #define P(a, b, c, d, k, s, t) \ | 98 #define P(a, b, c, d, k, s, t) \ |
| 99 { \ | 99 { \ |
| 100 a += F(b, c, d) + X[k] + t; \ | 100 a += F(b, c, d) + X[k] + t; \ |
| 101 a = S(a, s) + b; \ | 101 a = S(a, s) + b; \ |
| 102 } | 102 } |
| 103 A = ctx->state[0]; | 103 A = ctx->state[0]; |
| 104 B = ctx->state[1]; | 104 B = ctx->state[1]; |
| 105 C = ctx->state[2]; | 105 C = ctx->state[2]; |
| 106 D = ctx->state[3]; | 106 D = ctx->state[3]; |
| (...skipping 112 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 219 } | 219 } |
| 220 } | 220 } |
| 221 const uint8_t md5_padding[64] = { | 221 const uint8_t md5_padding[64] = { |
| 222 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, | 222 0x80, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
| 223 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, | 223 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, |
| 224 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}; | 224 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0}; |
| 225 void CRYPT_MD5Finish(void* pctx, uint8_t digest[16]) { | 225 void CRYPT_MD5Finish(void* pctx, uint8_t digest[16]) { |
| 226 struct md5_context* ctx = (struct md5_context*)pctx; | 226 struct md5_context* ctx = (struct md5_context*)pctx; |
| 227 uint32_t last, padn; | 227 uint32_t last, padn; |
| 228 uint8_t msglen[8]; | 228 uint8_t msglen[8]; |
| 229 PUT_FX_DWORD(ctx->total[0], msglen, 0); | 229 PUT_UINT32(ctx->total[0], msglen, 0); |
| 230 PUT_FX_DWORD(ctx->total[1], msglen, 4); | 230 PUT_UINT32(ctx->total[1], msglen, 4); |
| 231 last = (ctx->total[0] >> 3) & 0x3F; | 231 last = (ctx->total[0] >> 3) & 0x3F; |
| 232 padn = (last < 56) ? (56 - last) : (120 - last); | 232 padn = (last < 56) ? (56 - last) : (120 - last); |
| 233 CRYPT_MD5Update(ctx, md5_padding, padn); | 233 CRYPT_MD5Update(ctx, md5_padding, padn); |
| 234 CRYPT_MD5Update(ctx, msglen, 8); | 234 CRYPT_MD5Update(ctx, msglen, 8); |
| 235 PUT_FX_DWORD(ctx->state[0], digest, 0); | 235 PUT_UINT32(ctx->state[0], digest, 0); |
| 236 PUT_FX_DWORD(ctx->state[1], digest, 4); | 236 PUT_UINT32(ctx->state[1], digest, 4); |
| 237 PUT_FX_DWORD(ctx->state[2], digest, 8); | 237 PUT_UINT32(ctx->state[2], digest, 8); |
| 238 PUT_FX_DWORD(ctx->state[3], digest, 12); | 238 PUT_UINT32(ctx->state[3], digest, 12); |
| 239 } | 239 } |
| 240 void CRYPT_MD5Generate(const uint8_t* input, | 240 void CRYPT_MD5Generate(const uint8_t* input, |
| 241 uint32_t length, | 241 uint32_t length, |
| 242 uint8_t digest[16]) { | 242 uint8_t digest[16]) { |
| 243 md5_context ctx; | 243 md5_context ctx; |
| 244 CRYPT_MD5Start(&ctx); | 244 CRYPT_MD5Start(&ctx); |
| 245 CRYPT_MD5Update(&ctx, input, length); | 245 CRYPT_MD5Update(&ctx, input, length); |
| 246 CRYPT_MD5Finish(&ctx, digest); | 246 CRYPT_MD5Finish(&ctx, digest); |
| 247 } | 247 } |
| 248 static FX_BOOL (*g_PubKeyDecryptor)(const uint8_t* pData, | 248 static FX_BOOL (*g_PubKeyDecryptor)(const uint8_t* pData, |
| 249 uint32_t size, | 249 uint32_t size, |
| 250 uint8_t* data_buf, | 250 uint8_t* data_buf, |
| 251 uint32_t& data_len) = NULL; | 251 uint32_t& data_len) = NULL; |
| 252 void CRYPT_SetPubKeyDecryptor(FX_BOOL (*func)(const uint8_t* pData, | 252 void CRYPT_SetPubKeyDecryptor(FX_BOOL (*func)(const uint8_t* pData, |
| 253 uint32_t size, | 253 uint32_t size, |
| 254 uint8_t* data_buf, | 254 uint8_t* data_buf, |
| 255 uint32_t& data_len)) { | 255 uint32_t& data_len)) { |
| 256 g_PubKeyDecryptor = func; | 256 g_PubKeyDecryptor = func; |
| 257 } | 257 } |
| 258 #ifdef __cplusplus | 258 #ifdef __cplusplus |
| 259 }; | 259 }; |
| 260 #endif | 260 #endif |
| OLD | NEW |