akovalenko/lnd.xprv
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

lnwire: add bool types to codec for SCB format

Browse Source
This commit is contained in:
Olaoluwa Osuntokun 2019-03-10 16:28:18 -07:00
parent 64b8facf0e
commit a508a62b8c
No known key found for this signature in database
GPG Key ID: CE58F7F8E20FD9A2
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
lnwire/lnwire.go
View File

@ -416,6 +416,14 @@ func WriteElement(w io.Writer, element interface{}) error {
return err return err
} }
case bool:
var b [1]byte
if e {
b[0] = 1
}
if _, err := w.Write(b[:]); err != nil {
return err
}
default: default:
return fmt.Errorf("Unknown type in WriteElement: %T", e) return fmt.Errorf("Unknown type in WriteElement: %T", e)
} }
@ -440,6 +448,16 @@ func WriteElements(w io.Writer, elements ...interface{}) error {
func ReadElement(r io.Reader, element interface{}) error { func ReadElement(r io.Reader, element interface{}) error {
var err error var err error
switch e := element.(type) { switch e := element.(type) {
case *bool:
var b [1]byte
if _, err := io.ReadFull(r, b[:]); err != nil {
return err
}
if b[0] == 1 {
*e = true
}
case *NodeAlias: case *NodeAlias:
var a [32]byte var a [32]byte
if _, err := io.ReadFull(r, a[:]); err != nil { if _, err := io.ReadFull(r, a[:]); err != nil {