🔧 Fix SEGV when function has no body

7dc5f41b · nagayama15 · b407a709 · 7dc5f41b · 7dc5f41b
Commit 7dc5f41b authored Jul 07, 2019 by nagayama15
Show 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,9 +68,11 @@ namespace kyut::pass {
            OperandSwappingExtractingVisitor visitor{writer_};

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

    private:
        BitWriter &writer_;

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

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

    private:
        CircularBitStream &stream_;