mirror of
https://github.com/unidoc/unipdf.git
synced 2025-04-30 13:48:51 +08:00
1803 lines
48 KiB
Go
1803 lines
48 KiB
Go
/*
|
||
* This file is subject to the terms and conditions defined in
|
||
* file 'LICENSE.md', which is part of this source code package.
|
||
*/
|
||
|
||
package core
|
||
|
||
import (
|
||
"bytes"
|
||
"crypto/aes"
|
||
"crypto/cipher"
|
||
"crypto/md5"
|
||
"crypto/rand"
|
||
"crypto/rc4"
|
||
"crypto/sha256"
|
||
"crypto/sha512"
|
||
"encoding/binary"
|
||
"errors"
|
||
"fmt"
|
||
"hash"
|
||
"io"
|
||
"math"
|
||
"time"
|
||
|
||
"github.com/unidoc/unidoc/common"
|
||
)
|
||
|
||
type Version struct {
|
||
Major int
|
||
Minor int
|
||
}
|
||
|
||
type EncryptInfo struct {
|
||
Version
|
||
Encrypt *PdfObjectDictionary
|
||
ID0, ID1 string
|
||
}
|
||
|
||
// PdfCryptNewEncrypt makes the document crypt handler based on a specified crypt filter.
|
||
func PdfCryptNewEncrypt(cf CryptFilter, userPass, ownerPass []byte, perm AccessPermissions) (*PdfCrypt, *EncryptInfo, error) {
|
||
crypter := &PdfCrypt{
|
||
EncryptedObjects: make(map[PdfObject]bool),
|
||
CryptFilters: make(CryptFilters),
|
||
encryptStd: stdEncryptDict{
|
||
P: perm,
|
||
EncryptMetadata: true,
|
||
},
|
||
}
|
||
// TODO(dennwc): define it in the CF interface
|
||
var vers Version
|
||
switch cf.(type) {
|
||
case cryptFilterV2:
|
||
crypter.encrypt.V = 2
|
||
crypter.encryptStd.R = 3
|
||
case cryptFilterAESV2:
|
||
vers.Major, vers.Minor = 1, 5
|
||
crypter.encrypt.V = 4
|
||
crypter.encryptStd.R = 4
|
||
case cryptFilterAESV3:
|
||
vers.Major, vers.Minor = 2, 0
|
||
crypter.encrypt.V = 5
|
||
crypter.encryptStd.R = 6 // TODO(dennwc): a way to set R=5?
|
||
}
|
||
if cf != nil {
|
||
crypter.encrypt.Length = cf.KeyLength() * 8
|
||
}
|
||
const (
|
||
defaultFilter = StandardCryptFilter
|
||
)
|
||
crypter.CryptFilters[defaultFilter] = cf
|
||
if crypter.encrypt.V >= 4 {
|
||
crypter.StreamFilter = defaultFilter
|
||
crypter.StringFilter = defaultFilter
|
||
}
|
||
ed := crypter.newEncyptDict()
|
||
|
||
// Prepare the ID object for the trailer.
|
||
hashcode := md5.Sum([]byte(time.Now().Format(time.RFC850)))
|
||
id0 := string(hashcode[:])
|
||
b := make([]byte, 100)
|
||
rand.Read(b)
|
||
hashcode = md5.Sum(b)
|
||
id1 := string(hashcode[:])
|
||
common.Log.Trace("Random b: % x", b)
|
||
|
||
common.Log.Trace("Gen Id 0: % x", id0)
|
||
|
||
// Generate encryption parameters
|
||
if crypter.encryptStd.R < 5 {
|
||
crypter.Id0 = string(id0)
|
||
|
||
// Make the O and U objects.
|
||
O, err := crypter.Alg3(userPass, ownerPass)
|
||
if err != nil {
|
||
common.Log.Debug("ERROR: Error generating O for encryption (%s)", err)
|
||
return nil, nil, err
|
||
}
|
||
crypter.encryptStd.O = []byte(O)
|
||
common.Log.Trace("gen O: % x", O)
|
||
U, key, err := crypter.Alg5(userPass)
|
||
if err != nil {
|
||
common.Log.Debug("ERROR: Error generating O for encryption (%s)", err)
|
||
return nil, nil, err
|
||
}
|
||
common.Log.Trace("gen U: % x", U)
|
||
crypter.encryptStd.U = []byte(U)
|
||
crypter.EncryptionKey = key
|
||
|
||
ed.Set("O", MakeHexString(O))
|
||
ed.Set("U", MakeHexString(U))
|
||
} else { // R >= 5
|
||
err := crypter.GenerateParams(userPass, ownerPass)
|
||
if err != nil {
|
||
return nil, nil, err
|
||
}
|
||
ed.Set("O", MakeString(string(crypter.encryptStd.O)))
|
||
ed.Set("U", MakeString(string(crypter.encryptStd.U)))
|
||
ed.Set("OE", MakeString(string(crypter.encryptStd.OE)))
|
||
ed.Set("UE", MakeString(string(crypter.encryptStd.UE)))
|
||
ed.Set("EncryptMetadata", MakeBool(crypter.encryptStd.EncryptMetadata))
|
||
if crypter.encryptStd.R > 5 {
|
||
ed.Set("Perms", MakeString(string(crypter.encryptStd.Perms)))
|
||
}
|
||
}
|
||
if crypter.encrypt.V >= 4 {
|
||
if err := crypter.saveCryptFilters(ed); err != nil {
|
||
return nil, nil, err
|
||
}
|
||
}
|
||
|
||
return crypter, &EncryptInfo{
|
||
Version: vers,
|
||
Encrypt: ed,
|
||
ID0: id0, ID1: id1,
|
||
}, nil
|
||
}
|
||
|
||
// PdfCrypt provides PDF encryption/decryption support.
|
||
// The PDF standard supports encryption of strings and streams (Section 7.6).
|
||
// TODO (v3): Consider unexporting.
|
||
type PdfCrypt struct {
|
||
encrypt encryptDict
|
||
encryptStd stdEncryptDict
|
||
|
||
Id0 string
|
||
EncryptionKey []byte
|
||
DecryptedObjects map[PdfObject]bool
|
||
EncryptedObjects map[PdfObject]bool
|
||
Authenticated bool
|
||
// Crypt filters (V4).
|
||
CryptFilters CryptFilters
|
||
StreamFilter string
|
||
StringFilter string
|
||
|
||
parser *PdfParser
|
||
|
||
decryptedObjNum map[int]struct{}
|
||
ivAESZero []byte // a zero buffer used as an initialization vector for AES
|
||
}
|
||
|
||
func (crypt *PdfCrypt) newEncyptDict() *PdfObjectDictionary {
|
||
// Generate the encryption dictionary.
|
||
ed := MakeDict()
|
||
ed.Set("Filter", MakeName("Standard"))
|
||
ed.Set("V", MakeInteger(int64(crypt.encrypt.V)))
|
||
ed.Set("Length", MakeInteger(int64(crypt.encrypt.Length)))
|
||
ed.Set("P", MakeInteger(int64(crypt.encryptStd.P)))
|
||
ed.Set("R", MakeInteger(int64(crypt.encryptStd.R)))
|
||
return ed
|
||
}
|
||
|
||
// String returns a descriptive information string about the encryption method used.
|
||
func (crypt *PdfCrypt) String() string {
|
||
if crypt == nil {
|
||
return ""
|
||
}
|
||
// TODO(dennwc): define a String method on CF
|
||
str := crypt.encrypt.Filter + " - "
|
||
|
||
if crypt.encrypt.V == 0 {
|
||
str += "Undocumented algorithm"
|
||
} else if crypt.encrypt.V == 1 {
|
||
// RC4 or AES (bits: 40)
|
||
str += "RC4: 40 bits"
|
||
} else if crypt.encrypt.V == 2 {
|
||
str += fmt.Sprintf("RC4: %d bits", crypt.encrypt.Length)
|
||
} else if crypt.encrypt.V == 3 {
|
||
str += "Unpublished algorithm"
|
||
} else if crypt.encrypt.V >= 4 {
|
||
// Look at CF, StmF, StrF
|
||
str += fmt.Sprintf("Stream filter: %s - String filter: %s", crypt.StreamFilter, crypt.StringFilter)
|
||
str += "; Crypt filters:"
|
||
for name, cf := range crypt.CryptFilters {
|
||
str += fmt.Sprintf(" - %s: %s (%d)", name, cf.Name(), cf.KeyLength())
|
||
}
|
||
}
|
||
perms := crypt.GetAccessPermissions()
|
||
str += fmt.Sprintf(" - %#v", perms)
|
||
|
||
return str
|
||
}
|
||
|
||
type authEvent string
|
||
|
||
const (
|
||
authEventDocOpen = authEvent("DocOpen")
|
||
authEventEFOpen = authEvent("EFOpen")
|
||
)
|
||
|
||
type cryptFiltersDict map[string]cryptFilterDict
|
||
|
||
// encryptDict is a set of field common to all encryption dictionaries.
|
||
type encryptDict struct {
|
||
Filter string // (Required) The name of the preferred security handler for this document.
|
||
V int // (Required) A code specifying the algorithm to be used in encrypting and decrypting the document.
|
||
SubFilter string // Completely specifies the format and interpretation of the encryption dictionary.
|
||
Length int // The length of the encryption key, in bits.
|
||
CF cryptFiltersDict // Crypt filters dictionary.
|
||
StmF string // The filter that shall be used by default when decrypting streams.
|
||
StrF string // The filter that shall be used when decrypting all strings in the document.
|
||
EFF string // The filter that shall be used when decrypting embedded file streams.
|
||
}
|
||
|
||
// stdEncryptDict is a set of additional fields used in standard encryption dictionary.
|
||
type stdEncryptDict struct {
|
||
R int // (Required) A number specifying which revision of the standard security handler shall be used.
|
||
O, U []byte
|
||
OE, UE []byte // R=6
|
||
|
||
P AccessPermissions
|
||
Perms []byte // An encrypted copy of P (16 bytes). Used to verify permissions. R=6
|
||
EncryptMetadata bool // Indicates whether the document-level metadata stream shall be encrypted.
|
||
}
|
||
|
||
// AccessPermissions is a bitmask of access permissions for a PDF file.
|
||
type AccessPermissions uint32
|
||
|
||
const (
|
||
// PermOwner grants all permissions.
|
||
PermOwner = AccessPermissions(math.MaxUint32)
|
||
|
||
PermPrinting = AccessPermissions(1 << 2) // bit 3
|
||
PermModify = AccessPermissions(1 << 3) // bit 4
|
||
PermExtractGraphics = AccessPermissions(1 << 4) // bit 5
|
||
PermAnnotate = AccessPermissions(1 << 5) // bit 6
|
||
// PermFillForms allow form filling, if annotation is disabled? If annotation enabled, is not looked at.
|
||
PermFillForms = AccessPermissions(1 << 8) // bit 9
|
||
PermDisabilityExtract = AccessPermissions(1 << 9) // bit 10 // TODO: not clear what this means!
|
||
// PermRotateInsert allows rotating, editing page order.
|
||
PermRotateInsert = AccessPermissions(1 << 10) // bit 11
|
||
// PermFullPrintQuality limits print quality (lowres), assuming Printing bit is set.
|
||
PermFullPrintQuality = AccessPermissions(1 << 11) // bit 12
|
||
)
|
||
|
||
// Allowed checks if a set of permissions can be granted.
|
||
func (p AccessPermissions) Allowed(p2 AccessPermissions) bool {
|
||
return p&p2 == p2
|
||
}
|
||
|
||
const padding = "\x28\xBF\x4E\x5E\x4E\x75\x8A\x41\x64\x00\x4E\x56\xFF" +
|
||
"\xFA\x01\x08\x2E\x2E\x00\xB6\xD0\x68\x3E\x80\x2F\x0C" +
|
||
"\xA9\xFE\x64\x53\x69\x7A"
|
||
|
||
// StandardCryptFilter is a default name for a standard crypt filter.
|
||
const StandardCryptFilter = "StdCF"
|
||
|
||
func newCryptFiltersV2(length int) CryptFilters {
|
||
return CryptFilters{
|
||
StandardCryptFilter: NewCryptFilterV2(length),
|
||
}
|
||
}
|
||
|
||
// NewCryptFilterV2 creates a RC4-based filter with a specified key length (in bytes).
|
||
func NewCryptFilterV2(length int) CryptFilter {
|
||
f, err := newCryptFilterV2(cryptFilterDict{Length: length})
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
return f
|
||
}
|
||
|
||
// NewCryptFilterAESV2 creates an AES-based filter with a 128 bit key (AESV2).
|
||
func NewCryptFilterAESV2() CryptFilter {
|
||
f, err := newCryptFilterAESV2(cryptFilterDict{})
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
return f
|
||
}
|
||
|
||
// NewCryptFilterAESV3 creates an AES-based filter with a 256 bit key (AESV3).
|
||
func NewCryptFilterAESV3() CryptFilter {
|
||
f, err := newCryptFilterAESV3(cryptFilterDict{})
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
return f
|
||
}
|
||
|
||
// CryptFilters is a map of crypt filter name and underlying CryptFilter info.
|
||
// TODO (v3): Unexport.
|
||
type CryptFilters map[string]CryptFilter
|
||
|
||
// LoadCryptFilters loads crypt filter information from the encryption dictionary (V>=4).
|
||
// TODO (v3): Unexport.
|
||
func (crypt *PdfCrypt) LoadCryptFilters(ed *PdfObjectDictionary) error {
|
||
crypt.CryptFilters = CryptFilters{}
|
||
|
||
obj := ed.Get("CF")
|
||
obj = TraceToDirectObject(obj) // XXX may need to resolve reference...
|
||
if ref, isRef := obj.(*PdfObjectReference); isRef {
|
||
o, err := crypt.parser.LookupByReference(*ref)
|
||
if err != nil {
|
||
common.Log.Debug("Error looking up CF reference")
|
||
return err
|
||
}
|
||
obj = TraceToDirectObject(o)
|
||
}
|
||
|
||
cf, ok := obj.(*PdfObjectDictionary)
|
||
if !ok {
|
||
common.Log.Debug("Invalid CF, type: %T", obj)
|
||
return errors.New("Invalid CF")
|
||
}
|
||
|
||
for _, name := range cf.Keys() {
|
||
v := cf.Get(name)
|
||
|
||
if ref, isRef := v.(*PdfObjectReference); isRef {
|
||
o, err := crypt.parser.LookupByReference(*ref)
|
||
if err != nil {
|
||
common.Log.Debug("Error lookup up dictionary reference")
|
||
return err
|
||
}
|
||
v = TraceToDirectObject(o)
|
||
}
|
||
|
||
dict, ok := v.(*PdfObjectDictionary)
|
||
if !ok {
|
||
return fmt.Errorf("Invalid dict in CF (name %s) - not a dictionary but %T", name, v)
|
||
}
|
||
|
||
if name == "Identity" {
|
||
common.Log.Debug("ERROR - Cannot overwrite the identity filter - Trying next")
|
||
continue
|
||
}
|
||
|
||
var cfd cryptFilterDict
|
||
if err := cfd.ReadFrom(dict); err != nil {
|
||
return err
|
||
}
|
||
fnc, err := getCryptFilterMethod(cfd.CFM)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
cf, err := fnc(cfd)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
crypt.CryptFilters[string(name)] = cf
|
||
}
|
||
// Cannot be overwritten.
|
||
crypt.CryptFilters["Identity"] = cryptFilteridentity{}
|
||
|
||
// StrF strings filter.
|
||
crypt.StringFilter = "Identity"
|
||
if strf, ok := ed.Get("StrF").(*PdfObjectName); ok {
|
||
if _, exists := crypt.CryptFilters[string(*strf)]; !exists {
|
||
return fmt.Errorf("Crypt filter for StrF not specified in CF dictionary (%s)", *strf)
|
||
}
|
||
crypt.StringFilter = string(*strf)
|
||
}
|
||
|
||
// StmF streams filter.
|
||
crypt.StreamFilter = "Identity"
|
||
if stmf, ok := ed.Get("StmF").(*PdfObjectName); ok {
|
||
if _, exists := crypt.CryptFilters[string(*stmf)]; !exists {
|
||
return fmt.Errorf("Crypt filter for StmF not specified in CF dictionary (%s)", *stmf)
|
||
}
|
||
crypt.StreamFilter = string(*stmf)
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// saveCryptFilters saves crypt filter information to the encryption dictionary (V>=4).
|
||
// TODO (v3): Unexport.
|
||
func (crypt *PdfCrypt) saveCryptFilters(ed *PdfObjectDictionary) error {
|
||
if crypt.encrypt.V < 4 {
|
||
return errors.New("can only be used with V>=4")
|
||
}
|
||
cf := MakeDict()
|
||
ed.Set("CF", cf)
|
||
|
||
for name, filter := range crypt.CryptFilters {
|
||
if name == "Identity" {
|
||
continue
|
||
}
|
||
v := cryptFilterToDict(filter, "")
|
||
cf.Set(PdfObjectName(name), v)
|
||
}
|
||
ed.Set("StrF", MakeName(crypt.StringFilter))
|
||
ed.Set("StmF", MakeName(crypt.StreamFilter))
|
||
return nil
|
||
}
|
||
|
||
// PdfCryptNewDecrypt makes the document crypt handler based on the encryption dictionary
|
||
// and trailer dictionary. Returns an error on failure to process.
|
||
func PdfCryptNewDecrypt(parser *PdfParser, ed, trailer *PdfObjectDictionary) (*PdfCrypt, error) {
|
||
crypter := &PdfCrypt{
|
||
Authenticated: false,
|
||
DecryptedObjects: make(map[PdfObject]bool),
|
||
EncryptedObjects: make(map[PdfObject]bool),
|
||
decryptedObjNum: make(map[int]struct{}),
|
||
parser: parser,
|
||
}
|
||
|
||
filter, ok := ed.Get("Filter").(*PdfObjectName)
|
||
if !ok {
|
||
common.Log.Debug("ERROR Crypt dictionary missing required Filter field!")
|
||
return crypter, errors.New("Required crypt field Filter missing")
|
||
}
|
||
if *filter != "Standard" {
|
||
common.Log.Debug("ERROR Unsupported filter (%s)", *filter)
|
||
return crypter, errors.New("Unsupported Filter")
|
||
}
|
||
crypter.encrypt.Filter = string(*filter)
|
||
|
||
if subfilter, ok := ed.Get("SubFilter").(*PdfObjectString); ok {
|
||
crypter.encrypt.SubFilter = subfilter.Str()
|
||
common.Log.Debug("Using subfilter %s", subfilter)
|
||
}
|
||
|
||
if L, ok := ed.Get("Length").(*PdfObjectInteger); ok {
|
||
if (*L % 8) != 0 {
|
||
common.Log.Debug("ERROR Invalid encryption length")
|
||
return crypter, errors.New("Invalid encryption length")
|
||
}
|
||
crypter.encrypt.Length = int(*L)
|
||
} else {
|
||
crypter.encrypt.Length = 40
|
||
}
|
||
|
||
crypter.encrypt.V = 0
|
||
if v, ok := ed.Get("V").(*PdfObjectInteger); ok {
|
||
V := int(*v)
|
||
crypter.encrypt.V = V
|
||
if V >= 1 && V <= 2 {
|
||
// Default algorithm is V2.
|
||
crypter.CryptFilters = newCryptFiltersV2(crypter.encrypt.Length)
|
||
} else if V >= 4 && V <= 5 {
|
||
if err := crypter.LoadCryptFilters(ed); err != nil {
|
||
return crypter, err
|
||
}
|
||
} else {
|
||
common.Log.Debug("ERROR Unsupported encryption algo V = %d", V)
|
||
return crypter, errors.New("Unsupported algorithm")
|
||
}
|
||
}
|
||
|
||
R, ok := ed.Get("R").(*PdfObjectInteger)
|
||
if !ok {
|
||
return crypter, errors.New("Encrypt dictionary missing R")
|
||
}
|
||
// TODO(dennwc): according to spec, R should be validated according to V value
|
||
if *R < 2 || *R > 6 {
|
||
return crypter, fmt.Errorf("Invalid R (%d)", *R)
|
||
}
|
||
crypter.encryptStd.R = int(*R)
|
||
|
||
O, ok := ed.Get("O").(*PdfObjectString)
|
||
if !ok {
|
||
return crypter, errors.New("Encrypt dictionary missing O")
|
||
}
|
||
if crypter.encryptStd.R == 5 || crypter.encryptStd.R == 6 {
|
||
// the spec says =48 bytes, but Acrobat pads them out longer
|
||
if len(O.Str()) < 48 {
|
||
return crypter, fmt.Errorf("Length(O) < 48 (%d)", len(O.Str()))
|
||
}
|
||
} else if len(O.Str()) != 32 {
|
||
return crypter, fmt.Errorf("Length(O) != 32 (%d)", len(O.Str()))
|
||
}
|
||
crypter.encryptStd.O = O.Bytes()
|
||
|
||
U, ok := ed.Get("U").(*PdfObjectString)
|
||
if !ok {
|
||
return crypter, errors.New("Encrypt dictionary missing U")
|
||
}
|
||
if crypter.encryptStd.R == 5 || crypter.encryptStd.R == 6 {
|
||
// the spec says =48 bytes, but Acrobat pads them out longer
|
||
if len(U.Str()) < 48 {
|
||
return crypter, fmt.Errorf("Length(U) < 48 (%d)", len(U.Str()))
|
||
}
|
||
} else if len(U.Str()) != 32 {
|
||
// Strictly this does not cause an error.
|
||
// If O is OK and others then can still read the file.
|
||
common.Log.Debug("Warning: Length(U) != 32 (%d)", len(U.Str()))
|
||
//return crypter, errors.New("Length(U) != 32")
|
||
}
|
||
crypter.encryptStd.U = U.Bytes()
|
||
|
||
if crypter.encryptStd.R >= 5 {
|
||
OE, ok := ed.Get("OE").(*PdfObjectString)
|
||
if !ok {
|
||
return crypter, errors.New("Encrypt dictionary missing OE")
|
||
}
|
||
if len(OE.Str()) != 32 {
|
||
return crypter, fmt.Errorf("Length(OE) != 32 (%d)", len(OE.Str()))
|
||
}
|
||
crypter.encryptStd.OE = OE.Bytes()
|
||
|
||
UE, ok := ed.Get("UE").(*PdfObjectString)
|
||
if !ok {
|
||
return crypter, errors.New("Encrypt dictionary missing UE")
|
||
}
|
||
if len(UE.Str()) != 32 {
|
||
return crypter, fmt.Errorf("Length(UE) != 32 (%d)", len(UE.Str()))
|
||
}
|
||
crypter.encryptStd.UE = UE.Bytes()
|
||
}
|
||
|
||
P, ok := ed.Get("P").(*PdfObjectInteger)
|
||
if !ok {
|
||
return crypter, errors.New("Encrypt dictionary missing permissions attr")
|
||
}
|
||
crypter.encryptStd.P = AccessPermissions(*P)
|
||
|
||
if crypter.encryptStd.R == 6 {
|
||
Perms, ok := ed.Get("Perms").(*PdfObjectString)
|
||
if !ok {
|
||
return crypter, errors.New("Encrypt dictionary missing Perms")
|
||
}
|
||
if len(Perms.Str()) != 16 {
|
||
return crypter, fmt.Errorf("Length(Perms) != 16 (%d)", len(Perms.Str()))
|
||
}
|
||
crypter.encryptStd.Perms = Perms.Bytes()
|
||
}
|
||
|
||
em, ok := ed.Get("EncryptMetadata").(*PdfObjectBool)
|
||
if ok {
|
||
crypter.encryptStd.EncryptMetadata = bool(*em)
|
||
} else {
|
||
crypter.encryptStd.EncryptMetadata = true // True by default.
|
||
}
|
||
|
||
// Default: empty ID.
|
||
// Strictly, if file is encrypted, the ID should always be specified
|
||
// but clearly not everyone is following the specification.
|
||
id0 := ""
|
||
if idArray, ok := trailer.Get("ID").(*PdfObjectArray); ok && idArray.Len() >= 1 {
|
||
id0obj, ok := GetString(idArray.Get(0))
|
||
if !ok {
|
||
return crypter, errors.New("Invalid trailer ID")
|
||
}
|
||
id0 = id0obj.Str()
|
||
} else {
|
||
common.Log.Debug("Trailer ID array missing or invalid!")
|
||
}
|
||
crypter.Id0 = id0
|
||
|
||
return crypter, nil
|
||
}
|
||
|
||
// GetAccessPermissions returns the PDF access permissions as an AccessPermissions object.
|
||
func (crypt *PdfCrypt) GetAccessPermissions() AccessPermissions {
|
||
return crypt.encryptStd.P
|
||
}
|
||
|
||
// Check whether the specified password can be used to decrypt the document.
|
||
func (crypt *PdfCrypt) authenticate(password []byte) (bool, error) {
|
||
// Also build the encryption/decryption key.
|
||
|
||
crypt.Authenticated = false
|
||
if crypt.encryptStd.R >= 5 {
|
||
authenticated, err := crypt.alg2a(password)
|
||
if err != nil {
|
||
return false, err
|
||
}
|
||
crypt.Authenticated = authenticated
|
||
return authenticated, err
|
||
}
|
||
|
||
// Try user password.
|
||
common.Log.Trace("Debugging authentication - user pass")
|
||
authenticated, err := crypt.alg6(password)
|
||
if err != nil {
|
||
return false, err
|
||
}
|
||
if authenticated {
|
||
common.Log.Trace("this.Authenticated = True")
|
||
crypt.Authenticated = true
|
||
return true, nil
|
||
}
|
||
|
||
// Try owner password also.
|
||
// May not be necessary if only want to get all contents.
|
||
// (user pass needs to be known or empty).
|
||
common.Log.Trace("Debugging authentication - owner pass")
|
||
authenticated, err = crypt.alg7(password)
|
||
if err != nil {
|
||
return false, err
|
||
}
|
||
if authenticated {
|
||
common.Log.Trace("this.Authenticated = True")
|
||
crypt.Authenticated = true
|
||
return true, nil
|
||
}
|
||
|
||
return false, nil
|
||
}
|
||
|
||
// Check access rights and permissions for a specified password. If either user/owner password is specified,
|
||
// full rights are granted, otherwise the access rights are specified by the Permissions flag.
|
||
//
|
||
// The bool flag indicates that the user can access and can view the file.
|
||
// The AccessPermissions shows what access the user has for editing etc.
|
||
// An error is returned if there was a problem performing the authentication.
|
||
func (crypt *PdfCrypt) checkAccessRights(password []byte) (bool, AccessPermissions, error) {
|
||
// Try owner password -> full rights.
|
||
var (
|
||
isOwner bool
|
||
err error
|
||
)
|
||
if crypt.encryptStd.R >= 5 {
|
||
var h []byte
|
||
h, err = crypt.alg12(password)
|
||
if err != nil {
|
||
return false, 0, err
|
||
}
|
||
isOwner = len(h) != 0
|
||
} else {
|
||
isOwner, err = crypt.alg7(password)
|
||
}
|
||
if err != nil {
|
||
return false, 0, err
|
||
}
|
||
if isOwner {
|
||
// owner -> full rights.
|
||
return true, PermOwner, nil
|
||
}
|
||
|
||
// Try user password.
|
||
var isUser bool
|
||
if crypt.encryptStd.R >= 5 {
|
||
var h []byte
|
||
h, err = crypt.alg11(password)
|
||
if err != nil {
|
||
return false, 0, err
|
||
}
|
||
isUser = len(h) != 0
|
||
} else {
|
||
isUser, err = crypt.alg6(password)
|
||
}
|
||
if err != nil {
|
||
return false, 0, err
|
||
}
|
||
if isUser {
|
||
// User password specified correctly -> access granted with specified permissions.
|
||
return true, crypt.encryptStd.P, nil
|
||
}
|
||
|
||
// Cannot even view the file.
|
||
return false, 0, nil
|
||
}
|
||
|
||
func (crypt *PdfCrypt) paddedPass(pass []byte) []byte {
|
||
key := make([]byte, 32)
|
||
if len(pass) >= 32 {
|
||
for i := 0; i < 32; i++ {
|
||
key[i] = pass[i]
|
||
}
|
||
} else {
|
||
for i := 0; i < len(pass); i++ {
|
||
key[i] = pass[i]
|
||
}
|
||
for i := len(pass); i < 32; i++ {
|
||
key[i] = padding[i-len(pass)]
|
||
}
|
||
}
|
||
return key
|
||
}
|
||
|
||
// Generates a key for encrypting a specific object based on the
|
||
// object and generation number, as well as the document encryption key.
|
||
func (crypt *PdfCrypt) makeKey(filter string, objNum, genNum uint32, ekey []byte) ([]byte, error) {
|
||
f, ok := crypt.CryptFilters[filter]
|
||
if !ok {
|
||
return nil, fmt.Errorf("Unknown crypt filter (%s)", filter)
|
||
}
|
||
return f.MakeKey(objNum, genNum, ekey)
|
||
}
|
||
|
||
// encryptDictKeys list all required field for "Encrypt" dictionary.
|
||
// It is used as a fingerprint to detect old copies of this dictionary.
|
||
var encryptDictKeys = []PdfObjectName{
|
||
"V", "R", "O", "U", "P",
|
||
}
|
||
|
||
// Check if object has already been processed.
|
||
func (crypt *PdfCrypt) isDecrypted(obj PdfObject) bool {
|
||
_, ok := crypt.DecryptedObjects[obj]
|
||
if ok {
|
||
common.Log.Trace("Already decrypted")
|
||
return true
|
||
}
|
||
switch obj := obj.(type) {
|
||
case *PdfObjectStream:
|
||
if crypt.encryptStd.R != 5 {
|
||
if name, ok := obj.Get("Type").(*PdfObjectName); ok && *name == "XRef" {
|
||
return true // Cross-reference streams should not be encrypted
|
||
}
|
||
}
|
||
case *PdfIndirectObject:
|
||
if _, ok = crypt.decryptedObjNum[int(obj.ObjectNumber)]; ok {
|
||
return true
|
||
}
|
||
switch obj := obj.PdfObject.(type) {
|
||
case *PdfObjectDictionary:
|
||
// detect old copies of "Encrypt" dictionary
|
||
// TODO: find a better way to do it
|
||
ok := true
|
||
for _, key := range encryptDictKeys {
|
||
if obj.Get(key) == nil {
|
||
ok = false
|
||
break
|
||
}
|
||
}
|
||
if ok {
|
||
return true
|
||
}
|
||
}
|
||
}
|
||
|
||
common.Log.Trace("Not decrypted yet")
|
||
return false
|
||
}
|
||
|
||
// Decrypt a buffer with a selected crypt filter.
|
||
func (crypt *PdfCrypt) decryptBytes(buf []byte, filter string, okey []byte) ([]byte, error) {
|
||
common.Log.Trace("Decrypt bytes")
|
||
f, ok := crypt.CryptFilters[filter]
|
||
if !ok {
|
||
return nil, fmt.Errorf("Unknown crypt filter (%s)", filter)
|
||
}
|
||
return f.DecryptBytes(buf, okey)
|
||
}
|
||
|
||
// Decrypt an object with specified key. For numbered objects,
|
||
// the key argument is not used and a new one is generated based
|
||
// on the object and generation number.
|
||
// Traverses through all the subobjects (recursive).
|
||
//
|
||
// Does not look up references.. That should be done prior to calling.
|
||
func (crypt *PdfCrypt) Decrypt(obj PdfObject, parentObjNum, parentGenNum int64) error {
|
||
if crypt.isDecrypted(obj) {
|
||
return nil
|
||
}
|
||
|
||
switch obj := obj.(type) {
|
||
case *PdfIndirectObject:
|
||
crypt.DecryptedObjects[obj] = true
|
||
|
||
common.Log.Trace("Decrypting indirect %d %d obj!", obj.ObjectNumber, obj.GenerationNumber)
|
||
|
||
objNum := obj.ObjectNumber
|
||
genNum := obj.GenerationNumber
|
||
|
||
err := crypt.Decrypt(obj.PdfObject, objNum, genNum)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
case *PdfObjectStream:
|
||
// Mark as decrypted first to avoid recursive issues.
|
||
crypt.DecryptedObjects[obj] = true
|
||
dict := obj.PdfObjectDictionary
|
||
|
||
if crypt.encryptStd.R != 5 {
|
||
if s, ok := dict.Get("Type").(*PdfObjectName); ok && *s == "XRef" {
|
||
return nil // Cross-reference streams should not be encrypted
|
||
}
|
||
}
|
||
|
||
objNum := obj.ObjectNumber
|
||
genNum := obj.GenerationNumber
|
||
common.Log.Trace("Decrypting stream %d %d !", objNum, genNum)
|
||
|
||
// TODO: Check for crypt filter (V4).
|
||
// The Crypt filter shall be the first filter in the Filter array entry.
|
||
|
||
streamFilter := StandardCryptFilter // Default RC4.
|
||
if crypt.encrypt.V >= 4 {
|
||
streamFilter = crypt.StreamFilter
|
||
common.Log.Trace("this.StreamFilter = %s", crypt.StreamFilter)
|
||
|
||
if filters, ok := dict.Get("Filter").(*PdfObjectArray); ok {
|
||
// Crypt filter can only be the first entry.
|
||
if firstFilter, ok := GetName(filters.Get(0)); ok {
|
||
if *firstFilter == "Crypt" {
|
||
// Crypt filter overriding the default.
|
||
// Default option is Identity.
|
||
streamFilter = "Identity"
|
||
|
||
// Check if valid crypt filter specified in the decode params.
|
||
if decodeParams, ok := dict.Get("DecodeParms").(*PdfObjectDictionary); ok {
|
||
if filterName, ok := decodeParams.Get("Name").(*PdfObjectName); ok {
|
||
if _, ok := crypt.CryptFilters[string(*filterName)]; ok {
|
||
common.Log.Trace("Using stream filter %s", *filterName)
|
||
streamFilter = string(*filterName)
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
common.Log.Trace("with %s filter", streamFilter)
|
||
if streamFilter == "Identity" {
|
||
// Identity: pass unchanged.
|
||
return nil
|
||
}
|
||
}
|
||
|
||
err := crypt.Decrypt(dict, objNum, genNum)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
okey, err := crypt.makeKey(streamFilter, uint32(objNum), uint32(genNum), crypt.EncryptionKey)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
obj.Stream, err = crypt.decryptBytes(obj.Stream, streamFilter, okey)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
// Update the length based on the decrypted stream.
|
||
dict.Set("Length", MakeInteger(int64(len(obj.Stream))))
|
||
|
||
return nil
|
||
case *PdfObjectString:
|
||
common.Log.Trace("Decrypting string!")
|
||
|
||
stringFilter := StandardCryptFilter
|
||
if crypt.encrypt.V >= 4 {
|
||
// Currently only support Identity / RC4.
|
||
common.Log.Trace("with %s filter", crypt.StringFilter)
|
||
if crypt.StringFilter == "Identity" {
|
||
// Identity: pass unchanged: No action.
|
||
return nil
|
||
}
|
||
stringFilter = crypt.StringFilter
|
||
}
|
||
|
||
key, err := crypt.makeKey(stringFilter, uint32(parentObjNum), uint32(parentGenNum), crypt.EncryptionKey)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
// Overwrite the encrypted with decrypted string.
|
||
str := obj.Str()
|
||
decrypted := make([]byte, len(str))
|
||
for i := 0; i < len(str); i++ {
|
||
decrypted[i] = str[i]
|
||
}
|
||
common.Log.Trace("Decrypt string: %s : % x", decrypted, decrypted)
|
||
decrypted, err = crypt.decryptBytes(decrypted, stringFilter, key)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
obj.val = string(decrypted)
|
||
|
||
return nil
|
||
case *PdfObjectArray:
|
||
for _, o := range obj.Elements() {
|
||
err := crypt.Decrypt(o, parentObjNum, parentGenNum)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
return nil
|
||
case *PdfObjectDictionary:
|
||
isSig := false
|
||
if t := obj.Get("Type"); t != nil {
|
||
typeStr, ok := t.(*PdfObjectName)
|
||
if ok && *typeStr == "Sig" {
|
||
isSig = true
|
||
}
|
||
}
|
||
for _, keyidx := range obj.Keys() {
|
||
o := obj.Get(keyidx)
|
||
// How can we avoid this check, i.e. implement a more smart
|
||
// traversal system?
|
||
if isSig && string(keyidx) == "Contents" {
|
||
// Leave the Contents of a Signature dictionary.
|
||
continue
|
||
}
|
||
|
||
if string(keyidx) != "Parent" && string(keyidx) != "Prev" && string(keyidx) != "Last" { // Check not needed?
|
||
err := crypt.Decrypt(o, parentObjNum, parentGenNum)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// Check if object has already been processed.
|
||
func (crypt *PdfCrypt) isEncrypted(obj PdfObject) bool {
|
||
_, ok := crypt.EncryptedObjects[obj]
|
||
if ok {
|
||
common.Log.Trace("Already encrypted")
|
||
return true
|
||
}
|
||
|
||
common.Log.Trace("Not encrypted yet")
|
||
return false
|
||
}
|
||
|
||
// Encrypt a buffer with the specified crypt filter and key.
|
||
func (crypt *PdfCrypt) encryptBytes(buf []byte, filter string, okey []byte) ([]byte, error) {
|
||
common.Log.Trace("Encrypt bytes")
|
||
f, ok := crypt.CryptFilters[filter]
|
||
if !ok {
|
||
return nil, fmt.Errorf("Unknown crypt filter (%s)", filter)
|
||
}
|
||
return f.EncryptBytes(buf, okey)
|
||
}
|
||
|
||
// Encrypt an object with specified key. For numbered objects,
|
||
// the key argument is not used and a new one is generated based
|
||
// on the object and generation number.
|
||
// Traverses through all the subobjects (recursive).
|
||
//
|
||
// Does not look up references.. That should be done prior to calling.
|
||
func (crypt *PdfCrypt) Encrypt(obj PdfObject, parentObjNum, parentGenNum int64) error {
|
||
if crypt.isEncrypted(obj) {
|
||
return nil
|
||
}
|
||
switch obj := obj.(type) {
|
||
case *PdfIndirectObject:
|
||
crypt.EncryptedObjects[obj] = true
|
||
|
||
common.Log.Trace("Encrypting indirect %d %d obj!", obj.ObjectNumber, obj.GenerationNumber)
|
||
|
||
objNum := obj.ObjectNumber
|
||
genNum := obj.GenerationNumber
|
||
|
||
err := crypt.Encrypt(obj.PdfObject, objNum, genNum)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
return nil
|
||
case *PdfObjectStream:
|
||
crypt.EncryptedObjects[obj] = true
|
||
dict := obj.PdfObjectDictionary
|
||
|
||
if s, ok := dict.Get("Type").(*PdfObjectName); ok && *s == "XRef" {
|
||
return nil // Cross-reference streams should not be encrypted
|
||
}
|
||
|
||
objNum := obj.ObjectNumber
|
||
genNum := obj.GenerationNumber
|
||
common.Log.Trace("Encrypting stream %d %d !", objNum, genNum)
|
||
|
||
// TODO: Check for crypt filter (V4).
|
||
// The Crypt filter shall be the first filter in the Filter array entry.
|
||
|
||
streamFilter := StandardCryptFilter // Default RC4.
|
||
if crypt.encrypt.V >= 4 {
|
||
// For now. Need to change when we add support for more than
|
||
// Identity / RC4.
|
||
streamFilter = crypt.StreamFilter
|
||
common.Log.Trace("this.StreamFilter = %s", crypt.StreamFilter)
|
||
|
||
if filters, ok := dict.Get("Filter").(*PdfObjectArray); ok {
|
||
// Crypt filter can only be the first entry.
|
||
if firstFilter, ok := GetName(filters.Get(0)); ok {
|
||
if *firstFilter == "Crypt" {
|
||
// Crypt filter overriding the default.
|
||
// Default option is Identity.
|
||
streamFilter = "Identity"
|
||
|
||
// Check if valid crypt filter specified in the decode params.
|
||
if decodeParams, ok := dict.Get("DecodeParms").(*PdfObjectDictionary); ok {
|
||
if filterName, ok := decodeParams.Get("Name").(*PdfObjectName); ok {
|
||
if _, ok := crypt.CryptFilters[string(*filterName)]; ok {
|
||
common.Log.Trace("Using stream filter %s", *filterName)
|
||
streamFilter = string(*filterName)
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
|
||
common.Log.Trace("with %s filter", streamFilter)
|
||
if streamFilter == "Identity" {
|
||
// Identity: pass unchanged.
|
||
return nil
|
||
}
|
||
}
|
||
|
||
err := crypt.Encrypt(obj.PdfObjectDictionary, objNum, genNum)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
okey, err := crypt.makeKey(streamFilter, uint32(objNum), uint32(genNum), crypt.EncryptionKey)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
obj.Stream, err = crypt.encryptBytes(obj.Stream, streamFilter, okey)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
// Update the length based on the encrypted stream.
|
||
dict.Set("Length", MakeInteger(int64(len(obj.Stream))))
|
||
|
||
return nil
|
||
case *PdfObjectString:
|
||
common.Log.Trace("Encrypting string!")
|
||
|
||
stringFilter := StandardCryptFilter
|
||
if crypt.encrypt.V >= 4 {
|
||
common.Log.Trace("with %s filter", crypt.StringFilter)
|
||
if crypt.StringFilter == "Identity" {
|
||
// Identity: pass unchanged: No action.
|
||
return nil
|
||
}
|
||
stringFilter = crypt.StringFilter
|
||
}
|
||
|
||
key, err := crypt.makeKey(stringFilter, uint32(parentObjNum), uint32(parentGenNum), crypt.EncryptionKey)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
|
||
str := obj.Str()
|
||
encrypted := make([]byte, len(str))
|
||
for i := 0; i < len(str); i++ {
|
||
encrypted[i] = str[i]
|
||
}
|
||
common.Log.Trace("Encrypt string: %s : % x", encrypted, encrypted)
|
||
encrypted, err = crypt.encryptBytes(encrypted, stringFilter, key)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
obj.val = string(encrypted)
|
||
|
||
return nil
|
||
case *PdfObjectArray:
|
||
for _, o := range obj.Elements() {
|
||
err := crypt.Encrypt(o, parentObjNum, parentGenNum)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
return nil
|
||
case *PdfObjectDictionary:
|
||
isSig := false
|
||
if t := obj.Get("Type"); t != nil {
|
||
typeStr, ok := t.(*PdfObjectName)
|
||
if ok && *typeStr == "Sig" {
|
||
isSig = true
|
||
}
|
||
}
|
||
|
||
for _, keyidx := range obj.Keys() {
|
||
o := obj.Get(keyidx)
|
||
// How can we avoid this check, i.e. implement a more smart
|
||
// traversal system?
|
||
if isSig && string(keyidx) == "Contents" {
|
||
// Leave the Contents of a Signature dictionary.
|
||
continue
|
||
}
|
||
if string(keyidx) != "Parent" && string(keyidx) != "Prev" && string(keyidx) != "Last" { // Check not needed?
|
||
err := crypt.Encrypt(o, parentObjNum, parentGenNum)
|
||
if err != nil {
|
||
return err
|
||
}
|
||
}
|
||
}
|
||
return nil
|
||
}
|
||
|
||
return nil
|
||
}
|
||
|
||
// aesZeroIV allocates a zero-filled buffer that serves as an initialization vector for AESv3.
|
||
func (crypt *PdfCrypt) aesZeroIV() []byte {
|
||
if crypt.ivAESZero == nil {
|
||
crypt.ivAESZero = make([]byte, aes.BlockSize)
|
||
}
|
||
return crypt.ivAESZero
|
||
}
|
||
|
||
// alg2a retrieves the encryption key from an encrypted document (R >= 5).
|
||
// It returns false if the password was wrong.
|
||
// 7.6.4.3.2 Algorithm 2.A (page 83)
|
||
func (crypt *PdfCrypt) alg2a(pass []byte) (bool, error) {
|
||
// O & U: 32 byte hash + 8 byte Validation Salt + 8 byte Key Salt
|
||
|
||
// step a: Unicode normalization
|
||
// TODO(dennwc): make sure that UTF-8 strings are normalized
|
||
|
||
// step b: truncate to 127 bytes
|
||
if len(pass) > 127 {
|
||
pass = pass[:127]
|
||
}
|
||
|
||
// step c: test pass against the owner key
|
||
h, err := crypt.alg12(pass)
|
||
if err != nil {
|
||
return false, err
|
||
}
|
||
var (
|
||
data []byte // data to hash
|
||
ekey []byte // encrypted file key
|
||
ukey []byte // user key; set only when using owner's password
|
||
)
|
||
if len(h) != 0 {
|
||
// owner password valid
|
||
|
||
// step d: compute an intermediate owner key
|
||
str := make([]byte, len(pass)+8+48)
|
||
i := copy(str, pass)
|
||
i += copy(str[i:], crypt.encryptStd.O[40:48]) // owner Key Salt
|
||
i += copy(str[i:], crypt.encryptStd.U[0:48])
|
||
|
||
data = str
|
||
ekey = crypt.encryptStd.OE
|
||
ukey = crypt.encryptStd.U[0:48]
|
||
} else {
|
||
// check user password
|
||
h, err = crypt.alg11(pass)
|
||
if err == nil && len(h) == 0 {
|
||
// try default password
|
||
h, err = crypt.alg11([]byte(""))
|
||
}
|
||
if err != nil {
|
||
return false, err
|
||
} else if len(h) == 0 {
|
||
// wrong password
|
||
return false, nil
|
||
}
|
||
// step e: compute an intermediate user key
|
||
str := make([]byte, len(pass)+8)
|
||
i := copy(str, pass)
|
||
i += copy(str[i:], crypt.encryptStd.U[40:48]) // user Key Salt
|
||
|
||
data = str
|
||
ekey = crypt.encryptStd.UE
|
||
ukey = nil
|
||
}
|
||
ekey = ekey[:32]
|
||
|
||
// intermediate key
|
||
ikey := crypt.alg2b(data, pass, ukey)
|
||
|
||
ac, err := aes.NewCipher(ikey[:32])
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
|
||
iv := crypt.aesZeroIV()
|
||
cbc := cipher.NewCBCDecrypter(ac, iv)
|
||
fkey := make([]byte, 32)
|
||
cbc.CryptBlocks(fkey, ekey)
|
||
|
||
crypt.EncryptionKey = fkey
|
||
|
||
if crypt.encryptStd.R == 5 {
|
||
return true, nil
|
||
}
|
||
|
||
return crypt.alg13(fkey)
|
||
}
|
||
|
||
// alg2b computes a hash for R=5 and R=6.
|
||
func (crypt *PdfCrypt) alg2b(data, pwd, userKey []byte) []byte {
|
||
if crypt.encryptStd.R == 5 {
|
||
return alg2b_R5(data)
|
||
}
|
||
return alg2b(data, pwd, userKey)
|
||
}
|
||
|
||
// alg2b_R5 computes a hash for R=5, used in a deprecated extension.
|
||
// It's used the same way as a hash described in Algorithm 2.B, but it doesn't use the original password
|
||
// and the user key to calculate the hash.
|
||
func alg2b_R5(data []byte) []byte {
|
||
h := sha256.New()
|
||
h.Write(data)
|
||
return h.Sum(nil)
|
||
}
|
||
|
||
// repeat repeats first n bytes of buf until the end of the buffer.
|
||
// It assumes that the length of buf is a multiple of n.
|
||
func repeat(buf []byte, n int) {
|
||
bp := n
|
||
for bp < len(buf) {
|
||
copy(buf[bp:], buf[:bp])
|
||
bp *= 2
|
||
}
|
||
}
|
||
|
||
// alg2b computes a hash for R=6.
|
||
// 7.6.4.3.3 Algorithm 2.B (page 83)
|
||
func alg2b(data, pwd, userKey []byte) []byte {
|
||
var (
|
||
s256, s384, s512 hash.Hash
|
||
)
|
||
s256 = sha256.New()
|
||
hbuf := make([]byte, 64)
|
||
|
||
h := s256
|
||
h.Write(data)
|
||
K := h.Sum(hbuf[:0])
|
||
|
||
buf := make([]byte, 64*(127+64+48))
|
||
|
||
round := func(rnd int) (E []byte) {
|
||
// step a: repeat pass+K 64 times
|
||
n := len(pwd) + len(K) + len(userKey)
|
||
part := buf[:n]
|
||
i := copy(part, pwd)
|
||
i += copy(part[i:], K[:])
|
||
i += copy(part[i:], userKey)
|
||
if i != n {
|
||
panic("wrong size")
|
||
}
|
||
K1 := buf[:n*64]
|
||
repeat(K1, n)
|
||
|
||
// step b: encrypt K1 with AES-128 CBC
|
||
ac, err := aes.NewCipher(K[0:16])
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
cbc := cipher.NewCBCEncrypter(ac, K[16:32])
|
||
cbc.CryptBlocks(K1, K1)
|
||
E = K1
|
||
|
||
// step c: use 16 bytes of E as big-endian int, select the next hash
|
||
b := 0
|
||
for i := 0; i < 16; i++ {
|
||
b += int(E[i] % 3)
|
||
}
|
||
var h hash.Hash
|
||
switch b % 3 {
|
||
case 0:
|
||
h = s256
|
||
case 1:
|
||
if s384 == nil {
|
||
s384 = sha512.New384()
|
||
}
|
||
h = s384
|
||
case 2:
|
||
if s512 == nil {
|
||
s512 = sha512.New()
|
||
}
|
||
h = s512
|
||
}
|
||
|
||
// step d: take the hash of E, use as a new K
|
||
h.Reset()
|
||
h.Write(E)
|
||
K = h.Sum(hbuf[:0])
|
||
|
||
return E
|
||
}
|
||
|
||
for i := 0; ; {
|
||
E := round(i)
|
||
b := uint8(E[len(E)-1])
|
||
// from the spec, it appears that i should be incremented after
|
||
// the test, but that doesn't match what Adobe does
|
||
i++
|
||
if i >= 64 && b <= uint8(i-32) {
|
||
break
|
||
}
|
||
}
|
||
return K[:32]
|
||
}
|
||
|
||
// alg2 computes an encryption key.
|
||
func (crypt *PdfCrypt) alg2(pass []byte) []byte {
|
||
common.Log.Trace("alg2")
|
||
key := crypt.paddedPass(pass)
|
||
|
||
h := md5.New()
|
||
h.Write(key)
|
||
|
||
// Pass O.
|
||
h.Write(crypt.encryptStd.O)
|
||
|
||
// Pass P (Lower order byte first).
|
||
var p = uint32(crypt.encryptStd.P)
|
||
var pb []byte
|
||
for i := 0; i < 4; i++ {
|
||
pb = append(pb, byte(((p >> uint(8*i)) & 0xff)))
|
||
}
|
||
h.Write(pb)
|
||
common.Log.Trace("go P: % x", pb)
|
||
|
||
// Pass ID[0] from the trailer
|
||
h.Write([]byte(crypt.Id0))
|
||
|
||
common.Log.Trace("this.R = %d encryptMetadata %v", crypt.encryptStd.R, crypt.encryptStd.EncryptMetadata)
|
||
if (crypt.encryptStd.R >= 4) && !crypt.encryptStd.EncryptMetadata {
|
||
h.Write([]byte{0xff, 0xff, 0xff, 0xff})
|
||
}
|
||
hashb := h.Sum(nil)
|
||
|
||
if crypt.encryptStd.R >= 3 {
|
||
for i := 0; i < 50; i++ {
|
||
h = md5.New()
|
||
h.Write(hashb[0 : crypt.encrypt.Length/8])
|
||
hashb = h.Sum(nil)
|
||
}
|
||
}
|
||
|
||
if crypt.encryptStd.R >= 3 {
|
||
return hashb[0 : crypt.encrypt.Length/8]
|
||
}
|
||
|
||
return hashb[0:5]
|
||
}
|
||
|
||
// Create the RC4 encryption key.
|
||
func (crypt *PdfCrypt) alg3Key(pass []byte) []byte {
|
||
h := md5.New()
|
||
okey := crypt.paddedPass(pass)
|
||
h.Write(okey)
|
||
|
||
if crypt.encryptStd.R >= 3 {
|
||
for i := 0; i < 50; i++ {
|
||
hashb := h.Sum(nil)
|
||
h = md5.New()
|
||
h.Write(hashb)
|
||
}
|
||
}
|
||
|
||
encKey := h.Sum(nil)
|
||
if crypt.encryptStd.R == 2 {
|
||
encKey = encKey[0:5]
|
||
} else {
|
||
encKey = encKey[0 : crypt.encrypt.Length/8]
|
||
}
|
||
return encKey
|
||
}
|
||
|
||
// Alg3 computes the encryption dictionary’s O (owner password) value.
|
||
func (crypt *PdfCrypt) Alg3(upass, opass []byte) (string, error) {
|
||
// Return O string val.
|
||
O := ""
|
||
|
||
var encKey []byte
|
||
if len(opass) > 0 {
|
||
encKey = crypt.alg3Key(opass)
|
||
} else {
|
||
encKey = crypt.alg3Key(upass)
|
||
}
|
||
|
||
ociph, err := rc4.NewCipher(encKey)
|
||
if err != nil {
|
||
return O, errors.New("Failed rc4 ciph")
|
||
}
|
||
|
||
ukey := crypt.paddedPass(upass)
|
||
encrypted := make([]byte, len(ukey))
|
||
ociph.XORKeyStream(encrypted, ukey)
|
||
|
||
if crypt.encryptStd.R >= 3 {
|
||
encKey2 := make([]byte, len(encKey))
|
||
for i := 0; i < 19; i++ {
|
||
for j := 0; j < len(encKey); j++ {
|
||
encKey2[j] = encKey[j] ^ byte(i+1)
|
||
}
|
||
ciph, err := rc4.NewCipher(encKey2)
|
||
if err != nil {
|
||
return O, errors.New("Failed rc4 ciph")
|
||
}
|
||
ciph.XORKeyStream(encrypted, encrypted)
|
||
}
|
||
}
|
||
|
||
O = string(encrypted)
|
||
return O, nil
|
||
}
|
||
|
||
// alg4 computes the encryption dictionary’s U (user password) value (Security handlers of revision 2).
|
||
func (crypt *PdfCrypt) alg4(upass []byte) (string, []byte, error) {
|
||
U := ""
|
||
|
||
ekey := crypt.alg2(upass)
|
||
ciph, err := rc4.NewCipher(ekey)
|
||
if err != nil {
|
||
return U, ekey, errors.New("Failed rc4 ciph")
|
||
}
|
||
|
||
s := []byte(padding)
|
||
encrypted := make([]byte, len(s))
|
||
ciph.XORKeyStream(encrypted, s)
|
||
|
||
U = string(encrypted)
|
||
return U, ekey, nil
|
||
}
|
||
|
||
// Alg5 computes the encryption dictionary’s U (user password) value (Security handlers of revision 3 or greater).
|
||
func (crypt *PdfCrypt) Alg5(upass []byte) (string, []byte, error) {
|
||
U := ""
|
||
|
||
ekey := crypt.alg2(upass)
|
||
|
||
h := md5.New()
|
||
h.Write([]byte(padding))
|
||
h.Write([]byte(crypt.Id0))
|
||
hash := h.Sum(nil)
|
||
|
||
common.Log.Trace("Alg5")
|
||
common.Log.Trace("ekey: % x", ekey)
|
||
common.Log.Trace("ID: % x", crypt.Id0)
|
||
|
||
if len(hash) != 16 {
|
||
return U, ekey, errors.New("Hash length not 16 bytes")
|
||
}
|
||
|
||
ciph, err := rc4.NewCipher(ekey)
|
||
if err != nil {
|
||
return U, ekey, errors.New("Failed rc4 ciph")
|
||
}
|
||
encrypted := make([]byte, 16)
|
||
ciph.XORKeyStream(encrypted, hash)
|
||
|
||
// Do the following 19 times: Take the output from the previous
|
||
// invocation of the RC4 function and pass it as input to a new
|
||
// invocation of the function; use an encryption key generated by
|
||
// taking each byte of the original encryption key obtained in step
|
||
// (a) and performing an XOR (exclusive or) operation between that
|
||
// byte and the single-byte value of the iteration counter (from 1 to 19).
|
||
ekey2 := make([]byte, len(ekey))
|
||
for i := 0; i < 19; i++ {
|
||
for j := 0; j < len(ekey); j++ {
|
||
ekey2[j] = ekey[j] ^ byte(i+1)
|
||
}
|
||
ciph, err = rc4.NewCipher(ekey2)
|
||
if err != nil {
|
||
return U, ekey, errors.New("Failed rc4 ciph")
|
||
}
|
||
ciph.XORKeyStream(encrypted, encrypted)
|
||
common.Log.Trace("i = %d, ekey: % x", i, ekey2)
|
||
common.Log.Trace("i = %d -> % x", i, encrypted)
|
||
}
|
||
|
||
bb := make([]byte, 32)
|
||
for i := 0; i < 16; i++ {
|
||
bb[i] = encrypted[i]
|
||
}
|
||
|
||
// Append 16 bytes of arbitrary padding to the output from the final
|
||
// invocation of the RC4 function and store the 32-byte result as
|
||
// the value of the U entry in the encryption dictionary.
|
||
_, err = rand.Read(bb[16:32])
|
||
if err != nil {
|
||
return U, ekey, errors.New("Failed to gen rand number")
|
||
}
|
||
|
||
U = string(bb)
|
||
return U, ekey, nil
|
||
}
|
||
|
||
// alg6 authenticates the user password.
|
||
func (crypt *PdfCrypt) alg6(upass []byte) (bool, error) {
|
||
var uo string
|
||
var err error
|
||
var key []byte
|
||
if crypt.encryptStd.R == 2 {
|
||
uo, key, err = crypt.alg4(upass)
|
||
} else if crypt.encryptStd.R >= 3 {
|
||
uo, key, err = crypt.Alg5(upass)
|
||
} else {
|
||
return false, errors.New("invalid R")
|
||
}
|
||
|
||
if err != nil {
|
||
return false, err
|
||
}
|
||
|
||
common.Log.Trace("check: % x == % x ?", string(uo), string(crypt.encryptStd.U))
|
||
|
||
uGen := string(uo) // Generated U from specified pass.
|
||
uDoc := string(crypt.encryptStd.U) // U from the document.
|
||
if crypt.encryptStd.R >= 3 {
|
||
// comparing on the first 16 bytes in the case of security
|
||
// handlers of revision 3 or greater),
|
||
if len(uGen) > 16 {
|
||
uGen = uGen[0:16]
|
||
}
|
||
if len(uDoc) > 16 {
|
||
uDoc = uDoc[0:16]
|
||
}
|
||
}
|
||
|
||
if uGen == uDoc {
|
||
crypt.EncryptionKey = key
|
||
return true, nil
|
||
}
|
||
|
||
return false, nil
|
||
}
|
||
|
||
// alg7 authenticates the owner password.
|
||
func (crypt *PdfCrypt) alg7(opass []byte) (bool, error) {
|
||
encKey := crypt.alg3Key(opass)
|
||
|
||
decrypted := make([]byte, len(crypt.encryptStd.O))
|
||
if crypt.encryptStd.R == 2 {
|
||
ciph, err := rc4.NewCipher(encKey)
|
||
if err != nil {
|
||
return false, errors.New("Failed cipher")
|
||
}
|
||
ciph.XORKeyStream(decrypted, crypt.encryptStd.O)
|
||
} else if crypt.encryptStd.R >= 3 {
|
||
s := append([]byte{}, crypt.encryptStd.O...)
|
||
for i := 0; i < 20; i++ {
|
||
//newKey := encKey
|
||
newKey := append([]byte{}, encKey...)
|
||
for j := 0; j < len(encKey); j++ {
|
||
newKey[j] ^= byte(19 - i)
|
||
}
|
||
ciph, err := rc4.NewCipher(newKey)
|
||
if err != nil {
|
||
return false, errors.New("Failed cipher")
|
||
}
|
||
ciph.XORKeyStream(decrypted, s)
|
||
s = append([]byte{}, decrypted...)
|
||
}
|
||
} else {
|
||
return false, errors.New("invalid R")
|
||
}
|
||
|
||
auth, err := crypt.alg6(decrypted)
|
||
if err != nil {
|
||
return false, nil
|
||
}
|
||
|
||
return auth, nil
|
||
}
|
||
|
||
// GenerateParams generates encryption parameters for specified passwords.
|
||
// Can be called only for R>=5.
|
||
func (crypt *PdfCrypt) GenerateParams(upass, opass []byte) error {
|
||
if crypt.encryptStd.R < 5 {
|
||
// TODO(dennwc): move code for R<5 from PdfWriter.Encrypt
|
||
return errors.New("can be used only for R>=5")
|
||
}
|
||
crypt.EncryptionKey = make([]byte, 32)
|
||
if _, err := io.ReadFull(rand.Reader, crypt.EncryptionKey); err != nil {
|
||
return err
|
||
}
|
||
return crypt.generateR6(upass, opass)
|
||
}
|
||
|
||
// generateR6 is the algorithm opposite to alg2a (R>=5).
|
||
// It generates U,O,UE,OE,Perms fields using AESv3 encryption.
|
||
// There is no algorithm number assigned to this function in the spec.
|
||
func (crypt *PdfCrypt) generateR6(upass, opass []byte) error {
|
||
// all these field will be populated by functions below
|
||
crypt.encryptStd.U = nil
|
||
crypt.encryptStd.O = nil
|
||
crypt.encryptStd.UE = nil
|
||
crypt.encryptStd.OE = nil
|
||
crypt.encryptStd.Perms = nil // populated only for R=6
|
||
|
||
if len(upass) > 127 {
|
||
upass = upass[:127]
|
||
}
|
||
if len(opass) > 127 {
|
||
opass = opass[:127]
|
||
}
|
||
// generate U and UE
|
||
if err := crypt.alg8(upass); err != nil {
|
||
return err
|
||
}
|
||
// generate O and OE
|
||
if err := crypt.alg9(opass); err != nil {
|
||
return err
|
||
}
|
||
if crypt.encryptStd.R == 5 {
|
||
return nil
|
||
}
|
||
// generate Perms
|
||
return crypt.alg10()
|
||
}
|
||
|
||
// alg8 computes the encryption dictionary's U (user password) and UE (user encryption) values (R>=5).
|
||
// 7.6.4.4.6 Algorithm 8 (page 86)
|
||
func (crypt *PdfCrypt) alg8(upass []byte) error {
|
||
// step a: compute U (user password)
|
||
var rbuf [16]byte
|
||
if _, err := io.ReadFull(rand.Reader, rbuf[:]); err != nil {
|
||
return err
|
||
}
|
||
valSalt := rbuf[0:8]
|
||
keySalt := rbuf[8:16]
|
||
|
||
str := make([]byte, len(upass)+len(valSalt))
|
||
i := copy(str, upass)
|
||
i += copy(str[i:], valSalt)
|
||
|
||
h := crypt.alg2b(str, upass, nil)
|
||
|
||
U := make([]byte, len(h)+len(valSalt)+len(keySalt))
|
||
i = copy(U, h[:32])
|
||
i += copy(U[i:], valSalt)
|
||
i += copy(U[i:], keySalt)
|
||
|
||
crypt.encryptStd.U = U
|
||
|
||
// step b: compute UE (user encryption)
|
||
|
||
// str still contains a password, reuse it
|
||
i = len(upass)
|
||
i += copy(str[i:], keySalt)
|
||
|
||
h = crypt.alg2b(str, upass, nil)
|
||
|
||
ac, err := aes.NewCipher(h[:32])
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
|
||
iv := crypt.aesZeroIV()
|
||
cbc := cipher.NewCBCEncrypter(ac, iv)
|
||
UE := make([]byte, 32)
|
||
cbc.CryptBlocks(UE, crypt.EncryptionKey[:32])
|
||
crypt.encryptStd.UE = UE
|
||
|
||
return nil
|
||
}
|
||
|
||
// alg9 computes the encryption dictionary's O (owner password) and OE (owner encryption) values (R>=5).
|
||
// 7.6.4.4.7 Algorithm 9 (page 86)
|
||
func (crypt *PdfCrypt) alg9(opass []byte) error {
|
||
// step a: compute O (owner password)
|
||
var rbuf [16]byte
|
||
if _, err := io.ReadFull(rand.Reader, rbuf[:]); err != nil {
|
||
return err
|
||
}
|
||
valSalt := rbuf[0:8]
|
||
keySalt := rbuf[8:16]
|
||
userKey := crypt.encryptStd.U[:48]
|
||
|
||
str := make([]byte, len(opass)+len(valSalt)+len(userKey))
|
||
i := copy(str, opass)
|
||
i += copy(str[i:], valSalt)
|
||
i += copy(str[i:], userKey)
|
||
|
||
h := crypt.alg2b(str, opass, userKey)
|
||
|
||
O := make([]byte, len(h)+len(valSalt)+len(keySalt))
|
||
i = copy(O, h[:32])
|
||
i += copy(O[i:], valSalt)
|
||
i += copy(O[i:], keySalt)
|
||
|
||
crypt.encryptStd.O = O
|
||
|
||
// step b: compute OE (owner encryption)
|
||
|
||
// str still contains a password and a user key - reuse both, but overwrite the salt
|
||
i = len(opass)
|
||
i += copy(str[i:], keySalt)
|
||
// i += len(userKey)
|
||
|
||
h = crypt.alg2b(str, opass, userKey)
|
||
|
||
ac, err := aes.NewCipher(h[:32])
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
|
||
iv := crypt.aesZeroIV()
|
||
cbc := cipher.NewCBCEncrypter(ac, iv)
|
||
OE := make([]byte, 32)
|
||
cbc.CryptBlocks(OE, crypt.EncryptionKey[:32])
|
||
crypt.encryptStd.OE = OE
|
||
|
||
return nil
|
||
}
|
||
|
||
// alg10 computes the encryption dictionary's Perms (permissions) value (R=6).
|
||
// 7.6.4.4.8 Algorithm 10 (page 87)
|
||
func (crypt *PdfCrypt) alg10() error {
|
||
// step a: extend permissions to 64 bits
|
||
perms := uint64(uint32(crypt.encryptStd.P)) | (math.MaxUint32 << 32)
|
||
|
||
// step b: record permissions
|
||
Perms := make([]byte, 16)
|
||
binary.LittleEndian.PutUint64(Perms[:8], perms)
|
||
|
||
// step c: record EncryptMetadata
|
||
if crypt.encryptStd.EncryptMetadata {
|
||
Perms[8] = 'T'
|
||
} else {
|
||
Perms[8] = 'F'
|
||
}
|
||
|
||
// step d: write "adb" magic
|
||
copy(Perms[9:12], "adb")
|
||
|
||
// step e: write 4 bytes of random data
|
||
|
||
// spec doesn't specify them as generated "from a strong random source",
|
||
// but we will use the cryptographic random generator anyway
|
||
if _, err := io.ReadFull(rand.Reader, Perms[12:16]); err != nil {
|
||
return err
|
||
}
|
||
|
||
// step f: encrypt permissions
|
||
ac, err := aes.NewCipher(crypt.EncryptionKey[:32])
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
|
||
ecb := newECBEncrypter(ac)
|
||
ecb.CryptBlocks(Perms, Perms)
|
||
|
||
crypt.encryptStd.Perms = Perms[:16]
|
||
return nil
|
||
}
|
||
|
||
// alg11 authenticates the user password (R >= 5) and returns the hash.
|
||
func (crypt *PdfCrypt) alg11(upass []byte) ([]byte, error) {
|
||
str := make([]byte, len(upass)+8)
|
||
i := copy(str, upass)
|
||
i += copy(str[i:], crypt.encryptStd.U[32:40]) // user Validation Salt
|
||
|
||
h := crypt.alg2b(str, upass, nil)
|
||
h = h[:32]
|
||
if !bytes.Equal(h, crypt.encryptStd.U[:32]) {
|
||
return nil, nil
|
||
}
|
||
return h, nil
|
||
}
|
||
|
||
// alg12 authenticates the owner password (R >= 5) and returns the hash.
|
||
// 7.6.4.4.10 Algorithm 12 (page 87)
|
||
func (crypt *PdfCrypt) alg12(opass []byte) ([]byte, error) {
|
||
str := make([]byte, len(opass)+8+48)
|
||
i := copy(str, opass)
|
||
i += copy(str[i:], crypt.encryptStd.O[32:40]) // owner Validation Salt
|
||
i += copy(str[i:], crypt.encryptStd.U[0:48])
|
||
|
||
h := crypt.alg2b(str, opass, crypt.encryptStd.U[0:48])
|
||
h = h[:32]
|
||
if !bytes.Equal(h, crypt.encryptStd.O[:32]) {
|
||
return nil, nil
|
||
}
|
||
return h, nil
|
||
}
|
||
|
||
// alg13 validates user permissions (P+EncryptMetadata vs Perms) for R=6.
|
||
// 7.6.4.4.11 Algorithm 13 (page 87)
|
||
func (crypt *PdfCrypt) alg13(fkey []byte) (bool, error) {
|
||
perms := make([]byte, 16)
|
||
copy(perms, crypt.encryptStd.Perms[:16])
|
||
|
||
ac, err := aes.NewCipher(fkey[:32])
|
||
if err != nil {
|
||
panic(err)
|
||
}
|
||
|
||
ecb := newECBDecrypter(ac)
|
||
ecb.CryptBlocks(perms, perms)
|
||
|
||
if !bytes.Equal(perms[9:12], []byte("adb")) {
|
||
return false, errors.New("decoded permissions are invalid")
|
||
}
|
||
p := AccessPermissions(binary.LittleEndian.Uint32(perms[0:4]))
|
||
if p != crypt.encryptStd.P {
|
||
return false, errors.New("permissions validation failed")
|
||
}
|
||
encMeta := true
|
||
if perms[8] == 'T' {
|
||
encMeta = true
|
||
} else if perms[8] == 'F' {
|
||
encMeta = false
|
||
} else {
|
||
return false, errors.New("decoded metadata encryption flag is invalid")
|
||
}
|
||
if encMeta != crypt.encryptStd.EncryptMetadata {
|
||
return false, errors.New("metadata encryption validation failed")
|
||
}
|
||
return true, nil
|
||
}
|