Test if blob is returned. (#1535)

1 files changed, 10 insertions, 10 deletions

diff --git a/source/slang/slang.cpp b/source/slang/slang.cpp
index b9381eec3..c97da7125 100644
--- a/source/slang/slang.cpp
+++ b/source/slang/slang.cpp
@@ -3390,6 +3390,13 @@ SLANG_API void const* spGetEntryPointCode(
     size_t*                 outSize)
 {
     using namespace Slang;
+
+    // Zero the size initially, in case need to return nullptr for error.
+    if (outSize)
+    {
+        *outSize = 0;
+    }
+
     auto req = Slang::asInternal(request);
     auto linkage = req->getLinkage();
     auto program = req->getSpecializedGlobalAndEntryPointsComponentType();
@@ -3411,22 +3418,15 @@ SLANG_API void const* spGetEntryPointCode(
         return nullptr;
     CompileResult& result = targetProgram->getExistingEntryPointResult(entryPointIndex);
 
-    void const* data = nullptr;
-    size_t size = 0;
-
     ComPtr<ISlangBlob> blob;
-    if (SLANG_SUCCEEDED(result.getBlob(blob)))
-    {
-        data = blob->getBufferPointer();
-        size = blob->getBufferSize();
-    }
+    SLANG_RETURN_NULL_ON_FAIL(result.getBlob(blob));
 
     if (outSize)
     {
-        *outSize = size;
+        *outSize = blob->getBufferSize();
     }
 
-    return data;
+    return (void*)blob->getBufferPointer();
 }
 
 static SlangResult _getEntryPointResult(