Disk cache: Protect the code against misuse......

net/disk_cache/in_flight_io.cc

 // Copyright (c) 2006-2010 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 "net/disk_cache/in_flight_io.h"
 
 #include "base/logging.h"
 
 namespace disk_cache {
 
@@ skipping 19 matching lines @@
   while (!io_list_.empty()) {
     // Block the current thread until all pending IO completes.
     IOList::iterator it = io_list_.begin();
     InvokeCallback(*it, true);
   }
 }
 
 // Runs on a background thread.
 void InFlightIO::OnIOComplete(BackgroundIO* operation) {
 #ifndef NDEBUG
-  if (callback_thread_ == MessageLoop::current()) {
+  if (callback_thread_->BelongsToCurrentThread()) {
     DCHECK(single_thread_ || !running_);
     single_thread_ = true;
   }
   running_ = true;
 #endif
 
   callback_thread_->PostTask(FROM_HERE,
                              NewRunnableMethod(operation,
                                                &BackgroundIO::OnIOSignalled));
   operation->io_completed()->Signal();
 }
 
 // Runs on the primary thread.
 void InFlightIO::InvokeCallback(BackgroundIO* operation, bool cancel_task) {
   operation->io_completed()->Wait();
 
   if (cancel_task)
     operation->Cancel();
 
   // Make sure that we remove the operation from the list before invoking the
   // callback (so that a subsequent cancel does not invoke the callback again).
   DCHECK(io_list_.find(operation) != io_list_.end());
   io_list_.erase(operation);
   OnOperationComplete(operation, cancel_task);
 }
 
 // Runs on the primary thread.
 void InFlightIO::OnOperationPosted(BackgroundIO* operation) {
-  DCHECK(callback_thread_ == MessageLoop::current());
+  DCHECK(callback_thread_->BelongsToCurrentThread());
   io_list_.insert(operation);
 }
 
 }  // namespace disk_cache