Improve new field on Event, and refactored the Protocol field with a Protocol enum

This commit is contained in:
Mario
2022-05-10 22:50:29 +02:00
parent 98f9835079
commit 24ff11099c
2 changed files with 26 additions and 11 deletions

View File

@ -28,7 +28,7 @@ func (SSHStrategy *SecureShellStrategy) Init(beelzebubServiceConfiguration parse
tr.TraceEvent(tracer.Event{ tr.TraceEvent(tracer.Event{
Msg: "New SSH Session", Msg: "New SSH Session",
Protocol: beelzebubServiceConfiguration.Protocol, Protocol: tracer.SSH,
RemoteAddr: sess.RemoteAddr().String(), RemoteAddr: sess.RemoteAddr().String(),
Status: tracer.Start, Status: tracer.Start,
ID: uuidSession.String(), ID: uuidSession.String(),

View File

@ -2,6 +2,7 @@ package tracer
import ( import (
log "github.com/sirupsen/logrus" log "github.com/sirupsen/logrus"
"net/http"
) )
type Tracer struct { type Tracer struct {
@ -20,7 +21,7 @@ func (tracer *Tracer) TraceEvent(event Event) {
type Event struct { type Event struct {
RemoteAddr string RemoteAddr string
Protocol string Protocol Protocol
Command string Command string
Status Status Status Status
Msg string Msg string
@ -29,8 +30,22 @@ type Event struct {
User string User string
Password string Password string
Client string Client string
Headers http.Header
Cookies []*http.Cookie
UserAgent string
HostHTTPRequest string
Body string
HTTPMethod string
RequestURI string
} }
type Protocol int
const (
HTTP Protocol = iota
SSH
)
type Status int type Status int
const ( const (