/// <summary>
 /// Process user driven requests
 /// </summary>
 /// <param name="json">Json recieved form the external client shown here as a string.</param>
 public static void ProcessRequest(string json)
 {
     //Check if the json is in the HeartBeat class
     if (json.Contains("heartbeat"))
     {
         //Deserialize the json to a HeatBeat class
         HeartBeat beat      = JsonConvert.DeserializeObject <HeartBeat>(json);
         int       HighestID = 0;
         foreach (var beats in Program.Clients)
         {
             //Check if the client is already in the HeatBeat list, if it is not add it to the list
             if (beats.ipSocket == beat.ipSocket)
             {
                 ConsoleView.AddLog("Client already in database");
                 return;
             }
             if (beats.pcId > HighestID)
             {
                 HighestID = beats.pcId;
             }
         }
         Program.Clients.Add(
             new HeartBeat
         {
             pcId       = HighestID + 1,
             heartbeat  = beat.heartbeat,
             ipSocket   = beat.ipSocket,
             lastUpdate = beat.lastUpdate,
             pcName     = beat.pcName
         }
             );
         ConsoleView.AddLog("Added client on " + beat.ipSocket + " with ID:" + (HighestID + 1).ToString());
     }
     //If the json is a restart command tell the server application that the client is restarting
     if (json.Contains("restart"))
     {
         ConsoleView.AddLog("Restarting remote computer");
     }
     ConsoleView.Update();
 }
Exemplo n.º 2
0
        /// <summary>
        ///
        /// </summary>
        static void ClientHeartbeatTimer_Elapsed(object sender, ElapsedEventArgs e)
        {
            int clientIndex = 0;
            int clientId    = 0;

            if (Clients != null)
            {
                try {
                    foreach (var client in Clients)
                    {
                        if (!WebPost.SendRequest(client.ipSocket, true, "{\"heartbeat\":true, \"pcId\":" + client.pcId.ToString() + "}"))
                        {
                            //Remove client from Client
                            clientId = client.pcId;
                            Clients.RemoveAt(clientIndex);

                            //Remove client from ClientUsage
                            clientIndex = 0;
                            foreach (var clientU in ClientUsage)
                            {
                                if (clientU.pcId == clientId)
                                {
                                    ClientUsage.RemoveAt(clientIndex);
                                }
                                clientIndex++;
                            }
                            ConsoleView.AddLog("Heartbeat failed, removing client: " + client.ipSocket);
                            return;
                        }
                        clientIndex++;
                    }
                }catch (Exception)
                {
                }
            }
            ConsoleData.CurrentConnectedClients = clientIndex;
            ConsoleView.Update();
        }
Exemplo n.º 3
0
        static void Main(string[] args)
        {
            //First update of the console to draw UI
            ConsoleView.Update();
            //Set timer up initially and activate it
            ClientHeartbeatTimer          = new Timer(HeartbeatInterval * 1000);
            ClientHeartbeatTimer.Elapsed += new ElapsedEventHandler(ClientHeartbeatTimer_Elapsed);
            ClientHeartbeatTimer.Enabled  = true;

            while (true)
            {
                //Wait for user input
                string ConsoleString = Console.ReadLine();
                //Split console commands via a space or " " represented as a null
                String[] ConsoleCommand = ConsoleString.Split(null);
                switch (ConsoleCommand[0])
                {
                //The add command to add a new client to the Clients list
                case "add":
                    if (!WebPost.SendRequest(ConsoleCommand[1]))
                    {
                        ConsoleView.AddLog("Failed to add client on socket: " + ConsoleCommand[1]);
                    }
                    else
                    {
                        ConsoleData.CurrentConnectedClients += 1;
                    }
                    break;

                //Set the interval between
                case "update":
                    HeartbeatInterval             = Convert.ToInt32(ConsoleCommand[1]);
                    ClientHeartbeatTimer.Interval = (HeartbeatInterval * 1000);
                    ConsoleView.AddLog("Update rate modifier: " + ConsoleCommand[1]);
                    break;

                //Restart the remote computer
                case "restart":
                case "reboot":
                    if (!WebPost.SendRequest(ConsoleCommand[1], false, "{\"restartPc\" : true}"))
                    {
                        ConsoleView.AddLog("Failed to reboot remote computer");
                    }
                    else
                    {
                        ConsoleView.AddLog("Restarting remote computer " + ConsoleCommand[1]);
                    }
                    break;

                //show the next 5 clients in the list on the console ui
                case "list":
                    ConsoleView.WriteClients(Convert.ToInt32(ConsoleCommand[1]));
                    break;

                //Basic commands
                case "help":
                case "?":
                    Console.WriteLine("Commands Available:");
                    Console.WriteLine("add <socket>, clear");
                    break;
                }
                ConsoleView.ClearInput();
                ConsoleView.Update();
            }
        }