Code/Tools/Statoscope/Statoscope/SocketLogDataStream.cs

   1 // Copyright 2001-2019 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 using System;
   4 using System.Collections.Generic;
   5 using System.Net.Sockets;
   6
   7 namespace Statoscope
   8 {
   9         class SocketLogDataStream : ILogDataStream
  10         {
  11                 NetworkStream m_netStream;
  12                 Queue<string> m_lines = new Queue<string>();
  13                 string m_partLine = "";
  14                 Byte[] m_buffer = new Byte[16 * 1024];
  15
  16                 public SocketLogDataStream(NetworkStream netStream)
  17                 {
  18                         m_netStream = netStream;
  19                 }
  20
  21                 public void Dispose()
  22                 {
  23                 }
  24
  25                 public int Peek()
  26                 {
  27                         while (m_lines.Count == 0)
  28                         {
  29                                 GetMoreLines();
  30                         }
  31
  32                         string nextLine = m_lines.Peek();
  33
  34             //System.Console.WriteLine(nextLine);
  35
  36                         return nextLine != "" ? nextLine[0] : 0;
  37                 }
  38
  39                 public string ReadLine()
  40                 {
  41                         while (m_lines.Count == 0)
  42                         {
  43                                 GetMoreLines();
  44                         }
  45
  46                         return m_lines.Dequeue();
  47                 }
  48
  49                 public bool IsEndOfStream
  50                 {
  51                         get
  52                         {
  53                                 if (m_lines.Count == 0)
  54                                 {
  55                                         GetMoreLines();
  56                                 }
  57
  58                                 return (m_lines.Count == 0);
  59                         }
  60                 }
  61
  62                 void GetMoreLines()
  63     {
  64       while (m_lines.Count == 0)
  65       {
  66         while (!m_netStream.DataAvailable)
  67         {
  68         }
  69
  70         //we may need a few read attempts to get a whole line (eg for screen shots)
  71         int numBytesRead = m_netStream.Read(m_buffer, 0, m_buffer.Length);
  72
  73         if (numBytesRead <= 0)
  74         {
  75           throw new Exception("NetworkStream Read failed");
  76         }
  77
  78         string str = System.Text.Encoding.ASCII.GetString(m_buffer, 0, numBytesRead);
  79         string[] lines = str.Split(new char[] { '\n' });
  80         int numLines = lines.Length;
  81         bool strEndsWithNewLine = str.EndsWith("\n");
  82
  83         if (strEndsWithNewLine)
  84         {
  85           // Split will have put an empty string on the end of lines
  86           numLines--;
  87         }
  88
  89         m_partLine += lines[0];
  90
  91         if (numLines > 1 || strEndsWithNewLine)
  92         {
  93           // m_partLine must now be complete
  94           m_lines.Enqueue(m_partLine);
  95           m_partLine = "";
  96         }
  97
  98         if (numLines > 1 && !strEndsWithNewLine)
  99         {
 100           // the last line is incomplete so keep it in m_partLine for later completion
 101           // note: the previous m_partLine must have already been added to m_lines above
 102           numLines--;
 103           m_partLine = lines[numLines];
 104         }
 105
 106         // don't take the first one as this will have been taken already to complete the last part line
 107         for (int i = 1; i < numLines; i++)
 108         {
 109           m_lines.Enqueue(lines[i]);
 110         }
 111
 112         //if (m_lines.Count == 0)
 113         //{
 114         //System.Console.WriteLine("Waiting for lines");
 115         //}
 116       }
 117
 118       if (m_lines.Count == 0)
 119       {
 120         System.Console.WriteLine("No more lines");
 121       }
 122     }
 123         }
 124 }