src/cypherpunks.ru/govpn/action.go

   1 /*
   2 GoVPN -- simple secure free software virtual private network daemon
   3 Copyright (C) 2014-2016 Sergey Matveev <stargrave@stargrave.org>
   4
   5 This program is free software: you can redistribute it and/or modify
   6 it under the terms of the GNU General Public License as published by
   7 the Free Software Foundation, either version 3 of the License, or
   8 (at your option) any later version.
   9
  10 This program is distributed in the hope that it will be useful,
  11 but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 GNU General Public License for more details.
  14
  15 You should have received a copy of the GNU General Public License
  16 along with this program.  If not, see <http://www.gnu.org/licenses/>.
  17 */
  18
  19 package govpn
  20
  21 import (
  22         "os"
  23         "os/exec"
  24
  25         "github.com/pkg/errors"
  26 )
  27
  28 // PeerContext hold info about a peer that connect or disconnect
  29 // used for Up, PreUp and Down
  30 type PeerContext struct {
  31         RemoteAddress string
  32         Protocol      Protocol
  33         Config        PeerConf
  34 }
  35
  36 // TunnelAction is an action for either client or server that is
  37 // executed when tunnel goes down
  38 type TunnelAction func(PeerContext) error
  39
  40 // TunnelPreUpAction is an action for client or server that is executed
  41 // after user is authenticated
  42 type TunnelPreUpAction func(PeerContext) (*TAP, error)
  43
  44 // RunScriptAction convert the path to a script into a TunnelAction
  45 func RunScriptAction(path *string) TunnelAction {
  46         if path == nil {
  47                 return nil
  48         }
  49         return func(ctx PeerContext) error {
  50                 _, err := ScriptCall(*path, ctx.Config.Iface, ctx.RemoteAddress)
  51                 return errors.Wrapf(err, "ScriptCall path=%q interface=%q remote=%q", *path, ctx.Config.Iface, ctx.RemoteAddress)
  52         }
  53 }
  54
  55 // ScriptCall call external program/script.
  56 // You have to specify path to it and (inteface name as a rule) something
  57 // that will be the first argument when calling it. Function will return
  58 // it's output and possible error.
  59 func ScriptCall(path, ifaceName, remoteAddr string) ([]byte, error) {
  60         if path == "" {
  61                 return nil, nil
  62         }
  63         if _, err := os.Stat(path); err != nil && os.IsNotExist(err) {
  64                 return nil, errors.Wrap(err, "os.Path")
  65         }
  66         cmd := exec.Command(path)
  67         cmd.Env = append(cmd.Env, environmentKeyInterface+"="+ifaceName)
  68         cmd.Env = append(cmd.Env, environmentKeyRemote+"="+remoteAddr)
  69         out, err := cmd.CombinedOutput()
  70         return out, errors.Wrap(err, "cmd.CombinedOutput")
  71 }