Skip to content

Commit 2b36c9d

Browse files
err handling
Signed-off-by: SREENIDHI DR <[email protected]>
1 parent 82cbfc3 commit 2b36c9d

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

Diff for: token/services/ttx/endorse.go

+12-1
Original file line numberDiff line numberDiff line change
@@ -352,7 +352,10 @@ func (c *CollectEndorsementsView) signRemote(context view.Context, party view.Id
352352
if err != nil {
353353
return nil, errors.Wrap(err, "failed reading message")
354354
}
355-
355+
if len(sigma) == 0 {
356+
info := context.Session().Info()
357+
return nil, errors.Errorf("received empty message, session closed [%s:%v]", info.ID, info.Closed)
358+
}
356359
verifier, err := verifierGetter(party)
357360
if err != nil {
358361
return nil, errors.Wrapf(err, "failed getting verifier for [%s]", party)
@@ -563,6 +566,10 @@ func (c *CollectEndorsementsView) distributeEvnToParty(context view.Context, ent
563566
if err != nil {
564567
return errors.Wrapf(err, "failed reading message on session [%s]", session.Info().ID)
565568
}
569+
if len(sigma) == 0 {
570+
info := context.Session().Info()
571+
return errors.Errorf("received empty message, session closed [%s:%v]", info.ID, info.Closed)
572+
}
566573
logger.Debugf("received ack from [%s] [%s], checking signature on [%s]",
567574
entry.LongTerm, hash.Hashable(sigma).String(),
568575
hash.Hashable(txRaw).String())
@@ -747,6 +754,10 @@ func (f *ReceiveTransactionView) Call(context view.Context) (interface{}, error)
747754
if err != nil {
748755
span.RecordError(err)
749756
}
757+
if len(msg) == 0 {
758+
info := context.Session().Info()
759+
return nil, errors.Errorf("received empty message, session closed [%s:%v]", info.ID, info.Closed)
760+
}
750761
span.AddEvent("receive_tx")
751762

752763
if logger.IsEnabledFor(zapcore.DebugLevel) {

0 commit comments

Comments
 (0)