ShapeDetectionImpl: Android implementation of detectFace()

third_party/WebKit/Source/modules/shapedetection/FaceDetector.cpp

 // Copyright 2016 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 "modules/shapedetection/FaceDetector.h"
 
 #include "core/dom/DOMException.h"
 #include "core/dom/DOMRect.h"
 #include "core/dom/Document.h"
 #include "core/fetch/ImageResource.h"
@@ skipping 29 matching lines @@
 
   if (!image) {
     DLOG(ERROR) << "Failed to convert blink::Image to sk_sp<SkImage>.";
     return mojo::ScopedSharedBufferHandle();
   }
 
   const SkImageInfo skiaInfo =
       SkImageInfo::MakeN32(image->width(), image->height(), image->alphaType());
 
   const uint32_t allocationSize = skiaInfo.getSafeSize(skiaInfo.minRowBytes());
+
   mojo::ScopedSharedBufferHandle sharedBufferHandle =
       mojo::SharedBufferHandle::Create(allocationSize);
+  if (!sharedBufferHandle.is_valid()) {
+    // TODO(xianglu): Do something when the image is too large.
+    DLOG(ERROR) << "Failed to create a sharedBufferHandle. allocationSize = "
+                << allocationSize << "bytes. limit = 16777216";
+    return mojo::ScopedSharedBufferHandle();
+  }
+
   const mojo::ScopedSharedBufferMapping mappedBuffer =
       sharedBufferHandle->Map(allocationSize);
-  DCHECK(mappedBuffer);
 
   const SkPixmap pixmap(skiaInfo, mappedBuffer.get(), skiaInfo.minRowBytes());
   if (!image->readPixels(pixmap, 0, 0)) {
     DLOG(ERROR) << "Failed to read pixels from sk_sp<SkImage>.";
     return mojo::ScopedSharedBufferHandle();
   }
 
   return sharedBufferHandle;
 }
 
@@ skipping 24 matching lines @@
     resolver->reject(DOMException::create(
         SyntaxError, "The provided HTMLImageElement is empty."));
     return promise;
   }
 
   // TODO(xianglu): Add security check when the spec is ready.
   // https://crbug.com/646083
   mojo::ScopedSharedBufferHandle sharedBufferHandle =
       getSharedBufferHandle(img);
   if (!sharedBufferHandle->is_valid()) {
-    resolver->reject(
-        DOMException::create(SyntaxError, "Failed to get sharedBufferHandle."));
+    resolver->reject(DOMException::create(
+        SyntaxError, "Request for sharedBufferHandle failed."));
     return promise;
   }
 
   m_serviceRequests.add(resolver);
   DCHECK(m_service.is_bound());
   m_service->DetectFace(std::move(sharedBufferHandle), img->naturalWidth(),
                         img->naturalHeight(),
                         convertToBaseCallback(WTF::bind(
                             &FaceDetector::onDetectFace, wrapPersistent(this),
                             wrapPersistent(resolver))));
@@ skipping 16 matching lines @@
 
   resolver->resolve(detectedFaces);
   m_serviceRequests.remove(resolver);
 }
 
 DEFINE_TRACE(FaceDetector) {
   visitor->trace(m_serviceRequests);
 }
 
 }  // namespace blink