public void handleMessage(final Connection connection, final InputStream dataStream)
     throws IOException {
   final int cmd = readUnsignedByte(dataStream);
   if (cmd != Protocol.AUTH) {
     log.warnf(
         "Received unrecognized greeting code 0x%02x from %s",
         Integer.valueOf(cmd), connection.getPeerAddress());
     connection.close();
     return;
   }
   final int version = StreamUtils.readUnsignedByte(dataStream);
   if (version < 1) {
     log.warnf("Received connection with invalid version from %s", connection.getPeerAddress());
     connection.close();
     return;
   }
   final byte[] authCode = new byte[16];
   StreamUtils.readFully(dataStream, authCode);
   final ManagedProcess process = processController.getServerByAuthCode(authCode);
   if (process == null) {
     log.warnf(
         "Received connection with unknown credentials from %s", connection.getPeerAddress());
     StreamUtils.safeClose(connection);
     return;
   }
   log.tracef("Received authentic connection from %s", connection.getPeerAddress());
   connection.setMessageHandler(
       new ConnectedMessageHandler(processController, process.isInitial()));
   processController.addManagedConnection(connection);
   dataStream.close();
 }
Exemplo n.º 2
0
  private boolean lookupServerInModel(String host, String server) throws Exception {
    final ModelNode operation = new ModelNode();
    operation.get(OP).set(READ_RESOURCE_OPERATION);
    operation.get(OP_ADDR).set(getHostControllerServerAddress(host, server));

    final ModelControllerClient client = ModelControllerClient.Factory.create("localhost", HC_PORT);
    try {
      final ModelNode result = client.execute(operation);
      if (result.get(OUTCOME).asString().equals(SUCCESS)) {
        final ModelNode model = result.require(RESULT);
        if (model.hasDefined(NAME) && model.get(NAME).asString().equals(server)) {
          return true;
        }
      }
    } catch (IOException e) {
    } finally {
      StreamUtils.safeClose(client);
    }
    return false;
  }
 protected void printHelp(CommandContext ctx) {
   InputStream helpInput =
       SecurityActions.getClassLoader(CommandHandlerWithHelp.class).getResourceAsStream(filename);
   if (helpInput != null) {
     BufferedReader reader = new BufferedReader(new InputStreamReader(helpInput));
     try {
       String helpLine = reader.readLine();
       while (helpLine != null) {
         ctx.printLine(helpLine);
         helpLine = reader.readLine();
       }
     } catch (java.io.IOException e) {
       ctx.printLine("Failed to read help/help.txt: " + e.getLocalizedMessage());
     } finally {
       StreamUtils.safeClose(reader);
     }
   } else {
     ctx.printLine("Failed to locate command description " + filename);
   }
   return;
 }
Exemplo n.º 4
0
    List<String> listProcesses() {
      final Process p;
      try {
        p = Runtime.getRuntime().exec(getJpsCommand());
      } catch (IOException e) {
        throw new RuntimeException(e);
      }

      List<String> processes = new ArrayList<String>();
      BufferedReader input = new BufferedReader(new InputStreamReader(p.getInputStream()));
      try {
        String line;
        while ((line = input.readLine()) != null) {
          if (line.contains("jboss-modules.jar")) {
            processes.add(line);
          }
        }
      } catch (IOException e) {
        throw new RuntimeException(e);
      } finally {
        StreamUtils.safeClose(input);
      }
      return processes;
    }
 /** {@inheritDoc} */
 public synchronized void stop(StopContext context) {
   safeClose(client);
   client = null;
   channel = null;
 }
      public void handleMessage(final Connection connection, final InputStream dataStream)
          throws IOException {
        try {

          final int cmd = StreamUtils.readUnsignedByte(dataStream);
          switch (cmd) {
            case Protocol.SEND_STDIN:
              {
                // HostController only
                if (isHostController) {
                  final String processName = readUTFZBytes(dataStream);
                  log.tracef("Received send_stdin for process %s", processName);
                  processController.sendStdin(processName, dataStream);
                } else {
                  log.tracef("Ignoring send_stdin message from untrusted source");
                }
                dataStream.close();
                break;
              }
            case Protocol.ADD_PROCESS:
              {
                if (isHostController) {
                  final String processName = readUTFZBytes(dataStream);
                  final byte[] authKey = new byte[16];
                  readFully(dataStream, authKey);
                  final int commandCount = readInt(dataStream);
                  final String[] command = new String[commandCount];
                  for (int i = 0; i < commandCount; i++) {
                    command[i] = readUTFZBytes(dataStream);
                  }
                  final int envCount = readInt(dataStream);
                  final Map<String, String> env = new HashMap<String, String>();
                  for (int i = 0; i < envCount; i++) {
                    env.put(readUTFZBytes(dataStream), readUTFZBytes(dataStream));
                  }
                  final String workingDirectory = readUTFZBytes(dataStream);
                  log.tracef("Received add_process for process %s", processName);
                  processController.addProcess(
                      processName, Arrays.asList(command), env, workingDirectory, false);
                } else {
                  log.tracef("Ignoring add_process message from untrusted source");
                }
                dataStream.close();
                break;
              }
            case Protocol.START_PROCESS:
              {
                if (isHostController) {
                  final String processName = readUTFZBytes(dataStream);
                  processController.startProcess(processName);
                  log.tracef("Received start_process for process %s", processName);
                } else {
                  log.tracef("Ignoring start_process message from untrusted source");
                }
                dataStream.close();
                break;
              }
            case Protocol.STOP_PROCESS:
              {
                if (isHostController) {
                  final String processName = readUTFZBytes(dataStream);
                  // HostController only
                  processController.stopProcess(processName);
                } else {
                  log.tracef("Ignoring stop_process message from untrusted source");
                }
                dataStream.close();
                break;
              }
            case Protocol.REMOVE_PROCESS:
              {
                if (isHostController) {
                  final String processName = readUTFZBytes(dataStream);
                  processController.removeProcess(processName);
                } else {
                  log.tracef("Ignoring remove_process message from untrusted source");
                }
                dataStream.close();
                break;
              }
            case Protocol.REQUEST_PROCESS_INVENTORY:
              {
                if (isHostController) {
                  processController.sendInventory();
                } else {
                  log.tracef("Ignoring request_process_inventory message from untrusted source");
                }
                dataStream.close();
                break;
              }
            case Protocol.RECONNECT_PROCESS:
              {
                if (isHostController) {
                  final String processName = readUTFZBytes(dataStream);
                  final String hostName = readUTFZBytes(dataStream);
                  final int port = readInt(dataStream);
                  processController.sendReconnectProcess(processName, hostName, port);
                } else {
                  log.tracef("Ignoring reconnect_process message from untrusted source");
                }
                dataStream.close();
                break;
              }
            case Protocol.SHUTDOWN:
              {
                if (isHostController) {
                  new Thread(
                          new Runnable() {
                            public void run() {
                              processController.shutdown();
                              System.exit(0);
                            }
                          })
                      .start();
                } else {
                  log.tracef("Ignoring shutdown message from untrusted source");
                }
                break;
              }
            default:
              {
                log.warnf("Received unknown message with code 0x%02x", Integer.valueOf(cmd));
                // unknown
                dataStream.close();
              }
          }
        } finally {
          safeClose(dataStream);
        }
      }