fix: grpc web proxy must ensure to read full header (argoproj#6319)

Signed-off-by: Alexander Matyushentsev <[email protected]>
tkasuz · May 26, 2021 · aad1a26 · aad1a26
1 parent 32f60d3
commit aad1a26
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pkg/apiclient/grpcproxy.go b/pkg/apiclient/grpcproxy.go
@@ -152,7 +152,7 @@ func (c *client) startGRPCProxy() (*grpc.Server, net.Listener, error) {
 
 			for {
 				header := make([]byte, frameHeaderLength)
-				if _, err := resp.Body.Read(header); err != nil {
+				if _, err := io.ReadAtLeast(resp.Body, header, frameHeaderLength); err != nil {
 					if err == io.EOF {
 						err = io.ErrUnexpectedEOF
 					}