From d317e8f6df0e0e16445db606da1d683a6b35f531 Mon Sep 17 00:00:00 2001
From: Alexander Neonxp Kiryukhin <i@neonxp.ru>
Date: Tue, 30 Dec 2025 19:33:39 +0300
Subject: =?UTF-8?q?=D0=BD=D0=B0=D1=87=D0=B0=D0=BB=D1=8C=D0=BD=D1=8B=D0=B9?=
 =?UTF-8?q?=20=D0=BA=D0=BE=D0=BC=D0=BC=D0=B8=D1=82?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 internal/server/conn.go | 67 +++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)
 create mode 100644 internal/server/conn.go

(limited to 'internal/server/conn.go')

diff --git a/internal/server/conn.go b/internal/server/conn.go
new file mode 100644
index 0000000..1069b0b
--- /dev/null
+++ b/internal/server/conn.go
@@ -0,0 +1,67 @@
+package server
+
+import (
+	"context"
+	"fmt"
+	"log/slog"
+	"net"
+	"sync"
+
+	"golang.org/x/crypto/ssh"
+)
+
+func (s *Server) serveConn(ctx context.Context, nConn net.Conn, config *ssh.ServerConfig) error {
+	conn, chans, reqs, err := ssh.NewServerConn(nConn, config)
+	if err != nil {
+		return fmt.Errorf("failed to handshake: %w", err)
+	}
+	slog.Info("user connected", slog.Any("user", conn.User()), slog.String("ip", conn.RemoteAddr().String()))
+
+	var wg sync.WaitGroup
+	defer wg.Wait()
+
+	wg.Go(func() {
+		ssh.DiscardRequests(reqs)
+	})
+
+	for newChannel := range chans {
+		if newChannel.ChannelType() != "session" {
+			newChannel.Reject(ssh.UnknownChannelType, "unknown channel type")
+			continue
+		}
+		channel, requests, err := newChannel.Accept()
+		if err != nil {
+			return fmt.Errorf("could not accept channel: %w", err)
+		}
+
+		wg.Go(func() {
+			for req := range requests {
+				switch req.Type {
+				case "pty-req":
+					req.Reply(true, nil)
+				case "shell":
+					req.Reply(true, nil)
+				default:
+					req.Reply(false, nil)
+				}
+				slog.Debug(
+					"req",
+					slog.String("type", req.Type),
+					slog.Bool("want-reply", req.WantReply),
+					slog.String("payload", string(req.Payload)),
+				)
+			}
+		})
+
+		wg.Go(func() {
+			identify := conn.Permissions.ExtraData["identify"].(string)
+			user := s.chat.NewUser(conn.User(), identify)
+			slog.Info("joined", slog.String("user", user.NUsername()))
+			s.serveClient(ctx, channel, user)
+			slog.Info("disconnected", slog.String("user", user.NUsername()))
+			conn.Close()
+		})
+	}
+
+	return nil
+}
-- 
cgit v1.2.3