0c8efaa3bcb819ecb7325c494daf3b66bc956fba
[bgpcep.git] / pcep / ietf-stateful02 / src / main / java / org / opendaylight / protocol / pcep / ietf / stateful02 / Stateful02LspObjectParser.java
1 /*
2  * Copyright (c) 2014 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.stateful02;
9
10 import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeMedium;
11
12 import com.google.common.base.Preconditions;
13 import io.netty.buffer.ByteBuf;
14 import io.netty.buffer.Unpooled;
15 import java.util.BitSet;
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.ByteArray;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.PlspId;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.lsp.db.version.tlv.LspDbVersion;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.lsp.object.Lsp;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.lsp.object.LspBuilder;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.lsp.object.lsp.Tlvs;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.lsp.object.lsp.TlvsBuilder;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.rsvp.error.spec.tlv.RsvpErrorSpec;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.symbolic.path.name.tlv.SymbolicPathName;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.ObjectHeader;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Tlv;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.vendor.information.tlvs.VendorInformationTlv;
35
36 /**
37  * Parser for {@link Lsp}
38  */
39 public class Stateful02LspObjectParser extends AbstractObjectWithTlvsParser<TlvsBuilder> {
40
41     public static final int CLASS = 32;
42
43     public static final int TYPE = 1;
44
45     /*
46      * first 4b are restricted
47      */
48     private static final int DELEGATE_FLAG_OFFSET = 15;
49     private static final int SYNC_FLAG_OFFSET = 14;
50     private static final int OPERATIONAL_FLAG_OFFSET = 13;
51     private static final int REMOVE_FLAG_OFFSET = 12;
52
53     private static final int ONE_B_OFFSET = 12;
54     private static final int TWO_B_OFFSET = 4;
55
56     private static final int FLAGS_SIZE = 2;
57
58     public Stateful02LspObjectParser(final TlvRegistry tlvReg, final VendorInformationTlvRegistry viTlvReg) {
59         super(tlvReg, viTlvReg);
60     }
61
62     @Override
63     public Lsp parseObject(final ObjectHeader header, final ByteBuf bytes) throws PCEPDeserializerException {
64         Preconditions.checkArgument(bytes != null && bytes.isReadable(), "Array of bytes is mandatory. Can't be null or empty.");
65         final LspBuilder builder = new LspBuilder();
66         builder.setIgnore(header.isIgnore());
67         builder.setProcessingRule(header.isProcessingRule());
68         final int[] plspIdRaw = { bytes.readUnsignedByte(), bytes.readUnsignedByte(), bytes.getUnsignedByte(2) };
69         builder.setPlspId(new PlspId((long) ((plspIdRaw[0] << ONE_B_OFFSET) | (plspIdRaw[1] << TWO_B_OFFSET) | (plspIdRaw[2] >> TWO_B_OFFSET))));
70         final BitSet flags = ByteArray.bytesToBitSet(ByteArray.readBytes(bytes, FLAGS_SIZE));
71         builder.setDelegate(flags.get(DELEGATE_FLAG_OFFSET));
72         builder.setSync(flags.get(SYNC_FLAG_OFFSET));
73         builder.setRemove(flags.get(REMOVE_FLAG_OFFSET));
74         builder.setOperational(flags.get(OPERATIONAL_FLAG_OFFSET));
75         final TlvsBuilder b = new TlvsBuilder();
76         parseTlvs(b, bytes.slice());
77         builder.setTlvs(b.build());
78         return builder.build();
79     }
80
81     @Override
82     public void addTlv(final TlvsBuilder builder, final Tlv tlv) {
83         if (tlv instanceof RsvpErrorSpec) {
84             builder.setRsvpErrorSpec((RsvpErrorSpec) tlv);
85         } else if (tlv instanceof SymbolicPathName) {
86             builder.setSymbolicPathName((SymbolicPathName) tlv);
87         } else if (tlv instanceof LspDbVersion) {
88             builder.setLspDbVersion((LspDbVersion) tlv);
89         }
90     }
91
92     @Override
93     public void serializeObject(final Object object, final ByteBuf buffer) {
94         Preconditions.checkArgument(object instanceof Lsp, "Wrong instance of PCEPObject. Passed %s. Needed LspObject.", object.getClass());
95         final Lsp specObj = (Lsp) object;
96         final ByteBuf body = Unpooled.buffer();
97         final PlspId plsp = specObj.getPlspId();
98         Preconditions.checkArgument(plsp != null, "PLSP-ID not present");
99         writeMedium(plsp.getValue().intValue() << TWO_B_OFFSET, body);
100
101         final BitSet flags = new BitSet(2 * Byte.SIZE);
102         if (specObj.isDelegate() != null) {
103             flags.set(DELEGATE_FLAG_OFFSET, specObj.isDelegate());
104         }
105         if (specObj.isRemove() != null) {
106             flags.set(REMOVE_FLAG_OFFSET, specObj.isRemove());
107         }
108         if (specObj.isSync() != null) {
109             flags.set(SYNC_FLAG_OFFSET, specObj.isSync());
110         }
111         if (specObj.isOperational() != null) {
112             flags.set(OPERATIONAL_FLAG_OFFSET, specObj.isOperational());
113         }
114         body.writeByte(ByteArray.bitSetToBytes(flags, FLAGS_SIZE)[1]);
115         serializeTlvs(specObj.getTlvs(), body);
116         ObjectUtil.formatSubobject(TYPE, CLASS, object.isProcessingRule(), object.isIgnore(), body, buffer);
117     }
118
119     public void serializeTlvs(final Tlvs tlvs, final ByteBuf body) {
120         if (tlvs == null) {
121             return;
122         }
123         if (tlvs.getRsvpErrorSpec() != null) {
124             serializeTlv(tlvs.getRsvpErrorSpec(), body);
125         }
126         if (tlvs.getSymbolicPathName() != null) {
127             serializeTlv(tlvs.getSymbolicPathName(), body);
128         }
129         if (tlvs.getLspDbVersion() != null) {
130             serializeTlv(tlvs.getLspDbVersion(), body);
131         }
132         serializeVendorInformationTlvs(tlvs.getVendorInformationTlv(), body);
133     }
134
135     @Override
136     protected final void addVendorInformationTlvs(final TlvsBuilder builder, final List<VendorInformationTlv> tlvs) {
137         if (!tlvs.isEmpty()) {
138             builder.setVendorInformationTlv(tlvs);
139         }
140     }
141 }