F5Client.cs

// Copyright 2023 Keyfactor                                                   
// Licensed under the Apache License, Version 2.0 (the "License"); you may    
// not use this file except in compliance with the License.  You may obtain a 
// copy of the License at http://www.apache.org/licenses/LICENSE-2.0.  Unless 
// required by applicable law or agreed to in writing, software distributed   
// under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES   
// OR CONDITIONS OF ANY KIND, either express or implied. See the License for  
// thespecific language governing permissions and limitations under the       
// License. 
using Keyfactor.Orchestrators.Extensions;
using Keyfactor.Orchestrators.Common.Enums;
using Keyfactor.PKI.X509;
using Keyfactor.PKI.PEM;
using Microsoft.Extensions.Logging;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Security.Cryptography.X509Certificates;
using System.Text;
using System.Text.RegularExpressions;

using Newtonsoft.Json;
using System.Collections;

namespace Keyfactor.Extensions.Orchestrator.F5Orchestrator
{
    internal class F5Client
    {
        #region Properties

        protected ILogger logger;

        private const string INVALID_KEY_MSG_ID = "01070317:3";
        private const string INVALID_KEY_SUBSTR = "key(";
        private const string INVALID_KEY_BEG_DELIM = @"/";
        private const string INVALID_KEY_END_DELIM = ")";

        public CertificateStore CertificateStore { get; set; }
        public string ServerUserName { get; set; }
        public string ServerPassword { get; set; }
        public string Token { get; set; }
        public bool UseSSL { get; set; }
        public string PFXPassword { get; set; }
        public IEnumerable<PreviousInventoryItem> Inventory { get; set; }
        public string PrimaryNode { get; set; }
        public string F5Version { get; set; }
        public bool IgnoreSSLWarning { get; set; }
        public bool UseTokenAuth { get; set; }
        private RESTHandler REST { get; set; }
        private F5Transaction Transaction { get; set; }

        // Properties
        #endregion

        #region Constructors

        public F5Client(CertificateStore certificateStore, string serverUserName, string serverPassword, bool useSSL, string pfxPassword, bool ignoreSSLWarning, bool useTokenAuth, IEnumerable<PreviousInventoryItem> inventory)
        {
            CertificateStore = certificateStore;
            ServerUserName = serverUserName;
            ServerPassword = serverPassword;
            UseSSL = useSSL;
            PFXPassword = pfxPassword;
            IgnoreSSLWarning = ignoreSSLWarning;
            UseTokenAuth = useTokenAuth;
            Inventory = inventory;

            if (logger == null)
            {
                logger = Keyfactor.Logging.LogHandler.GetClassLogger(this.GetType());
            }

            REST = new RESTHandler(certificateStore.ClientMachine, serverUserName, serverPassword, useSSL, IgnoreSSLWarning);

            if (UseTokenAuth)
                REST.Token = GetToken(serverUserName, serverPassword);
        }

        // Constructors
        #endregion

        #region Methods

        #region Certificate/PFX Shared

        public void AddEntry(string partition, string name, string b64Certificate, string certificatePassword)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "AddEntry");
            LogHandlerCommon.Trace(logger, CertificateStore, $"Processing certificate for partition '{partition}' and name '{name}'");
            byte[] entryContents = Convert.FromBase64String(b64Certificate);
            string password = PFXPassword;
            CertificateConverter converter = CertificateConverterFactory.FromDER(entryContents, password);
            X509Certificate2 certificate = converter.ToX509Certificate2(password);
            if (certificate.HasPrivateKey)
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"Certificate for partition '{partition}' and name '{name}' has a private key - performing addition");
                AddPfx(entryContents, partition, name, password, null, certificatePassword);
                LogHandlerCommon.Trace(logger, CertificateStore, $"PFX addition for partition '{partition}' and name '{name}' completed");
            }
            else
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"Certificate for partition '{partition}' and name '{name}' does not have a private key - performing addition");
                AddCertificate(entryContents, partition, name);
                LogHandlerCommon.Trace(logger, CertificateStore, $"Certificate addition for partition '{partition}' and name '{name}' completed");
            }
            LogHandlerCommon.MethodExit(logger, CertificateStore, "AddEntry");
        }

        public void ReplaceEntry(string partition, string name, string b64Certificate, string certificatePassword)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "ReplaceEntry");
            LogHandlerCommon.Trace(logger, CertificateStore, $"Processing certificate for partition '{partition}' and name '{name}'");
            byte[] entryContents = Convert.FromBase64String(b64Certificate);
            string password = PFXPassword;
            CertificateConverter converter = CertificateConverterFactory.FromDER(entryContents, password);
            X509Certificate2 certificate = converter.ToX509Certificate2(password);

            if (certificate.HasPrivateKey)
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"Certificate for partition '{partition}' and name '{name}' has a private key - performing replacement");
                ReplacePfx(entryContents, partition, name, password, certificatePassword);
                LogHandlerCommon.Trace(logger, CertificateStore, $"PFX replacement for partition '{partition}' and name '{name}' completed");
            }
            else
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"Certificate for partition '{partition}' and name '{name}' does not have a private key - performing replacement");
                ReplaceCertificate(entryContents, partition, name);
                LogHandlerCommon.Trace(logger, CertificateStore, $"Certificate replacement for partition '{partition}' and name '{name}' completed");
            }
            LogHandlerCommon.MethodExit(logger, CertificateStore, "ReplaceEntry");
        }

        public void RemoveEntry(string partition, string name)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "RemoveEntry");
            LogHandlerCommon.Trace(logger, CertificateStore, $"Processing certificate for partition '{partition}' and name '{name}'");
            string timestamp = DateTime.Now.ToString("MM-dd-yy:H:mm:ss");
            if (KeyExists(partition, name))
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"Archiving key at '{partition}' and name '{name}'");
                ArchiveFile($"/config/filestore/files_d/{partition}_d/certificate_key_d/:{partition}:{name}_*", $"{partition}-{name}-{timestamp}.key");
                LogHandlerCommon.Trace(logger, CertificateStore, $"Removing certificate and key at '{partition}' and name '{name}'");

                string keyName = GetKeyName(name, true);
                REST.Delete($"/mgmt/tm/sys/file/ssl-key/~{partition}~{keyName}");
            }
            LogHandlerCommon.Trace(logger, CertificateStore, $"Archiving certificate at '{partition}' and name '{name}'");
            ArchiveFile($"/config/filestore/files_d/{partition}_d/certificate_d/:{partition}:{name}_*", $"{partition}-{name}-{timestamp}.crt");
            LogHandlerCommon.Trace(logger, CertificateStore, $"Removing certificate at '{partition}' and name '{name}'");

            string crtName = GetCrtName(name, true);
            REST.Delete($"/mgmt/tm/sys/file/ssl-cert/~{partition}~{crtName}");
            LogHandlerCommon.MethodExit(logger, CertificateStore, "RemoveEntry");
        }

        public bool KeyExists(string partition, string name)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "KeyExists");
            bool exists = false;

            try
            {
                string keyName = GetKeyName(name, true);
                string query = $"/mgmt/tm/sys/file/ssl-key/~{partition}~{keyName}";
                F5Key key = REST.Get<F5Key>(query);
                exists = (key != null);
            }
            catch (F5RESTException rex)
            {
                // A 404 will be returned if the key is not found
                if (rex.code != 404)
                {
                    throw;
                }
            }

            LogHandlerCommon.MethodExit(logger, CertificateStore, "KeyExists");
            return exists;
        }

        public bool CertificateExists(string partition, string name)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "CertificateExists");
            bool exists = false;

            try
            {
                string crtName = GetCrtName(name, true);
                string query = $"/mgmt/tm/sys/file/ssl-cert/~{partition}~{crtName}";
                F5SSLProfile certificate = REST.Get<F5SSLProfile>(query);
                exists = (certificate != null);
            }
            catch (F5RESTException rex)
            {
                // A 404 will be returned if the certificate is not found
                if (rex.code != 404)
                {
                    throw;
                }
            }

            LogHandlerCommon.MethodExit(logger, CertificateStore, "CertificateExists");
            return exists;
        }

        private void AddCertificate(byte[] entryContents, string partition, string name)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "AddCertificate");
            LogHandlerCommon.Trace(logger, CertificateStore, $"Uploading file to {partition}-{name}");
            REST.UploadFile($"{partition}-{name}", entryContents);

            LogHandlerCommon.Trace(logger, CertificateStore, $"Installing certificate to '{name}'");
            REST.PostInstallCryptoCommand(new F5InstallCommand
            {
                command = "install",
                name = $"{name}",
                localfile = $"/var/config/rest/downloads/{partition}-{name}",
                partition = partition
            }, "cert");
            LogHandlerCommon.MethodExit(logger, CertificateStore, "AddCertificate");
        }

        private void AddPfx(byte[] entryContents, string partition, string name, string password, string keyName, string certificatePassword)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "AddPfx");
            LogHandlerCommon.Trace(logger, CertificateStore, $"Uploading PFX to {partition}-{name}.p12");
            REST.UploadFile($"{partition}-{name}.p12", entryContents);

            LogHandlerCommon.Trace(logger, CertificateStore, $"Installing PFX to '{name}'");

            string keyNameParam = string.IsNullOrEmpty(keyName) ? string.Empty : $":key={keyName}";

            try
            {
                REST.PostInstallCryptoCommand(new F5InstallCommand
                {
                    command = "install",
                    name = $"{name}",
                    localfile = $"/var/config/rest/downloads/{partition}-{name}.p12{keyNameParam}",
                    passphrase = password,
                    keyPassphrase = String.IsNullOrEmpty(certificatePassword) ? string.Empty : certificatePassword,
                    keySecurityType = String.IsNullOrEmpty(certificatePassword) ? "normal" : "password",
                    partition = partition
                }, "pkcs12");
            }
            catch (F5RESTException ex)
            {
                // Check to see if the specific error message is produced that would mean the certificate and key names are 
                //  different.  The error message is then parsed to retrieve the key name and the /pkcs12 api call is tried
                //  again with that key name appended onto the localfile parameter.  An F5 hotfix is necessary to produce
                //  this message and use the updated /pkcs12 API that accepts the separate key name.
                if (string.IsNullOrEmpty(keyName) && ex.message.Contains(INVALID_KEY_MSG_ID))
                    AddPfx(entryContents, partition, name, password, GetKeyName(ex.message), certificatePassword);
                else
                    throw (name.Contains(".crt", StringComparison.OrdinalIgnoreCase) &&
                           ex.Message.Contains("expected to exist", StringComparison.OrdinalIgnoreCase) ?
                                new Exception("Certificate and Key name may be different. If so, an F5 hotfix may be required to allow for the automatic renewal of this certificate.", ex) :
                                ex);
            }

            LogHandlerCommon.MethodExit(logger, CertificateStore, "AddPfx");
        }

        // Method to parse error message from /pkcs12 API call that can occur when the certificate and key have different names.
        //  There is an F5 hotfix needed to be installed to produce the specific error message parsed by this method to get the
        //  separate key name.
        private string GetKeyName(string errorMessage)
        {
            int locBegSubstr = errorMessage.IndexOf(INVALID_KEY_SUBSTR) + INVALID_KEY_SUBSTR.Length;
            string errorMessageSubstr = errorMessage.Substring(locBegSubstr);
            int locEndSubstr = errorMessageSubstr.IndexOf(INVALID_KEY_END_DELIM);
            errorMessageSubstr = errorMessageSubstr.Substring(0, locEndSubstr);
            locBegSubstr = errorMessageSubstr.LastIndexOf(INVALID_KEY_BEG_DELIM);
            string keyName = errorMessageSubstr.Substring(locBegSubstr + 1);

            return keyName;
        }

        private void ReplaceCertificate(byte[] entryContents, string partition, string name)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "ReplaceCertificate");
            string timestamp = DateTime.Now.ToString("MM-dd-yy:H:mm:ss");

            LogHandlerCommon.Trace(logger, CertificateStore, $"Archiving the certificate for partition '{partition}' and name '{name}'");
            ArchiveFile($"/config/filestore/files_d/{partition}_d/certificate_d/:{partition}:{name}_*", $"{partition}-{name}-{timestamp}.crt");

            LogHandlerCommon.Trace(logger, CertificateStore, $"Adding certificate to partition '{partition}' and name '{name}'");
            AddCertificate(entryContents, partition, name);
            LogHandlerCommon.MethodExit(logger, CertificateStore, "ReplaceCertificate");
        }

        private void ReplacePfx(byte[] entryContents, string partition, string name, string password, string certificatePassword)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "ReplacePfx");
            string timestamp = DateTime.Now.ToString("MM-dd-yy:H:mm:ss");

            LogHandlerCommon.Trace(logger, CertificateStore, $"Archiving the key and certificate for partition '{partition}' and name '{name}'");
            ArchiveFile($"/config/filestore/files_d/{partition}_d/certificate_key_d/:{partition}:{name}_*", $"{partition}-{name}-{timestamp}.key");
            ArchiveFile($"/config/filestore/files_d/{partition}_d/certificate_d/:{partition}:{name}_*", $"{partition}-{name}-{timestamp}.crt");

            LogHandlerCommon.Trace(logger, CertificateStore, $"Adding PFX to partition '{partition}' and name '{name}'");
            AddPfx(entryContents, partition, name, password, null, certificatePassword);
            LogHandlerCommon.MethodExit(logger, CertificateStore, "ReplacePfx");
        }

        private X509Certificate2Collection GetCertificateEntry(string path)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "GetCertificateEntry");
            string certificateEntry = string.Empty;
            LogHandlerCommon.Trace(logger, CertificateStore, $"Getting certificate entry from: '{path}'");

            string crt = REST.PostBASHCommand(new F5BashCommand
            {
                command = "run",
                utilCmdArgs = $"-c 'cat {path} | base64'"
            });

            byte[] crtBytes;
            switch (crt.Substring(0, 1))
            {
                //DER --> M
                case "M":
                    LogHandlerCommon.Trace(logger, CertificateStore, "Certificate is DER encoded");
                    certificateEntry = crt;
                    break;
                //PEM(no headers)-- > T
                case "T":
                    LogHandlerCommon.Trace(logger, CertificateStore, "Certificate is PEM without headers");
                    crtBytes = System.Convert.FromBase64String(crt);
                    certificateEntry = System.Text.ASCIIEncoding.ASCII.GetString(crtBytes);
                    break;
                //PEM(w / headers)-- > L or I
                case "L":
                case "I":
                    LogHandlerCommon.Trace(logger, CertificateStore, "Certificate is PEM with headers");
                    crtBytes = System.Convert.FromBase64String(crt);
                    certificateEntry = System.Text.ASCIIEncoding.ASCII.GetString(crtBytes);
                    break;
                default:
                    crtBytes = System.Convert.FromBase64String(crt);
                    certificateEntry = System.Text.ASCIIEncoding.ASCII.GetString(crtBytes);
                    throw new Exception("Unknown certificate format found for device certificate");
            }

            LogHandlerCommon.MethodExit(logger, CertificateStore, "GetCertificateEntry");
            //LogHandlerCommon.Debug(logger, CertificateStore, certificateEntry);
            string certificateEntryAfterRemovalOfDelim = certificateEntry.Replace("-----BEGIN CERTIFICATE----- ", "-----BEGIN CERTIFICATE-----");
            CertificateCollectionConverter c = CertificateCollectionConverterFactory.FromPEM(certificateEntryAfterRemovalOfDelim);

            return c.ToX509Certificate2Collection();
        }

        private void SetItemStatus(CurrentInventoryItem agentInventoryItem)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "SetItemStatus");
            PreviousInventoryItem keyfactorInventoryItem = Inventory
                .SingleOrDefault(i => i.Alias.Equals(agentInventoryItem.Alias, StringComparison.OrdinalIgnoreCase));
            if (keyfactorInventoryItem == null)
            {
                LogHandlerCommon.Trace(logger, CertificateStore, "F5 item does not exist in Keyfactor Command and will be tagged as new");
                agentInventoryItem.ItemStatus = OrchestratorInventoryItemStatus.New;
                LogHandlerCommon.MethodExit(logger, CertificateStore, "SetItemStatus");
                return;
            }

            LogHandlerCommon.Trace(logger, CertificateStore, "Matching alias found, checking entry details");
            if (keyfactorInventoryItem.PrivateKeyEntry != agentInventoryItem.PrivateKeyEntry)
            {
                LogHandlerCommon.Trace(logger, CertificateStore, "Private key entry status does not match and will be tagged as modified");
                agentInventoryItem.ItemStatus = OrchestratorInventoryItemStatus.Modified;
                LogHandlerCommon.MethodExit(logger, CertificateStore, "SetItemStatus");
                return;
            }

            LogHandlerCommon.Trace(logger, CertificateStore, "Private key entry status matches, checking certificates");
            if (keyfactorInventoryItem.Thumbprints.Count() != agentInventoryItem.Certificates.Count())
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"F5 entry certificate count: {agentInventoryItem.Certificates.Count()} does not match Keyfactor Command count: {keyfactorInventoryItem.Thumbprints.Count()} and will be tagged as modified");
                agentInventoryItem.ItemStatus = OrchestratorInventoryItemStatus.Modified;
                LogHandlerCommon.MethodExit(logger, CertificateStore, "SetItemStatus");
                return;
            }

            LogHandlerCommon.Trace(logger, CertificateStore, "Certificate counts match, checking individual certificates");
            foreach (string pem in agentInventoryItem.Certificates)
            {
                string certificateBase64 = pem;
                certificateBase64 = Regex.Replace(certificateBase64, "\r", "");
                certificateBase64 = Regex.Replace(certificateBase64, "-----BEGIN CERTIFICATE-----\n", "");
                certificateBase64 = Regex.Replace(certificateBase64, "\\n-----END CERTIFICATE-----(\\n|)", "");

                LogHandlerCommon.Trace(logger, CertificateStore, "Getting X509 object from F5 certificate pem");
                X509Certificate2 x509 = new X509Certificate2(Convert.FromBase64String(certificateBase64));
                LogHandlerCommon.Trace(logger, CertificateStore, $"Looking for CMS thumbprint matching: '{x509.Thumbprint}'");
                if (!keyfactorInventoryItem.Thumbprints.Any(t => t.Equals(x509.Thumbprint, StringComparison.OrdinalIgnoreCase)))
                {
                    LogHandlerCommon.Trace(logger, CertificateStore, "Thumbprint not found and will be tagged as modified");
                    agentInventoryItem.ItemStatus = OrchestratorInventoryItemStatus.Modified;
                    LogHandlerCommon.MethodExit(logger, CertificateStore, "SetItemStatus");
                    return;
                }
            }

            LogHandlerCommon.Trace(logger, CertificateStore, "The inventory item is unchanged");
            agentInventoryItem.ItemStatus = OrchestratorInventoryItemStatus.Unchanged;
            LogHandlerCommon.MethodExit(logger, CertificateStore, "SetItemStatus");
        }

        private CurrentInventoryItem GetInventoryItem(string partition, string name, bool hasPrivateKey)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "GetInventoryItem");

            // Get the pfx/certificate contents from the filesystem (using a wildcard as the files have slightly randomized name suffixes)
            X509Certificate2Collection certificateCollection = GetCertificateEntry($"/config/filestore/files_d/{partition}_d/certificate_d/:{partition}:{name}_*");
            List<string> certContents = new List<string>();
            bool useChainLevel = certificateCollection.Count > 1;
            foreach (X509Certificate2 certificate in certificateCollection)
            {
                certContents.Add(Convert.ToBase64String(certificate.Export(X509ContentType.Cert)));
                //LogHandlerCommon.Debug(logger, CertificateStore, $"ALIAS: {name}: {Convert.ToBase64String(certificate.Export(X509ContentType.Cert))}");
            }

            string crtName = GetCrtName(name, false);
            CurrentInventoryItem inventoryItem = new CurrentInventoryItem
            {
                ItemStatus = OrchestratorInventoryItemStatus.Unknown,
                Alias = crtName,
                PrivateKeyEntry = hasPrivateKey,
                UseChainLevel = useChainLevel,
                Certificates = certContents.ToArray()
            };
            SetItemStatus(inventoryItem);
            LogHandlerCommon.MethodExit(logger, CertificateStore, "GetInventoryItem");
            return inventoryItem;
        }

        private string GetCrtName(string name, bool addExtension)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "GetCrtName");
            string crtName = name;

            switch (F5Version.ToLowerInvariant())
            {
                case "v12":
                    throw new Exception($"F5 Version 12 is not supported by the REST-based orchestrator. The legacy SOAP-based orchestrator should be used.");
                case "v13":
                    if (addExtension)
                    {
                        // The .crt extension must be added
                        if (!crtName.EndsWith(".crt", StringComparison.OrdinalIgnoreCase)) { crtName = $"{crtName}.crt"; }
                    }
                    else
                    {
                        // The .crt extension must be removed
                        if (crtName.EndsWith(".crt", StringComparison.OrdinalIgnoreCase)) { crtName = crtName.Substring(0, crtName.Length - 4); }
                    }
                    break;
            };

            LogHandlerCommon.MethodExit(logger, CertificateStore, "GetCrtName");
            return crtName;
        }

        private string GetKeyName(string name, bool addExtension)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "GetKeyName");
            string keyName = name;

            // No longer checking past version 14 for future-proofing
            switch (F5Version.ToLowerInvariant())
            {
                case "v12":
                    throw new Exception($"F5 Version 12 is not supported by the REST-based orchestrator. The legacy SOAP-based orchestrator should be used.");
                case "v13":
                    if (addExtension)
                    {
                        // The .key extension must be added
                        if (!keyName.EndsWith(".key", StringComparison.OrdinalIgnoreCase)) { keyName = $"{keyName}.key"; }
                    }
                    else
                    {
                        // The .key extension must be removed
                        if (keyName.EndsWith(".key", StringComparison.OrdinalIgnoreCase)) { keyName = keyName.Substring(0, keyName.Length - 4); }
                    }
                    break;
            };

            LogHandlerCommon.MethodExit(logger, CertificateStore, "GetKeyName");
            return keyName;
        }

        // Certificate PFX Shared
        #endregion

        #region Infrastructure

        public bool PrimaryNodeActive()
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "PrimaryNodeActive");
            F5NodeDevice device = REST.Get<F5NodeDevice>($"/mgmt/tm/cm/device/{PrimaryNode}?$select=name,failoverState");
            bool nodeActive = device.failoverState.Equals("active", StringComparison.OrdinalIgnoreCase);
            LogHandlerCommon.MethodExit(logger, CertificateStore, "PrimaryNodeActive");

            return nodeActive;
        }

        public string GetActiveNode()
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "GetActiveNode");
            string activeNode = string.Empty;
            F5NodeDeviceList devices = REST.Get<F5NodeDeviceList>($"/mgmt/tm/cm/device?$select=name,failoverState");
            foreach (F5NodeDevice device in devices.items)
            {
                if (device.failoverState.Equals("active", StringComparison.OrdinalIgnoreCase))
                {
                    activeNode = device.name;
                    break;
                }
            }
            if (!string.IsNullOrEmpty(activeNode))
            {
                LogHandlerCommon.Warn(logger, CertificateStore, "No active node found, returning an empty device name");
            }
            else
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"'{activeNode}' is currently active and will be considered the primary node");
            }
            LogHandlerCommon.MethodExit(logger, CertificateStore, "GetActiveNode");

            return activeNode;
        }

        public List<F5Partition> GetPartitions()
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "GetPartitions");
            F5PartitionList partitions = REST.Get<F5PartitionList>($"/mgmt/tm/auth/partition?$select=name,fullPath");
            LogHandlerCommon.MethodExit(logger, CertificateStore, "GetPartitions");

            return partitions.items.ToList<F5Partition>();
        }

        public string GetPartitionFromStorePath()
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "GetPartitionsFromStorePath");
            string[] pathParts = CertificateStore.StorePath.Split("/".ToCharArray(), StringSplitOptions.RemoveEmptyEntries);
            if (pathParts.Length < 1) { throw new Exception($"The store path '{CertificateStore.StorePath}' does not appear to contain a partition"); }
            LogHandlerCommon.MethodExit(logger, CertificateStore, "GetPartitionFromStorePath");
            return pathParts[0];
        }

        // Infrastructure
        #endregion

        #region Web Server

        public List<CurrentInventoryItem> GetWebServerInventory()
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "GetWebServerInventory");
            List<CurrentInventoryItem> inventory = new List<CurrentInventoryItem>();
            X509Certificate2Collection certificateCollection = GetCertificateEntry("/config/httpd/conf/ssl.crt/server.crt");
            List<string> webServerInventory = new List<string>();
            foreach (X509Certificate2 certificate in certificateCollection)
            {
                webServerInventory.Add(Convert.ToBase64String(certificate.RawData));
            }

            LogHandlerCommon.Trace(logger, CertificateStore, $"Obtained F5 web server device inventory");
            CurrentInventoryItem inventoryItem = new CurrentInventoryItem
            {
                Alias = "WebServer",
                PrivateKeyEntry = true,
                ItemStatus = OrchestratorInventoryItemStatus.Unknown,
                UseChainLevel = true,
                Certificates = webServerInventory.ToArray()
            };
            SetItemStatus(inventoryItem);
            inventory.Add(inventoryItem);
            LogHandlerCommon.MethodExit(logger, CertificateStore, "GetWebServerInventory");

            return inventory;
        }

        public void ReplaceWebServerCrt(string b64Certificate)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "ReplaceWebServerCrt");
            LogHandlerCommon.Trace(logger, CertificateStore, "Processing web server certificate");
            byte[] devicePfx = Convert.FromBase64String(b64Certificate);
            string password = PFXPassword;
            CertificateCollectionConverter converter = CertificateCollectionConverterFactory.FromDER(devicePfx, password);
            string pfxPem = converter.ToPEM(password);
            List<X509Certificate2> clist = converter.ToX509Certificate2List(password);

            StringBuilder certPemBuilder = new StringBuilder();

            //reordering of certificate chain necessary because of BouncyCastle bug.  Being fixed in a later release
            if (clist.Count > 1)
                clist = ReorderPEMLIst(clist);

            LogHandlerCommon.Trace(logger, CertificateStore, "Building certificate PEM");
            foreach (X509Certificate2 cert in clist)
            {
                certPemBuilder.AppendLine("-----BEGIN CERTIFICATE-----");
                certPemBuilder.AppendLine(
                    Convert.ToBase64String(cert.RawData, Base64FormattingOptions.InsertLineBreaks));
                certPemBuilder.AppendLine("-----END CERTIFICATE-----");
            }

            LogHandlerCommon.Trace(logger, CertificateStore, "Building key PEM");
            byte[] pkBytes = Keyfactor.PKI.PrivateKeys.PrivateKeyConverterFactory.FromPKCS12(devicePfx, password).ToPkcs8BlobUnencrypted();
            StringBuilder keyPemBuilder = new StringBuilder();
            keyPemBuilder.AppendLine("-----BEGIN PRIVATE KEY-----");
            keyPemBuilder.AppendLine(
                Convert.ToBase64String(pkBytes, Base64FormattingOptions.InsertLineBreaks));
            keyPemBuilder.AppendLine("-----END PRIVATE KEY-----");

            string timestamp = DateTime.Now.ToString("MM-dd-yy:H:mm:ss");

            LogHandlerCommon.Trace(logger, CertificateStore, "Uploading web server certificate");
            byte[] certbytes = Encoding.ASCII.GetBytes(certPemBuilder.ToString());
            RESTHandler rest = REST;
            rest.UploadFile($"kyfcert-{timestamp}.crt", certbytes);

            LogHandlerCommon.Trace(logger, CertificateStore, "Uploading the web server key");
            byte[] keybytes = Encoding.ASCII.GetBytes(keyPemBuilder.ToString());
            rest.UploadFile($"kyfcert-{timestamp}.key", keybytes);

            F5Transaction transaction = BeginTransaction();

            LogHandlerCommon.Trace(logger, CertificateStore, "Archiving the web server certificate and key");
            ArchiveFile("/config/httpd/conf/ssl.key/server.key", $"server-{timestamp}.key", transaction.transid);
            ArchiveFile("/config/httpd/conf/ssl.crt/server.crt", $"server-{timestamp}.crt", transaction.transid);

            LogHandlerCommon.Trace(logger, CertificateStore, "Replacing the web server certificate and key");
            CopyFile($"/var/config/rest/downloads/kyfcert-{timestamp}.key", "/config/httpd/conf/ssl.key/server.key", transaction.transid);
            CopyFile($"/var/config/rest/downloads/kyfcert-{timestamp}.crt", "/config/httpd/conf/ssl.crt/server.crt", transaction.transid);

            CommitTransaction(transaction);
            LogHandlerCommon.MethodExit(logger, CertificateStore, "ReplaceWebServerCrt");
        }

        // Put certificate chain in proper order - EE => issuing => intermediate1 => ... => intermediateN => root
        private List<X509Certificate2> ReorderPEMLIst(List<X509Certificate2> certList)
        {
            List<X509Certificate2> rtnList = new List<X509Certificate2>();
            X509Certificate2 root = certList.FirstOrDefault(p => p.IssuerName.RawData.SequenceEqual(p.SubjectName.RawData));
            if (root == null || string.IsNullOrEmpty(root.SerialNumber))
                throw new Exception("Invalid certificate chain.  No root CA certificate found.");

            rtnList.Add(root);

            X509Certificate2 parentCert = root;
            for (int i=1; i<certList.Count; i++)
            {
                X509Certificate2 childCert = certList.FirstOrDefault(p => p.IssuerName.RawData.SequenceEqual(parentCert.SubjectName.RawData) && !p.IssuerName.RawData.SequenceEqual(p.SubjectName.RawData));
                if (root == null || string.IsNullOrEmpty(root.SerialNumber))
                    throw new Exception("Invalid certificate chain.  End entity or issuing CA certificate not found.");

                rtnList.Insert(0, childCert);
                parentCert = childCert;
            }

            return rtnList;
        }

        // WebServer
        #endregion

        #region SSL Profiles

        public List<CurrentInventoryItem> GetSSLProfiles(int pageSize)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "GetSSLProfiles");
            string partition = CertificateStore.StorePath;
            string query = $"/mgmt/tm/sys/file/ssl-cert?$filter=partition+eq+{partition}&$select=name,keyType,isBundle&$top={pageSize}&$skip=0";
            F5PagedSSLProfiles pagedProfiles = REST.Get<F5PagedSSLProfiles>(query);
            List<F5SSLProfile> profiles = new List<F5SSLProfile>();
            List<CurrentInventoryItem> inventory = new List<CurrentInventoryItem>();

            if (pagedProfiles.totalItems == 0 || pagedProfiles.items?.Length == 0)
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"No SSL profiles found in partition '{partition}'");
                LogHandlerCommon.MethodExit(logger, CertificateStore, "GetSSLProfiles");
                return inventory;
            }
            else
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"Compiling {pagedProfiles.totalPages} pages containing {pagedProfiles.totalItems} total inventory entries");
            }

            // Collected all of the profile entry names
            for (int i = pagedProfiles.pageIndex; i <= pagedProfiles.totalPages; i++)
            {
                profiles.AddRange(pagedProfiles.items);

                // The current paged profile will contain a link to the next set, unless the end has been reached
                if (string.IsNullOrEmpty(pagedProfiles.nextLink)) { break; }

                // Get the next page of profiles
                query = pagedProfiles.nextLink.Replace("https://localhost", "");
                pagedProfiles = REST.Get<F5PagedSSLProfiles>(query);
            }

            // Compile the entries into inventory items
            for (int i = 0; i < profiles.Count; i++)
            {
                try
                {
                    LogHandlerCommon.Trace(logger, CertificateStore, $"Processing alias {profiles[i].name}");
                    // Exclude 'ca-bundle.crt' as that can only be managed by F5
                    if (profiles[i].name.Equals("ca-bundle.crt", StringComparison.OrdinalIgnoreCase)
                        || profiles[i].name.Equals("f5-ca-bundle.crt", StringComparison.OrdinalIgnoreCase))
                    {
                        LogHandlerCommon.Trace(logger, CertificateStore, $"Skipping '{profiles[i].name}' because it is managed by F5");
                        continue;
                    }
                    inventory.Add(GetInventoryItem(partition, profiles[i].name, true));
                }
                catch (Exception ex)
                {
                    LogHandlerCommon.Error(logger, CertificateStore, ExceptionHandler.FlattenExceptionMessages(ex, $"Unable to process inventory item {profiles[i].name}."));
                }
            }

            LogHandlerCommon.MethodExit(logger, CertificateStore, "GetSSLProfiles");
            return inventory;
        }

        // SSL Profiles
        #endregion

        #region Auth

        private string GetToken(string userName, string userPassword)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "GetToken");
            F5LoginRequest request = new F5LoginRequest() { username = userName, password = userPassword, loginProviderName = "tmos" };
            F5LoginResponse loginResponse = REST.Post<F5LoginResponse>($"/mgmt/shared/authn/login", JsonConvert.SerializeObject(request));
            LogHandlerCommon.MethodExit(logger, CertificateStore, "GetToken");

            return loginResponse.token.token;
        }
        #endregion

        #region Bundles

        public List<F5CABundle> GetCABundles(string partition, int pageSize)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "GetCABundles");
            string query = $"/mgmt/tm/sys/crypto/ca-bundle-manager?$filter=partition+eq+{partition}&$select=name,fullPath&$top={pageSize}&$skip=0";
            F5PagedCABundles pagedBundles = REST.Get<F5PagedCABundles>(query);
            List<F5CABundle> bundles = new List<F5CABundle>();

            if (pagedBundles.totalItems == 0 || pagedBundles.items?.Length == 0)
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"No CA bundles found in partition '{partition}'");
                LogHandlerCommon.MethodExit(logger, CertificateStore, "GetCABundles");
                return bundles;
            }
            else
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"Compiling {pagedBundles.totalPages} pages containing {pagedBundles.totalItems} total CA bundles");
            }

            for (int i = pagedBundles.pageIndex; i <= pagedBundles.totalPages; i++)
            {
                foreach (F5CABundle bundle in pagedBundles.items)
                {
                    // Remove 'ca-bundle'
                    if (bundle.name.Equals("ca-bundle", StringComparison.OrdinalIgnoreCase))
                    {
                        LogHandlerCommon.Trace(logger, CertificateStore, $"Skipping '{bundle.name}' because it is managed by F5");
                        continue;
                    }
                    bundles.Add(bundle);
                }

                if (string.IsNullOrEmpty(pagedBundles.nextLink)) { break; }

                query = pagedBundles.nextLink.Replace("https://localhost", "");
                pagedBundles = REST.Get<F5PagedCABundles>(query);
            }

            LogHandlerCommon.MethodExit(logger, CertificateStore, "GetCABundles");
            return bundles;
        }

        public List<CurrentInventoryItem> GetCABundleInventory()
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "GetCABundleInventory");
            List<CurrentInventoryItem> inventory = new List<CurrentInventoryItem>();
            string[] includeBundle = GetCABundleIncludes();

            LogHandlerCommon.Trace(logger, CertificateStore, $"Compiling {includeBundle.Length} bundled certificates");
            for (int i = 0; i < includeBundle.Length; i++)
            {
                try
                {
                    LogHandlerCommon.Trace(logger, CertificateStore, $"Processing certificate '{includeBundle[i]}'");
                    string[] crtPathParts = includeBundle[i].Split("/".ToCharArray(), StringSplitOptions.RemoveEmptyEntries);
                    if (crtPathParts.Length != 2) { throw new Exception($"Bundled certificate path: '{includeBundle[i]}' is invalid.  Expecting '/<partition>/<certificate>'."); }
                    string partition = crtPathParts[0];
                    string crtName = crtPathParts[1];

                    LogHandlerCommon.Trace(logger, CertificateStore, $"Adding inventory item for partition '{partition}' and name '{crtName}'");
                    inventory.Add(GetInventoryItem(partition, crtName, false));
                }
                catch (Exception ex)
                {
                    LogHandlerCommon.Error(logger, CertificateStore, ExceptionHandler.FlattenExceptionMessages(ex, $"Unable to process inventory item {includeBundle[i]}."));
                }
            }

            LogHandlerCommon.MethodExit(logger, CertificateStore, "GetCABundleInventory");
            return inventory;
        }

        public bool EntryExistsInBundle(string alias)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "EntryExistsInBundle");
            bool exists = false;

            List<string> bundleIncludes = new List<string>(GetCABundleIncludes());
            string partition = GetPartitionFromStorePath();

            string crtName = GetCrtName(alias, true);
            exists = bundleIncludes.Any<string>(i => i.Equals($"/{partition}/{crtName}", StringComparison.OrdinalIgnoreCase));

            LogHandlerCommon.MethodExit(logger, CertificateStore, "EntryExistsInBundle");
            return exists;
        }

        private string[] GetCABundleIncludes()
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "GetCABundleIncludes");
            string bundlePath = CertificateStore.StorePath;
            string[] bundlePathParts = bundlePath.Split("/".ToCharArray(), StringSplitOptions.RemoveEmptyEntries);
            if (bundlePathParts.Length != 2) { throw new Exception($"CA bundle path: '{bundlePath}' is invalid.  Expecting '/<partition>/<bundle>'."); }
            string partition = bundlePathParts[0];
            string bundleName = bundlePathParts[1];
            string query = $"/mgmt/tm/sys/crypto/ca-bundle-manager/~{partition}~{bundleName}?$select=includeBundle";
            F5CABundle bundle = REST.Get<F5CABundle>(query);
            string[] includeBundle;
            if (bundle.includeBundle == null)
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"Found 0 included bundles");
                includeBundle = new List<string>().ToArray();
            }
            else
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"Found {bundle.includeBundle.Count()} included bundles");
                includeBundle = bundle.includeBundle;
            }

            LogHandlerCommon.MethodExit(logger, CertificateStore, "GetCABundleIncludes");
            return includeBundle;
        }

        public void AddBundleEntry(string bundle, string partition, string name, string b64Certificate, string alias, bool overwrite)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "AddBundleEntry");

            // Add the entry to inventory
            if (!CertificateExists(partition, name))
            {
                LogHandlerCommon.Debug(logger, CertificateStore, $"Add entry '{name}' in '{CertificateStore.StorePath}'");
                AddEntry(partition, name, b64Certificate, null);
            }
            else
            {
                if (!overwrite) { throw new Exception($"An entry named '{name}' exists and 'overwrite' was not selected"); }

                LogHandlerCommon.Debug(logger, CertificateStore, $"Replace entry '{name}' in '{CertificateStore.StorePath}'");
                ReplaceEntry(partition, name, b64Certificate, null);
            }

            // Add the entry to the bundle
            string crtName = GetCrtName(name, true);
            string crt = $"/{partition}/{crtName}";
            List<string> bundleIncludes = new List<string>(GetCABundleIncludes());
            if (!bundleIncludes.Contains(crt))
            {
                bundleIncludes.Add(crt);
                F5BundleInclude bundleInclude = new F5BundleInclude { includeBundle = bundleIncludes.ToArray() };
                REST.Patch<F5BundleInclude>($"/mgmt/tm/sys/crypto/ca-bundle-manager/{bundle.Replace('/', '~')}", bundleInclude);
            }
            LogHandlerCommon.MethodExit(logger, CertificateStore, "AddBundleEntry");
        }

        public void RemoveBundleEntry(string bundle, string partition, string name)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "RemoveBundleEntry");

            string crtName = GetCrtName(name, true);
            string crtEntry = $"/{partition}/{crtName}";

            LogHandlerCommon.Trace(logger, CertificateStore, $"Preparing to remove bundle entry '{crtEntry}'");
            List<string> bundleIncludes = new List<string>(GetCABundleIncludes());
            if (bundleIncludes.Contains(crtEntry))
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"The current bundle contains entry '{crtEntry}' - adding removal operation to transaction");
                bundleIncludes.Remove(crtEntry);
                F5BundleInclude bundleInclude = new F5BundleInclude { includeBundle = bundleIncludes.ToArray() };
                REST.Patch<F5BundleInclude>($"/mgmt/tm/sys/crypto/ca-bundle-manager/{bundle.Replace('/', '~')}", bundleInclude);
            }
            else
            {
                LogHandlerCommon.Trace(logger, CertificateStore, $"The current bundle does not contain entry '{crtEntry}'");
            }
            LogHandlerCommon.MethodExit(logger, CertificateStore, "RemoveBundleEntry");
        }

        // Bundles
        #endregion

        #region File Handling

        private void ArchiveFile(string sourcePath, string targetFilename, string transactionId = "")
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "ArchiveFile");

            // Make the 'keyfactor' directory if it doesn't exist
            string mkdirResult = REST.PostBASHCommand(new F5BashCommand
            {
                command = "run",
                utilCmdArgs = $"-c 'mkdir -p /var/config/rest/downloads/keyfactor'"
            }, transactionId);

            CopyFile(sourcePath, $"/var/config/rest/downloads/keyfactor/{targetFilename}", transactionId);
            LogHandlerCommon.MethodExit(logger, CertificateStore, "ArchiveFile");
        }

        private void CopyFile(string source, string target, string transactionId = "")
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "CopyFile");
            string copyResult = REST.PostBASHCommand(new F5BashCommand
            {
                command = "run",
                utilCmdArgs = $"-c 'cp {source} {target}'"
            }, transactionId);
            LogHandlerCommon.MethodExit(logger, CertificateStore, "CopyFile");
        }

        private void MoveFile(string source, string target, string transactionId = "")
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "MoveFile");
            string moveResult = REST.PostBASHCommand(new F5BashCommand
            {
                command = "run",
                utilCmdArgs = $"-c 'mv {source} {target}'"
            }, transactionId);
            LogHandlerCommon.MethodExit(logger, CertificateStore, "MoveFile");
        }

        private void RemoveFile(string source, string transactionId = "")
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "RemoveFile");
            string removeResult = REST.PostBASHCommand(new F5BashCommand
            {
                command = "run",
                utilCmdArgs = $"-c 'rm {source}'"
            }, transactionId);
            LogHandlerCommon.MethodExit(logger, CertificateStore, "RemoveFile");
        }
        // File Handling
        #endregion

        #region Transactions

        private F5Transaction BeginTransaction()
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "BeingTransaction");
            F5Transaction transaction = REST.Post<F5Transaction>("/mgmt/tm/transaction", "{}");
            LogHandlerCommon.Trace(logger, CertificateStore, $"Initiated transaction '{transaction.transid}'");
            LogHandlerCommon.MethodExit(logger, CertificateStore, "BeingTransaction");
            return transaction;
        }

        private void CommitTransaction(F5Transaction transaction)
        {
            LogHandlerCommon.MethodEntry(logger, CertificateStore, "CommitTransaction");
            LogHandlerCommon.Trace(logger, CertificateStore, $"Committing transaction '{transaction.transid};");
            REST.Patch<F5CommitTransaction>($"/mgmt/tm/transaction/{transaction.transid}", new F5CommitTransaction { state = "VALIDATING", validateOnly = false });
            LogHandlerCommon.MethodExit(logger, CertificateStore, "CommitTransaction");
        }

        // Transactions
        #endregion

        // Methods
        #endregion
    }
}