🔧 Fix SEGV when function has no body

7dc5f41b · nagayama15 · b407a709 · 7dc5f41b · 7dc5f41b
Commit 7dc5f41b authored Jul 07, 2019 by nagayama15
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 2 deletions

operand_swapping_extractor.cpp src/kyut/pass/operand_swapping_extractor.cpp +3 -1

operand_swapping_watermarker.cpp src/kyut/pass/operand_swapping_watermarker.cpp +3 -1

No files found.
--- a/src/kyut/pass/operand_swapping_extractor.cpp
+++ b/src/kyut/pass/operand_swapping_extractor.cpp
@@ -68,7 +68,9 @@ namespace kyut::pass {
            OperandSwappingExtractingVisitor visitor{writer_};

            for (const auto &func : module->functions) {
-                visitor.visit(func->body);
+                if (func->body) {
+                    visitor.visit(func->body);
+                }
            }
        }


--- a/src/kyut/pass/operand_swapping_watermarker.cpp
+++ b/src/kyut/pass/operand_swapping_watermarker.cpp
@@ -70,7 +70,9 @@ namespace kyut::pass {
            OperandSwappingWatermarkingVisitor visitor{stream_};

            for (const auto &func : module->functions) {
-                visitor.visit(func->body);
+                if (func->body) {
+                    visitor.visit(func->body);
+                }
            }
        }