1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
|
// Copyright (C) 2018-2020 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 vpn
import (
"fmt"
"log"
"strings"
"github.com/apparentlymart/go-openvpn-mgmt/openvpn"
)
const (
On = "on"
Off = "off"
Starting = "starting"
Stopping = "stopping"
Failed = "failed"
)
var statusNames = map[string]string{
"CONNECTING": Starting,
"WAIT": Starting,
"AUTH": Starting,
"GET_CONFIG": Starting,
"ASSIGN_IP": Starting,
"ADD_ROUTES": Starting,
"CONNECTED": On,
"RECONNECTING": Starting,
"EXITING": Stopping,
"OFF": Off,
"FAILED": Off,
}
func (b *Bitmask) openvpnManagement() {
// TODO: we should warn the user on ListenAndServe errors
newConnection := func(conn openvpn.IncomingConn) {
eventCh := make(chan openvpn.Event, 10)
log.Println("New connection into the management")
b.managementClient = conn.Open(eventCh)
b.managementClient.SetStateEvents(true)
b.eventHandler(eventCh)
}
log.Fatal(openvpn.ListenAndServe(
fmt.Sprintf("%s:%s", openvpnManagementAddr, openvpnManagementPort),
openvpn.IncomingConnHandlerFunc(newConnection),
))
}
func (b *Bitmask) eventHandler(eventCh <-chan openvpn.Event) {
for event := range eventCh {
log.Printf("Event: %v", event)
stateEvent, ok := event.(*openvpn.StateEvent)
if !ok {
continue
}
statusName := stateEvent.NewState()
status, ok := statusNames[statusName]
if ok {
b.statusCh <- status
}
if statusName == "CONNECTED" {
ip := strings.Split(stateEvent.String(), ": ")[1]
gw, err := b.bonafide.GetGatewayByIP(ip)
if err == nil {
b.onGateway = gw
log.Println("Connected to gateway:", b.onGateway.Host)
} else {
log.Println("ERROR: connected to unknown gateway", ip)
}
}
}
b.statusCh <- Off
}
func (b *Bitmask) GetCurrentGateway() string {
return b.onGateway.Host
}
func (b *Bitmask) GetCurrentLocation() string {
return b.onGateway.LocationName
}
func (b *Bitmask) GetCurrentCountry() string {
return b.onGateway.CountryCode
}
func (b *Bitmask) IsManualLocation() bool {
return b.bonafide.IsManualLocation()
}
func (b *Bitmask) getOpenvpnState() (string, error) {
if b.managementClient == nil {
return "", fmt.Errorf("No management connected")
}
stateEvent, err := b.managementClient.LatestState()
if err != nil {
return "", err
}
status, ok := statusNames[stateEvent.NewState()]
if !ok {
return "", fmt.Errorf("Unkonw status")
}
return status, nil
}
|
