1 files changed, 46 insertions, 0 deletions
diff --git a/source/slang/ir-missing-return.cpp b/source/slang/ir-missing-return.cpp
new file mode 100644
index 000000000..0396fbce9
--- /dev/null
+++ b/source/slang/ir-missing-return.cpp
@@ -0,0 +1,46 @@
+// ir-missing-return.cpp
+#include "ir-missing-return.h"
+
+#include "ir.h"
+#include "ir-insts.h"
+
+namespace Slang {
+
+class DiagnosticSink;
+struct IRModule;
+
+void checkForMissingReturnsRec(
+    IRInst*         inst,
+    DiagnosticSink* sink)
+{
+    if( auto code = as<IRGlobalValueWithCode>(inst) )
+    {
+        for( auto block : code->getBlocks() )
+        {
+            auto terminator = block->getTerminator();
+
+            if( auto missingReturn = as<IRMissingReturn>(terminator) )
+            {
+                sink->diagnose(missingReturn, Diagnostics::missingReturn);
+            }
+        }
+    }
+
+    if( auto parentInst = as<IRParentInst>(inst) )
+    {
+        for( auto childInst : parentInst->getChildren() )
+        {
+            checkForMissingReturnsRec(childInst, sink);
+        }
+    }
+}
+
+void checkForMissingReturns(
+    IRModule*       module,
+    DiagnosticSink* sink)
+{
+    // Look for any `missingReturn` instructions
+    checkForMissingReturnsRec(module->getModuleInst(), sink);
+}
+
+}