From b2106e70567c28fdb6d44226a46b46c296b28ec7 Mon Sep 17 00:00:00 2001
From: Simone Lazzaris <simone.lazzaris@qcom.it>
Date: Fri, 24 Jan 2020 15:02:59 +0100
Subject: [PATCH] Reputation tracking on redis db

---
 reputation.go | 58 +++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 58 insertions(+)
 create mode 100644 reputation.go

diff --git a/reputation.go b/reputation.go
new file mode 100644
index 0000000..10f691c
--- /dev/null
+++ b/reputation.go
@@ -0,0 +1,58 @@
+package main
+import (
+	"fmt"
+	"github.com/gomodule/redigo/redis"
+	"log"
+	"time"
+)
+
+
+var (
+	rdb_pool *redis.Pool
+)
+
+func verify_reputation(sender string, ip string) (bool,string) {
+	if _,ok:=cfg["reputation_redis_address"]; !ok {
+		return true,""
+	}
+	xsender:=fmt.Sprintf("<%s>",sender)
+	score:=reputation_score(xsender, ip)
+	log_msg:=fmt.Sprintf(" reputation score %.3f",score)
+	if score > 0.2 {
+		if *xdebug {fmt.Printf("Accepting sender: %s:%s score %f\n", sender, ip, score)}
+		return true, log_msg
+	}
+	xlog.Info(fmt.Sprintf("Rejecting bad sender: %s:%s score %f ", sender, ip, score))
+	return false, log_msg
+}
+
+func reputation_score(sender string, ip string) float64 {
+	if *xdebug { fmt.Printf("REPUTATION: Sender %s from ip %s\n",sender,ip) }
+	rdb:=rdb_pool.Get()
+	defer rdb.Close()
+	rep,err:=redis.Float64(rdb.Do("GET", fmt.Sprintf("TRACK:%s:%s",sender,ip)))
+	if err!=nil {
+		rep,err=redis.Float64(rdb.Do("GET", fmt.Sprintf("TRACK:IP:%s",ip)))
+	}
+	if err!=nil {
+		log.Println(err)
+		return 1.0
+	}
+	return rep
+		
+}
+
+func InitReputation() {
+	if addr,ok:=cfg["reputation_redis_address"]; ok {
+		if *xdebug {fmt.Println("Initializing pool")}
+		rdb_pool=&redis.Pool {
+			MaxIdle: 10,
+			IdleTimeout: 120*time.Second,
+			Dial: func() (redis.Conn, error) {
+				if *xdebug {fmt.Println("Connecting to REDIS "+addr)}
+				return redis.Dial("tcp", addr)
+			},
+		}
+	}
+}
+