honeypot/ports/server_test.go

@@ -18,12 +18,11 @@ func TestServer(t *testing.T) {
 	server := ports.New(store)
 	server.IP = "127.0.0.1"
 
-	server.AddTCPPort("2555")
+	server.AddTCPPort("25")
 
 	go server.Start(ctx)
-	time.Sleep(1 * time.Second)
 
-	rAddr, err := net.ResolveTCPAddr("tcp", net.JoinHostPort(server.IP, "2555"))
+	rAddr, err := net.ResolveTCPAddr("tcp", net.JoinHostPort(server.IP, "25"))
 	if err != nil {
 		t.Fatalf("Error resolving remote address: %s", err)
 	}

honeypot/ssh/store/cache.go

@@ -2,8 +2,6 @@ package store
 
 import "git.t-juice.club/torjus/apiary/models"
 
-var _ LoginAttemptStore = &CachingStore{}
-
 type CachingStore struct {
 	backend LoginAttemptStore
 
@@ -35,16 +33,14 @@ func NewCachingStore(backend LoginAttemptStore) *CachingStore {
 		//TODO: Handle better maybe?
 		panic(err)
 	}
-	var loginCount int
 
-	for attempt := range all {
+	cs.totalLoginsCount = len(all)
+	for _, attempt := range all {
 		cs.uniqueUsernames[attempt.Username] = struct{}{}
 		cs.uniquePasswords[attempt.Password] = struct{}{}
 		cs.uniqueIPs[attempt.RemoteIP.String()] = struct{}{}
 		cs.uniqueCountries[attempt.Country] = struct{}{}
-		loginCount++
 	}
-	cs.totalLoginsCount = loginCount
 
 	return cs
 }
@@ -62,7 +58,7 @@ func (s *CachingStore) AddAttempt(l *models.LoginAttempt) error {
 	return s.backend.AddAttempt(l)
 }
 
-func (s *CachingStore) All() (<-chan models.LoginAttempt, error) {
+func (s *CachingStore) All() ([]models.LoginAttempt, error) {
 	return s.backend.All()
 }
 

honeypot/ssh/store/memory.go

@@ -9,8 +9,6 @@ import (
 	"git.t-juice.club/torjus/apiary/models"
 )
 
-var _ LoginAttemptStore = &MemoryStore{}
-
 type MemoryStore struct {
 	lock      sync.RWMutex
 	attempts  []models.LoginAttempt
@@ -34,17 +32,8 @@ func (ms *MemoryStore) AddAttempt(l *models.LoginAttempt) error {
 	return nil
 }
 
-func (ms *MemoryStore) All() (<-chan models.LoginAttempt, error) {
+func (ms *MemoryStore) All() ([]models.LoginAttempt, error) {
-	ch := make(chan models.LoginAttempt)
+	return ms.attempts, nil
-	go func() {
-		ms.lock.RLock()
-		defer ms.lock.RUnlock()
-		for _, attempt := range ms.attempts {
-			ch <- attempt
-		}
-		close(ch)
-	}()
-	return ch, nil
 }
 
 func (ms *MemoryStore) Stats(statType LoginStats, limit int) ([]StatsResult, error) {

honeypot/ssh/store/metrics.go

@@ -8,8 +8,6 @@ import (
 	"github.com/prometheus/client_golang/prometheus"
 )
 
-var _ LoginAttemptStore = &MetricsCollectingStore{}
-
 const tickDuration = 5 * time.Second
 
 type MetricsCollectingStore struct {
@@ -91,7 +89,7 @@ func (s *MetricsCollectingStore) AddAttempt(l *models.LoginAttempt) error {
 	return err
 }
 
-func (s *MetricsCollectingStore) All() (<-chan models.LoginAttempt, error) {
+func (s *MetricsCollectingStore) All() ([]models.LoginAttempt, error) {
 	return s.store.All()
 }
 

honeypot/ssh/store/postgres.go

@@ -9,8 +9,6 @@ import (
 	_ "github.com/jackc/pgx/v4/stdlib"
 )
 
-var _ LoginAttemptStore = &PostgresStore{}
-
 const DBSchema = `
 CREATE TABLE IF NOT EXISTS login_attempts(
 	id serial PRIMARY KEY,
@@ -63,30 +61,27 @@ RETURNING id;`
 	return tx.Commit()
 }
 
-func (s *PostgresStore) All() (<-chan models.LoginAttempt, error) {
+func (s *PostgresStore) All() ([]models.LoginAttempt, error) {
 	stmt := `SELECT date, remote_ip, username, password, client_version, connection_uuid, country FROM login_attempts`
 
 	rows, err := s.db.Query(stmt)
 	if err != nil {
 		return nil, err
 	}
+	defer rows.Close()
 
-	ch := make(chan models.LoginAttempt)
+	var attempts []models.LoginAttempt
-	go func() {
+	for rows.Next() {
-		defer rows.Close()
+		var a models.LoginAttempt
-		for rows.Next() {
+		var ip string
-			var a models.LoginAttempt
+		if err := rows.Scan(&a.Date, &ip, &a.Username, &a.Password, &a.SSHClientVersion, &a.SSHClientVersion, &a.Country); err != nil {
-			var ip string
+			return nil, err
-			if err := rows.Scan(&a.Date, &ip, &a.Username, &a.Password, &a.SSHClientVersion, &a.SSHClientVersion, &a.Country); err != nil {
-				panic(err)
-			}
-			a.RemoteIP = net.ParseIP(ip)
-			ch <- a
 		}
-		close(ch)
+		a.RemoteIP = net.ParseIP(ip)
-	}()
+		attempts = append(attempts, a)
+	}
 
-	return ch, nil
+	return attempts, nil
 }
 
 func (s *PostgresStore) Stats(statType LoginStats, limit int) ([]StatsResult, error) {

honeypot/ssh/store/store.go

@@ -38,7 +38,7 @@ type AttemptQuery struct {
 }
 type LoginAttemptStore interface {
 	AddAttempt(l *models.LoginAttempt) error
-	All() (<-chan models.LoginAttempt, error)
+	All() ([]models.LoginAttempt, error)
 	Stats(statType LoginStats, limit int) ([]StatsResult, error)
 	Query(query AttemptQuery) ([]models.LoginAttempt, error)
 	HealthCheker

honeypot/ssh/store/store_test.go

@@ -25,12 +25,8 @@ func testLoginAttemptStore(s store.LoginAttemptStore, t *testing.T) {
 		if err != nil {
 			t.Fatalf("Error getting all attempts: %s", err)
 		}
-		var count int
+		if len(all) != len(testAttempts) {
-		for range all {
+			t.Errorf("All returned wrong amount. Got %d want %d", len(all), len(testAttempts))
-			count++
-		}
-		if count != len(testAttempts) {
-			t.Errorf("All returned wrong amount. Got %d want %d", count, len(testAttempts))
 		}
 		stats, err := s.Stats(store.LoginStatsTotals, 1)
 		if err != nil {
@@ -233,11 +229,7 @@ func benchmarkLoginAttemptStore(setupFunc func() store.LoginAttemptStore, b *tes
 			if err != nil {
 				b.Fatalf("Error fetchin all: %s", err)
 			}
-			var count int
+			_ = len(all)
-			for range all {
-				count++
-			}
-			_ = count
 		}
 	})
 }