diff options
author | Ruben Pollan <meskio@sindominio.net> | 2018-06-26 19:15:52 +0200 |
---|---|---|
committer | Kali Kaneko (leap communications) <kali@leap.se> | 2018-06-27 20:31:49 +0200 |
commit | 8b0ad0c146015cfeef6abd4373ca99a3b4e57826 (patch) | |
tree | 5fe5098117950cb8d4a6657570c614a9d329b3f1 /helper/helper.go | |
parent | 05bc3260b9c958a72096e7a28956f4666bd65603 (diff) |
[feat] first implementation of the helper in go
Diffstat (limited to 'helper/helper.go')
-rw-r--r-- | helper/helper.go | 104 |
1 files changed, 104 insertions, 0 deletions
diff --git a/helper/helper.go b/helper/helper.go new file mode 100644 index 0000000..a3816de --- /dev/null +++ b/helper/helper.go @@ -0,0 +1,104 @@ +// Copyright (C) 2018 LEAP +// +// This program is free software: you can redistribute it and/or modify +// it under the terms of the GNU General Public License as published by +// the Free Software Foundation, either version 3 of the License, or +// (at your option) any later version. +// +// This program is distributed in the hope that it will be useful, +// but WITHOUT ANY WARRANTY; without even the implied warranty of +// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +// GNU General Public License for more details. +// +// You should have received a copy of the GNU General Public License +// along with this program. If not, see <http://www.gnu.org/licenses/>. + +package main + +import ( + "encoding/json" + "log" + "net/http" + "os/exec" +) + +const ( + bindAddr = "localhost:7171" +) + +type openvpnT struct { + cmd *exec.Cmd +} + +func main() { + openvpn := openvpnT{nil} + firewall := firewallT{} + http.HandleFunc("/openvpn/start", openvpn.start) + http.HandleFunc("/openvpn/stop", openvpn.stop) + http.HandleFunc("/firewall/start", firewall.start) + http.HandleFunc("/firewall/stop", firewall.stop) + + log.Fatal(http.ListenAndServe(bindAddr, nil)) +} + +func (openvpn *openvpnT) start(w http.ResponseWriter, r *http.Request) { + args, err := getArgs(r) + if err != nil { + log.Printf("An error has occurred processing flags: %v", err) + w.Write([]byte(err.Error())) + return + } + + log.Printf("start openvpn: %v", args) + err = openvpn.run(args) + if err != nil { + log.Printf("Error starting openvpn: %v", err) + w.Write([]byte(err.Error())) + } +} + +func (openvpn *openvpnT) run(args []string) error { + if openvpn.cmd != nil { + log.Printf("openvpn was running, stop it first") + err := openvpn.kill() + if err != nil { + return err + } + } + + // TODO: if it dies we should restart it + openvpn.cmd = exec.Command(getOpenvpnPath(), args...) + return openvpn.cmd.Start() +} + +func (openvpn *openvpnT) stop(w http.ResponseWriter, r *http.Request) { + log.Println("stop openvpn") + if openvpn.cmd == nil || openvpn.cmd.ProcessState != nil { + openvpn.cmd = nil + return + } + + err := openvpn.kill() + if err != nil { + log.Printf("Error stoping openvpn: %v", err) + w.Write([]byte(err.Error())) + } +} + +func (openvpn *openvpnT) kill() error { + err := kill(openvpn.cmd) + if err != nil { + return err + } + openvpn.cmd.Wait() + + openvpn.cmd = nil + return nil +} + +func getArgs(r *http.Request) ([]string, error) { + args := []string{} + decoder := json.NewDecoder(r.Body) + err := decoder.Decode(&args) + return args, err +} |