Bug 8745: Migrate EditConfig to the new XML parser
[netconf.git] / netconf / mdsal-netconf-connector / src / main / java / org / opendaylight / netconf / mdsal / connector / ops / EditConfig.java
1 /*
2  * Copyright (c) 2015 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
9 package org.opendaylight.netconf.mdsal.connector.ops;
10
11 import com.google.common.base.Optional;
12 import com.google.common.base.Preconditions;
13 import com.google.common.collect.ImmutableMap;
14 import java.io.StringReader;
15 import java.net.URI;
16 import java.net.URISyntaxException;
17 import java.util.List;
18 import java.util.ListIterator;
19 import java.util.Map;
20 import java.util.stream.Collectors;
21 import org.opendaylight.controller.config.util.xml.DocumentedException;
22 import org.opendaylight.controller.config.util.xml.DocumentedException.ErrorSeverity;
23 import org.opendaylight.controller.config.util.xml.DocumentedException.ErrorTag;
24 import org.opendaylight.controller.config.util.xml.DocumentedException.ErrorType;
25 import org.opendaylight.controller.config.util.xml.XmlElement;
26 import org.opendaylight.controller.config.util.xml.XmlUtil;
27 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
28 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
29 import org.opendaylight.controller.md.sal.dom.api.DOMDataReadWriteTransaction;
30 import org.opendaylight.netconf.api.NetconfDocumentedException;
31 import org.opendaylight.netconf.api.xml.XmlNetconfConstants;
32 import org.opendaylight.netconf.mdsal.connector.CurrentSchemaContext;
33 import org.opendaylight.netconf.mdsal.connector.TransactionProvider;
34 import org.opendaylight.netconf.mdsal.connector.ops.DataTreeChangeTracker.DataTreeChange;
35 import org.opendaylight.netconf.util.mapping.AbstractSingletonNetconfOperation;
36 import org.opendaylight.yangtools.util.xml.UntrustedXML;
37 import org.opendaylight.yangtools.yang.common.QName;
38 import org.opendaylight.yangtools.yang.data.api.ModifyAction;
39 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
40 import org.opendaylight.yangtools.yang.data.api.schema.AugmentationNode;
41 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
42 import org.opendaylight.yangtools.yang.data.api.schema.MapNode;
43 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
44 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
45 import org.opendaylight.yangtools.yang.data.codec.xml.XmlParserStream;
46 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
47 import org.opendaylight.yangtools.yang.data.impl.schema.NormalizedNodeResult;
48 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
49 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
50 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
51 import org.opendaylight.yangtools.yang.model.api.Module;
52 import org.opendaylight.yangtools.yang.model.api.SchemaNode;
53 import org.opendaylight.yangtools.yang.model.util.SchemaContextUtil;
54 import org.slf4j.Logger;
55 import org.slf4j.LoggerFactory;
56 import org.w3c.dom.Document;
57 import org.w3c.dom.Element;
58 import org.w3c.dom.NodeList;
59
60 public class EditConfig extends AbstractSingletonNetconfOperation {
61
62     private static final Logger LOG = LoggerFactory.getLogger(EditConfig.class);
63
64     private static final String OPERATION_NAME = "edit-config";
65     private static final String CONFIG_KEY = "config";
66     private static final String TARGET_KEY = "target";
67     private static final String DEFAULT_OPERATION_KEY = "default-operation";
68     private final CurrentSchemaContext schemaContext;
69     private final TransactionProvider transactionProvider;
70
71     public EditConfig(final String netconfSessionIdForReporting, final CurrentSchemaContext schemaContext,
72             final TransactionProvider transactionProvider) {
73         super(netconfSessionIdForReporting);
74         this.schemaContext = schemaContext;
75         this.transactionProvider = transactionProvider;
76     }
77
78     @Override
79     protected Element handleWithNoSubsequentOperations(final Document document, final XmlElement operationElement)
80             throws DocumentedException {
81         final Datastore targetDatastore = extractTargetParameter(operationElement);
82         if (targetDatastore == Datastore.running) {
83             throw new DocumentedException("edit-config on running datastore is not supported",
84                     ErrorType.PROTOCOL,
85                     ErrorTag.OPERATION_NOT_SUPPORTED,
86                     ErrorSeverity.ERROR);
87         }
88
89         final ModifyAction defaultAction = getDefaultOperation(operationElement);
90
91         final XmlElement configElement = getElement(operationElement, CONFIG_KEY);
92
93         for (final XmlElement element : configElement.getChildElements()) {
94             final String ns = element.getNamespace();
95             final DataSchemaNode schemaNode = getSchemaNodeFromNamespace(ns, element).get();
96
97             final DataTreeChangeTracker changeTracker = new DataTreeChangeTracker(defaultAction);
98
99             parseIntoNormalizedNode(schemaNode, element, changeTracker);
100             executeOperations(changeTracker);
101         }
102
103         return XmlUtil.createElement(document, XmlNetconfConstants.OK, Optional.absent());
104     }
105
106     private void executeOperations(final DataTreeChangeTracker changeTracker) throws DocumentedException {
107         final DOMDataReadWriteTransaction rwTx = transactionProvider.getOrCreateTransaction();
108         final List<DataTreeChange> aa = changeTracker.getDataTreeChanges();
109         final ListIterator<DataTreeChange> iterator = aa.listIterator(aa.size());
110
111         while (iterator.hasPrevious()) {
112             final DataTreeChange dtc = iterator.previous();
113             executeChange(rwTx, dtc);
114         }
115     }
116
117     private void executeChange(final DOMDataReadWriteTransaction rwtx, final DataTreeChange change)
118             throws DocumentedException {
119         final YangInstanceIdentifier path = YangInstanceIdentifier.create(change.getPath());
120         final NormalizedNode<?, ?> changeData = change.getChangeRoot();
121         switch (change.getAction()) {
122             case NONE:
123                 return;
124             case MERGE:
125                 mergeParentMixin(rwtx, path, changeData);
126                 rwtx.merge(LogicalDatastoreType.CONFIGURATION, path, changeData);
127                 break;
128             case CREATE:
129                 try {
130                     final Optional<NormalizedNode<?, ?>> readResult =
131                             rwtx.read(LogicalDatastoreType.CONFIGURATION, path).checkedGet();
132                     if (readResult.isPresent()) {
133                         throw new DocumentedException("Data already exists, cannot execute CREATE operation",
134                             ErrorType.PROTOCOL, ErrorTag.DATA_EXISTS, ErrorSeverity.ERROR);
135                     }
136                     mergeParentMixin(rwtx, path, changeData);
137                     rwtx.put(LogicalDatastoreType.CONFIGURATION, path, changeData);
138                 } catch (final ReadFailedException e) {
139                     LOG.warn("Read from datastore failed when trying to read data for create operation", change, e);
140                 }
141                 break;
142             case REPLACE:
143                 mergeParentMixin(rwtx, path, changeData);
144                 rwtx.put(LogicalDatastoreType.CONFIGURATION, path, changeData);
145                 break;
146             case DELETE:
147                 try {
148                     final Optional<NormalizedNode<?, ?>> readResult =
149                             rwtx.read(LogicalDatastoreType.CONFIGURATION, path).checkedGet();
150                     if (!readResult.isPresent()) {
151                         throw new DocumentedException("Data is missing, cannot execute DELETE operation",
152                             ErrorType.PROTOCOL, ErrorTag.DATA_MISSING, ErrorSeverity.ERROR);
153                     }
154                     rwtx.delete(LogicalDatastoreType.CONFIGURATION, path);
155                 } catch (final ReadFailedException e) {
156                     LOG.warn("Read from datastore failed when trying to read data for delete operation", change, e);
157                 }
158                 break;
159             case REMOVE:
160                 rwtx.delete(LogicalDatastoreType.CONFIGURATION, path);
161                 break;
162             default:
163                 LOG.warn("Unknown/not implemented operation, not executing");
164         }
165     }
166
167     private void mergeParentMixin(final DOMDataReadWriteTransaction rwtx, final YangInstanceIdentifier path,
168                                 final NormalizedNode change) {
169         final YangInstanceIdentifier parentNodeYid = path.getParent();
170         if (change instanceof MapEntryNode) {
171             final SchemaNode schemaNode = SchemaContextUtil.findNodeInSchemaContext(
172                     schemaContext.getCurrentContext(),
173                     parentNodeYid.getPathArguments().stream()
174                             // filter out identifiers not present in the schema tree
175                             .filter(arg -> !(arg instanceof YangInstanceIdentifier.NodeIdentifierWithPredicates))
176                             .filter(arg -> !(arg instanceof YangInstanceIdentifier.AugmentationIdentifier))
177                             .map(YangInstanceIdentifier.PathArgument::getNodeType).collect(Collectors.toList()));
178
179             // we should have the schema node that points to the parent list now, enforce it
180             Preconditions.checkState(schemaNode instanceof ListSchemaNode, "Schema node is not pointing to a list.");
181
182             //merge empty ordered or unordered map
183             if (((ListSchemaNode) schemaNode).isUserOrdered()) {
184                 final MapNode mixinNode = Builders.orderedMapBuilder()
185                         .withNodeIdentifier(
186                                 new YangInstanceIdentifier.NodeIdentifier(
187                                         parentNodeYid.getLastPathArgument().getNodeType()))
188                         .build();
189                 rwtx.merge(LogicalDatastoreType.CONFIGURATION, parentNodeYid, mixinNode);
190                 return;
191             }
192
193             final MapNode mixinNode = Builders.mapBuilder()
194                     .withNodeIdentifier(
195                             new YangInstanceIdentifier.NodeIdentifier(
196                                         parentNodeYid.getLastPathArgument().getNodeType()))
197                     .build();
198             rwtx.merge(LogicalDatastoreType.CONFIGURATION, parentNodeYid, mixinNode);
199         } else if (parentNodeYid.getLastPathArgument() instanceof YangInstanceIdentifier.AugmentationIdentifier) {
200             // merge empty augmentation node
201             final YangInstanceIdentifier.AugmentationIdentifier augmentationYid =
202                 (YangInstanceIdentifier.AugmentationIdentifier) parentNodeYid.getLastPathArgument();
203             final AugmentationNode augmentationNode = Builders.augmentationBuilder()
204                 .withNodeIdentifier(augmentationYid).build();
205             rwtx.merge(LogicalDatastoreType.CONFIGURATION, parentNodeYid, augmentationNode);
206         }
207     }
208
209     @SuppressWarnings("checkstyle:IllegalCatch")
210     private NormalizedNode<?, ?> parseIntoNormalizedNode(final DataSchemaNode schemaNode, final XmlElement element,
211             final DataTreeChangeTracker changeTracker) throws DocumentedException {
212         if (!(schemaNode instanceof ContainerSchemaNode) && !(schemaNode instanceof ListSchemaNode)) {
213             //this should never happen since edit-config on any other node type should not be possible nor makes sense
214             LOG.debug("DataNode from module is not ContainerSchemaNode nor ListSchemaNode, aborting..");
215             throw new UnsupportedOperationException("implement exception if parse fails");
216         }
217
218         final NormalizedNodeResult resultHolder = new NormalizedNodeResult();
219         final NormalizedNodeStreamWriter writer = new EditOperationNormalizedNodeStreamWriter(resultHolder,
220                 changeTracker);
221         final XmlParserStream xmlParser = XmlParserStream.create(writer, schemaContext.getCurrentContext(), schemaNode);
222         try {
223             xmlParser.parse(UntrustedXML.createXMLStreamReader(new StringReader(XmlUtil.toString(element))));
224         } catch (final Exception ex) {
225             throw new NetconfDocumentedException("Error parsing input: " + ex.getMessage(), ex, ErrorType.PROTOCOL,
226                     ErrorTag.MALFORMED_MESSAGE, ErrorSeverity.ERROR);
227         }
228
229         return resultHolder.getResult();
230     }
231
232     private Optional<DataSchemaNode> getSchemaNodeFromNamespace(final String namespace, final XmlElement element)
233             throws DocumentedException {
234         Optional<DataSchemaNode> dataSchemaNode = Optional.absent();
235         try {
236             // returns module with newest revision since findModuleByNamespace returns a set of modules and we only
237             // need the newest one
238             final Module module =
239                     schemaContext.getCurrentContext().findModuleByNamespaceAndRevision(new URI(namespace), null);
240             if (module == null) {
241                 // no module is present with this namespace
242                 throw new NetconfDocumentedException("Unable to find module by namespace: " + namespace,
243                         ErrorType.APPLICATION, ErrorTag.UNKNOWN_NAMESPACE, ErrorSeverity.ERROR);
244             }
245             final DataSchemaNode schemaNode =
246                     module.getDataChildByName(QName.create(module.getQNameModule(), element.getName()));
247             if (schemaNode != null) {
248                 dataSchemaNode = Optional.of(schemaNode);
249             } else {
250                 throw new DocumentedException(
251                         "Unable to find node with namespace: " + namespace + "in module: " + module.toString(),
252                         ErrorType.APPLICATION,
253                         ErrorTag.UNKNOWN_NAMESPACE,
254                         ErrorSeverity.ERROR);
255             }
256         } catch (final URISyntaxException e) {
257             LOG.debug("Unable to create URI for namespace : {}", namespace);
258         }
259
260         return dataSchemaNode;
261     }
262
263     private static Datastore extractTargetParameter(final XmlElement operationElement) throws DocumentedException {
264         final NodeList elementsByTagName = operationElement.getDomElement().getElementsByTagName(TARGET_KEY);
265         // Direct lookup instead of using XmlElement class due to performance
266         if (elementsByTagName.getLength() == 0) {
267             final Map<String, String> errorInfo = ImmutableMap.of("bad-attribute", TARGET_KEY, "bad-element",
268                 OPERATION_NAME);
269             throw new DocumentedException("Missing target element", ErrorType.PROTOCOL, ErrorTag.MISSING_ATTRIBUTE,
270                 ErrorSeverity.ERROR, errorInfo);
271         } else if (elementsByTagName.getLength() > 1) {
272             throw new DocumentedException("Multiple target elements", ErrorType.RPC, ErrorTag.UNKNOWN_ATTRIBUTE,
273                 ErrorSeverity.ERROR);
274         } else {
275             final XmlElement targetChildNode =
276                     XmlElement.fromDomElement((Element) elementsByTagName.item(0)).getOnlyChildElement();
277             return Datastore.valueOf(targetChildNode.getName());
278         }
279     }
280
281     private static ModifyAction getDefaultOperation(final XmlElement operationElement) throws DocumentedException {
282         final NodeList elementsByTagName = operationElement.getDomElement().getElementsByTagName(DEFAULT_OPERATION_KEY);
283         if (elementsByTagName.getLength() == 0) {
284             return ModifyAction.MERGE;
285         } else if (elementsByTagName.getLength() > 1) {
286             throw new DocumentedException("Multiple " + DEFAULT_OPERATION_KEY + " elements", ErrorType.RPC,
287                 ErrorTag.UNKNOWN_ATTRIBUTE, ErrorSeverity.ERROR);
288         } else {
289             return ModifyAction.fromXmlValue(elementsByTagName.item(0).getTextContent());
290         }
291
292     }
293
294     private static XmlElement getElement(final XmlElement operationElement, final String elementName)
295             throws DocumentedException {
296         final Optional<XmlElement> childNode = operationElement.getOnlyChildElementOptionally(elementName);
297         if (!childNode.isPresent()) {
298             throw new DocumentedException(elementName + " element is missing",
299                     ErrorType.PROTOCOL,
300                     ErrorTag.MISSING_ELEMENT,
301                     ErrorSeverity.ERROR);
302         }
303
304         return childNode.get();
305     }
306
307     @Override
308     protected String getOperationName() {
309         return OPERATION_NAME;
310     }
311
312 }