Moved all gas and payment values to uint64 to make sure we are safe here

This commit is contained in:
Ethan Frey
2017-08-04 14:11:01 +02:00
parent 2f4f875dd4
commit 640f06998a
11 changed files with 24 additions and 24 deletions
+3 -3
View File
@@ -103,11 +103,11 @@ func (PriceHandler) DeliverTx(ctx basecoin.Context, store state.SimpleDB,
// PriceShowTx lets us bounce back a given fee/gas on CheckTx
type PriceShowTx struct {
GasAllocated uint
GasPayment uint
GasAllocated uint64
GasPayment uint64
}
func NewPriceShowTx(gasAllocated, gasPayment uint) basecoin.Tx {
func NewPriceShowTx(gasAllocated, gasPayment uint64) basecoin.Tx {
return PriceShowTx{GasAllocated: gasAllocated, GasPayment: gasPayment}.Wrap()
}
+2 -2
View File
@@ -77,7 +77,7 @@ func runAllDelivers(ctx basecoin.Context, store state.SimpleDB, txs []basecoin.T
func combineChecks(all []basecoin.CheckResult) basecoin.CheckResult {
datas := make([]data.Bytes, len(all))
logs := make([]string, len(all))
var allocated, payments uint
var allocated, payments uint64
for i, r := range all {
datas[i] = r.Data
logs[i] = r.Log
@@ -97,7 +97,7 @@ func combineChecks(all []basecoin.CheckResult) basecoin.CheckResult {
func combineDelivers(all []basecoin.DeliverResult) basecoin.DeliverResult {
datas := make([]data.Bytes, len(all))
logs := make([]string, len(all))
var used uint
var used uint64
var diffs []*abci.Validator
for i, r := range all {
datas[i] = r.Data
+2 -2
View File
@@ -43,8 +43,8 @@ func TestMultiplexer(t *testing.T) {
cases := [...]struct {
tx basecoin.Tx
valid bool
gasAllocated uint
gasPayment uint
gasAllocated uint64
gasPayment uint64
log string
data data.Bytes
}{
+3 -3
View File
@@ -16,9 +16,9 @@ const (
//NameCoin - name space of the coin module
NameCoin = "coin"
// CostSend is GasAllocation per input/output
CostSend = uint(10)
CostSend = uint64(10)
// CostCredit is GasAllocation of a credit allocation
CostCredit = uint(20)
CostCredit = uint64(20)
)
// Handler includes an accountant
@@ -53,7 +53,7 @@ func (h Handler) CheckTx(ctx basecoin.Context, store state.SimpleDB,
switch t := tx.Unwrap().(type) {
case SendTx:
// price based on inputs and outputs
used := uint(len(t.Inputs) + len(t.Outputs))
used := uint64(len(t.Inputs) + len(t.Outputs))
return basecoin.NewCheck(used*CostSend, ""), h.checkSendTx(ctx, store, t)
case CreditTx:
// default price of 20, constant work
+2 -2
View File
@@ -110,7 +110,7 @@ func TestCheckDeliverSendTx(t *testing.T) {
tx basecoin.Tx
perms []basecoin.Actor
final []money // nil for error
cost uint // gas allocated (if not error)
cost uint64 // gas allocated (if not error)
}{
{
[]money{{addr1, moreCoins}},
@@ -175,7 +175,7 @@ func TestCheckDeliverSendTx(t *testing.T) {
assert.Nil(err, "%d: %+v", i, err)
assert.Nil(err2, "%d: %+v", i, err2)
// make sure proper gas is set
assert.Equal(uint(0), cres.GasPayment, "%d", i)
assert.Equal(uint64(0), cres.GasPayment, "%d", i)
assert.Equal(tc.cost, cres.GasAllocated, "%d", i)
// make sure the final balances are correct
for _, f := range tc.final {
+2 -2
View File
@@ -55,14 +55,14 @@ func (h SimpleFeeMiddleware) CheckTx(ctx basecoin.Context, store state.SimpleDB,
return res, err
}
var paid, used uint
var paid, used uint64
if !fee.Fee.IsZero() { // now, try to make a IPC call to coins...
send := coin.NewSendOneTx(fee.Payer, h.Collector, coin.Coins{fee.Fee})
sendRes, err := next.CheckTx(ctx, store, send)
if err != nil {
return res, err
}
paid = uint(fee.Fee.Amount)
paid = uint64(fee.Fee.Amount)
used = sendRes.GasAllocated
}
+1 -1
View File
@@ -71,7 +71,7 @@ func TestFeeChecks(t *testing.T) {
left coin.Coins
collected coin.Coins
// expected gas allocated
expectedCost uint
expectedCost uint64
}{
// make sure it works with no fee (control group)
{true, app1, act1, false, act1, zero, mixed, nil, 0},
+2 -2
View File
@@ -10,9 +10,9 @@ const (
//NameRole - name space of the roles module
NameRole = "role"
// CostCreate is the cost to create a new role
CostCreate = uint(40)
CostCreate = uint64(40)
// CostAssume is the cost to assume a role as part of a tx
CostAssume = uint(5)
CostAssume = uint64(5)
)
// Handler allows us to create new roles
+1 -1
View File
@@ -44,7 +44,7 @@ func TestCreateRole(t *testing.T) {
assert.Nil(err, "%d/%s: %+v", i, tc.role, err)
assert.Nil(err2, "%d/%s: %+v", i, tc.role, err2)
assert.Equal(roles.CostCreate, cres.GasAllocated)
assert.Equal(uint(0), cres.GasPayment)
assert.Equal(uint64(0), cres.GasPayment)
} else {
assert.NotNil(err, "%d/%s", i, tc.role)
assert.NotNil(err2, "%d/%s", i, tc.role)
+2 -2
View File
@@ -99,8 +99,8 @@ func TestAssumeRole(t *testing.T) {
assert.Nil(err, "%d: %+v", i, err)
assert.Nil(err2, "%d: %+v", i, err2)
// make sure we charge for each role
assert.Equal(roles.CostAssume*uint(len(tc.roles)), cres.GasAllocated)
assert.Equal(uint(0), cres.GasPayment)
assert.Equal(roles.CostAssume*uint64(len(tc.roles)), cres.GasAllocated)
assert.Equal(uint64(0), cres.GasPayment)
} else {
assert.NotNil(err, "%d", i)
assert.NotNil(err2, "%d", i)