Merge "BUG-9048 Fix actor crash when writing incorrect data"
[netconf.git] / netconf / sal-netconf-connector / src / main / java / org / opendaylight / netconf / sal / connect / util / NetconfTopologyRPCProvider.java
1 /*
2  * Copyright (c) 2017 Brocade Communication Systems 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.netconf.sal.connect.util;
9
10 import com.google.common.base.Preconditions;
11 import com.google.common.util.concurrent.CheckedFuture;
12 import com.google.common.util.concurrent.FutureCallback;
13 import com.google.common.util.concurrent.Futures;
14 import com.google.common.util.concurrent.MoreExecutors;
15 import com.google.common.util.concurrent.SettableFuture;
16 import java.util.concurrent.Future;
17 import org.opendaylight.aaa.encrypt.AAAEncryptionService;
18 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
19 import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
20 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
21 import org.opendaylight.controller.md.sal.common.api.data.TransactionCommitFailedException;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.AddNetconfNodeInput;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNode;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeBuilder;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeTopologyService;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.LoginPassword;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.LoginPasswordBuilder;
28 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
29 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
30 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.TopologyId;
31 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.Topology;
32 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.TopologyKey;
33 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
34 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
35 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
36 import org.opendaylight.yangtools.yang.common.RpcResult;
37 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
38 import org.slf4j.Logger;
39 import org.slf4j.LoggerFactory;
40
41 public class NetconfTopologyRPCProvider implements NetconfNodeTopologyService {
42     private final AAAEncryptionService encryptionService;
43     private final DataBroker dataBroker;
44     private final String topologyId;
45     private static final Logger LOG = LoggerFactory.getLogger(NetconfTopologyRPCProvider.class);
46
47     public NetconfTopologyRPCProvider(final DataBroker dataBroker,
48                                       final AAAEncryptionService encryptionService,
49                                       final String topologyId) {
50         this.dataBroker = dataBroker;
51         this.encryptionService = Preconditions.checkNotNull(encryptionService);
52         this.topologyId = Preconditions.checkNotNull(topologyId);
53     }
54
55     @Override
56     public Future<RpcResult<Void>> addNetconfNode(AddNetconfNodeInput input) {
57         NetconfNode node = this.encryptPassword(input);
58         final SettableFuture<RpcResult<Void>> futureResult = SettableFuture.create();
59         NodeId nodeId = new NodeId(input.getNodeId());
60         writeToConfigDS(node, nodeId, topologyId, futureResult);
61         return futureResult;
62     }
63
64     private NetconfNode encryptPassword(AddNetconfNodeInput input) {
65         NetconfNodeBuilder builder = new NetconfNodeBuilder();
66         builder.fieldsFrom(input);
67
68         boolean encrypt = input.isEncrypt();
69         LoginPassword loginPassword = (LoginPassword) input.getCredentials();
70         if (encrypt) {
71             String encryptedPassword = encryptionService.encrypt(loginPassword.getPassword());
72             LoginPassword newCreds = new LoginPasswordBuilder().setPassword(encryptedPassword)
73                     .setUsername(loginPassword.getUsername()).build();
74             builder.setCredentials(newCreds);
75         }
76
77         NetconfNode node = builder.build();
78         return node;
79     }
80
81     private void writeToConfigDS(NetconfNode node, NodeId nodeId, String topologyId,
82                                  final SettableFuture<RpcResult<Void>> futureResult) {
83
84         WriteTransaction writeTransaction = dataBroker.newWriteOnlyTransaction();
85         final InstanceIdentifier<NetworkTopology> networkTopologyId =
86                 InstanceIdentifier.builder(NetworkTopology.class).build();
87         final InstanceIdentifier<NetconfNode> niid = networkTopologyId.child(Topology.class,
88                 new TopologyKey(new TopologyId(topologyId))).child(Node.class,
89                 new NodeKey(nodeId)).augmentation(NetconfNode.class);
90         writeTransaction.merge(LogicalDatastoreType.CONFIGURATION, niid, node, true);
91         final CheckedFuture<Void, TransactionCommitFailedException> future = writeTransaction.submit();
92         Futures.addCallback(future, new FutureCallback<Void>() {
93
94             @Override
95             public void onSuccess(Void result) {
96                 LOG.info("add-netconf-node RPC: Added netconf node successfully.");
97                 futureResult.set(RpcResultBuilder.<Void>success().build());
98             }
99
100             @Override
101             public void onFailure(Throwable exception) {
102                 LOG.error("add-netconf-node RPC: Unable to add netconf node.", exception);
103                 futureResult.setException(exception);
104             }
105         }, MoreExecutors.directExecutor());
106     }
107
108 }