chrissewell/beelzebub
1
0
Fork 0
mirror of https://github.com/mariocandela/beelzebub.git synced 2025-07-01 18:47:26 -04:00
Code Issues Packages Projects Releases Wiki Activity

Configure tracer on secureShellStrategy.go

Browse Source
This commit is contained in:
Mario
2022-05-09 23:18:20 +02:00
parent 4b8f74d18c
commit b43d09ecf2
1 changed files with 34 additions and 41 deletions
Download Patch File Download Diff File Expand all files Collapse all files

75
protocols/secureShellStrategy.go
View File

@ -2,19 +2,21 @@ package protocols
import ( import (
"beelzebub/parser" "beelzebub/parser"
"beelzebub/tracer"
"fmt" "fmt"
"github.com/gliderlabs/ssh" "github.com/gliderlabs/ssh"
"github.com/google/uuid" "github.com/google/uuid"
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
"golang.org/x/crypto/ssh/terminal" "golang.org/x/crypto/ssh/terminal"
"regexp" "regexp"
"strings"
"time" "time"
) )
type SecureShellStrategy struct { type SecureShellStrategy struct {
} }
func (SSHStrategy *SecureShellStrategy) Init(beelzebubServiceConfiguration parser.BeelzebubServiceConfiguration) error { func (SSHStrategy *SecureShellStrategy) Init(beelzebubServiceConfiguration parser.BeelzebubServiceConfiguration, tr tracer.Tracer) error {
go func() { go func() {
server := &ssh.Server{ server := &ssh.Server{
Addr: beelzebubServiceConfiguration.Address, Addr: beelzebubServiceConfiguration.Address,
@ -23,14 +25,30 @@ func (SSHStrategy *SecureShellStrategy) Init(beelzebubServiceConfiguration parse
Version: beelzebubServiceConfiguration.ServerVersion, Version: beelzebubServiceConfiguration.ServerVersion,
Handler: func(sess ssh.Session) { Handler: func(sess ssh.Session) {
uuidSession := uuid.New() uuidSession := uuid.New()
traceSessionStart(sess, uuidSession)
tr.TraceEvent(tracer.Event{
Msg: "New SSH Session",
Protocol: beelzebubServiceConfiguration.Protocol,
RemoteAddr: sess.RemoteAddr().String(),
Status: tracer.Start,
ID: uuidSession.String(),
Environ: strings.Join(sess.Environ(), ","),
User: sess.User(),
})
term := terminal.NewTerminal(sess, buildPrompt(sess.User(), beelzebubServiceConfiguration.ServerName)) term := terminal.NewTerminal(sess, buildPrompt(sess.User(), beelzebubServiceConfiguration.ServerName))
for { for {
commandInput, err := term.ReadLine() commandInput, err := term.ReadLine()
if err != nil { if err != nil {
break break
} }
traceCommand(commandInput, uuidSession) tr.TraceEvent(tracer.Event{
Msg: "New SSH Command",
RemoteAddr: sess.RemoteAddr().String(),
Status: tracer.Interaction,
Command: commandInput,
ID: uuidSession.String(),
})
if commandInput == "exit" { if commandInput == "exit" {
break break
} }
@ -47,10 +65,21 @@ func (SSHStrategy *SecureShellStrategy) Init(beelzebubServiceConfiguration parse
} }
} }
} }
traceSessionEnd(sess, uuidSession) tr.TraceEvent(tracer.Event{
Msg: "End SSH Session",
Status: tracer.End,
ID: uuidSession.String(),
})
}, },
PasswordHandler: func(ctx ssh.Context, password string) bool { PasswordHandler: func(ctx ssh.Context, password string) bool {
traceAttempt(ctx, password) tr.TraceEvent(tracer.Event{
Msg: "New SSH attempt",
Status: tracer.Stateless,
User: ctx.User(),
Password: password,
Client: ctx.ClientVersion(),
RemoteAddr: ctx.RemoteAddr().String(),
})
matched, err := regexp.MatchString(beelzebubServiceConfiguration.PasswordRegex, password) matched, err := regexp.MatchString(beelzebubServiceConfiguration.PasswordRegex, password)
if err != nil { if err != nil {
log.Errorf("Error regex: %s, %s", beelzebubServiceConfiguration.PasswordRegex, err.Error()) log.Errorf("Error regex: %s, %s", beelzebubServiceConfiguration.PasswordRegex, err.Error())
@ -75,39 +104,3 @@ func (SSHStrategy *SecureShellStrategy) Init(beelzebubServiceConfiguration parse
func buildPrompt(user string, serverName string) string { func buildPrompt(user string, serverName string) string {
return fmt.Sprintf("%s@%s:~$ ", user, serverName) return fmt.Sprintf("%s@%s:~$ ", user, serverName)
} }
func traceAttempt(ctx ssh.Context, password string) {
log.WithFields(log.Fields{
"remoteAddr": ctx.RemoteAddr(),
"user": ctx.User(),
"password": password,
"client": ctx.ClientVersion(),
}).Info("New SSH attempt")
}
func traceSessionStart(sess ssh.Session, uuidSession uuid.UUID) {
log.WithFields(log.Fields{
"uuidSession": uuidSession,
"remoteAddr": sess.RemoteAddr(),
"command": sess.Command(),
"environ": sess.Environ(),
"user": sess.User(),
}).Info("New SSH Session")
}
func traceSessionEnd(sess ssh.Session, uuidSession uuid.UUID) {
log.WithFields(log.Fields{
"uuidSession": uuidSession,
"remoteAddr": sess.RemoteAddr(),
"command": sess.Command(),
"environ": sess.Environ(),
"user": sess.User(),
}).Info("End SSH Session")
}
func traceCommand(command string, uuidSession uuid.UUID) {
log.WithFields(log.Fields{
"uuidSession": uuidSession,
"command": command,
}).Info("New SSH Command")
}