aboutsummaryrefslogtreecommitdiff
path: root/tde/tde.go
blob: 664778de541d5a85cb624e1d76fda974b2a938be (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
48
49
package tde

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

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
	}
	log.Println(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
}