*/
package org.opendaylight.protocol.pcep.impl.object;
-import org.opendaylight.protocol.pcep.PCEPDeserializerException;
-import org.opendaylight.protocol.pcep.PCEPDocumentedException;
-import org.opendaylight.protocol.pcep.impl.Util;
+import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeUnsignedByte;
+
+import com.google.common.base.Preconditions;
+import com.google.common.primitives.UnsignedBytes;
+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.TlvHandlerRegistry;
-import org.opendaylight.protocol.util.ByteArray;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.NotificationObject;
+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.OverloadDurationTlv;
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.Tlvs;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.notification.object.TlvsBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.notification.object.tlvs.OverloadDurationBuilder;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcntf.message.pcntf.message.notifications.NotificationsBuilder;
+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 NotificationObject}
+ * Parser for {@link CNotification}
*/
-public class PCEPNotificationObjectParser extends AbstractObjectWithTlvsParser<NotificationsBuilder> {
-
- public static final int CLASS = 12;
-
- public static final int TYPE = 1;
-
- /*
- * 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;
-
- public PCEPNotificationObjectParser(final TlvHandlerRegistry tlvReg) {
- super(tlvReg);
- }
-
- @Override
- public NotificationObject parseObject(final ObjectHeader header, final byte[] bytes) throws PCEPDeserializerException,
- PCEPDocumentedException {
- if (bytes == null || bytes.length == 0) {
- throw new IllegalArgumentException("Array of bytes is mandatory. Can't be null or empty.");
- }
-
- final NotificationsBuilder builder = new NotificationsBuilder();
-
- parseTlvs(builder, ByteArray.cutBytes(bytes, TLVS_OFFSET));
-
- builder.setIgnore(header.isIgnore());
- builder.setProcessingRule(header.isProcessingRule());
-
- builder.setType((short) (bytes[NT_F_OFFSET] & 0xFF));
- builder.setValue((short) (bytes[NV_F_OFFSET] & 0xFF));
-
- return builder.build();
- }
-
- @Override
- public void addTlv(final NotificationsBuilder builder, final Tlv tlv) {
- if (tlv instanceof OverloadDurationTlv && builder.getType() == 2 && builder.getValue() == 1) {
- builder.setTlvs(new TlvsBuilder().setOverloadDuration(
- new OverloadDurationBuilder().setDuration(((OverloadDurationTlv) tlv).getDuration()).build()).build());
- }
- }
-
- @Override
- public byte[] serializeObject(final Object object) {
- if (!(object instanceof NotificationObject)) {
- throw new IllegalArgumentException("Wrong instance of PCEPObject. Passed " + object.getClass() + ". Needed NotificationObject.");
- }
-
- final NotificationObject notObj = (NotificationObject) object;
-
- final byte[] tlvs = serializeTlvs(notObj.getTlvs());
- int tlvsLength = 0;
- if (tlvs != null) {
- tlvsLength = tlvs.length;
- }
- final byte[] retBytes = new byte[TLVS_OFFSET + tlvsLength + Util.getPadding(TLVS_OFFSET + tlvs.length, PADDED_TO)];
-
- if (tlvs != null) {
- ByteArray.copyWhole(tlvs, retBytes, TLVS_OFFSET);
- }
-
- retBytes[NT_F_OFFSET] = ByteArray.shortToBytes(notObj.getType())[1];
- retBytes[NV_F_OFFSET] = ByteArray.shortToBytes(notObj.getValue())[1];
-
- return retBytes;
- }
-
- public byte[] serializeTlvs(final Tlvs tlvs) {
- if (tlvs.getOverloadDuration() != null) {
- // FIXME : add
- // return serializeTlv(new NoPathVectorBuilder().setFlags(tlvs.getNoPathVector()).build());
- }
- return null;
- }
-
- @Override
- public int getObjectType() {
- return TYPE;
- }
-
- @Override
- public int getObjectClass() {
- return CLASS;
- }
+public class PCEPNotificationObjectParser extends AbstractObjectWithTlvsParser<CNotificationBuilder> {
+
+ public static final int CLASS = 12;
+
+ public static final int TYPE = 1;
+
+ /*
+ * offsets of fields
+ */
+ private static final int NT_F_OFFSET = 2;
+
+ public PCEPNotificationObjectParser(final TlvRegistry tlvReg, final VendorInformationTlvRegistry viTlvReg) {
+ super(tlvReg, viTlvReg);
+ }
+
+ @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.readerIndex(bytes.readerIndex() + NT_F_OFFSET);
+ builder.setType((short) UnsignedBytes.toInt(bytes.readByte()));
+ builder.setValue((short) UnsignedBytes.toInt(bytes.readByte()));
+ parseTlvs(builder, bytes.slice());
+ return builder.build();
+ }
+
+ @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());
+ }
+ }
+
+ @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);
+ }
+
+ 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());
+ }
+ }
}
Code Review / bgpcep.git / blobdiff