telebit/rvpn/connection/connection.go

117 lines
2.2 KiB
Go
Raw Normal View History

package connection
import (
"encoding/hex"
"github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}
// Connection track websocket and faciliates in and out data
type Connection struct {
connectionTable *Table
// The websocket connection.
conn *websocket.Conn
// Buffered channel of outbound messages.
send chan []byte
// Address of the Remote End Point
source string
// bytes in
bytesIn int64
// bytes out
bytesOut int64
// communications channel between go routines
commCh chan bool
//initialDomains - a list of domains from the JWT
initialDomains []interface{}
}
//NewConnection -- Constructor
func NewConnection(connectionTable *Table, conn *websocket.Conn, remoteAddress string, initialDomains []interface{}) (p *Connection) {
p = new(Connection)
p.connectionTable = connectionTable
p.conn = conn
p.source = remoteAddress
p.bytesIn = 0
p.bytesOut = 0
p.send = make(chan []byte, 256)
p.commCh = make(chan bool)
p.initialDomains = initialDomains
return
}
func (c *Connection) addIn(num int64) {
c.bytesIn = c.bytesIn + num
}
func (c *Connection) addOut(num int64) {
c.bytesOut = c.bytesOut + num
}
//ConnectionTable -- property
func (c *Connection) ConnectionTable() (table *Table) {
table = c.connectionTable
return
}
//CommCh -- Property
func (c *Connection) CommCh() chan bool {
return c.commCh
}
//Reader -- export the reader function
func (c *Connection) Reader() {
defer func() {
c.connectionTable.unregister <- c
c.conn.Close()
}()
c.conn.SetReadLimit(1024)
for {
_, message, err := c.conn.ReadMessage()
if err != nil {
if websocket.IsUnexpectedCloseError(err, websocket.CloseGoingAway) {
loginfo.Printf("error: %v", err)
}
break
}
loginfo.Println(hex.Dump(message))
c.addIn(int64(len(message)))
loginfo.Println(c)
}
}
//Writer -- expoer the writer function
func (c *Connection) Writer() {
defer func() {
c.conn.Close()
}()
for {
select {
case message := <-c.send:
w, err := c.conn.NextWriter(websocket.TextMessage)
if err != nil {
return
}
w.Write(message)
if err := w.Close(); err != nil {
return
}
c.addOut(int64(len(message)))
}
}
}