From e04c605c930eb3184b0e31f0c7b36629c87da676 Mon Sep 17 00:00:00 2001
From: pagran <pagran@protonmail.com>
Date: Sun, 15 Jan 2023 23:03:09 +0100
Subject: [PATCH] preallocate data in shuffle and split literal obfuscator

---
 internal/literals/shuffle.go | 12 ++++--------
 internal/literals/split.go   | 14 +++++---------
 2 files changed, 9 insertions(+), 17 deletions(-)

diff --git a/internal/literals/shuffle.go b/internal/literals/shuffle.go
index be1de5a..0590e80 100644
--- a/internal/literals/shuffle.go
+++ b/internal/literals/shuffle.go
@@ -52,14 +52,10 @@ func (shuffle) obfuscate(obfRand *mathrand.Rand, data []byte) *ast.BlockStmt {
 			Tok: token.DEFINE,
 			Rhs: []ast.Expr{ah.DataToByteSlice(shuffledFullData)},
 		},
-		&ast.DeclStmt{
-			Decl: &ast.GenDecl{
-				Tok: token.VAR,
-				Specs: []ast.Spec{&ast.ValueSpec{
-					Names: []*ast.Ident{ast.NewIdent("data")},
-					Type:  &ast.ArrayType{Elt: ast.NewIdent("byte")},
-				}},
-			},
+		&ast.AssignStmt{
+			Lhs: []ast.Expr{ast.NewIdent("data")},
+			Tok: token.DEFINE,
+			Rhs: []ast.Expr{ah.CallExpr(ast.NewIdent("make"), &ast.ArrayType{Elt: ast.NewIdent("byte")}, ah.IntLit(0), ah.IntLit(len(data)+1))},
 		},
 		&ast.AssignStmt{
 			Lhs: []ast.Expr{ast.NewIdent("data")},
diff --git a/internal/literals/split.go b/internal/literals/split.go
index 584c5c2..2aacf76 100644
--- a/internal/literals/split.go
+++ b/internal/literals/split.go
@@ -158,15 +158,11 @@ func (split) obfuscate(obfRand *mathrand.Rand, data []byte) *ast.BlockStmt {
 	}
 
 	return ah.BlockStmt(
-		&ast.DeclStmt{Decl: &ast.GenDecl{
-			Tok: token.VAR,
-			Specs: []ast.Spec{
-				&ast.ValueSpec{
-					Names: []*ast.Ident{ast.NewIdent("data")},
-					Type:  &ast.ArrayType{Elt: ast.NewIdent("byte")},
-				},
-			},
-		}},
+		&ast.AssignStmt{
+			Lhs: []ast.Expr{ast.NewIdent("data")},
+			Tok: token.DEFINE,
+			Rhs: []ast.Expr{ah.CallExpr(ast.NewIdent("make"), &ast.ArrayType{Elt: ast.NewIdent("byte")}, ah.IntLit(0), ah.IntLit(len(data)+1))},
+		},
 		&ast.AssignStmt{
 			Lhs: []ast.Expr{ast.NewIdent("i")},
 			Tok: token.DEFINE,