add public function

signintech · Oct 7, 2016 · 9a0d6dc · 9a0d6dc
1 parent b3f9279
commit 9a0d6dc
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 3 deletions.
diff --git a/encryption_obj.go b/encryption_obj.go
@@ -46,3 +46,13 @@ func (e *EncryptionObj) escape(b []byte) string {
 	s = strings.Replace(s, "\r", "\\r", -1)
 	return s
 }
+
+//GetObjBuff get buffer
+func (e *EncryptionObj) GetObjBuff() *bytes.Buffer {
+	return e.getObjBuff()
+}
+
+//Build build buffer
+func (e *EncryptionObj) Build(objID int) error {
+	return e.build(objID)
+}
diff --git a/image_obj.go b/image_obj.go
@@ -23,7 +23,7 @@ type ImageObj struct {
 }
 
 func (i *ImageObj) init(funcGetRoot func() *GoPdf) {
-	//i.getRoot = funcGetRoot
+
 }
 
 func (i *ImageObj) setProtection(p *PDFProtection) {

diff --git a/pdf_protection.go b/pdf_protection.go
@@ -35,15 +35,20 @@ type PDFProtection struct {
 	encryptionKey []byte
 }
 
+//SetProtection set protection infomation
+func (p *PDFProtection) SetProtection(permissions int, userPass []byte, ownerPass []byte) error {
+	return p.setProtection(permissions, userPass, ownerPass)
+}
+
 func (p *PDFProtection) setProtection(permissions int, userPass []byte, ownerPass []byte) error {
 	protection := 192 | permissions
 	if ownerPass == nil || len(ownerPass) == 0 {
 		ownerPass = p.randomPass(24)
 	}
-	return p.generateencryptionkey(userPass, ownerPass, protection)
+	return p.generateEncryptionKey(userPass, ownerPass, protection)
 }
 
-func (p *PDFProtection) generateencryptionkey(userPass []byte, ownerPass []byte, protection int) error {
+func (p *PDFProtection) generateEncryptionKey(userPass []byte, ownerPass []byte, protection int) error {
 
 	userPass = append(userPass, protectionPadding...)
 	userPassWithPadding := userPass[0:32]
@@ -67,6 +72,11 @@ func (p *PDFProtection) generateencryptionkey(userPass []byte, ownerPass []byte,
 	return nil
 }
 
+//EncryptionObj get Encryption Object
+func (p *PDFProtection) EncryptionObj() *EncryptionObj {
+	return p.encryptionObj()
+}
+
 func (p *PDFProtection) encryptionObj() *EncryptionObj {
 	var en EncryptionObj
 	en.oValue = p.oValue
@@ -118,6 +128,11 @@ func (p *PDFProtection) randomPass(strlen int) []byte {
 	return result
 }
 
+//Objectkey create object key from ObjID
+func (p *PDFProtection) Objectkey(objID int) []byte {
+	return p.objectkey(objID)
+}
+
 func (p *PDFProtection) objectkey(n int) []byte {
 	tmp := make([]byte, 8, 8)
 	binary.LittleEndian.PutUint32(tmp, uint32(n))
-Original file line number
+Diff line change
@@ Expand Up / @@ -23,7 +23,7 @@ type ImageObj struct { @@
     }
     func (i *ImageObj) init(funcGetRoot func() *GoPdf) {
-    	//i.getRoot = funcGetRoot
     }
     func (i *ImageObj) setProtection(p *PDFProtection) {
@@ Expand Down @@