add missing branch detection of default in switch. all reloop but sauer, dlmalloc/clang

4 files changed, 32 insertions, 6 deletions

diff --git a/src/analyzer.js b/src/analyzer.js
index 54f766cc..b44165b1 100644
--- a/src/analyzer.js
+++ b/src/analyzer.js
@@ -348,7 +348,7 @@ function analyzer(data) {
         labels.forEach(function(label) {
           label.lines.forEach(function(line) {
             function process(item) {
-              ['label', 'labelTrue', 'labelFalse', 'toLabel', 'unwindLabel'].forEach(function(id) {
+              ['label', 'labelTrue', 'labelFalse', 'toLabel', 'unwindLabel', 'defaultLabel'].forEach(function(id) {
                 if (item[id]) {
                   if (item[id][0] == 'B') { // BREAK, BCONT, BNOPP
                     label.hasBreak = true;
@@ -460,7 +460,7 @@ function analyzer(data) {
         if (entries.length > 1) return emulated();
         var entry = entries[0];
         assert(entry);
-        dprint('relooping', 'Relooping: ' + entry + ',' + labels.length + ' labels');
+        dprint('relooping', 'makeBlock: ' + entry + ',' + labels.length + ' labels');
 
         var entryLabel = labelsDict[entry];
         assert(entryLabel);
@@ -469,11 +469,12 @@ function analyzer(data) {
 
         var canReturn = values(entryLabel.inLabels).length > 0;
 
+        labels = labels.filter(function(label) { return label === entryLabel || label.ident in entryLabel.allOutLabels }); // ignore unreachables
+
         // === (simple) 'emulated' ===
 
         if (!canReturn && values(entryLabel.outLabels).length == 1) {
           dprint('relooping', '   Creating simple emulated, outlabels: ' + keys(entryLabel.outLabels));
-          assertEq(lastLine.intertype, 'branch');
           var next = keys(entryLabel.outLabels)[0];
           if (next in exitLabels) {
             exitLabelsHit[next] = true;
@@ -506,6 +507,19 @@ function analyzer(data) {
           var internals = split_.leftIn;
           var currExitLabels = set(getLabelIds(externals));
 
+          // Verify that no external can reach an internal
+          var inLabels = set(getLabelIds(internals));
+          var fail = false;
+          externals.forEach(function(external) {
+            if (fail || values(setIntersect(external.outLabels, inLabels)).length > 0) {
+              fail = true;
+            }
+          });
+          if (fail) {
+            dprint('relooping', 'Found an external that wants to reach an internal, fallback to flow emulation');
+            return emulated();
+          }
+
           dprint('relooping', function() { return '   Creating reloop: Inner: ' + dump(getLabelIds(internals)) + ', Exxer: ' + dump(currExitLabels) });
 
           replaceLabelLabels(internals, searchable(entry), 'BCONT' + entry); // we will be in a loop, |continue| gets us back to the entry
diff --git a/src/intertyper.js b/src/intertyper.js
index ed670e65..3092a310 100644
--- a/src/intertyper.js
+++ b/src/intertyper.js
@@ -559,7 +559,7 @@ function intertyper(data) {
         intertype: 'switch',
         type: item.tokens[1].text,
         ident: item.tokens[2].text,
-        defaultLabel: item.tokens[5].text,
+        defaultLabel: toNiceIdent(item.tokens[5].text),
         switchLabels: parseSwitchLabels(item.tokens[6]),
         lineNum: item.lineNum,
       }];
diff --git a/src/jsifier.js b/src/jsifier.js
index a4d7b96e..69bc7d1e 100644
--- a/src/jsifier.js
+++ b/src/jsifier.js
@@ -260,6 +260,7 @@ function JSify(data) {
         if (!block.entry && block.entries.length == 1) block.entry = block.entries[0];
         dprint('relooping', 'walking block: ' + block.type + ',' + block.entry + ',' + block.entries + ' : ' + block.labels.length);
         function getLabelLines(label, indent) {
+          if (!label) return '';
           var ret = '';
           if (LABEL_DEBUG) {
             ret += indent + "print(INDENT + '" + func.ident + ":" + label.ident + "');\n";
@@ -427,7 +428,7 @@ function JSify(data) {
     if (label[0] == 'B') {
       if (label[1] == 'R') {
         assert(oldLabel);
-        return '__label__ = ' + getLabelId(oldLabel) + '; ' + // TODO: optimize away
+        return '__label__ = ' + getLabelId(oldLabel) + '; /* ' + label + ' */' + // TODO: optimize away
                'break ' + label.substr(5) + ';';
       } else if (label[1] == 'C') {
         return 'continue ' + label.substr(5) + ';';
@@ -435,7 +436,7 @@ function JSify(data) {
         return ';'; // Returning no text might confuse this parser
       }
     } else {
-      return '__label__ = ' + getLabelId(label) + '; break;';
+      return '__label__ = ' + getLabelId(label) + '; /* ' + label + ' */ break;';
     }
   }
 
diff --git a/src/utility.js b/src/utility.js
index 3d8ce86e..d5583ffb 100644
--- a/src/utility.js
+++ b/src/utility.js
@@ -202,3 +202,14 @@ function setSub(x, y) {
   return ret;
 }
 
+// Intersection of 2 sets. Faster if |xx| << |yy|
+function setIntersect(x, y) {
+  var ret = {};
+  for (xx in x) {
+    if (xx in y) {
+      ret[xx] = true;
+    }
+  }
+  return ret;
+}
+