Added first working web interface version
This commit is contained in:
parent
48685c7e2f
commit
c44a18acb3
|
@ -52,3 +52,8 @@ coverage.xml
|
|||
# Sphinx documentation
|
||||
docs/_build/
|
||||
|
||||
*.swp
|
||||
pdns.conf
|
||||
pdns_backend.py
|
||||
ddns
|
||||
dump.rdb
|
||||
|
|
|
@ -0,0 +1,82 @@
|
|||
package connection
|
||||
|
||||
import (
|
||||
"crypto/sha1"
|
||||
"fmt"
|
||||
"github.com/garyburd/redigo/redis"
|
||||
"log"
|
||||
"time"
|
||||
"strings"
|
||||
)
|
||||
|
||||
const HostExpirationSeconds int = 10 * 24 * 60 * 60 // 10 Days
|
||||
|
||||
func HandleErr(err error) {
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func OpenConnection() *RedisConnection {
|
||||
conn, conn_err := redis.Dial("tcp", ":6379")
|
||||
if conn_err != nil {
|
||||
log.Fatal(conn_err)
|
||||
}
|
||||
|
||||
return &RedisConnection{conn}
|
||||
}
|
||||
|
||||
type RedisConnection struct {
|
||||
redis.Conn
|
||||
}
|
||||
|
||||
func (self *RedisConnection) GetHost(name string) *Host {
|
||||
host := Host{Hostname: name}
|
||||
|
||||
if self.HostExist(name) {
|
||||
data, err := redis.Values(self.Do("HGETALL", host.Hostname))
|
||||
HandleErr(err)
|
||||
|
||||
HandleErr(redis.ScanStruct(data, &host))
|
||||
}
|
||||
|
||||
return &host
|
||||
}
|
||||
|
||||
func (self *RedisConnection) SaveHost(host *Host) {
|
||||
_, err := self.Do("HMSET", redis.Args{}.Add(host.Hostname).AddFlat(host)...)
|
||||
HandleErr(err)
|
||||
|
||||
_, err = self.Do("EXPIRE", host.Hostname, HostExpirationSeconds)
|
||||
HandleErr(err)
|
||||
}
|
||||
|
||||
func (self *RedisConnection) HostExist(name string) bool {
|
||||
exists, err := redis.Bool(self.Do("EXISTS", name))
|
||||
HandleErr(err)
|
||||
|
||||
return exists
|
||||
}
|
||||
|
||||
type Host struct {
|
||||
Hostname string `redis:"-"`
|
||||
Ip string `redis:"ip"`
|
||||
Token string `redis:"token"`
|
||||
}
|
||||
|
||||
func (self *Host) GenerateAndSetToken() {
|
||||
hash := sha1.New()
|
||||
hash.Write([]byte(fmt.Sprintf("%d", time.Now().UnixNano())))
|
||||
hash.Write([]byte(self.Hostname))
|
||||
|
||||
self.Token = fmt.Sprintf("%x", hash.Sum(nil))
|
||||
}
|
||||
|
||||
// Returns true when this host has a IPv4 Address and false if IPv6
|
||||
func (self *Host) IsIPv4() bool {
|
||||
if strings.Contains(self.Ip, ".") {
|
||||
return true
|
||||
}
|
||||
|
||||
return False
|
||||
}
|
|
@ -0,0 +1,99 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"github.com/gin-gonic/gin"
|
||||
"github.com/pboehm/ddns/connection"
|
||||
"log"
|
||||
"net"
|
||||
"os"
|
||||
)
|
||||
|
||||
func HandleErr(err error) {
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
}
|
||||
|
||||
func RunBackend() {
|
||||
/* code */
|
||||
}
|
||||
|
||||
func RunWebService() {
|
||||
conn := connection.OpenConnection()
|
||||
defer conn.Close()
|
||||
|
||||
r := gin.Default()
|
||||
r.GET("/new/:hostname", func(c *gin.Context) {
|
||||
hostname := c.Params.ByName("hostname")
|
||||
|
||||
if conn.HostExist(hostname) {
|
||||
c.String(403, "This hostname has already been registered.")
|
||||
return
|
||||
}
|
||||
|
||||
host := &connection.Host{Hostname: hostname, Ip: "127.0.0.1"}
|
||||
host.GenerateAndSetToken()
|
||||
|
||||
conn.SaveHost(host)
|
||||
|
||||
c.String(200, fmt.Sprintf(
|
||||
"Go to /update/%s/%s for updating your IP address",
|
||||
host.Hostname, host.Token))
|
||||
})
|
||||
|
||||
r.GET("/update/:hostname/:token", func(c *gin.Context) {
|
||||
hostname := c.Params.ByName("hostname")
|
||||
token := c.Params.ByName("token")
|
||||
|
||||
if !conn.HostExist(hostname) {
|
||||
c.String(404,
|
||||
"This hostname has not been registered or is expired.")
|
||||
return
|
||||
}
|
||||
|
||||
host := conn.GetHost(hostname)
|
||||
|
||||
if host.Token != token {
|
||||
c.String(403,
|
||||
"You have supplied the wrong token to manipulate this host")
|
||||
return
|
||||
}
|
||||
|
||||
ip, _, err := net.SplitHostPort(c.Req.RemoteAddr)
|
||||
if err != nil {
|
||||
c.String(500, "You sender IP address is not in the right format")
|
||||
}
|
||||
|
||||
host.Ip = ip
|
||||
conn.SaveHost(host)
|
||||
|
||||
c.String(200, fmt.Sprintf("Your current IP is %s", ip))
|
||||
})
|
||||
|
||||
r.Run(":8080")
|
||||
}
|
||||
|
||||
func main() {
|
||||
|
||||
if len(os.Args) < 2 {
|
||||
usage()
|
||||
}
|
||||
|
||||
cmd := os.Args[1]
|
||||
|
||||
switch cmd {
|
||||
case "backend":
|
||||
log.Printf("Starting PDNS Backend\n")
|
||||
RunBackend()
|
||||
case "web":
|
||||
log.Printf("Starting Web Service\n")
|
||||
RunWebService()
|
||||
default:
|
||||
usage()
|
||||
}
|
||||
}
|
||||
|
||||
func usage() {
|
||||
log.Fatal("Usage: ./ddns [backend|web]")
|
||||
}
|
Loading…
Reference in New Issue