helpers/tcp.psm1
|
function Invoke-TCP { [CmdletBinding()] param ( [Parameter(Mandatory=$true, Position=0)] [PSObject] $Receiver, [Parameter(Mandatory=$true, Position=1)] [PSObject] $Sender, [Parameter(Mandatory=$true, Position=2)] $LogDir ) $ModuleBase = (Get-Module Test-NetStack -ListAvailable | Select-Object -First 1).ModuleBase # CTS Traffic Rate Limit is specified in bytes/second $ServerLinkSpeed = $Receiver.LinkSpeed.split(" ") Switch($ServerLinkSpeed[1]) { ("Gbps") {$ServerLinkSpeedBps = [Int]::Parse($ServerLinkSpeed[0]) * [Math]::Pow(10, 9) / 8} ("Mbps") {$ServerLinkSpeedBps = [Int]::Parse($ServerLinkSpeed[0]) * [Math]::Pow(10, 6) / 8} ("Kbps") {$ServerLinkSpeedBps = [Int]::Parse($ServerLinkSpeed[0]) * [Math]::Pow(10, 3) / 8} ("bps") {$ServerLinkSpeedBps = [Int]::Parse($ServerLinkSpeed[0]) / 8} } $ClientLinkSpeed = $Sender.LinkSpeed.split(" ") Switch($ClientLinkSpeed[1]) { ("Gbps") {$ClientLinkSpeedBps = [Int]::Parse($ClientLinkSpeed[0]) * [Math]::Pow(10, 9) / 8} ("Mbps") {$ClientLinkSpeedBps = [Int]::Parse($ClientLinkSpeed[0]) * [Math]::Pow(10, 6) / 8} ("Kbps") {$ClientLinkSpeedBps = [Int]::Parse($ClientLinkSpeed[0]) * [Math]::Pow(10, 3) / 8} ("bps") {$ClientLinkSpeedBps = [Int]::Parse($ClientLinkSpeed[0]) / 8} } $BufferSize = 65536 if ($ServerLinkSpeed -ge 50) { $BufferSize = $BufferSize * 10 } $ServerOutput = Start-Job -ScriptBlock { param ([string] $ServerName, [string] $ServerIP, $ModuleBase, $LogDir, $BufferSize) Invoke-Command -ComputerName $ServerName -ScriptBlock { param ([string] $ServerIP, [string] $ModuleBase, $LogDir, $BufferSize) Set-Location $ModuleBase cmd /c ".\tools\NTttcp\ntttcp.exe -r -m 64,*,$ServerIP -l $BufferSize -a 16 -v -t 20" | Out-File "$($LogDir)\NTttcp_$($ServerIP)_Recv_$(Get-Date -f yyyy-MM-dd-HHmmss).txt" -Append } -ArgumentList $ServerIP, $ModuleBase, $LogDir, $BufferSize } -ArgumentList $Receiver.NodeName, $Receiver.IPAddress, $ModuleBase, $LogDir, $BufferSize $ServerRecvCounter = Start-Job -ScriptBlock { param ([string] $ServerName, [string] $ServerInterfaceDescription) $ServerInterfaceDescription = (((($ServerInterfaceDescription) -replace '#', '_') -replace '[(]', '[') -replace '[)]', ']') -replace '/', '_' Invoke-Command -ComputerName $ServerName -ScriptBlock { param([string]$ServerInterfaceDescription) Get-Counter -Counter "\Network Adapter($ServerInterfaceDescription)\Bytes Received/sec" -MaxSamples 20 -ErrorAction Ignore } -ArgumentList $ServerInterfaceDescription } -ArgumentList $Receiver.NodeName, $Receiver.InterfaceDescription $ClientSendCounter = Start-Job -ScriptBlock { param([string] $ClientName, [string] $ClientInterfaceDescription) $ClientInterfaceDescription = (((($ClientInterfaceDescription) -replace '#', '_') -replace '[(]', '[') -replace '[)]', ']') -replace '/', '_' Invoke-Command -ComputerName $ClientName -ScriptBlock { param ([string] $ClientInterfaceDescription) Get-Counter -Counter "\Network Adapter($ClientInterfaceDescription)\Bytes Sent/sec" -MaxSamples 20 } -ArgumentList $ClientInterfaceDescription } -ArgumentList $Sender.NodeName,$Sender.InterfaceDescription $ClientOutput = Start-Job -ScriptBlock { param ([string] $ClientName, [string] $ServerIP, [string] $ClientIP, [string] $ModuleBase, $LogDir, $BufferSize) Invoke-Command -ComputerName $ClientName -ScriptBlock { param ([string] $ServerIP, [string] $ClientIP, $ModuleBase, $LogDir, $BufferSize) Set-Location $ModuleBase cmd /c ".\tools\NTttcp\ntttcp.exe -s -m 64,*,$ServerIP -l $BufferSize -a 16 -v -t 20" | Out-File "$($LogDir)\NTttcp_$($clientip)_Send_$(Get-Date -f yyyy-MM-dd-HHmmss).txt" -Append } -ArgumentList $ServerIP, $ClientIP, $ModuleBase, $LogDir, $BufferSize } -ArgumentList $Sender.NodeName, $Receiver.IPAddress, $Sender.IPAddress, $ModuleBase, $LogDir, $BufferSize Sleep 20 $ServerRecv = Receive-Job $ServerRecvCounter $ClientSend = Receive-Job $ClientSendCounter $FlatServerRecvOutput = $ServerRecv.Readings.split(":") | ForEach-Object { try {[uint64]($_) * 8} catch {} } $FlatClientSendOutput = $ClientSend.Readings.split(":") | ForEach-Object { try {[uint64]($_) * 8} catch {} } $ServerRecvBitsPerSecond = [Math]::Round(($FlatServerRecvOutput | Measure-Object -Maximum).Maximum, 2) $ClientSendBitsPerSecond = [Math]::Round(($FlatClientSendOutput | Measure-Object -Maximum).Maximum, 2) Write-Verbose "Server Recv bps: $ServerRecvBitsPerSecond" Write-Verbose "Client Send bps: $ClientSendBitsPerSecond" $ServerOutput = Receive-Job $ServerOutput $ClientOutput = Receive-Job $ClientOutput $ServerLinkSpeedBitsPerSecond = $ServerLinkSpeedBps * 8 $ClientLinkSpeedBitsPerSecond = $ClientLinkSpeedBps * 8 $MinLinkSpeedBitsPerSecond = ($ServerLinkSpeedBitsPerSecond, $ClientLinkSpeedBitsPerSecond | Measure-Object -Minimum).Minimum Write-Verbose "Minimum Link Speed bps: $MinLinkSpeedBitsPerSecond" $RawData = New-Object -TypeName psobject $RawData | Add-Member -MemberType NoteProperty -Name ServerRxbps -Value $ServerRecvBitsPerSecond $RawData | Add-Member -MemberType NoteProperty -Name ClientTxbps -Value $ClientSendBitsPerSecond $RawData | Add-Member -MemberType NoteProperty -Name MinLinkSpeedbps -Value $MinLinkSpeedBitsPerSecond $ReceiverLinkSpeedGbps = [Math]::Round($ServerLinkSpeedBitsPerSecond * [Math]::Pow(10, -9), 2) $ReceivedGbps = [Math]::Round($ServerRecvBitsPerSecond * [Math]::Pow(10, -9), 2) $ReceivedPercentageOfLinkSpeed = [Math]::Round(($ReceivedGbps / $ReceiverLinkSpeedGbps) * 100, 2) $TCPResults = New-Object -TypeName psobject $TCPResults | Add-Member -MemberType NoteProperty -Name ReceiverLinkSpeedGbps -Value $ReceiverLinkSpeedGbps $TCPResults | Add-Member -MemberType NoteProperty -Name ReceivedGbps -Value $ReceivedGbps $TCPResults | Add-Member -MemberType NoteProperty -Name ReceivedPctgOfLinkSpeed -Value $ReceivedPercentageOfLinkSpeed $TCPResults | Add-Member -MemberType NoteProperty -Name RawData -Value $RawData Return $TCPResults } |