BUG-2794 : refactored code to use BitArray
[bgpcep.git] / pcep / ietf-stateful07 / src / main / java / org / opendaylight / protocol / pcep / ietf / stateful07 / Stateful07LspObjectParser.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.ietf.stateful07;
9
10 import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeMedium;
11
12 import com.google.common.base.Preconditions;
13 import com.google.common.primitives.UnsignedBytes;
14 import io.netty.buffer.ByteBuf;
15 import io.netty.buffer.Unpooled;
16 import java.util.List;
17 import org.opendaylight.protocol.pcep.spi.AbstractObjectWithTlvsParser;
18 import org.opendaylight.protocol.pcep.spi.ObjectUtil;
19 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
20 import org.opendaylight.protocol.pcep.spi.TlvRegistry;
21 import org.opendaylight.protocol.pcep.spi.VendorInformationTlvRegistry;
22 import org.opendaylight.protocol.util.BitArray;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.OperationalStatus;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.PlspId;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.error.code.tlv.LspErrorCode;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.identifiers.tlv.LspIdentifiers;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.Lsp;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.LspBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.Tlvs;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.TlvsBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.rsvp.error.spec.tlv.RsvpErrorSpec;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.symbolic.path.name.tlv.SymbolicPathName;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.ObjectHeader;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Tlv;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.vendor.information.tlvs.VendorInformationTlv;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.vs.tlv.VsTlv;
38
39 /**
40  * Parser for {@link Lsp}
41  */
42 public class Stateful07LspObjectParser extends AbstractObjectWithTlvsParser<TlvsBuilder> {
43
44     public static final int CLASS = 32;
45
46     public static final int TYPE = 1;
47
48     /*
49      * 12b extended to 16b so first 4b are restricted (belongs to LSP ID)
50      */
51     protected static final int DELEGATE = 11;
52     protected static final int SYNC = 10;
53     protected static final int REMOVE = 9;
54     protected static final int ADMINISTRATIVE = 8;
55     protected static final int OPERATIONAL = 5;
56
57     protected static final int FOUR_BITS_SHIFT = 4;
58     protected static final int FLAGS_SIZE = 12;
59
60     public Stateful07LspObjectParser(final TlvRegistry tlvReg, final VendorInformationTlvRegistry viTlvReg) {
61         super(tlvReg, viTlvReg);
62     }
63
64     @Override
65     public Lsp parseObject(final ObjectHeader header, final ByteBuf bytes) throws PCEPDeserializerException {
66         Preconditions.checkArgument(bytes != null && bytes.isReadable(), "Array of bytes is mandatory. Can't be null or empty.");
67         final LspBuilder builder = new LspBuilder();
68         builder.setIgnore(header.isIgnore());
69         builder.setProcessingRule(header.isProcessingRule());
70         final int[] plspIdRaw = new int[] { bytes.readUnsignedByte(), bytes.readUnsignedByte(), bytes.getUnsignedByte(2), };
71         builder.setPlspId(new PlspId((long) ((plspIdRaw[0] << FLAGS_SIZE) | (plspIdRaw[1] << FOUR_BITS_SHIFT) | (plspIdRaw[2] >> FOUR_BITS_SHIFT))));
72         final BitArray flags = BitArray.valueOf(bytes, FLAGS_SIZE);
73         builder.setDelegate(flags.get(DELEGATE));
74         builder.setSync(flags.get(SYNC));
75         builder.setRemove(flags.get(REMOVE));
76         builder.setAdministrative(flags.get(ADMINISTRATIVE));
77         short s = 0;
78         s |= flags.get(OPERATIONAL + 2) ? 1 : 0;
79         s |= (flags.get(OPERATIONAL + 1) ? 1 : 0) << 1;
80         s |= (flags.get(OPERATIONAL) ? 1 : 0) << 2;
81         builder.setOperational(OperationalStatus.forValue(s));
82         final TlvsBuilder b = new TlvsBuilder();
83         parseTlvs(b, bytes.slice());
84         builder.setTlvs(b.build());
85         return builder.build();
86     }
87
88     @Override
89     public void addTlv(final TlvsBuilder builder, final Tlv tlv) {
90         if (tlv instanceof LspErrorCode) {
91             builder.setLspErrorCode((LspErrorCode) tlv);
92         } else if (tlv instanceof LspIdentifiers) {
93             builder.setLspIdentifiers((LspIdentifiers) tlv);
94         } else if (tlv instanceof RsvpErrorSpec) {
95             builder.setRsvpErrorSpec((RsvpErrorSpec) tlv);
96         } else if (tlv instanceof SymbolicPathName) {
97             builder.setSymbolicPathName((SymbolicPathName) tlv);
98         } else if (tlv instanceof VsTlv) {
99             builder.setVsTlv((VsTlv) tlv);
100         }
101     }
102
103     @Override
104     public void serializeObject(final Object object, final ByteBuf buffer) {
105         Preconditions.checkArgument(object instanceof Lsp, "Wrong instance of PCEPObject. Passed %s . Needed LspObject.", object.getClass());
106         final Lsp specObj = (Lsp) object;
107         final ByteBuf body = Unpooled.buffer();
108         Preconditions.checkArgument(specObj.getPlspId() != null, "PLSP-ID not present");
109         writeMedium(specObj.getPlspId().getValue().intValue() << FOUR_BITS_SHIFT, body);
110         final BitArray flags = new BitArray(FLAGS_SIZE);
111         flags.set(DELEGATE, specObj.isDelegate());
112         flags.set(REMOVE, specObj.isRemove());
113         flags.set(SYNC, specObj.isSync());
114         flags.set(ADMINISTRATIVE, specObj.isAdministrative());
115         byte op = 0;
116         if (specObj.getOperational() != null) {
117             op = UnsignedBytes.checkedCast(specObj.getOperational().getIntValue());
118             op = (byte) (op << FOUR_BITS_SHIFT);
119         }
120         final byte[] res = flags.array();
121         res[res.length -1] = (byte) (res[res.length -1] | op);
122         body.writeByte(res[res.length -1]);
123         serializeTlvs(specObj.getTlvs(), body);
124         ObjectUtil.formatSubobject(TYPE, CLASS, object.isProcessingRule(), object.isIgnore(), body, buffer);
125     }
126
127     public void serializeTlvs(final Tlvs tlvs, final ByteBuf body) {
128         if (tlvs == null) {
129             return;
130         }
131         if (tlvs.getLspErrorCode() != null) {
132             serializeTlv(tlvs.getLspErrorCode(), body);
133         }
134         if (tlvs.getLspIdentifiers() != null) {
135             serializeTlv(tlvs.getLspIdentifiers(), body);
136         }
137         if (tlvs.getRsvpErrorSpec() != null) {
138             serializeTlv(tlvs.getRsvpErrorSpec(), body);
139         }
140         if (tlvs.getSymbolicPathName() != null) {
141             serializeTlv(tlvs.getSymbolicPathName(), body);
142         }
143         if (tlvs.getVsTlv() != null) {
144             serializeTlv(tlvs.getVsTlv(), body);
145         }
146         serializeVendorInformationTlvs(tlvs.getVendorInformationTlv(), body);
147     }
148
149     @Override
150     protected final void addVendorInformationTlvs(final TlvsBuilder builder, final List<VendorInformationTlv> tlvs) {
151         if (!tlvs.isEmpty()) {
152             builder.setVendorInformationTlv(tlvs);
153         }
154     }
155 }