aboutsummaryrefslogtreecommitdiff
path: root/peer/peer.go
blob: 02b638dcf722d2d5ea6222db10c5d5ccbd438485 (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
package peer

import (
	"bytes"
	"net"

	"dim13.org/btget/bitfield"
)

const BlockSize = 1 << 14 // 16k

const (
	Idle = iota
	Connecting
	InHandshake
	Connected
	Failed
)

type Flags struct {
	Unchocked  bool
	Interested bool
}

type Peer struct {
	conn       net.Conn
	Addr       *net.TCPAddr
	ID         []byte
	Valid      bool
	Local      Flags
	Remote     Flags
	Optimistic bool
	buf        *bytes.Buffer
	bitField   bitfield.BitField
}

func NewPeer(addr string) (Peer, error) {
	a, err := net.ResolveTCPAddr("tcp", addr)
	if err != nil {
		return Peer{}, err
	}
	return Peer{Addr: a}, nil
}