liblttng-ust-java-agent/java/lttng-ust-agent-common/org/lttng/ust/agent/client/SessiondCommand.java

   1 /*
   2  * Copyright (C) 2015-2016 - EfficiOS Inc., Alexandre Montplaisir <alexmonthy@efficios.com>
   3  * Copyright (C) 2013 - David Goulet <dgoulet@efficios.com>
   4  *
   5  * This library is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU Lesser General Public License, version 2.1 only,
   7  * as published by the Free Software Foundation.
   8  *
   9  * This library is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE. See the GNU Lesser General Public License
  12  * for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public License
  15  * along with this library; if not, write to the Free Software Foundation,
  16  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
  17  */
  18
  19 package org.lttng.ust.agent.client;
  20
  21 import java.nio.ByteBuffer;
  22
  23 /**
  24  * Base class to represent all commands sent from the session daemon to the Java
  25  * agent. The agent is then expected to execute the command and provide a
  26  * response.
  27  *
  28  * @author Alexandre Montplaisir
  29  */
  30 abstract class SessiondCommand {
  31
  32         enum CommandType {
  33
  34                 /** List logger(s). */
  35                 CMD_LIST(1),
  36                 /** Enable logger by name. */
  37                 CMD_ENABLE(2),
  38                 /** Disable logger by name. */
  39                 CMD_DISABLE(3),
  40                 /** Registration done */
  41                 CMD_REG_DONE(4);
  42
  43                 private int code;
  44
  45                 private CommandType(int c) {
  46                         code = c;
  47                 }
  48
  49                 public int getCommandType() {
  50                         return code;
  51                 }
  52         }
  53
  54         /**
  55          * Execute the command handler's action on the specified tracing agent.
  56          *
  57          * @param agent
  58          *            The agent on which to execute the command
  59          * @return If the command completed successfully or not
  60          */
  61         public abstract LttngAgentResponse execute(ILttngTcpClientListener agent);
  62
  63         /**
  64          * Utility method to read agent-protocol strings passed on the socket. The
  65          * buffer will contain a 32-bit integer representing the length, immediately
  66          * followed by the string itself.
  67          *
  68          * @param buffer
  69          *            The ByteBuffer from which to read. It should already be setup
  70          *            and positioned where the read should begin.
  71          * @return The string that was read, or <code>null</code> if it was badly
  72          *         formatted.
  73          */
  74         protected static String readNextString(ByteBuffer buffer) {
  75                 int length = buffer.getInt();
  76                 if (length < 0) {
  77                         /* The string length should be positive */
  78                         return null;
  79                 }
  80                 if (length == 0) {
  81                         /* The string is explicitly an empty string */
  82                         return "";
  83                 }
  84
  85                 byte[] stringBytes = new byte[length];
  86                 buffer.get(stringBytes);
  87                 return new String(stringBytes).trim();
  88         }
  89 }