コード例 #1
0
  /** Main dispatch method called from the LatherBoss. */
  public LatherValue dispatch(LatherContext ctx, String method, LatherValue arg)
      throws LatherRemoteException {

    Integer agentId = null;
    if (!haService.alertTriggersHaveInitialized()) {
      if (log.isDebugEnabled()) {
        log.debug("Not ready - received request for " + method + " from " + ctx.getCallerIP());
      }
      throw new LatherRemoteException("Server still initializing");
    }

    if (log.isDebugEnabled()) {
      log.debug("Request for " + method + "() from " + ctx.getCallerIP());
    }

    if (!HAUtil.isMasterNode()) {
      log.warn("Non-primary server received communication from an agent.  Request will be denied.");
      throw new LatherRemoteException(
          "This server is not the primary node in the HA configuration. Agent request denied.");
    }

    if (secureCommands.contains(method)) {
      if (!(arg instanceof SecureAgentLatherValue)) {
        log.warn(
            "Authenticated call made from "
                + ctx.getCallerIP()
                + " which did not subclass the correct authentication class");
        throw new LatherRemoteException("Unauthorized agent denied");
      }

      String agentToken = ((SecureAgentLatherValue) arg).getAgentToken();
      validateAgent(ctx, agentToken);
      synchronized (tokensToTime) {
        tokensToTime.put(agentToken, System.currentTimeMillis());
      }
      try {
        Agent a = agentManager.getAgent(agentToken);
        agentId = a.getId();
      } catch (AgentNotFoundException e) {
        log.debug(e, e);
      }
    }

    AgentConnection conn = null;
    long start = 0;
    try {
      conn = agentManager.getAgentConnection(method, ctx.getCallerIP(), agentId);
      start = now();
      return runCommand(ctx, method, arg);
    } catch (LatherRemoteException e) {
      concurrentStatsCollector.addStat(1, LATHER_REMOTE_EXCEPTION);
      throw e;
    } finally {
      if (conn != null) {
        agentManager.disconnectAgent(conn);
      }
      long duration = now() - start;
      concurrentStatsCollector.addStat(duration, LATHER_RUN_COMMAND_TIME);
    }
  }
コード例 #2
0
 private void checkUserCanManageAgent(
     LatherContext ctx, String user, String pword, String operation) throws PermissionException {
   try {
     authManager.authenticate(user, pword);
     AuthzSubject subject =
         authzSubjectManager.findSubjectByAuth(user, HQConstants.ApplicationName);
     permissionManager.checkCreatePlatformPermission(subject);
   } catch (SecurityException exc) {
     log.warn(
         "Security exception when '"
             + user
             + "' tried to "
             + operation
             + " an Agent @ "
             + ctx.getCallerIP(),
         exc);
     throw new PermissionException();
   } catch (PermissionException exc) {
     log.warn(
         "Permission denied when '"
             + user
             + "' tried to "
             + operation
             + " an Agent @ "
             + ctx.getCallerIP());
     throw new PermissionException();
   } catch (ApplicationException exc) {
     log.warn(
         "Application exception when '"
             + user
             + "' tried to "
             + operation
             + " an Agent @ "
             + ctx.getCallerIP(),
         exc);
     throw new PermissionException();
   } catch (SystemException exc) {
     log.warn(
         "System exception when '"
             + user
             + "' tried to "
             + operation
             + " an Agent @ "
             + ctx.getCallerIP(),
         exc);
     throw new PermissionException();
   }
 }
コード例 #3
0
 private LatherValue runCommand(LatherContext ctx, String method, LatherValue arg)
     throws LatherRemoteException {
   LatherValue rtn = null;
   long start = now();
   if (method.equals(CommandInfo.CMD_PING)) {
     rtn = cmdPing(arg);
     concurrentStatsCollector.addStat(now() - start, ConcurrentStatsCollector.CMD_PING);
   } else if (method.equals(CommandInfo.CMD_USERISVALID)) {
     rtn = cmdUserIsValid(ctx, (UserIsValid_args) arg);
     concurrentStatsCollector.addStat(now() - start, ConcurrentStatsCollector.CMD_USERISVALID);
   } else if (method.equals(CommandInfo.CMD_MEASUREMENT_SEND_REPORT)) {
     rtn = cmdMeasurementSendReport((MeasurementSendReport_args) arg);
     concurrentStatsCollector.addStat(
         now() - start, ConcurrentStatsCollector.CMD_MEASUREMENT_SEND_REPORT);
   } else if (method.equals(CommandInfo.CMD_MEASUREMENT_GET_CONFIGS)) {
     rtn = cmdMeasurementGetConfigs((MeasurementGetConfigs_args) arg);
     concurrentStatsCollector.addStat(
         now() - start, ConcurrentStatsCollector.CMD_MEASUREMENT_GET_CONFIGS);
   } else if (method.equals(CommandInfo.CMD_REGISTER_AGENT)) {
     rtn = cmdRegisterAgent(ctx, (RegisterAgent_args) arg);
     concurrentStatsCollector.addStat(now() - start, ConcurrentStatsCollector.CMD_REGISTER_AGENT);
   } else if (method.equals(CommandInfo.CMD_UPDATE_AGENT)) {
     rtn = cmdUpdateAgent(ctx, (UpdateAgent_args) arg);
     concurrentStatsCollector.addStat(now() - start, ConcurrentStatsCollector.CMD_UPDATE_AGENT);
   } else if (method.equals(CommandInfo.CMD_AI_SEND_REPORT)) {
     rtn = cmdAiSendReport((AiSendReport_args) arg);
     concurrentStatsCollector.addStat(now() - start, ConcurrentStatsCollector.CMD_AI_SEND_REPORT);
   } else if (method.equals(CommandInfo.CMD_AI_SEND_RUNTIME_REPORT)) {
     rtn = cmdAiSendRuntimeReport((AiSendRuntimeReport_args) arg);
     concurrentStatsCollector.addStat(
         now() - start, ConcurrentStatsCollector.CMD_AI_SEND_RUNTIME_REPORT);
   } else if (method.equals(CommandInfo.CMD_TRACK_SEND_LOG)) {
     rtn = cmdTrackLogMessage((TrackSend_args) arg);
     concurrentStatsCollector.addStat(now() - start, ConcurrentStatsCollector.CMD_TRACK_SEND_LOG);
   } else if (method.equals(CommandInfo.CMD_TRACK_SEND_CONFIG_CHANGE)) {
     rtn = cmdTrackConfigChange((TrackSend_args) arg);
     concurrentStatsCollector.addStat(
         now() - start, ConcurrentStatsCollector.CMD_TRACK_SEND_CONFIG_CHANGE);
   } else if (method.equals(CommandInfo.CMD_CONTROL_GET_PLUGIN_CONFIG)) {
     rtn = cmdControlGetPluginConfig((ControlGetPluginConfig_args) arg);
     concurrentStatsCollector.addStat(
         now() - start, ConcurrentStatsCollector.CMD_CONTROL_GET_PLUGIN_CONFIG);
   } else if (method.equals(CommandInfo.CMD_CONTROL_SEND_COMMAND_RESULT)) {
     rtn = cmdControlSendCommandResult((ControlSendCommandResult_args) arg);
     concurrentStatsCollector.addStat(
         now() - start, ConcurrentStatsCollector.CMD_CONTROL_SEND_COMMAND_RESULT);
   } else if (method.equals(CommandInfo.CMD_PLUGIN_SEND_REPORT)) {
     rtn = cmdAgentPluginReport((PluginReport_args) arg);
     concurrentStatsCollector.addStat(
         now() - start, ConcurrentStatsCollector.CMD_PLUGIN_SEND_REPORT);
   } else if (method.equals(CommandInfo.CMD_TOPN_SEND_REPORT)) {
     rtn = cmdTopNSendReport((TopNSendReport_args) arg);
   } else {
     log.warn(
         ctx.getCallerIP() + " attempted to invoke '" + method + "' which could not be found");
     throw new LatherRemoteException("Unknown method, '" + method + "'");
   }
   return rtn;
 }
コード例 #4
0
 /** Command to make sure that a user is valid. */
 private UserIsValid_result cmdUserIsValid(LatherContext ctx, UserIsValid_args arg) {
   try {
     authManager.authenticate(arg.getUser(), arg.getPword());
   } catch (Exception exc) {
     log.warn(
         "An invalid user(" + arg.getUser() + ") connected from " + ctx.getCallerIP() + ": " + exc,
         exc);
     return new UserIsValid_result(false);
   }
   return new UserIsValid_result(true);
 }