从NetworkStream中读取特定的字节数

时间:2011-09-24 22:08:13

标签: sockets stream

我正在尝试从网络流中读取已知长度的消息。 我有点期待NetworkStream.Read()等待返回,直到我给它的缓冲区数组已满。如果没有,那么ReadTimeout属性的重点是什么?

我用来测试我的理论的示例代码

public static void Main(string[] args)
{
    TcpListener listener = new TcpListener(IPAddress.Any, 10001);
    listener.Start();

    Console.WriteLine("Waiting for connection...");

    ThreadPool.QueueUserWorkItem(WriterThread);

    using (TcpClient client = listener.AcceptTcpClient())
    using (NetworkStream stream = client.GetStream())
    {
        Console.WriteLine("Connected. Waiting for data...");

        client.ReceiveTimeout = (int)new TimeSpan(0, 1, 0).TotalMilliseconds;
        stream.ReadTimeout = (int)new TimeSpan(0, 1, 0).TotalMilliseconds;

        byte[] buffer = new byte[1024];
        int bytesRead = stream.Read(buffer, 0, buffer.Length);

        Console.WriteLine("Got {0} bytes.", bytesRead);
    }

    listener.Stop();

    Console.WriteLine("Press any key to exit...");
    Console.ReadKey(true);
}

private static void WriterThread(object state)
{
    using (TcpClient client = new TcpClient())
    {
        client.Connect(new IPEndPoint(IPAddress.Parse("127.0.0.1"), 10001));
        using (NetworkStream stream = client.GetStream())
        {
            byte[] bytes = Encoding.UTF8.GetBytes("obviously less than 1024 bytes");
            Console.WriteLine("Sending {0} bytes...", bytes.Length);
            stream.Write(bytes, 0, bytes.Length);
            Thread.Sleep(new TimeSpan(0, 2, 0));
        }
    }
}

结果是:

Waiting for connection...
Sending 30 bytes...
Connected. Waiting for data...
Got 30 bytes.
Press any key to exit...

是否有一种标准的同步读取方式,只有在读取指定的字节数时才会返回?我确信自己编写一个并不太复杂,但TcpClientNetworkStream上的超时属性的存在表明它应该已经以这种方式工作。

2 个答案:

答案 0 :(得分:5)

您所保证的是(其中之一):

  • 0字节(流结束)
  • 至少1个字节(某些数据可用;并不意味着没有更多或已经可用)
  • 错误(超时等)

读取指定数量的字节... loop:

int read = 0, offset = 0, toRead = ...
while(toRead > 0 && (read = stream.Read(buffer, offset, toRead)) > 0) {
    toRead -= read;
    offset += read;
}
if(toRead > 0) throw new EndOfStreamException();

答案 1 :(得分:2)

TCP是一种字节流协议,不保留应用程序消息边界。它根本无法以这种方式将字节“粘合”在一起。读取超时的目的是指定要阻止读取的时间。但只要至少返回一个字节的数据,读操作就不会阻塞。

如果您需要在循环中调用read,直到您阅读完整的消息,请执行此操作。 TCP层并不关心你认为是完整的消息,这不是它的工作。