third-party-mirrors/ssh-chat
1
0
Fork 0
mirror of https://github.com/shazow/ssh-chat.git synced 2025-04-15 08:30:36 +03:00
Code Issues Projects Releases Wiki Activity

Configurable rate limiting for sshd

Browse Source
This commit is contained in:
Andrey Petrov 2015-01-16 12:35:57 -08:00
parent b99083ee6e
commit cc25d17bdc
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
cmd.go
View File

@ -102,6 +102,7 @@ func main() {
os.Exit(4) os.Exit(4)
} }
defer s.Close() defer s.Close()
s.RateLimit = true
fmt.Printf("Listening for connections on %v\n", s.Addr().String()) fmt.Printf("Listening for connections on %v\n", s.Addr().String())

11
sshd/net.go
View File

@ -11,7 +11,8 @@ import (
// Container for the connection and ssh-related configuration // Container for the connection and ssh-related configuration
type SSHListener struct { type SSHListener struct {
net.Listener net.Listener
config *ssh.ServerConfig config *ssh.ServerConfig
RateLimit bool
} }
// Make an SSH listener socket // Make an SSH listener socket
@ -20,13 +21,17 @@ func ListenSSH(laddr string, config *ssh.ServerConfig) (*SSHListener, error) {
if err != nil { if err != nil {
return nil, err return nil, err
} }
l := SSHListener{socket, config} l := SSHListener{Listener: socket, config: config}
return &l, nil return &l, nil
} }
func (l *SSHListener) handleConn(conn net.Conn) (*Terminal, error) { func (l *SSHListener) handleConn(conn net.Conn) (*Terminal, error) {
if l.RateLimit {
// TODO: Configurable Limiter?
conn = ReadLimitConn(conn, rateio.NewGracefulLimiter(1000, time.Minute*2, time.Second*3))
}
// Upgrade TCP connection to SSH connection // Upgrade TCP connection to SSH connection
conn = ReadLimitConn(conn, rateio.NewGracefulLimiter(1000, time.Minute*2, time.Second*3))
sshConn, channels, requests, err := ssh.NewServerConn(conn, l.config) sshConn, channels, requests, err := ssh.NewServerConn(conn, l.config)
if err != nil { if err != nil {
return nil, err return nil, err