Refactoring of some vars and funcs name
Theses change breaks old configuration files but make the code readability better thanks to VSCode plugins.
This commit is contained in:
parent
7cf1ae926b
commit
efffac8b0a
8 changed files with 51 additions and 52 deletions
|
|
@ -2,9 +2,9 @@ package core
|
|||
|
||||
import "github.com/miekg/dns"
|
||||
|
||||
//Handle the DNS request using miekg/dns
|
||||
//HandleDNSRequest : Handle the DNS request using miekg/dns
|
||||
//Requires dns.ReponseWriter and dns.Msg args
|
||||
func HandleDnsRequest(w dns.ResponseWriter, r *dns.Msg) {
|
||||
func HandleDNSRequest(w dns.ResponseWriter, r *dns.Msg) {
|
||||
|
||||
//dns.Msg object
|
||||
//Will be passed to the parseQuery() function
|
||||
|
|
|
|||
|
|
@ -1,20 +1,20 @@
|
|||
App_mode = "production" #Anything != production will show DEBUG messages
|
||||
|
||||
[App]
|
||||
Ip = ":"
|
||||
IP = ":"
|
||||
Port = 5353
|
||||
Logfile = true
|
||||
Logdir = "/var/log/"
|
||||
|
||||
[Database]
|
||||
Ip = "127.0.0.1"
|
||||
IP = "127.0.0.1"
|
||||
Username = "sacrebleu"
|
||||
Password = "superSecretPassword"
|
||||
Port = "3306"
|
||||
Db = "sacrebleudatabase"
|
||||
|
||||
[Redis]
|
||||
Ip = "127.0.0.1"
|
||||
IP = "127.0.0.1"
|
||||
Port = 6379
|
||||
Password = ""
|
||||
Db = 0
|
||||
|
|
|
|||
11
main.go
11
main.go
|
|
@ -13,7 +13,10 @@ import (
|
|||
)
|
||||
|
||||
//Global vars
|
||||
//conf Configuration
|
||||
var conf *utils.Conf
|
||||
|
||||
//DB the SQL database
|
||||
var DB *sql.DB
|
||||
|
||||
//Main loop
|
||||
|
|
@ -31,17 +34,17 @@ func main() {
|
|||
utils.InitLogger(conf)
|
||||
|
||||
//Attach DNS request handler func for all domains
|
||||
dns.HandleFunc(".", core.HandleDnsRequest)
|
||||
dns.HandleFunc(".", core.HandleDNSRequest)
|
||||
|
||||
//Initialize the redis database
|
||||
utils.RedisDatabase(conf)
|
||||
|
||||
//Initialize the sql database
|
||||
utils.SqlDatabase(conf)
|
||||
utils.SQLDatabase(conf)
|
||||
|
||||
//Start the DNS server
|
||||
server := &dns.Server{Addr: conf.App.Ip + strconv.Itoa(conf.App.Port), Net: "udp"} //define the server
|
||||
logrus.WithFields(logrus.Fields{"ip": conf.App.Ip, "port": conf.App.Port}).Infof("SERVER : Started") //log
|
||||
server := &dns.Server{Addr: conf.App.IP + strconv.Itoa(conf.App.Port), Net: "udp"} //define the server
|
||||
logrus.WithFields(logrus.Fields{"ip": conf.App.IP, "port": conf.App.Port}).Infof("SERVER : Started") //log
|
||||
err = server.ListenAndServe() //start it
|
||||
utils.CheckErr(err)
|
||||
|
||||
|
|
|
|||
|
|
@ -8,7 +8,7 @@ import (
|
|||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
//Check the SQL and REDIS database for a Record.
|
||||
//GetRecord : Check the SQL and REDIS database for a Record.
|
||||
//A Record struct is used as input and output
|
||||
func GetRecord(entry Record) Record {
|
||||
//Check for strict record in Redis cache
|
||||
|
|
|
|||
|
|
@ -16,13 +16,13 @@ var ctx = context.Background()
|
|||
//Redis client as global var
|
||||
var redisDb *redis.Client
|
||||
|
||||
//Initialize the Redis Database
|
||||
//RedisDatabase : Initialize the Redis Database
|
||||
//Requires a conf struct
|
||||
//Return a *redis.Client
|
||||
func RedisDatabase(conf *Conf) *redis.Client {
|
||||
logrus.WithFields(logrus.Fields{"ip": conf.Redis.Ip, "port": conf.Redis.Port}).Infof("REDIS : Connection to DB")
|
||||
logrus.WithFields(logrus.Fields{"ip": conf.Redis.IP, "port": conf.Redis.Port}).Infof("REDIS : Connection to DB")
|
||||
rdb := redis.NewClient(&redis.Options{
|
||||
Addr: fmt.Sprintf("%s:%v", conf.Redis.Ip, conf.Redis.Port),
|
||||
Addr: fmt.Sprintf("%s:%v", conf.Redis.IP, conf.Redis.Port),
|
||||
Password: conf.Redis.Password,
|
||||
DB: conf.Redis.Db,
|
||||
}) //Connect to the DB
|
||||
|
|
@ -54,10 +54,8 @@ func redisCheckForRecord(redisKey string, entry Record) (Record, error) {
|
|||
err := json.Unmarshal([]byte(val), &entry)
|
||||
logrus.Debugf("REDIS : %s => %s", redisKey, entry.Content)
|
||||
return entry, err
|
||||
} else {
|
||||
//Else return nil
|
||||
return entry, redis.Nil
|
||||
}
|
||||
return entry, redis.Nil
|
||||
}
|
||||
|
||||
//Add a record in the Redis database
|
||||
|
|
|
|||
29
utils/sql.go
29
utils/sql.go
|
|
@ -6,26 +6,26 @@ import (
|
|||
"time"
|
||||
|
||||
"github.com/go-redis/redis"
|
||||
_ "github.com/go-sql-driver/mysql"
|
||||
_ "github.com/go-sql-driver/mysql" //MySQL driver
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
//SQL database as global var
|
||||
//DB SQL database as global var
|
||||
var DB *sql.DB
|
||||
|
||||
//Initialize the (My)SQL Database
|
||||
//SQLDatabase Initialize the (My)SQL Database
|
||||
//Requires a conf struct
|
||||
func SqlDatabase(conf *Conf) {
|
||||
func SQLDatabase(conf *Conf) {
|
||||
logrus.WithFields(logrus.Fields{"database": conf.Database.Db}).Infof("SQL : Connection to DB")
|
||||
//Connect to the Database
|
||||
db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@tcp(%s:%s)/%s", conf.Database.Username, conf.Database.Password, conf.Database.Ip, conf.Database.Port, conf.Database.Db))
|
||||
db, err := sql.Open("mysql", fmt.Sprintf("%s:%s@tcp(%s:%s)/%s", conf.Database.Username, conf.Database.Password, conf.Database.IP, conf.Database.Port, conf.Database.Db))
|
||||
CheckErr(err)
|
||||
DB = db
|
||||
SqlTest() //Test SQL connexion
|
||||
SQLTest() //Test SQL connexion
|
||||
}
|
||||
|
||||
//Test the SQL connexion by selecting all records from the database
|
||||
func SqlTest() {
|
||||
//SQLTest : Test the SQL connexion by selecting all records from the database
|
||||
func SQLTest() {
|
||||
_, err := DB.Query("SELECT name, content FROM records")
|
||||
CheckErr(err) //Panic if any error
|
||||
}
|
||||
|
|
@ -34,7 +34,7 @@ func SqlTest() {
|
|||
func sqlCheckForRecord(redisKey string, dKey string, entry Record) (Record, int) {
|
||||
dbg := DB.QueryRow(
|
||||
"SELECT id, content, ttl FROM records WHERE `name` = ? AND `type` = ?;", dKey, entry.Qtype).Scan(
|
||||
&entry.Id,
|
||||
&entry.ID,
|
||||
&entry.Content,
|
||||
&entry.TTL,
|
||||
)
|
||||
|
|
@ -50,10 +50,10 @@ func sqlCheckForRecord(redisKey string, dKey string, entry Record) (Record, int)
|
|||
logrus.Debugf("REDIS : Set entry for %s", redisKey)
|
||||
_ = redisSet(redisDb, redisKey, 30*time.Second, entry) //Set it in the Redis database for 30sec
|
||||
return entry, 0
|
||||
} else {
|
||||
//Else return 1 for err
|
||||
return entry, 1
|
||||
}
|
||||
//Else return 1 for err
|
||||
return entry, 1
|
||||
|
||||
}
|
||||
|
||||
//Check for a wildcard record in the SQL database
|
||||
|
|
@ -65,7 +65,7 @@ func sqlCheckForReverse6Wildcard(redisKey string, dKey string, entry Record) (Re
|
|||
|
||||
//For each result check if it match the reverse IP
|
||||
for results.Next() {
|
||||
err = results.Scan(&returnedEntry.Id, &returnedEntry.Content, &returnedEntry.Fqdn)
|
||||
err = results.Scan(&returnedEntry.ID, &returnedEntry.Content, &returnedEntry.Fqdn)
|
||||
CheckErr(err)
|
||||
|
||||
//Check if the record is matching the reversed IP
|
||||
|
|
@ -74,9 +74,8 @@ func sqlCheckForReverse6Wildcard(redisKey string, dKey string, entry Record) (Re
|
|||
//Cache the request in Redis if any result
|
||||
_ = redisSet(redisDb, redisKey, 10*time.Second, returnedEntry)
|
||||
return returnedEntry, err
|
||||
} else {
|
||||
logrus.Debug("REVERSE : WRONG wildcard reverse .")
|
||||
}
|
||||
logrus.Debug("REVERSE : WRONG wildcard reverse .")
|
||||
}
|
||||
|
||||
return entry, redis.Nil
|
||||
|
|
|
|||
|
|
@ -1,53 +1,53 @@
|
|||
package utils
|
||||
|
||||
//Struct for App (dns server) configuration in the config.ini file
|
||||
//App : Struct for App (dns server) configuration in the config.ini file
|
||||
type App struct {
|
||||
Port int
|
||||
Ip string
|
||||
IP string
|
||||
Logdir string
|
||||
Logfile bool
|
||||
}
|
||||
|
||||
//Struct for SQL Database configuration in the config.ini file
|
||||
//Database : Struct for SQL Database configuration in the config.ini file
|
||||
type Database struct {
|
||||
Ip string
|
||||
IP string
|
||||
Port string
|
||||
Username string
|
||||
Password string
|
||||
Db string
|
||||
}
|
||||
|
||||
//Struct for Redis Database configuration in the config.ini file
|
||||
//Redis : Struct for Redis Database configuration in the config.ini file
|
||||
type Redis struct {
|
||||
Ip string
|
||||
IP string
|
||||
Port int
|
||||
Password string
|
||||
Db int
|
||||
Ttl int
|
||||
TTL int
|
||||
}
|
||||
|
||||
//Struct for the whole config.ini file when it will be parsed by go-ini
|
||||
//Conf : Struct for the whole config.ini file when it will be parsed by go-ini
|
||||
type Conf struct {
|
||||
App_mode string
|
||||
AppMode string
|
||||
App
|
||||
Database
|
||||
Redis
|
||||
}
|
||||
|
||||
//Struct for a Domain (not used currently).
|
||||
//Domain : Struct for a Domain (not used currently).
|
||||
type Domain struct {
|
||||
ID int `json:"id"`
|
||||
FriendlyName string
|
||||
Fqdn string
|
||||
OwnerId int
|
||||
OwnerID int
|
||||
LastEdit string
|
||||
}
|
||||
|
||||
//Struct for a domain record
|
||||
//Record : Struct for a domain record
|
||||
//Defined by it's ID, DomainID (parent domain), Fqdn (or name), Content (value of the record), Type (as Qtype/int), TTL (used only for the DNS response and not the Redis TTL)
|
||||
type Record struct {
|
||||
Id int
|
||||
DomainId int
|
||||
ID int
|
||||
DomainID int
|
||||
Fqdn string
|
||||
Content string
|
||||
Type int
|
||||
|
|
|
|||
|
|
@ -10,26 +10,26 @@ import (
|
|||
"github.com/snowzach/rotatefilehook"
|
||||
)
|
||||
|
||||
//If fatal error, log it and panic
|
||||
//CheckErr : If fatal error, log it and panic
|
||||
func CheckErr(err error) {
|
||||
if err != nil {
|
||||
log.Fatalf("%s\n ", err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
//If basic error, log it as classic error but don't panic and keep kalm
|
||||
//DbgErr : If basic error, log it as classic error but don't panic and keep kalm
|
||||
func DbgErr(err error) {
|
||||
if err != nil {
|
||||
log.Errorf("%s\n ", err.Error())
|
||||
}
|
||||
}
|
||||
|
||||
//Init the logrus logger with rotateFileHook.
|
||||
//InitLogger : Init the logrus logger with rotateFileHook.
|
||||
//Conf struct passed to get informations about the logger (debug or not)
|
||||
func InitLogger(conf *Conf) {
|
||||
var logLevel = logrus.InfoLevel //By default the level is Info.
|
||||
|
||||
if conf.App_mode != "production" { //If the configuration contains anything different than "production"; the level is set to Debug
|
||||
if conf.AppMode != "production" { //If the configuration contains anything different than "production"; the level is set to Debug
|
||||
logLevel = logrus.DebugLevel
|
||||
}
|
||||
|
||||
|
|
@ -62,7 +62,7 @@ func InitLogger(conf *Conf) {
|
|||
logrus.AddHook(rotateFileHook)
|
||||
}
|
||||
|
||||
log.WithFields(log.Fields{"app_mode": conf.App_mode}).Info("Application mode")
|
||||
log.WithFields(log.Fields{"app_mode": conf.AppMode}).Info("Application mode")
|
||||
log.WithFields(log.Fields{"logLevel": logLevel}).Debug("Log level")
|
||||
}
|
||||
|
||||
|
|
@ -74,7 +74,6 @@ func checkReverse6(entry Record, result Record) bool {
|
|||
logrus.Debugf("REVERSE checkReverse6 : %s", check)
|
||||
if strings.Contains(check, IP6arpa) {
|
||||
return false
|
||||
} else {
|
||||
return true
|
||||
}
|
||||
return true
|
||||
}
|
||||
|
|
|
|||
Loading…
Reference in a new issue