include/oox/ole/vbaexport.hxx

   1 /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
   2 /*
   3  * This file is part of the LibreOffice project.
   4  *
   5  * This Source Code Form is subject to the terms of the Mozilla Public
   6  * License, v. 2.0. If a copy of the MPL was not distributed with this
   7  * file, You can obtain one at http://mozilla.org/MPL/2.0/.
   8  */
   9
  10 #ifndef INCLUDED_OOX_OLE_VBAEXPORT_HXX
  11 #define INCLUDED_OOX_OLE_VBAEXPORT_HXX
  12
  13 #include <cstddef>
  14
  15 #include <com/sun/star/uno/Reference.hxx>
  16 #include <oox/dllapi.h>
  17 #include <rtl/ustring.hxx>
  18 #include <sal/types.h>
  19
  20 class SotStorage;
  21 class SvStream;
  22 class SvMemoryStream;
  23
  24 namespace com::sun::star {
  25     namespace container { class XNameContainer; }
  26     namespace frame { class XModel; }
  27     namespace script { class XLibraryContainer; }
  28 }
  29
  30 class OOX_DLLPUBLIC VbaExport
  31 {
  32 public:
  33     VbaExport(css::uno::Reference<css::frame::XModel> xModel);
  34
  35     void exportVBA(SotStorage* pRootStorage);
  36
  37     bool containsVBAProject();
  38
  39 private:
  40
  41     css::uno::Reference<css::container::XNameContainer>
  42         getBasicLibrary() const;
  43
  44     css::uno::Reference<css::script::XLibraryContainer>
  45         getLibraryContainer() const;
  46
  47     OUString getProjectName() const;
  48     rtl_TextEncoding getVBATextEncoding() const;
  49
  50     css::uno::Reference<css::frame::XModel> mxModel;
  51 };
  52
  53 class VBACompressionChunk
  54 {
  55 public:
  56
  57     VBACompressionChunk(SvStream& rCompressedStream, const sal_uInt8* pData, std::size_t nChunkSize);
  58
  59     void write();
  60
  61 private:
  62     SvStream& mrCompressedStream;
  63     const sal_uInt8* mpUncompressedData;
  64     sal_uInt8* mpCompressedChunkStream;
  65
  66     // same as DecompressedChunkEnd in the spec
  67     std::size_t mnChunkSize;
  68
  69     // CompressedCurrent according to the spec
  70     sal_uInt64 mnCompressedCurrent;
  71
  72     // CompressedEnd according to the spec
  73     sal_uInt64 mnCompressedEnd;
  74
  75     // DecompressedCurrent according to the spec
  76     sal_uInt64 mnDecompressedCurrent;
  77
  78     // DecompressedEnd according to the spec
  79     sal_uInt64 mnDecompressedEnd;
  80
  81     static void PackCompressedChunkSize(size_t nSize, sal_uInt16& rHeader);
  82
  83     static void PackCompressedChunkFlag(bool bCompressed, sal_uInt16& rHeader);
  84
  85     static void PackCompressedChunkSignature(sal_uInt16& rHeader);
  86
  87     void compressTokenSequence();
  88
  89     void compressToken(size_t index, sal_uInt8& nFlagByte);
  90
  91     static void SetFlagBit(size_t index, bool bVal, sal_uInt8& rFlag);
  92
  93     sal_uInt16 CopyToken(size_t nLength, size_t nOffset);
  94
  95     void match(size_t& rLength, size_t& rOffset);
  96
  97     void CopyTokenHelp(sal_uInt16& rLengthMask, sal_uInt16& rOffsetMask,
  98             sal_uInt16& rBitCount, sal_uInt16& rMaximumLength);
  99
 100     void writeRawChunk();
 101
 102     sal_uInt16 handleHeader(bool bCompressed);
 103 };
 104
 105 class OOX_DLLPUBLIC VBACompression
 106 {
 107 public:
 108     VBACompression(SvStream& rCompressedStream,
 109             SvMemoryStream& rUncompressedStream);
 110
 111     void write();
 112
 113 private:
 114     SvStream& mrCompressedStream;
 115     SvMemoryStream& mrUncompressedStream;
 116 };
 117
 118 class OOX_DLLPUBLIC VBAEncryption
 119 {
 120 public:
 121     VBAEncryption(const sal_uInt8* pData,
 122                   const sal_uInt16 nLength,
 123                   SvStream& rEncryptedData,
 124                   sal_uInt8 nProjKey,
 125                   rtl_TextEncoding eTextEncoding);
 126
 127     void write();
 128
 129     static sal_uInt8 calculateProjKey(const OUString& rString);
 130
 131 private:
 132     const sal_uInt8* mpData; // an array of bytes to be obfuscated
 133     const sal_uInt16 mnLength; // the length of Data
 134     SvStream& mrEncryptedData; // Encrypted Data Structure
 135     sal_uInt8 mnUnencryptedByte1; // the last unencrypted byte read or written
 136     sal_uInt8 mnEncryptedByte1; // the last encrypted byte read or written
 137     sal_uInt8 mnEncryptedByte2; // the next-to-last encrypted byte read or written
 138     sal_Unicode mnProjKey; // a project-specific encryption key
 139     sal_uInt8 mnIgnoredLength; // the length in bytes of IgnoredEnc
 140
 141     sal_uInt8 mnSeed; // the seed value
 142     sal_uInt8 mnVersionEnc; // the version encoding
 143     rtl_TextEncoding meTextEncoding; // the VBA text encoding on export
 144
 145     void writeSeed();
 146     void writeVersionEnc();
 147     void writeProjKeyEnc();
 148     void writeIgnoredEnc();
 149     void writeDataLengthEnc();
 150     void writeDataEnc();
 151 };
 152
 153 #endif
 154
 155 /* vim:set shiftwidth=4 softtabstop=4 expandtab: */