mirror of
https://github.com/waku-org/go-multiaddr.git
synced 2025-02-22 19:18:14 +00:00
8af58b283a
* Enforce minimal varint encoding when decoding multiaddrs. * Ensure codecs fit in 32bit integers for compatibility. We can increase the size, but we'd need to change the `Protocol.Codec` type from `int` to `uint64`.
28 lines
533 B
Go
28 lines
533 B
Go
package multiaddr
|
|
|
|
import (
|
|
"math"
|
|
|
|
"github.com/multiformats/go-varint"
|
|
)
|
|
|
|
// CodeToVarint converts an integer to a varint-encoded []byte
|
|
func CodeToVarint(num int) []byte {
|
|
if num < 0 || num > math.MaxInt32 {
|
|
panic("invalid code")
|
|
}
|
|
return varint.ToUvarint(uint64(num))
|
|
}
|
|
|
|
func ReadVarintCode(b []byte) (int, int, error) {
|
|
code, n, err := varint.FromUvarint(b)
|
|
if err != nil {
|
|
return 0, 0, err
|
|
}
|
|
if code > math.MaxInt32 {
|
|
// we only allow 32bit codes.
|
|
return 0, 0, varint.ErrOverflow
|
|
}
|
|
return int(code), n, err
|
|
}
|