BUG-612 : switched ERO to ByteBuf.
[bgpcep.git] / pcep / impl / src / main / java / org / opendaylight / protocol / pcep / impl / subobject / EROLabelSubobjectParser.java
index 175439f260bb4a86a39f10dfb31a8cabfeebefcb..a817874bf220fa7c0dbb0dba48b2af6a5e2ad37a 100644 (file)
@@ -7,83 +7,79 @@
  */
 package org.opendaylight.protocol.pcep.impl.subobject;
 
-import java.util.Arrays;
+import io.netty.buffer.ByteBuf;
+
 import java.util.BitSet;
 
-import org.opendaylight.protocol.pcep.PCEPDeserializerException;
-import org.opendaylight.protocol.pcep.impl.Util.BiParsersMap;
-import org.opendaylight.protocol.pcep.subobject.EROLabelSubobject;
+import org.opendaylight.protocol.pcep.impl.object.EROSubobjectUtil;
+import org.opendaylight.protocol.pcep.spi.EROSubobjectParser;
+import org.opendaylight.protocol.pcep.spi.EROSubobjectSerializer;
+import org.opendaylight.protocol.pcep.spi.LabelRegistry;
+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.Subobject;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.ero.SubobjectBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.basic.explicit.route.subobjects.subobject.type.LabelCase;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.basic.explicit.route.subobjects.subobject.type.LabelCaseBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.basic.explicit.route.subobjects.subobject.type.label._case.Label;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.basic.explicit.route.subobjects.subobject.type.label._case.LabelBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.label.subobject.LabelType;
 
-public class EROLabelSubobjectParser {
+import com.google.common.base.Preconditions;
+import com.google.common.primitives.UnsignedBytes;
 
-       public static final int RES_F_LENGTH = 1;
+public class EROLabelSubobjectParser implements EROSubobjectParser, EROSubobjectSerializer {
 
-       public static final int C_TYPE_F_LENGTH = 1;
+       public static final int TYPE = 3;
 
-       public static final int RES_F_OFFSET = 0;
+       private static final int RES_F_LENGTH = 1;
 
-       public static final int C_TYPE_F_OFFSET = RES_F_OFFSET + RES_F_LENGTH;
+       private static final int C_TYPE_F_LENGTH = 1;
 
-       public static final int HEADER_LENGTH = C_TYPE_F_OFFSET + C_TYPE_F_LENGTH;
+       private static final int RES_F_OFFSET = 0;
 
-       public static final int U_FLAG_OFFSET = 0;
+       private static final int C_TYPE_F_OFFSET = RES_F_OFFSET + RES_F_LENGTH;
 
-       private static class MapOfParsers extends BiParsersMap<Class<? extends EROLabelSubobject>, Integer, EROLabelParser> {
-               private final static MapOfParsers instance = new MapOfParsers();
+       private static final int HEADER_LENGTH = C_TYPE_F_OFFSET + C_TYPE_F_LENGTH;
 
-               private MapOfParsers() {
-                       this.fillInMap();
-               }
+       private static final int U_FLAG_OFFSET = 0;
 
-               private void fillInMap() {
+       private final LabelRegistry registry;
 
-               }
-
-               public static MapOfParsers getInstance() {
-                       return instance;
-               }
+       public EROLabelSubobjectParser(final LabelRegistry labelReg) {
+               this.registry = Preconditions.checkNotNull(labelReg);
        }
 
-       public static EROLabelSubobject parse(final byte[] soContentsBytes, final boolean loose) throws PCEPDeserializerException {
-               if (soContentsBytes == null || soContentsBytes.length == 0)
-                       throw new IllegalArgumentException("Array of bytes is mandatory. Can't be null or empty.");
-               if (soContentsBytes.length < HEADER_LENGTH)
-                       throw new PCEPDeserializerException("Wrong length of array of bytes. Passed: " + soContentsBytes.length + "; Expected: >"
+       @Override
+       public Subobject parseSubobject(final ByteBuf buffer, final boolean loose) throws PCEPDeserializerException {
+               Preconditions.checkArgument(buffer != null && buffer.isReadable(), "Array of bytes is mandatory. Can't be null or empty.");
+               if (buffer.readableBytes() < HEADER_LENGTH) {
+                       throw new PCEPDeserializerException("Wrong length of array of bytes. Passed: " + buffer.readableBytes() + "; Expected: >"
                                        + HEADER_LENGTH + ".");
-
-               final BitSet reserved = ByteArray.bytesToBitSet(Arrays.copyOfRange(soContentsBytes, RES_F_OFFSET, RES_F_LENGTH));
-
-               final int c_type = soContentsBytes[C_TYPE_F_OFFSET] & 0xFF;
-
-               final EROLabelParser parser = MapOfParsers.getInstance().getValueFromKeyValue(c_type);
-
-               if (parser == null) {
-                       throw new PCEPDeserializerException("Unknown C-TYPE for ero label subobject. Passed: " + c_type);
                }
+               final BitSet reserved = ByteArray.bytesToBitSet(ByteArray.readBytes(buffer, RES_F_LENGTH));
+               final short cType = (short) UnsignedBytes.toInt(buffer.readByte());
 
-               return parser.parse(ByteArray.cutBytes(soContentsBytes, HEADER_LENGTH), reserved.get(U_FLAG_OFFSET), loose);
+               //FIXME: switch to ByteBuf
+               final LabelType labelType = this.registry.parseLabel(cType, ByteArray.readAllBytes(buffer));
+               if (labelType == null) {
+                       throw new PCEPDeserializerException("Unknown C-TYPE for ero label subobject. Passed: " + cType);
+               }
+               final LabelBuilder builder = new LabelBuilder();
+               builder.setUniDirectional(reserved.get(U_FLAG_OFFSET));
+               builder.setLabelType(labelType);
+               return new SubobjectBuilder().setLoose(loose).setSubobjectType(new LabelCaseBuilder().setLabel(builder.build()).build()).build();
        }
 
-       public static byte[] put(final EROLabelSubobject objToSerialize) {
-               final Integer c_type = MapOfParsers.getInstance().getKeyValueFromKey(objToSerialize.getClass());
-               final EROLabelParser parser = MapOfParsers.getInstance().getValueFromKeyValue(c_type);
-
-               if (c_type == null || parser == null)
-                       throw new IllegalArgumentException("Unknown EROLabelSubobject instance. Passed " + objToSerialize.getClass());
-
-               final byte[] labelbytes = parser.put(objToSerialize);
-
-               final byte[] retBytes = new byte[labelbytes.length + HEADER_LENGTH];
-
-               System.arraycopy(labelbytes, 0, retBytes, HEADER_LENGTH, labelbytes.length);
-
-               final BitSet reserved = new BitSet();
-               reserved.set(U_FLAG_OFFSET, objToSerialize.isUpStream());
-               System.arraycopy(ByteArray.bitSetToBytes(reserved, RES_F_LENGTH), 0, retBytes, RES_F_OFFSET, RES_F_LENGTH);
-
-               retBytes[C_TYPE_F_OFFSET] = (byte) c_type.intValue();
-
-               return retBytes;
+       @Override
+       public byte[] serializeSubobject(final Subobject subobject) {
+               Preconditions.checkNotNull(subobject.getSubobjectType(), "Subobject type cannot be empty.");
+               final Label label = ((LabelCase) subobject.getSubobjectType()).getLabel();
+               final byte[] labelbytes = this.registry.serializeLabel(label.isUniDirectional(), false, label.getLabelType());
+               if (labelbytes == null) {
+                       throw new IllegalArgumentException("Unknown EROLabelSubobject instance. Passed "
+                                       + label.getLabelType().getImplementedInterface());
+               }
+               return EROSubobjectUtil.formatSubobject(TYPE, subobject.isLoose(), labelbytes);
        }
 }