Add navigator.sendBeacon() use counters.

Source/modules/beacon/NavigatorBeacon.cpp

 // 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 "config.h"
 #include "modules/beacon/NavigatorBeacon.h"
 
 #include "bindings/core/v8/ExceptionState.h"
 #include "core/dom/ExceptionCode.h"
 #include "core/dom/ExecutionContext.h"
 #include "core/fileapi/Blob.h"
 #include "core/frame/LocalFrame.h"
 #include "core/frame/Settings.h"
+#include "core/frame/UseCounter.h"
 #include "core/frame/csp/ContentSecurityPolicy.h"
 #include "core/html/DOMFormData.h"
 #include "core/loader/BeaconLoader.h"
 #include "wtf/ArrayBufferView.h"
 
 namespace blink {
 
 NavigatorBeacon::NavigatorBeacon(Navigator& navigator)
     : m_transmittedBytes(0)
     , m_navigator(navigator)
@@ skipping 45 matching lines @@
     const Settings* settings = m_navigator.frame()->settings();
     if (settings) {
         int maxAllowed = settings->maxBeaconTransmission();
         if (maxAllowed < m_transmittedBytes)
             return 0;
         return maxAllowed - m_transmittedBytes;
     }
     return m_transmittedBytes;
 }
 
-void NavigatorBeacon::updateTransmittedBytes(int length)
+bool NavigatorBeacon::beaconResult(ExecutionContext* context, bool allowed, int sentBytes)
 {
-    ASSERT(length >= 0);
-    m_transmittedBytes += length;
+    if (allowed) {
+        ASSERT(sentBytes >= 0);
+        m_transmittedBytes += sentBytes;
+    } else {
+        UseCounter::count(context, UseCounter::SendBeaconQuotaExceeded);
+    }
+    return allowed;
 }
 
 bool NavigatorBeacon::sendBeacon(ExecutionContext* context, Navigator& navigator, const String& urlstring, const String& data, ExceptionState& exceptionState)
 {
     return NavigatorBeacon::from(navigator).sendBeacon(context, urlstring, data, exceptionState);
 }
 
 bool NavigatorBeacon::sendBeacon(ExecutionContext* context, const String& urlstring, const String& data, ExceptionState& exceptionState)
 {
     KURL url = context->completeURL(urlstring);
     if (!canSendBeacon(context, url, exceptionState))
         return false;
 
     int bytes = 0;
     bool result = BeaconLoader::sendBeacon(m_navigator.frame(), maxAllowance(), url, data, bytes);
-    if (result)
-        updateTransmittedBytes(bytes);
-
-    return result;
+    return beaconResult(context, result, bytes);
 }
 
 bool NavigatorBeacon::sendBeacon(ExecutionContext* context, Navigator& navigator, const String& urlstring, PassRefPtr<ArrayBufferView> data, ExceptionState& exceptionState)
 {
     return NavigatorBeacon::from(navigator).sendBeacon(context, urlstring, data, exceptionState);
 }
 
 bool NavigatorBeacon::sendBeacon(ExecutionContext* context, const String& urlstring, PassRefPtr<ArrayBufferView> data, ExceptionState& exceptionState)
 {
     KURL url = context->completeURL(urlstring);
     if (!canSendBeacon(context, url, exceptionState))
         return false;
 
     int bytes = 0;
     bool result = BeaconLoader::sendBeacon(m_navigator.frame(), maxAllowance(), url, data, bytes);
-    if (result)
-        updateTransmittedBytes(bytes);
-
-    return result;
+    return beaconResult(context, result, bytes);
 }
 
 bool NavigatorBeacon::sendBeacon(ExecutionContext* context, Navigator& navigator, const String& urlstring, PassRefPtrWillBeRawPtr<Blob> data, ExceptionState& exceptionState)
 {
     return NavigatorBeacon::from(navigator).sendBeacon(context, urlstring, data, exceptionState);
 }
 
 bool NavigatorBeacon::sendBeacon(ExecutionContext* context, const String& urlstring, PassRefPtrWillBeRawPtr<Blob> data, ExceptionState& exceptionState)
 {
     KURL url = context->completeURL(urlstring);
     if (!canSendBeacon(context, url, exceptionState))
         return false;
 
     int bytes = 0;
     bool result = BeaconLoader::sendBeacon(m_navigator.frame(), maxAllowance(), url, data, bytes);
-    if (result)
-        updateTransmittedBytes(bytes);
-
-    return result;
+    return beaconResult(context, result, bytes);
 }
 
 bool NavigatorBeacon::sendBeacon(ExecutionContext* context, Navigator& navigator, const String& urlstring, PassRefPtrWillBeRawPtr<DOMFormData> data, ExceptionState& exceptionState)
 {
     return NavigatorBeacon::from(navigator).sendBeacon(context, urlstring, data, exceptionState);
 }
 
 bool NavigatorBeacon::sendBeacon(ExecutionContext* context, const String& urlstring, PassRefPtrWillBeRawPtr<DOMFormData> data, ExceptionState& exceptionState)
 {
     KURL url = context->completeURL(urlstring);
     if (!canSendBeacon(context, url, exceptionState))
         return false;
 
     int bytes = 0;
     bool result = BeaconLoader::sendBeacon(m_navigator.frame(), maxAllowance(), url, data, bytes);
-    if (result)
-        updateTransmittedBytes(bytes);
-
-    return result;
+    return beaconResult(context, result, bytes);
 }
 
 } // namespace blink