Getting rid of readerIndex handling.
[bgpcep.git] / pcep / impl / src / main / java / org / opendaylight / protocol / pcep / impl / object / PCEPNotificationObjectParser.java
index 7f277b7130ca1db7cd46de6b6a1b40c056e6f8e3..bf095a438e83a1e338e755f5871a32b66409892c 100644 (file)
@@ -7,62 +7,93 @@
  */
 package org.opendaylight.protocol.pcep.impl.object;
 
-import org.opendaylight.protocol.pcep.PCEPDeserializerException;
-import org.opendaylight.protocol.pcep.PCEPObject;
-import org.opendaylight.protocol.pcep.impl.PCEPObjectParser;
-import org.opendaylight.protocol.pcep.impl.PCEPTlvParser;
-import org.opendaylight.protocol.pcep.object.PCEPNotificationObject;
-import org.opendaylight.protocol.util.ByteArray;
+import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeUnsignedByte;
+
+import com.google.common.base.Preconditions;
+import io.netty.buffer.ByteBuf;
+import io.netty.buffer.Unpooled;
+import java.util.List;
+import org.opendaylight.protocol.pcep.spi.AbstractObjectWithTlvsParser;
+import org.opendaylight.protocol.pcep.spi.ObjectUtil;
+import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
+import org.opendaylight.protocol.pcep.spi.TlvRegistry;
+import org.opendaylight.protocol.pcep.spi.VendorInformationTlvRegistry;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.ObjectHeader;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Tlv;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.notification.object.CNotification;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.notification.object.CNotificationBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.notification.object.c.notification.Tlvs;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.notification.object.c.notification.TlvsBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.overload.duration.tlv.OverloadDuration;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.vendor.information.tlvs.VendorInformationTlv;
 
 /**
- * Parser for {@link org.opendaylight.protocol.pcep.object.PCEPNotificationObject
- * PCEPNotificationObject}
+ * Parser for {@link CNotification}
  */
-public class PCEPNotificationObjectParser implements PCEPObjectParser {
-
-       /*
-        * lengths of fields
-        */
-       public static final int FLAGS_F_LENGTH = 1;
-       public static final int NT_F_LENGTH = 1;
-       public static final int NV_F_LENGTH = 1;
-
-       /*
-        * offsets of fields
-        */
-       public static final int FLAGS_F_OFFSET = 1; //added reserved filed of size 1
-       public static final int NT_F_OFFSET = FLAGS_F_OFFSET + FLAGS_F_LENGTH;
-       public static final int NV_F_OFFSET = NT_F_OFFSET + NT_F_LENGTH;
-       public static final int TLVS_OFFSET = NV_F_OFFSET + NV_F_LENGTH;
-
-       @Override
-       public PCEPObject parse(byte[] bytes, boolean processed, boolean ignored) throws PCEPDeserializerException {
-               if (bytes == null || bytes.length == 0)
-                       throw new IllegalArgumentException("Array of bytes is mandatory. Can't be null or empty.");
+public class PCEPNotificationObjectParser extends AbstractObjectWithTlvsParser<CNotificationBuilder> {
 
-               if (bytes.length < TLVS_OFFSET)
-                       throw new PCEPDeserializerException("Wrong length of array of bytes. Passed: " + bytes.length + "; Expected: >=" + TLVS_OFFSET + ".");
+    public static final int CLASS = 12;
 
-               return new PCEPNotificationObject((short) (bytes[NT_F_OFFSET] & 0xFF), (short) (bytes[NV_F_OFFSET] & 0xFF), PCEPTlvParser.parse(ByteArray.cutBytes(
-                               bytes, TLVS_OFFSET)));
-       }
+    public static final int TYPE = 1;
 
-       @Override
-       public byte[] put(PCEPObject obj) {
-               if (!(obj instanceof PCEPNotificationObject))
-                       throw new IllegalArgumentException("Wrong instance of PCEPObject. Passed " + obj.getClass() + ". Needed PCEPNotificationObject.");
+    /*
+     * offsets of fields
+     */
+    private static final int NT_F_OFFSET = 2;
 
-               final PCEPNotificationObject notObj = (PCEPNotificationObject) obj;
+    public PCEPNotificationObjectParser(final TlvRegistry tlvReg, final VendorInformationTlvRegistry viTlvReg) {
+        super(tlvReg, viTlvReg);
+    }
 
-               final byte[] tlvs = PCEPTlvParser.put(notObj.getTlvs());
-               final byte[] retBytes = new byte[TLVS_OFFSET + tlvs.length];
+    @Override
+    public CNotification parseObject(final ObjectHeader header, final ByteBuf bytes) throws PCEPDeserializerException {
+        Preconditions.checkArgument(bytes != null && bytes.isReadable(), "Array of bytes is mandatory. Can't be null or empty.");
+        final CNotificationBuilder builder = new CNotificationBuilder();
+        builder.setIgnore(header.isIgnore());
+        builder.setProcessingRule(header.isProcessingRule());
+        bytes.skipBytes(NT_F_OFFSET);
+        builder.setType(bytes.readUnsignedByte());
+        builder.setValue(bytes.readUnsignedByte());
+        parseTlvs(builder, bytes.slice());
+        return builder.build();
+    }
 
-               ByteArray.copyWhole(tlvs, retBytes, TLVS_OFFSET);
+    @Override
+    public void addTlv(final CNotificationBuilder builder, final Tlv tlv) {
+        if (tlv instanceof OverloadDuration && builder.getType() == 2 && builder.getValue() == 1) {
+            builder.setTlvs(new TlvsBuilder().setOverloadDuration((OverloadDuration) tlv).build());
+        }
+    }
 
-               retBytes[NT_F_OFFSET] = ByteArray.shortToBytes(notObj.getType())[1];
-               retBytes[NV_F_OFFSET] = ByteArray.shortToBytes(notObj.getValue())[1];
+    @Override
+    public void serializeObject(final Object object, final ByteBuf buffer) {
+        Preconditions.checkArgument(object instanceof CNotification, "Wrong instance of PCEPObject. Passed %s. Needed CNotificationObject.", object.getClass());
+        final CNotification notObj = (CNotification) object;
+        final ByteBuf body = Unpooled.buffer();
+        body.writeZero(NT_F_OFFSET);
+        Preconditions.checkArgument(notObj.getType() != null, "Type is mandatory.");
+        writeUnsignedByte(notObj.getType(), body);
+        Preconditions.checkArgument(notObj.getValue() != null, "Value is mandatory.");
+        writeUnsignedByte(notObj.getValue(), body);
+        serializeTlvs(notObj.getTlvs(), body);
+        ObjectUtil.formatSubobject(TYPE, CLASS, object.isProcessingRule(), object.isIgnore(), body, buffer);
+    }
 
-               return retBytes;
-       }
+    public void serializeTlvs(final Tlvs tlvs, final ByteBuf body) {
+        if (tlvs == null) {
+            return;
+        }
+        if (tlvs.getOverloadDuration() != null) {
+            serializeTlv(tlvs.getOverloadDuration(), body);
+        }
+        serializeVendorInformationTlvs(tlvs.getVendorInformationTlv(), body);
+    }
 
+    @Override
+    protected final void addVendorInformationTlvs(final CNotificationBuilder builder, final List<VendorInformationTlv> tlvs) {
+        if (!tlvs.isEmpty()) {
+            builder.setTlvs(new TlvsBuilder(builder.getTlvs()).setVendorInformationTlv(tlvs).build());
+        }
+    }
 }