public void getResources(int resourceType, String menu) { try { ByteArrayWriter baw = new ByteArrayWriter(); baw.writeInt(resourceType); Request request = new Request("getResources", baw.toByteArray()); if (agent.getConnection().sendRequest(request, true)) { if (request.getRequestData() != null) { ByteArrayReader reader = new ByteArrayReader(request.getRequestData()); int count = (int) reader.readInt(); if (count > 0) { if (agent.getGUI().isMenuExists(menu)) agent.getGUI().clearMenu(menu); else agent.getGUI().addMenu(menu); for (int i = 0; i < count; i++) { int resourceId = (int) reader.readInt(); agent .getGUI() .addMenuItem(menu, new ResourceLaunchAction(resourceId, reader.readString())); } } else { if (agent.getGUI().isMenuExists(menu)) agent.getGUI().removeMenu(menu); } } } } catch (IOException e) { // #ifdef DEBUG log.error("Failed to get resources.", e); // #endif } }
/** * Get a set of the resource ids of all active tunnels (local and remote). * * @param session * @return resource IDs of active tunnels */ public Set<Integer> getActiveTunnels(SessionInfo session) { try { MultiplexedConnection tunnel = DefaultAgentManager.getInstance().getAgentBySession(session); if (tunnel == null) return null; HashSet<Integer> activeTunnelIds = new HashSet<Integer>(); // The agent keeps track of 'local' tunnels Request request = new Request(ACTIVE_LOCAL_TUNNELS); if (tunnel.sendRequest(request, true) && request.getRequestData() != null) { ByteArrayReader reader = new ByteArrayReader(request.getRequestData()); int count = (int) reader.readInt(); for (int i = 0; i < count; i++) { activeTunnelIds.add(new Integer((int) reader.readInt())); } } // The server keeps track of 'remote' tunnels Collection<RemoteTunnel> activeRemoteTunnels = RemoteTunnelManagerFactory.getInstance().getRemoteTunnels(session); if (activeRemoteTunnels != null) { synchronized (activeRemoteTunnels) { for (RemoteTunnel r : activeRemoteTunnels) { activeTunnelIds.add(r.getTunnel().getResourceId()); } } } return activeTunnelIds; } catch (IOException e) { log.error("Failed to get active tunnel list from agent", e); return null; } }
public boolean processRequest(Request request, MultiplexedConnection connection) { AgentTunnel agent = (AgentTunnel) connection; if (request.getRequestName().equals(SETUP_AND_LAUNCH_TUNNEL) && request.getRequestData() != null) { try { ByteArrayReader reader = new ByteArrayReader(request.getRequestData()); int id = (int) reader.readInt(); Tunnel resource = (Tunnel) TunnelPlugin.SSL_TUNNEL_RESOURCE_TYPE.getResourceById(id); if (resource == null) { throw new Exception("No resource with ID " + id); } Policy policy = LaunchSessionManager.getLaunchRequestPolicy(null, agent.getSession(), resource); if (resource.sessionPasswordRequired(agent.getSession())) { // TODO: prompt user for credentials through agent! return true; } else { LaunchSession launchSession = LaunchSessionFactory.getInstance() .createLaunchSession(agent.getSession(), resource, policy); launchSession.checkAccessRights(null, agent.getSession()); if (resource.getType() == TransportType.LOCAL_TUNNEL_ID) { try { Request req = buildLocalTunnel(resource, launchSession); request.setRequestData(req.getRequestData()); return true; } catch (IOException ioe) { throw new TunnelException(TunnelException.INTERNAL_ERROR, ioe); } } else if (resource.getType() == TransportType.REMOTE_TUNNEL_ID) { startRemoteTunnel(agent, resource, launchSession); request.setRequestData(null); return true; } else { throw new TunnelException( TunnelException.INTERNAL_ERROR, (Throwable) null, "Unknown tunnel type " + resource.getType()); } } } catch (Exception e) { log.error("Failed to start tunnel.", e); return false; } } return false; }