Create common parent for extensions families

[bgpcep.git] / bgp / extensions / flowspec / src / main / java / org / opendaylight / protocol / bgp / flowspec / FSSourcePortHandler.java

diff --git a/bgp/extensions/flowspec/src/main/java/org/opendaylight/protocol/bgp/flowspec/FSSourcePortHandler.java b/bgp/extensions/flowspec/src/main/java/org/opendaylight/protocol/bgp/flowspec/FSSourcePortHandler.java
new file mode 100644 (file)
index 0000000..8c4b703
--- /dev/null
+++ b/bgp/extensions/flowspec/src/main/java/org/opendaylight/protocol/bgp/flowspec/FSSourcePortHandler.java
@@ -0,0 +1,61 @@
+/*
+ * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
+package org.opendaylight.protocol.bgp.flowspec;
+
+import static java.util.Objects.requireNonNull;
+
+import com.google.common.base.Preconditions;
+import io.netty.buffer.ByteBuf;
+import java.util.ArrayList;
+import java.util.List;
+import org.opendaylight.protocol.bgp.flowspec.handlers.AbstractOperandParser;
+import org.opendaylight.protocol.bgp.flowspec.handlers.FlowspecTypeParser;
+import org.opendaylight.protocol.bgp.flowspec.handlers.FlowspecTypeSerializer;
+import org.opendaylight.protocol.bgp.flowspec.handlers.NumericOneByteOperandParser;
+import org.opendaylight.protocol.bgp.flowspec.handlers.NumericTwoByteOperandParser;
+import org.opendaylight.protocol.util.ByteArray;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.flowspec.rev180329.NumericOperand;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.flowspec.rev180329.flowspec.destination.flowspec.FlowspecType;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.flowspec.rev180329.flowspec.destination.flowspec.flowspec.type.SourcePortCase;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.flowspec.rev180329.flowspec.destination.flowspec.flowspec.type.SourcePortCaseBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.flowspec.rev180329.flowspec.destination.flowspec.flowspec.type.source.port._case.SourcePorts;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.flowspec.rev180329.flowspec.destination.flowspec.flowspec.type.source.port._case.SourcePortsBuilder;
+
+public final class FSSourcePortHandler implements FlowspecTypeParser, FlowspecTypeSerializer {
+    public static final int SOURCE_PORT_VALUE = 6;
+
+    @Override
+    public void serializeType(FlowspecType fsType, ByteBuf output) {
+        Preconditions.checkArgument(fsType instanceof SourcePortCase, "SourcePortCase class is mandatory!");
+        output.writeByte(SOURCE_PORT_VALUE);
+        NumericTwoByteOperandParser.INSTANCE.serialize(((SourcePortCase) fsType).getSourcePorts(), output);
+    }
+
+    @Override
+    public FlowspecType parseType(ByteBuf buffer) {
+        requireNonNull(buffer, "input buffer is null, missing data to parse.");
+        return new SourcePortCaseBuilder().setSourcePorts(parseSourcePort(buffer)).build();
+    }
+
+    private static List<SourcePorts> parseSourcePort(final ByteBuf nlri) {
+        final List<SourcePorts> ports = new ArrayList<>();
+        boolean end = false;
+        // we can do this as all fields will be rewritten in the cycle
+        final SourcePortsBuilder builder = new SourcePortsBuilder();
+        while (!end) {
+            final byte b = nlri.readByte();
+            final NumericOperand op = NumericOneByteOperandParser.INSTANCE.parse(b);
+            builder.setOp(op);
+            final short length = AbstractOperandParser.parseLength(b);
+            builder.setValue(ByteArray.bytesToInt(ByteArray.readBytes(nlri, length)));
+            end = op.isEndOfList();
+            ports.add(builder.build());
+        }
+        return ports;
+    }
+}