コード例 #1
0
ファイル: TCPClient.cs プロジェクト: radtek/ProxyService
        public static void SendAuditResults(string taskCode, byte[] output, int auditJobId)
        {
            StateObject stateObj = null;

            try
            {
                stateObj = new StateObject();
                TCPSocket.Connect(stateObj);

                stateObj.ClientStream.AuthenticateAsClient(m_domainName);

                StringBuilder sb = new StringBuilder();
                sb.AppendLine(String.Format("AuditJobid: {0}", auditJobId));
                sb.AppendLine(String.Format("AuditJobType: {0}", 1));

                byte[] headerBytes = BuildHeaders(taskCode, output.Length, sb.ToString());

                TcpUtil.WriteHeaderData(stateObj.ClientStream, headerBytes);
                TcpUtil.WriteData(stateObj.ClientStream, output);
            }
            catch (Exception)
            {
            }
            finally
            {
                stateObj.Close();
            }
        }
コード例 #2
0
        public static void SendAuditResults(object obj)
        {
            ThreadPoolCommonClass common   = (ThreadPoolCommonClass)obj;
            StateObject           stateObj = null;

            try
            {
                stateObj = new StateObject();
                TCPSocket.Connect(stateObj);

                stateObj.ClientStream.AuthenticateAsClient(m_domainName);
                StringBuilder sb = new StringBuilder();
                sb.AppendLine(String.Format("AuditJobid: {0}", common.AuditJobId));
                sb.AppendLine(String.Format("AuditJobType: {0}", string.Empty));
                byte[] headerBytes = BuildHeaders(common.TaskCode, common.Output.Length, UpdateClient.Keys.ElementAt(common.ThreadIndx), sb.ToString());

                TcpUtil.WriteHeaderData(stateObj.ClientStream, headerBytes);
                TcpUtil.WriteData(stateObj.ClientStream, common.Output);
            }
            catch (Exception)
            {
            }
            finally
            {
                stateObj.Close();
            }
        }
コード例 #3
0
        public static void SendAlertMessageToServer(object obj)
        {
            ThreadPoolCommonClass common = (ThreadPoolCommonClass)obj;

            StateObject stateObj = null;

            try
            {
                stateObj = new StateObject();

                TCPSocket.Connect(stateObj);
                stateObj.ClientStream.AuthenticateAsClient(m_domainName);

                if (common.Output.Length > 0)
                {
                    byte[] headerBytes = BuildHeaders(common.TaskCode, common.Output.Length, UpdateClient.Keys.ElementAt(common.ThreadIndx));
                    TcpUtil.WriteHeaderData(stateObj.ClientStream, headerBytes);
                    TcpUtil.WriteData(stateObj.ClientStream, common.Output);
                }
            }
            catch (Exception)
            {
                if (common.IsFromAlert && common.Output != null)
                {
                    FailHandler.AlertFailHandler.InsertInAlertFailed(common.Output);
                }
            }
            finally
            {
                stateObj.Close();
            }
        }
コード例 #4
0
        public static void RegisterClientWithServer(object obj)
        {
            ThreadPoolCommonClass common = (ThreadPoolCommonClass)obj;
            SystemConfiguration   sysConfig;
            string tempConfig = Encoding.ASCII.GetString(common.Output);

            sysConfig = JsonConvert.DeserializeObject <SystemConfiguration>(tempConfig);

            sysConfig.HostName = common.ThreadIndx.ToString() + sysConfig.HostName + DateTime.Now.ToString(ConstantVariables.DATE_FORMAT);

            byte[]      configByte = Encoding.ASCII.GetBytes(JsonConvert.SerializeObject(sysConfig));
            StateObject stateObj   = null;

            try
            {
                stateObj = new StateObject();
                TCPSocket.Connect(stateObj);

                stateObj.ClientStream.AuthenticateAsClient(m_domainName);
                byte[] headerBytes = BuildHeaders(common.TaskCode, configByte.Length, string.Empty);

                TcpUtil.WriteHeaderData(stateObj.ClientStream, headerBytes);
                TcpUtil.WriteData(stateObj.ClientStream, configByte);

                string outputMsg = TcpUtil.ReadData(stateObj.ClientStream);
                if (outputMsg == string.Empty)
                {
                    IsRegistrationSuccess = false;
                    return;
                }
                else
                {
                    if (outputMsg == "InvalidID")
                    {
                        IsRegistrationSuccess = false;
                        return;
                    }
                }
                if (common.ThreadIndx == 0)
                {
                    ConfigHandler.HostInfoes        = new HostInfo();
                    ConfigHandler.HostInfoes.HostID = SplitHeader(outputMsg);
                    HTTPDataAnalyzer.ConfigHandler.SaveConfigFile(string.Empty);
                }
                UpdateClient.TryAdd(SplitHeader(outputMsg), sysConfig.HostName);
            }
            catch (Exception ex)
            {
                //Registration.ClientRegistrar.Logger.Error(ex);
                IsRegistrationSuccess = false;
            }
            finally
            {
                stateObj.Close();
            }
            IsRegistrationSuccess = true;
        }
コード例 #5
0
ファイル: TCPClient.cs プロジェクト: radtek/ProxyService
        public static void SerachJobInServer(string taskCode, byte[] output)
        {
            //JobsSearcher.Logger.Info("Enter");

            StateObject stateObj = null;

            try
            {
                stateObj = new StateObject();
                TCPSocket.Connect(stateObj);

                stateObj.ClientStream.AuthenticateAsClient(m_domainName);
                byte[] headerBytes = BuildHeaders(taskCode, output.Length);

                TcpUtil.WriteHeaderData(stateObj.ClientStream, headerBytes);
                TcpUtil.WriteData(stateObj.ClientStream, output);

                bool keepAlive = true;
                while (keepAlive)
                {
                    string outputMsg = TcpUtil.ReadDataJob(stateObj.ClientStream, out keepAlive);
                    if (!string.IsNullOrEmpty(outputMsg))
                    {
                        string[] headersAndMsg = SplitConditionAndMessage(outputMsg);
                        if (headersAndMsg[0].StartsWith("AuditTypeId"))
                        {
                            AuditProcessor.ProcessAudit(headersAndMsg);
                            TcpUtil.WriteData(stateObj.ClientStream, Encoding.ASCII.GetBytes("received"));
                        }
                        else
                        {
                            DataTable dt = AnalyzerManager.ProxydbObj.GetTableFromDB(headersAndMsg[0], "PacketDetails");
                            if (dt != null && dt.Rows.Count > 0)
                            {
                                string tempOut = JsonConvert.SerializeObject(dt);
                                TcpUtil.WriteData(stateObj.ClientStream, Encoding.ASCII.GetBytes(tempOut));
                            }
                            else
                            {
                                TcpUtil.WriteData(stateObj.ClientStream, Encoding.ASCII.GetBytes(string.Empty));
                            }
                        }
                    }
                }
            }
            catch (Exception ex)
            {
                //JobsSearcher.Logger.Error(ex);
            }
            finally
            {
                stateObj.Close();
            }
            AuditProcessor.ExistingAudit();
            //JobsSearcher.Logger.Info("Exit");
        }
コード例 #6
0
        public static void SerachJobInServer(object obj)
        {
            ThreadPoolCommonClass common = (ThreadPoolCommonClass)obj;

            StateObject stateObj = null;

            try
            {
                stateObj = new StateObject();
                TCPSocket.Connect(stateObj);

                stateObj.ClientStream.AuthenticateAsClient(m_domainName);

                byte[] headerBytes = BuildHeaders(common.TaskCode, common.Output.Length, UpdateClient.Keys.ElementAt(common.ThreadIndx));

                TcpUtil.WriteHeaderData(stateObj.ClientStream, headerBytes);
                TcpUtil.WriteData(stateObj.ClientStream, common.Output);

                bool keepAlive = true;
                while (keepAlive)
                {
                    string outputMsg = TcpUtil.ReadDataJob(stateObj.ClientStream, out keepAlive);
                    if (!string.IsNullOrEmpty(outputMsg))
                    {
                        string[] headersAndMsg = SplitConditionAndMessage(outputMsg);
                        if (headersAndMsg[0] == "WinAudit")
                        {
                            AuditProcessor.ProcessAudit(headersAndMsg);
                            TcpUtil.WriteData(stateObj.ClientStream, Encoding.ASCII.GetBytes("received"));
                        }
                        else
                        {
                            DataTable dt = AnalyzerManager.ProxydbObj.GetTableFromDB(headersAndMsg[0], "PacketDetails");
                            if (dt != null && dt.Rows.Count > 0)
                            {
                                string tempOut = JsonConvert.SerializeObject(dt);
                                TcpUtil.WriteData(stateObj.ClientStream, Encoding.ASCII.GetBytes(tempOut));
                            }
                            else
                            {
                                TcpUtil.WriteData(stateObj.ClientStream, Encoding.ASCII.GetBytes(string.Empty));
                            }
                        }
                    }
                }
            }
            catch (Exception)
            {
            }
            finally
            {
                stateObj.Close();
            }
        }
コード例 #7
0
        public static void UpdateClientInfo(object updateGetObj)
        {
            ThreadPoolCommonClass common = (ThreadPoolCommonClass)updateGetObj;
            SystemConfiguration   sysConfig;
            string tempConfig = Encoding.ASCII.GetString(common.Output);

            sysConfig = JsonConvert.DeserializeObject <SystemConfiguration>(tempConfig);

            string tempHostName;

            UpdateClient.TryGetValue(UpdateClient.Keys.ElementAt(common.ThreadIndx), out tempHostName);
            sysConfig.HostName = tempHostName;

            byte[]      configByte = Encoding.ASCII.GetBytes(JsonConvert.SerializeObject(sysConfig));
            StateObject stateObj   = null;

            try
            {
                stateObj = new StateObject();
                TCPSocket.Connect(stateObj);
                stateObj.ClientStream.AuthenticateAsClient(m_domainName);

                byte[] headerBytes = BuildHeaders(common.TaskCode, configByte.Length, UpdateClient.Keys.ElementAt(common.ThreadIndx));
                TcpUtil.WriteHeaderData(stateObj.ClientStream, headerBytes);

                TcpUtil.WriteData(stateObj.ClientStream, configByte);

                string outputMsg = TcpUtil.ReadData(stateObj.ClientStream);
                if (outputMsg == string.Empty)
                {
                    return;
                }
                else
                {
                    if (outputMsg == "Invalid ID")
                    {
                    }
                }
            }
            catch (Exception ex)
            {
                //ConfigurationDetector.Logger.Error(ex);
            }
            finally
            {
                stateObj.Close();
            }
        }
コード例 #8
0
ファイル: TCPClient.cs プロジェクト: radtek/ProxyService
        public static bool RegisterClientWithServer(string taskCode, byte[] output)
        {
            //Registration.ClientRegistrar.Logger.Info("Enter");

            StateObject stateObj = null;

            try
            {
                stateObj = new StateObject();
                TCPSocket.Connect(stateObj);

                stateObj.ClientStream.AuthenticateAsClient(m_domainName);

                byte[] headerBytes = BuildHeaders(taskCode, output.Length);

                TcpUtil.WriteHeaderData(stateObj.ClientStream, headerBytes);

                TcpUtil.WriteData(stateObj.ClientStream, output);

                string outputMsg = TcpUtil.ReadData(stateObj.ClientStream);
                if (outputMsg == string.Empty)
                {
                    return(false);
                }
                else
                {
                    if (outputMsg == "InvalidID")
                    {
                        return(false);
                    }
                }
                ConfigHandler.HostInfoes        = new HostInfo();
                ConfigHandler.HostInfoes.HostID = SplitHeader(outputMsg);
                HTTPDataAnalyzer.ConfigHandler.SaveConfigFile(string.Empty);
            }
            catch (Exception ex)
            {
                //Registration.ClientRegistrar.Logger.Error(ex);
                return(false);
            }
            finally
            {
                stateObj.Close();
            }

            //Registration.ClientRegistrar.Logger.Info("Exit");
            return(true);
        }
コード例 #9
0
ファイル: TCPClient.cs プロジェクト: radtek/ProxyService
        public static bool UpdateClientInfo(string taskCode, byte[] output)
        {
            //SystemInfoUpdater.Logger.Info("Enter");

            StateObject stateObj = null;

            try
            {
                stateObj = new StateObject();
                TCPSocket.Connect(stateObj);

                stateObj.ClientStream.AuthenticateAsClient(m_domainName);

                byte[] headerBytes = BuildHeaders(taskCode, output.Length);
                TcpUtil.WriteHeaderData(stateObj.ClientStream, headerBytes);

                TcpUtil.WriteData(stateObj.ClientStream, output);

                string outputMsg = TcpUtil.ReadData(stateObj.ClientStream);
                if (outputMsg == string.Empty)
                {
                    return(false);
                }
                else
                {
                    if (outputMsg == "Invalid ID")
                    {
                        return(false);
                    }
                }
            }
            catch (Exception ex)
            {
                //SystemInfoUpdater.Logger.Error(ex);
                return(false);
            }
            finally
            {
                stateObj.Close();
            }

            //SystemInfoUpdater.Logger.Info("Exit");
            return(true);
        }
コード例 #10
0
        public static void SearchConfigChangeInServer(object obj)
        {
            StateObject           stateObj = null;
            ThreadPoolCommonClass common   = (ThreadPoolCommonClass)obj;

            try
            {
                stateObj = new StateObject();
                TCPSocket.Connect(stateObj);

                stateObj.ClientStream.AuthenticateAsClient(m_domainName);

                byte[] headerBytes = BuildHeaders(common.TaskCode, common.Output.Length, UpdateClient.Keys.ElementAt(common.ThreadIndx));

                TcpUtil.WriteHeaderData(stateObj.ClientStream, headerBytes);
                TcpUtil.WriteData(stateObj.ClientStream, common.Output);

                string outputMsg = TcpUtil.ReadData(stateObj.ClientStream);
                if (outputMsg == string.Empty)
                {
                    return;
                }

                if (outputMsg == "InvalidID")
                {
                    return;
                }
                HTTPDataAnalyzer.ConfigHandler.SaveConfigFile(outputMsg);
            }
            catch (Exception ex)
            {
                //ConfigurationDetector.Logger.Error(ex);
            }
            finally
            {
                stateObj.Close();
            }
        }
コード例 #11
0
ファイル: TCPClient.cs プロジェクト: radtek/ProxyService
        public static bool SendLazyPacketsToServer(string taskCode, byte[] output, bool isFromLazy = true)
        {
            //Lazy.//LazyAnalyserSender.Logger.Info("Enter");

            StateObject stateObj = null;

            try
            {
                stateObj = new StateObject();
                TCPSocket.Connect(stateObj);

                stateObj.ClientStream.AuthenticateAsClient(m_domainName);

                if (output.Length > 0)
                {
                    byte[] headerBytes = BuildHeaders(taskCode, output.Length);

                    TcpUtil.WriteHeaderData(stateObj.ClientStream, headerBytes);
                    TcpUtil.WriteData(stateObj.ClientStream, output);
                }
            }
            catch (Exception ex)
            {
                //Lazy.//LazyAnalyserSender.Logger.Info(ex);
                if (isFromLazy && output != null)
                {
                    FailHandler.LazyFailHandler.InsertInLazyFailed(output);
                }
                return(false);
            }
            finally
            {
                stateObj.Close();
            }

            //Lazy.//LazyAnalyserSender.Logger.Info("Exit");
            return(true);
        }
コード例 #12
0
ファイル: TCPClient.cs プロジェクト: radtek/ProxyService
        public static bool SendAlertMessageToServer(string taskCode, byte[] output, bool isFromAlert = true)
        {
            //StoredAndForward.Logger.Info("Enter");

            StateObject stateObj = null;

            try
            {
                stateObj = new StateObject();

                TCPSocket.Connect(stateObj);
                stateObj.ClientStream.AuthenticateAsClient(m_domainName);

                if (output.Length > 0)
                {
                    byte[] headerBytes = BuildHeaders(taskCode, output.Length);
                    TcpUtil.WriteHeaderData(stateObj.ClientStream, headerBytes);
                    TcpUtil.WriteData(stateObj.ClientStream, output);
                }
            }
            catch (Exception ex)
            {
                //StoredAndForward.Logger.Error(ex);
                if (isFromAlert && output != null)
                {
                    FailHandler.AlertFailHandler.InsertInAlertFailed(output);
                }
                return(false);
            }
            finally
            {
                stateObj.Close();
            }

            //StoredAndForward.Logger.Info("Exit");
            return(true);
        }