37 lines
594 B
Go
37 lines
594 B
Go
|
package krpc
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"github.com/anacrolix/torrent/bencode"
|
||
|
|
)
|
||
|
|
|
||
|
|
type ID [20]byte
|
||
|
|
|
||
|
|
var _ interface {
|
||
|
|
bencode.Marshaler
|
||
|
|
bencode.Unmarshaler
|
||
|
|
} = (*ID)(nil)
|
||
|
|
|
||
|
|
func IdFromString(s string) (id ID) {
|
||
|
|
if n := copy(id[:], s); n != 20 {
|
||
|
|
panic(n)
|
||
|
|
}
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
func (id *ID) MarshalBencode() ([]byte, error) {
|
||
|
|
return []byte("20:" + string(id[:])), nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (id *ID) UnmarshalBencode(b []byte) error {
|
||
|
|
var s string
|
||
|
|
if err := bencode.Unmarshal(b, &s); err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if n := copy(id[:], s); n != 20 {
|
||
|
|
return fmt.Errorf("string has wrong length: %d", n)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|