netconf/mdsal-netconf-connector/src/main/java/org/opendaylight/netconf/mdsal/connector/ops/get/AbstractGet.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 package org.opendaylight.netconf.mdsal.connector.ops.get;
   9
  10 import static com.google.common.base.Preconditions.checkArgument;
  11
  12 import com.google.common.annotations.VisibleForTesting;
  13 import java.io.IOException;
  14 import java.util.Optional;
  15 import javax.xml.stream.XMLOutputFactory;
  16 import javax.xml.stream.XMLStreamException;
  17 import javax.xml.stream.XMLStreamWriter;
  18 import javax.xml.transform.dom.DOMResult;
  19 import org.opendaylight.netconf.api.DocumentedException;
  20 import org.opendaylight.netconf.api.xml.XmlElement;
  21 import org.opendaylight.netconf.api.xml.XmlNetconfConstants;
  22 import org.opendaylight.netconf.mdsal.connector.CurrentSchemaContext;
  23 import org.opendaylight.netconf.mdsal.connector.ops.Datastore;
  24 import org.opendaylight.netconf.util.mapping.AbstractSingletonNetconfOperation;
  25 import org.opendaylight.yangtools.yang.common.ErrorSeverity;
  26 import org.opendaylight.yangtools.yang.common.ErrorTag;
  27 import org.opendaylight.yangtools.yang.common.ErrorType;
  28 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  29 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  30 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  31 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
  32 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeWriter;
  33 import org.opendaylight.yangtools.yang.data.api.schema.stream.YangInstanceIdentifierWriter;
  34 import org.opendaylight.yangtools.yang.data.codec.xml.XMLStreamNormalizedNodeStreamWriter;
  35 import org.opendaylight.yangtools.yang.model.api.EffectiveModelContext;
  36 import org.w3c.dom.Document;
  37 import org.w3c.dom.Element;
  38 import org.w3c.dom.Node;
  39
  40 // FIXME: seal when we have JDK17+
  41 abstract class AbstractGet extends AbstractSingletonNetconfOperation {
  42     private static final XMLOutputFactory XML_OUTPUT_FACTORY;
  43     private static final String FILTER = "filter";
  44
  45     static {
  46         XML_OUTPUT_FACTORY = XMLOutputFactory.newFactory();
  47         XML_OUTPUT_FACTORY.setProperty(XMLOutputFactory.IS_REPAIRING_NAMESPACES, true);
  48     }
  49
  50     private final CurrentSchemaContext schemaContext;
  51     private final FilterContentValidator validator;
  52
  53     AbstractGet(final String netconfSessionIdForReporting, final CurrentSchemaContext schemaContext) {
  54         super(netconfSessionIdForReporting);
  55         this.schemaContext = schemaContext;
  56         validator = new FilterContentValidator(schemaContext);
  57     }
  58
  59     // FIXME: throw a DocumentedException
  60     private Node transformNormalizedNode(final Document document, final NormalizedNode data,
  61                                          final YangInstanceIdentifier dataRoot) {
  62         final DOMResult result = new DOMResult(document.createElement(XmlNetconfConstants.DATA_KEY));
  63         final XMLStreamWriter xmlWriter = getXmlStreamWriter(result);
  64         final EffectiveModelContext currentContext = schemaContext.getCurrentContext();
  65
  66         final NormalizedNodeStreamWriter nnStreamWriter = XMLStreamNormalizedNodeStreamWriter.create(xmlWriter,
  67             currentContext);
  68
  69         try {
  70             if (dataRoot.isEmpty()) {
  71                 writeRoot(nnStreamWriter, data);
  72             } else {
  73                 write(nnStreamWriter, currentContext, dataRoot.coerceParent(), data);
  74             }
  75         } catch (IOException e) {
  76             // FIXME: throw DocumentedException
  77             throw new IllegalStateException(e);
  78         }
  79
  80         return result.getNode();
  81     }
  82
  83     private static void write(final NormalizedNodeStreamWriter nnStreamWriter,
  84             final EffectiveModelContext currentContext, final YangInstanceIdentifier parent, final NormalizedNode data)
  85                 throws IOException {
  86         try (var yiidWriter = YangInstanceIdentifierWriter.open(nnStreamWriter, currentContext, parent)) {
  87             try (var nnWriter = NormalizedNodeWriter.forStreamWriter(nnStreamWriter, true)) {
  88                 nnWriter.write(data);
  89             }
  90         }
  91     }
  92
  93     private static void writeRoot(final NormalizedNodeStreamWriter nnStreamWriter, final NormalizedNode data)
  94             throws IOException {
  95         checkArgument(data instanceof ContainerNode, "Unexpected root data %s", data);
  96
  97         try (var nnWriter = NormalizedNodeWriter.forStreamWriter(nnStreamWriter, true)) {
  98             for (var child : ((ContainerNode) data).body()) {
  99                 nnWriter.write(child);
 100             }
 101         }
 102     }
 103
 104     private static XMLStreamWriter getXmlStreamWriter(final DOMResult result) {
 105         try {
 106             return XML_OUTPUT_FACTORY.createXMLStreamWriter(result);
 107         } catch (final XMLStreamException e) {
 108             // FIXME: throw DocumentedException
 109             throw new IllegalStateException(e);
 110         }
 111     }
 112
 113     final Element serializeNodeWithParentStructure(final Document document, final YangInstanceIdentifier dataRoot,
 114                                                    final NormalizedNode node) {
 115         return (Element) transformNormalizedNode(document, node, dataRoot);
 116     }
 117
 118     /**
 119      * Obtain data root according to filter from operation element.
 120      *
 121      * @param operationElement operation element
 122      * @return if filter is present and not empty returns Optional of the InstanceIdentifier to the read location
 123      *      in datastore. Empty filter returns Optional.absent() which should equal an empty &lt;data/&gt;
 124      *      container in the response. If filter is not present we want to read the entire datastore - return ROOT.
 125      * @throws DocumentedException if not possible to get identifier from filter
 126      */
 127     final Optional<YangInstanceIdentifier> getDataRootFromFilter(final XmlElement operationElement)
 128             throws DocumentedException {
 129         final Optional<XmlElement> filterElement = operationElement.getOnlyChildElementOptionally(FILTER);
 130         if (filterElement.isPresent()) {
 131             if (filterElement.get().getChildElements().size() == 0) {
 132                 return Optional.empty();
 133             }
 134             return Optional.of(getInstanceIdentifierFromFilter(filterElement.get()));
 135         }
 136
 137         return Optional.of(YangInstanceIdentifier.empty());
 138     }
 139
 140     @VisibleForTesting
 141     protected final YangInstanceIdentifier getInstanceIdentifierFromFilter(final XmlElement filterElement)
 142             throws DocumentedException {
 143
 144         if (filterElement.getChildElements().size() != 1) {
 145             throw new DocumentedException("Multiple filter roots not supported yet",
 146                     ErrorType.APPLICATION, ErrorTag.OPERATION_NOT_SUPPORTED, ErrorSeverity.ERROR);
 147         }
 148
 149         final XmlElement element = filterElement.getOnlyChildElement();
 150         return validator.validate(element);
 151     }
 152
 153     protected static final class GetConfigExecution {
 154         private final Optional<Datastore> datastore;
 155
 156         GetConfigExecution(final Optional<Datastore> datastore) {
 157             this.datastore = datastore;
 158         }
 159
 160         static GetConfigExecution fromXml(final XmlElement xml, final String operationName) throws DocumentedException {
 161             try {
 162                 validateInputRpc(xml, operationName);
 163             } catch (final DocumentedException e) {
 164                 throw new DocumentedException("Incorrect RPC: " + e.getMessage(), e, e.getErrorType(), e.getErrorTag(),
 165                         e.getErrorSeverity(), e.getErrorInfo());
 166             }
 167
 168             final Optional<Datastore> sourceDatastore;
 169             try {
 170                 sourceDatastore = parseSource(xml);
 171             } catch (final DocumentedException e) {
 172                 throw new DocumentedException("Get-config source attribute error: " + e.getMessage(), e,
 173                         e.getErrorType(), e.getErrorTag(), e.getErrorSeverity(), e.getErrorInfo());
 174             }
 175
 176             return new GetConfigExecution(sourceDatastore);
 177         }
 178
 179         private static Optional<Datastore> parseSource(final XmlElement xml) throws DocumentedException {
 180             final Optional<XmlElement> sourceElement = xml.getOnlyChildElementOptionally(XmlNetconfConstants.SOURCE_KEY,
 181                     XmlNetconfConstants.URN_IETF_PARAMS_XML_NS_NETCONF_BASE_1_0);
 182             return sourceElement.isPresent()
 183                     ? Optional.of(Datastore.valueOf(sourceElement.get().getOnlyChildElement().getName()))
 184                     : Optional.empty();
 185         }
 186
 187         private static void validateInputRpc(final XmlElement xml, final String operationName) throws
 188                 DocumentedException {
 189             xml.checkName(operationName);
 190             xml.checkNamespace(XmlNetconfConstants.URN_IETF_PARAMS_XML_NS_NETCONF_BASE_1_0);
 191         }
 192
 193         public Optional<Datastore> getDatastore() {
 194             return datastore;
 195         }
 196     }
 197 }