--- /dev/null
+/*
+ * Copyright (c) 2015 Ericsson India Global Services Pvt Ltd. 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.bgpmanager;
+
+import java.io.BufferedReader;
+import java.io.IOException;
+import java.io.InputStreamReader;
+import java.io.PrintWriter;
+import java.net.Socket;
+import java.net.SocketTimeoutException;
+import java.net.UnknownHostException;
+import org.apache.karaf.shell.commands.Command;
+import org.apache.karaf.shell.commands.Option;
+import org.apache.karaf.shell.console.OsgiCommandSupport;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+@Command(scope = "odl", name = "show-bgp", description = "")
+public class VtyshCli extends OsgiCommandSupport {
+ // public class DisplayBgpConfigCli {
+
+ @Option(name = "--cmd", description = "command to run", required = true, multiValued = false)
+ String cmd = null;
+
+ private static final Logger LOGGER = LoggerFactory.getLogger(VtyshCli.class);
+
+ private static int serverPort = 2605;
+ private static String serverName = "localhost";
+ private static int handlerModule = 0;
+ private static final int BGPd = 1;
+ public static String passwordCheckStr = "Password:";
+ public static String vtyPassword = "sdncbgpc";
+ public static String noPaginationCmd = "terminal length 0";
+ public static int sockTimeout = 5;
+
+ String[] validCommands = new String[]{
+ "display routing ip bgp vpnv4 all",
+ "display routing ip bgp vpnv4 rd <rd>",
+ "display routing ip bgp vpnv4 all neighbors",
+ "display routing ip bgp vpnv4 all neighbors <ip> routes",
+ "display routing ip bgp vpnv4 all <ip/mask>",
+ "display routing ip bgp vpnv4 all summary",
+ "display routing ip bgp vpnv4 all tags",
+ "display routing ip bgp vpnv4 rd <rd> tags",
+ "display routing ip bgp vpnv4 rd <rd> <ip>",
+ "display routing ip bgp vpnv4 rd <rd> <ip/mask>",
+ "display routing ip bgp neighbors",
+ "display routing ip bgp summary",
+ "display routing ip bgp ipv4 unicast",
+ "display routing ip bgp ipv4 unicast <ip/mask>",
+ "display routing bgp neighbors",
+ "display routing bgp neighbors <ip>",
+ "display routing bgp ipv4 unicast <ip>",
+ "display routing bgp ipv4 unicast <ip/mask>"
+ };
+ private static final Logger logger = LoggerFactory.getLogger(VtyshCli.class);
+
+ // @Override
+ protected Object doExecute() throws Exception {
+ String sArg;
+ cmd = cmd.trim();
+ if (cmd.equals("") || cmd.equals("help") ||
+ cmd.equals("-help") || cmd.equals("--help")) {
+ for (String help : validCommands) {
+ System.out.println(help);
+ }
+ return null;
+ }
+ String args[] = cmd.split(" ");
+ if (args.length == 0) {
+ return null;
+ }
+ sArg = args[0];
+ if (sArg == null || sArg.trim().equals("")) {
+ System.out.println("Please provide a valid input.");
+ return null;
+ }
+ switch (sArg) {
+ case "ip":
+ case "bgp":
+ handlerModule = BGPd;
+ break;
+ default:
+ System.out.println("Unknown command");
+ return null;
+ }
+
+ switch (handlerModule) {
+ case BGPd:
+ try {
+ handleCommand(sArg, cmd);
+ } catch (IOException ioe) {
+ System.out.println("IOException thrown.");
+ }
+ break;
+ default:
+ break;
+ }
+ return null;
+ }
+
+ public static void setHostAddr(String hostAddr) {
+ serverName = hostAddr;
+ }
+
+ public String getHostAddr() {
+ return serverName;
+ }
+
+ public static void handleCommand(String arg, String cmd) throws IOException {
+
+ StringBuilder inputBgpCmd = new StringBuilder();
+
+ String str, prompt, replacedStr, inputCmd = null;
+ char cbuf[] = new char[10];
+ char op_buf[];
+ Socket socket = null;
+ PrintWriter out_to_socket = null;
+ BufferedReader in_from_socket = null;
+ StringBuilder sb = new StringBuilder();
+ int ip = 0, ret;
+ StringBuilder temp, temp2;
+ char ch, gt = '>';
+
+ inputBgpCmd.append("show " + cmd);
+
+ inputCmd = inputBgpCmd.toString();
+
+ try {
+ socket = new Socket(serverName, serverPort);
+
+ } catch (UnknownHostException ioe) {
+ System.out.println("No host exists: " + ioe.getMessage());
+ return;
+ } catch (IOException ioe) {
+ System.out.println("I/O error occured " + ioe.getMessage());
+ return;
+ }
+ try {
+ socket.setSoTimeout(sockTimeout * 1000);
+ out_to_socket = new PrintWriter(socket.getOutputStream(), true);
+ in_from_socket = new BufferedReader(new InputStreamReader(socket.getInputStream()));
+
+ } catch (IOException ioe) {
+ System.out.println("IOException thrown.");
+ socket.close();
+ return;
+ }
+ while (true) {
+ try {
+ ret = in_from_socket.read(cbuf);
+
+ } catch (SocketTimeoutException ste) {
+ System.out.println("Read from Socket timed Out while asking for password.");
+ socket.close();
+ return;
+ }
+ if (ret == -1) {
+ System.out.println("Connection closed by BGPd.");
+ socket.close();
+ return;
+ } else {
+ sb.append(cbuf);
+
+ if (sb.toString().contains(passwordCheckStr)) {
+
+ break;
+ }
+ }
+ }
+
+ sb.setLength(0);
+ out_to_socket.println(vtyPassword);
+
+ while (true) {
+ try {
+ ip = in_from_socket.read();
+ } catch (SocketTimeoutException ste) {
+ System.out.println(sb.toString());
+ System.out.println("Read from Socket timed Out while verifying the password.");
+ socket.close();
+ return;
+ }
+ if (ip == (int) gt) {
+ sb.append(gt);
+ break;
+ } else if (ip == -1) {
+ System.out.println(sb.toString());
+ System.out.println("Connection closed by BGPd.");
+ socket.close();
+ return;
+ } else {
+ ch = (char) ip;
+ sb.append(ch);
+
+ }
+ }
+
+ String promptStr = sb.toString();
+ prompt = promptStr.trim();
+ sb.setLength(0);
+ out_to_socket.println(noPaginationCmd);
+ while (true) {
+ try {
+ ip = in_from_socket.read();
+ } catch (SocketTimeoutException ste) {
+ System.out.println(sb.toString());
+ System.out.println("Read from Socket timed Out while sending the term len command..");
+ socket.close();
+ return;
+ }
+ if (ip == (int) gt) {
+ break;
+ } else if (ip == -1) {
+ System.out.println(sb.toString());
+ System.out.println("Connection closed by BGPd.");
+ socket.close();
+ return;
+ } else {
+ ch = (char) ip;
+ sb.append(ch);
+
+ }
+ }
+ sb.setLength(0);
+
+ out_to_socket.println(inputCmd);
+ StringBuffer output = new StringBuffer();
+ String errorMsg = "";
+ while (true) {
+ op_buf = new char[100];
+ temp = new StringBuilder();
+ temp2 = new StringBuilder();
+ try {
+ ret = in_from_socket.read(op_buf);
+
+ } catch (SocketTimeoutException ste) {
+ errorMsg = "Read from Socket timed Out while getting the data.";
+ break;
+ }
+ if (ret == -1) {
+ errorMsg = "Connection closed by BGPd";
+ break;
+ }
+ temp2.append(op_buf);
+
+ if (temp2.toString().contains(inputCmd)) {
+
+ replacedStr = temp2.toString().replaceAll(inputCmd, "");
+ temp.append(replacedStr);
+ temp2.setLength(0);
+
+ } else {
+ temp.append(op_buf);
+ temp2.setLength(0);
+
+ }
+
+ String outputStr = temp.toString();
+ outputStr.replaceAll("^\\s+|\\s+$", "");
+ output.append(outputStr);
+ if (output.toString().trim().endsWith(prompt)) {
+ int index = output.toString().lastIndexOf(prompt);
+ String newString = output.toString().substring(0, index);
+ output.setLength(0);
+ output.append(newString);
+ break;
+ }
+ temp.setLength(0);
+ }
+ System.out.println(output.toString().trim());
+ if (errorMsg.length() > 0) {
+ System.out.println(errorMsg);
+ }
+ socket.close();
+ return;
+
+ }
+
+}
Code Review / vpnservice.git / blobdiff