-
Notifications
You must be signed in to change notification settings - Fork 0
/
net.go
90 lines (75 loc) · 1.85 KB
/
net.go
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
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
package client
import (
"errors"
"fmt"
"log"
"net"
"github.com/Lama06/Oinky-Party/protocol"
)
func (c *client) connect(address string) error {
conn, err := net.Dial("tcp", fmt.Sprintf("%s:%d", address, protocol.Port))
if err != nil {
return fmt.Errorf("failed dial the server: %w", err)
}
err = conn.(*net.TCPConn).SetKeepAlive(true)
if err != nil {
return fmt.Errorf("failed to change the keep alive state: %w", err)
}
c.conn = conn
go c.forwardMessagesToServer()
go c.forwardMessagesFromServer()
return nil
}
func (c *client) forwardMessagesFromServer() {
defer c.disconnect()
for {
msgInSizeBuffer := make([]byte, 4)
n, err := c.conn.Read(msgInSizeBuffer)
if err != nil || n != 4 {
return
}
msgInSize := protocol.BytesToInt32([4]byte{msgInSizeBuffer[0], msgInSizeBuffer[1], msgInSizeBuffer[2], msgInSizeBuffer[3]})
msgIn := make([]byte, 0, msgInSize)
for len(msgIn) != int(msgInSize) {
msgInBuffer := make([]byte, int(msgInSize)-len(msgIn))
n, err = c.conn.Read(msgInBuffer)
if err != nil {
return
}
msgIn = append(msgIn, msgInBuffer[:n]...)
}
c.receive <- msgIn
}
}
func (c *client) forwardMessagesToServer() {
defer c.disconnect()
for msgOut := range c.send {
msgOutSize := protocol.Int32ToBytes(int32(len(msgOut)))
_, err := c.conn.Write([]byte{msgOutSize[0], msgOutSize[1], msgOutSize[2], msgOutSize[3]})
if err != nil {
return
}
_, err = c.conn.Write(msgOut)
if err != nil {
return
}
}
}
func (c *client) SendPacket(packet []byte) {
select {
case c.send <- packet:
return
default:
log.Println(errors.New("packet buffer is full"))
c.disconnect()
}
}
func (c *client) disconnect() {
c.disconnectOnce.Do(func() {
err := c.conn.Close()
if err != nil {
log.Println(fmt.Errorf("error while closing connection to server: %w", err))
}
c.disconnected <- struct{}{}
})
}