pcep/testtool/src/main/java/org/opendaylight/protocol/pcep/testtool/Main.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.testtool;
   9
  10 import java.net.InetAddress;
  11 import java.net.InetSocketAddress;
  12
  13 import org.opendaylight.protocol.framework.DispatcherImpl;
  14 import org.opendaylight.protocol.framework.ProtocolServer;
  15 import org.opendaylight.protocol.pcep.PCEPConnection;
  16 import org.opendaylight.protocol.pcep.PCEPConnectionFactory;
  17 import org.opendaylight.protocol.pcep.PCEPSessionListener;
  18 import org.opendaylight.protocol.pcep.PCEPSessionListenerFactory;
  19 import org.opendaylight.protocol.pcep.PCEPSessionProposal;
  20 import org.opendaylight.protocol.pcep.PCEPSessionProposalChecker;
  21 import org.opendaylight.protocol.pcep.PCEPSessionProposalCheckerFactory;
  22 import org.opendaylight.protocol.pcep.PCEPSessionProposalFactory;
  23 import org.opendaylight.protocol.pcep.impl.PCEPConnectionImpl;
  24 import org.opendaylight.protocol.pcep.impl.PCEPDispatcherImpl;
  25 import org.opendaylight.protocol.pcep.impl.PCEPMessageFactory;
  26 import org.opendaylight.protocol.pcep.impl.PCEPSessionProposalCheckerFactoryImpl;
  27 import org.opendaylight.protocol.pcep.impl.PCEPSessionProposalFactoryImpl;
  28
  29 public class Main {
  30
  31         public static String usage = "DESCRIPTION:\n" + "\tCreates a server with given parameters. As long as it runs, it accepts connections "
  32                         + "from PCCs.\n" + "USAGE:\n" + "\t-a, --address\n" + "\t\tthe ip address to which is this server bound.\n"
  33                         + "\t\tFormat: x.x.x.x:y where y is port number.\n\n" +
  34
  35                         "\t-d, --deadtimer\n" + "\t\tin seconds, value of the desired deadtimer\n"
  36                         + "\t\tAccording to RFC5440, recommended value for deadtimer is 4 times the value\n"
  37                         + "\t\tof KeepAlive timer. If it's not, a warning is printed.\n"
  38                         + "\t\tIf not set, it's value will be derived from KeepAlive timer value.\n\n" +
  39
  40                         "\t-ka, --keepalive\n" + "\t\tin seconds, value of the desired KeepAlive timer.\n"
  41                         + "\t\tIf not present, KeepAlive timer will be set to recommended value (30s).\n\n" +
  42
  43                         "\t--stateful\n" + "\t\tpassive stateful\n\n" +
  44
  45                         "\t--active\n" + "\t\tactive stateful (implies --stateful)\n\n" +
  46
  47                         "\t--versioned\n" + "\t\tversioned stateful (implies --stateful)\n\n" +
  48
  49                         "\t--instant\n"
  50                         + "\t\tinstantiated stateful, <seconds> cleanup timeout (default value, if not included = 0) (implies --stateful)\n\n" +
  51
  52                         "\t-arm, --autoResponseMessages <path to file>\n"
  53                         + "\t\t <path to file> with groovy script which implements MessageGeneratorService.\n"
  54                         + "\t\t Messages are used as auto response for every message received. Purely for testing puposes! \n\n" +
  55
  56                         "\t-psm, --periodicallySendMessages <path to file> <period>\n"
  57                         + "\t\t <path to file> with groovy script which implements MessageGeneratorService followed by <period> in seconds.\n"
  58                         + "\t\t Messages which are sent periodically. Purely for testing puposes! \n\n" +
  59
  60                         "\t-snm, --sendNowMessage <path to file>\n"
  61                         + "\t\t <path to file> with groovy script which implements MessageGeneratorService.\n"
  62                         + "\t\t Messages are sent in defined states defined by programmer. Purely for testing puposes! \n\n" +
  63
  64                         "\t--help\n" + "\t\tdisplay this help and exits\n\n" +
  65
  66                         "With no parameters, this help is printed.";
  67
  68         public static void main(final String[] args) throws Exception {
  69                 if (args.length == 0 || (args.length == 1 && args[0].equalsIgnoreCase("--help"))) {
  70                         System.out.println(Main.usage);
  71                         return;
  72                 }
  73
  74                 InetSocketAddress address = null;
  75                 int keepAliveValue = 30;
  76                 int deadTimerValue = 0;
  77                 boolean stateful = false;
  78                 boolean active = false;
  79                 boolean versioned = false;
  80                 boolean instant = false;
  81                 int timeout = 0;
  82
  83                 int i = 0;
  84                 while (i < args.length) {
  85                         if (args[i].equalsIgnoreCase("-a") || args[i].equalsIgnoreCase("--address")) {
  86                                 final String[] ip = args[i + 1].split(":");
  87                                 address = new InetSocketAddress(InetAddress.getByName(ip[0]), Integer.valueOf(ip[1]));
  88                                 i++;
  89                         } else if (args[i].equalsIgnoreCase("-d") || args[i].equalsIgnoreCase("--deadtimer")) {
  90                                 deadTimerValue = Integer.valueOf(args[i + 1]);
  91                                 i++;
  92                         } else if (args[i].equalsIgnoreCase("-ka") || args[i].equalsIgnoreCase("--keepalive")) {
  93                                 keepAliveValue = Integer.valueOf(args[i + 1]);
  94                                 i++;
  95                         } else if (args[i].equalsIgnoreCase("--stateful")) {
  96                                 stateful = true;
  97                         } else if (args[i].equalsIgnoreCase("--active")) {
  98                                 stateful = true;
  99                                 active = true;
 100                         } else if (args[i].equalsIgnoreCase("--versioned")) {
 101                                 stateful = true;
 102                                 versioned = true;
 103                         } else if (args[i].equalsIgnoreCase("--instant")) {
 104                                 stateful = true;
 105                                 instant = true;
 106                                 if (i == args.length - 1) {
 107                                         timeout = 0;
 108                                 } else if (Integer.valueOf(args[i + 1]) > 0 && Integer.valueOf(args[i + 1]) < Integer.MAX_VALUE) {
 109                                         timeout = Integer.valueOf(args[i + 1]);
 110                                         i++;
 111                                 }
 112                         } else {
 113                                 System.out.println("WARNING: Unrecognized argument: " + args[i]);
 114                         }
 115                         i++;
 116                 }
 117                 if (deadTimerValue != 0 && deadTimerValue != keepAliveValue * 4) {
 118                         System.out.println("WARNING: The value of DeadTimer should be 4 times the value of KeepAlive.");
 119                 }
 120                 if (deadTimerValue == 0) {
 121                         deadTimerValue = keepAliveValue * 4;
 122                 }
 123
 124                 final PCEPSessionListenerFactory slf = new TestingSessionListenerFactory();
 125                 final PCEPSessionProposalFactory spf = new PCEPSessionProposalFactoryImpl(deadTimerValue, keepAliveValue, stateful, active, versioned, instant, timeout);
 126                 final PCEPSessionProposalCheckerFactory spcf = new PCEPSessionProposalCheckerFactoryImpl();
 127
 128                 final PCEPSessionProposal prefs = spf.getSessionProposal(address, 0);
 129
 130                 final DispatcherImpl d = new DispatcherImpl(new PCEPMessageFactory());
 131                 final PCEPDispatcherImpl dispatcher = new PCEPDispatcherImpl(d, spf);
 132
 133                 ProtocolServer s = null;
 134
 135                 try {
 136                         s = dispatcher.createServer(address, new PCEPConnectionFactory() {
 137                                 @Override
 138                                 public PCEPConnection createProtocolConnection(final InetSocketAddress address) {
 139                                         final PCEPSessionProposalChecker checker = spcf.getPreferencesChecker(address);
 140                                         final PCEPSessionListener lsnr = slf.getSessionListener(address.getAddress());
 141
 142                                         return new PCEPConnectionImpl(address, lsnr, prefs.getProposal(), checker);
 143                                 }
 144
 145                                 @Override
 146                                 public void setProposal(final PCEPSessionProposalFactory proposals, final InetSocketAddress address, final int sessionId) {
 147                                 }
 148                         }).get();
 149
 150                         // try {
 151                         // Thread.sleep(10000);
 152                         // } catch (final InterruptedException e) {
 153                         // e.printStackTrace();
 154                         // }
 155                         //
 156                         // s.close();
 157
 158                 } finally {
 159                         ((PCEPSessionProposalCheckerFactoryImpl) spcf).close();
 160                         // d.stop();
 161                 }
 162         }
 163 }