Enable lost URL detection in the Physical Web scanner

ios/chrome/common/physical_web/physical_web_scanner.mm

 // Copyright 2015 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.
 
 #import "ios/chrome/common/physical_web/physical_web_scanner.h"
 
 #import <CoreBluetooth/CoreBluetooth.h>
 
 #include <string>
 #include <vector>
 
 #import "base/ios/weak_nsobject.h"
 #include "base/logging.h"
 #import "base/mac/scoped_nsobject.h"
 #include "base/memory/ptr_util.h"
 #include "base/strings/sys_string_conversions.h"
 #include "base/values.h"
 #include "device/bluetooth/uribeacon/uri_encoder.h"
 #import "ios/chrome/common/physical_web/physical_web_device.h"
 #import "ios/chrome/common/physical_web/physical_web_request.h"
 #import "ios/chrome/common/physical_web/physical_web_types.h"
 
 namespace {
 
 NSString* const kUriBeaconServiceUUID = @"FED8";
 NSString* const kEddystoneBeaconServiceUUID = @"FEAA";
 
+// The length of time in seconds since a URL was last seen before it should be
+// considered lost (ie, no longer nearby).
+const NSTimeInterval kLostThresholdSeconds = 15.0;
+// The time interval in seconds between checks for lost URLs.
+const NSTimeInterval kUpdateIntervalSeconds = 6.0;
+
 enum BeaconType {
   BEACON_TYPE_NONE,
   BEACON_TYPE_URIBEACON,
   BEACON_TYPE_EDDYSTONE,
 };
 
 }  // namespace
 
 @interface PhysicalWebScanner ()<CBCentralManagerDelegate>
 
 // Decodes the UriBeacon information in the given |data| and a beacon |type| to
 // return an unresolved PhysicalWebDevice instance. It also stores the given
 // |rssi| in the result.
 + (PhysicalWebDevice*)newDeviceFromData:(NSData*)data
                                    rssi:(int)rssi
                                    type:(BeaconType)type;
-// Starts the CoreBluetooth scanner when the bluetooth is powered on.
+// Starts the CoreBluetooth scanner when the bluetooth is powered on and starts
+// the update timer.
 - (void)reallyStart;
+// Stops the CoreBluetooth scanner and update timer.
+- (void)reallyStop;
+// Timer callback to check for lost URLs based on the elapsed time since they
+// were last seen.
+- (void)onUpdate:(NSTimer*)timer;
 // Requests metadata of a device if the same URL has not been requested before.
 - (void)requestMetadataForDevice:(PhysicalWebDevice*)device;
 // Returns the beacon type given the advertisement data.
 + (BeaconType)beaconTypeForAdvertisementData:(NSDictionary*)advertisementData;
 
 @end
 
 @implementation PhysicalWebScanner {
   // Delegate that will be notified when the list of devices change.
   id<PhysicalWebScannerDelegate> delegate_;
   // The value of |started_| is YES when the scanner has been started and NO
   // when it's been stopped. The initial value is NO.
   BOOL started_;
   // The value is valid when the scanner has been started. If bluetooth is not
   // powered on, the value is YES, if it's powered on and the CoreBluetooth
   // scanner has started, the value is NO.
   BOOL pendingStart_;
   // List of PhysicalWebRequest that we're waiting the response from.
   base::scoped_nsobject<NSMutableArray> pendingRequests_;
   // List of resolved PhysicalWebDevice.
   base::scoped_nsobject<NSMutableArray> devices_;
   // List of URLs that have been resolved or have a pending resolution from a
   // PhysicalWebRequest.
   base::scoped_nsobject<NSMutableSet> devicesUrls_;
   // List of final URLs that have been resolved. This set will help us
   // deduplicate the final URLs.
   base::scoped_nsobject<NSMutableSet> finalUrls_;
   // CoreBluetooth scanner.
   base::scoped_nsobject<CBCentralManager> centralManager_;
+  // When YES, we will notify the delegate if a previously nearby URL is lost
+  // and remove it from the list of nearby devices.
+  BOOL onLostDetectionEnabled_;
   // The value is YES if network requests can be sent.
   BOOL networkRequestEnabled_;
   // List of unresolved PhysicalWebDevice when network requests are not enabled.
   base::scoped_nsobject<NSMutableArray> unresolvedDevices_;
+  // Map from URLs to timestamps of the last time the URL was nearby.
+  base::scoped_nsobject<NSMutableDictionary> scanTimestamps_;
+  // A repeating timer to check for lost URLs. If the elapsed time since an URL
+  // was last seen exceeds a threshold, the URL is considered lost.
+  base::scoped_nsobject<NSTimer> updateTimer_;
 }
 
+@synthesize onLostDetectionEnabled = onLostDetectionEnabled_;
 @synthesize networkRequestEnabled = networkRequestEnabled_;
 
 - (instancetype)initWithDelegate:(id<PhysicalWebScannerDelegate>)delegate {
   self = [super init];
   if (self) {
     delegate_ = delegate;
     devices_.reset([[NSMutableArray alloc] init]);
     devicesUrls_.reset([[NSMutableSet alloc] init]);
     finalUrls_.reset([[NSMutableSet alloc] init]);
     pendingRequests_.reset([[NSMutableArray alloc] init]);
     centralManager_.reset([[CBCentralManager alloc]
         initWithDelegate:self
                    queue:dispatch_get_main_queue()]);
     unresolvedDevices_.reset([[NSMutableArray alloc] init]);
+    scanTimestamps_.reset([[NSMutableDictionary alloc] init]);
     [[NSHTTPCookieStorage sharedHTTPCookieStorage]
         setCookieAcceptPolicy:NSHTTPCookieAcceptPolicyNever];
   }
   return self;
 }
 
 - (instancetype)init {
   NOTREACHED();
   return nil;
 }
 
 - (void)dealloc {
   [centralManager_ setDelegate:nil];
   centralManager_.reset();
+  if (updateTimer_.get()) {
+    [updateTimer_ invalidate];
+    updateTimer_.reset();
+  }
   [super dealloc];
 }
 
 - (void)start {
   [self stop];
   [finalUrls_ removeAllObjects];
   [devicesUrls_ removeAllObjects];
   [devices_ removeAllObjects];
+  [scanTimestamps_ removeAllObjects];
   started_ = YES;
   if ([self bluetoothEnabled])
     [self reallyStart];
   else
     pendingStart_ = YES;
 }
 
 - (void)stop {
   if (!started_)
     return;
   for (PhysicalWebRequest* request in pendingRequests_.get()) {
     [request cancel];
   }
   [pendingRequests_ removeAllObjects];
   if (!pendingStart_ && [self bluetoothEnabled]) {
-    [centralManager_ stopScan];
+    [self reallyStop];
   }
   pendingStart_ = NO;
   started_ = NO;
 }
 
 - (NSArray*)devices {
   return [devices_ sortedArrayUsingComparator:^(id obj1, id obj2) {
     PhysicalWebDevice* device1 = obj1;
     PhysicalWebDevice* device2 = obj2;
     // Sorts in ascending order.
@@ skipping 39 matching lines @@
   if (!networkRequestEnabled_)
     return;
 
   // Sends the pending requests.
   for (PhysicalWebDevice* device in unresolvedDevices_.get()) {
     [self requestMetadataForDevice:device];
   }
   [unresolvedDevices_ removeAllObjects];
 }
 
+- (void)setOnLostDetectionEnabled:(BOOL)enabled {
+  BOOL oldOnLostDetectionEnabled = onLostDetectionEnabled_;
+  onLostDetectionEnabled_ = enabled;
+  if (started_ && oldOnLostDetectionEnabled != onLostDetectionEnabled_) {
+    [self start];

[Olivier, 2016/10/13 13:00:19]

This will reset all the lists. Is that expected?
Should we document this?

[mattreynolds, 2016/10/18 18:56:47]

I added a comment documenting the behavior

+  }
+}
+
 - (int)unresolvedBeaconsCount {
   return [unresolvedDevices_ count];
 }
 
 - (BOOL)bluetoothEnabled {
 // TODO(crbug.com/619982): The CBManager base class appears to still be in
 // flux.  Unwind this #ifdef once the APIs settle.
 #if defined(__IPHONE_10_0) && __IPHONE_OS_VERSION_MAX_ALLOWED >= __IPHONE_10_0
   return [centralManager_ state] == CBManagerStatePoweredOn;
 #else
   return [centralManager_ state] == CBCentralManagerStatePoweredOn;
 #endif
 }
 
 - (void)reallyStart {
   pendingStart_ = NO;
+
+  if (updateTimer_.get()) {
+    [updateTimer_ invalidate];
+    updateTimer_.reset();
+  }
+
   NSArray* serviceUUIDs = @[
     [CBUUID UUIDWithString:kUriBeaconServiceUUID],
     [CBUUID UUIDWithString:kEddystoneBeaconServiceUUID]
   ];
-  [centralManager_ scanForPeripheralsWithServices:serviceUUIDs options:nil];
+  NSDictionary* options = nil;
+  if (onLostDetectionEnabled_) {
+    // To detect lost URLs, we configure the scanner with the AllowDuplicates
+    // option enabled, which will notify us each time an advertising packet is
+    // received rather than only the first time. A URL is considered lost if the
+    // most recent sighting was more than |kLostThresholdSeconds| ago.
+    options =
+        [NSDictionary dictionaryWithObjectsAndKeys:
+                          [NSNumber numberWithBool:YES],
+                          CBCentralManagerScanOptionAllowDuplicatesKey, nil];
+
+    // Register a repeating timer to periodically check for lost URLs.
+    updateTimer_.reset([NSTimer
+        scheduledTimerWithTimeInterval:kUpdateIntervalSeconds
+                                target:self
+                              selector:@selector(onUpdate:)
+                              userInfo:nil
+                               repeats:YES]);
+  }
+  [centralManager_ scanForPeripheralsWithServices:serviceUUIDs options:options];
+}
+
+- (void)reallyStop {
+  if (updateTimer_.get()) {
+    [updateTimer_ invalidate];
+    updateTimer_.reset();
+  }
+
+  [centralManager_ stopScan];
+}
+
+- (void)onUpdate:(NSTimer*)timer {
+  NSDate* now = [NSDate date];
+  NSMutableArray* lostDevices = [NSMutableArray array];
+  NSMutableArray* lostUnresolvedDevices = [NSMutableArray array];
+  NSMutableArray* lostScannedUrls = [NSMutableArray array];
+  NSMutableArray* lostResolvedUrls = [NSMutableArray array];
+
+  for (PhysicalWebDevice* device in [self devices]) {
+    NSDate* scanTimestamp = [scanTimestamps_ objectForKey:[device requestURL]];
+    NSTimeInterval elapsedSeconds = [now timeIntervalSinceDate:scanTimestamp];
+    if (elapsedSeconds > kLostThresholdSeconds) {
+      [lostDevices addObject:device];
+      [lostScannedUrls addObject:[device requestURL]];
+      [lostResolvedUrls addObject:[device url]];
+      [devicesUrls_ removeObject:[device requestURL]];
+      [finalUrls_ removeObject:[device url]];
+    }
+  }
+
+  for (PhysicalWebDevice* device in unresolvedDevices_.get()) {
+    NSDate* scanTimestamp = [scanTimestamps_ objectForKey:[device requestURL]];
+    NSTimeInterval elapsedSeconds = [now timeIntervalSinceDate:scanTimestamp];
+    if (elapsedSeconds > kLostThresholdSeconds) {
+      [lostUnresolvedDevices addObject:device];
+      [lostScannedUrls addObject:[device requestURL]];
+      [devicesUrls_ removeObject:[device requestURL]];
+    }
+  }
+
+  [devices_ removeObjectsInArray:lostDevices];
+  [unresolvedDevices_ removeObjectsInArray:lostUnresolvedDevices];

[Olivier, 2016/10/13 13:00:19]

should we cancel the resolve URL query?

[mattreynolds, 2016/10/18 18:56:47]

Done.

+  [scanTimestamps_ removeObjectsForKeys:lostScannedUrls];
+
+  if ([lostDevices count]) {
+    [delegate_ scannerUpdatedDevices:self];
+  }
+
+  // For unknown reasons, when scanning for longer periods (on the order of

[Olivier, 2016/10/13 13:00:19]

Can you file a radar on this, file a crbug that reference the radar?

[mattreynolds, 2016/10/18 18:56:47]

Done.

+  // minutes), the scanner is less reliable at detecting all nearby URLs. As a
+  // workaround, we restart the scanner each time we check for lost URLs.
+  NSArray* serviceUUIDs = @[
+    [CBUUID UUIDWithString:kUriBeaconServiceUUID],
+    [CBUUID UUIDWithString:kEddystoneBeaconServiceUUID]
+  ];
+  NSDictionary* options = [NSDictionary
+      dictionaryWithObjectsAndKeys:[NSNumber numberWithBool:YES],
+                                   CBCentralManagerScanOptionAllowDuplicatesKey,
+                                   nil];
+  [centralManager_ stopScan];
+  [centralManager_ scanForPeripheralsWithServices:serviceUUIDs options:options];
 }
 
 #pragma mark -
 #pragma mark CBCentralManagerDelegate methods
 
 - (void)centralManagerDidUpdateState:(CBCentralManager*)central {
   if ([self bluetoothEnabled]) {
     if (pendingStart_)
       [self reallyStart];
   } else {
     if (started_ && !pendingStart_) {
       pendingStart_ = YES;
-      [centralManager_ stopScan];
+      [self reallyStop];
     }
   }
   [delegate_ scannerBluetoothStatusUpdated:self];
 }
 
 + (BeaconType)beaconTypeForAdvertisementData:(NSDictionary*)advertisementData {
   NSDictionary* serviceData =
       [advertisementData objectForKey:CBAdvertisementDataServiceDataKey];
   if ([serviceData objectForKey:[CBUUID UUIDWithString:kUriBeaconServiceUUID]])
     return BEACON_TYPE_URIBEACON;
@@ skipping 31 matching lines @@
   DCHECK(data);
 
   base::scoped_nsobject<PhysicalWebDevice> device([PhysicalWebScanner
       newDeviceFromData:data
                    rssi:[RSSI intValue]
                    type:type]);
   // Skip if the data couldn't be parsed.
   if (!device.get())
     return;
 
+  [scanTimestamps_ setObject:[NSDate date] forKey:[device requestURL]];

[Olivier, 2016/10/13 13:00:19]

Would it make sense to add this to the device object instead?

[mattreynolds, 2016/10/18 18:56:47]

Done.

+
   // Skip if the URL has already been seen.
-  if ([devicesUrls_ containsObject:[device url]])
+  if ([devicesUrls_ containsObject:[device requestURL]])
     return;
-  [devicesUrls_ addObject:[device url]];
+  [devicesUrls_ addObject:[device requestURL]];
 
   if (networkRequestEnabled_) {
     [self requestMetadataForDevice:device];
   } else {
     [unresolvedDevices_ addObject:device];
     [delegate_ scannerUpdatedDevices:self];
   }
 }
 
 #pragma mark -
@@ skipping 28 matching lines @@
   std::string utf8URI;
   device::DecodeUriBeaconUri(encodedURI, utf8URI);
   NSString* uriString = base::SysUTF8ToNSString(utf8URI);
   NSURL* url = [NSURL URLWithString:uriString];
 
   // Ensure URL is valid.
   if (!url)
     return nil;
 
   return [[PhysicalWebDevice alloc] initWithURL:url
-                                     requestURL:nil
+                                     requestURL:url
                                            icon:nil
                                           title:nil
                                     description:nil
                                   transmitPower:transmitPower
                                            rssi:rssi
                                            rank:physical_web::kMaxRank];
 }
 
 - (void)requestMetadataForDevice:(PhysicalWebDevice*)device {
   base::scoped_nsobject<PhysicalWebRequest> request(
@@ skipping 12 matching lines @@
         [strongSelf.get()->devices_ addObject:device];
         [strongSelf.get()->delegate_ scannerUpdatedDevices:weakSelf];
         [strongSelf.get()->finalUrls_ addObject:[device url]];
       }
     }
     [strongSelf.get()->pendingRequests_ removeObject:strongRequest];
   }];
 }
 
 @end