/**
* Copyright (c) 2014 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.openflowplugin.testcommon;
import static org.opendaylight.openflowjava.util.ByteBufUtils.macAddressToString;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.atomic.AtomicIntegerFieldUpdater;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev100924.MacAddress;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.DropActionCaseBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.drop.action._case.DropAction;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.action.drop.action._case.DropActionBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.list.Action;
import org.opendaylight.yang.gen.v1.urn.opendaylight.action.types.rev131112.action.list.ActionBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.Instructions;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.InstructionsBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.Match;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.flow.MatchBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.instruction.instruction.ApplyActionsCaseBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.instruction.instruction.apply.actions._case.ApplyActionsBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.instruction.list.Instruction;
import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.types.rev131026.instruction.list.InstructionBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeConnectorRef;
import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
import org.opendaylight.yang.gen.v1.urn.opendaylight.model.match.types.rev131026.ethernet.match.fields.EthernetSourceBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.model.match.types.rev131026.match.EthernetMatchBuilder;
import org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service.rev130709.PacketProcessingListener;
import org.opendaylight.yang.gen.v1.urn.opendaylight.packet.service.rev130709.PacketReceived;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
abstract class AbstractDropTest implements PacketProcessingListener, AutoCloseable {
private static final Logger LOG = LoggerFactory.getLogger(AbstractDropTest.class);
protected static final Integer PRIORITY = 4;
protected static final Long BUFFER_ID = 0L;
protected static final Integer HARD_TIMEOUT = 300;
protected static final Integer IDLE_TIMEOUT = 240;
protected static final short TABLE_ID = 0;
static final long STARTUP_LOOP_TICK = 500L;
static final int STARTUP_LOOP_MAX_RETRIES = 8;
static private final int PROCESSING_THREAD_NUMBER = 8;
private final ExecutorService[] executorServiceLot;
private static final AtomicIntegerFieldUpdater SENT_UPDATER = AtomicIntegerFieldUpdater.newUpdater(AbstractDropTest.class, "sent");
private volatile int sent;
private static final AtomicIntegerFieldUpdater RCVD_UPDATER = AtomicIntegerFieldUpdater.newUpdater(AbstractDropTest.class, "rcvd");
private volatile int rcvd;
private static final AtomicIntegerFieldUpdater EXCS_UPDATER = AtomicIntegerFieldUpdater.newUpdater(AbstractDropTest.class, "excs");
private volatile int excs;
public AbstractDropTest() {
executorServiceLot = new ExecutorService[PROCESSING_THREAD_NUMBER];
for (int i = 0; i < PROCESSING_THREAD_NUMBER; i++) {
executorServiceLot[i] = Executors.newFixedThreadPool(1);
}
}
public final DropTestStats getStats() {
return new DropTestStats(this.sent, this.rcvd, this.excs);
}
public final void clearStats() {
this.sent = 0;
this.rcvd = 0;
this.excs = 0;
}
@Override
public final void onPacketReceived(final PacketReceived notification) {
LOG.debug("onPacketReceived - Entering - {}", notification);
RCVD_UPDATER.incrementAndGet(this);
NodeKey nodeKey = notification.getIngress().getValue().firstKeyOf(Node.class, NodeKey.class);
int nodeIdHash = nodeKey.getId().getValue().hashCode();
executorServiceLot[nodeIdHash % PROCESSING_THREAD_NUMBER].submit(new Runnable() {
@Override
public void run() {
processPacket(notification);
}
});
LOG.debug("onPacketReceived - Leaving", notification);
}
private void processPacket(final PacketReceived notification) {
try {
// TODO Auto-generated method stub
final byte[] rawPacket = notification.getPayload();
final byte[] srcMac = Arrays.copyOfRange(rawPacket, 6, 12);
final MatchBuilder match = new MatchBuilder();
final EthernetMatchBuilder ethernetMatch = new EthernetMatchBuilder();
final EthernetSourceBuilder ethSourceBuilder = new EthernetSourceBuilder();
//TODO: use HEX, use binary form
//Hex.decodeHex("000000000001".toCharArray());
ethSourceBuilder.setAddress(new MacAddress(macAddressToString(srcMac)));
ethernetMatch.setEthernetSource(ethSourceBuilder.build());
match.setEthernetMatch(ethernetMatch.build());
final DropActionBuilder dab = new DropActionBuilder();
final DropAction dropAction = dab.build();
final ActionBuilder ab = new ActionBuilder();
ab.setOrder(0);
ab.setAction(new DropActionCaseBuilder().setDropAction(dropAction).build());
// Add our drop action to a list
final List actionList = Collections.singletonList(ab.build());
// Create an Apply Action
final ApplyActionsBuilder aab = new ApplyActionsBuilder();
aab.setAction(actionList);
// Wrap our Apply Action in an Instruction
final InstructionBuilder ib = new InstructionBuilder();
ib.setInstruction(new ApplyActionsCaseBuilder().setApplyActions(aab.build()).build()).setOrder(0);
// Put our Instruction in a list of Instructions
final InstructionsBuilder isb = new InstructionsBuilder();
final List instructions = Collections.singletonList(ib.build());
isb.setInstruction(instructions);
// Get the Ingress nodeConnectorRef
final NodeConnectorRef ncr = notification.getIngress();
// Get the instance identifier for the nodeConnectorRef
final InstanceIdentifier> ncri = ncr.getValue();
// Get the instanceID for the Node in the tree above us
final NodeKey node = ncri.firstKeyOf(Node.class, NodeKey.class);
processPacket(node, match.build(), isb.build());
SENT_UPDATER.incrementAndGet(this);
} catch (final Exception e) {
// TODO Auto-generated catch block
LOG.warn("Failed to process packet: {}", e.getMessage());
LOG.debug("Failed to process packet.. ", e);
EXCS_UPDATER.incrementAndGet(this);
}
}
protected abstract void processPacket(NodeKey node, Match match, Instructions instructions);
@Override
public void close() {
for (ExecutorService service : executorServiceLot) {
service.shutdown();
}
}
}