Metal: Handle potential OOM in other places.

This commit adds checks to (hopefully) all the remaining places in the
Metal backend where object creation can fail.

Bug: dawn:801

Change-Id: Ic27803e956beef822f94ca8449f7816ddd17c1bc
Reviewed-on: https://dawn-review.googlesource.com/c/dawn/+/58722
Commit-Queue: Stephen White <senorblanco@chromium.org>
Auto-Submit: Corentin Wallez <cwallez@chromium.org>
Reviewed-by: Austin Eng <enga@chromium.org>
Reviewed-by: Stephen White <senorblanco@chromium.org>

src/dawn_native/metal/CommandBufferMTL.mm

@@ -223,7 +223,8 @@ namespace dawn_native { namespace metal {
         }
 
         // Helper functions for Toggle AlwaysResolveIntoZeroLevelAndLayer
-        NSPRef<id<MTLTexture>> CreateResolveTextureForWorkaround(Device* device,
+        ResultOrError<NSPRef<id<MTLTexture>>> CreateResolveTextureForWorkaround(
+            Device* device,
             MTLPixelFormat mtlFormat,
             uint32_t width,
             uint32_t height) {
@@ -241,7 +242,12 @@ namespace dawn_native { namespace metal {
             mtlDesc.storageMode = MTLStorageModePrivate;
             mtlDesc.sampleCount = 1;
 
-            return AcquireNSPRef([device->GetMTLDevice() newTextureWithDescriptor:mtlDesc]);
+            id<MTLTexture> texture = [device->GetMTLDevice() newTextureWithDescriptor:mtlDesc];
+            if (texture == nil) {
+                return DAWN_OUT_OF_MEMORY_ERROR("Allocation of temporary texture failed.");
+            }
+
+            return AcquireNSPRef(texture);
         }
 
         void CopyIntoTrueResolveTarget(CommandRecordingContext* commandContext,
@@ -1144,8 +1150,8 @@ namespace dawn_native { namespace metal {
                 trueResolveSlices[i] = mtlRenderPass.colorAttachments[i].resolveSlice;
 
                 const MTLPixelFormat mtlFormat = trueResolveTextures[i].pixelFormat;
-                temporaryResolveTextures[i] =
-                    CreateResolveTextureForWorkaround(device, mtlFormat, width, height);
+                DAWN_TRY_ASSIGN(temporaryResolveTextures[i], CreateResolveTextureForWorkaround(
+                                                                 device, mtlFormat, width, height));
 
                 mtlRenderPass.colorAttachments[i].resolveTexture =
                     temporaryResolveTextures[i].Get();

src/dawn_native/metal/ComputePipelineMTL.mm

@@ -43,9 +43,10 @@ namespace dawn_native { namespace metal {
             newComputePipelineStateWithFunction:computeData.function.Get()
                                           error:&error]);
         if (error != nullptr) {
-            NSLog(@" error => %@", error);
-            return DAWN_INTERNAL_ERROR("Error creating pipeline state");
+            return DAWN_INTERNAL_ERROR("Error creating pipeline state" +
+                                       std::string([error.localizedDescription UTF8String]));
         }
+        ASSERT(mMtlComputePipelineState != nil);
 
         // Copy over the local workgroup size as it is passed to dispatch explicitly in Metal
         Origin3D localSize = GetStage(SingleShaderStage::Compute).metadata->localWorkgroupSize;

src/dawn_native/metal/QuerySetMTL.mm

@@ -53,8 +53,8 @@ namespace dawn_native { namespace metal {
                 [device->GetMTLDevice() newCounterSampleBufferWithDescriptor:descriptor
                                                                        error:&error];
             if (error != nullptr) {
-                const char* errorString = [error.localizedDescription UTF8String];
-                return DAWN_INTERNAL_ERROR(std::string("Error creating query set: ") + errorString);
+                return DAWN_OUT_OF_MEMORY_ERROR(std::string("Error creating query set: ") +
+                                                [error.localizedDescription UTF8String]);
             }
 
             return counterSampleBuffer;
@@ -80,6 +80,10 @@ namespace dawn_native { namespace metal {
                 mVisibilityBuffer = AcquireNSPRef([device->GetMTLDevice()
                     newBufferWithLength:bufferSize
                                 options:MTLResourceStorageModePrivate]);
+
+                if (mVisibilityBuffer == nil) {
+                    return DAWN_OUT_OF_MEMORY_ERROR("Failed to allocate query set.");
+                }
                 break;
             }
             case wgpu::QueryType::PipelineStatistics:

src/dawn_native/metal/RenderPipelineMTL.mm

@@ -406,20 +406,19 @@ namespace dawn_native { namespace metal {
         descriptorMTL.sampleCount = GetSampleCount();
         descriptorMTL.alphaToCoverageEnabled = IsAlphaToCoverageEnabled();
 
-        {
         NSError* error = nullptr;
         mMtlRenderPipelineState =
             AcquireNSPRef([mtlDevice newRenderPipelineStateWithDescriptor:descriptorMTL
                                                                     error:&error]);
         if (error != nullptr) {
-                NSLog(@" error => %@", error);
-                return DAWN_INTERNAL_ERROR("Error creating rendering pipeline state");
-            }
+            return DAWN_INTERNAL_ERROR(std::string("Error creating pipeline state") +
+                                       [error.localizedDescription UTF8String]);
         }
+        ASSERT(mMtlRenderPipelineState != nil);
 
         // Create depth stencil state and cache it, fetch the cached depth stencil state when we
-        // call setDepthStencilState() for a given render pipeline in CommandEncoder, in order to
-        // improve performance.
+        // call setDepthStencilState() for a given render pipeline in CommandEncoder, in order
+        // to improve performance.
         NSRef<MTLDepthStencilDescriptor> depthStencilDesc =
             MakeDepthStencilDesc(GetDepthStencilState());
         mMtlDepthStencilState =

src/dawn_native/metal/SamplerMTL.h

@@ -33,7 +33,8 @@ namespace dawn_native { namespace metal {
         id<MTLSamplerState> GetMTLSamplerState();
 
       private:
-        Sampler(Device* device, const SamplerDescriptor* descriptor);
+        using SamplerBase::SamplerBase;
+        MaybeError Initialize(const SamplerDescriptor* descriptor);
 
         NSPRef<id<MTLSamplerState>> mMtlSamplerState;
     };

src/dawn_native/metal/SamplerMTL.mm

@@ -58,11 +58,12 @@ namespace dawn_native { namespace metal {
             return DAWN_VALIDATION_ERROR("Sampler compare function not supported.");
         }
 
-        return AcquireRef(new Sampler(device, descriptor));
+        Ref<Sampler> sampler = AcquireRef(new Sampler(device, descriptor));
+        DAWN_TRY(sampler->Initialize(descriptor));
+        return sampler;
     }
 
-    Sampler::Sampler(Device* device, const SamplerDescriptor* descriptor)
-        : SamplerBase(device, descriptor) {
+    MaybeError Sampler::Initialize(const SamplerDescriptor* descriptor) {
         NSRef<MTLSamplerDescriptor> mtlDescRef = AcquireNSRef([MTLSamplerDescriptor new]);
         MTLSamplerDescriptor* mtlDesc = mtlDescRef.Get();
 
@@ -87,8 +88,13 @@ namespace dawn_native { namespace metal {
             // Metal debug device errors.
         }
 
-        mMtlSamplerState =
-            AcquireNSPRef([device->GetMTLDevice() newSamplerStateWithDescriptor:mtlDesc]);
+        mMtlSamplerState = AcquireNSPRef(
+            [ToBackend(GetDevice())->GetMTLDevice() newSamplerStateWithDescriptor:mtlDesc]);
+
+        if (mMtlSamplerState == nil) {
+            return DAWN_OUT_OF_MEMORY_ERROR("Failed to allocate sampler.");
+        }
+        return {};
     }
 
     id<MTLSamplerState> Sampler::GetMTLSamplerState() {

src/dawn_native/metal/ShaderModuleMTL.mm

@@ -332,11 +332,11 @@ namespace dawn_native { namespace metal {
                                                     error:&error]);
         if (error != nullptr) {
             if (error.code != MTLLibraryErrorCompileWarning) {
-                const char* errorString = [error.localizedDescription UTF8String];
                 return DAWN_VALIDATION_ERROR(std::string("Unable to create library object: ") +
-                                             errorString);
+                                             [error.localizedDescription UTF8String]);
             }
         }
+        ASSERT(library != nil);
 
         NSRef<NSString> name =
             AcquireNSRef([[NSString alloc] initWithUTF8String:remappedEntryPointName.c_str()]);