BUG-2794 : refactored code to use BitArray
[bgpcep.git] / pcep / impl / src / main / java / org / opendaylight / protocol / pcep / impl / subobject / RROIpv4PrefixSubobjectParser.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 static org.opendaylight.protocol.util.ByteBufWriteUtil.writeIpv4Prefix;
11
12 import com.google.common.base.Preconditions;
13 import io.netty.buffer.ByteBuf;
14 import io.netty.buffer.Unpooled;
15 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
16 import org.opendaylight.protocol.pcep.spi.RROSubobjectParser;
17 import org.opendaylight.protocol.pcep.spi.RROSubobjectSerializer;
18 import org.opendaylight.protocol.pcep.spi.RROSubobjectUtil;
19 import org.opendaylight.protocol.util.BitArray;
20 import org.opendaylight.protocol.util.ByteArray;
21 import org.opendaylight.protocol.util.Ipv4Util;
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.reported.route.object.rro.Subobject;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.reported.route.object.rro.SubobjectBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.IpPrefixSubobject;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.record.route.subobjects.subobject.type.IpPrefixCase;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.record.route.subobjects.subobject.type.IpPrefixCaseBuilder;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.record.route.subobjects.subobject.type.ip.prefix._case.IpPrefixBuilder;
29
30 /**
31  * Parser for {@link IpPrefixCase}
32  */
33 public class RROIpv4PrefixSubobjectParser implements RROSubobjectParser, RROSubobjectSerializer {
34
35     public static final int TYPE = 1;
36
37     private static final int PREFIX_F_LENGTH = 1;
38     private static final int FLAGS_SIZE = 8;
39
40     private static final int PREFIX4_F_OFFSET = 0 + Ipv4Util.IP4_LENGTH;
41
42     private static final int CONTENT4_LENGTH = Ipv4Util.IP4_LENGTH + PREFIX_F_LENGTH + FLAGS_SIZE / Byte.SIZE;
43
44     private static final int LPA_F_OFFSET = 7;
45     private static final int LPIU_F_OFFSET = 6;
46
47     @Override
48     public Subobject parseSubobject(final ByteBuf buffer) throws PCEPDeserializerException {
49         Preconditions.checkArgument(buffer != null && buffer.isReadable(), "Array of bytes is mandatory. Can't be null or empty.");
50         if (buffer.readableBytes() != CONTENT4_LENGTH) {
51             throw new PCEPDeserializerException("Wrong length of array of bytes. Passed: " + buffer.readableBytes() + ";");
52         }
53         final SubobjectBuilder builder = new SubobjectBuilder();
54         final int length = buffer.getUnsignedByte(PREFIX4_F_OFFSET);
55         final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.rsvp.rev130820.record.route.subobjects.subobject.type.ip.prefix._case.IpPrefix prefix = new IpPrefixBuilder().setIpPrefix(
56                 new IpPrefix(Ipv4Util.prefixForBytes(ByteArray.readBytes(buffer, Ipv4Util.IP4_LENGTH), length))).build();
57         buffer.skipBytes(PREFIX_F_LENGTH);
58         final BitArray flags = BitArray.valueOf(buffer, FLAGS_SIZE);
59         builder.setProtectionAvailable(flags.get(LPA_F_OFFSET));
60         builder.setProtectionInUse(flags.get(LPIU_F_OFFSET));
61         builder.setSubobjectType(new IpPrefixCaseBuilder().setIpPrefix(prefix).build());
62         return builder.build();
63     }
64
65     @Override
66     public void serializeSubobject(final Subobject subobject, final ByteBuf buffer) {
67         Preconditions.checkArgument(subobject.getSubobjectType() instanceof IpPrefixCase, "Unknown subobject instance. Passed %s. Needed IpPrefixCase.", subobject.getSubobjectType().getClass());
68         final IpPrefixSubobject specObj = ((IpPrefixCase) subobject.getSubobjectType()).getIpPrefix();
69         final IpPrefix prefix = specObj.getIpPrefix();
70         Preconditions.checkArgument(prefix.getIpv4Prefix() != null || prefix.getIpv6Prefix() != null, "Unknown AbstractPrefix instance. Passed %s.", prefix.getClass());
71         if (prefix.getIpv6Prefix() != null) {
72             new RROIpv6PrefixSubobjectParser().serializeSubobject(subobject, buffer);
73         } else {
74             final BitArray flags = new BitArray(FLAGS_SIZE);
75             flags.set(LPA_F_OFFSET, subobject.isProtectionAvailable());
76             flags.set(LPIU_F_OFFSET, subobject.isProtectionInUse());
77             final ByteBuf body = Unpooled.buffer(CONTENT4_LENGTH);
78             Preconditions.checkArgument(prefix.getIpv4Prefix() != null, "Ipv4Prefix is mandatory.");
79             writeIpv4Prefix(prefix.getIpv4Prefix(), body);
80             flags.toByteBuf(body);
81             RROSubobjectUtil.formatSubobject(TYPE, body, buffer);
82         }
83     }
84 }