lordwelch/breakglass
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add an optional banner to greet the user

Browse Source
This commit is contained in:
Christian Heusel 2022-03-06 23:06:23 +01:00 committed by Michael Stapelberg
parent ae89d9b070
commit f41ca45656
1 changed files with 18 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
breakglass.go
View File

@ -36,6 +36,10 @@ var (
"22", "22",
"port for breakglass to listen on") "port for breakglass to listen on")
enableBanner = flag.Bool("enable_banner",
false,
"Adds a banner to greet the user on login")
forwarding = flag.String("forward", forwarding = flag.String("forward",
"", "",
"allow port forwarding. Use `loopback` for loopback interfaces and `private-network` for private networks") "allow port forwarding. Use `loopback` for loopback interfaces and `private-network` for private networks")
@ -131,6 +135,20 @@ func main() {
} }
return nil, fmt.Errorf("public key not found in %s", *authorizedKeysPath) return nil, fmt.Errorf("public key not found in %s", *authorizedKeysPath)
}, },
BannerCallback: func(conn ssh.ConnMetadata) string {
if !*enableBanner {
return ""
}
bannerMessage := fmt.Sprintf("#\n# Welcome to gokrazy, %s!\n", conn.User())
bannerInfo := fmt.Sprintf("# This installation is running on a %q!\n#\n", gokrazy.Model())
maxChars := len(bannerInfo)
if maxChars < len(bannerMessage) {
maxChars = len(bannerMessage)
}
border := strings.Repeat("#", maxChars) + "\n"
bannerMessage = border + bannerMessage + bannerInfo + border
return bannerMessage
},
} }
signer, err := loadHostKey(*hostKeyPath) signer, err := loadHostKey(*hostKeyPath)