Negative height paramter's value leads to undefined behaviour

Follow-up of #4505 Issue: #4501
cosmos · Jun 7, 2019 · e6df636 · e6df636
1 parent a32d5a4
commit e6df636
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/types/rest/rest.go b/types/rest/rest.go
@@ -208,6 +208,11 @@ func ParseQueryHeightOrReturnBadRequest(w http.ResponseWriter, cliCtx context.CL
 			return cliCtx, false
 		}
 
+		if height < 0 {
+			WriteErrorResponse(w, http.StatusBadRequest, "height must be equal or greater than zero")
+			return cliCtx, false
+		}
+
 		if height > 0 {
 			cliCtx = cliCtx.WithHeight(height)
 		}

diff --git a/types/rest/rest_test.go b/types/rest/rest_test.go
@@ -122,7 +122,7 @@ func TestParseQueryHeight(t *testing.T) {
 		{"no height", req0, httptest.NewRecorder(), context.CLIContext{}, emptyHeight, true},
 		{"height", req1, httptest.NewRecorder(), context.CLIContext{}, height, true},
 		{"invalid height", req2, httptest.NewRecorder(), context.CLIContext{}, emptyHeight, false},
-		{"negative height", req3, httptest.NewRecorder(), context.CLIContext{}, emptyHeight, true},
+		{"negative height", req3, httptest.NewRecorder(), context.CLIContext{}, emptyHeight, false},
 	}
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {