ZipCursorTest.cpp - mozsearch

Enable keyboard shortcuts

/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include "gtest/gtest.h"

#include "mozilla/gtest/MozAssertions.h"

#include "nsDirectoryServiceDefs.h"

#include "nsDirectoryServiceUtils.h"

#include "nsXPCOM.h"

#include "nsZipArchive.h"

static void testBug2051664(const nsAString& archiveName, bool shouldSucceed) {

  nsCOMPtr<nsIFile> zipFile;

  ASSERT_NS_SUCCEEDED(NS_GetSpecialDirectory(NS_OS_CURRENT_WORKING_DIR,

                                             getter_AddRefs(zipFile)));

  ASSERT_NS_SUCCEEDED(zipFile->Append(archiveName));

  RefPtr<nsZipArchive> archive = nsZipArchive::OpenArchive(zipFile);

  ASSERT_TRUE(archive);

  nsZipItem* item = archive->GetItem("test.txt"_ns);

  ASSERT_TRUE(item);

  uint32_t bufSize = item->RealSize();

  EXPECT_EQ(bufSize, 14u);

  auto buffer = mozilla::MakeUnique<uint8_t[]>(bufSize);

  ASSERT_TRUE(buffer);

  nsZipCursor cursor(item, archive.get(), buffer.get(), bufSize, true);

  uint32_t bytesRead = 0;

  uint8_t* data = cursor.Copy(&bytesRead);

  if (shouldSucceed) {

    ASSERT_EQ(data, buffer.get());

    ASSERT_EQ(strncmp(reinterpret_cast<const char*>(data), "Hello, world!\n",

                      bufSize),

0);

  } else {

    ASSERT_EQ(data, nullptr);

  EXPECT_EQ(bytesRead, bufSize);

TEST(nsZipCursor, CopyUncompressedEntry)

  testBug2051664(u"bug_2051664_happy.zip"_ns, true);

TEST(nsZipCursor, CopyUncompressedEntryCorrupted)

  testBug2051664(u"bug_2051664_badcrc.zip"_ns, false);