From bfadd6dfb5969f231097353a76ada3b839964a19 Mon Sep 17 00:00:00 2001
From: sarthurdev <965089+sarthurdev@users.noreply.github.com>
Date: Tue, 20 Jul 2021 17:33:00 +0200
Subject: pki: eapol: T3642: Migrate EAPoL to use PKI configuration

---
 src/conf_mode/interfaces-ethernet.py | 31 +++++++++++++++++++++++++++++++
 1 file changed, 31 insertions(+)

(limited to 'src/conf_mode/interfaces-ethernet.py')

diff --git a/src/conf_mode/interfaces-ethernet.py b/src/conf_mode/interfaces-ethernet.py
index 378f400b8..78c24952b 100755
--- a/src/conf_mode/interfaces-ethernet.py
+++ b/src/conf_mode/interfaces-ethernet.py
@@ -32,6 +32,8 @@ from vyos.configverify import verify_vlan_config
 from vyos.configverify import verify_vrf
 from vyos.ethtool import Ethtool
 from vyos.ifconfig import EthernetIf
+from vyos.pki import wrap_certificate
+from vyos.pki import wrap_private_key
 from vyos.template import render
 from vyos.util import call
 from vyos.util import dict_search
@@ -40,6 +42,7 @@ from vyos import airbag
 airbag.enable()
 
 # XXX: wpa_supplicant works on the source interface
+cfg_dir = '/run/wpa_supplicant'
 wpa_suppl_conf = '/run/wpa_supplicant/{ifname}.conf'
 
 def get_config(config=None):
@@ -52,8 +55,15 @@ def get_config(config=None):
     else:
         conf = Config()
     base = ['interfaces', 'ethernet']
+
+    tmp_pki = conf.get_config_dict(['pki'], key_mangling=('-', '_'),
+                                get_first_key=True, no_tag_node_value_mangle=True)
+
     ethernet = get_interface_dict(conf, base)
 
+    if 'deleted' not in ethernet:
+        ethernet['pki'] = tmp_pki
+
     return ethernet
 
 def verify(ethernet):
@@ -126,6 +136,27 @@ def generate(ethernet):
     if 'eapol' in ethernet:
         render(wpa_suppl_conf.format(**ethernet),
                'ethernet/wpa_supplicant.conf.tmpl', ethernet)
+        
+        ifname = ethernet['ifname']
+        cert_file_path = os.path.join(cfg_dir, f'{ifname}_cert.pem')
+        cert_key_path = os.path.join(cfg_dir, f'{ifname}_cert.key')
+
+        cert_name = ethernet['eapol']['certificate']
+        pki_cert = ethernet['pki']['certificate'][cert_name]
+
+        with open(cert_file_path, 'w') as f:
+            f.write(wrap_certificate(pki_cert['certificate']))
+
+        with open(cert_key_path, 'w') as f:
+            f.write(wrap_private_key(pki_cert['private']['key']))
+
+        if 'ca_certificate' in ethernet['eapol']:
+            ca_cert_file_path = os.path.join(cfg_dir, f'{ifname}_ca.pem')
+            ca_cert_name = ethernet['eapol']['ca_certificate']
+            pki_ca_cert = ethernet['pki']['ca'][cert_name]
+
+            with open(ca_cert_file_path, 'w') as f:
+                f.write(wrap_certificate(pki_ca_cert['certificate']))
     else:
         # delete configuration on interface removal
         if os.path.isfile(wpa_suppl_conf.format(**ethernet)):
-- 
cgit v1.2.3