opendaylight/netconf/netconf-cli/src/main/java/org/opendaylight/controller/netconf/cli/commands/local/Connect.java

   1 /*
   2  * Copyright (c) 2014 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.controller.netconf.cli.commands.local;
   9
  10 import com.google.common.base.Optional;
  11 import com.google.common.base.Preconditions;
  12 import com.google.common.collect.Lists;
  13 import io.netty.util.concurrent.GlobalEventExecutor;
  14 import java.net.InetAddress;
  15 import java.net.InetSocketAddress;
  16 import java.net.UnknownHostException;
  17 import java.util.ArrayList;
  18 import java.util.Set;
  19 import org.opendaylight.controller.netconf.cli.NetconfDeviceConnectionManager;
  20 import org.opendaylight.controller.netconf.cli.commands.AbstractCommand;
  21 import org.opendaylight.controller.netconf.cli.commands.Command;
  22 import org.opendaylight.controller.netconf.cli.commands.input.Input;
  23 import org.opendaylight.controller.netconf.cli.commands.input.InputDefinition;
  24 import org.opendaylight.controller.netconf.cli.commands.output.Output;
  25 import org.opendaylight.controller.netconf.cli.commands.output.OutputDefinition;
  26 import org.opendaylight.controller.netconf.client.conf.NetconfClientConfiguration;
  27 import org.opendaylight.controller.netconf.client.conf.NetconfClientConfigurationBuilder;
  28 import org.opendaylight.controller.netconf.nettyutil.handler.ssh.authentication.LoginPassword;
  29 import org.opendaylight.protocol.framework.NeverReconnectStrategy;
  30 import org.opendaylight.protocol.framework.ReconnectStrategy;
  31 import org.opendaylight.yangtools.yang.common.QName;
  32 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeIdentifier;
  33 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier.NodeWithValue;
  34 import org.opendaylight.yangtools.yang.data.api.schema.DataContainerChild;
  35 import org.opendaylight.yangtools.yang.data.api.schema.LeafNode;
  36 import org.opendaylight.yangtools.yang.data.api.schema.LeafSetEntryNode;
  37 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  38 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafNodeBuilder;
  39 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafSetEntryNodeBuilder;
  40 import org.opendaylight.yangtools.yang.data.impl.schema.builder.impl.ImmutableLeafSetNodeBuilder;
  41 import org.opendaylight.yangtools.yang.model.api.RpcDefinition;
  42
  43 /**
  44  * Local command to connect to a remote device
  45  */
  46 public class Connect extends AbstractCommand {
  47
  48     private final NetconfDeviceConnectionManager connectManager;
  49     private final Integer connectionTimeout;
  50
  51     private Connect(final QName qName, final InputDefinition args, final OutputDefinition output,
  52                     final NetconfDeviceConnectionManager connectManager, final String description, final Integer connectionTimeout) {
  53         super(qName, args, output, description);
  54         this.connectManager = connectManager;
  55         this.connectionTimeout = connectionTimeout;
  56     }
  57
  58     @Override
  59     public Output invoke(final Input inputArgs) {
  60         final NetconfClientConfigurationBuilder config = getConfig(inputArgs);
  61         return invoke(config, getArgument(inputArgs, "address-name", String.class), inputArgs);
  62     }
  63
  64     private Output invoke(final NetconfClientConfigurationBuilder config, final String addressName, final Input inputArgs) {
  65         final Set<String> remoteCmds = connectManager.connectBlocking(addressName, getAdress(inputArgs), config);
  66
  67         final ArrayList<DataContainerChild<?, ?>> output = Lists.newArrayList();
  68         output.add(ImmutableLeafNodeBuilder.create()
  69                 .withNodeIdentifier(new NodeIdentifier(QName.create(getCommandId(), "status")))
  70                 .withValue("Connection initiated").build());
  71
  72         final ArrayList<LeafSetEntryNode<Object>> leafListChildren = Lists.newArrayList();
  73         for (final String cmdId : remoteCmds) {
  74             leafListChildren.add(ImmutableLeafSetEntryNodeBuilder.create()
  75                     .withNodeIdentifier(new NodeWithValue(QName.create(getCommandId(), "remote-commands"), cmdId))
  76                     .withValue(cmdId).build());
  77         }
  78
  79         return new Output(ImmutableLeafSetNodeBuilder.create()
  80                 .withNodeIdentifier(new NodeIdentifier(QName.create(getCommandId(), "remote-commands")))
  81                 .withValue(leafListChildren).build());
  82     }
  83
  84     private NetconfClientConfigurationBuilder getConfig(final Input inputArgs) {
  85
  86         final ReconnectStrategy strategy = getReconnectStrategy();
  87
  88         final String address = getArgument(inputArgs, "address-name", String.class);
  89         final Integer port = getArgument(inputArgs, "address-port", Integer.class);
  90         final String username = getArgument(inputArgs, "user-name", String.class);
  91         final String passwd = getArgument(inputArgs, "user-password", String.class);
  92
  93         final InetSocketAddress inetAddress;
  94         try {
  95             inetAddress = new InetSocketAddress(InetAddress.getByName(address), port);
  96         } catch (final UnknownHostException e) {
  97             throw new IllegalArgumentException("Unable to use address: " + address, e);
  98         }
  99
 100         return NetconfClientConfigurationBuilder.create().withAddress(inetAddress)
 101                 .withConnectionTimeoutMillis(connectionTimeout)
 102                 .withReconnectStrategy(strategy)
 103                 .withAuthHandler(new LoginPassword(username, passwd))
 104                 .withProtocol(NetconfClientConfiguration.NetconfClientProtocol.SSH);
 105     }
 106
 107     private InetSocketAddress getAdress(final Input inputArgs) {
 108         final String address = getArgument(inputArgs, "address-name", String.class);
 109         final InetSocketAddress inetAddress;
 110         try {
 111             inetAddress = new InetSocketAddress(InetAddress.getByName(address), getArgument(inputArgs, "address-port", Integer.class));
 112         } catch (final UnknownHostException e) {
 113             throw new IllegalArgumentException("Unable to use address: " + address, e);
 114         }
 115         return inetAddress;
 116     }
 117
 118     private <T> Optional<T> getArgumentOpt(final Input inputArgs, final String argName, final Class<T> type) {
 119         final QName argQName = QName.create(getCommandId(), argName);
 120         final NormalizedNode<?, ?> argumentNode = inputArgs.getArg(argName);
 121         if (argumentNode == null) {
 122             return Optional.absent();
 123         }
 124         Preconditions.checkArgument(argumentNode instanceof LeafNode, "Only simple type argument supported, %s",
 125                 argQName);
 126
 127         final Object value = argumentNode.getValue();
 128         Preconditions.checkArgument(type.isInstance(value), "Unexpected instance type: %s for argument: %s",
 129                 value.getClass(), argQName);
 130         return Optional.of(type.cast(value));
 131     }
 132
 133     private <T> T getArgument(final Input inputArgs, final String argName, final Class<T> type) {
 134         final Optional<T> argumentOpt = getArgumentOpt(inputArgs, argName, type);
 135         Preconditions.checkState(argumentOpt.isPresent(), "Argument: %s is missing but is required", argName);
 136         return argumentOpt.get();
 137     }
 138
 139     public static ReconnectStrategy getReconnectStrategy() {
 140         // FIXME move to args either start-up args or input nodes for connect or both
 141         return new NeverReconnectStrategy(GlobalEventExecutor.INSTANCE, 1000);
 142     }
 143
 144     public static Command create(final RpcDefinition rpcDefinition, final NetconfDeviceConnectionManager connectManager, final Integer connectionTimeout) {
 145         return new Connect(rpcDefinition.getQName(), getInputDefinition(rpcDefinition),
 146                 getOutputDefinition(rpcDefinition), connectManager, rpcDefinition.getDescription(), connectionTimeout);
 147     }
 148 }