71 lines
1.9 KiB
Go
71 lines
1.9 KiB
Go
package wire
|
|
|
|
import (
|
|
"encoding/binary"
|
|
"io"
|
|
"os"
|
|
"time"
|
|
|
|
"github.com/zach-klippenstein/goadb/util"
|
|
)
|
|
|
|
type SyncSender interface {
|
|
// SendOctetString sends a 4-byte string.
|
|
SendOctetString(string) error
|
|
SendInt32(int32) error
|
|
SendFileMode(os.FileMode) error
|
|
SendTime(time.Time) error
|
|
|
|
// Sends len(bytes) as an octet, followed by bytes.
|
|
SendString(str string) error
|
|
}
|
|
|
|
type realSyncSender struct {
|
|
io.Writer
|
|
}
|
|
|
|
func NewSyncSender(w io.Writer) SyncSender {
|
|
return &realSyncSender{w}
|
|
}
|
|
|
|
func (s *realSyncSender) SendOctetString(str string) error {
|
|
if len(str) != 4 {
|
|
return util.AssertionErrorf("octet string must be exactly 4 bytes: '%s'", str)
|
|
}
|
|
|
|
wrappedErr := util.WrapErrorf(writeFully(s.Writer, []byte(str)),
|
|
util.NetworkError, "error sending octet string on sync sender")
|
|
|
|
return wrappedErr
|
|
}
|
|
|
|
func (s *realSyncSender) SendInt32(val int32) error {
|
|
return util.WrapErrorf(binary.Write(s.Writer, binary.LittleEndian, val),
|
|
util.NetworkError, "error sending int on sync sender")
|
|
}
|
|
|
|
func (s *realSyncSender) SendFileMode(mode os.FileMode) error {
|
|
return util.WrapErrorf(binary.Write(s.Writer, binary.LittleEndian, mode),
|
|
util.NetworkError, "error sending filemode on sync sender")
|
|
}
|
|
|
|
func (s *realSyncSender) SendTime(t time.Time) error {
|
|
return util.WrapErrorf(s.SendInt32(int32(t.Unix())),
|
|
util.NetworkError, "error sending time on sync sender")
|
|
}
|
|
|
|
func (s *realSyncSender) SendString(str string) error {
|
|
length := len(str)
|
|
if length > MaxChunkSize {
|
|
// This limit might not apply to filenames, but it's big enough
|
|
// that I don't think it will be a problem.
|
|
return util.AssertionErrorf("str must be <= %d in length", MaxChunkSize)
|
|
}
|
|
|
|
if err := s.SendInt32(int32(length)); err != nil {
|
|
return util.WrapErrorf(err, util.NetworkError, "error sending string length on sync sender")
|
|
}
|
|
return util.WrapErrorf(writeFully(s.Writer, []byte(str)),
|
|
util.NetworkError, "error sending string on sync sender")
|
|
}
|