netbird/relay/client/dialer/ws/conn.go

65 lines
1.1 KiB
Go
Raw Normal View History

2024-06-28 11:17:06 +02:00
package ws
import (
"context"
"fmt"
"net"
"time"
"nhooyr.io/websocket"
)
type Conn struct {
ctx context.Context
2024-06-19 18:16:23 +02:00
*websocket.Conn
}
2024-07-02 11:57:17 +02:00
func NewConn(wsConn *websocket.Conn) net.Conn {
return &Conn{
2024-07-02 11:57:17 +02:00
ctx: context.Background(),
Conn: wsConn,
}
}
func (c *Conn) Read(b []byte) (n int, err error) {
t, ioReader, err := c.Conn.Reader(c.ctx)
if err != nil {
return 0, err
}
if t != websocket.MessageBinary {
return 0, fmt.Errorf("unexpected message type")
}
return ioReader.Read(b)
}
func (c *Conn) Write(b []byte) (n int, err error) {
err = c.Conn.Write(c.ctx, websocket.MessageBinary, b)
return len(b), err
}
func (c *Conn) RemoteAddr() net.Addr {
return WebsocketAddr{}
}
func (c *Conn) LocalAddr() net.Addr {
return WebsocketAddr{}
}
func (c *Conn) SetReadDeadline(t time.Time) error {
2024-07-09 16:29:38 +02:00
return fmt.Errorf("SetReadDeadline is not implemented")
}
func (c *Conn) SetWriteDeadline(t time.Time) error {
2024-07-09 16:29:38 +02:00
return fmt.Errorf("SetWriteDeadline is not implemented")
}
func (c *Conn) SetDeadline(t time.Time) error {
2024-07-09 16:29:38 +02:00
return fmt.Errorf("SetDeadline is not implemented")
}
func (c *Conn) Close() error {
2024-06-09 12:41:52 +02:00
return c.Conn.CloseNow()
}