PowerShell实时TCP接收问题：实现客户端同时处理终端输入与服务器实时输出

我懂你现在的困扰——想让你的TCP客户端同时监听服务器的实时消息，又能正常接收用户输入发送命令，而不是每次输入后才手动拉取作业输出对吧？你的思路方向是对的，但Start-Job的局限性导致没法实时自动更新，而且作业运行在独立的会话空间，原来的$connection、$reader这些变量在作业里根本访问不到，这也是你之前代码可能没正常工作的原因之一。

咱们换用Runspace来实现异步监听，它和主线程共享进程空间，能更高效地处理并行任务，还能实时输出服务器消息。下面是修改后的完整代码，我会标注关键改动点：

$server = "127.0.0.1"
$port = 2137
$connection = New-Object System.Net.Sockets.TcpClient($server, $port)
$stream = New-Object System.Net.Security.SslStream $connection.GetStream(),$false,({$true} -as [System.Net.Security.RemoteCertificateValidationCallback])
$stream.AuthenticateAsClient("Foo")
$reader = New-Object System.IO.BinaryReader($stream)
$writer = New-Object System.IO.StreamWriter($stream)
$writer.WriteLine("aadmin")
$writer.Flush()
$writer = New-Object System.IO.BinaryWriter($stream)
$read = $reader.ReadInt32()

if($read -eq 0) {
    $writer.Write(0)
    Write-Host "Password: " -NoNewline
    $password = Read-Host
    $writer.Write($password)
    $read = $reader.ReadInt32()
    
    if($read -eq 0) {
        # 1. 创建Runspace并传递需要共享的变量
        $initialSessionState = [System.Management.Automation.Runspaces.InitialSessionState]::CreateDefault()
        # 将reader和connection变量注入到Runspace中
        $initialSessionState.Variables.Add((New-Object System.Management.Automation.Runspaces.SessionStateVariableEntry -ArgumentList "reader", $reader, ""))
        $initialSessionState.Variables.Add((New-Object System.Management.Automation.Runspaces.SessionStateVariableEntry -ArgumentList "connection", $connection, ""))
        
        $runspace = [System.Management.Automation.Runspaces.RunspaceFactory]::CreateRunspace($initialSessionState)
        $runspace.Open()
        
        # 2. 定义异步读取服务器消息的脚本块
        $scriptBlock = {
            while($connection.Connected) {
                # 检查是否有可读取的数据
                if($connection.Available -gt 0) {
                    try {
                        $serverMessage = $reader.ReadString()
                        # 实时输出服务器消息，注意用Write-Host直接打印到终端
                        Write-Host "`n[Server] $serverMessage`n[admin@server] > " -NoNewline
                    } catch {
                        # 处理连接断开或读取错误
                        break
                    }
                }
                # 短暂休眠，避免占用过多CPU
                Start-Sleep -Milliseconds 100
            }
        }
        
        # 3. 启动Runspace任务
        $powershell = [System.Management.Automation.PowerShell]::Create().AddScript($scriptBlock)
        $powershell.Runspace = $runspace
        $asyncResult = $powershell.BeginInvoke()
        
        # 4. 主线程处理用户输入
        do {
            Write-Host "[admin@server] > " -NoNewline
            $in = Read-Host
            if($in -ne "exit" -and $connection.Connected) {
                $writer.Write($in)
                $writer.Flush() # 别忘了Flush，确保消息发送出去
            }
        } while($in -ne "exit" -and $connection.Connected)
        
        # 5. 清理资源
        $powershell.EndInvoke($asyncResult)
        $powershell.Dispose()
        $runspace.Close()
        $runspace.Dispose()
    } else {
        Write-Host "Error: Authentication failed"
    }
    
    # 关闭所有连接和流
    $reader.Close()
    $writer.Close()
    $stream.Close()
    $connection.Close()
} else {
    Write-Host "Error: Failed to initiate connection"
}

关键改动说明： #

• Runspace替代Start-Job：Runspace在同一个进程内运行，能直接访问我们注入的$reader和$connection变量，避免了Job的会话隔离问题。
• 实时输出处理：在异步脚本块里，每次读取到服务器消息后，用Write-Host带着换行和提示符前缀输出，这样不会打断用户输入的提示符位置。
• 资源清理：添加了Flush()确保用户输入的消息能立即发送，同时在结束时正确关闭Runspace和PowerShell实例，避免资源泄漏。
• 错误处理：添加了try-catch块处理读取过程中的异常（比如服务器突然断开连接），让脚本更健壮。

这样修改后，你的终端就能一边等待用户输入命令，一边实时弹出服务器返回的消息，完全不需要手动调用Receive-Job啦！

内容的提问来源于stack exchange，提问作者Ja3cob