diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/read_file_command.h | 60 | ||||
-rw-r--r-- | src/utility.h | 48 |
2 files changed, 108 insertions, 0 deletions
diff --git a/src/read_file_command.h b/src/read_file_command.h new file mode 100644 index 0000000..88c20bf --- /dev/null +++ b/src/read_file_command.h @@ -0,0 +1,60 @@ +#include <xalanc/Include/PlatformDefinitions.hpp> +#include <xercesc/util/PlatformUtils.hpp> +#include <xalanc/XalanTransformer/XalanTransformer.hpp> +#include <xalanc/XPath/XObjectFactory.hpp> +#include <xalanc/XPath/Function.hpp> + +#include "utility.h" + +XALAN_USING_XALAN(Function) +XALAN_USING_XALAN(Locator) +XALAN_USING_XALAN(XPathExecutionContext) +XALAN_USING_XALAN(XalanDOMString) +XALAN_USING_XALAN(XalanNode) +XALAN_USING_XALAN(XObjectPtr) +XALAN_USING_XALAN(MemoryManager) +XALAN_USING_XALAN(XalanCopyConstruct) + +class FunctionFileRead : public Function { + public: + virtual XObjectPtr execute( + XPathExecutionContext& executionContext, + XalanNode* context, + const XObjectArgVectorType& args, + const Locator* locator + ) const { + if ( args.size() != 1 ) { + XPathExecutionContext::GetAndReleaseCachedString guard(executionContext); + + generalError(executionContext, context, locator); + } + + xalanc_1_11::CharVectorType tmpFileName; + std::string fileName; + + args[0]->str().transcode(tmpFileName); + std::move(tmpFileName.begin(), tmpFileName.end(), fileName.begin()); + + std::string content(readFile(fileName)); + + return executionContext.getXObjectFactory().createString( + XalanDOMString(content.data()) + ); + } + + virtual FunctionFileRead* clone(MemoryManager& manager) const { + return XalanCopyConstruct(manager, *this); + } + + protected: + const XalanDOMString& getError(XalanDOMString& result) const { + result.assign("The read-file() function expects one argument."); + + return result; + } + + private: + FunctionFileRead& operator=(const FunctionFileRead&); + bool operator==(const FunctionFileRead&) const; + +}; diff --git a/src/utility.h b/src/utility.h new file mode 100644 index 0000000..7050cee --- /dev/null +++ b/src/utility.h @@ -0,0 +1,48 @@ +#include <sys/stat.h> +#include <fcntl.h> +#include <unistd.h> + +#include <cstddef> +#include <cstdio> + +namespace { + +const int OpenFlags = O_RDONLY; +const mode_t OpenMode = S_IRUSR | S_IWUSR; + +} + +std::string readFile(const std::string& path) { + int descriptor( + open(path.data(), OpenFlags, OpenMode) + ); + + if ( descriptor == -1 ) { + close(descriptor); + + return "io error"; + } else { + struct stat info; + fstat(descriptor, &info); + const std::size_t size(info.st_size); + + char* buffer(new char[size]); + + ssize_t readSize(read( + descriptor, + reinterpret_cast<void*>(buffer), + size + )); + + close(descriptor); + + std::string content( + buffer, + readSize + ); + + delete[] buffer; + + return content; + } +} |