Added ValidateBasic() to all Tx for easy stateless pre-filtering
This commit is contained in:
+32
@@ -2,6 +2,7 @@ package txs
|
||||
|
||||
import (
|
||||
"github.com/tendermint/basecoin"
|
||||
"github.com/tendermint/basecoin/errors"
|
||||
"github.com/tendermint/basecoin/types"
|
||||
"github.com/tendermint/go-wire/data"
|
||||
)
|
||||
@@ -30,6 +31,10 @@ const (
|
||||
TypeMultiSig = "multisig"
|
||||
)
|
||||
|
||||
const (
|
||||
rawMaxSize = 2000 * 1000
|
||||
)
|
||||
|
||||
func init() {
|
||||
basecoin.TxMapper.
|
||||
RegisterImplementation(Raw{}, TypeRaw, ByteRaw).
|
||||
@@ -47,6 +52,13 @@ func (r Raw) Wrap() basecoin.Tx {
|
||||
return basecoin.Tx{r}
|
||||
}
|
||||
|
||||
func (r Raw) ValidateBasic() error {
|
||||
if len(r.Bytes) > rawMaxSize {
|
||||
return errors.TooLarge()
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func NewRaw(d []byte) Raw {
|
||||
return Raw{data.Bytes(d)}
|
||||
}
|
||||
@@ -65,6 +77,11 @@ func NewFee(tx basecoin.Tx, fee types.Coin, addr []byte) *Fee {
|
||||
return &Fee{Tx: tx, Fee: fee, Payer: addr}
|
||||
}
|
||||
|
||||
func (f *Fee) ValidateBasic() error {
|
||||
// TODO: more checks
|
||||
return f.Tx.ValidateBasic()
|
||||
}
|
||||
|
||||
func (f *Fee) Wrap() basecoin.Tx {
|
||||
return basecoin.Tx{f}
|
||||
}
|
||||
@@ -82,6 +99,16 @@ func (mt *MultiTx) Wrap() basecoin.Tx {
|
||||
return basecoin.Tx{mt}
|
||||
}
|
||||
|
||||
func (mt *MultiTx) ValidateBasic() error {
|
||||
for _, t := range mt.Txs {
|
||||
err := t.ValidateBasic()
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
/*** Chain ****/
|
||||
|
||||
// Chain locks this tx to one chain, wrap with this before signing
|
||||
@@ -97,3 +124,8 @@ func NewChain(tx basecoin.Tx, chainID string) *Chain {
|
||||
func (c *Chain) Wrap() basecoin.Tx {
|
||||
return basecoin.Tx{c}
|
||||
}
|
||||
|
||||
func (c *Chain) ValidateBasic() error {
|
||||
// TODO: more checks? chainID?
|
||||
return c.Tx.ValidateBasic()
|
||||
}
|
||||
|
||||
+20
@@ -14,6 +14,7 @@ a basecoin.Tx.
|
||||
package txs
|
||||
|
||||
import (
|
||||
// TODO: merge in usage of pkg/errors into basecoin/errors and remove this
|
||||
"github.com/pkg/errors"
|
||||
|
||||
crypto "github.com/tendermint/go-crypto"
|
||||
@@ -21,6 +22,7 @@ import (
|
||||
"github.com/tendermint/go-wire/data"
|
||||
|
||||
"github.com/tendermint/basecoin"
|
||||
berrs "github.com/tendermint/basecoin/errors"
|
||||
)
|
||||
|
||||
// Signed holds one signature of the data
|
||||
@@ -59,6 +61,14 @@ func (s *OneSig) Wrap() basecoin.Tx {
|
||||
return basecoin.Tx{s}
|
||||
}
|
||||
|
||||
func (s *OneSig) ValidateBasic() error {
|
||||
// TODO: VerifyBytes here, we do it in Signers?
|
||||
if s.Empty() || !s.Pubkey.VerifyBytes(s.SignBytes(), s.Sig) {
|
||||
return berrs.Unauthorized()
|
||||
}
|
||||
return s.Tx.ValidateBasic()
|
||||
}
|
||||
|
||||
// TxBytes returns the full data with signatures
|
||||
func (s *OneSig) TxBytes() ([]byte, error) {
|
||||
return data.ToWire(s)
|
||||
@@ -121,6 +131,16 @@ func (s *MultiSig) Wrap() basecoin.Tx {
|
||||
return basecoin.Tx{s}
|
||||
}
|
||||
|
||||
func (s *MultiSig) ValidateBasic() error {
|
||||
// TODO: more efficient
|
||||
_, err := s.Signers()
|
||||
if err != nil {
|
||||
// TODO: better return value
|
||||
return berrs.Unauthorized()
|
||||
}
|
||||
return s.Tx.ValidateBasic()
|
||||
}
|
||||
|
||||
// TxBytes returns the full data with signatures
|
||||
func (s *MultiSig) TxBytes() ([]byte, error) {
|
||||
return data.ToWire(s)
|
||||
|
||||
Reference in New Issue
Block a user