Added first working web interface version

This commit is contained in:
Philipp Böhm 2014-07-12 22:37:00 +02:00
parent 48685c7e2f
commit c44a18acb3
3 changed files with 186 additions and 0 deletions

5
.gitignore vendored
View File

@ -52,3 +52,8 @@ coverage.xml
# Sphinx documentation
docs/_build/
*.swp
pdns.conf
pdns_backend.py
ddns
dump.rdb

82
connection/connection.go Normal file
View File

@ -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
}

99
ddns.go Normal file
View File

@ -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]")
}