astelm
/
grpcurl
mirror of https://github.com/fullstorydev/grpcurl.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix nil-dereference panic (#395)

This commit is contained in:
Joshua Humphries 2023-05-12 12:50:32 -04:00 committed by GitHub
parent 0efcfa65f2
commit d5b8e4d4ce
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
invoke.go
View File

@ -264,9 +264,11 @@ func invokeServerStream(ctx context.Context, stub grpcdynamic.Stub, md *desc.Met
// Now we can actually invoke the RPC! // Now we can actually invoke the RPC!
str, err := stub.InvokeRpcServerStream(ctx, md, req) str, err := stub.InvokeRpcServerStream(ctx, md, req)
if str != nil {
if respHeaders, err := str.Header(); err == nil { if respHeaders, err := str.Header(); err == nil {
handler.OnReceiveHeaders(respHeaders) handler.OnReceiveHeaders(respHeaders)
} }
}
// Download each response message // Download each response message
for err == nil { for err == nil {
@ -288,7 +290,9 @@ func invokeServerStream(ctx context.Context, stub grpcdynamic.Stub, md *desc.Met
return fmt.Errorf("grpc call for %q failed: %v", md.GetFullyQualifiedName(), err) return fmt.Errorf("grpc call for %q failed: %v", md.GetFullyQualifiedName(), err)
} }
if str != nil {
handler.OnReceiveTrailers(stat, str.Trailer()) handler.OnReceiveTrailers(stat, str.Trailer())
}
return nil return nil
} }