Reliably cancel in-progress CryptoResult(Impl) upon shutdown.

Source/platform/exported/WebCryptoResult.cpp

 /*
  * Copyright (C) 2013 Google Inc. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions are
  * met:
  *
  *     * Redistributions of source code must retain the above copyright
  * notice, this list of conditions and the following disclaimer.
  *     * Redistributions in binary form must reproduce the above
@@ skipping 15 matching lines @@
  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  * OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 
 #include "config.h"
 #include "public/platform/WebCrypto.h"
 
 #include "platform/CryptoResult.h"
 #include "platform/heap/Heap.h"
-#include <string.h>
+#include "wtf/Atomics.h"
 
 namespace blink {
 
 void WebCryptoResult::completeWithError(WebCryptoErrorType errorType, const WebString& errorDetails)
 {
+    if (cancelled())
+        return;
     m_impl->completeWithError(errorType, errorDetails);
-    reset();
+    clear();
 }
 
 void WebCryptoResult::completeWithBuffer(const void* bytes, unsigned bytesSize)
 {
+    if (cancelled())
+        return;
     m_impl->completeWithBuffer(bytes, bytesSize);
-    reset();
+    clear();
 }
 
 void WebCryptoResult::completeWithJson(const char* utf8Data, unsigned length)
 {
+    if (cancelled())
+        return;
     m_impl->completeWithJson(utf8Data, length);
-    reset();
+    clear();
 }
 
 void WebCryptoResult::completeWithBoolean(bool b)
 {
+    if (cancelled())
+        return;
     m_impl->completeWithBoolean(b);
-    reset();
+    clear();
 }
 
 void WebCryptoResult::completeWithKey(const WebCryptoKey& key)
 {
+    if (cancelled())
+        return;
     ASSERT(!key.isNull());
     m_impl->completeWithKey(key);
-    reset();
+    clear();
 }
 
 void WebCryptoResult::completeWithKeyPair(const WebCryptoKey& publicKey, const WebCryptoKey& privateKey)
 {
+    if (cancelled())
+        return;
     ASSERT(!publicKey.isNull());
     ASSERT(!privateKey.isNull());
     m_impl->completeWithKeyPair(publicKey, privateKey);
-    reset();
+    clear();
 }
 
 bool WebCryptoResult::cancelled() const
 {
-    return m_impl->cancelled();
+    return acquireLoad(&m_cancelled);
 }
 
 WebCryptoResult::WebCryptoResult(const PassRefPtrWillBeRawPtr<CryptoResult>& impl)
-    : m_impl(impl)
+    : m_cancelled(0)
+    , m_impl(impl)
 {
     ASSERT(m_impl.get());
+    m_impl->assignedTo(this);
 }
 
 void WebCryptoResult::reset()
 {
     m_impl.reset();
 }
 
+void WebCryptoResult::clear()
+{
+    if (!m_impl.isNull()) {
+        releaseStore(&m_cancelled, 1);
+        m_impl->assignedTo(nullptr);
+    }
+    reset();
+}
+
 void WebCryptoResult::assign(const WebCryptoResult& o)
 {
+    ASSERT(!o.m_cancelled);
+    ASSERT(!o.m_impl.isNull());
+    m_cancelled = o.m_cancelled;
     m_impl = o.m_impl;
+    m_impl->assignedTo(this);
 }
 
 } // namespace blink