New Zip::ZipFromMemory API.

third_party/zlib/google/zip_reader_unittest.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "third_party/zlib/google/zip_reader.h"
 
 #include <set>
 #include <string>
 
 #include "base/bind.h"
 #include "base/file_util.h"
 #include "base/files/scoped_temp_dir.h"
 #include "base/logging.h"
 #include "base/md5.h"
+#include "base/memory/ref_counted_memory.h"
 #include "base/path_service.h"
 #include "base/platform_file.h"
 #include "base/run_loop.h"
+#include "base/strings/stringprintf.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/time/time.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/platform_test.h"
 #include "third_party/zlib/google/zip_internal.h"
 
 namespace {
 
 const static std::string kQuuxExpectedMD5 = "d1ae4ac8a17a0e09317113ab284b57a6";
 
@@ skipping 35 matching lines @@
 
  private:
   base::PlatformFile file_;
 };
 
 // A mock that provides methods that can be used as callbacks in asynchronous
 // unzip functions.  Tracks the number of calls and number of bytes reported.
 // Assumes that progress callbacks will be executed in-order.
 class MockUnzipListener : public base::SupportsWeakPtr<MockUnzipListener> {
  public:
-  MockUnzipListener() 
+  MockUnzipListener()
       : success_calls_(0),
         failure_calls_(0),
         progress_calls_(0),
         current_progress_(0) {
   }
 
   // Success callback for async functions.
   void OnUnzipSuccess() {
     success_calls_++;
   }
@@ skipping 462 matching lines @@
 
   base::RunLoop().RunUntilIdle();
 
   EXPECT_EQ(1, listener.success_calls());
   EXPECT_EQ(0, listener.failure_calls());
   EXPECT_GE(0, listener.progress_calls());
 
   ASSERT_TRUE(base::DirectoryExists(target_file));
 }
 
+TEST_F(ZipReaderTest, ExtractCurrentEntryToRefCountedMemory) {
+  // test_mismatch_size.zip contains files with names from 0.txt to 7.txt with
+  // sizes from 0 to 7 bytes respectively, but the metadata in the zip file says
+  // the uncompressed size is 3 bytes. The ZipReader code needs to be clever
+  // enough to get all the data out.
+  base::FilePath test_zip_file =
+      test_data_dir_.AppendASCII("test_mismatch_size.zip");
+
+  ZipReader reader;
+  scoped_refptr<base::RefCountedMemory> contents;
+  ASSERT_TRUE(reader.Open(test_zip_file));
+
+  for (size_t k = 0; k < 8; k++) {
+    SCOPED_TRACE(base::StringPrintf("<loop:%d>", static_cast<int>(k)));
+
+    base::FilePath file_name = base::FilePath(
+        base::StringPrintf(FILE_PATH_LITERAL("%d.txt"), static_cast<int>(k)));
+    ASSERT_TRUE(reader.LocateAndOpenEntry(file_name));
+    EXPECT_EQ(3, reader.current_entry_info()->original_size());
+
+    if (k > 1) {
+      // Off by one byte read limit: must fail.
+      EXPECT_FALSE(reader.ExtractCurrentEntryToRefCountedMemory(
+          k - 1, &contents));
+    }
+
+    if (k > 0) {
+      // Exact byte read limit: must pass.
+      EXPECT_TRUE(reader.ExtractCurrentEntryToRefCountedMemory(k, &contents));
+      EXPECT_EQ(k, contents->size());
+      EXPECT_EQ(0, memcmp(contents->front(), "0123456", k));
+    }
+
+    // More than necessary byte read limit: must pass.
+    EXPECT_TRUE(reader.ExtractCurrentEntryToRefCountedMemory(16, &contents));
+    EXPECT_EQ(k, contents->size());
+    EXPECT_EQ(0, memcmp(contents->front(), "0123456", k));
+  }
+  reader.Close();
+}
+
 }  // namespace zip