Merge pull request #46 from hagbard-01/current

T427: wireguard support

1 files changed, 131 insertions, 98 deletions

diff --git a/src/conf_mode/wireguard.py b/src/conf_mode/wireguard.py
index a4f876397..9848914e3 100755
--- a/src/conf_mode/wireguard.py
+++ b/src/conf_mode/wireguard.py
@@ -26,101 +26,98 @@ from vyos.config import Config
 from vyos import ConfigError
 
 dir = r'/config/auth/wireguard'
-pk  = dir + '/private.key'
+pk = dir + '/private.key'
 pub = dir + '/public.key'
 
-### check_kmod may be removed in the future, 
-### just want to have everything smoothly running after reboot
 def check_kmod():
   if not os.path.exists('/sys/module/wireguard'):
-    sl.syslog(sl.LOG_NOTICE, "loading wirguard kmod") 
+    sl.syslog(sl.LOG_NOTICE, "loading wirguard kmod")
     if  os.system('sudo modprobe wireguard') != 0:
       sl.syslog(sl.LOG_NOTICE, "modprobe wireguard failed")
       raise ConfigError("modprobe wireguard failed")
 
 def get_config():
-  config_data = {
-    'interfaces' : {}
-  }
-
   c = Config()
   if not c.exists('interfaces wireguard'):
     return None
-  
-  c.set_level('interfaces') 
+
+  c.set_level('interfaces')
   intfcs = c.list_nodes('wireguard')
   intfcs_eff = c.list_effective_nodes('wireguard')
-  new_lst = list( set(intfcs) - set(intfcs_eff) ) 
-  del_lst = list( set(intfcs_eff) - set(intfcs) )
+  new_lst = list(set(intfcs) - set(intfcs_eff))
+  del_lst = list(set(intfcs_eff) - set(intfcs))
 
-  ### setting deafult and determine status of the config 
+  config_data = {
+      'interfaces' : {}
+  }
+  ### setting defaults and determine status of the config
   for intfc in intfcs:
     cnf = 'wireguard ' + intfc
     # default data struct
-    config_data['interfaces'].update (
-      {
-        intfc : {
-          'addr'        : '',
-          'descr'       : intfc, ## snmp ifAlias
-          'lport'       : '',
-          'status'      : 'exists',
-          'state'       : 'enabled',
-          'mtu'         : 1420,
-          'peer'        : {}
-          }
+    config_data['interfaces'].update(
+        {
+            intfc : {
+                'addr'        : '',
+                'descr'       : intfc, ## snmp ifAlias
+                'lport'       : '',
+                'status'      : 'exists',
+                'state'       : 'enabled',
+                'mtu'         : '1420',
+                'peer'        : {}
+            }
         }
-    ) 
+    )
 
+  ### determine status either delete or create
   for i in new_lst:
-    config_data['interfaces'][i]['status'] = 'create' 
+    config_data['interfaces'][i]['status'] = 'create'
 
   for i in del_lst:
-    config_data['interfaces'].update (
-      {
-        i : {
-          'status': 'delete'
+    config_data['interfaces'].update(
+        {
+            i : {
+                'status': 'delete'
+            }
         }
-      }
     )
 
-  ### based on the status, set real values
+  ### based on the status, setup conf values
   for intfc in intfcs:
     cnf = 'wireguard ' + intfc
     if config_data['interfaces'][intfc]['status'] != 'delete':
-      #### addresses
+      ### addresses
       if c.exists(cnf + ' address'):
         config_data['interfaces'][intfc]['addr'] = c.return_values(cnf + ' address')
       ### listen port
-      if c.exists(cnf + ' listen-port'):
-        config_data['interfaces'][intfc]['lport'] = c.return_value(cnf + ' listen-port')
+      if c.exists(cnf + ' port'):
+        config_data['interfaces'][intfc]['lport'] = c.return_value(cnf + ' port')
       ### description
       if c.exists(cnf + ' description'):
         config_data['interfaces'][intfc]['descr'] = c.return_value(cnf + ' description')
       ### mtu
       if c.exists(cnf + ' mtu'):
         config_data['interfaces'][intfc]['mtu'] = c.return_value(cnf + ' mtu')
-      
       ### peers
       if c.exists(cnf + ' peer'):
         for p in c.list_nodes(cnf + ' peer'):
-          config_data['interfaces'][intfc]['peer'].update  (
-            {
-              p : {
-                'allowed-ips' : [],
-                'endpoint'  : '' 
+          config_data['interfaces'][intfc]['peer'].update(
+              {
+                  p : {
+                      'allowed-ips' : [],
+                      'endpoint'  : '',
+                      'pubkey'  : ''
+                  }
               }
-            }
           )
+          if c.exists(cnf + ' peer ' + p + ' pubkey'):
+            config_data['interfaces'][intfc]['peer'][p]['pubkey'] = c.return_value(cnf + ' peer ' + p + ' pubkey')
           if c.exists(cnf + ' peer ' + p + ' allowed-ips'):
             config_data['interfaces'][intfc]['peer'][p]['allowed-ips'] = c.return_values(cnf + ' peer ' + p + ' allowed-ips')
           if c.exists(cnf + ' peer ' + p + ' endpoint'):
             config_data['interfaces'][intfc]['peer'][p]['endpoint'] = c.return_value(cnf + ' peer ' + p + ' endpoint')
-  
-      ### persistent-keepalive
-      if c.exists(cnf + ' peer ' + p + ' persistent-keepalive'):
-        config_data['interfaces'][intfc]['peer'][p]['persistent-keepalive'] = c.return_value(cnf + ' peer ' + p + ' persistent-keepalive')
+          if c.exists(cnf + ' peer ' + p + ' persistent-keepalive'):
+            config_data['interfaces'][intfc]['peer'][p]['persistent-keepalive'] = c.return_value(cnf + ' peer ' + p + ' persistent-keepalive')
 
-  #print (config_data)
   return config_data
 
 def verify(c):
@@ -130,34 +127,31 @@ def verify(c):
   for i in c['interfaces']:
     if c['interfaces'][i]['status'] != 'delete':
       if not c['interfaces'][i]['addr']:
-        raise ConfigError("address required for interface " + i) 
-      if not c['interfaces'][i]['lport']:
-        raise ConfigError("listen-port required for interface " + i)
+        raise ConfigError("address required for interface " + i)
       if not c['interfaces'][i]['peer']:
         raise ConfigError("peer required on interface " + i)
-      else:
-        for p in c['interfaces'][i]['peer']:
-          if not c['interfaces'][i]['peer'][p]['allowed-ips']:
-            raise ConfigError("allowed-ips required on interface " + i + " for peer " + p)
 
-    ### eventually check allowed-ips (if it's an ip and valid CIDR or so)
-    ### endpoint needs to be IP:port
+      for p in c['interfaces'][i]['peer']:
+        if not c['interfaces'][i]['peer'][p]['allowed-ips']:
+          raise ConfigError("allowed-ips required on interface " + i + " for peer " + p)
+        if not c['interfaces'][i]['peer'][p]['pubkey']:
+          raise ConfigError("pubkey from your peer is mandatory on " + i + " for peer " + p)
+
 
 def apply(c):
   ### no wg config left, delete all wireguard devices on the os
   if not c:
     net_devs = os.listdir('/sys/class/net/')
     for dev in net_devs:
-      buf = open('/sys/class/net/' + dev + '/uevent','r').read()
+      buf = open('/sys/class/net/' + dev + '/uevent', 'r').read()
       if re.search("DEVTYPE=wireguard", buf, re.I|re.M):
-        wg_intf = re.sub("INTERFACE=","", re.search("INTERFACE=.*", buf, re.I|re.M).group(0) ) 
+        wg_intf = re.sub("INTERFACE=", "", re.search("INTERFACE=.*", buf, re.I|re.M).group(0))
         sl.syslog(sl.LOG_NOTICE, "removing interface " + wg_intf)
         subprocess.call(['ip l d dev ' + wg_intf + ' >/dev/null'], shell=True)
     return None
-  
+
   ###
-  ## to find the diffs between old config an new config
-  ## so we only configure/delete what was not previously configured
+  ## find the diffs between effective config an new config
   ###
   c_eff = Config()
   c_eff.set_level('interfaces wireguard')
@@ -175,88 +169,129 @@ def apply(c):
 
       subprocess.call(['ip l a dev ' + intf + ' type wireguard 2>/dev/null'], shell=True)
       for addr in c['interfaces'][intf]['addr']:
-        add_addr(intf, addr) 
-        configure_interface(c,intf)
-      subprocess.call(['ip l set up dev ' + intf + ' &>/dev/null'], shell=True)
+        add_addr(intf, addr)
+
+      subprocess.call(['ip l set up dev ' + intf + ' mtu ' + c['interfaces'][intf]['mtu'] + ' &>/dev/null'], shell=True)
+      configure_interface(c, intf)
 
     ### config updates
     if c['interfaces'][intf]['status'] == 'exists':
       ### IP address change
-      addr_eff  = re.sub("\'", "", c_eff.return_effective_values(intf + ' address')).split() 
-      addr_rem  = list( set(addr_eff) - set(c['interfaces'][intf]['addr']) )
-      addr_add  = list( set(c['interfaces'][intf]['addr']) - set(addr_eff) )
+      addr_eff = re.sub("\'", "", c_eff.return_effective_values(intf + ' address')).split()
+      addr_rem = list(set(addr_eff) - set(c['interfaces'][intf]['addr']))
+      addr_add = list(set(c['interfaces'][intf]['addr']) - set(addr_eff))
 
-      if len(addr_rem) !=0:
+      if len(addr_rem) != 0:
         for addr in addr_rem:
           del_addr(intf, addr)
 
-      if len(addr_add) !=0:
+      if len(addr_add) != 0:
         for addr in addr_add:
           add_addr(intf, addr)
 
-      ### persistent-keepalive 
+      ## mtu update
+      mtu = c['interfaces'][intf]['mtu']
+      if mtu != 1420:
+        sl.syslog(sl.LOG_NOTICE, "setting mtu to " + mtu + " on " + intf)
+        subprocess.call(['ip l set mtu ' + mtu + ' dev ' + intf + ' &>/dev/null'], shell=True)
+
+      ### persistent-keepalive
       for p in c_eff.list_nodes(intf + ' peer'):
         val_eff = ""
-        val = "" 
+        val = ""
 
         if c_eff.exists_effective(intf + ' peer ' + p + ' persistent-keepalive'):
           val_eff = c_eff.return_effective_value(intf + ' peer ' + p + ' persistent-keepalive')
 
         if 'persistent-keepalive' in c['interfaces'][intf]['peer'][p]:
           val = c['interfaces'][intf]['peer'][p]['persistent-keepalive']
-        
+
         ### disable keepalive
         if val_eff and not val:
-          c['interfaces'][intf]['peer'][p]['persistent-keepalive'] = 0 
-        
+          c['interfaces'][intf]['peer'][p]['persistent-keepalive'] = 0
+
         ### set new keepalive value
         if not val_eff and val:
           c['interfaces'][intf]['peer'][p]['persistent-keepalive'] = val
 
       ## wg command call
-      configure_interface(c,intf)
+      configure_interface(c, intf)
 
-    ### ifalias for snmp from description   
+    ### ifalias for snmp from description
     descr_eff = c_eff.return_effective_value(intf + ' description')
     cnf_descr = c['interfaces'][intf]['descr']
     if descr_eff != cnf_descr:
-      open('/sys/class/net/' + str(intf) + '/ifalias','w').write(str(cnf_descr))
+      with open('/sys/class/net/' + str(intf) + '/ifalias', 'w') as fh:
+        fh.write(str(cnf_descr))
 
 def configure_interface(c, intf):
+  wg_config = {
+      'interface'   : intf,
+      'port' : 0,
+      'private-key' : '/config/auth/wireguard/private.key',
+      'peer'        :
+          {
+              'pubkey'  : ''
+          },
+      'allowed-ips' : [],
+      'fwmark'      : 0x00,
+      'endpoint'    : None,
+      'keepalive'   : 0
+
+  }
+
   for p in c['interfaces'][intf]['peer']:
-    cmd = "wg set " + intf + \
-          " listen-port " + c['interfaces'][intf]['lport'] + \
-          " private-key " + pk + \
-          " peer " + p 
+    ## mandatory settings
+    wg_config['peer']['pubkey'] = c['interfaces'][intf]['peer'][p]['pubkey']
+    wg_config['allowed-ips'] = c['interfaces'][intf]['peer'][p]['allowed-ips']
+
+    ## optional settings
+    # listen-port
+    if c['interfaces'][intf]['lport']:
+      wg_config['port'] = c['interfaces'][intf]['lport']
+
+    ## endpoint
+    if c['interfaces'][intf]['peer'][p]['endpoint']:
+      wg_config['endpoint'] = c['interfaces'][intf]['peer'][p]['endpoint']
+
+    ## persistent-keepalive
+    if 'persistent-keepalive' in c['interfaces'][intf]['peer'][p]:
+      wg_config['keepalive'] = c['interfaces'][intf]['peer'][p]['persistent-keepalive']
+
+    ### assemble wg command
+    cmd = "sudo wg set " + intf
+    cmd += " listen-port " + str(wg_config['port'])
+    cmd += " private-key " + wg_config['private-key']
+    cmd += " peer " + wg_config['peer']['pubkey']
     cmd += " allowed-ips "
+    for ap in wg_config['allowed-ips']:
+      if ap != wg_config['allowed-ips'][-1]:
+        cmd += ap + ","
+      else:
+        cmd += ap
 
-  for ap in c['interfaces'][intf]['peer'][p]['allowed-ips']:
-    if ap != c['interfaces'][intf]['peer'][p]['allowed-ips'][-1]:
-      cmd += ap + ","
-    else:
-      cmd += ap
-
-  ## endpoint is only required if wg runs as client
-  if c['interfaces'][intf]['peer'][p]['endpoint']:
-    cmd += " endpoint " + c['interfaces'][intf]['peer'][p]['endpoint']
+    if wg_config['endpoint']:
+      cmd += " endpoint " + wg_config['endpoint']
 
-  if 'persistent-keepalive' in c['interfaces'][intf]['peer'][p]:
-    cmd += " persistent-keepalive " + str( c['interfaces'][intf]['peer'][p]['persistent-keepalive'])
+    if wg_config['keepalive'] != 0:
+      cmd += " persistent-keepalive " + wg_config['keepalive']
+    else:
+      cmd += " persistent-keepalive 0"
 
-  sl.syslog(sl.LOG_NOTICE, "sudo " + cmd)
-  subprocess.call([ 'sudo ' + cmd], shell=True)
+    sl.syslog(sl.LOG_NOTICE, cmd)
+    subprocess.call([cmd], shell=True)
 
 def add_addr(intf, addr):
   ret = subprocess.call(['ip a a dev ' + intf + ' ' + addr + ' &>/dev/null'], shell=True)
   if ret != 0:
-    raise ConfigError('Can\'t set IP ' + addr + ' on ' + intf )
+    raise ConfigError('Can\'t set IP ' + addr + ' on ' + intf)
   else:
     sl.syslog(sl.LOG_NOTICE, "ip a a dev " + intf + " " + addr)
 
 def del_addr(intf, addr):
   ret = subprocess.call(['ip a d dev ' + intf + ' ' + addr + ' &>/dev/null'], shell=True)
   if ret != 0:
-    raise ConfigError('Can\'t delete IP ' + addr + ' on ' + intf )
+    raise ConfigError('Can\'t delete IP ' + addr + ' on ' + intf)
   else:
     sl.syslog(sl.LOG_NOTICE, "ip a d dev " + intf + " " + addr)
 
@@ -265,9 +300,7 @@ if __name__ == '__main__':
     check_kmod()
     c = get_config()
     verify(c)
-    #generate(c)
     apply(c)
   except ConfigError as e:
     print(e)
     sys.exit(1)
-