OrgGo/unipdf
1
0
Fork 0
mirror of https://github.com/unidoc/unipdf.git synced 2025-05-02 22:17:06 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

core: refactor encryption check

Browse Source
This commit is contained in:
Denys Smirnov 2018-09-13 22:37:53 +03:00
parent 72a6c7640b
commit 443efd4c4c
1 changed files with 33 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

66
pdf/core/parser.go
View File

@ -1548,42 +1548,42 @@ func NewParser(rs io.ReadSeeker) (*PdfParser, error) {
func (parser *PdfParser) IsEncrypted() (bool, error) { func (parser *PdfParser) IsEncrypted() (bool, error) {
if parser.crypter != nil { if parser.crypter != nil {
return true, nil return true, nil
} else if parser.trailer == nil {
return false, nil
} }
if parser.trailer != nil { common.Log.Trace("Checking encryption dictionary!")
common.Log.Trace("Checking encryption dictionary!") encDictRef, isEncrypted := parser.trailer.Get("Encrypt").(*PdfObjectReference)
encDictRef, isEncrypted := parser.trailer.Get("Encrypt").(*PdfObjectReference) if !isEncrypted {
if isEncrypted { return false, nil
common.Log.Trace("Is encrypted!")
common.Log.Trace("0: Look up ref %q", encDictRef)
encObj, err := parser.LookupByReference(*encDictRef)
common.Log.Trace("1: %q", encObj)
if err != nil {
return false, err
}
encIndObj, ok := encObj.(*PdfIndirectObject)
if !ok {
common.Log.Debug("Encryption object not an indirect object")
return false, errors.New("Type check error")
}
encDict, ok := encIndObj.PdfObject.(*PdfObjectDictionary)
common.Log.Trace("2: %q", encDict)
if !ok {
return false, errors.New("Trailer Encrypt object non dictionary")
}
crypter, err := PdfCryptMakeNew(parser, encDict, parser.trailer)
if err != nil {
return false, err
}
parser.crypter = &crypter
common.Log.Trace("Crypter object %b", crypter)
return true, nil
}
} }
return false, nil common.Log.Trace("Is encrypted!")
common.Log.Trace("0: Look up ref %q", encDictRef)
encObj, err := parser.LookupByReference(*encDictRef)
common.Log.Trace("1: %q", encObj)
if err != nil {
return false, err
}
encIndObj, ok := encObj.(*PdfIndirectObject)
if !ok {
common.Log.Debug("Encryption object not an indirect object")
return false, errors.New("Type check error")
}
encDict, ok := encIndObj.PdfObject.(*PdfObjectDictionary)
common.Log.Trace("2: %q", encDict)
if !ok {
return false, errors.New("Trailer Encrypt object non dictionary")
}
crypter, err := PdfCryptMakeNew(parser, encDict, parser.trailer)
if err != nil {
return false, err
}
parser.crypter = &crypter
common.Log.Trace("Crypter object %b", crypter)
return true, nil
} }
// Decrypt attempts to decrypt the PDF file with a specified password. Also tries to // Decrypt attempts to decrypt the PDF file with a specified password. Also tries to