q3rcon-ps/lib/base.ps1

class Base {
    static [int]$DEFAULT_RECEIVE_TIMEOUT = 100
    static [int]$DEFAULT_BUFFER_SIZE = 4096
    static [int]$DEFAULT_SEND_TIMEOUT = 5000

    [string]$hostname
    [int]$port
    [string]$passwd
    [Object]$request
    [Object]$response
    hidden [System.Net.Sockets.Socket] $_socket
    hidden [bool]$_disposed = $false
    hidden [byte[]]$_receiveBuffer

    Base ([string]$hostname, [int]$port, [string]$passwd) {
        if ([string]::IsNullOrWhiteSpace($hostname)) {
            throw [System.ArgumentException]::new('Hostname cannot be null or empty', 'hostname')
        }
        if ($port -le 0 -or $port -gt 65535) {
            throw [System.ArgumentOutOfRangeException]::new('port', 'Port must be between 1-65535')
        }
        if ([string]::IsNullOrWhiteSpace($passwd)) {
            throw [System.ArgumentException]::new('Password cannot be null or empty', 'passwd')
        }

        $this.hostname = $hostname
        $this.port = $port
        $this.passwd = $passwd

        $this.request = New-RequestPacket($this.passwd)
        $this.response = New-ResponsePacket
        $this._receiveBuffer = [byte[]]::new([Base]::DEFAULT_BUFFER_SIZE)

        $this._InitializeConnection()
    }

    hidden [void] _InitializeConnection() {
        try {
            $hostEntry = [System.Net.Dns]::GetHostEntry($this.hostname)
            if ($hostEntry.AddressList.Length -eq 0) {
                throw [System.Net.Sockets.SocketException]::new([int][System.Net.Sockets.SocketError]::HostNotFound)
            }
            
            $ipv4Address = $hostEntry.AddressList | Where-Object { $_.AddressFamily -eq [System.Net.Sockets.AddressFamily]::InterNetwork } | Select-Object -First 1
            if (-not $ipv4Address) {
                throw [System.InvalidOperationException]::new("No IPv4 address found for hostname: $($this.hostname)")
            }

            $endpoint = [System.Net.IPEndPoint]::new($ipv4Address, $this.port)
            
            $this._socket = [System.Net.Sockets.Socket]::new(
                [System.Net.Sockets.AddressFamily]::InterNetwork,
                [System.Net.Sockets.SocketType]::Dgram,
                [System.Net.Sockets.ProtocolType]::UDP
            )
            
            $this._socket.Connect($endpoint)
            $this._socket.ReceiveTimeout = [Base]::DEFAULT_RECEIVE_TIMEOUT
            $this._socket.SendTimeout = [Base]::DEFAULT_SEND_TIMEOUT
        }
        catch [System.Net.Sockets.SocketException] {
            $this._Cleanup()
            throw [System.InvalidOperationException]::new(
                "Failed to create UDP connection to $($this.hostname):$($this.port). Error: $($_.Exception.Message)", 
                $_.Exception
            )
        }
        catch {
            $this._Cleanup()
            throw [System.InvalidOperationException]::new(
                "Failed to initialize connection to $($this.hostname):$($this.port). Error: $($_.Exception.Message)", 
                $_.Exception
            )
        }
    }

    hidden [void] _ThrowIfDisposed() {
        if ($this._disposed) {
            throw [System.ObjectDisposedException]::new($this.GetType().Name)
        }
    }

    hidden [bool] _IsConnected() {
        return $this._socket -and -not $this._disposed -and $this._socket.Connected
    }

    [string] ToString () {
        $status = if ($this._IsConnected()) { 'Connected' } else { 'Disconnected' }
        return 'Rcon connection {0}:{1} ({2})' -f $this.hostname, $this.port, $status
    }

    [string] _send([string]$msg) {
        return $this._send($msg, [Base]::DEFAULT_RECEIVE_TIMEOUT)
    }

    [string] _send([string]$msg, [int]$timeout) {
        if ([string]::IsNullOrEmpty($msg)) {
            throw [System.ArgumentException]::new('Message cannot be null or empty', 'msg')
        }
        if ($timeout -le 0) {
            throw [System.ArgumentOutOfRangeException]::new('timeout', 'Timeout must be positive')
        }

        $this._ThrowIfDisposed()
        
        if (-not $this._IsConnected()) {
            throw [System.InvalidOperationException]::new('Socket is not connected')
        }

        try {
            $payload = $this.request.Payload($msg)
            $bytesSent = $this._socket.Send($payload)
            if ($bytesSent -ne $payload.Length) {
                Write-Warning "Not all bytes were sent. Expected: $($payload.Length), Sent: $bytesSent"
            }

            $responseData = [System.Text.StringBuilder]::new()
            $sw = [System.Diagnostics.Stopwatch]::StartNew()
            $headerLength = $this.response.Header().Length
            
            do {
                try {
                    $bytesReceived = $this._socket.Receive($this._receiveBuffer)
                    if ($bytesReceived -gt 0) {
                        $dataStartIndex = [Math]::Min($headerLength - 1, $bytesReceived)
                        $responseText = [System.Text.Encoding]::ASCII.GetString($this._receiveBuffer, $dataStartIndex, $bytesReceived - $dataStartIndex)
                        $responseData.Append($responseText) | Out-Null
                    }
                }
                catch [System.Net.Sockets.SocketException] {
                    if ($_.Exception.SocketErrorCode -eq 'TimedOut') {
                        Write-Debug 'Socket receive timeout - continuing to wait for more data'
                        continue
                    }
                    else {
                        throw [System.InvalidOperationException]::new(
                            "Socket error during receive: $($_.Exception.Message)", 
                            $_.Exception
                        )
                    }
                }
            } while ($sw.ElapsedMilliseconds -lt $timeout)
            
            $sw.Stop()
            return $responseData.ToString()
        }
        catch [System.Net.Sockets.SocketException] {
            throw [System.InvalidOperationException]::new(
                "Network error during send/receive: $($_.Exception.Message)", 
                $_.Exception
            )
        }
    }

    hidden [void] _Cleanup() {
        if ($this._socket) {
            try {
                if ($this._socket.Connected) {
                    $this._socket.Shutdown([System.Net.Sockets.SocketShutdown]::Both)
                }
            }
            catch {
                Write-Debug "Error during socket shutdown: $($_.Exception.Message)"
            }
            finally {
                $this._socket.Close()
                $this._socket = $null
            }
        }
    }

    [void] _close() {
        $this.Dispose()
    }

    # Dispose implementation (following IDisposable pattern)
    [void] Dispose() {
        $this.Dispose($true)
        [System.GC]::SuppressFinalize($this)
    }

    hidden [void] Dispose([bool]$disposing) {
        if (-not $this._disposed) {
            if ($disposing) {
                $this._Cleanup()
            }
            $this._disposed = $true
        }
    }
}

function New-Base {
    param(
        [Parameter(Mandatory)]
        [string]$hostname,
        
        [Parameter(Mandatory)]
        [ValidateRange(1, 65535)]
        [int]$port,
        
        [Parameter(Mandatory)]
        [string]$passwd
    )

    [Base]::new($hostname, $port, $passwd)
}