Pārlūkot izejas kodu

make the server banner configurable

Philip Hofstetter 6 gadi atpakaļ
vecāks
revīzija
88a288ccfe
2 mainītis faili ar 3 papildinājumiem un 1 dzēšanām
  1. 1 0
      config/config.go
  2. 2 1
      sftpd/server.go

+ 1 - 0
config/config.go

@@ -28,6 +28,7 @@ func init() {
 	// create a default configuration to use if no config file is provided
 	// create a default configuration to use if no config file is provided
 	globalConf = globalConfig{
 	globalConf = globalConfig{
 		SFTPD: sftpd.Configuration{
 		SFTPD: sftpd.Configuration{
+			Banner:       "SFTPServer",
 			BindPort:     2022,
 			BindPort:     2022,
 			BindAddress:  "",
 			BindAddress:  "",
 			IdleTimeout:  15,
 			IdleTimeout:  15,

+ 2 - 1
sftpd/server.go

@@ -27,6 +27,7 @@ import (
 
 
 // Configuration server configuration
 // Configuration server configuration
 type Configuration struct {
 type Configuration struct {
+	Banner       string  `json:"banner"`
 	BindPort     int     `json:"bind_port"`
 	BindPort     int     `json:"bind_port"`
 	BindAddress  string  `json:"bind_address"`
 	BindAddress  string  `json:"bind_address"`
 	IdleTimeout  int     `json:"idle_timeout"`
 	IdleTimeout  int     `json:"idle_timeout"`
@@ -63,7 +64,7 @@ func (c Configuration) Initialize(configDir string) error {
 
 
 			return sp, nil
 			return sp, nil
 		},
 		},
-		ServerVersion: "SSH-2.0-SFTPServer",
+		ServerVersion: "SSH-2.0-" + c.Banner,
 	}
 	}
 
 
 	if _, err := os.Stat(filepath.Join(configDir, "id_rsa")); os.IsNotExist(err) {
 	if _, err := os.Stat(filepath.Join(configDir, "id_rsa")); os.IsNotExist(err) {