Extend the IOBuffer to the disk cache.

net/disk_cache/stress_cache.cc

 // Copyright (c) 2006-2008 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.
 
 // This is a simple application that stress-tests the crash recovery of the disk
 // cache. The main application starts a copy of itself on a loop, checking the
 // exit code of the child process. When the child dies in an unexpected way,
 // the main application quits.
 
 // The child application has two threads: one to exercise the cache in an
 // infinite loop, and another one to asynchronously kill the process.
 
 #include <string>
 #include <vector>
 
 #include "base/at_exit.h"
 #include "base/command_line.h"
 #include "base/debug_util.h"
 #include "base/logging.h"
 #include "base/message_loop.h"
 #include "base/path_service.h"
 #include "base/platform_thread.h"
 #include "base/process_util.h"
 #include "base/string_util.h"
 #include "base/thread.h"
+#include "net/base/io_buffer.h"
 #include "net/disk_cache/disk_cache.h"
 #include "net/disk_cache/disk_cache_test_util.h"
 
 using base::Time;
 
 const int kError = -1;
 const int kExpectedCrash = 100;
 
 // Starts a new process.
 int RunSlave(int iteration) {
@@ skipping 53 matching lines @@
 
   const int kNumKeys = 5000;
   const int kNumEntries = 30;
   std::string keys[kNumKeys];
   disk_cache::Entry* entries[kNumEntries] = {0};
 
   for (int i = 0; i < kNumKeys; i++) {
     keys[i] = GenerateKey(true);
   }
 
-  const int kDataLen = 4000;
-  char data[kDataLen];
-  memset(data, 'k', kDataLen);
+  const int kSize = 4000;
+  scoped_refptr<net::IOBuffer> buffer = new net::IOBuffer(kSize);
+  memset(buffer->data(), 'k', kSize);
 
   for (int i = 0;; i++) {
     int slot = rand() % kNumEntries;
     int key = rand() % kNumKeys;
 
     if (entries[slot])
       entries[slot]->Close();
 
     if (!cache->OpenEntry(keys[key], &entries[slot]))
       CHECK(cache->CreateEntry(keys[key], &entries[slot]));
 
-    base::snprintf(data, kDataLen, "%d %d", iteration, i);
-    CHECK(kDataLen == entries[slot]->WriteData(0, 0, data, kDataLen, NULL,
-                                               false));
+    base::snprintf(buffer->data(), kSize, "%d %d", iteration, i);
+    CHECK(kSize == entries[slot]->WriteData(0, 0, buffer, kSize, NULL, false));
 
     if (rand() % 100 > 80) {
       key = rand() % kNumKeys;
       cache->DoomEntry(keys[key]);
     }
 
     if (!(i % 100))
       printf("Entries: %d    \r", i);
   }
 }
@@ skipping 55 matching lines @@
   // Setup an AtExitManager so Singleton objects will be destructed.
   base::AtExitManager at_exit_manager; 
 
   if (argc < 2)
     return MasterCode();
 
   logging::SetLogAssertHandler(CrashHandler);
 
   // Some time for the memory manager to flush stuff.
   PlatformThread::Sleep(3000);
-  MessageLoop message_loop;
+  MessageLoop message_loop(MessageLoop::TYPE_IO);
 
   char* end;
   long int iteration = strtol(argv[1], &end, 0);
 
   if (!StartCrashThread()) {
     printf("failed to start thread\n");
     return kError;
   }
 
   StressTheCache(iteration);
   return 0;
 }