Bug 498 - Fixed PCEP and BGP testtools
[bgpcep.git] / 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 io.netty.channel.nio.NioEventLoopGroup;
11 import io.netty.util.HashedWheelTimer;
12
13 import java.net.InetAddress;
14 import java.net.InetSocketAddress;
15
16 import org.opendaylight.protocol.pcep.PCEPSessionProposalFactory;
17 import org.opendaylight.protocol.pcep.ietf.initiated00.Stateful07SessionProposalFactory;
18 import org.opendaylight.protocol.pcep.ietf.stateful07.StatefulActivator;
19 import org.opendaylight.protocol.pcep.impl.DefaultPCEPSessionNegotiatorFactory;
20 import org.opendaylight.protocol.pcep.impl.PCEPDispatcherImpl;
21 import org.opendaylight.protocol.pcep.spi.pojo.ServiceLoaderPCEPExtensionProviderContext;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.Open;
23 import org.slf4j.Logger;
24 import org.slf4j.LoggerFactory;
25
26 public final class Main {
27
28         private static final Logger LOG = LoggerFactory.getLogger(Main.class);
29
30         public static final String USAGE = "DESCRIPTION:\n"
31                         + "\tCreates a server with given parameters. As long as it runs, it accepts connections " + "from PCCs.\n" + "USAGE:\n"
32                         + "\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--instant\n"
48                         + "\t\tinstantiated stateful, <seconds> cleanup timeout (default value, if not included = 0) (implies --stateful)\n\n" +
49
50                         "\t-arm, --autoResponseMessages <path to file>\n"
51                         + "\t\t <path to file> with groovy script which implements MessageGeneratorService.\n"
52                         + "\t\t Messages are used as auto response for every message received. Purely for testing puposes! \n\n" +
53
54                         "\t-psm, --periodicallySendMessages <path to file> <period>\n"
55                         + "\t\t <path to file> with groovy script which implements MessageGeneratorService followed by <period> in seconds.\n"
56                         + "\t\t Messages which are sent periodically. Purely for testing puposes! \n\n" +
57
58                         "\t-snm, --sendNowMessage <path to file>\n"
59                         + "\t\t <path to file> with groovy script which implements MessageGeneratorService.\n"
60                         + "\t\t Messages are sent in defined states defined by programmer. Purely for testing puposes! \n\n" +
61
62                         "\t--help\n" + "\t\tdisplay this help and exits\n\n" +
63
64                         "With no parameters, this help is printed.";
65
66         private Main() {
67
68         }
69
70         public static void main(final String[] args) throws Exception {
71                 if (args.length == 0 || (args.length == 1 && args[0].equalsIgnoreCase("--help"))) {
72                         System.err.println(Main.USAGE);
73                         return;
74                 }
75
76                 InetSocketAddress address = null;
77                 int keepAliveValue = 30;
78                 int deadTimerValue = 0;
79                 boolean stateful = false;
80                 boolean active = false;
81                 final boolean versioned = false;
82                 boolean instant = false;
83
84                 int i = 0;
85                 while (i < args.length) {
86                         if (args[i].equalsIgnoreCase("-a") || args[i].equalsIgnoreCase("--address")) {
87                                 final String[] ip = args[i + 1].split(":");
88                                 address = new InetSocketAddress(InetAddress.getByName(ip[0]), Integer.valueOf(ip[1]));
89                                 i++;
90                         } else if (args[i].equalsIgnoreCase("-d") || args[i].equalsIgnoreCase("--deadtimer")) {
91                                 deadTimerValue = Integer.valueOf(args[i + 1]);
92                                 i++;
93                         } else if (args[i].equalsIgnoreCase("-ka") || args[i].equalsIgnoreCase("--keepalive")) {
94                                 keepAliveValue = Integer.valueOf(args[i + 1]);
95                                 i++;
96                         } else if (args[i].equalsIgnoreCase("--stateful")) {
97                                 stateful = true;
98                         } else if (args[i].equalsIgnoreCase("--active")) {
99                                 stateful = true;
100                                 active = true;
101                         } else if (args[i].equalsIgnoreCase("--instant")) {
102                                 stateful = true;
103                                 instant = true;
104                         } else {
105                                 LOG.warn("WARNING: Unrecognized argument: {}", args[i]);
106                         }
107                         i++;
108                 }
109                 if (deadTimerValue != 0 && deadTimerValue != keepAliveValue * 4) {
110                         LOG.warn("WARNING: The value of DeadTimer should be 4 times the value of KeepAlive.");
111                 }
112                 if (deadTimerValue == 0) {
113                         deadTimerValue = keepAliveValue * 4;
114                 }
115
116                 final PCEPSessionProposalFactory spf = new Stateful07SessionProposalFactory(deadTimerValue, keepAliveValue, stateful, active, instant);
117
118                 final Open prefs = spf.getSessionProposal(address, 0);
119
120                 StatefulActivator activator07 = new StatefulActivator();
121                 activator07.start(ServiceLoaderPCEPExtensionProviderContext.getSingletonInstance());
122
123                 final PCEPDispatcherImpl dispatcher = new PCEPDispatcherImpl(ServiceLoaderPCEPExtensionProviderContext.getSingletonInstance().getMessageHandlerRegistry(), new DefaultPCEPSessionNegotiatorFactory(new HashedWheelTimer(), prefs, 5), new NioEventLoopGroup(), new NioEventLoopGroup());
124
125                 dispatcher.createServer(address, new TestingSessionListenerFactory()).get();
126         }
127 }