123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108 |
- // Copyright (C) 2016 The Protocol Authors.
- package protocol
- import (
- "bytes"
- "encoding/binary"
- "encoding/hex"
- "io"
- "testing"
- )
- func TestVersion14Hello(t *testing.T) {
- // Tests that we can send and receive a version 0.14 hello message.
- expected := Hello{
- DeviceName: "test device",
- ClientName: "syncthing",
- ClientVersion: "v0.14.5",
- }
- msgBuf, err := expected.Marshal()
- if err != nil {
- t.Fatal(err)
- }
- hdrBuf := make([]byte, 6)
- binary.BigEndian.PutUint32(hdrBuf, HelloMessageMagic)
- binary.BigEndian.PutUint16(hdrBuf[4:], uint16(len(msgBuf)))
- outBuf := new(bytes.Buffer)
- outBuf.Write(hdrBuf)
- outBuf.Write(msgBuf)
- inBuf := new(bytes.Buffer)
- conn := &readWriter{outBuf, inBuf}
- send := &Hello{
- DeviceName: "this device",
- ClientName: "other client",
- ClientVersion: "v0.14.6",
- }
- res, err := ExchangeHello(conn, send)
- if err != nil {
- t.Fatal(err)
- }
- if res.ClientName != expected.ClientName {
- t.Errorf("incorrect ClientName %q != expected %q", res.ClientName, expected.ClientName)
- }
- if res.ClientVersion != expected.ClientVersion {
- t.Errorf("incorrect ClientVersion %q != expected %q", res.ClientVersion, expected.ClientVersion)
- }
- if res.DeviceName != expected.DeviceName {
- t.Errorf("incorrect DeviceName %q != expected %q", res.DeviceName, expected.DeviceName)
- }
- }
- func TestOldHelloMsgs(t *testing.T) {
- // Tests that we can correctly identify old/missing/unknown hello
- // messages.
- cases := []struct {
- msg string
- err error
- }{
- {"00010001", ErrTooOldVersion}, // v12
- {"9F79BC40", ErrTooOldVersion}, // v13
- {"12345678", ErrUnknownMagic},
- }
- for _, tc := range cases {
- msg, _ := hex.DecodeString(tc.msg)
- outBuf := new(bytes.Buffer)
- outBuf.Write(msg)
- inBuf := new(bytes.Buffer)
- conn := &readWriter{outBuf, inBuf}
- send := &Hello{
- DeviceName: "this device",
- ClientName: "other client",
- ClientVersion: "v1.0.0",
- }
- _, err := ExchangeHello(conn, send)
- if err != tc.err {
- t.Errorf("unexpected error %v != %v", err, tc.err)
- }
- }
- }
- type readWriter struct {
- r io.Reader
- w io.Writer
- }
- func (rw *readWriter) Write(data []byte) (int, error) {
- return rw.w.Write(data)
- }
- func (rw *readWriter) Read(data []byte) (int, error) {
- return rw.r.Read(data)
- }
|