Add docs and reformat

internal/literals/split.go

@@ -22,10 +22,7 @@ func splitIntoRandomChunks(data []byte) [][]byte {
 	}
 
 	var chunks [][]byte
-	for {
-		if len(data) == 0 {
-			break
-		}
+	for len(data) > 0 {
 		chunkSize := 1 + mathrand.Intn(maxChunkSize)
 		if chunkSize > len(data) {
 			chunkSize = len(data)
@@ -45,6 +42,8 @@ func splitIntoOneByteChunks(data []byte) [][]byte {
 	return chunks
 }
 
+// Shuffles the passed array and returns it back.
+// Applies for inline declaration of randomly shuffled statement arrays
 func shuffleStmts(stmts []ast.Stmt) []ast.Stmt {
 	mathrand.Shuffle(len(stmts), func(i, j int) {
 		stmts[i], stmts[j] = stmts[j], stmts[i]
@@ -54,7 +53,8 @@ func shuffleStmts(stmts []ast.Stmt) []ast.Stmt {
 
 func (x split) obfuscate(data []byte) *ast.BlockStmt {
 	var chunks [][]byte
-	if len(data)/maxChunkSize < minCaseCount { // Short arrays should be divided into single-byte fragments
+	// Short arrays should be divided into single-byte fragments
+	if len(data)/maxChunkSize < minCaseCount {
 		chunks = splitIntoOneByteChunks(data)
 	} else {
 		chunks = splitIntoRandomChunks(data)
@@ -79,8 +79,7 @@ func (x split) obfuscate(data []byte) *ast.BlockStmt {
 		}
 	}
 
-	switchCases := []ast.Stmt{
-		&ast.CaseClause{
+	switchCases := []ast.Stmt{&ast.CaseClause{
 		List: []ast.Expr{ah.IntLit(decryptIndex)},
 		Body: shuffleStmts([]ast.Stmt{
 			&ast.AssignStmt{
@@ -99,8 +98,7 @@ func (x split) obfuscate(data []byte) *ast.BlockStmt {
 				}),
 			},
 		}),
-		},
-	}
+	}}
 	for i := range chunks {
 		index := indexes[i]
 		nextIndex := indexes[i+1]
@@ -142,8 +140,8 @@ func (x split) obfuscate(data []byte) *ast.BlockStmt {
 		})
 	}
 
-	return ah.BlockStmt(&ast.DeclStmt{
-		Decl: &ast.GenDecl{
+	return ah.BlockStmt(
+		&ast.DeclStmt{Decl: &ast.GenDecl{
 			Tok: token.VAR,
 			Specs: []ast.Spec{
 				&ast.ValueSpec{
@@ -151,8 +149,7 @@ func (x split) obfuscate(data []byte) *ast.BlockStmt {
 					Type:  &ast.ArrayType{Elt: ah.Ident("byte")},
 				},
 			},
-		},
-	},
+		}},
 		&ast.AssignStmt{
 			Lhs: []ast.Expr{ah.Ident("i")},
 			Tok: token.DEFINE,