From 65302d3c1e8e990080e5309ad7a9f6fef43aa685 Mon Sep 17 00:00:00 2001
From: Simon Ser <contact@emersion.fr>
Date: Tue, 11 Aug 2020 10:36:14 +0200
Subject: [PATCH] Add an ident server

Closes: https://todo.sr.ht/~emersion/soju/69
---
 cmd/soju/main.go | 16 ++++++++++++++++
 server.go        |  1 +
 user.go          | 22 ++++++++++++++++++++++
 3 files changed, 39 insertions(+)

diff --git a/cmd/soju/main.go b/cmd/soju/main.go
index 976a3d0..ad670f1 100644
--- a/cmd/soju/main.go
+++ b/cmd/soju/main.go
@@ -124,6 +124,22 @@ func main() {
 			go func() {
 				log.Fatal(httpSrv.ListenAndServe())
 			}()
+		case "ident":
+			if srv.Identd == nil {
+				srv.Identd = soju.NewIdentd()
+			}
+
+			host := u.Host
+			if _, _, err := net.SplitHostPort(host); err != nil {
+				host = host + ":113"
+			}
+			ln, err := net.Listen("tcp", host)
+			if err != nil {
+				log.Fatalf("failed to start listener on %q: %v", listen, err)
+			}
+			go func() {
+				log.Fatal(srv.Identd.Serve(ln))
+			}()
 		default:
 			log.Fatalf("failed to listen on %q: unsupported scheme", listen)
 		}
diff --git a/server.go b/server.go
index bb93106..883b9b5 100644
--- a/server.go
+++ b/server.go
@@ -51,6 +51,7 @@ type Server struct {
 	Debug          bool
 	HTTPOrigins    []string
 	AcceptProxyIPs config.IPSet
+	Identd         *Identd // can be nil
 
 	db *DB
 
diff --git a/user.go b/user.go
index c9c3fb8..0eec316 100644
--- a/user.go
+++ b/user.go
@@ -1,6 +1,9 @@
 package soju
 
 import (
+	"crypto/sha256"
+	"encoding/base64"
+	"encoding/binary"
 	"fmt"
 	"time"
 
@@ -99,6 +102,17 @@ func (net *network) isStopped() bool {
 	}
 }
 
+func userIdent(u *User) string {
+	// The ident is a string we will send to upstream servers in clear-text.
+	// For privacy reasons, make sure it doesn't expose any meaningful user
+	// metadata. We just use the base64-encoded hashed ID, so that people don't
+	// start relying on the string being an integer or following a pattern.
+	var b [64]byte
+	binary.LittleEndian.PutUint64(b[:], uint64(u.ID))
+	h := sha256.Sum256(b[:])
+	return base64.RawStdEncoding.EncodeToString(h[:])
+}
+
 func (net *network) run() {
 	var lastTry time.Time
 	for {
@@ -120,6 +134,10 @@ func (net *network) run() {
 			continue
 		}
 
+		if net.user.srv.Identd != nil {
+			net.user.srv.Identd.Store(uc.RemoteAddr().String(), uc.LocalAddr().String(), userIdent(&net.user.User))
+		}
+
 		uc.register()
 		if err := uc.runUntilRegistered(); err != nil {
 			uc.logger.Printf("failed to register: %v", err)
@@ -138,6 +156,10 @@ func (net *network) run() {
 		}
 		uc.Close()
 		net.user.events <- eventUpstreamDisconnected{uc}
+
+		if net.user.srv.Identd != nil {
+			net.user.srv.Identd.Delete(uc.RemoteAddr().String(), uc.LocalAddr().String())
+		}
 	}
 }