From 206ea10b45e8659eb3227239d3012b20dbb24231 Mon Sep 17 00:00:00 2001 From: zeripath Date: Thu, 30 Sep 2021 20:31:02 +0100 Subject: [PATCH] Prevent NPE on invalid diff (#17197) * Prevent NPE on invalid diff If ParseCompareInfo returns a nil compare info the defer function needs to ensure that it does not attempt to close the HeadGitRepo. Fix #17193 Signed-off-by: Andrew Thornton * add TEST Co-authored-by: 6543 <6543@obermui.de> --- integrations/compare_test.go | 5 +++++ routers/web/repo/compare.go | 2 +- routers/web/repo/pull.go | 2 +- 3 files changed, 7 insertions(+), 2 deletions(-) diff --git a/integrations/compare_test.go b/integrations/compare_test.go index 08468cbfd6..e8efe7cd09 100644 --- a/integrations/compare_test.go +++ b/integrations/compare_test.go @@ -6,6 +6,7 @@ package integrations import ( "net/http" + "strings" "testing" "github.com/stretchr/testify/assert" @@ -21,4 +22,8 @@ func TestCompareTag(t *testing.T) { selection := htmlDoc.doc.Find(".choose.branch .filter.dropdown") // A dropdown for both base and head. assert.Lenf(t, selection.Nodes, 2, "The template has changed") + + req = NewRequest(t, "GET", "/user2/repo1/compare/invalid") + resp = session.MakeRequest(t, req, http.StatusNotFound) + assert.False(t, strings.Contains(resp.Body.String(), "/assets/img/500.png"), "expect 404 page not 500") } diff --git a/routers/web/repo/compare.go b/routers/web/repo/compare.go index 71ca4df93e..556a6218a6 100644 --- a/routers/web/repo/compare.go +++ b/routers/web/repo/compare.go @@ -635,7 +635,7 @@ func getBranchesAndTagsForRepo(user *models.User, repo *models.Repository) (bool func CompareDiff(ctx *context.Context) { ci := ParseCompareInfo(ctx) defer func() { - if ci.HeadGitRepo != nil { + if ci != nil && ci.HeadGitRepo != nil { ci.HeadGitRepo.Close() } }() diff --git a/routers/web/repo/pull.go b/routers/web/repo/pull.go index 1b15ea9a79..9c9cf0f21d 100644 --- a/routers/web/repo/pull.go +++ b/routers/web/repo/pull.go @@ -1043,7 +1043,7 @@ func CompareAndPullRequestPost(ctx *context.Context) { ci := ParseCompareInfo(ctx) defer func() { - if ci.HeadGitRepo != nil { + if ci != nil && ci.HeadGitRepo != nil { ci.HeadGitRepo.Close() } }()