// SPDX-License-Identifier: AGPL-3.0-only
/**
* Digital Voice Modem - Fixed Network Equipment Core Library
* AGPLv3 Open Source. Use is subject to license terms.
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
*
* @package DVM / Fixed Network Equipment Core Library
* @license AGPLv3 License (https://opensource.org/licenses/AGPL-3.0)
*
*   Copyright (C) 2022-2025 Bryan Biedenkapp, N2PLL
*   Copyright (C) 2024 Caleb, KO4UYJ
*
*/

using System;
using System.IO;
using System.Net;
using System.Net.Sockets;
using System.Threading;
using System.Threading.Tasks;
using System.Text;
using System.Text.Json;
using System.Linq;
using System.Collections.Generic;

using fnecore.DMR;
using fnecore.P25;
using fnecore.NXDN;
using fnecore.Analog;
using fnecore.P25.KMM;

namespace fnecore
{
    /// <summary>
    /// Callback used to process a raw network frame.
    /// </summary>
    /// <param name="frame"><see cref="UdpFrame"/></param>
    /// <param name="peerId">Peer ID</param>
    /// <param name="streamId">Stream ID</param>
    /// <returns>True, if the frame was handled, otherwise false.</returns>
    public delegate bool RawNetworkFrame(UdpFrame frame, uint peerId, uint streamId);
    /// <summary>
    /// Callback used to process a raw network frame.
    /// </summary>
    /// <param name="frame"><see cref="UdpFrame"/></param>
    /// <param name="peerId">Peer ID</param>
    /// <param name="streamId">Stream ID</param>
    /// <param name="rtpHeader">RTP Header</param>
    /// <param name="fneHeader">RTP FNE Header</param>
    /// <param name="message">Raw packet message</param>
    /// <returns>True, if the frame was handled, otherwise false.</returns>
    public delegate bool UserPacketHandler(UdpFrame frame, uint peerId, uint streamId,
        RtpHeader rtpHeader, RtpFNEHeader fneHeader, byte[] message);

    /// <summary>
    /// Implements an FNE "peer".
    /// </summary>
    public class FnePeer : FneBase
    {
        private const int MAX_MISSED_PEER_PINGS = 5;

        private UdpReceiver client = null;

        private bool abortListening = false;

        private CancellationTokenSource listenCancelToken = new CancellationTokenSource();
        private Task listenTask = null;
        private CancellationTokenSource maintainenceCancelToken = new CancellationTokenSource();
        private Task maintainenceTask = null;

        private PeerInformation info;
        private IPEndPoint masterEndpoint = null;

        private ushort currPktSeq = 0;
        private uint streamId = 0;

        private List<TalkgroupEntry> announcedTGs = new List<TalkgroupEntry>();
        private List<PeerHAIPEntry> haIPs = new List<PeerHAIPEntry>();
        private int currentHAIP;

        private int retryCount;
        private int maxRetryCount = Constants.MAX_RETRY_BEFORE_RECONNECT;

        /*
        ** Properties
        */

        /// <summary>
        /// Gets/sets the peer information.
        /// </summary>
        public PeerInformation Information
        {
            get { return info; }
            set { info = value; }
        }

        /// <summary>
        /// Gets/sets the password used for connecting to a master.
        /// </summary>
        public string Passphrase
        {
            get;
            set;
        }

        /// <summary>
        /// Gets the list of talkgroups announced from the master FNE.
        /// </summary>
        public List<TalkgroupEntry> AnnouncedTGs => announcedTGs;

        /// <summary>
        /// Gets the number of pings sent.
        /// </summary>
        public int PingsSent
        {
            get;
            private set;
        }

        /// <summary>
        /// Gets the number of pings acked.
        /// </summary>
        public int PingsAcked
        {
            get;
            private set;
        }

        /// <summary>
        /// Gets/sets flag indicating whether or not the user packet handler is also handling
        /// protocol packets.
        /// </summary>
        public bool UserProtocolHandler
        {
            get;
            set;
        }

        /*
        ** Events/Callbacks
        */

        /// <summary>
        /// Event action that handles a raw network frame directly.
        /// </summary>
        public RawNetworkFrame NetworkFrameHandler = null;
        /// <summary>
        /// Event action that handles a RTP network packet.
        /// </summary>
        public UserPacketHandler UserPacketHandler = null;

        /*
        ** Methods
        */

        /// <summary>
        /// Initializes a new instance of the <see cref="FnePeer"/> class.
        /// </summary>
        /// <param name="systemName"></param>
        /// <param name="peerId"></param>
        /// <param name="address"></param>
        /// <param name="port"></param>
        public FnePeer(string systemName, uint peerId, string address, int port, string PresharedKey = null) : this(systemName, peerId, new IPEndPoint(IPAddress.Parse(address), port), PresharedKey)
        {
            /* stub */
        }

        /// <summary>
        /// Initializes a new instance of the <see cref="FnePeer"/> class.
        /// </summary>
        /// <param name="systemName"></param>
        /// <param name="peerId"></param>
        /// <param name="endpoint"></param>
        public FnePeer(string systemName, uint peerId, IPEndPoint endpoint, string PresharedKey = null) : base(systemName, peerId)
        {
            masterEndpoint = endpoint;
            client = new UdpReceiver();

            if (PresharedKey != null)
                client.SetPresharedKey(FneUtils.ConvertHexStringToPresharedKey(PresharedKey));

            info = new PeerInformation();
            info.PeerID = peerId;
            info.State = ConnectionState.WAITING_LOGIN;

            UserProtocolHandler = false;

            PingsAcked = 0;
        }

        /// <summary>
        /// Starts the main execution loop for this <see cref="FnePeer"/>.
        /// </summary>
        public override void Start()
        {
            if (isStarted)
                throw new InvalidOperationException("Cannot start listening when already started.");

            Logger(LogLevel.INFO, $"({systemName}) starting network services, {masterEndpoint}");

            // attempt initial connection
            try
            {
                client.Connect(masterEndpoint);
            }
            catch (SocketException se)
            {
                Log(LogLevel.FATAL, $"({systemName}) SOCKET ERROR: {se.SocketErrorCode}; {se.Message}");
            }

            abortListening = false;
            listenTask = Task.Factory.StartNew(Listen, listenCancelToken.Token);
            maintainenceTask = Task.Factory.StartNew(Maintainence, maintainenceCancelToken.Token);

            isStarted = true;
        }

        /// <summary>
        /// Stops the main execution loop for this <see cref="FnePeer"/>.
        /// </summary>
        public override void Stop()
        {
            if (!isStarted)
                throw new InvalidOperationException("Cannot stop listening when not started.");

            Logger(LogLevel.INFO, $"({systemName}) stopping network services, {masterEndpoint}");

            // send shutdown opcode to server
            SendMaster(CreateOpcode(Constants.NET_FUNC_RPT_CLOSING, Constants.NET_SUBFUNC_NOP), new byte[1], 1, CreateStreamID(), true);

            // stop UDP listen task
            if (listenTask != null)
            {
                abortListening = true;
                listenCancelToken.Cancel();

                try
                {
                    listenTask.GetAwaiter().GetResult();
                }
                catch (OperationCanceledException) { /* stub */ }
                finally
                {
                    listenCancelToken.Dispose();
                }
            }

            // stop maintainence task
            if (maintainenceTask != null)
            {
                maintainenceCancelToken.Cancel();

                try
                {
                    maintainenceTask.GetAwaiter().GetResult();
                }
                catch (OperationCanceledException) { /* stub */ }
                finally
                {
                    maintainenceCancelToken.Dispose();
                }
            }

            isStarted = false;
        }

        /// <summary>
        /// Helper to send a raw UDP frame.
        /// </summary>
        /// <param name="frame">UDP frame to send</param>
        public void Send(UdpFrame frame)
        {
            if (RawPacketTrace)
                Log(LogLevel.DEBUG, $"({systemName}) Network Sent (to {frame.Endpoint}) -- {FneUtils.HexDump(frame.Message, 0)}");

            client.Send(frame);
        }

        /// <summary>
        /// Helper to send a data message to the master.
        /// </summary>
        /// <param name="opcode">Opcode</param>
        /// <param name="message">Byte array containing message to send</param>
        /// <param name="pktSeq">RTP Packet Sequence</param>
        /// <param name="streamId"></param>
        /// <param name="forceZeroStream"></param>
        public void SendMaster(Tuple<byte, byte> opcode, byte[] message, ushort pktSeq, uint streamId = 0, bool forceZeroStream = false)
        {
            if (streamId == 0 && !forceZeroStream)
                streamId = this.streamId;

            Send(new UdpFrame()
            {
                Endpoint = masterEndpoint,
                Message = WriteFrame(message, peerId, this.peerId, opcode, pktSeq, streamId)
            });
        }

        /// <summary>
        /// Helper to send a data message to the master.
        /// </summary>
        /// <param name="opcode">Opcode</param>
        /// <param name="message">Byte array containing message to send</param>
        public void SendMaster(Tuple<byte, byte> opcode, byte[] message)
        {
            SendMaster(opcode, message, pktSeq());
        }

        /// <summary>
        /// Helper to send group affiliation announcements to the master.
        /// </summary>
        /// <param name="srcId"></param>
        /// <param name="dstId"></param>
        public void SendMasterGroupAffiliation(uint srcId, uint dstId)
        {
            // send message to master
            byte[] res = new byte[6];

            FneUtils.Write3Bytes(srcId, ref res, 0);
            FneUtils.Write3Bytes(dstId, ref res, 3);

            SendMaster(CreateOpcode(Constants.NET_FUNC_ANNOUNCE, Constants.NET_ANNC_SUBFUNC_GRP_AFFIL), res, 0, 0, true);
        }

        /// <summary>
        /// Helper to send group affiliation dictionary to the master
        /// </summary>
        /// <param name="affs"></param>
        public void SendMasterAffiliationUpdate(List<Tuple<uint, uint>> affs)
        {
            int bufferSize = 4 + (affs.Count * 8);
            byte[] buffer = new byte[bufferSize];

            FneUtils.WriteBytes((uint)affs.Count, ref buffer, 0);

            int offset = 4;
            foreach (var entry in affs)
            {
                FneUtils.Write3Bytes(entry.Item1, ref buffer, offset);
                FneUtils.Write3Bytes(entry.Item2, ref buffer, offset + 4);

                offset += 8;
            }

            SendMaster(CreateOpcode(Constants.NET_FUNC_ANNOUNCE, Constants.NET_ANNC_SUBFUNC_AFFILS), buffer, 0, 0, true);
        }

        /// <summary>
        /// Helper to send group affiliation removal announcement to the master.
        /// </summary>
        /// <param name="srcId"></param>
        public void SendMasterGroupAffiliationRemoval(uint srcId)
        {
            // send message to master
            byte[] res = new byte[3];

            FneUtils.Write3Bytes(srcId, ref res, 0);

            SendMaster(CreateOpcode(Constants.NET_FUNC_ANNOUNCE, Constants.NET_ANNC_SUBFUNC_GRP_UNAFFIL), res, 0, 0, true);
        }

        /// <summary>
        /// Helper to send unit registration announcements to the master.
        /// </summary>
        /// <param name="srcId"></param>
        public void SendMasterUnitRegistration(uint srcId)
        {
            // send message to master
            byte[] res = new byte[3];

            FneUtils.Write3Bytes(srcId, ref res, 0);

            SendMaster(CreateOpcode(Constants.NET_FUNC_ANNOUNCE, Constants.NET_ANNC_SUBFUNC_UNIT_REG), res, 0, 0, true);
        }

        /// <summary>
        /// Helper to send unit deregistration announcements to the master.
        /// </summary>
        /// <param name="srcId"></param>
        public void SendMasterUnitDeRegistration(uint srcId)
        {
            // send message to master
            byte[] res = new byte[3];

            FneUtils.Write3Bytes(srcId, ref res, 0);

            SendMaster(CreateOpcode(Constants.NET_FUNC_ANNOUNCE, Constants.NET_ANNC_SUBFUNC_UNIT_DEREG), res, 0, 0, true);
        }

        /// <summary>
        /// Helper to send a key request to the master
        /// </summary>
        /// <param name="algId"></param>
        /// <param name="kId"></param>
        public void SendMasterKeyRequest(byte algId, ushort kId)
        {
            byte[] res = new byte[32];

            KmmModifyKey modifyKey = new KmmModifyKey
            {
                AlgId = algId,
                KeyId = kId
            };

            KeysetItem ks = new KeysetItem
            {
                KeysetId = 0,
                AlgId = algId,
                KeyLength = 32
            };

            modifyKey.KeysetItem = ks;
            modifyKey.MessageLength = 21;

            byte[] payload = new byte[modifyKey.MessageLength];

            modifyKey.Encode(payload);

            Array.Copy(payload, 0, res, 11, payload.Length);

            SendMaster(CreateOpcode(Constants.NET_FUNC_KEY_REQ, Constants.NET_SUBFUNC_NOP), res, Constants.RtpCallEndSeq, 0, true);
        }

        /// <summary>
        /// Helper to update the RTP packet sequence.
        /// </summary>
        /// <param name="reset"></param>
        /// <returns>RTP packet sequence.</returns>
        public ushort pktSeq(bool reset = false)
        {
            if (reset)
            {
                currPktSeq = 0;
                return currPktSeq;
            }

            ushort curr = currPktSeq;
            ++currPktSeq;
            if (currPktSeq > (Constants.RtpCallEndSeq - 1))
                currPktSeq = 0;

            return curr;
        }

        /// <summary>
        /// Helper to rotate the master endpoint to the next HA endpoint.
        /// </summary>
        private void RotateMasterEndpont()
        {
            // are we rotating IPs for HA reconnect?
            if (haIPs.Count() > 0 && retryCount > 0U && maxRetryCount == Constants.MAX_RETRY_HA_RECONNECT)
            {

                PeerHAIPEntry entry = haIPs[currentHAIP];
                currentHAIP++;

                if (currentHAIP > haIPs.Count)
                {
                    currentHAIP = 0;
                }

                Log(LogLevel.ERROR, $"({systemName}) Not connected or lost connection to {masterEndpoint}; trying next HA {entry.EndPoint}...");
                masterEndpoint = entry.EndPoint;
            }

            ++retryCount;
        }

        /// <summary>
        /// Internal UDP listen routine.
        /// </summary>
        private async void Listen()
        {
            CancellationToken ct = listenCancelToken.Token;
            ct.ThrowIfCancellationRequested();

            while (!abortListening)
            {
                try
                {
                    UdpFrame frame = await client.Receive();
                    if (RawPacketTrace)
                        Log(LogLevel.DEBUG, $"Network Received (from {frame.Endpoint}) -- {FneUtils.HexDump(frame.Message, 0)}");

                    // decode RTP frame
                    if (frame.Message.Length <= 0)
                        continue;

                    RtpHeader rtpHeader;
                    RtpFNEHeader fneHeader;
                    int messageLength = 0;
                    byte[] message = ReadFrame(frame, out messageLength, out rtpHeader, out fneHeader);
                    if (message == null)
                    {
                        Log(LogLevel.ERROR, $"({systemName}) Malformed packet (from {frame.Endpoint}); failed to decode RTP frame");
                        continue;
                    }

                    if (message.Length < 1)
                    {
                        Log(LogLevel.WARNING, $"({systemName}) Malformed packet (from {frame.Endpoint}) -- {FneUtils.HexDump(message, 0)}");
                        continue;
                    }

                    // validate frame endpoint
                    if (frame.Endpoint.ToString() == masterEndpoint.ToString())
                    {
                        uint peerId = fneHeader.PeerID;

                        if (streamId != fneHeader.StreamID)
                            pktSeq(true);

                        // update current peer stream ID
                        streamId = fneHeader.StreamID;

                        // see if the peer is defining its own frame handler, if it is try to handle the frame there
                        if (NetworkFrameHandler != null)
                        {
                            if (NetworkFrameHandler(frame, peerId, streamId))
                                continue;
                        }

                        // process incoming message frame opcodes
                        switch (fneHeader.Function)
                        {
                            case Constants.NET_FUNC_PROTOCOL:
                                {
                                    // are we handling protocol packets with the user packet handler?
                                    if (UserPacketHandler != null && UserProtocolHandler)
                                    {
                                        UserPacketHandler(frame, peerId, streamId, rtpHeader, fneHeader, message);
                                        break;
                                    }

                                    if (fneHeader.SubFunction == Constants.NET_PROTOCOL_SUBFUNC_DMR)            // Encapsulated DMR data frame
                                    {
                                        if (peerId != this.peerId)
                                        {
                                            //Log(LogLevel.WARNING, $"({systemName}) PEER {peerId}; routed traffic, rewriting PEER {this.peerId}");
                                            peerId = this.peerId;
                                        }

                                        // is this for our peer?
                                        if (peerId == this.peerId)
                                        {
                                            // TODO: port the mux validation logic from dvmhost:src/common/network/Network.cpp

                                            byte seqNo = message[4];
                                            uint srcId = FneUtils.Bytes3ToUInt32(message, 5);
                                            uint dstId = FneUtils.Bytes3ToUInt32(message, 8);

                                            byte bits = message[15];
                                            byte slot = (byte)(((bits & 0x80) == 0x80) ? 1 : 0);
                                            CallType callType = ((bits & 0x40) == 0x40) ? CallType.PRIVATE : CallType.GROUP;
                                            FrameType frameType = (FrameType)((bits & 0x30) >> 4);

                                            DMRDataType dataType = DMRDataType.IDLE;
                                            if ((bits & 0x20) == 0x20)
                                                dataType = (DMRDataType)(bits & ~(0x20));

                                            byte n = (byte)(bits & 0xF);
#if DEBUG
                                            Log(LogLevel.DEBUG, $"{systemName} DMRD: SRC_PEER {peerId} SRC_ID {srcId} DST_ID {dstId} TS {slot} [STREAM ID {streamId}]");
#endif
                                            // perform any userland actions with the data
                                            FireDMRDataReceived(new DMRDataReceivedEvent(peerId, srcId, dstId, slot, callType, frameType, dataType, n, rtpHeader.Sequence, streamId, message));
                                        }
                                    }
                                    else if (fneHeader.SubFunction == Constants.NET_PROTOCOL_SUBFUNC_P25)       // Encapsulated P25 data frame
                                    {
                                        if (peerId != this.peerId)
                                        {
                                            //Log(LogLevel.WARNING, $"({systemName}) PEER {peerId}; routed traffic, rewriting PEER {this.peerId}");
                                            peerId = this.peerId;
                                        }

                                        // is this for our peer?
                                        if (peerId == this.peerId)
                                        {
                                            // TODO: port the mux validation logic from dvmhost:src/common/network/Network.cpp

                                            uint srcId = FneUtils.Bytes3ToUInt32(message, 5);
                                            uint dstId = FneUtils.Bytes3ToUInt32(message, 8);
                                            CallType callType = (message[4] == P25Defines.LC_PRIVATE) ? CallType.PRIVATE : CallType.GROUP;
                                            P25DUID duid = (P25DUID)message[22];
                                            FrameType frameType = ((duid != P25DUID.TDU) && (duid != P25DUID.TDULC)) ? FrameType.VOICE : FrameType.TERMINATOR;
#if DEBUG
                                            Log(LogLevel.DEBUG, $"{systemName} P25D: SRC_PEER {peerId} SRC_ID {srcId} DST_ID {dstId} [STREAM ID {streamId}]");
#endif
                                            // perform any userland actions with the data
                                            FireP25DataReceived(new P25DataReceivedEvent(peerId, srcId, dstId, callType, duid, frameType, rtpHeader.Sequence, streamId, message));
                                        }
                                    }
                                    else if (fneHeader.SubFunction == Constants.NET_PROTOCOL_SUBFUNC_NXDN)      // Encapsulated NXDN data frame
                                    {
                                        if (peerId != this.peerId)
                                        {
                                            //Log(LogLevel.WARNING, $"({systemName}) PEER {peerId}; routed traffic, rewriting PEER {this.peerId}");
                                            peerId = this.peerId;
                                        }

                                        // is this for our peer?
                                        if (peerId == this.peerId)
                                        {
                                            // TODO: port the mux validation logic from dvmhost:src/common/network/Network.cpp

                                            NXDNMessageType messageType = (NXDNMessageType)message[4];
                                            uint srcId = FneUtils.Bytes3ToUInt32(message, 5);
                                            uint dstId = FneUtils.Bytes3ToUInt32(message, 8);

                                            byte bits = message[15];
                                            CallType callType = ((bits & 0x40) == 0x40) ? CallType.PRIVATE : CallType.GROUP;
                                            FrameType frameType = (messageType != NXDNMessageType.MESSAGE_TYPE_TX_REL) ? FrameType.VOICE : FrameType.TERMINATOR;
#if DEBUG
                                            Log(LogLevel.DEBUG, $"{systemName} NXDD: SRC_PEER {peerId} SRC_ID {srcId} DST_ID {dstId} [STREAM ID {streamId}]");
#endif
                                            // perform any userland actions with the data
                                            FireNXDNDataReceived(new NXDNDataReceivedEvent(peerId, srcId, dstId, callType, messageType, frameType, rtpHeader.Sequence, streamId, message));
                                        }
                                    }
                                    else if (fneHeader.SubFunction == Constants.NET_PROTOCOL_SUBFUNC_ANALOG)    // Encapsulated Analog data frame
                                    {
                                        if (peerId != this.peerId)
                                        {
                                            //Log(LogLevel.WARNING, $"({systemName}) PEER {peerId}; routed traffic, rewriting PEER {this.peerId}");
                                            peerId = this.peerId;
                                        }

                                        // is this for our peer?
                                        if (peerId == this.peerId)
                                        {
                                            // TODO: port the mux validation logic from dvmhost:src/common/network/Network.cpp

                                            uint srcId = FneUtils.Bytes3ToUInt32(message, 5);
                                            uint dstId = FneUtils.Bytes3ToUInt32(message, 8);
                                            CallType callType = CallType.GROUP; /* analog calls cannot be private calls right now ... */
                                            AudioFrameType audioFrameType = (AudioFrameType)(message[15] & 0x0F);
                                            FrameType frameType = (audioFrameType != AudioFrameType.TERMINATOR) ? FrameType.VOICE : FrameType.TERMINATOR;
#if DEBUG
                                            Log(LogLevel.DEBUG, $"{systemName} P25D: SRC_PEER {peerId} SRC_ID {srcId} DST_ID {dstId} [STREAM ID {streamId}]");
#endif
                                            // perform any userland actions with the data
                                            FireAnalogDataReceived(new AnalogDataReceivedEvent(peerId, srcId, dstId, callType, audioFrameType, frameType, rtpHeader.Sequence, streamId, message));
                                        }
                                    }
                                    else
                                    {
                                        Log(LogLevel.ERROR, $"({systemName}) Unknown protocol opcode {FneUtils.BytesToString(message, 0, 4)} -- {FneUtils.HexDump(message, 0)}");
                                    }
                                }
                                break;

                            case Constants.NET_FUNC_MASTER:                                                 // Master
                                {
                                    if (this.peerId == peerId)
                                    {
                                        // process incoming message subfunction opcodes
                                        switch (fneHeader.SubFunction)
                                        {
                                            case Constants.NET_MASTER_SUBFUNC_ACTIVE_TGS:                   // Talkgroup Active IDs
                                                {
                                                    uint len = FneUtils.ToUInt32(message, 6);
                                                    int offs = 11;
                                                    for (int i = 0; i < len; i++)
                                                    {
                                                        uint id = FneUtils.Bytes3ToUInt32(message, offs);
                                                        byte slot = (byte)(message[offs + 3] & 0x03);
                                                        bool affiliated = (message[offs + 3] & 0x40) == 0x40;
                                                        bool nonPreferred = (message[offs + 3] & 0x80) == 0x80;

                                                        TalkgroupEntry entry = new TalkgroupEntry()
                                                        {
                                                            ID = id,
                                                            Slot = slot,
                                                            Affiliated = affiliated,
                                                            NonPreferred = nonPreferred,
                                                            Invalid = false
                                                        };

                                                        int idx = announcedTGs.FindIndex(x => x.ID == id && x.Slot == slot);
                                                        if (idx != -1)
                                                            announcedTGs[idx] = entry;
                                                        else
                                                            announcedTGs.Add(entry);

                                                        offs += 5;
                                                    }

                                                    Log(LogLevel.INFO, $"Activated {len} TGs; loaded {announcedTGs.Count} entries into talkgroup table");
                                                }
                                                break;
                                            case Constants.NET_MASTER_SUBFUNC_DEACTIVE_TGS:                 // Talkgroup Deactivated IDs
                                                {
                                                    uint len = FneUtils.ToUInt32(message, 6);
                                                    int offs = 11;
                                                    for (int i = 0; i < len; i++)
                                                    {
                                                        uint id = FneUtils.Bytes3ToUInt32(message, offs);
                                                        byte slot = message[offs + 3];

                                                        int idx = announcedTGs.FindIndex(x => x.ID == id && x.Slot == slot);
                                                        if (idx != -1)
                                                            announcedTGs[idx].Invalid = true;
                                                        else
                                                        {
                                                            TalkgroupEntry entry = new TalkgroupEntry()
                                                            {
                                                                ID = id,
                                                                Slot = slot,
                                                                Affiliated = false,
                                                                NonPreferred = false,
                                                                Invalid = true
                                                            };
                                                            announcedTGs.Add(entry);
                                                        }

                                                        offs += 5;
                                                    }

                                                    Log(LogLevel.INFO, $"Deactivated {len} TGs; loaded {announcedTGs.Count} entries into talkgroup table");
                                                }
                                                break;

                                            case Constants.NET_MASTER_SUBFUNC_HA_PARAMS:                    // HA Parameters
                                                {
                                                    haIPs.Clear();
                                                    currentHAIP = 0;
                                                    maxRetryCount = Constants.MAX_RETRY_HA_RECONNECT;

                                                    // always add the configured address to the HA IP list
                                                    haIPs.Add(new PeerHAIPEntry(masterEndpoint.Address.ToString(), masterEndpoint.Port));

                                                    uint len = FneUtils.ToUInt32(message, 6);
                                                    if (len > 0)
                                                        len /= Constants.HAParamsEntryLen;

                                                    int offs = 10;
                                                    for (int i = 0; i < len; i++, offs += (int)Constants.HAParamsEntryLen)
                                                    {
                                                        uint ipAddr = FneUtils.ToUInt32(message, offs + 4);
                                                        ushort port = FneUtils.ToUInt16(message, offs + 8);

                                                        IPAddress address = new IPAddress(ipAddr);
                                                        haIPs.Add(new PeerHAIPEntry(address.ToString(), port));
                                                    }

                                                    if (haIPs.Count > 1)
                                                    {
                                                        currentHAIP = 1;
                                                        Log(LogLevel.INFO, $"Loaded {haIPs.Count} HA IPs from master");
                                                    }
                                                }
                                                break;

                                            default:
                                                Log(LogLevel.ERROR, $"({systemName}) Unknown protocol opcode {FneUtils.BytesToString(message, 0, 4)} -- {FneUtils.HexDump(message, 0)}");
                                                break;
                                        }
                                    }
                                }
                                break;

                            case Constants.NET_FUNC_INCALL_CTRL:                                            // In-Call Control
                                {
                                    if (this.peerId == peerId)
                                    {
                                        // process incoming message subfunction opcodes
                                        switch (fneHeader.SubFunction)
                                        {
                                            case Constants.NET_PROTOCOL_SUBFUNC_DMR:                        // DMR In-Call Control
                                                {
                                                    byte command = message[10];
                                                    uint dstId = FneUtils.Bytes3ToUInt32(message, 11);
                                                    byte slot = message[14];

                                                    // fire off DMR in-call callback if we have one
                                                    FireDMRInCallControl(new DMRInCallControlEvent(peerId, dstId, slot, command));
                                                }
                                                break;
                                            case Constants.NET_PROTOCOL_SUBFUNC_P25:                        // P25 In-Call Control
                                                {
                                                    byte command = message[10];
                                                    uint dstId = FneUtils.Bytes3ToUInt32(message, 11);

                                                    // fire off P25 in-call callback if we have one
                                                    FireP25InCallControl(new P25InCallControlEvent(peerId, dstId, command));
                                                }
                                                break;
                                            case Constants.NET_PROTOCOL_SUBFUNC_NXDN:                       // NXDN In-Call Control
                                                {
                                                    byte command = message[10];
                                                    uint dstId = FneUtils.Bytes3ToUInt32(message, 11);

                                                    // fire off NXDN in-call callback if we have one
                                                    FireNXDNInCallControl(new NXDNInCallControlEvent(peerId, dstId, command));
                                                }
                                                break;
                                            case Constants.NET_PROTOCOL_SUBFUNC_ANALOG:                     // Analog In-Call Control
                                                {
                                                    byte command = message[10];
                                                    uint dstId = FneUtils.Bytes3ToUInt32(message, 11);

                                                    // fire off analog in-call callback if we have one
                                                    FireAnalogInCallControl(new AnalogInCallControlEvent(peerId, dstId, command));
                                                }
                                                break;

                                            default:
                                                Log(LogLevel.ERROR, $"({systemName}) Unknown incall control opcode {FneUtils.BytesToString(message, 0, 4)} -- {FneUtils.HexDump(message, 0)}");
                                                break;
                                        }
                                    }
                                }
                                break;

                            case Constants.NET_FUNC_NAK:                                                    // Master NAK
                                {
                                    if (this.peerId == peerId)
                                    {
                                        // DVM 3.6 adds support to respond with a NAK reason, as such we just check if the NAK response is greater
                                        // then 10 bytes and process the reason value
                                        ConnectionMSTNAK reason = ConnectionMSTNAK.INVALID;
                                        if (message.Length > 10)
                                        {
                                            reason = (ConnectionMSTNAK)FneUtils.ToUInt16(message, 10);
                                            switch (reason)
                                            {
                                                case ConnectionMSTNAK.MODE_NOT_ENABLED:
                                                    Log(LogLevel.WARNING, $"({systemName}) PEER {this.peerId} master NAK; digital mode not enabled on FNE");
                                                    break;
                                                case ConnectionMSTNAK.ILLEGAL_PACKET:
                                                    Log(LogLevel.WARNING, $"({systemName}) PEER {this.peerId} master NAK; illegal/unknown packet");
                                                    break;
                                                case ConnectionMSTNAK.FNE_UNAUTHORIZED:
                                                    Log(LogLevel.WARNING, $"({systemName}) PEER {this.peerId} master NAK; unauthorized");
                                                    break;
                                                case ConnectionMSTNAK.BAD_CONN_STATE:
                                                    Log(LogLevel.WARNING, $"({systemName}) PEER {this.peerId} master NAK; bad connection state");
                                                    break;
                                                case ConnectionMSTNAK.INVALID_CONFIG_DATA:
                                                    Log(LogLevel.WARNING, $"({systemName}) PEER {this.peerId} master NAK; invalid configuration data");
                                                    break;
                                                case ConnectionMSTNAK.FNE_MAX_CONN:
                                                    Log(LogLevel.WARNING, $"({systemName}) PEER {this.peerId} master NAK; FNE has reached maximum permitted connections");
                                                    break;
                                                case ConnectionMSTNAK.PEER_RESET:
                                                    Log(LogLevel.WARNING, $"({systemName}) PEER {this.peerId} master NAK; FNE demanded connection reset");
                                                    break;
                                                case ConnectionMSTNAK.PEER_ACL:
                                                    Log(LogLevel.ERROR, $"({systemName}) PEER {this.peerId} master NAK; ACL rejection, network disabled");
                                                    info.State = ConnectionState.WAITING_LOGIN;
                                                    Stop();
                                                    break;

                                                case ConnectionMSTNAK.GENERAL_FAILURE:
                                                default:
                                                    Log(LogLevel.WARNING, $"({systemName}) PEER {this.peerId} master NAK; general failure");
                                                    break;
                                            }
                                        }

                                        if (info.State == ConnectionState.RUNNING && (reason == ConnectionMSTNAK.FNE_MAX_CONN))
                                        {
                                            Log(LogLevel.WARNING, $"({systemName}) PEER {this.peerId} master NAK; attemping to relogin");

                                            // reset states
                                            PingsSent = 0;
                                            PingsAcked = 0;
                                            info.State = ConnectionState.WAITING_LOGIN;
                                        }
                                        else
                                        {
                                            Log(LogLevel.ERROR, $"({systemName}) PEER {this.peerId} master NAK; network reconnect");

                                            // reset states
                                            PingsSent = 0;
                                            PingsAcked = 0;
                                            info.State = ConnectionState.WAITING_LOGIN;
                                            break;
                                        }
                                    }
                                }
                                break;
                            case Constants.NET_FUNC_ACK:                                                    // Repeater ACK
                                {
                                    if (info.State == ConnectionState.WAITING_LOGIN)                        // Repeater Login
                                    {
                                        uint salt = FneUtils.ToUInt32(message, 6);
                                        Log(LogLevel.INFO, $"({systemName}) PEER {this.peerId} login ACK received with ID {salt}");

                                        info.Salt = salt;

                                        // calculate our own hash
                                        byte[] inBuf = new byte[4 + Passphrase.Length];
                                        FneUtils.WriteBytes(info.Salt, ref inBuf, 0);
                                        FneUtils.StringToBytes(Passphrase, inBuf, 4, Passphrase.Length);
                                        byte[] calcHash = FneUtils.sha256_hash(inBuf);

                                        // send message to master
                                        byte[] res = new byte[calcHash.Length + 8];
                                        FneUtils.StringToBytes(Constants.TAG_REPEATER_AUTH, res, 0, 4);
                                        FneUtils.WriteBytes(peerId, ref res, 4);
                                        Buffer.BlockCopy(calcHash, 0, res, 8, calcHash.Length);
                                        SendMaster(CreateOpcode(Constants.NET_FUNC_RPTK), res);

                                        info.State = ConnectionState.WAITING_AUTHORISATION;
                                    }
                                    else if (info.State == ConnectionState.WAITING_AUTHORISATION)           // Repeater Authorization
                                    {
                                        if (this.peerId == peerId)
                                        {
                                            string json = string.Empty;
                                            using (MemoryStream stream = new MemoryStream())
                                            {
                                                using (Utf8JsonWriter jsonWriter = new Utf8JsonWriter(stream))
                                                {
                                                    jsonWriter.WriteStartObject();

                                                    // identity
                                                    jsonWriter.WriteString("identity", info.Details.Identity);
                                                    jsonWriter.WriteNumber("rxFrequency", info.Details.RxFrequency);
                                                    jsonWriter.WriteNumber("txFrequency", info.Details.TxFrequency);

                                                    // peer types
                                                    jsonWriter.WriteBoolean("externalPeer", info.Details.ExternalPeer);
                                                    jsonWriter.WriteBoolean("conventionalPeer", info.Details.ConventionalPeer);
                                                    jsonWriter.WriteBoolean("sysView", info.Details.SysView);

                                                    // system info
                                                    {
                                                        jsonWriter.WritePropertyName("info");
                                                        jsonWriter.WriteStartObject();

                                                        jsonWriter.WriteNumber("latitude", info.Details.Latitude);
                                                        jsonWriter.WriteNumber("longitude", info.Details.Longitude);
                                                        jsonWriter.WriteNumber("height", info.Details.Height);
                                                        jsonWriter.WriteString("location", info.Details.Location);

                                                        jsonWriter.WriteEndObject();
                                                    }

                                                    // channel data
                                                    {
                                                        jsonWriter.WritePropertyName("channel");
                                                        jsonWriter.WriteStartObject();

                                                        jsonWriter.WriteNumber("txPower", info.Details.TxPower);
                                                        jsonWriter.WriteNumber("txOffsetMhz", (double)info.Details.TxOffsetMhz);
                                                        jsonWriter.WriteNumber("chBandwidthKhz", (double)info.Details.ChBandwidthKhz);
                                                        jsonWriter.WriteNumber("channelId", info.Details.ChannelID);
                                                        jsonWriter.WriteNumber("channelNo", info.Details.ChannelNo);

                                                        jsonWriter.WriteEndObject();
                                                    }

                                                    // RCON
                                                    {
                                                        jsonWriter.WritePropertyName("rcon");
                                                        jsonWriter.WriteStartObject();

                                                        jsonWriter.WriteString("password", info.Details.Password);
                                                        jsonWriter.WriteNumber("port", info.Details.Port);

                                                        jsonWriter.WriteEndObject();
                                                    }

                                                    jsonWriter.WriteString("software", info.Details.Software);

                                                    jsonWriter.WriteEndObject();
                                                }

                                                json = Encoding.UTF8.GetString(stream.ToArray());
                                            }

                                            // send message to master
                                            byte[] res = new byte[json.Length + 8];
                                            FneUtils.StringToBytes(Constants.TAG_REPEATER_CONFIG, res, 0, 4);
                                            FneUtils.WriteBytes(peerId, ref res, 4);
                                            FneUtils.StringToBytes(json, res, 8, json.Length);
                                            SendMaster(CreateOpcode(Constants.NET_FUNC_RPTC), res);

                                            info.State = ConnectionState.WAITING_CONFIG;
                                        }
                                        else
                                        {
                                            info.State = ConnectionState.WAITING_LOGIN;
                                            Log(LogLevel.ERROR, $"({systemName}) PEER {this.peerId} master ACK contained wrong ID - connection reset");
                                        }
                                    }
                                    else if (info.State == ConnectionState.WAITING_CONFIG)                  // Repeater Configuration
                                    {
                                        if (this.peerId == peerId)
                                        {
                                            info.State = ConnectionState.RUNNING;
                                            Log(LogLevel.INFO, $"({systemName}) PEER {this.peerId} connection to MASTER completed");

                                            this.streamId = 0;
                                            retryCount = 0; // reset retry count

                                            // userland actions
                                            FirePeerConnected(new PeerConnectedEvent(peerId, info));
                                        }
                                        else
                                        {
                                            info.State = ConnectionState.WAITING_LOGIN;
                                            Log(LogLevel.ERROR, $"({systemName}) PEER {this.peerId} master ACK contained wrong ID - connection reset");
                                        }
                                    }
                                }
                                break;
                            case Constants.NET_FUNC_MST_CLOSING:                                            // Master Closing (Disconnect)
                                {
                                    if (this.peerId == peerId)
                                    {
                                        info.State = ConnectionState.WAITING_LOGIN;
                                        Log(LogLevel.DEBUG, $"({systemName}) PEER {this.peerId} MSTCL received");

                                        // userland actions
                                        if (PeerDisconnected != null)
                                            PeerDisconnected(peerId);
                                    }
                                }
                                break;
                            case Constants.NET_FUNC_PONG:                                                   // Master Ping Response
                                {
                                    if (this.peerId == peerId)
                                    {
                                        PingsAcked++;
                                        Log(LogLevel.DEBUG, $"({systemName}) PEER {this.peerId} MSTPONG received, pongs since connected {PingsAcked}");
                                    }
                                }
                                break;
                            case Constants.NET_FUNC_KEY_RSP:                                               // Key Response
                                if (this.peerId == peerId)
                                {
                                    byte[] payload = message.Skip(11).ToArray();

                                    byte messageId = payload[0];

                                    if (messageId == (byte)KmmMessageType.MODIFY_KEY_CMD)
                                    {
                                        KmmModifyKey modifyKey = new KmmModifyKey();
                                        modifyKey.Decode(payload);

                                        FireKeyResponse(new KeyResponseEvent(messageId, modifyKey, message));
                                    }
                                    else
                                    {
                                        Log(LogLevel.WARNING, $"Unknown KEY_RSP Message ID: {messageId}");
                                    }
                                }
                                break;

                            default:
                                // are we handling anything else with a user handler?
                                if (UserPacketHandler != null)
                                {
                                    UserPacketHandler(frame, peerId, streamId, rtpHeader, fneHeader, message);
                                    break;
                                }

                                Log(LogLevel.ERROR, $"({systemName}) Unknown opcode {FneUtils.BytesToString(message, 0, 4)} -- {FneUtils.HexDump(message, 0)}");
                                break;
                        }
                    }
                }
                catch (InvalidOperationException)
                {
                    Log(LogLevel.ERROR, $"({systemName}) Not connected or lost connection to {masterEndpoint}; reconnecting...");
                    RotateMasterEndpont();

                    // reset states
                    PingsSent = 0;
                    PingsAcked = 0;
                    info.State = ConnectionState.WAITING_LOGIN;

                    client.Connect(masterEndpoint);
                }
                catch (SocketException se)
                {
                    // what kind of socket error do we have?
                    switch (se.SocketErrorCode)
                    {
                        case SocketError.NotConnected:
                        case SocketError.ConnectionReset:
                        case SocketError.ConnectionAborted:
                        case SocketError.ConnectionRefused:
                            Log(LogLevel.ERROR, $"({systemName}) Not connected or lost connection to {masterEndpoint}; reconnecting...");
                            RotateMasterEndpont();

                            // reset states
                            PingsSent = 0;
                            PingsAcked = 0;
                            info.State = ConnectionState.WAITING_LOGIN;

                            client.Connect(masterEndpoint);
                            break;
                        default:
                            Log(LogLevel.FATAL, $"({systemName}) SOCKET ERROR: {se.SocketErrorCode}; {se.Message}");
                            break;
                    }
                }

                if (ct.IsCancellationRequested)
                    abortListening = true;
            }
        }

        /// <summary>
        /// Internal maintainence routine.
        /// </summary>
        private async void Maintainence()
        {
            CancellationToken ct = maintainenceCancelToken.Token;
            while (!abortListening)
            {
                try
                {
                    // if we're not connected, zero out the connection stats and send a login request to the master
                    if (info.State == ConnectionState.WAITING_LOGIN)
                    {
                        // reset states
                        PingsSent = 0;
                        PingsAcked = 0;
                        info.State = ConnectionState.WAITING_LOGIN;

                        this.streamId = CreateStreamID();

                        // send message to master
                        byte[] res = new byte[8];
                        FneUtils.StringToBytes(Constants.TAG_REPEATER_LOGIN, res, 0, 4);
                        FneUtils.WriteBytes(peerId, ref res, 4);
                        SendMaster(CreateOpcode(Constants.NET_FUNC_RPTL), res);

                        Log(LogLevel.INFO, $"({systemName}) Sending login request to MASTER {masterEndpoint}");
                    }

                    // if we are connected, sent a ping to the master and increment the counter
                    if (info.State == ConnectionState.RUNNING)
                    {
                        if (PingsSent > (PingsAcked + MAX_MISSED_PEER_PINGS))
                        {
                            Log(LogLevel.WARNING, $"({systemName} Peer connection lost to {masterEndpoint}; reconnecting...");
                            RotateMasterEndpont();

                            // reset states
                            PingsSent = 0;
                            PingsAcked = 0;
                            info.State = ConnectionState.WAITING_LOGIN;

                            client.Connect(masterEndpoint);
                        }
                        else
                        {
                            // send message to master
                            byte[] res = new byte[1];
                            SendMaster(CreateOpcode(Constants.NET_FUNC_PING), res, Constants.RtpCallEndSeq);

                            PingsSent++;
                            Log(LogLevel.DEBUG, $"({systemName}) RPTPING sent to MASTER {masterEndpoint}; pings since connected {PingsSent}");
                        }
                    }
                }
                catch (InvalidOperationException)
                {
                    Log(LogLevel.ERROR, $"({systemName}) Not connected or lost connection to {masterEndpoint}; reconnecting...");
                    RotateMasterEndpont();

                    // reset states
                    PingsSent = 0;
                    PingsAcked = 0;
                    info.State = ConnectionState.WAITING_LOGIN;

                    client.Connect(masterEndpoint);
                }
                catch (SocketException se)
                {
                    // what kind of socket error do we have?
                    switch (se.SocketErrorCode)
                    {
                        case SocketError.NotConnected:
                        case SocketError.ConnectionReset:
                        case SocketError.ConnectionAborted:
                        case SocketError.ConnectionRefused:
                            Log(LogLevel.ERROR, $"({systemName}) Not connected or lost connection to {masterEndpoint}; reconnecting...");
                            RotateMasterEndpont();

                            // reset states
                            PingsSent = 0;
                            PingsAcked = 0;
                            info.State = ConnectionState.WAITING_LOGIN;

                            client.Connect(masterEndpoint);
                            break;
                        default:
                            Log(LogLevel.FATAL, $"({systemName}) SOCKET ERROR: {se.SocketErrorCode}; {se.Message}");
                            abortListening = true;
                            break;
                    }
                }

                try
                {
                    await Task.Delay(PingTime * 1000, ct);
                }
                catch (TaskCanceledException) { /* stub */ }
            }
        }
    } // public class FnePeer
} // namespace fnecore