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/fpdfapi/fpdf_parser/cpdf_standard_security_handler.h" | 7 #include "core/fpdfapi/fpdf_parser/cpdf_standard_security_handler.h" |
8 | 8 |
9 #include <time.h> | 9 #include <time.h> |
10 | 10 |
(...skipping 148 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
159 } | 159 } |
160 return TRUE; | 160 return TRUE; |
161 } | 161 } |
162 | 162 |
163 FX_BOOL CPDF_StandardSecurityHandler::LoadDict(CPDF_Dictionary* pEncryptDict) { | 163 FX_BOOL CPDF_StandardSecurityHandler::LoadDict(CPDF_Dictionary* pEncryptDict) { |
164 m_pEncryptDict = pEncryptDict; | 164 m_pEncryptDict = pEncryptDict; |
165 m_Version = pEncryptDict->GetIntegerBy("V"); | 165 m_Version = pEncryptDict->GetIntegerBy("V"); |
166 m_Revision = pEncryptDict->GetIntegerBy("R"); | 166 m_Revision = pEncryptDict->GetIntegerBy("R"); |
167 m_Permissions = pEncryptDict->GetIntegerBy("P", -1); | 167 m_Permissions = pEncryptDict->GetIntegerBy("P", -1); |
168 if (m_Version < 4) { | 168 if (m_Version < 4) { |
169 return _LoadCryptInfo(pEncryptDict, CFX_ByteString(), m_Cipher, m_KeyLen); | 169 return _LoadCryptInfo(pEncryptDict, CFX_ByteStringC(), m_Cipher, m_KeyLen); |
170 } | 170 } |
171 CFX_ByteString stmf_name = pEncryptDict->GetStringBy("StmF"); | 171 CFX_ByteString stmf_name = pEncryptDict->GetStringBy("StmF"); |
172 CFX_ByteString strf_name = pEncryptDict->GetStringBy("StrF"); | 172 CFX_ByteString strf_name = pEncryptDict->GetStringBy("StrF"); |
173 if (stmf_name != strf_name) { | 173 if (stmf_name != strf_name) { |
174 return FALSE; | 174 return FALSE; |
175 } | 175 } |
176 if (!_LoadCryptInfo(pEncryptDict, strf_name, m_Cipher, m_KeyLen)) { | 176 if (!_LoadCryptInfo(pEncryptDict, strf_name.AsByteStringC(), m_Cipher, |
| 177 m_KeyLen)) { |
177 return FALSE; | 178 return FALSE; |
178 } | 179 } |
179 return TRUE; | 180 return TRUE; |
180 } | 181 } |
181 | 182 |
182 FX_BOOL CPDF_StandardSecurityHandler::LoadDict(CPDF_Dictionary* pEncryptDict, | 183 FX_BOOL CPDF_StandardSecurityHandler::LoadDict(CPDF_Dictionary* pEncryptDict, |
183 uint32_t type, | 184 uint32_t type, |
184 int& cipher, | 185 int& cipher, |
185 int& key_len) { | 186 int& key_len) { |
186 m_pEncryptDict = pEncryptDict; | 187 m_pEncryptDict = pEncryptDict; |
187 m_Version = pEncryptDict->GetIntegerBy("V"); | 188 m_Version = pEncryptDict->GetIntegerBy("V"); |
188 m_Revision = pEncryptDict->GetIntegerBy("R"); | 189 m_Revision = pEncryptDict->GetIntegerBy("R"); |
189 m_Permissions = pEncryptDict->GetIntegerBy("P", -1); | 190 m_Permissions = pEncryptDict->GetIntegerBy("P", -1); |
190 CFX_ByteString strf_name, stmf_name; | 191 CFX_ByteString strf_name, stmf_name; |
191 if (m_Version >= 4) { | 192 if (m_Version >= 4) { |
192 stmf_name = pEncryptDict->GetStringBy("StmF"); | 193 stmf_name = pEncryptDict->GetStringBy("StmF"); |
193 strf_name = pEncryptDict->GetStringBy("StrF"); | 194 strf_name = pEncryptDict->GetStringBy("StrF"); |
194 if (stmf_name != strf_name) { | 195 if (stmf_name != strf_name) { |
195 return FALSE; | 196 return FALSE; |
196 } | 197 } |
197 } | 198 } |
198 if (!_LoadCryptInfo(pEncryptDict, strf_name, cipher, key_len)) { | 199 if (!_LoadCryptInfo(pEncryptDict, strf_name.AsByteStringC(), cipher, |
| 200 key_len)) { |
199 return FALSE; | 201 return FALSE; |
200 } | 202 } |
201 m_Cipher = cipher; | 203 m_Cipher = cipher; |
202 m_KeyLen = key_len; | 204 m_KeyLen = key_len; |
203 return TRUE; | 205 return TRUE; |
204 } | 206 } |
205 | 207 |
206 FX_BOOL CPDF_StandardSecurityHandler::GetCryptInfo(int& cipher, | 208 FX_BOOL CPDF_StandardSecurityHandler::GetCryptInfo(int& cipher, |
207 const uint8_t*& buffer, | 209 const uint8_t*& buffer, |
208 int& keylen) { | 210 int& keylen) { |
(...skipping 496 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
705 buf[11] = 'b'; | 707 buf[11] = 'b'; |
706 uint8_t* aes = FX_Alloc(uint8_t, 2048); | 708 uint8_t* aes = FX_Alloc(uint8_t, 2048); |
707 CRYPT_AESSetKey(aes, 16, key, 32, TRUE); | 709 CRYPT_AESSetKey(aes, 16, key, 32, TRUE); |
708 uint8_t iv[16], buf1[16]; | 710 uint8_t iv[16], buf1[16]; |
709 FXSYS_memset(iv, 0, 16); | 711 FXSYS_memset(iv, 0, 16); |
710 CRYPT_AESSetIV(aes, iv); | 712 CRYPT_AESSetIV(aes, iv); |
711 CRYPT_AESEncrypt(aes, buf1, buf, 16); | 713 CRYPT_AESEncrypt(aes, buf1, buf, 16); |
712 FX_Free(aes); | 714 FX_Free(aes); |
713 pEncryptDict->SetAtString("Perms", CFX_ByteString(buf1, 16)); | 715 pEncryptDict->SetAtString("Perms", CFX_ByteString(buf1, 16)); |
714 } | 716 } |
OLD | NEW |