q3rcon-ps/lib/base.ps1

class Base {
    [string]$hostname
    [int]$port
    [string]$passwd
    [Object]$request
    [Object]$response
    hidden [System.Net.Sockets.Socket] $_socket

    Base ([string]$hostname, [int]$port, [string]$passwd) {
        $this.hostname = $hostname
        $this.port = $port
        $this.passwd = $passwd

        $this.request = New-RequestPacket($this.passwd)
        $this.response = New-ResponsePacket

        $ip = [system.net.IPAddress]::Parse([System.Net.Dns]::GetHostAddresses($this.hostname)[0].IPAddressToString)
        
        $endpoint = New-Object System.Net.IPEndPoint $ip, $this.port 
        try {
            $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 = 100
        }
        catch [System.Net.Sockets.SocketException] {
            throw "Failed to create UDP connection to server."          
        }
    }

    [string] ToString () {
        return "Rcon connection {0}:{1} with pass {2}" -f $this.hostname, $this.port, $this.passwd
    }

    [string] _send([string]$msg) {
        $this._socket.Send($this.request.Payload($msg))

        $buf = New-Object System.Byte[] 4096
        try {
            $this._socket.Receive($buf)
        }
        catch [System.Net.Sockets.SocketException] {
            if ( $_.Exception.SocketErrorCode -eq 'TimedOut' ) {
                "finished waiting for fragment" | Write-Debug
            }
        }
        return [System.Text.Encoding]::ASCII.GetString($($buf | Select-Object -Skip $($this.response.Header().Length - 1)))
    }

    [string] _send([string]$msg, [int]$timeout) {
        $this._socket.Send($this.request.Payload($msg))

        [string[]]$data = @()
        $sw = [Diagnostics.Stopwatch]::StartNew()
        While ($sw.ElapsedMilliseconds -lt $timeout) {
            try {
                $buf = New-Object System.Byte[] 4096
                $this._socket.Receive($buf)
                $data += [System.Text.Encoding]::ASCII.GetString($($buf | Select-Object -Skip $($this.response.Header().Length - 1)))
            }
            catch [System.Net.Sockets.SocketException] {
                if ( $_.Exception.SocketErrorCode -eq 'TimedOut' ) {
                    "finished waiting for fragment" | Write-Debug
                }
            }
        }
        $sw.Stop()
        return [string]::Join("", $data)
    }

    [void] _close() {
        $this._socket.Close()
    }
}

Function New-Base {
    param([string]$hostname, [int]$port, [string]$passwd)

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