GitHub/ergo
3
0
Fork 0
mirror of https://github.com/ergochat/ergo.git synced 2024-11-25 13:29:27 +01:00
Code Issues Projects Releases Wiki Activity

implement candidate compromise proposal for websockets

Browse Source 
1. Text and binary frames are accepted
2. Text frames are sent by default
3. Binary frames are sent to clients who negotiate `binary.ircv3.net`
4. Non-UTF8 data is not accepted (enabling websockets still enables UTFONLY)
This commit is contained in:
Shivaram Lingamneni 2021-02-24 14:08:04 -05:00
parent 640572e151
commit d547d05205
3 changed files with 17 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
irc/client.go
View File

@ -691,7 +691,7 @@ func (client *Client) run(session *Session) {
} else if err != nil {
var quitMessage string
switch err {
case ircreader.ErrReadQ, errWSBinaryMessage:
case ircreader.ErrReadQ:
quitMessage = err.Error()
default:
quitMessage = "connection closed"

21
irc/ircconn.go
View File

@ -5,7 +5,6 @@ package irc
import (
"bytes"
"errors"
"net"
"unicode/utf8"
@ -23,7 +22,6 @@ const (
var (
crlf = []byte{'\r', '\n'}
errWSBinaryMessage = errors.New("WebSocket binary messages are unsupported")
)
// IRCConn abstracts away the distinction between a regular
@ -91,10 +89,12 @@ func (cc *IRCStreamConn) Close() (err error) {
// IRCWSConn is an IRCConn over a websocket.
type IRCWSConn struct {
conn *websocket.Conn
binary bool
}
func NewIRCWSConn(conn *websocket.Conn) IRCWSConn {
return IRCWSConn{conn: conn}
binary := conn.Subprotocol() == "binary.ircv3.net"
return IRCWSConn{conn: conn, binary: binary}
}
func (wc IRCWSConn) UnderlyingConn() *utils.WrappedConn {
@ -106,7 +106,11 @@ func (wc IRCWSConn) UnderlyingConn() *utils.WrappedConn {
func (wc IRCWSConn) WriteLine(buf []byte) (err error) {
buf = bytes.TrimSuffix(buf, crlf)
// #1483: if we have websockets at all, then we're enforcing utf8
return wc.conn.WriteMessage(websocket.TextMessage, buf)
messageType := websocket.TextMessage
if wc.binary {
messageType = websocket.BinaryMessage
}
return wc.conn.WriteMessage(messageType, buf)
}
func (wc IRCWSConn) WriteLines(buffers [][]byte) (err error) {
@ -122,11 +126,12 @@ func (wc IRCWSConn) WriteLines(buffers [][]byte) (err error) {
func (wc IRCWSConn) ReadLine() (line []byte, err error) {
messageType, line, err := wc.conn.ReadMessage()
if err == nil {
if messageType == websocket.TextMessage {
return line, nil
} else {
return nil, errWSBinaryMessage
if messageType == websocket.BinaryMessage && globalUtf8EnforcementSetting {
if !utf8.Valid(line) {
return line, errInvalidUtf8
}
}
return line, nil
} else if err == websocket.ErrReadLimit {
return line, ircreader.ErrReadQ
} else {

2
irc/listeners.go
View File

@ -166,7 +166,7 @@ func (wl *WSListener) handle(w http.ResponseWriter, r *http.Request) {
}
return false
},
Subprotocols: []string{"text.ircv3.net"},
Subprotocols: []string{"text.ircv3.net", "binary.ircv3.net"},
}
conn, err := wsUpgrader.Upgrade(w, r, nil)