From a508a62b8c79dbe6b68c447f544e24bde4145475 Mon Sep 17 00:00:00 2001 From: Olaoluwa Osuntokun Date: Sun, 10 Mar 2019 16:28:18 -0700 Subject: [PATCH] lnwire: add bool types to codec for SCB format --- lnwire/lnwire.go | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/lnwire/lnwire.go b/lnwire/lnwire.go index e99304973..28dca7dff 100644 --- a/lnwire/lnwire.go +++ b/lnwire/lnwire.go @@ -416,6 +416,14 @@ func WriteElement(w io.Writer, element interface{}) error { return err } + case bool: + var b [1]byte + if e { + b[0] = 1 + } + if _, err := w.Write(b[:]); err != nil { + return err + } default: 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 { var err error 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: var a [32]byte if _, err := io.ReadFull(r, a[:]); err != nil {