aboutsummaryrefslogtreecommitdiff
path: root/tde/tde.go
blob: 44ee6e3b5cb0420fa8d210fb7acbf4e0d0db34f9 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
package tde

import (
	"encoding/binary"
	"io"
	"net"
)

type TDE struct {
	net.Conn
}

var nbo = binary.BigEndian // Network Byte Order

func (t TDE) Write(p []byte) (n int, err error) {
	size := int16(len(p))
	if err := binary.Write(t.Conn, nbo, size); err != nil {
		return 0, err
	}
	if err := binary.Write(t.Conn, nbo, p); err != nil {
		return 0, err
	}
	return int(size), nil
}

func (t TDE) Read(p []byte) (n int, err error) {
	var size int16
	if err := binary.Read(t.Conn, nbo, &size); err != nil {
		return 0, err
	}
	if int(size) < len(p) {
		p = p[:size]
		err = io.EOF
	}
	if err := binary.Read(t.Conn, nbo, p); err != nil {
		return 0, err
	}
	return int(size), err
}

func Dial(service string) (TDE, error) {
	conn, err := net.Dial("tcp", service)
	if err != nil {
		return TDE{}, err
	}
	return TDE{conn}, nil
}