*/
package org.opendaylight.protocol.pcep.impl.object;
+import com.google.common.base.Preconditions;
+import com.google.common.primitives.UnsignedBytes;
+import io.netty.buffer.ByteBuf;
+import io.netty.buffer.ByteBufUtil;
+import java.util.ArrayList;
import java.util.List;
-
-import org.opendaylight.protocol.pcep.spi.EROSubobjectHandlerRegistry;
-import org.opendaylight.protocol.pcep.spi.EROSubobjectSerializer;
+import org.opendaylight.protocol.pcep.spi.EROSubobjectRegistry;
import org.opendaylight.protocol.pcep.spi.ObjectParser;
import org.opendaylight.protocol.pcep.spi.ObjectSerializer;
import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
-import org.opendaylight.protocol.util.ByteArray;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.ero.Subobjects;
+import org.opendaylight.protocol.util.Values;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.ero.Subobject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
-import com.google.common.base.Preconditions;
-import com.google.common.collect.Lists;
-import com.google.common.primitives.UnsignedBytes;
-
public abstract class AbstractEROWithSubobjectsParser implements ObjectParser, ObjectSerializer {
- private static final Logger LOG = LoggerFactory.getLogger(AbstractEROWithSubobjectsParser.class);
-
- private static final int SUB_TYPE_FLAG_F_LENGTH = 1;
- private static final int SUB_LENGTH_F_LENGTH = 1;
- private static final int SUB_HEADER_LENGTH = SUB_TYPE_FLAG_F_LENGTH + SUB_LENGTH_F_LENGTH;
-
- private static final int TYPE_FLAG_F_OFFSET = 0;
- private static final int LENGTH_F_OFFSET = TYPE_FLAG_F_OFFSET + SUB_TYPE_FLAG_F_LENGTH;
- private static final int SO_CONTENTS_OFFSET = LENGTH_F_OFFSET + SUB_LENGTH_F_LENGTH;
-
- private final EROSubobjectHandlerRegistry subobjReg;
-
- protected AbstractEROWithSubobjectsParser(final EROSubobjectHandlerRegistry subobjReg) {
- this.subobjReg = Preconditions.checkNotNull(subobjReg);
- }
-
- protected List<Subobjects> parseSubobjects(final byte[] bytes) throws PCEPDeserializerException {
- if (bytes == null) {
- throw new IllegalArgumentException("Byte array is mandatory.");
- }
-
- boolean loose = false;
- int type;
- byte[] soContentsBytes;
- int length;
- int offset = 0;
-
- final List<Subobjects> subs = Lists.newArrayList();
-
- while (offset < bytes.length) {
-
- loose = ((bytes[offset + TYPE_FLAG_F_OFFSET] & (1 << 7)) != 0) ? true : false;
- length = ByteArray.bytesToInt(ByteArray.subByte(bytes, offset + LENGTH_F_OFFSET, SUB_LENGTH_F_LENGTH));
-
- type = (bytes[offset + TYPE_FLAG_F_OFFSET] & 0xff) & ~(1 << 7);
-
- if (length > bytes.length - offset) {
- throw new PCEPDeserializerException("Wrong length specified. Passed: " + length + "; Expected: <= "
- + (bytes.length - offset));
- }
-
- soContentsBytes = new byte[length - SO_CONTENTS_OFFSET];
- System.arraycopy(bytes, offset + SO_CONTENTS_OFFSET, soContentsBytes, 0, length - SO_CONTENTS_OFFSET);
-
- LOG.debug("Attempt to parse subobject from bytes: {}", ByteArray.bytesToHexString(soContentsBytes));
- final Subobjects sub = this.subobjReg.getSubobjectParser(type).parseSubobject(soContentsBytes, loose);
- LOG.debug("Subobject was parsed. {}", sub);
-
- subs.add(sub);
-
- offset += length;
- }
- return subs;
- }
-
- protected final byte[] serializeSubobject(final List<Subobjects> subobjects) {
-
- final List<byte[]> result = Lists.newArrayList();
-
- int finalLength = 0;
-
- for (final Subobjects subobject : subobjects) {
-
- final EROSubobjectSerializer serializer = this.subobjReg.getSubobjectSerializer(subobject.getSubobjectType());
-
- final byte[] valueBytes = serializer.serializeSubobject(subobject);
-
- final byte[] bytes = new byte[SUB_HEADER_LENGTH + valueBytes.length];
-
- final byte typeBytes = (byte) (UnsignedBytes.checkedCast(serializer.getType()) | (subobject.isLoose() ? 1 << 7 : 0));
- final byte lengthBytes = UnsignedBytes.checkedCast(valueBytes.length + SUB_HEADER_LENGTH);
-
- bytes[0] = typeBytes;
- bytes[1] = lengthBytes;
- System.arraycopy(valueBytes, 0, bytes, SUB_HEADER_LENGTH, valueBytes.length);
-
- finalLength += bytes.length;
- result.add(bytes);
- }
-
- final byte[] resultBytes = new byte[finalLength];
- int byteOffset = 0;
- for (final byte[] b : result) {
- System.arraycopy(b, 0, resultBytes, byteOffset, b.length);
- byteOffset += b.length;
- }
- return resultBytes;
- }
+ private static final Logger LOG = LoggerFactory.getLogger(AbstractEROWithSubobjectsParser.class);
+
+ private static final int HEADER_LENGTH = 2;
+
+ private final EROSubobjectRegistry subobjReg;
+
+ protected AbstractEROWithSubobjectsParser(final EROSubobjectRegistry subobjReg) {
+ this.subobjReg = Preconditions.checkNotNull(subobjReg);
+ }
+
+ protected List<Subobject> parseSubobjects(final ByteBuf buffer) throws PCEPDeserializerException {
+ // Explicit approval of empty ERO
+ Preconditions.checkArgument(buffer != null, "Array of bytes is mandatory. Can't be null.");
+ final List<Subobject> subs = new ArrayList<>();
+ while (buffer.isReadable()) {
+ final boolean loose = ((buffer.getByte(buffer.readerIndex()) & (1 << Values.FIRST_BIT_OFFSET)) != 0) ? true : false;
+ final int type = (buffer.readByte() & Values.BYTE_MAX_VALUE_BYTES) & ~(1 << Values.FIRST_BIT_OFFSET);
+ final int length = UnsignedBytes.toInt(buffer.readByte()) - HEADER_LENGTH;
+ if (length > buffer.readableBytes()) {
+ throw new PCEPDeserializerException("Wrong length specified. Passed: " + length + "; Expected: <= "
+ + buffer.readableBytes());
+ }
+ LOG.debug("Attempt to parse subobject from bytes: {}", ByteBufUtil.hexDump(buffer));
+ final Subobject sub = this.subobjReg.parseSubobject(type, buffer.readSlice(length), loose);
+ if (sub == null) {
+ LOG.warn("Unknown subobject type: {}. Ignoring subobject.", type);
+ } else {
+ LOG.debug("Subobject was parsed. {}", sub);
+ subs.add(sub);
+ }
+ }
+ return subs;
+ }
+
+ protected final void serializeSubobject(final List<Subobject> subobjects, final ByteBuf buffer) {
+ if(subobjects != null && !subobjects.isEmpty()) {
+ for (final Subobject subobject : subobjects) {
+ this.subobjReg.serializeSubobject(subobject, buffer);
+ }
+ }
+ }
}
Code Review / bgpcep.git / blobdiff