BUG-612 : switched PCEP XRO subobject serializers to ByteBuf
[bgpcep.git] / pcep / impl / src / main / java / org / opendaylight / protocol / pcep / impl / subobject / XROIpv4PrefixSubobjectParser.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 package org.opendaylight.protocol.pcep.impl.subobject;
9
10 import com.google.common.base.Preconditions;
11 import com.google.common.primitives.UnsignedBytes;
12
13 import io.netty.buffer.ByteBuf;
14 import io.netty.buffer.Unpooled;
15
16 import org.opendaylight.protocol.concepts.Ipv4Util;
17 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
18 import org.opendaylight.protocol.pcep.spi.XROSubobjectParser;
19 import org.opendaylight.protocol.pcep.spi.XROSubobjectSerializer;
20 import org.opendaylight.protocol.pcep.spi.XROSubobjectUtil;
21 import org.opendaylight.protocol.util.ByteArray;
22 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.IpPrefix;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.exclude.route.object.xro.Subobject;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.exclude.route.object.xro.SubobjectBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.ExcludeRouteSubobjects.Attribute;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.IpPrefixSubobject;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.basic.explicit.route.subobjects.subobject.type.IpPrefixCase;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.basic.explicit.route.subobjects.subobject.type.IpPrefixCaseBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.basic.explicit.route.subobjects.subobject.type.ip.prefix._case.IpPrefixBuilder;
30
31 /**
32  * Parser for {@link IpPrefixCase}
33  */
34 public class XROIpv4PrefixSubobjectParser implements XROSubobjectParser, XROSubobjectSerializer {
35
36     public static final int TYPE = 1;
37
38     private static final int PREFIX_F_LENGTH = 1;
39
40     private static final int PREFIX4_F_OFFSET = Ipv4Util.IP4_LENGTH;
41
42     private static final int CONTENT4_LENGTH = PREFIX4_F_OFFSET + PREFIX_F_LENGTH + 1;
43
44     @Override
45     public Subobject parseSubobject(final ByteBuf buffer, final boolean mandatory) throws PCEPDeserializerException {
46         Preconditions.checkArgument(buffer != null && buffer.isReadable(), "Array of bytes is mandatory. Can't be null or empty.");
47         final SubobjectBuilder builder = new SubobjectBuilder();
48         builder.setMandatory(mandatory);
49         if (buffer.readableBytes() != CONTENT4_LENGTH) {
50             throw new PCEPDeserializerException("Wrong length of array of bytes. Passed: " + buffer.readableBytes() + ";");
51         }
52         final int length = UnsignedBytes.toInt(buffer.getByte(PREFIX4_F_OFFSET));
53         IpPrefixBuilder prefix = new IpPrefixBuilder().setIpPrefix(new IpPrefix(Ipv4Util.prefixForBytes(ByteArray.readBytes(buffer,
54                 Ipv4Util.IP4_LENGTH), length)));
55         builder.setSubobjectType(new IpPrefixCaseBuilder().setIpPrefix(prefix.build()).build());
56         buffer.readerIndex(buffer.readerIndex() + PREFIX_F_LENGTH);
57         builder.setAttribute(Attribute.forValue(UnsignedBytes.toInt(buffer.readByte())));
58         return builder.build();
59     }
60
61     @Override
62     public void serializeSubobject(final Subobject subobject, final ByteBuf buffer) {
63         Preconditions.checkArgument(subobject.getSubobjectType() instanceof IpPrefixCase, "Unknown subobject instance. Passed %s. Needed IpPrefixCase.", subobject.getSubobjectType().getClass());
64         final IpPrefixSubobject specObj = ((IpPrefixCase) subobject.getSubobjectType()).getIpPrefix();
65         final IpPrefix prefix = specObj.getIpPrefix();
66         Preconditions.checkArgument(prefix.getIpv4Prefix() != null || prefix.getIpv6Prefix() != null, "Unknown AbstractPrefix instance. Passed %s.", prefix.getClass());
67         if (prefix.getIpv6Prefix() != null) {
68             new XROIpv6PrefixSubobjectParser().serializeSubobject(subobject, buffer);
69         } else {
70             final ByteBuf body = Unpooled.buffer(CONTENT4_LENGTH);
71             body.writeBytes(Ipv4Util.bytesForPrefix(prefix.getIpv4Prefix()));
72             body.writeByte(subobject.getAttribute().getIntValue());
73             XROSubobjectUtil.formatSubobject(TYPE, subobject.isMandatory(), body, buffer);
74         }
75     }
76 }