a9fd15b409c6655fad681a6b810b2d46c6402caa
[epoint] / logic / logic.go
1 package logic
2
3 // main transfer logic
4
5 import (
6         "bytes"
7         "crypto/openpgp"
8         "epoint/document"
9         "epoint/store"
10         "fmt"
11         "time"
12 )
13
14 // TODO: do in docs?
15 const IntLimit = 1e15
16
17 var db *store.Conn
18
19 func AddKeys(d []byte) (err error) {
20         entities, err := openpgp.ReadArmoredKeyRing(bytes.NewBuffer(d))
21         if err != nil {
22                 return
23         }
24         // TODO: allow multiple key uploads at once?
25         if len(entities) > 100 {
26                 err = fmt.Errorf("expected at most 100 keys; got %d", len(entities))
27                 return
28         }
29         for _, e := range entities {
30                 // TODO: various checks..
31                 b := new(bytes.Buffer)
32                 err = e.Serialize(b)
33                 if err != nil {
34                         return
35                 }
36                 fpr := fmt.Sprintf("%040X", e.PrimaryKey.Fingerprint)
37                 err = db.Set("key", fpr, b.Bytes())
38                 if err != nil {
39                         return
40                 }
41                 err = db.Append("fprlist/64", fpr[len(fpr)-16:], []byte(fpr))
42                 if err != nil {
43                         return
44                 }
45                 err = db.Append("fprlist/32", fpr[len(fpr)-8:], []byte(fpr))
46                 if err != nil {
47                         return
48                 }
49         }
50         return
51 }
52
53 func CertByDraft(draftid string) (d []byte, err error) {
54         certid, err := db.Get("certid/debit", draftid)
55         if err != nil {
56                 // TODO: we have the draft but the cert is not ready
57                 return
58         }
59         d, err = db.Get("cert", string(certid))
60         if err != nil {
61                 // shouldn't happen, cert is not available
62                 return
63         }
64         return
65 }
66
67 func CertByDebitCert(debitid string) (d []byte, err error) {
68         creditid, err := db.Get("certid/credit", debitid)
69         if err != nil {
70                 // TODO: we have the debit cert but the credit cert is not ready
71                 return
72         }
73         d, err = db.Get("cert", string(creditid))
74         if err != nil {
75                 // shouldn't happen, cert is not available
76                 return
77         }
78         return
79 }
80
81 // parse clear signed draft and verify it
82 func ParseDraft(d []byte) (draft *document.Draft, draftid string, err error) {
83         iv, signed, err := document.Parse(d)
84         if err != nil {
85                 return
86         }
87         draft, ok := iv.(*document.Draft)
88         if !ok {
89                 err = fmt.Errorf("ParseDraft: expected a draft docuent")
90                 return
91         }
92         draftid = document.Id(signed)
93 /*
94         k, err := db.Get("key", draft.Drawer)
95         if err != nil {
96                 return
97         }
98         kr, err := openpgp.ReadKeyRing(bytes.NewBuffer(k))
99         if err != nil {
100                 // internal error: pubkey cannot be parsed
101                 return
102         }
103         cleaned, err = document.Verify(signed, kr)
104         if err != nil {
105                 return
106         }
107         // TODO: verify issuer
108         _, err = db.Get("key", draft.Beneficiary)
109         if err != nil {
110                 return
111         }
112 */
113         // TODO: do various format checks (AuthorizedBy check etc)
114         if draft.Amount <= 0 || draft.Amount >= IntLimit {
115                 err = fmt.Errorf("draft amount is invalid: %d", draft.Amount)
116                 return
117         }
118         return
119 }
120
121 func ParseDebitCert(d []byte) (cert *document.DebitCert, certid string, err error) {
122         iv, signed, err := document.Parse(d)
123         if err != nil {
124                 return
125         }
126         cert, ok := iv.(*document.DebitCert)
127         if !ok {
128                 err = fmt.Errorf("ParseDebitCert: expected a debit docuent")
129                 return
130         }
131 /*
132         // TODO: keep our key at hand
133         k, err := db.Get("key", cert.AuthorizedBy)
134         if err != nil {
135                 return
136         }
137         kr, err := openpgp.ReadKeyRing(bytes.NewBuffer(k))
138         if err != nil {
139                 // internal error: pubkey cannot be parsed
140                 return
141         }
142         // must clean up to make sure the hash is ok
143         cleaned, err = document.Verify(signed, kr)
144         if err != nil {
145                 return
146         }
147 */
148         certid = document.Id(signed)
149         return
150 }
151
152 func NewDebitCert(draftid string, draft *document.Draft) (*document.DebitCert, error) {
153         cert := new(document.DebitCert)
154         cert.Holder = draft.Drawer
155         cert.Date = time.Seconds()
156         cert.Denomination = "epoint"
157         cert.Issuer = draft.Issuer
158         cert.AuthorizedBy = draft.AuthorizedBy
159         cert.Difference = -draft.Amount
160         cert.Draft = draftid
161         cert.Beneficiary = draft.Beneficiary
162
163         oid, err := db.Get("certid/last", draft.Drawer)
164         oldcertid := string(oid)
165         if err != nil {
166                 // first cert: drawer is issuer
167                 if draft.Drawer != draft.Issuer {
168                         return nil, fmt.Errorf("drawer must be the issuer when drawing an empty account")
169                 }
170                 cert.Serial = 1
171                 cert.Balance = cert.Difference
172                 cert.LastDebitSerial = 0
173                 cert.LastCreditSerial = 0
174         } else {
175                 d, err := db.Get("cert", oldcertid)
176                 if err != nil {
177                         return nil, err
178                 }
179                 iv, _, err := document.Parse(d)
180                 if err != nil {
181                         // internal error
182                         return nil, err
183                 }
184                 // TODO: this is a hack
185                 oldcert, err := document.ToCert(iv)
186                 // TODO: sanity checks? oldcert.Holder == draft.Drawer
187                 cert.Serial = oldcert.Serial + 1
188                 cert.Balance = oldcert.Balance + cert.Difference
189                 if cert.Balance <= -IntLimit {
190                         return nil, fmt.Errorf("balance limit exceeded: %d", cert.Balance)
191                 }
192                 if oldcert.Balance > 0 && cert.Balance < 0 {
193                         return nil, fmt.Errorf("insufficient funds: %d", oldcert.Balance)
194                 }
195                 cert.LastDebitSerial = oldcert.LastDebitSerial
196                 cert.LastCreditSerial = oldcert.LastCreditSerial
197                 if oldcert.IsDebit {
198                         cert.LastDebitSerial = oldcert.Serial
199                 } else {
200                         cert.LastCreditSerial = oldcert.Serial
201                 }
202                 cert.LastCert = &oldcertid
203         }
204         return cert, nil
205 }
206
207 func NewCreditCert(draftid string, draft *document.Draft, dcertid string, dcert *document.DebitCert) (*document.CreditCert, error) {
208         cert := new(document.CreditCert)
209         // TODO: get from old cert instead?
210         cert.Holder = dcert.Beneficiary
211         cert.Date = time.Seconds()
212         // TODO: get these from the cert holder pubkey
213         cert.Denomination = "epoint"
214         cert.Issuer = draft.Issuer
215         cert.AuthorizedBy = dcert.AuthorizedBy // TODO: draft vs dcert vs serverside decision
216         cert.Difference = -dcert.Difference
217         cert.Draft = draftid
218         cert.Drawer = dcert.Holder
219         cert.DebitCert = dcertid
220
221         oid, err := db.Get("certid/last", dcert.Beneficiary)
222         oldcertid := string(oid)
223         if err != nil {
224                 // this is the first cert
225                 cert.Serial = 1
226                 cert.Balance = cert.Difference
227                 cert.LastDebitSerial = 0
228                 cert.LastCreditSerial = 0
229         } else {
230                 d, err := db.Get("cert", oldcertid)
231                 if err != nil {
232                         // internal error
233                         return nil, err
234                 }
235                 iv, _, err := document.Parse(d)
236                 if err != nil {
237                         // internal error
238                         return nil, err
239                 }
240                 // TODO: this is a hack
241                 oldcert, err := document.ToCert(iv)
242                 if err != nil {
243                         // internal error
244                         return nil, err
245                 }
246                 cert.Serial = oldcert.Serial + 1
247                 cert.Balance = oldcert.Balance + cert.Difference
248                 if cert.Balance >= IntLimit {
249                         return nil, fmt.Errorf("balance limit exceeded: %d", cert.Balance)
250                 }
251                 cert.LastDebitSerial = oldcert.LastDebitSerial
252                 cert.LastCreditSerial = oldcert.LastCreditSerial
253                 if oldcert.IsDebit {
254                         cert.LastDebitSerial = oldcert.Serial
255                 } else {
256                         cert.LastCreditSerial = oldcert.Serial
257                 }
258                 cert.LastCert = &oldcertid
259         }
260         return cert, nil
261 }
262
263 // TODO: draft ref
264 func EvalDraft(d []byte, sk *openpgp.Entity) (r []byte, err error) {
265         draft, draftid, err := ParseDraft(d)
266         if err != nil {
267                 return
268         }
269         _, err = db.Get("draft", draftid)
270         if err == nil {
271                 // found
272                 return CertByDraft(draftid)
273         }
274         // if draft is ok we save it
275         err = db.Set("draft", draftid, d)
276         if err != nil {
277                 // internal error
278                 return
279         }
280
281         // debit cert
282         cert, err := NewDebitCert(draftid, draft)
283         if err != nil {
284                 return
285         }
286         r, signed, err := document.Format(cert, sk)
287         certid := document.Id(signed)
288         err = db.Set("cert", certid, r)
289         if err != nil {
290                 // internal error
291                 return
292         }
293         err = db.Set("certid/debit", draftid, []byte(certid))
294         if err != nil {
295                 // internal error
296                 return
297         }
298         err = db.Set("certid/last", cert.Holder, []byte(certid))
299         if err != nil {
300                 // internal error
301                 return
302         }
303         // TODO: append?
304         err = db.Set("certid/all", fmt.Sprintf("%s.%09d", cert.Holder, cert.Serial), []byte(certid))
305         if err != nil {
306                 // internal error
307                 return
308         }
309         return
310 }
311
312 func EvalDebitCert(d []byte, sk *openpgp.Entity) (r []byte, err error) {
313         dcert, dcertid, err := ParseDebitCert(d)
314         if err != nil {
315                 return
316         }
317         r, err = CertByDebitCert(dcertid)
318         if err == nil {
319                 // found
320                 return
321         }
322         // TODO: we only need the draft to know the issuer (+beneficiary)
323         // it should be in the pubkey
324         d, err = db.Get("draft", dcert.Draft)
325         if err != nil {
326                 // internal error
327                 return
328         }
329         iv, _, err := document.Parse(d)
330         if err != nil {
331                 // internal error
332                 return
333         }
334         draft, ok := iv.(*document.Draft)
335         if !ok {
336                 // internal error
337                 err = fmt.Errorf("EvalDebitCert: expected draft from internal db")
338                 return
339         }
340
341         // credit side
342         // TODO: check pubkey etc
343         cert, err := NewCreditCert(dcert.Draft, draft, dcertid, dcert)
344         if err != nil {
345                 // internal error
346                 return
347         }
348         r, signed, err := document.Format(cert, sk)
349         if err != nil {
350                 // internal error
351                 return
352         }
353         certid := document.Id(signed)
354         err = db.Set("cert", certid, r)
355         if err != nil {
356                 // internal error
357                 return
358         }
359         err = db.Set("certid/credit", dcertid, []byte(certid))
360         if err != nil {
361                 // internal error
362                 return
363         }
364         err = db.Set("certid/last", cert.Holder, []byte(certid))
365         if err != nil {
366                 // internal error
367                 return
368         }
369         // TODO: append?
370         err = db.Set("certid/all", fmt.Sprintf("%s.%09d", cert.Holder, cert.Serial), []byte(certid))
371         if err != nil {
372                 // internal error
373                 return
374         }
375         return
376 }
377
378 func Init() (err error) {
379         db, err = store.Open("teststore")
380         if err != nil {
381                 return
382         }
383         err = db.Ensure("key")
384         if err != nil {
385                 return
386         }
387         err = db.Ensure("cert")
388         if err != nil {
389                 return
390         }
391         err = db.Ensure("draft")
392         if err != nil {
393                 return
394         }
395         err = db.Ensure("certid/credit")
396         if err != nil {
397                 return
398         }
399         err = db.Ensure("certid/debit")
400         if err != nil {
401                 return
402         }
403         err = db.Ensure("certid/last")
404         if err != nil {
405                 return
406         }
407         err = db.Ensure("certid/all")
408         if err != nil {
409                 return
410         }
411         err = db.Ensure("fprlist/64")
412         if err != nil {
413                 return
414         }
415         err = db.Ensure("fprlist/32")
416         if err != nil {
417                 return
418         }
419         return
420 }