mirror of
				https://gitee.com/gitea/gitea
				synced 2025-11-04 00:20:25 +08:00 
			
		
		
		
	Refactor SSH init code, fix directory creation for TrustedUserCAKeys file (#20299)
* Refactor SSH init code, fix directory creation for TrustedUserCAKeys file * Update modules/ssh/init.go Co-authored-by: zeripath <art27@cantab.net> * fix lint copyright * Update modules/ssh/init.go Co-authored-by: zeripath <art27@cantab.net> Co-authored-by: Lunny Xiao <xiaolunwen@gmail.com>
This commit is contained in:
		
							
								
								
									
										55
									
								
								modules/ssh/init.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										55
									
								
								modules/ssh/init.go
									
									
									
									
									
										Normal file
									
								
							@@ -0,0 +1,55 @@
 | 
			
		||||
// Copyright 2022 The Gitea Authors. All rights reserved.
 | 
			
		||||
// Use of this source code is governed by a MIT-style
 | 
			
		||||
// license that can be found in the LICENSE file.
 | 
			
		||||
 | 
			
		||||
package ssh
 | 
			
		||||
 | 
			
		||||
import (
 | 
			
		||||
	"fmt"
 | 
			
		||||
	"net"
 | 
			
		||||
	"os"
 | 
			
		||||
	"path/filepath"
 | 
			
		||||
	"strconv"
 | 
			
		||||
	"strings"
 | 
			
		||||
 | 
			
		||||
	"code.gitea.io/gitea/modules/log"
 | 
			
		||||
	"code.gitea.io/gitea/modules/setting"
 | 
			
		||||
)
 | 
			
		||||
 | 
			
		||||
func Init() error {
 | 
			
		||||
	if setting.SSH.Disabled {
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if setting.SSH.StartBuiltinServer {
 | 
			
		||||
		Listen(setting.SSH.ListenHost, setting.SSH.ListenPort, setting.SSH.ServerCiphers, setting.SSH.ServerKeyExchanges, setting.SSH.ServerMACs)
 | 
			
		||||
		log.Info("SSH server started on %s. Cipher list (%v), key exchange algorithms (%v), MACs (%v)",
 | 
			
		||||
			net.JoinHostPort(setting.SSH.ListenHost, strconv.Itoa(setting.SSH.ListenPort)),
 | 
			
		||||
			setting.SSH.ServerCiphers, setting.SSH.ServerKeyExchanges, setting.SSH.ServerMACs,
 | 
			
		||||
		)
 | 
			
		||||
		return nil
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	builtinUnused()
 | 
			
		||||
 | 
			
		||||
	// FIXME: why 0o644 for a directory .....
 | 
			
		||||
	if err := os.MkdirAll(setting.SSH.KeyTestPath, 0o644); err != nil {
 | 
			
		||||
		return fmt.Errorf("failed to create directory %q for ssh key test: %w", setting.SSH.KeyTestPath, err)
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	if len(setting.SSH.TrustedUserCAKeys) > 0 && setting.SSH.AuthorizedPrincipalsEnabled {
 | 
			
		||||
		caKeysFileName := setting.SSH.TrustedUserCAKeysFile
 | 
			
		||||
		caKeysFileDir := filepath.Dir(caKeysFileName)
 | 
			
		||||
 | 
			
		||||
		err := os.MkdirAll(caKeysFileDir, 0o700) // SSH.RootPath by default (That is `~/.ssh` in most cases)
 | 
			
		||||
		if err != nil {
 | 
			
		||||
			return fmt.Errorf("failed to create directory %q for ssh trusted ca keys: %w", caKeysFileDir, err)
 | 
			
		||||
		}
 | 
			
		||||
 | 
			
		||||
		if err := os.WriteFile(caKeysFileName, []byte(strings.Join(setting.SSH.TrustedUserCAKeys, "\n")), 0o600); err != nil {
 | 
			
		||||
			return fmt.Errorf("failed to write ssh trusted ca keys to %q: %w", caKeysFileName, err)
 | 
			
		||||
		}
 | 
			
		||||
	}
 | 
			
		||||
 | 
			
		||||
	return nil
 | 
			
		||||
}
 | 
			
		||||
@@ -29,7 +29,7 @@ func listen(server *ssh.Server) {
 | 
			
		||||
	log.Info("SSH Listener: %s Closed", server.Addr)
 | 
			
		||||
}
 | 
			
		||||
 | 
			
		||||
// Unused informs our cleanup routine that we will not be using a ssh port
 | 
			
		||||
func Unused() {
 | 
			
		||||
// builtinUnused informs our cleanup routine that we will not be using a ssh port
 | 
			
		||||
func builtinUnused() {
 | 
			
		||||
	graceful.GetManager().InformCleanup()
 | 
			
		||||
}
 | 
			
		||||
 
 | 
			
		||||
		Reference in New Issue
	
	Block a user