KnowBe4-Partner

0.1.2

Private/Helpers/New-SAMLInteractive.ps1

function New-SAMLInteractive {
    [CmdletBinding()]
    param (
        [Parameter(Mandatory)]
        [string]$LoginIDP
    )

    begin {
        $RegEx = '(?i)name="SAMLResponse"(?: type="hidden")? value=\"(.*?)\"(?:.*)?\/>'
        Add-Type -AssemblyName System.Windows.Forms
        Add-Type -AssemblyName System.Web
    }

    process {
        # Create window for embedded browser
        $form = New-Object Windows.Forms.Form
        $form.StartPosition = [System.Windows.Forms.FormStartPosition]::CenterScreen
        $form.Width = 660
        $form.Height = 760
        $form.showIcon = $false
        $form.Topmost = $true

        $web = New-Object Windows.Forms.WebBrowser
        $web.Size = $form.ClientSize
        $web.Anchor = "Left,Top,Right,Bottom"
        $web.ScriptErrorsSuppressed = $true

        $form.Controls.Add($web)
        $web.Navigate($LoginIDP)
        $web.add_Navigating({

                if ($web.DocumentText -match "SAMLResponse") {
                    $_.cancel = $true
                    if ($web.DocumentText -match $RegEx) {
                        $form.Close()
                        $Script:SAMLResponse = $(($Matches[1] -replace '+', '+') -replace '=', '=')
                    }
                }
            })
        # Show browser window, waits for window to close
        if ($form.ShowDialog() -ne "OK") {
            if ($null -ne $Script:SAMLResponse) {
                Write-Output $Script:SAMLResponse
                $form.Close()
                Remove-Variable -Name SAMLResponse -Scope Script -ErrorAction SilentlyContinue
            } else {
                throw "SAMLResponse not matched"
            }
        }
    }
    end {
        $form.Dispose()
    }
}