les: return error for missing header also in les backend - StateAndHe…

…aderByNumber
ethereum · Mar 5, 2019 · 4ef987e · 4ef987e
1 parent 6434c5a
commit 4ef987e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/les/api_backend.go b/les/api_backend.go
@@ -77,9 +77,12 @@ func (b *LesApiBackend) BlockByNumber(ctx context.Context, blockNr rpc.BlockNumb
 
 func (b *LesApiBackend) StateAndHeaderByNumber(ctx context.Context, blockNr rpc.BlockNumber) (*state.StateDB, *types.Header, error) {
 	header, err := b.HeaderByNumber(ctx, blockNr)
-	if header == nil || err != nil {
+	if err != nil {
 		return nil, nil, err
 	}
+	if header == nil {
+		return nil, nil, errors.New("header not found")
+	}
 	return light.NewState(ctx, header, b.eth.odr), header, nil
 }