Code/Tools/Statoscope/Statoscope/SocketLogProcessor.cs

   1 // Copyright 2001-2019 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 using System;
   4 using System.IO;
   5 using System.Collections.Generic;
   6 using System.Linq;
   7 using System.Text;
   8 using System.Net.Sockets;
   9 using System.Windows.Forms;
  10 using System.Threading;
  11
  12 namespace Statoscope
  13 {
  14         class SocketLogProcessor
  15         {
  16                 public LogData ProcessLog(StatoscopeForm ssForm)
  17                 {
  18                         TcpClient client = null;
  19                         bool bShouldLogToFile = false;
  20                         string connectionName = "";
  21
  22                         using (ConnectMessageBox connectMsgBox = new ConnectMessageBox())
  23                         {
  24                                 bool successful = false;
  25                                 while (!successful)
  26                                 {
  27                                         if (connectMsgBox.ShowDialog() == DialogResult.OK)
  28                                         {
  29                                                 try
  30                                                 {
  31                                                         client = new TcpClient(connectMsgBox.IpAddress, connectMsgBox.Port);
  32                                                         bShouldLogToFile = connectMsgBox.logToFileCheckBox.Checked;
  33                                                         connectionName = connectMsgBox.ConnectionName;
  34                                                         successful = true;
  35                                                 }
  36                                                 catch (Exception ex)
  37                                                 {
  38                                                         ex.ToString();
  39                                                         ConnectionErrorDialog ced = new ConnectionErrorDialog();
  40                                                         ced.errorLabel.Text += connectMsgBox.IpAddress + ":" + connectMsgBox.Port;
  41                                                         ced.ShowDialog();
  42                                                 }
  43                                         }
  44                                         else
  45                                         {
  46                                                 return null;
  47                                         }
  48                                 }
  49                         }
  50
  51                         string logWriteFilename = "";
  52
  53                         if (bShouldLogToFile)
  54                         {
  55                                 SaveFileDialog fd = new SaveFileDialog();
  56                                 fd.CheckFileExists = false;
  57                                 fd.FileName = "log.bin";
  58                                 fd.Filter = "Binary log files (*.bin)|*.bin|All files (*.*)|*.*";
  59                                 fd.FilterIndex = 0;
  60
  61                                 if (fd.ShowDialog() == DialogResult.OK)
  62                                 {
  63                                         logWriteFilename = fd.FileName;
  64                                 }
  65                         }
  66
  67                         SocketSessionInfo sessionInfo = new SocketSessionInfo(connectionName, logWriteFilename);
  68                         SocketLogData logData = new SocketLogData(sessionInfo);
  69                         logData.ProcessRecords();
  70
  71                         List<LogRange> logRanges = new List<LogRange>();
  72                         LogRange logRange = new LogRange(logData, null);
  73
  74                         logRanges.Add(logRange);
  75                         ssForm.AddToLogRangeList(logRange);
  76
  77                         ssForm.CreateLogControlTabPage(logRanges, connectionName, null);
  78
  79                         ThreadPool.QueueUserWorkItem(new WaitCallback(ProcessSocketDataStreamCB), new SSocketParseArgs(logData, logWriteFilename, client));
  80
  81                         return logData;
  82                 }
  83
  84                 struct SSocketParseArgs
  85                 {
  86                         public SocketLogData m_logData;
  87                         public TcpClient m_client;
  88
  89                         public SSocketParseArgs(SocketLogData logData, string logWriteFilename, TcpClient client)
  90                         {
  91                                 m_logData = logData;
  92                                 m_logData.m_logWriteFilename = logWriteFilename;
  93                                 m_client = client;
  94                         }
  95                 }
  96
  97                 static void ProcessSocketDataStreamCB(object state)
  98                 {
  99                         SSocketParseArgs args = (SSocketParseArgs)state;
 100                         NetworkStream netStream = args.m_client.GetStream();
 101
 102                         if (args.m_logData.m_logWriteFilename == "")
 103                         {
 104                                 args.m_logData.m_logWriteFilename = Path.GetTempFileName();
 105                                 args.m_logData.m_bTempFile = true;
 106                         }
 107
 108                         using (SocketLogBinaryDataStream logDataStream = new SocketLogBinaryDataStream(netStream, args.m_logData.m_logWriteFilename))
 109                         {
 110                                 try
 111                                 {
 112                                         args.m_logData.DataStream = logDataStream;
 113                                         BinaryLogParser logParser = new BinaryLogParser(logDataStream, args.m_logData, "");
 114                                         logParser.ProcessDataStream();
 115                                 }
 116                                 catch (Exception)
 117                                 {
 118                                 }
 119                         }
 120
 121                         netStream.Close();
 122                         args.m_client.Close();
 123                 }
 124         }
 125 }