Fix out of range panics (#182)

Signed-off-by: Dmitry Novikov <novikov.dmitri.a@gmail.com>
daixiang0 · Feb 19, 2024 · 5ae85f3 · 5ae85f3
1 parent 667afeb
commit 5ae85f3
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/pkg/gci/testdata.go b/pkg/gci/testdata.go
@@ -1273,6 +1273,26 @@ import (
 	testing "github.com/daixiang0/test"
 	g "github.com/golang"
 )
+`,
+	},
+	{
+		"no-trailing-newline",
+
+		`sections:
+  - Standard
+`,
+		`package main
+
+import (
+	"net"
+	"fmt"
+)`,
+		`package main
+
+import (
+	"fmt"
+	"net"
+)
 `,
 	},
 }
diff --git a/pkg/parse/parse.go b/pkg/parse/parse.go
@@ -111,6 +111,9 @@ func ParseFile(src []byte, filename string) (ImportList, int, int, int, int, err
 					headEnd = int(decl.Pos()) - 1
 				}
 				tailStart = int(decl.End())
+				if tailStart > len(src) {
+					tailStart = len(src)
+				}
 
 				for _, spec := range genDecl.Specs {
 					imp := spec.(*ast.ImportSpec)