Revert "add timeouts to mysql and Kafka application discovery"

cmd/scan.go

@@ -1,7 +1,6 @@
 package cmd
 
 import (
-	"context"
 	"encoding/json"
 	"fmt"
 	"net"
@@ -80,7 +79,7 @@ func scan(cmd *cobra.Command, args []string) error {
 	for _, target := range scanResults {
 		// Perform service discovery for open TCP ports
 		for _, port := range target.TCPPorts {
-			discoveryResult, err := ScanTargets(context.Background(), target.Host, port)
+			discoveryResult, err := ScanTargets(target.Host, port)
 			if err != nil {
 				fmt.Fprintf(os.Stderr, "Error while discovering services on %s:%d: %s\n", target.Host, port, err)
 				continue
@@ -112,7 +111,7 @@ func scan(cmd *cobra.Command, args []string) error {
 		}
 		// Perform service discovery for open UDP ports
 		for _, port := range target.UDPPorts {
-			discoveryResult, err := ScanTargets(context.Background(), target.Host, port)
+			discoveryResult, err := ScanTargets(target.Host, port)
 			if err != nil {
 				fmt.Fprintf(os.Stderr, "Error while discovering services on %s:%d: %s\n", target.Host, port, err)
 				continue

cmd/servicediscovery.go

@@ -1,7 +1,6 @@
 package cmd
 
 import (
-	"context"
 	"fmt"
 	"io"
 	"sync"
@@ -22,7 +21,7 @@ type DiscoveryResult struct {
 	Properties        map[string]interface{}
 }
 
-func ScanTargets(ctx context.Context, host string, port int) (result DiscoveryResult, err error) {
+func ScanTargets(host string, port int) (result DiscoveryResult, err error) {
 	var sessionWg sync.WaitGroup
 	var presentationWg sync.WaitGroup
 	var applicationWg sync.WaitGroup
@@ -106,7 +105,6 @@ func ScanTargets(ctx context.Context, host string, port int) (result DiscoveryRe
 								if err != nil {
 									return
 								}
-
 								applicationLayerChan <- applicationDiscoveryResult
 							}(applicationDiscoveryItem)
 						}
@@ -127,11 +125,10 @@ func ScanTargets(ctx context.Context, host string, port int) (result DiscoveryRe
 							break // Stop checking application layer protocol
 						}
 					}
+
 					break // Stop checking presentation layer protocols
 				}
-
 			}
-
 			if presentationDiscoveryResult == nil || !presentationDiscoveryResult.GetIsDetected() {
 				// Continue to discover application layer protocols
 				applicationLayerChan := make(chan applicationLayerDiscoveryResult)

pkg/networkscanner/servicediscovery/applicationlayerdiscovery/al_kafka.go

@@ -47,24 +47,19 @@ func (k *KafkaDiscovery) Discover(sessionHandler servicediscovery.ISessionHandle
 	// Configure the producer
 	config := sarama.NewConfig()
 	config.Producer.RequiredAcks = sarama.WaitForAll
-	// add a more strict connection timeout
-	config.Net.ReadTimeout = 10 * time.Second
-	config.Net.WriteTimeout = 10 * time.Second
 	config.Producer.Retry.Max = 1
 	config.Producer.Timeout = 500 * time.Millisecond
 	config.Producer.Return.Successes = true
 
 	// Create a new SyncProducer
 	producer, err := sarama.NewSyncProducer(brokerList, config)
-
 	if err != nil {
 		return &KafkaDiscoveryResult{
 			isDetected:      false,
 			isAuthenticated: true,
 			properties:      nil, // Set properties to nil as it's not used in this case
 		}, err
 	}
-
 	defer func() {
 		if err := producer.Close(); err != nil {
 			log.Debugf("Failed to close Kafka producer: %s", err)

pkg/networkscanner/servicediscovery/applicationlayerdiscovery/al_mysql.go

@@ -1,12 +1,10 @@
 package applicationlayerdiscovery
 
 import (
-	"context"
 	"fmt"
 	"io"
 	"log"
 	"strings"
-	"time"
 
 	"database/sql"
 
@@ -57,10 +55,8 @@ func (d *MysqlDiscovery) Discover(sessionHandler servicediscovery.ISessionHandle
 		}, err
 	}
 
-	// Ping the server with passed context()
-	ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
-	defer cancel()
-	err = db.PingContext(ctx)
+	// Ping the server
+	err = db.Ping()
 	if err != nil {
 		if strings.Contains(err.Error(), "Access denied") {
 			return &MysqlDiscoveryResult{