Add IsEncrypted() method to File struct

go.mod

@@ -1,4 +1,4 @@
-module github.com/bodgit/sevenzip
+module github.com/todylcom/sevenzip
 
 go 1.17
 

internal/bcj2/reader.go

@@ -6,7 +6,7 @@ import (
 	"errors"
 	"io"
 
-	"github.com/bodgit/sevenzip/internal/util"
+	"github.com/todylcom/sevenzip/internal/util"
 	"github.com/hashicorp/go-multierror"
 )
 

internal/deflate/reader.go

@@ -5,7 +5,7 @@ import (
 	"io"
 	"sync"
 
-	"github.com/bodgit/sevenzip/internal/util"
+	"github.com/todylcom/sevenzip/internal/util"
 	"github.com/klauspost/compress/flate"
 )
 

internal/pool/pool.go

@@ -6,7 +6,7 @@ import (
 	"sort"
 	"sync"
 
-	"github.com/bodgit/sevenzip/internal/util"
+	"github.com/todylcom/sevenzip/internal/util"
 )
 
 // Pooler is the interface implemented by a pool.

reader.go

@@ -18,9 +18,9 @@ import (
 	"time"
 
 	"github.com/bodgit/plumbing"
-	"github.com/bodgit/sevenzip/internal/pool"
-	"github.com/bodgit/sevenzip/internal/util"
 	"github.com/hashicorp/go-multierror"
+	"github.com/todylcom/sevenzip/internal/pool"
+	"github.com/todylcom/sevenzip/internal/util"
 	"go4.org/readerutil"
 )
 
@@ -146,6 +146,19 @@ func (f *File) Open() (io.ReadCloser, error) {
 	}, nil
 }
 
+// IsEncrypted returns true if file is encrypted with AES
+func (f *File) IsEncrypted() bool {
+	if f.zip == nil || f.zip.si == nil || f.zip.si.unpackInfo == nil {
+		return false
+	}
+	for _, coder := range f.zip.si.unpackInfo.folder[f.folder].coder {
+		if bytes.Equal(aesCbc256CoderSignature, coder.id) {
+			return true
+		}
+	}
+	return false
+}
+
 // OpenReaderWithPassword will open the 7-zip file specified by name using
 // password as the basis of the decryption key and return a ReadCloser. If
 // name has a ".001" suffix it is assumed there are multiple volumes and each

reader_test.go

@@ -9,8 +9,8 @@ import (
 	"testing"
 	"testing/fstest"
 
-	"github.com/bodgit/sevenzip"
-	"github.com/bodgit/sevenzip/internal/util"
+	"github.com/todylcom/sevenzip"
+	"github.com/todylcom/sevenzip/internal/util"
 	"github.com/stretchr/testify/assert"
 )
 

register.go

@@ -5,16 +5,20 @@ import (
 	"io"
 	"sync"
 
-	"github.com/bodgit/sevenzip/internal/aes7z"
-	"github.com/bodgit/sevenzip/internal/bcj2"
-	"github.com/bodgit/sevenzip/internal/brotli"
-	"github.com/bodgit/sevenzip/internal/bzip2"
-	"github.com/bodgit/sevenzip/internal/deflate"
-	"github.com/bodgit/sevenzip/internal/delta"
-	"github.com/bodgit/sevenzip/internal/lz4"
-	"github.com/bodgit/sevenzip/internal/lzma"
-	"github.com/bodgit/sevenzip/internal/lzma2"
-	"github.com/bodgit/sevenzip/internal/zstd"
+	"github.com/todylcom/sevenzip/internal/aes7z"
+	"github.com/todylcom/sevenzip/internal/bcj2"
+	"github.com/todylcom/sevenzip/internal/brotli"
+	"github.com/todylcom/sevenzip/internal/bzip2"
+	"github.com/todylcom/sevenzip/internal/deflate"
+	"github.com/todylcom/sevenzip/internal/delta"
+	"github.com/todylcom/sevenzip/internal/lz4"
+	"github.com/todylcom/sevenzip/internal/lzma"
+	"github.com/todylcom/sevenzip/internal/lzma2"
+	"github.com/todylcom/sevenzip/internal/zstd"
+)
+
+var (
+	aesCbc256CoderSignature = []byte{0x06, 0xf1, 0x07, 0x01}
 )
 
 // Decompressor describes the function signature that decompression/decryption
@@ -55,7 +59,7 @@ func init() {
 	// LZ4
 	RegisterDecompressor([]byte{0x04, 0xf7, 0x11, 0x04}, Decompressor(lz4.NewReader))
 	// AES-CBC-256 & SHA-256
-	RegisterDecompressor([]byte{0x06, 0xf1, 0x07, 0x01}, Decompressor(aes7z.NewReader))
+	RegisterDecompressor(aesCbc256CoderSignature, Decompressor(aes7z.NewReader))
 	// LZMA2
 	RegisterDecompressor([]byte{0x21}, Decompressor(lzma2.NewReader))
 }

struct.go

@@ -11,7 +11,7 @@ import (
 	"time"
 
 	"github.com/bodgit/plumbing"
-	"github.com/bodgit/sevenzip/internal/util"
+	"github.com/todylcom/sevenzip/internal/util"
 )
 
 var errAlgorithm = errors.New("sevenzip: unsupported compression algorithm")

types.go

@@ -10,7 +10,7 @@ import (
 	"math/bits"
 	"time"
 
-	"github.com/bodgit/sevenzip/internal/util"
+	"github.com/todylcom/sevenzip/internal/util"
 	"github.com/bodgit/windows"
 	"golang.org/x/text/encoding/unicode"
 	"golang.org/x/text/transform"