55e17c56b38e5d47b656ad37f7e5c9e5f11e78db
[controller.git] / opendaylight / md-sal / sal-netconf-connector / src / main / java / org / opendaylight / controller / sal / connect / netconf / NetconfMapping.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.sal.connect.netconf;
9
10 import com.google.common.base.Optional;
11 import com.google.common.base.Predicate;
12 import com.google.common.collect.Collections2;
13 import com.google.common.collect.ImmutableList;
14 import com.google.common.collect.Iterables;
15 import com.google.common.collect.Lists;
16 import java.net.URI;
17 import java.util.ArrayList;
18 import java.util.Collections;
19 import java.util.List;
20 import java.util.Map;
21 import java.util.Set;
22 import java.util.concurrent.atomic.AtomicInteger;
23 import javax.activation.UnsupportedDataTypeException;
24 import javax.annotation.Nullable;
25 import org.opendaylight.controller.netconf.api.NetconfDocumentedException;
26 import org.opendaylight.controller.netconf.api.NetconfMessage;
27 import org.opendaylight.controller.netconf.util.messages.NetconfMessageUtil;
28 import org.opendaylight.controller.netconf.util.xml.XmlUtil;
29 import org.opendaylight.controller.sal.common.util.Rpcs;
30 import org.opendaylight.yangtools.yang.common.QName;
31 import org.opendaylight.yangtools.yang.common.RpcError;
32 import org.opendaylight.yangtools.yang.common.RpcResult;
33 import org.opendaylight.yangtools.yang.data.api.CompositeNode;
34 import org.opendaylight.yangtools.yang.data.api.InstanceIdentifier;
35 import org.opendaylight.yangtools.yang.data.api.InstanceIdentifier.NodeIdentifierWithPredicates;
36 import org.opendaylight.yangtools.yang.data.api.InstanceIdentifier.PathArgument;
37 import org.opendaylight.yangtools.yang.data.api.Node;
38 import org.opendaylight.yangtools.yang.data.impl.CompositeNodeTOImpl;
39 import org.opendaylight.yangtools.yang.data.impl.ImmutableCompositeNode;
40 import org.opendaylight.yangtools.yang.data.impl.SimpleNodeTOImpl;
41 import org.opendaylight.yangtools.yang.data.impl.codec.xml.XmlDocumentUtils;
42 import org.opendaylight.yangtools.yang.data.impl.util.CompositeNodeBuilder;
43 import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
44 import org.opendaylight.yangtools.yang.model.api.RpcDefinition;
45 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
46 import org.w3c.dom.Document;
47 import org.w3c.dom.Element;
48
49 public class NetconfMapping {
50
51     public static URI NETCONF_URI = URI.create("urn:ietf:params:xml:ns:netconf:base:1.0");
52     public static String NETCONF_MONITORING_URI = "urn:ietf:params:xml:ns:yang:ietf-netconf-monitoring";
53     public static URI NETCONF_NOTIFICATION_URI = URI.create("urn:ietf:params:xml:ns:netconf:notification:1.0");
54     public static URI NETCONF_ROLLBACK_ON_ERROR_URI = URI.create("urn:ietf:params:netconf:capability:rollback-on-error:1.0");
55
56     public static QName NETCONF_QNAME = QName.create(NETCONF_URI, null, "netconf");
57     public static QName NETCONF_RPC_QNAME = QName.create(NETCONF_QNAME, "rpc");
58     public static QName NETCONF_GET_QNAME = QName.create(NETCONF_QNAME, "get");
59     public static QName NETCONF_FILTER_QNAME = QName.create(NETCONF_QNAME, "filter");
60     public static QName NETCONF_TYPE_QNAME = QName.create(NETCONF_QNAME, "type");
61     public static QName NETCONF_GET_CONFIG_QNAME = QName.create(NETCONF_QNAME, "get-config");
62     public static QName NETCONF_EDIT_CONFIG_QNAME = QName.create(NETCONF_QNAME, "edit-config");
63     public static QName NETCONF_DELETE_CONFIG_QNAME = QName.create(NETCONF_QNAME, "delete-config");
64     public static QName NETCONF_OPERATION_QNAME = QName.create(NETCONF_QNAME, "operation");
65     public static QName NETCONF_COMMIT_QNAME = QName.create(NETCONF_QNAME, "commit");
66
67     public static QName NETCONF_CONFIG_QNAME = QName.create(NETCONF_QNAME, "config");
68     public static QName NETCONF_SOURCE_QNAME = QName.create(NETCONF_QNAME, "source");
69     public static QName NETCONF_TARGET_QNAME = QName.create(NETCONF_QNAME, "target");
70
71     public static QName NETCONF_CANDIDATE_QNAME = QName.create(NETCONF_QNAME, "candidate");
72     public static QName NETCONF_RUNNING_QNAME = QName.create(NETCONF_QNAME, "running");
73
74     public static QName NETCONF_ERROR_OPTION_QNAME = QName.create(NETCONF_QNAME, "error-option");
75     public static String ROLLBACK_ON_ERROR_OPTION = "rollback-on-error";
76
77     public static QName NETCONF_RPC_REPLY_QNAME = QName.create(NETCONF_QNAME, "rpc-reply");
78     public static QName NETCONF_OK_QNAME = QName.create(NETCONF_QNAME, "ok");
79     public static QName NETCONF_DATA_QNAME = QName.create(NETCONF_QNAME, "data");
80     public static QName NETCONF_CREATE_SUBSCRIPTION_QNAME = QName.create(NETCONF_NOTIFICATION_URI, null,
81             "create-subscription");
82     public static QName NETCONF_CANCEL_SUBSCRIPTION_QNAME = QName.create(NETCONF_NOTIFICATION_URI, null,
83             "cancel-subscription");
84     public static QName IETF_NETCONF_MONITORING_MODULE = QName.create(NETCONF_MONITORING_URI, "2010-10-04",
85             "ietf-netconf-monitoring");
86
87     static List<Node<?>> RUNNING = Collections.<Node<?>> singletonList(new SimpleNodeTOImpl(NETCONF_RUNNING_QNAME,
88             null, null));
89
90     public static CompositeNode CONFIG_SOURCE_RUNNING = new CompositeNodeTOImpl(NETCONF_SOURCE_QNAME, null, RUNNING);
91
92     static AtomicInteger messageId = new AtomicInteger(0);
93
94     static Node<?> toFilterStructure(InstanceIdentifier identifier) {
95         Node<?> previous = null;
96         if (identifier.getPath().isEmpty()) {
97             return null;
98         }
99
100         for (org.opendaylight.yangtools.yang.data.api.InstanceIdentifier.PathArgument component : Lists
101                 .reverse(identifier.getPath())) {
102             previous = toNode(component, previous);
103         }
104         return filter("subtree", previous);
105     }
106
107     static Node<?> toNode(NodeIdentifierWithPredicates argument, Node<?> node) {
108         List<Node<?>> list = new ArrayList<>();
109         for (Map.Entry<QName, Object> arg : argument.getKeyValues().entrySet()) {
110             list.add(new SimpleNodeTOImpl(arg.getKey(), null, arg.getValue()));
111         }
112         if (node != null) {
113             list.add(node);
114         }
115         return new CompositeNodeTOImpl(argument.getNodeType(), null, list);
116     }
117
118     static Node<?> toNode(PathArgument argument, Node<?> node) {
119         if (node != null) {
120             return new CompositeNodeTOImpl(argument.getNodeType(), null, Collections.<Node<?>> singletonList(node));
121         } else {
122             return new SimpleNodeTOImpl(argument.getNodeType(), null, null);
123         }
124     }
125
126     static CompositeNode toCompositeNode(NetconfMessage message, Optional<SchemaContext> ctx) {
127         // TODO: implement general normalization to normalize incoming Netconf
128         // Message
129         // for Schema Context counterpart
130         return null;
131     }
132
133     static CompositeNode toNotificationNode(NetconfMessage message, Optional<SchemaContext> ctx) {
134         if (ctx.isPresent()) {
135             SchemaContext schemaContext = ctx.get();
136             Set<NotificationDefinition> notifications = schemaContext.getNotifications();
137             Document document = message.getDocument();
138             return XmlDocumentUtils.notificationToDomNodes(document, Optional.fromNullable(notifications), ctx.get());
139         }
140         return null;
141     }
142
143     static NetconfMessage toRpcMessage(QName rpc, CompositeNode node, Optional<SchemaContext> ctx) {
144         CompositeNodeTOImpl rpcPayload = wrap(NETCONF_RPC_QNAME, flattenInput(node));
145         Document w3cPayload = null;
146         try {
147             w3cPayload = XmlDocumentUtils.toDocument(rpcPayload, XmlDocumentUtils.defaultValueCodecProvider());
148         } catch (UnsupportedDataTypeException e) {
149             throw new IllegalArgumentException("Unable to create message", e);
150         }
151         w3cPayload.getDocumentElement().setAttribute("message-id", "m-" + messageId.getAndIncrement());
152         return new NetconfMessage(w3cPayload);
153     }
154
155     static CompositeNode flattenInput(final CompositeNode node) {
156         final QName inputQName = QName.create(node.getNodeType(), "input");
157         CompositeNode input = node.getFirstCompositeByName(inputQName);
158         if (input == null)
159             return node;
160         if (input instanceof CompositeNode) {
161
162             List<Node<?>> nodes = ImmutableList.<Node<?>> builder() //
163                     .addAll(input.getChildren()) //
164                     .addAll(Collections2.filter(node.getChildren(), new Predicate<Node<?>>() {
165                         @Override
166                         public boolean apply(@Nullable final Node<?> input) {
167                             return input.getNodeType() != inputQName;
168                         }
169                     })) //
170                     .build();
171
172             return ImmutableCompositeNode.create(node.getNodeType(), nodes);
173         }
174
175         return input;
176     }
177
178     static RpcResult<CompositeNode> toRpcResult(NetconfMessage message, final QName rpc, Optional<SchemaContext> context) {
179         CompositeNode rawRpc;
180         if (context.isPresent())
181             if (isDataRetrieQNameReply(rpc)) {
182
183                 Element xmlData = getDataSubtree(message.getDocument());
184
185                 List<org.opendaylight.yangtools.yang.data.api.Node<?>> dataNodes = XmlDocumentUtils.toDomNodes(xmlData,
186                         Optional.of(context.get().getDataDefinitions()));
187
188                 CompositeNodeBuilder<ImmutableCompositeNode> it = ImmutableCompositeNode.builder();
189                 it.setQName(NETCONF_RPC_REPLY_QNAME);
190                 it.add(ImmutableCompositeNode.create(NETCONF_DATA_QNAME, dataNodes));
191
192                 rawRpc = it.toInstance();
193                 // sys(xmlData)
194             } else {
195                 rawRpc = (CompositeNode) toCompositeNode(message, context);
196             }
197         else {
198             rawRpc = (CompositeNode) toCompositeNode(message.getDocument());
199         }
200         // rawRpc.
201         return Rpcs.getRpcResult(true, rawRpc, Collections.<RpcError> emptySet());
202     }
203
204     static Element getDataSubtree(Document doc) {
205         return (Element) doc.getElementsByTagNameNS(NETCONF_URI.toString(), "data").item(0);
206     }
207
208     static boolean isDataRetrieQNameReply(QName it) {
209         return NETCONF_URI == it.getNamespace()
210                 && (it.getLocalName() == NETCONF_GET_CONFIG_QNAME.getLocalName() || it.getLocalName() == NETCONF_GET_QNAME
211                         .getLocalName());
212     }
213
214     static CompositeNodeTOImpl wrap(QName name, Node<?> node) {
215         if (node != null) {
216             return new CompositeNodeTOImpl(name, null, Collections.<Node<?>> singletonList(node));
217         } else {
218             return new CompositeNodeTOImpl(name, null, Collections.<Node<?>> emptyList());
219         }
220     }
221
222     static CompositeNodeTOImpl wrap(QName name, Node<?> additional, Node<?> node) {
223         if (node != null) {
224             return new CompositeNodeTOImpl(name, null, ImmutableList.of(additional, node));
225         } else {
226             return new CompositeNodeTOImpl(name, null, ImmutableList.<Node<?>> of(additional));
227         }
228     }
229
230     static ImmutableCompositeNode filter(String type, Node<?> node) {
231         CompositeNodeBuilder<ImmutableCompositeNode> it = ImmutableCompositeNode.builder(); //
232         it.setQName(NETCONF_FILTER_QNAME);
233         it.setAttribute(NETCONF_TYPE_QNAME, type);
234         if (node != null) {
235             return it.add(node).toInstance();
236         } else {
237             return it.toInstance();
238         }
239     }
240
241     public static Node<?> toCompositeNode(Document document) {
242         return XmlDocumentUtils.toDomNode(document);
243     }
244
245     public static void checkValidReply(NetconfMessage input, NetconfMessage output) {
246         String inputMsgId = input.getDocument().getDocumentElement().getAttribute("message-id");
247         String outputMsgId = output.getDocument().getDocumentElement().getAttribute("message-id");
248
249         if(inputMsgId.equals(outputMsgId) == false) {
250             String requestXml = XmlUtil.toString(input.getDocument());
251             String responseXml = XmlUtil.toString(output.getDocument());
252             throw new IllegalStateException(String.format("Rpc request and reply message IDs must be same. Request: %s, response: %s", requestXml, responseXml));
253         }
254     }
255
256     public static void checkSuccessReply(NetconfMessage output) throws NetconfDocumentedException {
257         if(NetconfMessageUtil.isErrorMessage(output)) {
258             throw new IllegalStateException(String.format("Response contains error: %s", XmlUtil.toString(output.getDocument())));
259         }
260     }
261 }