Implement chrome.bluetoothLowEnergy.resetAdvertising().

extensions/browser/api/bluetooth_low_energy/bluetooth_low_energy_api.cc

 // Copyright 2014 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 "extensions/browser/api/bluetooth_low_energy/bluetooth_low_energy_api.h"
 
 #include <stdint.h>
 #include <algorithm>
 #include <iterator>
 #include <utility>
@@ skipping 1071 matching lines @@
   DCHECK(advertisements_manager_);
   return advertisements_manager_->Get(extension_id(), advertisement_id);
 }
 
 void BluetoothLowEnergyAdvertisementFunction::RemoveAdvertisement(
     int advertisement_id) {
   DCHECK(advertisements_manager_);
   advertisements_manager_->Remove(extension_id(), advertisement_id);
 }
 
+const base::hash_set<int>*
+BluetoothLowEnergyAdvertisementFunction::GetAdvertisementIds() {
+  return advertisements_manager_->GetResourceIds(extension_id());
+}
+
 bool BluetoothLowEnergyAdvertisementFunction::RunAsync() {
   Initialize();
+
+  // Check permission in the manifest.
+  if (!BluetoothManifestData::CheckPeripheralPermitted(extension())) {
+    SetError(kErrorPermissionDenied);
+    return false;
+  }
+
+  // For advertisement API to be available the app has to be either auto
+  // launched in Kiosk Mode or the enable-ble-advertisement-in-apps
+  // should be set.
+  if (!(IsAutoLaunchedKioskApp(extension()->id()) ||
+        IsPeripheralFlagEnabled())) {
+    SetError(kErrorPermissionDenied);
+    return false;
+  }
+
   return BluetoothLowEnergyExtensionFunctionDeprecated::RunAsync();
 }
 
 void BluetoothLowEnergyAdvertisementFunction::Initialize() {
   advertisements_manager_ =
       ApiResourceManager<BluetoothApiAdvertisement>::Get(browser_context());
 }
 
 // RegisterAdvertisement:
 
 bool BluetoothLowEnergyRegisterAdvertisementFunction::DoWork() {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 
-  // Check permissions in manifest.
-  if (!BluetoothManifestData::CheckPeripheralPermitted(extension())) {
-    error_ = kErrorPermissionDenied;
-    SendResponse(false);
-    return false;
-  }
-
-  // For this API to be available the app has to be either auto
-  // launched in Kiosk Mode or the enable-ble-advertisement-in-apps
-  // should be set.
-  if (!(IsAutoLaunchedKioskApp(extension()->id()) ||
-        IsPeripheralFlagEnabled())) {
-    error_ = kErrorPermissionDenied;
-    SendResponse(false);
-    return false;
-  }
-
   BluetoothLowEnergyEventRouter* event_router =
       GetEventRouter(browser_context());
 
   // The adapter must be initialized at this point, but return an error instead
   // of asserting.
   if (!event_router->HasAdapter()) {
     SetError(kErrorAdapterNotInitialized);
     SendResponse(false);
     return false;
   }
@@ skipping 58 matching lines @@
       SetError(kErrorOperationFailed);
   }
   SendResponse(false);
 }
 
 // UnregisterAdvertisement:
 
 bool BluetoothLowEnergyUnregisterAdvertisementFunction::DoWork() {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 
-  // Check permission in the manifest.
-  if (!BluetoothManifestData::CheckPeripheralPermitted(extension())) {
-    error_ = kErrorPermissionDenied;
-    SendResponse(false);
-    return false;
-  }
-
-  // For this API to be available the app has to be either auto
-  // launched in Kiosk Mode or the enable-ble-advertisement-in-apps
-  // should be set.
-  if (!(IsAutoLaunchedKioskApp(extension()->id()) ||
-        IsPeripheralFlagEnabled())) {
-    error_ = kErrorPermissionDenied;
-    SendResponse(false);
-    return false;
-  }
-
   BluetoothLowEnergyEventRouter* event_router =
       GetEventRouter(browser_context());
 
   // If we don't have an initialized adapter, unregistering is a no-op.
   if (!event_router->HasAdapter())
     return true;
 
   std::unique_ptr<apibtle::UnregisterAdvertisement::Params> params(
       apibtle::UnregisterAdvertisement::Params::Create(*args_));
   EXTENSION_FUNCTION_VALIDATE(params.get() != NULL);
@@ skipping 31 matching lines @@
     case device::BluetoothAdvertisement::ErrorCode::
         ERROR_ADVERTISEMENT_DOES_NOT_EXIST:
       SetError(kStatusAdvertisementDoesNotExist);
       break;
     default:
       SetError(kErrorOperationFailed);
   }
   SendResponse(false);
 }
 
+// ResetAdvertising:
+
+bool BluetoothLowEnergyResetAdvertisingFunction::DoWork() {
+  DCHECK_CURRENTLY_ON(BrowserThread::UI);
+
+  BluetoothLowEnergyEventRouter* event_router =
+      GetEventRouter(browser_context());
+
+  // If the adapter is not initialized, there is nothing to reset.
+  if (!event_router->HasAdapter()) {
+    SendResponse(true);
+    return true;
+  }
+
+  const base::hash_set<int>* advertisement_ids = GetAdvertisementIds();
+  if (!advertisement_ids || advertisement_ids->empty()) {
+    SendResponse(true);
+    return true;
+  }
+
+  // Copy the hash set, as RemoveAdvertisement can change advertisement_ids
+  // while we are iterating on it.

[tbarzic, 2017/06/27 22:49:12]

nit: s/iterating on/iterating over/

[Sonny Sasaka, 2017/06/27 23:14:01]

Done.

+  base::hash_set<int> advertisement_ids_tmp = *advertisement_ids;
+  for (int advertisement_id : advertisement_ids_tmp) {
+    RemoveAdvertisement(advertisement_id);
+  }
+
+  event_router->adapter()->ResetAdvertising(
+      base::Bind(&BluetoothLowEnergyResetAdvertisingFunction::SuccessCallback,
+                 this),
+      base::Bind(&BluetoothLowEnergyResetAdvertisingFunction::ErrorCallback,
+                 this));
+
+  return true;
+}
+
+void BluetoothLowEnergyResetAdvertisingFunction::SuccessCallback() {
+  SendResponse(true);
+}
+
+void BluetoothLowEnergyResetAdvertisingFunction::ErrorCallback(
+    device::BluetoothAdvertisement::ErrorCode status) {
+#if defined(OS_CHROMEOS) || defined(OS_LINUX)

[tbarzic, 2017/06/27 22:49:12]

no ifdefs?

[Sonny Sasaka, 2017/06/27 23:14:01]

Done.

+  error_ = kErrorOperationFailed;
+  SendResponse(false);
+#endif
+}
+
 // SetAdvertisingInterval:
 
 template class BLEPeripheralExtensionFunction<
     apibtle::SetAdvertisingInterval::Params>;
 
 void BluetoothLowEnergySetAdvertisingIntervalFunction::DoWork() {
 #if defined(OS_CHROMEOS) || defined(OS_LINUX)
   BluetoothLowEnergyEventRouter* event_router =
       GetEventRouter(browser_context());
   event_router->adapter()->SetAdvertisingInterval(
@@ skipping 216 matching lines @@
                         params_->response.value->end());
   }
   event_router_->HandleRequestResponse(
       extension(), params_->response.request_id, params_->response.is_error,
       uint8_vector);
   Respond(NoArguments());
 }
 
 }  // namespace api
 }  // namespace extensions