collect statistics on bytes read and written to sockets

Fixes #1296

irc/getters.go

@@ -64,6 +64,8 @@ type SessionData struct {
 	connInfo     string
 	sessionID    int64
 	caps         []string
+	bytesRead    uint64
+	bytesWritten uint64
 }
 
 func (client *Client) AllSessionData(currentSession *Session, hasPrivs bool) (data []SessionData, currentIndex int) {
@@ -76,6 +78,7 @@ func (client *Client) AllSessionData(currentSession *Session, hasPrivs bool) (da
 		if session == currentSession {
 			currentIndex = i
 		}
+		bytesRead, bytesWritten := session.socket.Stats()
 		data[i] = SessionData{
 			atime:        session.lastActive,
 			ctime:        session.ctime,
@@ -83,6 +86,8 @@ func (client *Client) AllSessionData(currentSession *Session, hasPrivs bool) (da
 			certfp:       session.certfp,
 			deviceID:     session.deviceID,
 			sessionID:    session.sessionID,
+			bytesRead:    bytesRead,
+			bytesWritten: bytesWritten,
 		}
 		if session.proxiedIP != nil {
 			data[i].ip = session.proxiedIP

irc/nickserv.go

@@ -1275,6 +1275,7 @@ func nsClientsListHandler(service *ircService, server *Server, client *Client, p
 				service.Notice(rb, fmt.Sprintf(client.t("IRCv3 CAPs:  %s"), capStr))
 			}
 		}
+		service.Notice(rb, fmt.Sprintf(client.t("Bytes RX/TX: %d / %d"), session.bytesRead, session.bytesWritten))
 	}
 }
 

irc/socket.go

@@ -35,6 +35,9 @@ type Socket struct {
 	sendQExceeded bool
 	finalData     []byte // what to send when we die
 	finalized     bool
+
+	bytesRead    uint64
+	bytesWritten uint64
 }
 
 // NewSocket returns a new Socket.
@@ -56,6 +59,12 @@ func (socket *Socket) Close() {
 	socket.wakeWriter()
 }
 
+func (socket *Socket) Stats() (bytesRead, bytesWritten uint64) {
+	socket.Lock()
+	defer socket.Unlock()
+	return socket.bytesRead, socket.bytesWritten
+}
+
 // Read returns a single IRC line from a Socket.
 func (socket *Socket) Read() (string, error) {
 	// immediately fail if Close() has been called, even if there's
@@ -67,6 +76,10 @@ func (socket *Socket) Read() (string, error) {
 	lineBytes, err := socket.conn.ReadLine()
 	line := string(lineBytes)
 
+	socket.Lock()
+	socket.bytesRead += uint64(len(lineBytes))
+	socket.Unlock()
+
 	if err == io.EOF {
 		socket.Close()
 	}
@@ -96,6 +109,7 @@ func (socket *Socket) Write(data []byte) (err error) {
 		} else {
 			socket.buffers = append(socket.buffers, data)
 			socket.totalLength = prospectiveLen
+			socket.bytesWritten += uint64(len(data))
 		}
 	}
 	socket.Unlock()
@@ -136,6 +150,10 @@ func (socket *Socket) BlockingWrite(data []byte) (err error) {
 		return io.EOF
 	}
 
+	socket.Lock()
+	socket.bytesWritten += uint64(len(data))
+	socket.Unlock()
+
 	err = socket.conn.WriteLine(data)
 	if err != nil {
 		socket.finalize()