Jiri Simsa | d7616c9 | 2015-03-24 23:44:30 -0700 | [diff] [blame] | 1 | // Copyright 2015 The Vanadium Authors. All rights reserved. |
| 2 | // Use of this source code is governed by a BSD-style |
| 3 | // license that can be found in the LICENSE file. |
| 4 | |
Asim Shankar | ae8d4c5 | 2014-10-08 13:03:31 -0700 | [diff] [blame] | 5 | package security |
Ankur | 100eb27 | 2014-09-15 16:48:12 -0700 | [diff] [blame] | 6 | |
| 7 | import ( |
gauthamt | 1e313bc | 2014-11-10 15:45:56 -0800 | [diff] [blame] | 8 | "io" |
Ankur | 100eb27 | 2014-09-15 16:48:12 -0700 | [diff] [blame] | 9 | |
Jiri Simsa | 6ac9522 | 2015-02-23 16:11:49 -0800 | [diff] [blame] | 10 | "v.io/v23/security" |
Mike Burrows | 7f7088d | 2015-03-25 13:05:00 -0700 | [diff] [blame] | 11 | "v.io/v23/verror" |
Jiri Simsa | 6ac9522 | 2015-02-23 16:11:49 -0800 | [diff] [blame] | 12 | "v.io/v23/vom" |
Todd Wang | b351149 | 2015-04-07 23:32:34 -0700 | [diff] [blame] | 13 | "v.io/x/ref/lib/security/serialization" |
Ankur | 100eb27 | 2014-09-15 16:48:12 -0700 | [diff] [blame] | 14 | ) |
| 15 | |
Mike Burrows | 7f7088d | 2015-03-25 13:05:00 -0700 | [diff] [blame] | 16 | var ( |
| 17 | errBadDataOrSig = verror.Register(pkgPath+".errBadDataOrSig", verror.NoRetry, "{1:}{2:} invalid data/signature handles data:{3} sig:{4}{:_}") |
| 18 | ) |
| 19 | |
gauthamt | 1e313bc | 2014-11-10 15:45:56 -0800 | [diff] [blame] | 20 | func encodeAndStore(obj interface{}, data, signature io.WriteCloser, signer serialization.Signer) error { |
| 21 | if data == nil || signature == nil { |
Mike Burrows | 7f7088d | 2015-03-25 13:05:00 -0700 | [diff] [blame] | 22 | return verror.New(errBadDataOrSig, nil, data, signature) |
Ankur | 100eb27 | 2014-09-15 16:48:12 -0700 | [diff] [blame] | 23 | } |
gauthamt | 1e313bc | 2014-11-10 15:45:56 -0800 | [diff] [blame] | 24 | swc, err := serialization.NewSigningWriteCloser(data, signature, signer, nil) |
Ankur | 100eb27 | 2014-09-15 16:48:12 -0700 | [diff] [blame] | 25 | if err != nil { |
| 26 | return err |
| 27 | } |
Jungho Ahn | 5d1fe97 | 2015-04-27 17:51:32 -0700 | [diff] [blame] | 28 | enc := vom.NewEncoder(swc) |
Suharsh Sivakumar | 51afaa3 | 2015-01-05 15:36:10 -0800 | [diff] [blame] | 29 | if err := enc.Encode(obj); err != nil { |
Ankur | 100eb27 | 2014-09-15 16:48:12 -0700 | [diff] [blame] | 30 | swc.Close() |
| 31 | return err |
| 32 | } |
gauthamt | 1e313bc | 2014-11-10 15:45:56 -0800 | [diff] [blame] | 33 | return swc.Close() |
Ankur | 100eb27 | 2014-09-15 16:48:12 -0700 | [diff] [blame] | 34 | } |
| 35 | |
gauthamt | 1e313bc | 2014-11-10 15:45:56 -0800 | [diff] [blame] | 36 | func decodeFromStorage(obj interface{}, data, signature io.ReadCloser, publicKey security.PublicKey) error { |
| 37 | if data == nil || signature == nil { |
Mike Burrows | 7f7088d | 2015-03-25 13:05:00 -0700 | [diff] [blame] | 38 | return verror.New(errBadDataOrSig, nil, data, signature) |
Ankur | 100eb27 | 2014-09-15 16:48:12 -0700 | [diff] [blame] | 39 | } |
gauthamt | 1e313bc | 2014-11-10 15:45:56 -0800 | [diff] [blame] | 40 | defer data.Close() |
| 41 | defer signature.Close() |
| 42 | vr, err := serialization.NewVerifyingReader(data, signature, publicKey) |
Ankur | 100eb27 | 2014-09-15 16:48:12 -0700 | [diff] [blame] | 43 | if err != nil { |
| 44 | return err |
| 45 | } |
Jungho Ahn | 5d1fe97 | 2015-04-27 17:51:32 -0700 | [diff] [blame] | 46 | dec := vom.NewDecoder(vr) |
Suharsh Sivakumar | 51afaa3 | 2015-01-05 15:36:10 -0800 | [diff] [blame] | 47 | return dec.Decode(obj) |
Ankur | 100eb27 | 2014-09-15 16:48:12 -0700 | [diff] [blame] | 48 | } |