TCP PowerShell Server
I would like to ask you how multiple connection streams can be handled.
I have implemented a TCP server like this:
$endpoint = New-Object System.Net.IPEndPoint ([System.Net.IPAddress]::Any, 8989)
$listener = New-Object System.Net.Sockets.TcpListener $endpoint
$listener.Start()
do {
$client = $listener.AcceptTcpClient() # will block here until connection
$stream = $client.GetStream();
$reader = New-Object System.IO.StreamReader $stream
do {
$line = $reader.ReadLine()
Write-Host $line -fore cyan
} while ($line -and $line -ne ([char]4))
$reader.Dispose()
$stream.Dispose()
$client.Dispose()
} while ($line -ne ([char]4))
$listener.Stop()
This code can only handle one thread at a time. Can you give me some advice on how to create a TCP server in PowerShell that can handle multiple clients?
+3
source to share
1 answer
You need multiple threads to handle multiple clients and you need to use runspaces for that . Below is a working code that accepts multiple clients and processes each client on a separate thread (space)
$Global: Listener = [HashTable]::Synchronized(@{})
$Global: CnQueue = [System.Collections.Queue]::Synchronized((New - Object System.collections.queue))
$Global: space = [RunSpaceFactory]::CreateRunspace()
$space.Open()
$space.SessionStateProxy.setVariable("CnQueue", $CnQueue)
$space.SessionStateProxy.setVariable("Listener", $Listener)
$Global: newPowerShell = [PowerShell]::Create()
$newPowerShell.Runspace = $space
$Timer = New - Object Timers.Timer
$Timer.Enabled = $true
$Timer.Interval = 1000
Register - ObjectEvent - SourceIdentifier MonitorClientConnection - InputObject $Timer - EventName Elapsed - Action {
While($CnQueue.count - ne 0) {
$client = $CnQueue.Dequeue()
$newRunspace = [RunSpaceFactory]::CreateRunspace()
$newRunspace.Open()
$newRunspace.SessionStateProxy.setVariable("client", $client)
$newPowerShell = [PowerShell]::Create()
$newPowerShell.Runspace = $newRunspace
$process = {
$stream = $client.GetStream();
$reader = New - Object System.IO.StreamReader $stream
[console]::WriteLine("Inside Processing")
# You have client here so do whatever you want to do here.
# This is a separate thread so if you write blocking code here, it will not impact any other part of the program
}
$jobHandle = $newPowerShell.AddScript($process).BeginInvoke()
#jobHandle you need to save for future to cleanup
}
}
$listener = {
$Listener['listener'] = New - Object System.Net.Sockets.TcpListener("127.0.0.1", "1234")
$Listener['listener'].Start()[console]::WriteLine("Listening on :1234")
while ($true) {
$c = $Listener['listener'].AcceptTcpClient()
If($c - ne $Null) {
[console]::WriteLine("{0} >> Accepted Client " - f(Get - Date).ToString())
$CnQueue.Enqueue($c)
}
Else {
[console]::WriteLine("Shutting down")
Break
}
}
}
$Timer.Start()
$Global: handle = $newPowerShell.AddScript($listener).BeginInvoke()
For a more detailed example go here
+1
source to share