fix: Add ARM64 filter support

Author: bodgit

README.md

@@ -21,7 +21,7 @@ Current status:
 * Handles archives split into multiple volumes, (`7za a -v100m test.7z ...`).
 * Handles self-extracting archives, (`7za a -sfx archive.exe ...`).
 * Validates CRC values as it parses the file.
-* Supports ARM, BCJ, BCJ2, Brotli, Bzip2, Copy, Deflate, Delta, LZ4, LZMA, LZMA2, PPC, SPARC and Zstandard methods.
+* Supports ARM, ARM64, BCJ, BCJ2, Brotli, Bzip2, Copy, Deflate, Delta, LZ4, LZMA, LZMA2, PPC, SPARC and Zstandard methods.
 * Implements the `fs.FS` interface so you can treat an opened 7-zip archive like a filesystem.
 
 More examples of 7-zip archives are needed to test all of the different combinations/algorithms possible.

internal/bra/arm64.go

@@ -0,0 +1,85 @@
+package bra
+
+import (
+	"encoding/binary"
+	"io"
+)
+
+const arm64Alignment = 4
+
+type arm64 struct {
+	ip uint32
+}
+
+func (c *arm64) Size() int { return arm64Alignment }
+
+//nolint:funlen
+func (c *arm64) Convert(b []byte, encoding bool) int {
+	if len(b) < c.Size() {
+		return 0
+	}
+
+	var i int
+
+	for i = 0; i < len(b) & ^(arm64Alignment-1); i += arm64Alignment {
+		v := binary.LittleEndian.Uint32(b[i:])
+
+		if (v-0x94000000)&0xfc000000 == 0 {
+			v <<= 2
+
+			if encoding {
+				v += c.ip
+			} else {
+				v -= c.ip
+			}
+
+			v >>= 2
+			v &= 0x03ffffff
+			v |= 0x94000000
+
+			binary.LittleEndian.PutUint32(b[i:], v)
+
+			goto next
+		}
+
+		v -= 0x90000000
+
+		if v&0x9f000000 == 0 {
+			const (
+				flag = uint32(1) << (24 - 4)
+				mask = uint32(1)<<24 - flag<<1
+			)
+
+			v += flag
+
+			if v&mask > 0 {
+				goto next
+			}
+
+			z, ip := v&0xffffffe0|v>>26, (c.ip>>(12-3)) & ^uint32(7)
+
+			if encoding {
+				z += ip
+			} else {
+				z -= ip
+			}
+
+			v &= 0x1f
+			v |= 0x90000000
+			v |= z << 26
+			v |= 0x00ffffe0 & ((z & (flag<<1 - 1)) - flag)
+
+			binary.LittleEndian.PutUint32(b[i:], v)
+		}
+
+	next:
+		c.ip += uint32(arm64Alignment)
+	}
+
+	return i
+}
+
+// NewARM64Reader returns a new ARM64 io.ReadCloser.
+func NewARM64Reader(_ []byte, _ uint64, readers []io.ReadCloser) (io.ReadCloser, error) {
+	return newReader(readers, new(arm64))
+}

reader_test.go

@@ -197,6 +197,10 @@ func TestOpenReader(t *testing.T) {
 			file: "COMPRESS-492.7z",
 			err:  sevenzip.ErrMissingUnpackInfo,
 		},
+		{
+			name: "arm64",
+			file: "arm64.7z",
+		},
 	}
 
 	for _, table := range tables {
@@ -657,3 +661,7 @@ func BenchmarkARM(b *testing.B) {
 func BenchmarkSPARC(b *testing.B) {
 	benchmarkArchive(b, "sparc.7z", "", true)
 }
+
+func BenchmarkARM64(b *testing.B) {
+	benchmarkArchive(b, "arm64.7z", "", true)
+}

register.go

@@ -69,6 +69,8 @@ func init() {
 	RegisterDecompressor([]byte{0x04, 0xf7, 0x11, 0x04}, Decompressor(lz4.NewReader))
 	// AES-CBC-256 & SHA-256
 	RegisterDecompressor([]byte{0x06, 0xf1, 0x07, 0x01}, Decompressor(aes7z.NewReader))
+	// ARM64
+	RegisterDecompressor([]byte{0x0a}, Decompressor(bra.NewARM64Reader))
 	// LZMA2
 	RegisterDecompressor([]byte{0x21}, Decompressor(lzma2.NewReader))
 }