opendaylight/netconf/netconf-util/src/main/java/org/opendaylight/controller/netconf/util/handler/NetconfMessageAggregator.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8
   9 package org.opendaylight.controller.netconf.util.handler;
  10
  11 import com.google.common.base.Charsets;
  12 import io.netty.buffer.ByteBuf;
  13 import io.netty.channel.ChannelHandlerContext;
  14 import io.netty.handler.codec.ByteToMessageDecoder;
  15 import org.opendaylight.controller.netconf.util.messages.FramingMechanism;
  16 import org.opendaylight.controller.netconf.util.messages.NetconfMessageConstants;
  17 import org.slf4j.Logger;
  18 import org.slf4j.LoggerFactory;
  19
  20 import java.util.List;
  21
  22 public class NetconfMessageAggregator extends ByteToMessageDecoder {
  23
  24     private final static Logger logger = LoggerFactory.getLogger(NetconfMessageAggregator.class);
  25
  26     private byte[] eom = NetconfMessageConstants.endOfMessage;
  27
  28     public NetconfMessageAggregator(FramingMechanism framingMechanism) {
  29         if (framingMechanism == FramingMechanism.CHUNK) {
  30             eom = NetconfMessageConstants.endOfChunk;
  31         }
  32     }
  33
  34     @Override
  35     protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) throws Exception {
  36         int index = indexOfSequence(in, eom);
  37         if (index == -1) {
  38             logger.debug("Message is not complete, read again.");
  39             if (logger.isTraceEnabled()) {
  40                 String str = in.toString(Charsets.UTF_8);
  41                 logger.trace("Message read so far: {}", str);
  42             }
  43             ctx.read();
  44         } else {
  45             ByteBuf msg = in.readBytes(index);
  46             in.readBytes(eom.length);
  47             in.discardReadBytes();
  48             logger.debug("Message is complete.");
  49             out.add(msg);
  50         }
  51     }
  52
  53     private int indexOfSequence(ByteBuf in, byte[] sequence) {
  54         int index = -1;
  55         for (int i = 0; i < in.readableBytes() - sequence.length + 1; i++) {
  56             if (in.getByte(i) == sequence[0]) {
  57                 index = i;
  58                 for (int j = 1; j < sequence.length; j++) {
  59                     if (in.getByte(i + j) != sequence[j]) {
  60                         index = -1;
  61                         break;
  62                     }
  63                 }
  64                 if (index != -1) {
  65                     return index;
  66                 }
  67             }
  68         }
  69         return index;
  70     }
  71
  72 }