Ejemplo n.º 1
0
 @Override
 public void decodeRest(HornetQBuffer buffer) {
   super.decodeRest(buffer);
   handler = buffer.readSimpleString();
   voteBuffer = buffer.readSlice(buffer.readableBytes());
 }
Ejemplo n.º 2
0
  public StompFrame unmarshal(HornetQBuffer in) throws IOException {

    try {
      String action = null;

      // skip white space to next real action line
      while (true) {
        action = readLine(in, MAX_COMMAND_LENGTH, "The maximum command length was exceeded");
        if (action == null) {
          throw new IOException("connection was closed");
        } else {
          action = action.trim();
          if (action.length() > 0) {
            break;
          }
        }
      }

      // Parse the headers
      HashMap headers = new HashMap(25);
      while (true) {
        String line = readLine(in, MAX_HEADER_LENGTH, "The maximum header length was exceeded");
        if (line != null && line.trim().length() > 0) {

          if (headers.size() > MAX_HEADERS) {
            throw new StompException("The maximum number of headers was exceeded", true);
          }

          try {
            int seperator_index = line.indexOf(Stomp.Headers.SEPERATOR);
            String name = line.substring(0, seperator_index).trim();
            String value = line.substring(seperator_index + 1, line.length()).trim();
            headers.put(name, value);
          } catch (Exception e) {
            throw new StompException("Unable to parser header line [" + line + "]", true);
          }
        } else {
          break;
        }
      }

      // Read in the data part.
      byte[] data = NO_DATA;
      String contentLength = (String) headers.get(Stomp.Headers.CONTENT_LENGTH);
      if (contentLength != null) {

        // Bless the client, he's telling us how much data to read in.
        int length;
        try {
          length = Integer.parseInt(contentLength.trim());
        } catch (NumberFormatException e) {
          throw new StompException("Specified content-length is not a valid integer", true);
        }

        if (length > MAX_DATA_LENGTH) {
          throw new StompException("The maximum data length was exceeded", true);
        }

        data = new byte[length];
        in.readBytes(data);

        if (in.readByte() != 0) {
          throw new StompException(
              Stomp.Headers.CONTENT_LENGTH
                  + " bytes were read and "
                  + "there was no trailing null byte",
              true);
        }
      } else {

        // We don't know how much to read.. data ends when we hit a 0
        byte b;
        ByteArrayOutputStream baos = null;
        while (in.readableBytes() > 0 && (b = in.readByte()) != 0) {

          if (baos == null) {
            baos = new ByteArrayOutputStream();
          } else if (baos.size() > MAX_DATA_LENGTH) {
            throw new StompException("The maximum data length was exceeded", true);
          }

          baos.write(b);
        }

        if (baos != null) {
          baos.close();
          data = baos.toByteArray();
        }
      }

      return new StompFrame(action, headers, data);
    } catch (StompException e) {
      return new StompFrameError(e);
    }
  }