LCOV - code coverage report
Current view: top level - src/utils/xml - SUMOSAXReader.cpp (source / functions) Coverage Total Hit
Test: lcov.info Lines: 88.6 % 105 93
Test Date: 2024-11-20 15:55:46 Functions: 91.7 % 12 11

            Line data    Source code
       1              : /****************************************************************************/
       2              : // Eclipse SUMO, Simulation of Urban MObility; see https://eclipse.dev/sumo
       3              : // Copyright (C) 2012-2024 German Aerospace Center (DLR) and others.
       4              : // This program and the accompanying materials are made available under the
       5              : // terms of the Eclipse Public License 2.0 which is available at
       6              : // https://www.eclipse.org/legal/epl-2.0/
       7              : // This Source Code may also be made available under the following Secondary
       8              : // Licenses when the conditions for such availability set forth in the Eclipse
       9              : // Public License 2.0 are satisfied: GNU General Public License, version 2
      10              : // or later which is available at
      11              : // https://www.gnu.org/licenses/old-licenses/gpl-2.0-standalone.html
      12              : // SPDX-License-Identifier: EPL-2.0 OR GPL-2.0-or-later
      13              : /****************************************************************************/
      14              : /// @file    SUMOSAXReader.cpp
      15              : /// @author  Daniel Krajzewicz
      16              : /// @author  Jakob Erdmann
      17              : /// @author  Michael Behrisch
      18              : /// @date    Nov 2012
      19              : ///
      20              : // SAX-reader encapsulation
      21              : /****************************************************************************/
      22              : #include <config.h>
      23              : 
      24              : #include <string>
      25              : #include <memory>
      26              : #include <iostream>
      27              : #include <xercesc/sax2/XMLReaderFactory.hpp>
      28              : #include <xercesc/framework/LocalFileInputSource.hpp>
      29              : #include <xercesc/framework/MemBufInputSource.hpp>
      30              : 
      31              : #include <utils/common/FileHelpers.h>
      32              : #include <utils/common/MsgHandler.h>
      33              : #include <utils/common/ToString.h>
      34              : #include <utils/common/StringUtils.h>
      35              : #include "GenericSAXHandler.h"
      36              : #ifdef HAVE_ZLIB
      37              : #include <foreign/zstr/zstr.hpp>
      38              : #endif
      39              : #include "IStreamInputSource.h"
      40              : #include "SUMOSAXReader.h"
      41              : 
      42              : using XERCES_CPP_NAMESPACE::SAX2XMLReader;
      43              : using XERCES_CPP_NAMESPACE::XMLUni;
      44              : 
      45              : 
      46              : // ===========================================================================
      47              : // method definitions
      48              : // ===========================================================================
      49              : 
      50        99635 : SUMOSAXReader::SUMOSAXReader(GenericSAXHandler& handler, const std::string& validationScheme, XERCES_CPP_NAMESPACE::XMLGrammarPool* grammarPool) :
      51        99635 :     myHandler(nullptr),
      52        99635 :     myValidationScheme(validationScheme),
      53        99635 :     myGrammarPool(grammarPool),
      54        99635 :     myXMLReader(nullptr),
      55              :     myIStream(nullptr),
      56              :     myInputStream(nullptr),
      57        99635 :     mySchemaResolver(true, false),
      58        99635 :     myLocalResolver(false, false),
      59        99635 :     myNoOpResolver(false, true),
      60              :     myNextSection(-1, nullptr) {
      61        99635 :     setHandler(handler);
      62        99635 : }
      63              : 
      64              : 
      65        96842 : SUMOSAXReader::~SUMOSAXReader() {
      66        96842 :     delete myXMLReader;
      67        96842 :     delete myNextSection.second;
      68       193684 : }
      69              : 
      70              : 
      71              : void
      72       197243 : SUMOSAXReader::setHandler(GenericSAXHandler& handler) {
      73       197243 :     myHandler = &handler;
      74       197243 :     if (myXMLReader != nullptr) {
      75        97608 :         myXMLReader->setContentHandler(&handler);
      76        97608 :         myXMLReader->setErrorHandler(&handler);
      77              :     }
      78       197243 : }
      79              : 
      80              : 
      81              : void
      82       187921 : SUMOSAXReader::setValidation(std::string validationScheme) {
      83              :     // The settings ensure that by default (validationScheme "local" or "never") no network access occurs
      84              :     // this is achieved by either resolving no entities at all (myNoOpResolver) or resolving only
      85              :     // to local files (myLocalResolver). Thus we can safely disable the Sonar warnings in the parse methods below.
      86       187921 :     if (myXMLReader != nullptr && validationScheme != myValidationScheme) {
      87       116536 :         if (validationScheme == "") {
      88              :             validationScheme = myValidationScheme;
      89              :         }
      90              :         // see here https://svn.apache.org/repos/asf/xerces/c/trunk/samples/src/SAX2Count/SAX2Count.cpp for the way to set features
      91       116536 :         if (validationScheme == "never") {
      92        81470 :             myXMLReader->setEntityResolver(&myNoOpResolver);
      93        81470 :             myXMLReader->setProperty(XMLUni::fgXercesScannerName, (void*)XMLUni::fgWFXMLScanner);
      94              :         } else {
      95        35066 :             myXMLReader->setEntityResolver(validationScheme == "local" ? &myLocalResolver : &mySchemaResolver);
      96        35066 :             myXMLReader->setProperty(XMLUni::fgXercesScannerName, (void*)XMLUni::fgIGXMLScanner);
      97        35066 :             myXMLReader->setFeature(XMLUni::fgXercesSchema, true);
      98        35066 :             myXMLReader->setFeature(XMLUni::fgSAX2CoreValidation, true);
      99        35066 :             myXMLReader->setFeature(XMLUni::fgXercesDynamic, validationScheme == "local" || validationScheme == "auto");
     100        35066 :             myXMLReader->setFeature(XMLUni::fgXercesUseCachedGrammarInParse, myValidationScheme == "always");
     101              :         }
     102              :     }
     103       187921 :     myValidationScheme = validationScheme;
     104       187921 : }
     105              : 
     106              : 
     107              : void
     108       143213 : SUMOSAXReader::parse(std::string systemID) {
     109       286426 :     if (!FileHelpers::isReadable(systemID)) {
     110            0 :         throw IOError(TLF("Cannot read file '%'!", systemID));
     111              :     }
     112       286426 :     if (FileHelpers::isDirectory(systemID)) {
     113            0 :         throw IOError(TLF("File '%' is a directory!", systemID));
     114              :     }
     115       143213 :     ensureSAXReader();
     116              : #ifdef HAVE_ZLIB
     117       143213 :     zstr::ifstream istream(StringUtils::transcodeToLocal(systemID).c_str(), std::fstream::in | std::fstream::binary);
     118       143213 :     myXMLReader->parse(IStreamInputSource(istream));  // NOSONAR
     119              : #else
     120              :     myXMLReader->parse(StringUtils::transcodeToLocal(systemID).c_str());  // NOSONAR
     121              : #endif
     122       143213 : }
     123              : 
     124              : 
     125              : void
     126            0 : SUMOSAXReader::parseString(std::string content) {
     127            0 :     ensureSAXReader();
     128            0 :     XERCES_CPP_NAMESPACE::MemBufInputSource memBufIS((const XMLByte*)content.c_str(), content.size(), "registrySettings");
     129            0 :     myXMLReader->parse(memBufIS);  // NOSONAR
     130            0 : }
     131              : 
     132              : 
     133              : bool
     134        44721 : SUMOSAXReader::parseFirst(std::string systemID) {
     135        89442 :     if (!FileHelpers::isReadable(systemID)) {
     136           39 :         throw IOError(TLF("Cannot read file '%'!", systemID));
     137              :     }
     138        89416 :     if (FileHelpers::isDirectory(systemID)) {
     139            0 :         throw IOError(TLF("File '%' is a directory!", systemID));
     140              :     }
     141        44708 :     ensureSAXReader();
     142        44708 :     myToken = XERCES_CPP_NAMESPACE::XMLPScanToken();
     143              : #ifdef HAVE_ZLIB
     144        44708 :     myIStream = std::unique_ptr<zstr::ifstream>(new zstr::ifstream(StringUtils::transcodeToLocal(systemID).c_str(), std::fstream::in | std::fstream::binary));
     145        44708 :     myInputStream = std::unique_ptr<IStreamInputSource>(new IStreamInputSource(*myIStream));
     146        44708 :     return myXMLReader->parseFirst(*myInputStream, myToken);  // NOSONAR
     147              : #else
     148              :     return myXMLReader->parseFirst(StringUtils::transcodeToLocal(systemID).c_str(), myToken);  // NOSONAR
     149              : #endif
     150              : }
     151              : 
     152              : 
     153              : bool
     154      3022725 : SUMOSAXReader::parseNext() {
     155      3022725 :     if (myXMLReader == nullptr) {
     156            0 :         throw ProcessError(TL("The XML-parser was not initialized."));
     157              :     }
     158      3022725 :     return myXMLReader->parseNext(myToken);
     159              : }
     160              : 
     161              : 
     162              : bool
     163          461 : SUMOSAXReader::parseSection(SumoXMLTag element) {
     164          461 :     if (myXMLReader == nullptr) {
     165            0 :         throw ProcessError(TL("The XML-parser was not initialized."));
     166              :     }
     167              :     bool started = false;
     168          461 :     if (myNextSection.first != -1) {
     169          287 :         started = myNextSection.first == element;
     170          287 :         if (!started) {
     171              :             // This enforces that the next parsed section starts right after the last one.
     172              :             // If we want to skip sections we need to change this.
     173            2 :             WRITE_WARNINGF("Expected different XML section '%', some content may be missing.", toString(element));
     174              :         }
     175          287 :         myHandler->myStartElement(myNextSection.first, *myNextSection.second);
     176          287 :         delete myNextSection.second;
     177          287 :         myNextSection.first = -1;
     178          287 :         myNextSection.second = nullptr;
     179              :     }
     180          461 :     myHandler->setSection(element, started);
     181      3494793 :     while (!myHandler->sectionFinished()) {
     182      3494505 :         if (!myXMLReader->parseNext(myToken)) {
     183              :             return false;
     184              :         }
     185              :     }
     186              :     myNextSection = myHandler->retrieveNextSectionStart();
     187          288 :     return true;
     188              : }
     189              : 
     190              : 
     191              : void
     192       187921 : SUMOSAXReader::ensureSAXReader() {
     193       187921 :     if (myXMLReader == nullptr) {
     194        99606 :         myXMLReader = XERCES_CPP_NAMESPACE::XMLReaderFactory::createXMLReader(XERCES_CPP_NAMESPACE::XMLPlatformUtils::fgMemoryManager, myGrammarPool);
     195        99606 :         if (myXMLReader == nullptr) {
     196            0 :             throw ProcessError(TL("The XML-parser could not be build."));
     197              :         }
     198        99606 :         setValidation();
     199        99606 :         myXMLReader->setContentHandler(myHandler);
     200        99606 :         myXMLReader->setErrorHandler(myHandler);
     201              :     }
     202       187921 : }
     203              : 
     204              : 
     205       298905 : SUMOSAXReader::LocalSchemaResolver::LocalSchemaResolver(const bool haveFallback, const bool noOp) :
     206       298905 :     myHaveFallback(haveFallback),
     207       298905 :     myNoOp(noOp) {
     208       298905 : }
     209              : 
     210              : 
     211              : XERCES_CPP_NAMESPACE::InputSource*
     212       146731 : SUMOSAXReader::LocalSchemaResolver::resolveEntity(const XMLCh* const /* publicId */, const XMLCh* const systemId) {
     213       146731 :     if (myNoOp) {
     214            0 :         return new XERCES_CPP_NAMESPACE::MemBufInputSource((const XMLByte*)"", 0, "");
     215              :     }
     216       146731 :     const std::string url = StringUtils::transcode(systemId);
     217              :     const std::string::size_type pos = url.find("/xsd/");
     218       146731 :     if (pos != std::string::npos) {
     219        36268 :         const char* sumoPath = std::getenv("SUMO_HOME");
     220              :         // no need for a warning if SUMO_HOME is not set, global preparsing should have done it.
     221        36268 :         if (sumoPath != nullptr) {
     222       108780 :             const std::string file = sumoPath + std::string("/data") + url.substr(pos);
     223        72520 :             if (FileHelpers::isReadable(file)) {
     224        36231 :                 XMLCh* t = XERCES_CPP_NAMESPACE::XMLString::transcode(file.c_str());
     225        36231 :                 XERCES_CPP_NAMESPACE::InputSource* const result = new XERCES_CPP_NAMESPACE::LocalFileInputSource(t);
     226        36231 :                 XERCES_CPP_NAMESPACE::XMLString::release(&t);
     227              :                 return result;
     228              :             } else {
     229           89 :                 WRITE_WARNING("Cannot read local schema '" + file + (myHaveFallback ? "', will try website lookup." : "', XML validation will fail."));
     230              :             }
     231              :         }
     232              :     }
     233       546610 :     if (myHaveFallback || (!StringUtils::startsWith(url, "http:") && !StringUtils::startsWith(url, "https:") && !StringUtils::startsWith(url, "ftp:"))) {
     234              :         return nullptr;
     235              :     }
     236            2 :     return new XERCES_CPP_NAMESPACE::MemBufInputSource((const XMLByte*)"", 0, "");
     237              : }
     238              : 
     239              : 
     240              : /****************************************************************************/
        

Generated by: LCOV version 2.0-1