Tools/SmtpServer.ps1
|
function Start-SmtpServer { # ensure we have smtp handlers if ($null -eq (Get-PodeTcpHandler -Type 'SMTP')) { throw 'No SMTP handler has been passed' } # grab the relavant port $port = $PodeSession.Server.IP.Port if ($port -eq 0) { $port = 25 } # create the listener for smtp $endpoint = New-Object System.Net.IPEndPoint($PodeSession.Server.IP.Address, $port) $listener = New-Object System.Net.Sockets.TcpListener -ArgumentList $endpoint try { # start listener $listener.Start() } catch { $Error[0] | Out-Default if ($null -ne $listener) { $listener.Stop() } throw $_.Exception } # state where we're running Write-Host "Listening on smtp://$($PodeSession.Server.IP.Name):$($port) [$($PodeSession.Threads) thread(s)]" -ForegroundColor Yellow # script for listening out of for incoming requests $listenScript = { param ( [Parameter(Mandatory=$true)] [ValidateNotNull()] $Listener, [Parameter(Mandatory=$true)] [int] $ThreadId ) # scriptblock for the core smtp message processing logic $process = { # if there's no client, just return if ($null -eq $TcpSession.Client) { return } # variables to store data for later processing $mail_from = [string]::Empty $rcpt_tos = @() $data = [string]::Empty # open response to smtp request tcp write "220 $($PodeSession.Server.IP.Name) -- Pode Proxy Server" $msg = [string]::Empty # respond to smtp request while ($true) { try { $msg = (tcp read) } catch { break } try { if (!(Test-Empty $msg)) { if ($msg.StartsWith('QUIT')) { tcp write '221 Bye' if ($null -ne $TcpSession.Client -and $TcpSession.Client.Connected) { dispose $TcpSession.Client -Close } break } if ($msg.StartsWith('EHLO') -or $msg.StartsWith('HELO')) { tcp write '250 OK' } if ($msg.StartsWith('RCPT TO')) { tcp write '250 OK' $rcpt_tos += (Get-SmtpEmail $msg) } if ($msg.StartsWith('MAIL FROM')) { tcp write '250 OK' $mail_from = Get-SmtpEmail $msg } if ($msg.StartsWith('DATA')) { tcp write '354 Start mail input; end with <CR><LF>.<CR><LF>' $data = (tcp read) tcp write '250 OK' # set session data $SmtpSession.From = $mail_from $SmtpSession.To = $rcpt_tos $SmtpSession.Data = $data $SmtpSession.Lockable = $PodeSession.Lockable # call user handlers for processing smtp data Invoke-ScriptBlock -ScriptBlock (Get-PodeTcpHandler -Type 'SMTP') -Arguments $SmtpSession -Scoped # reset the to list $rcpt_tos = @() } } } catch [exception] { throw $_.exception } } } try { while (!$PodeSession.Tokens.Cancellation.IsCancellationRequested) { # get an incoming request $task = $Listener.AcceptTcpClientAsync() $task.Wait($PodeSession.Tokens.Cancellation.Token) $client = $task.Result # convert the ip $ip = (ConvertTo-IPAddress -Endpoint $client.Client.RemoteEndPoint) # ensure the request ip is allowed if (!(Test-IPAccess -IP $ip) -or !(Test-IPLimit -IP $ip)) { dispose $client -Close } # deal with smtp call else { $SmtpSession = @{} $TcpSession = @{ 'Client' = $client; 'Lockable' = $PodeSession.Lockable } Invoke-ScriptBlock -ScriptBlock $process } } } catch [System.OperationCanceledException] {} catch { $Error[0] | Out-Default throw $_.Exception } } # start the runspace for listening on x-number of threads 1..$PodeSession.Threads | ForEach-Object { Add-PodeRunspace -Type 'Main' -ScriptBlock $listenScript ` -Parameters @{ 'Listener' = $listener; 'ThreadId' = $_ } } # script to keep smtp server listening until cancelled $waitScript = { param ( [Parameter(Mandatory=$true)] [ValidateNotNull()] $Listener ) try { while (!$PodeSession.Tokens.Cancellation.IsCancellationRequested) { Start-Sleep -Seconds 1 } } catch [System.OperationCanceledException] {} catch { $Error[0] | Out-Default throw $_.Exception } finally { if ($null -ne $Listener) { $Listener.Stop() } } } Add-PodeRunspace -Type 'Main' -ScriptBlock $waitScript -Parameters @{ 'Listener' = $listener } } function Get-SmtpEmail { param ( [Parameter()] [string] $Value ) $tmp = ($Value -isplit ':') if ($tmp.Length -gt 1) { return $tmp[1].Trim().Trim('<', '>') } return [string]::Empty } |