Fix NULL pointer dereference.

This commit is contained in:
Mark D. Roth
2016-07-01 08:50:57 -07:00
parent 84352c6400
commit 0eec1573d0

View File

@ -525,7 +525,7 @@ func serverNewPayload(t testpb.PayloadType, size int32) (*testpb.Payload, error)
func (s *testServer) UnaryCall(ctx context.Context, in *testpb.SimpleRequest) (*testpb.SimpleResponse, error) {
resp := in.GetResponseStatus()
if *resp.Code != 0 {
if resp != nil && *resp.Code != 0 {
return nil, grpc.Errorf(codes.Code(*resp.Code), *resp.Message)
}
pl, err := serverNewPayload(in.GetResponseType(), in.GetResponseSize())
@ -539,7 +539,7 @@ func (s *testServer) UnaryCall(ctx context.Context, in *testpb.SimpleRequest) (*
func (s *testServer) StreamingOutputCall(args *testpb.StreamingOutputCallRequest, stream testpb.TestService_StreamingOutputCallServer) error {
resp := args.GetResponseStatus()
if *resp.Code != 0 {
if resp != nil && *resp.Code != 0 {
return grpc.Errorf(codes.Code(*resp.Code), *resp.Message)
}
cs := args.GetResponseParameters()