Improve initial TCP honeypot structure

configurations/beelzebub.yaml

@@ -5,5 +5,5 @@ core:
     logDisableTimestamp: true
     logsPath: ./logs
   tracing:
-    rabbitMQEnabled: true
-    rabbitMQURI: "amqp://beelzebub:b33lz3b4b@localhost:8888/"
+    rabbitMQEnabled: false
+    rabbitMQURI: ""

configurations/services/tcp-3306.yaml

@@ -2,6 +2,5 @@ apiVersion: "v1"
 protocol: "tcp"
 address: ":3306"
 description: "Mysql"
-commands:
-  - regex: "hello"
-    handler: "WORLD"
+banner: "mysql 4.0"
+deadlineTimeoutSeconds: 60

parser/configurationsParser.go

@@ -38,6 +38,7 @@ type BeelzebubServiceConfiguration struct {
 	DeadlineTimeoutSeconds int       `yaml:"deadlineTimeoutSeconds"`
 	PasswordRegex          string    `yaml:"passwordRegex"`
 	Description            string    `yaml:"description"`
+	Banner                 string    `yaml:"banner"`
 }
 
 type Command struct {

protocols/transmissionControlProtocolStrategy.go

@@ -0,0 +1,49 @@
+package protocols
+
+import (
+	"beelzebub/parser"
+	"beelzebub/tracer"
+	log "github.com/sirupsen/logrus"
+	"net"
+	"time"
+)
+
+type TransmissionControlProtocolStrategy struct {
+}
+
+func (TCPStrategy *TransmissionControlProtocolStrategy) Init(beelzebubServiceConfiguration parser.BeelzebubServiceConfiguration, tr tracer.Tracer) error {
+	listen, err := net.Listen("TCP", beelzebubServiceConfiguration.Address)
+	if err != nil {
+		log.Errorf("Error during init TCP Protocol: %s", err.Error())
+		return err
+	}
+	defer listen.Close()
+
+	go func() {
+		for {
+			if conn, err := listen.Accept(); err == nil {
+				conn.SetDeadline(time.Now().Add(time.Duration(beelzebubServiceConfiguration.DeadlineTimeoutSeconds) * time.Second))
+				go handleIncomingRequest(conn)
+			}
+		}
+	}()
+
+	log.WithFields(log.Fields{
+		"port":   beelzebubServiceConfiguration.Address,
+		"banner": beelzebubServiceConfiguration.Banner,
+	}).Infof("Init service %s", beelzebubServiceConfiguration.Protocol)
+	return nil
+}
+
+func handleIncomingRequest(conn net.Conn) {
+	buffer := make([]byte, 1024)
+	_, err := conn.Read(buffer)
+	if err != nil {
+		log.Fatal(err)
+	}
+	// respond
+	conn.Write([]byte("Hi back!\n"))
+
+	// close conn
+	conn.Close()
+}