Initial tests
This commit is contained in:
commit
cdaf0dd68d
|
@ -0,0 +1,3 @@
|
|||
config.json
|
||||
test.sh
|
||||
openvpn-dm-mgt-server
|
|
@ -0,0 +1,51 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"log"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
type HttpServer struct {
|
||||
Port string
|
||||
ovpn *OpenVpnMgt
|
||||
}
|
||||
|
||||
func (h *HttpServer) handler(w http.ResponseWriter, r *http.Request) {
|
||||
fmt.Fprintf(w, "Hi there, I love %s!", r.URL.Path[1:])
|
||||
}
|
||||
|
||||
func (h *HttpServer) versionHandler(w http.ResponseWriter, r *http.Request) {
|
||||
err, message := h.ovpn.Version()
|
||||
if err != nil {
|
||||
fmt.Fprintf(w, "Error : %s", err)
|
||||
}
|
||||
|
||||
jsonStr, err := json.Marshal(message)
|
||||
if err != nil {
|
||||
fmt.Fprintf(w, "Error : %s", err)
|
||||
}
|
||||
fmt.Fprintf(w, "%s", jsonStr)
|
||||
}
|
||||
|
||||
func (h *HttpServer) helpHandler(w http.ResponseWriter, r *http.Request) {
|
||||
err, message := h.ovpn.Help()
|
||||
if err != nil {
|
||||
fmt.Fprintf(w, "Error : %s", err)
|
||||
} else {
|
||||
fmt.Fprintf(w, "%s", message)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
func NewHTTPServer(port string, s *OpenVpnMgt) {
|
||||
h := &HttpServer{
|
||||
Port: port,
|
||||
ovpn: s,
|
||||
}
|
||||
http.HandleFunc("/help", h.helpHandler)
|
||||
http.HandleFunc("/version", h.versionHandler)
|
||||
http.HandleFunc("/", h.handler)
|
||||
log.Fatal(http.ListenAndServe(port, nil))
|
||||
}
|
|
@ -0,0 +1,7 @@
|
|||
package main
|
||||
|
||||
func main() {
|
||||
server := NewVPNServer(":4000")
|
||||
go server.Run()
|
||||
NewHTTPServer(":8080", server)
|
||||
}
|
|
@ -0,0 +1,11 @@
|
|||
|
||||
|
||||
client-deny 0 0 "Need OTP" "CRV1:R:blabla:ZXVjbGlkZQ==:OTP Code "
|
||||
|
||||
|
||||
client-auth 1 0
|
||||
ifconfig-push 10.8.66.3 10.8.66.1
|
||||
push "dhcp-option DNS 10.190.32.2"
|
||||
push "dhcp-option DNS 10.190.32.20"
|
||||
END
|
||||
|
|
@ -0,0 +1,159 @@
|
|||
package main
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"errors"
|
||||
"fmt"
|
||||
"io"
|
||||
"log"
|
||||
"net"
|
||||
"strings"
|
||||
"sync"
|
||||
)
|
||||
|
||||
// Server represents the server
|
||||
type OpenVpnMgt struct {
|
||||
Port string
|
||||
buf *bufio.ReadWriter
|
||||
connected bool
|
||||
m sync.RWMutex
|
||||
ret chan []string
|
||||
}
|
||||
|
||||
// NewServer returns a pointer to a new server
|
||||
func NewVPNServer(port string) *OpenVpnMgt {
|
||||
return &OpenVpnMgt{
|
||||
Port: port,
|
||||
ret: make(chan []string),
|
||||
}
|
||||
}
|
||||
|
||||
// Run starts a the server
|
||||
func (s *OpenVpnMgt) Run() {
|
||||
// Resolve the passed port into an address
|
||||
addrs, err := net.ResolveTCPAddr("tcp", s.Port)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
// start listening to client connections
|
||||
listener, err := net.ListenTCP("tcp", addrs)
|
||||
if err != nil {
|
||||
fmt.Println(err)
|
||||
}
|
||||
// Infinite loop since we dont want the server to shut down
|
||||
for {
|
||||
// Accept the incomming connections
|
||||
conn, err := listener.Accept()
|
||||
if err != nil {
|
||||
// continue accepting connection even if an error occurs (if error occurs dont shut down)
|
||||
continue
|
||||
}
|
||||
// run it as a go routine to allow multiple clients to connect at the same time
|
||||
go s.handleConn(conn)
|
||||
}
|
||||
}
|
||||
|
||||
func (s *OpenVpnMgt) sendCommand(msg []string) (error, []string) {
|
||||
if !s.connected {
|
||||
return errors.New("No openvpn server present"), nil
|
||||
}
|
||||
for _, line := range msg {
|
||||
if _, err := s.buf.WriteString(line + "\r\n"); err != nil {
|
||||
return err, nil
|
||||
}
|
||||
}
|
||||
|
||||
if err := s.buf.Flush(); err != nil {
|
||||
return err, nil
|
||||
}
|
||||
|
||||
// wait for the response
|
||||
ret := <-s.ret
|
||||
return nil, ret
|
||||
}
|
||||
|
||||
func (s *OpenVpnMgt) Help() (error, []string) {
|
||||
err, msg := s.sendCommand([]string{"help"})
|
||||
if err != nil {
|
||||
return err, nil
|
||||
}
|
||||
return nil, msg
|
||||
}
|
||||
|
||||
func (s *OpenVpnMgt) Version() (error, []string) {
|
||||
err, msg := s.sendCommand([]string{"version"})
|
||||
if err != nil {
|
||||
return err, nil
|
||||
}
|
||||
return nil, msg
|
||||
}
|
||||
|
||||
func (s *OpenVpnMgt) ClientDisconnect(line string) {
|
||||
msg := <-s.ret
|
||||
log.Println(msg)
|
||||
}
|
||||
|
||||
func (s *OpenVpnMgt) ClientConnect(line string) {
|
||||
msg := <-s.ret
|
||||
log.Println(msg)
|
||||
}
|
||||
|
||||
func (s *OpenVpnMgt) handleConn(conn net.Conn) {
|
||||
defer conn.Close()
|
||||
|
||||
// we don't want multiple connexions, only one openvpn server at a time
|
||||
s.m.Lock()
|
||||
if s.connected {
|
||||
conn.Write([]byte("Sorry, only one server allowed\n"))
|
||||
s.m.Unlock()
|
||||
return
|
||||
}
|
||||
s.connected = true
|
||||
s.m.Unlock()
|
||||
|
||||
// we store the buffer pointer in the struct, to be accessed from other methods
|
||||
s.buf = bufio.NewReadWriter(bufio.NewReader(conn), bufio.NewWriter(conn))
|
||||
|
||||
// most response are multilined, use response to concatenate them
|
||||
response := []string{}
|
||||
|
||||
for {
|
||||
line, err := s.buf.ReadString('\n')
|
||||
|
||||
// manage basic errors
|
||||
switch {
|
||||
case err == io.EOF:
|
||||
log.Println("Reached EOF - close this connection.\n")
|
||||
s.connected = false
|
||||
return
|
||||
case err != nil:
|
||||
log.Println("Error reading line. Got: '"+line+"'\n", err)
|
||||
s.connected = false
|
||||
return
|
||||
}
|
||||
line = strings.Trim(line, "\n\r ")
|
||||
|
||||
switch {
|
||||
// a new openvpn server is connected
|
||||
case strings.HasPrefix(line, ">INFO"):
|
||||
|
||||
// new bloc for a disconnect event.
|
||||
// We start the receiving handler, which will wait for the Channel message
|
||||
case strings.HasPrefix(line, ">CLIENT:DISCONNECT"):
|
||||
go s.ClientDisconnect(line)
|
||||
|
||||
// new bloc for a connect event.
|
||||
// We start the receiving handler, which will wait for the Channel message
|
||||
case strings.HasPrefix(line, ">CLIENT:CONNECT"):
|
||||
go s.ClientConnect(line)
|
||||
|
||||
// write the cumulated lines into the channel to the current handler
|
||||
case strings.HasPrefix(line, "END") || strings.HasPrefix(line, ">CLIENT:ENV,END"):
|
||||
s.ret <- response
|
||||
response = nil
|
||||
default:
|
||||
response = append(response, line)
|
||||
}
|
||||
log.Print(line)
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue