1 /*
   2  * Copyright (c) 2018, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 #ifndef GTEST_UTILITIES_HELPER_INLINE_HPP
  25 #define GTEST_UTILITIES_HELPER_INLINE_HPP
  26 
  27 #include "runtime/mutex.hpp"
  28 #include "runtime/semaphore.hpp"
  29 #include "runtime/thread.hpp"
  30 #include "runtime/vmThread.hpp"
  31 #include "runtime/vm_operations.hpp"
  32 #include "unittest.hpp"
  33 
  34 class VM_StopSafepoint : public VM_Operation {
  35 public:
  36   Semaphore* _test_complete;
  37   VM_StopSafepoint(Semaphore* wait_for) : _test_complete(wait_for) {}
  38   VMOp_Type type() const          { return VMOp_Dummy; }
  39   Mode evaluation_mode() const    { return _no_safepoint; }
  40   bool is_cheap_allocated() const { return false; }
  41   void doit()                     { _test_complete->wait(); }
  42 };
  43 
  44 // This class and thread keep the non-safepoint op running while we do our testing.
  45 class VMThreadBlocker : public JavaThread {
  46 public:
  47   Semaphore* _unblock;
  48   Semaphore* _done;
  49   VMThreadBlocker(Semaphore* ub, Semaphore* done) : _unblock(ub), _done(done) {
  50   }
  51   virtual ~VMThreadBlocker() {}
  52   void run() {
  53     this->set_thread_state(_thread_in_vm);
  54     {
  55       MutexLocker ml(Threads_lock);
  56       Threads::add(this);
  57     }
  58     VM_StopSafepoint ss(_unblock);
  59     VMThread::execute(&ss);
  60     _done->signal();
  61     Threads::remove(this);
  62     this->smr_delete();
  63   }
  64   void doit() {
  65     if (os::create_thread(this, os::os_thread)) {
  66       os::start_thread(this);
  67     } else {
  68       ASSERT_TRUE(false);
  69     }
  70   }
  71 };
  72 
  73 // For testing in a real JavaThread.
  74 class JavaTestThread : public JavaThread {
  75 public:
  76   Semaphore* _post;
  77   JavaTestThread(Semaphore* post)
  78     : _post(post) {
  79   }
  80   virtual ~JavaTestThread() {}
  81 
  82   void prerun() {
  83     this->set_thread_state(_thread_in_vm);
  84     {
  85       MutexLocker ml(Threads_lock);
  86       Threads::add(this);
  87     }
  88     {
  89       MutexLockerEx ml(SR_lock(), Mutex::_no_safepoint_check_flag);
  90     }
  91   }
  92 
  93   virtual void main_run() = 0;
  94 
  95   void run() {
  96     prerun();
  97     main_run();
  98     postrun();
  99   }
 100 
 101   void postrun() {
 102     Threads::remove(this);
 103     _post->signal();
 104     this->smr_delete();
 105   }
 106 
 107   void doit() {
 108     if (os::create_thread(this, os::os_thread)) {
 109       os::start_thread(this);
 110     } else {
 111       ASSERT_TRUE(false);
 112     }
 113   }
 114 };
 115 
 116 template <typename FUNC>
 117 class SingleTestThread : public JavaTestThread {
 118 public:
 119   FUNC& _f;
 120   SingleTestThread(Semaphore* post, FUNC& f)
 121     : JavaTestThread(post), _f(f) {
 122   }
 123 
 124   virtual ~SingleTestThread(){}
 125 
 126   virtual void main_run() {
 127     _f(this);
 128   }
 129 };
 130 
 131 template <typename TESTFUNC>
 132 static void nomt_test_doer(TESTFUNC &f) {
 133   Semaphore post, block_done, vmt_done;
 134   VMThreadBlocker* blocker = new VMThreadBlocker(&block_done, &vmt_done);
 135   blocker->doit();
 136   SingleTestThread<TESTFUNC>* stt = new SingleTestThread<TESTFUNC>(&post, f);
 137   stt->doit();
 138   post.wait();
 139   block_done.signal();
 140   vmt_done.wait();
 141 }
 142 
 143 template <typename RUNNER>
 144 static void mt_test_doer() {
 145   Semaphore post, block_done, vmt_done;
 146   VMThreadBlocker* blocker = new VMThreadBlocker(&block_done, &vmt_done);
 147   blocker->doit();
 148   RUNNER* runner = new RUNNER(&post);
 149   runner->doit();
 150   post.wait();
 151   block_done.signal();
 152   vmt_done.wait();
 153 }
 154 
 155 #endif // include guard