Fixing race conditions in chrome.alarms

chrome/browser/extensions/api/alarms/alarms_api.cc

 // Copyright (c) 2012 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 "chrome/browser/extensions/api/alarms/alarms_api.h"
 
 #include "base/strings/string_number_conversions.h"
 #include "base/time/clock.h"
 #include "base/time/default_clock.h"
 #include "base/values.h"
@@ skipping 89 matching lines @@
 }
 
 bool AlarmsCreateFunction::RunImpl() {
   scoped_ptr<alarms::Create::Params> params(
       alarms::Create::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get());
   const std::string& alarm_name =
       params->name.get() ? *params->name : kDefaultAlarmName;
   std::vector<std::string> warnings;
   if (!ValidateAlarmCreateInfo(
-          alarm_name, params->alarm_info, GetExtension(), &error_, &warnings))
+          alarm_name, params->alarm_info, GetExtension(), &error_, &warnings)) {
     return false;
+  }
   for (std::vector<std::string>::const_iterator it = warnings.begin();
        it != warnings.end(); ++it)
     WriteToConsole(content::CONSOLE_MESSAGE_LEVEL_WARNING, *it);
 
   Alarm alarm(alarm_name,
               params->alarm_info,
               base::TimeDelta::FromMinutes(
                   Manifest::IsUnpackedLocation(GetExtension()->location()) ?
                   kDevDelayMinimum : kReleaseDelayMinimum),
               clock_->Now());
-  AlarmManager::Get(profile())->AddAlarm(extension_id(), alarm);
+  AlarmManager::Get(profile())->AddAlarm(extension_id(), alarm, base::Bind(
+      &AlarmsCreateFunction::Callback, this));
 
   return true;
 }
 
+void AlarmsCreateFunction::Callback() {
+  SendResponse(true);
+}
+
 bool AlarmsGetFunction::RunImpl() {
   scoped_ptr<alarms::Get::Params> params(alarms::Get::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get());
 
   std::string name = params->name.get() ? *params->name : kDefaultAlarmName;
-  const Alarm* alarm =
-      AlarmManager::Get(profile())->GetAlarm(extension_id(), name);
+  AlarmManager::Get(profile())->GetAlarm(extension_id(), name, base::Bind(
+      &AlarmsGetFunction::Callback, this, name));
 
-  if (!alarm) {
-    error_ = ErrorUtils::FormatErrorMessage(kAlarmNotFound, name);
-    return false;
-  }
-
-  results_ = alarms::Get::Results::Create(*alarm->js_alarm);
   return true;
 }
 
+void AlarmsGetFunction::Callback(
+    const std::string& name, extensions::Alarm* alarm) {
+  if (alarm) {
+    results_ = alarms::Get::Results::Create(*alarm->js_alarm);
+    SendResponse(true);
+  } else {
+    error_ = ErrorUtils::FormatErrorMessage(kAlarmNotFound, name);
+    SendResponse(false);
+  }
+}
+
 bool AlarmsGetAllFunction::RunImpl() {
-  const AlarmManager::AlarmList* alarms =
-      AlarmManager::Get(profile())->GetAllAlarms(extension_id());
+  AlarmManager::Get(profile())->GetAllAlarms(extension_id(), base::Bind(
+      &AlarmsGetAllFunction::Callback, this));
+  return true;
+}
+
+void AlarmsGetAllFunction::Callback(const AlarmList* alarms) {
   if (alarms) {
     std::vector<linked_ptr<extensions::api::alarms::Alarm> > create_arg;
     create_arg.reserve(alarms->size());
     for (size_t i = 0, size = alarms->size(); i < size; ++i) {
       create_arg.push_back((*alarms)[i].js_alarm);
     }
     results_ = alarms::GetAll::Results::Create(create_arg);
   } else {
     SetResult(new base::ListValue());
   }
-  return true;
+  SendResponse(true);
 }
 
 bool AlarmsClearFunction::RunImpl() {
   scoped_ptr<alarms::Clear::Params> params(
       alarms::Clear::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get());
 
   std::string name = params->name.get() ? *params->name : kDefaultAlarmName;
-  bool success = AlarmManager::Get(profile())->RemoveAlarm(extension_id(),
-                                                           name);
-
-  if (!success) {
-    error_ = ErrorUtils::FormatErrorMessage(kAlarmNotFound, name);
-    return false;
-  }
+  AlarmManager::Get(profile())->RemoveAlarm(extension_id(), name, base::Bind(
+      &AlarmsClearFunction::Callback, this, name));
 
   return true;
 }
 
+void AlarmsClearFunction::Callback(const std::string& name, bool success) {
+  if (!success)
+    error_ = ErrorUtils::FormatErrorMessage(kAlarmNotFound, name);
+
+  SendResponse(success);
+}
+
 bool AlarmsClearAllFunction::RunImpl() {
-  AlarmManager::Get(profile())->RemoveAllAlarms(extension_id());
+  AlarmManager::Get(profile())->RemoveAllAlarms(extension_id(), base::Bind(
+      &AlarmsClearAllFunction::Callback, this));
   return true;
 }
 
+void AlarmsClearAllFunction::Callback() {
+  SendResponse(true);
+}
+
 }  // namespace extensions