[feat] pass udp flag to bitmask-root

1 files changed, 7 insertions, 1 deletions

diff --git a/pkg/vpn/main.go b/pkg/vpn/main.go
index 0671877..0b1d316 100644
--- a/pkg/vpn/main.go
+++ b/pkg/vpn/main.go
@@ -39,6 +39,7 @@ type Bitmask struct {
 	shapes           *shapeshifter.ShapeShifter
 	certPemPath      string
 	openvpnArgs      []string
+	udp              bool
 	failed           bool
 }
 
@@ -54,7 +55,7 @@ func Init() (*Bitmask, error) {
 	if err != nil {
 		return nil, err
 	}
-	b := Bitmask{tempdir, bonafide.Gateway{}, bonafide.Gateway{}, statusCh, nil, bf, launch, "", nil, "", []string{}, false}
+	b := Bitmask{tempdir, bonafide.Gateway{}, bonafide.Gateway{}, statusCh, nil, bf, launch, "", nil, "", []string{}, false, false}
 
 	b.launch.firewallStop()
 	/*
@@ -107,3 +108,8 @@ func (b *Bitmask) NeedsCredentials() bool {
 func (b *Bitmask) DoLogin(username, password string) (bool, error) {
 	return b.bonafide.DoLogin(username, password)
 }
+
+func (b *Bitmask) UseUDP(udp bool) error {
+	b.udp = udp
+	return nil
+}