Implement swap obfuscator

internal/literals/obfuscators.go

@@ -19,6 +19,7 @@ var (
 	// obfuscators contains all types which implement the obfuscator Interface
 	obfuscators = []obfuscator{
 		xor{},
+		swap{},
 	}
 	envGarbleSeed = os.Getenv("GARBLE_SEED")
 )
@@ -58,3 +59,15 @@ func genRandBytes(buffer []byte) {
 func genRandInt() int {
 	return mathrand.Int()
 }
+
+func genRandIntn(max int) int {
+	return mathrand.Intn(max)
+}
+
+func generateIntSlice(max, count int) []int {
+	indexes := make([]int, count)
+	for i := 0; i < count; i++ {
+		indexes[i] = genRandIntn(max)
+	}
+	return indexes
+}

internal/literals/swap.go

@@ -0,0 +1,108 @@
+package literals
+
+import (
+	"go/ast"
+	"go/token"
+	"strconv"
+)
+
+type swap struct{}
+
+// check that the obfuscator interface is implemented
+var _ obfuscator = swap{}
+
+func dataToIntSlice(data []int) *ast.CompositeLit {
+	arr := &ast.CompositeLit{
+		Type: &ast.ArrayType{
+			Len: &ast.Ellipsis{}, // Performance optimization
+			Elt: &ast.Ident{
+				Name: "int",
+			},
+		},
+		Elts: []ast.Expr{},
+	}
+	for _, data := range data {
+		arr.Elts = append(arr.Elts, intLiteral(strconv.Itoa(data)))
+	}
+	return arr
+}
+
+func (x swap) obfuscate(data []byte) *ast.BlockStmt {
+	maxJunkIdxCount := len(data) / 2
+	if maxJunkIdxCount == 0 {
+		maxJunkIdxCount = 1
+	}
+	count := len(data) + genRandIntn(maxJunkIdxCount)
+	if count%2 != 0 {
+		count++
+	}
+	indexes := generateIntSlice(len(data), count)
+	for i := len(indexes) - 2; i >= 0; i -= 2 {
+		data[indexes[i]], data[indexes[i+1]] = data[indexes[i+1]], data[indexes[i]]
+	}
+
+	return &ast.BlockStmt{List: []ast.Stmt{
+		&ast.AssignStmt{
+			Lhs: []ast.Expr{ident("data")},
+			Tok: token.DEFINE,
+			Rhs: []ast.Expr{dataToByteSlice(data)},
+		},
+		&ast.AssignStmt{
+			Lhs: []ast.Expr{
+				ident("indexes"),
+			},
+			Tok: token.DEFINE,
+			Rhs: []ast.Expr{
+				dataToIntSlice(indexes),
+			},
+		},
+		&ast.ForStmt{
+			Init: &ast.AssignStmt{
+				Lhs: []ast.Expr{
+					ident("i"),
+				},
+				Tok: token.DEFINE,
+				Rhs: []ast.Expr{
+					intLiteral("0"),
+				},
+			},
+			Cond: &ast.BinaryExpr{
+				X:  ident("i"),
+				Op: token.LSS,
+				Y:  intLiteral(strconv.Itoa(len(indexes))),
+			},
+			Post: &ast.AssignStmt{
+				Lhs: []ast.Expr{
+					ident("i"),
+				},
+				Tok: token.ADD_ASSIGN,
+				Rhs: []ast.Expr{
+					intLiteral("2"),
+				},
+			},
+			Body: &ast.BlockStmt{
+				List: []ast.Stmt{
+					&ast.AssignStmt{
+						Lhs: []ast.Expr{
+							indexExpr("data", indexExpr("indexes", ident("i"))),
+							indexExpr("data", indexExpr("indexes", &ast.BinaryExpr{
+								X:  ident("i"),
+								Op: token.ADD,
+								Y:  intLiteral("1"),
+							})),
+						},
+						Tok: token.ASSIGN,
+						Rhs: []ast.Expr{
+							indexExpr("data", indexExpr("indexes", &ast.BinaryExpr{
+								X:  ident("i"),
+								Op: token.ADD,
+								Y:  intLiteral("1"),
+							})),
+							indexExpr("data", indexExpr("indexes", ident("i"))),
+						},
+					},
+				},
+			},
+		},
+	}}
+}