From 797fa10f7cd643502e7c139b064fa46ccc5829ae Mon Sep 17 00:00:00 2001
From: Matej Ferencevic <matej.ferencevic@memgraph.io>
Date: Wed, 2 Aug 2017 17:03:40 +0200
Subject: [PATCH] Added antlr sigsegv test.

Reviewers: buda, mislav.bradac

Reviewed By: mislav.bradac

Subscribers: pullbot

Differential Revision: https://phabricator.memgraph.io/D622
---
 tests/manual/CMakeLists.txt     |   2 +
 tests/manual/antlr_sigsegv.cpp  | 106 ++++++++++++++++++++++++++++++++
 tests/manual/test_antlr_sigsegv |   8 +++
 3 files changed, 116 insertions(+)
 create mode 100644 tests/manual/antlr_sigsegv.cpp
 create mode 100755 tests/manual/test_antlr_sigsegv

diff --git a/tests/manual/CMakeLists.txt b/tests/manual/CMakeLists.txt
index 7e830adff..e0d04a77d 100644
--- a/tests/manual/CMakeLists.txt
+++ b/tests/manual/CMakeLists.txt
@@ -32,5 +32,7 @@ foreach(test_cpp ${test_type_cpps})
 
     # link libraries
     target_link_libraries(${target_name} memgraph_lib)
+    # gtest
+    target_link_libraries(${target_name} gtest gtest_main gmock)
 
 endforeach()
diff --git a/tests/manual/antlr_sigsegv.cpp b/tests/manual/antlr_sigsegv.cpp
new file mode 100644
index 000000000..615b91053
--- /dev/null
+++ b/tests/manual/antlr_sigsegv.cpp
@@ -0,0 +1,106 @@
+#include <atomic>
+#include <chrono>
+#include <thread>
+#include <vector>
+
+#include <glog/logging.h>
+#include <gtest/gtest.h>
+
+#include "query/frontend/opencypher/parser.hpp"
+#include "utils/signals/handler.hpp"
+#include "utils/stacktrace.hpp"
+
+using namespace std::chrono_literals;
+
+TEST(Antlr, Sigsegv) {
+  // start clients
+  int N = 8;
+  std::vector<std::thread> threads;
+
+  std::atomic<bool> run{false};
+
+  for (int i = 0; i < N; ++i) {
+    threads.push_back(std::thread([&run]() {
+      while (!run);
+      while (run) {
+        query::frontend::opencypher::Parser parser(
+            "CREATE (:Label_T7 {x: 903}) CREATE (:Label_T7 {x: 720}) CREATE "
+            "(:Label_T7 {x: 13}) CREATE (:Label_T7 {x: 643}) CREATE (:Label_T7 "
+            "{x: 245}) CREATE (:Label_T7 {x: 441}) CREATE (:Label_T7 {x: 47}) "
+            "CREATE (:Label_T7 {x: 583}) CREATE (:Label_T7 {x: 262}) CREATE "
+            "(:Label_T7 {x: 472}) CREATE (:Label_T7 {x: 443}) CREATE "
+            "(:Label_T7 {x: 955}) CREATE (:Label_T7 {x: 197}) CREATE "
+            "(:Label_T7 {x: 524}) CREATE (:Label_T7 {x: 416}) CREATE "
+            "(:Label_T7 {x: 403}) CREATE (:Label_T7 {x: 100}) CREATE "
+            "(:Label_T7 {x: 524}) CREATE (:Label_T7 {x: 702}) CREATE "
+            "(:Label_T7 {x: 586}) CREATE (:Label_T7 {x: 459}) CREATE "
+            "(:Label_T7 {x: 844}) CREATE (:Label_T7 {x: 959}) CREATE "
+            "(:Label_T7 {x: 81}) CREATE (:Label_T7 {x: 423}) CREATE (:Label_T7 "
+            "{x: 553}) CREATE (:Label_T7 {x: 987}) CREATE (:Label_T7 {x: 579}) "
+            "CREATE (:Label_T7 {x: 980}) CREATE (:Label_T7 {x: 869}) CREATE "
+            "(:Label_T7 {x: 86}) CREATE (:Label_T7 {x: 521}) CREATE (:Label_T7 "
+            "{x: 171}) CREATE (:Label_T7 {x: 161}) CREATE (:Label_T7 {x: 260}) "
+            "CREATE (:Label_T7 {x: 857}) CREATE (:Label_T7 {x: 34}) CREATE "
+            "(:Label_T7 {x: 969}) CREATE (:Label_T7 {x: 948}) CREATE "
+            "(:Label_T7 {x: 519}) CREATE (:Label_T7 {x: 333}) CREATE "
+            "(:Label_T7 {x: 733}) CREATE (:Label_T7 {x: 870}) CREATE "
+            "(:Label_T7 {x: 129}) CREATE (:Label_T7 {x: 77}) CREATE (:Label_T7 "
+            "{x: 54}) CREATE (:Label_T7 {x: 109}) CREATE (:Label_T7 {x: 908}) "
+            "CREATE (:Label_T7 {x: 560}) CREATE (:Label_T7 {x: 928}) CREATE "
+            "(:Label_T7 {x: 934}) CREATE (:Label_T7 {x: 475}) CREATE "
+            "(:Label_T7 {x: 619}) CREATE (:Label_T7 {x: 787}) CREATE "
+            "(:Label_T7 {x: 739}) CREATE (:Label_T7 {x: 622}) CREATE "
+            "(:Label_T7 {x: 588}) CREATE (:Label_T7 {x: 917}) CREATE "
+            "(:Label_T7 {x: 45}) CREATE (:Label_T7 {x: 696}) CREATE (:Label_T7 "
+            "{x: 998}) CREATE (:Label_T7 {x: 232}) CREATE (:Label_T7 {x: 600}) "
+            "CREATE (:Label_T7 {x: 763}) CREATE (:Label_T7 {x: 511}) CREATE "
+            "(:Label_T7 {x: 501}) CREATE (:Label_T7 {x: 555}) CREATE "
+            "(:Label_T7 {x: 491}) CREATE (:Label_T7 {x: 896}) CREATE "
+            "(:Label_T7 {x: 95}) CREATE (:Label_T7 {x: 361}) CREATE (:Label_T7 "
+            "{x: 132}) CREATE (:Label_T7 {x: 637}) CREATE (:Label_T7 {x: 427}) "
+            "CREATE (:Label_T7 {x: 37}) CREATE (:Label_T7 {x: 78}) CREATE "
+            "(:Label_T7 {x: 405}) CREATE (:Label_T7 {x: 821}) CREATE "
+            "(:Label_T7 {x: 186}) CREATE (:Label_T7 {x: 66}) CREATE (:Label_T7 "
+            "{x: 450}) CREATE (:Label_T7 {x: 709}) CREATE (:Label_T7 {x: 424}) "
+            "CREATE (:Label_T7 {x: 698}) CREATE (:Label_T7 {x: 42}) CREATE "
+            "(:Label_T7 {x: 629}) CREATE (:Label_T7 {x: 121}) CREATE "
+            "(:Label_T7 {x: 891}) CREATE (:Label_T7 {x: 823}) CREATE "
+            "(:Label_T7 {x: 337}) CREATE (:Label_T7 {x: 139}) CREATE "
+            "(:Label_T7 {x: 704}) CREATE (:Label_T7 {x: 141}) CREATE "
+            "(:Label_T7 {x: 438}) CREATE (:Label_T7 {x: 793}) CREATE "
+            "(:Label_T7 {x: 42}) CREATE (:Label_T7 {x: 142}) CREATE (:Label_T7 "
+            "{x: 526}) CREATE (:Label_T7 {x: 732}) CREATE (:Label_T7 {x: "
+            "336})");
+        parser.tree();
+      }
+    }));
+  }
+
+  run = true;
+  std::this_thread::sleep_for(10ms);
+  run = false;
+
+  // cleanup threads
+  for (int i = 0; i < N; ++i) threads[i].join();
+}
+
+int main(int argc, char **argv) {
+  google::InitGoogleLogging(argv[0]);
+  ::testing::InitGoogleTest(&argc, argv);
+
+  // Signal handling init.
+  SignalHandler::register_handler(Signal::SegmentationFault, []() {
+    // Log that we got SIGSEGV and abort the program, because returning from
+    // SIGSEGV handler is undefined behaviour.
+    std::cerr << "SegmentationFault signal raised" << std::endl;
+    std::abort();  // This will continue into our SIGABRT handler.
+  });
+  SignalHandler::register_handler(Signal::Abort, []() {
+    // Log the stacktrace and let the abort continue.
+    Stacktrace stacktrace;
+    std::cerr << "Abort signal raised" << std::endl
+              << stacktrace.dump() << std::endl;
+  });
+
+  return RUN_ALL_TESTS();
+}
diff --git a/tests/manual/test_antlr_sigsegv b/tests/manual/test_antlr_sigsegv
new file mode 100755
index 000000000..c8179b771
--- /dev/null
+++ b/tests/manual/test_antlr_sigsegv
@@ -0,0 +1,8 @@
+#!/bin/bash
+
+DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" && pwd )"
+cd "$DIR"
+
+for i in {1..100}; do
+    ../../build/tests/manual/antlr_sigsegv || exit 1
+done