public void DecompressionProducesExpectedOutput() { var decompressed = VbaCompression.Decompress(_expectedCompressedBytes); Assert.Equal(_expectedDecompressedBytes.Length, decompressed.Length); Assert.True(_expectedDecompressedBytes.SequenceEqual(decompressed)); }
public void CompressionProducesExpectedOutput() { var compressed = VbaCompression.Compress(_compressionInputBytes); Assert.Equal(_expectedCompressedBytes.Length, compressed.Length); Assert.True(_expectedCompressedBytes.SequenceEqual(compressed)); }
public void GivenLargeByteSequenceWithLowCompressibilityCompressingAndDecompressionProducesSameInput() { var data = GetLargeByteSequenceWithLowCompressibility().ToArray(); var compressedData = VbaCompression.Compress(data); var convertedData = VbaCompression.Decompress(compressedData); Assert.True(data != convertedData); Assert.True(data.LongLength == convertedData.LongLength); Assert.True(data.SequenceEqual(convertedData)); }
/** * Create VBA Project from file * * @param sPath path of macro enabled Excel, Powerpoint or Word file. */ public VBAProject(string sPath) { //Initialize pPath pPath = sPath; //Get unzip location unzipTempPath = CreateUniqueTempDirectory(); /* EvilClippy brute forces the file open, It does this in a fairly unclean manner. We will assume the file, * is named correctly (extension wise). This, although requiring more boiler plate will look a lot cleaner. */ // Get extension of file: string ext = Path.GetExtension(sPath); //Parse extension and create sOLEFileName switch (ext) { case ".xlsm": case ".xlam": isOpenXML = true; sOLEFileName = Path.Combine(unzipTempPath, "xl", "vbaProject.bin"); break; case ".docm": isOpenXML = true; sOLEFileName = Path.Combine(unzipTempPath, "word", "vbaProject.bin"); break; case ".pptm": isOpenXML = true; sOLEFileName = Path.Combine(unzipTempPath, "ppt", "vbaProject.bin"); //untested break; case ".xls": case ".doc": case ".ppt": //Copy path to sOLEFileName to prevent overwriting (we'll overrite this file directly) sOLEFileName = Path.Combine(unzipTempPath, Path.GetFileName(sPath)); File.Copy(sPath, sOLEFileName); break; default: Console.WriteLine("ERROR: Could not open file " + sPath); Console.WriteLine("Please make sure this file exists, has a valid extension and is of a valid type."); Console.WriteLine(); break; } //Unzip to unzipTemoPath if isOpenXML. Otherwise create compound file. if (isOpenXML) { ZipFile.ExtractToDirectory(sPath, unzipTempPath); } //Create Compound file from VBProject.bin or xls,doc,ppt file: cf = new CompoundFile(sOLEFileName, CFSUpdateMode.Update, 0); // Read relevant streams switch (ext) { case ".xlsm": case ".xlam": case ".docm": case ".pptm": commonStorage = cf.RootStorage; break; case ".doc": commonStorage = cf.RootStorage.GetStorage("Macros"); //Note you can also use `cf.RootStorage.TryGetStorage("Macros")` which returns null if not found. break; case ".ppt": //untested - don't know if this is the case for ppts case ".xls": commonStorage = cf.RootStorage.GetStorage("_VBA_PROJECT_CUR"); break; } vbaProjectStream = commonStorage.GetStorage("VBA").GetStream("_VBA_PROJECT").GetData(); //Contains VBA Project properties, including module names. Remove module names from this stream to hide them. //PROJECT stream: https://docs.microsoft.com/en-us/openspecs/office_file_formats/ms-ovba/cc848a02-6f87-49a4-ad93-6edb3103f593 //More information: https://docs.microsoft.com/en-us/openspecs/office_file_formats/ms-ovba/d88cb9d8-a475-423d-b370-cc0caaf78628 projectStream = commonStorage.GetStream("project").GetData(); //PROJECTwm stream contains all names of all Modules. //PROJECTwm stream: https://docs.microsoft.com/en-us/openspecs/office_file_formats/ms-ovba/514acc65-ea7b-4813-aaf7-fabb1bca0ba2 //More Information: https://docs.microsoft.com/en-us/openspecs/office_file_formats/ms-ovba/c458f2e6-f2cc-4c2d-96c7-91a3e63f2fe1 projectwmStream = commonStorage.GetStream("PROJECTwm").GetData(); //Contains names of visible modules and module source code text (compressed) //VBA Storage: https://docs.microsoft.com/en-us/openspecs/office_file_formats/ms-ovba/b693e0ba-489f-4ac8-ac9d-6387fb5779bb //VBA StorageInfo: https://docs.microsoft.com/en-us/openspecs/office_file_formats/ms-ovba/170f52a0-4cd6-4729-b51a-d08155cbd213 //Dir Stream: https://docs.microsoft.com/en-us/openspecs/office_file_formats/ms-ovba/672d276c-d83c-4452-993b-ca9eca3d8917 dirStream = VbaCompression.Decompress(commonStorage.GetStorage("VBA").GetStream("dir").GetData()); // Read project streams as string string projectStreamString = System.Text.Encoding.UTF8.GetString(projectStream); string projectwmStreamString = System.Text.Encoding.UTF8.GetString(projectwmStream); Console.WriteLine(System.Text.Encoding.UTF8.GetString(projectStream)); Console.WriteLine(System.Text.Encoding.UTF8.GetString(projectwmStream)); // Find all VBA modules in current file vbaModulesEx = ParseModulesFromDirStream(dirStream); // Write streams to debug log (if verbosity enabled) Console.WriteLine("Hex dump of original _VBA_PROJECT stream:\n" + Utils.HexDump(vbaProjectStream)); Console.WriteLine("Hex dump of original dir stream:\n" + Utils.HexDump(dirStream)); }