simplify variable-length argument handling

Instead of having the transformers deal with variable length argument prefixes,
just do this in the bytesToString/stringToBytes functions.
This commit is contained in:
Steven Allen 2018-06-18 12:23:33 -07:00
parent b68c4a264d
commit 5416c663d5
3 changed files with 26 additions and 47 deletions

View File

@ -50,6 +50,9 @@ func stringToBytes(s string) ([]byte, error) {
if err != nil { if err != nil {
return nil, fmt.Errorf("failed to parse %s: %s %s", p.Name, sp[0], err) return nil, fmt.Errorf("failed to parse %s: %s %s", p.Name, sp[0], err)
} }
if p.Size < 0 { // varint size.
b.Write(CodeToVarint(len(a)))
}
b.Write(a) b.Write(a)
sp = sp[1:] sp = sp[1:]
} }
@ -74,11 +77,13 @@ func validateBytes(b []byte) (err error) {
continue continue
} }
size, err := sizeForAddr(p, b) n, size, err := sizeForAddr(p, b)
if err != nil { if err != nil {
return err return err
} }
b = b[n:]
if len(b) < size || size < 0 { if len(b) < size || size < 0 {
return fmt.Errorf("invalid value for size") return fmt.Errorf("invalid value for size")
} }
@ -109,11 +114,13 @@ func bytesToString(b []byte) (ret string, err error) {
continue continue
} }
size, err := sizeForAddr(p, b) n, size, err := sizeForAddr(p, b)
if err != nil { if err != nil {
return "", err return "", err
} }
b = b[n:]
if len(b) < size || size < 0 { if len(b) < size || size < 0 {
return "", fmt.Errorf("invalid value for size") return "", fmt.Errorf("invalid value for size")
} }
@ -125,6 +132,9 @@ func bytesToString(b []byte) (ret string, err error) {
if err != nil { if err != nil {
return "", err return "", err
} }
if p.Path && len(a) > 0 && a[0] == '/' {
a = a[1:]
}
if len(a) > 0 { if len(a) > 0 {
s += "/" + a s += "/" + a
} }
@ -134,24 +144,24 @@ func bytesToString(b []byte) (ret string, err error) {
return s, nil return s, nil
} }
func sizeForAddr(p Protocol, b []byte) (int, error) { func sizeForAddr(p Protocol, b []byte) (skip, size int, err error) {
switch { switch {
case p.Size > 0: case p.Size > 0:
return (p.Size / 8), nil return 0, (p.Size / 8), nil
case p.Size == 0: case p.Size == 0:
return 0, nil return 0, 0, nil
case p.Path: case p.Path:
size, n, err := ReadVarintCode(b) size, n, err := ReadVarintCode(b)
if err != nil { if err != nil {
return 0, err return 0, 0, err
} }
return size + n, nil return n, size, nil
default: default:
size, n, err := ReadVarintCode(b) size, n, err := ReadVarintCode(b)
if err != nil { if err != nil {
return 0, err return 0, 0, err
} }
return size + n, nil return n, size, nil
} }
} }
@ -168,12 +178,12 @@ func bytesSplit(b []byte) ([][]byte, error) {
return nil, fmt.Errorf("no protocol with code %d", b[0]) return nil, fmt.Errorf("no protocol with code %d", b[0])
} }
size, err := sizeForAddr(p, b[n:]) n2, size, err := sizeForAddr(p, b[n:])
if err != nil { if err != nil {
return nil, err return nil, err
} }
length := n + size length := n + n2 + size
ret = append(ret, b[:length]) ret = append(ret, b[:length])
b = b[length:] b = b[length:]
} }

View File

@ -86,12 +86,12 @@ func (m *multiaddr) Protocols() []Protocol {
ps = append(ps, p) ps = append(ps, p)
b = b[n:] b = b[n:]
size, err := sizeForAddr(p, b) n, size, err := sizeForAddr(p, b)
if err != nil { if err != nil {
panic(err) panic(err)
} }
b = b[size:] b = b[n+size:]
} }
return ps return ps
} }

View File

@ -3,7 +3,6 @@ package multiaddr
import ( import (
"encoding/base32" "encoding/base32"
"encoding/binary" "encoding/binary"
"errors"
"fmt" "fmt"
"net" "net"
"strconv" "strconv"
@ -129,22 +128,10 @@ func p2pStB(s string) ([]byte, error) {
if err != nil { if err != nil {
return nil, fmt.Errorf("failed to parse p2p addr: %s %s", s, err) return nil, fmt.Errorf("failed to parse p2p addr: %s %s", s, err)
} }
size := CodeToVarint(len(m)) return m, nil
b := append(size, m...)
return b, nil
} }
func p2pBtS(b []byte) (string, error) { func p2pBtS(b []byte) (string, error) {
// the address is a varint-prefixed multihash string representation
size, n, err := ReadVarintCode(b)
if err != nil {
return "", err
}
b = b[n:]
if len(b) != size {
return "", errors.New("inconsistent lengths")
}
m, err := mh.Cast(b) m, err := mh.Cast(b)
if err != nil { if err != nil {
return "", err return "", err
@ -155,27 +142,9 @@ func p2pBtS(b []byte) (string, error) {
var TranscoderUnix = NewTranscoderFromFunctions(unixStB, unixBtS) var TranscoderUnix = NewTranscoderFromFunctions(unixStB, unixBtS)
func unixStB(s string) ([]byte, error) { func unixStB(s string) ([]byte, error) {
// the address is the whole remaining string, prefixed by a varint len return []byte(s), nil
size := CodeToVarint(len(s))
b := append(size, []byte(s)...)
return b, nil
} }
func unixBtS(b []byte) (string, error) { func unixBtS(b []byte) (string, error) {
// the address is a varint len prefixed string return string(b), nil
size, n, err := ReadVarintCode(b)
if err != nil {
return "", err
}
b = b[n:]
if len(b) != size {
return "", errors.New("inconsistent lengths")
}
if size == 0 {
return "", errors.New("invalid length")
}
s := string(b)
s = s[1:] // remove starting slash
return s, nil
} }