Merge pull request #13575 from AdamKorcz/fuzz1

v2auth: fix out of bounds in test
etcd-io · Jan 16, 2022 · f4266e2 · f4266e2
2 parents b8c5d44 + 6aac35d
commit f4266e2
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/server/etcdserver/api/v2auth/auth_test.go b/server/etcdserver/api/v2auth/auth_test.go
@@ -176,6 +176,11 @@ func (td *testDoer) Do(_ context.Context, req etcdserverpb.Request) (etcdserver.
 		}, nil
 	}
 	if (req.Method == "GET" || req.Method == "QGET") && td.get != nil {
+		if td.getindex >= len(td.get) {
+			return etcdserver.Response{}, &v2error.Error{
+				ErrorCode: v2error.EcodeKeyNotFound,
+			}
+		}
 		res := td.get[td.getindex]
 		if res.Event == nil {
 			td.getindex++
@@ -187,6 +192,11 @@ func (td *testDoer) Do(_ context.Context, req etcdserverpb.Request) (etcdserver.
 		return res, nil
 	}
 	if req.Method == "PUT" && td.put != nil {
+		if td.putindex >= len(td.put) {
+			return etcdserver.Response{}, &v2error.Error{
+				ErrorCode: v2error.EcodeKeyNotFound,
+			}
+		}
 		res := td.put[td.putindex]
 		if res.Event == nil {
 			td.putindex++