lsm/PdfPig
1
0
Fork 0
mirror of https://github.com/UglyToad/PdfPig.git synced 2025-09-22 12:09:50 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

finish support for revision 5 encryption using aes 256 #34

Browse Source
This commit is contained in:
Eliot Jones
2019-06-13 19:46:08 +01:00
parent d0a3cd398f
commit 4c716fcbd6
6 changed files with 52 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

56
src/UglyToad.PdfPig.Tests/Integration/LocalTests.cs
View File

@@ -1,39 +1,39 @@
namespace UglyToad.PdfPig.Tests.Integration
{
//using System;
//using System.Diagnostics;
//using System.IO;
//using Xunit;
using System;
using System.Diagnostics;
using System.IO;
using Xunit;
/// <summary>
/// A class for testing files which are not checked in to source control.
/// </summary>
public class LocalTests
{
//[Fact]
//public void Tests()
//{
// var files = Directory.GetFiles(@"C:\temp\pdfs");
[Fact]
public void Tests()
{
var files = Directory.GetFiles(@"C:\temp\pdfs");
// foreach (var file in files)
// {
// try
// {
// using (var document = PdfDocument.Open(file, new ParsingOptions { UseLenientParsing = false }))
// {
// for (var i = 1; i <= document.NumberOfPages; i++)
// {
// var page = document.GetPage(i);
// var text = page.Text;
// Trace.WriteLine(text);
// }
// }
// }
// catch (Exception ex)
// {
// throw new InvalidOperationException($"Error parsing: {Path.GetFileName(file)}.", ex);
// }
// }
//}
foreach (var file in files)
{
try
{
using (var document = PdfDocument.Open(file, new ParsingOptions { UseLenientParsing = false }))
{
for (var i = 1; i <= document.NumberOfPages; i++)
{
var page = document.GetPage(i);
var text = page.Text;
Trace.WriteLine(text);
}
}
}
catch (Exception ex)
{
throw new InvalidOperationException($"Error parsing: {Path.GetFileName(file)}.", ex);
}
}
}
}
}

8
src/UglyToad.PdfPig/Encryption/CryptDictionary.cs
View File

@@ -45,7 +45,13 @@
/// (PDF 1.6) The application asks the security handler for the encryption key and implicitly decrypts data using the AES algorithm in Cipher Block Chaining (CBC) mode
/// with a 16-byte block size and an initialization vector that is randomly generated and placed as the first 16 bytes in the stream or string.
/// </summary>
AesV2
AesV2,
/// <summary>
/// The application asks the security handler for the encryption key and implicitly decrypts data using the AES-256 algorithm in Cipher Block Chaining (CBC) with padding mode
/// with a 16-byte block size and an initialization vector that is randomly generated and placed as the first 16 bytes in the stream or string.
/// The key size shall be 256 bits.
/// </summary>
AesV3
}
/// <summary>

4
src/UglyToad.PdfPig/Encryption/CryptHandler.cs
View File

@@ -72,6 +72,10 @@
{
method = CryptDictionary.Method.AesV2;
}
else if (cfmName == NameToken.Aesv3)
{
method = CryptDictionary.Method.AesV3;
}
else
{
throw new PdfDocumentEncryptedException($"Unrecognized CFM option for crypt filter {cfm}: {cryptDictionaryToken}.");

8
src/UglyToad.PdfPig/Encryption/EncryptionAlgorithmCode.cs
View File

@@ -22,8 +22,12 @@
/// </summary>
UnpublishedAlgorithm40To128BitKey = 3,
/// <summary>
/// The security handler defines the use of encryption and decryption in the document.
/// The security handler defines the use of encryption and decryption in the document with a key length of 128 bits.
/// </summary>
SecurityHandlerInDocument
SecurityHandlerInDocument = 4,
/// <summary>
/// The security handler defines the use of encryption and decryption in the document with a key length of 256 bits.
/// </summary>
SecurityHandlerInDocument256 = 5
}
}

3
src/UglyToad.PdfPig/Encryption/EncryptionDictionary.cs
View File

@@ -75,7 +75,8 @@
{
cryptHandler = null;
if (EncryptionAlgorithmCode != EncryptionAlgorithmCode.SecurityHandlerInDocument)
if (EncryptionAlgorithmCode != EncryptionAlgorithmCode.SecurityHandlerInDocument
&& EncryptionAlgorithmCode != EncryptionAlgorithmCode.SecurityHandlerInDocument256)
{
return false;
}

8
src/UglyToad.PdfPig/Encryption/EncryptionHandler.cs
View File

@@ -55,7 +55,8 @@
useAes = false;
if (encryptionDictionary.EncryptionAlgorithmCode == EncryptionAlgorithmCode.SecurityHandlerInDocument)
if (encryptionDictionary.EncryptionAlgorithmCode == EncryptionAlgorithmCode.SecurityHandlerInDocument
|| encryptionDictionary.EncryptionAlgorithmCode == EncryptionAlgorithmCode.SecurityHandlerInDocument256)
{
if (!encryptionDictionary.TryGetCryptHandler(out var cryptHandlerLocal))
{
@@ -64,7 +65,8 @@
cryptHandler = cryptHandlerLocal;
useAes = cryptHandlerLocal?.StreamDictionary?.Name == CryptDictionary.Method.AesV2;
useAes = cryptHandlerLocal?.StreamDictionary?.Name == CryptDictionary.Method.AesV2
|| cryptHandlerLocal?.StreamDictionary?.Name == CryptDictionary.Method.AesV3;
}
var charset = OtherEncodings.Iso88591;
@@ -407,7 +409,7 @@
{
if (useAes && encryptionKey.Length == 32)
{
throw new PdfDocumentEncryptedException("Decryption for AES-256 not currently supported.", encryptionDictionary);
return AesEncryptionHelper.Decrypt(data, encryptionKey);
}
var finalKey = GetObjectKey(reference);