From 905d4e26c6d51f39e1625cbbd3daba57062efddb Mon Sep 17 00:00:00 2001
From: licsber <admin@licsber.site>
Date: Thu, 23 Jan 2025 00:32:23 +0800
Subject: [PATCH] change rsa to ed25519.

---
 db.go   |  2 +-
 main.go | 11 ++++++-----
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/db.go b/db.go
index 451ca9b..81a6ac4 100644
--- a/db.go
+++ b/db.go
@@ -56,7 +56,7 @@ func logConnect(conn net.Conn) {
 	insertSQL := `INSERT INTO connect_attempts (time, ip, port) VALUES (?, ?, ?)`
 	_, err := db.Exec(insertSQL, currentTime, host, portInt)
 	if err != nil {
-		log.Println("Failed to insert connect attempt:", err)
+		log.Fatal("Failed to insert connect attempt:", err)
 		return
 	}
 }
diff --git a/main.go b/main.go
index 4ee1fcd..ef08431 100644
--- a/main.go
+++ b/main.go
@@ -1,8 +1,8 @@
 package main
 
 import (
+	"crypto/ed25519"
 	"crypto/rand"
-	"crypto/rsa"
 	"errors"
 	"log"
 	"net"
@@ -15,8 +15,9 @@ import (
 var (
 	errBadPassword = errors.New("permission denied")
 	serverVersions = []string{
-		"SSH-2.0-OpenSSH_6.0p1 Debian-4+deb7u2",
-		"SSH-2.0-OpenSSH_9.9",
+		"SSH-2.0-OpenSSH_9.2p1 Debian-2+deb12u4", // My Server Truely
+		"SSH-2.0-OpenSSH_6.0p1 Debian-4+deb7u2", // cowrie default
+		"SSH-2.0-OpenSSH_9.9", // Manjaro
 	}
 )
 
@@ -29,12 +30,12 @@ func main() {
 	defer db.Close()
 
 	serverConfig := &ssh.ServerConfig{
-		MaxAuthTries:     6,
+		MaxAuthTries:     3,
 		PasswordCallback: passwordCallback,
 		ServerVersion:    serverVersions[0],
 	}
 
-	privateKey, _ := rsa.GenerateKey(rand.Reader, 2048)
+	_, privateKey, _ := ed25519.GenerateKey(rand.Reader)
 	signer, _ := ssh.NewSignerFromSigner(privateKey)
 	serverConfig.AddHostKey(signer)