From 5e9fc897c292d0c51e263f6d8a4f6cd31da131cc Mon Sep 17 00:00:00 2001 From: nsz Date: Sun, 29 Jan 2012 20:00:26 +0100 Subject: [PATCH] fix balance check in server logic --- pkg/server/server.go | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/pkg/server/server.go b/pkg/server/server.go index 93d8da2..8be10b2 100644 --- a/pkg/server/server.go +++ b/pkg/server/server.go @@ -429,6 +429,10 @@ func handleDebit(w *work) (c []byte, err error) { return } +func isIssuer(c *document.Cert) bool { + return c.Issuer == c.Holder +} + func newDebitCert(w *work) (*document.DebitCert, error) { cert := new(document.DebitCert) cert.Holder = w.draft.Drawer @@ -461,6 +465,7 @@ func newDebitCert(w *work) (*document.DebitCert, error) { // internal error return nil, err } + // TODO: make sure oldcert and newcert cannot become inconsistent // TODO: this is a hack oldcert, err := document.ToCert(iv) if err != nil { @@ -473,8 +478,8 @@ func newDebitCert(w *work) (*document.DebitCert, error) { if cert.Balance <= -IntLimit { return nil, fmt.Errorf("balance limit exceeded: %d", cert.Balance) } - if oldcert.Balance > 0 && cert.Balance < 0 { - return nil, fmt.Errorf("insufficient funds: %d", oldcert.Balance) + if !isIssuer(&cert.Cert) && cert.Balance < 0 { + return nil, fmt.Errorf("insufficient funds: %d, draft: %d", oldcert.Balance, cert.Difference) } cert.LastDebitSerial = oldcert.LastDebitSerial cert.LastCreditSerial = oldcert.LastCreditSerial @@ -536,6 +541,10 @@ func newCreditCert(w *work) (*document.CreditCert, error) { if cert.Balance >= IntLimit { return nil, fmt.Errorf("balance limit exceeded: %d", cert.Balance) } + // sanity check + if isIssuer(&cert.Cert) && cert.Balance > 0 { + return nil, fmt.Errorf("internal error") + } cert.LastDebitSerial = oldcert.LastDebitSerial cert.LastCreditSerial = oldcert.LastCreditSerial if _, ok := iv.(*document.DebitCert); ok { -- 2.20.1