heap/MachineStackMarker.h

   1 /*
   2  *  Copyright (C) 1999-2000 Harri Porten (porten@kde.org)
   3  *  Copyright (C) 2001 Peter Kelly (pmk@post.com)
   4  *  Copyright (C) 2003, 2004, 2005, 2006, 2007, 2008, 2009 Apple Inc. All rights reserved.
   5  *
   6  *  This library is free software; you can redistribute it and/or
   7  *  modify it under the terms of the GNU Lesser General Public
   8  *  License as published by the Free Software Foundation; either
   9  *  version 2 of the License, or (at your option) any later version.
  10  *
  11  *  This library is distributed in the hope that it will be useful,
  12  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  *  Lesser General Public License for more details.
  15  *
  16  *  You should have received a copy of the GNU Lesser General Public
  17  *  License along with this library; if not, write to the Free Software
  18  *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  19  *
  20  */
  21
  22 #ifndef MachineThreads_h
  23 #define MachineThreads_h
  24
  25 #include <setjmp.h>
  26 #include <wtf/Noncopyable.h>
  27 #include <wtf/ThreadSpecific.h>
  28 #include <wtf/ThreadingPrimitives.h>
  29
  30 namespace JSC {
  31
  32     class CodeBlockSet;
  33     class ConservativeRoots;
  34     class Heap;
  35     class JITStubRoutineSet;
  36
  37     class MachineThreads {
  38         WTF_MAKE_NONCOPYABLE(MachineThreads);
  39     public:
  40         typedef jmp_buf RegisterState;
  41
  42         MachineThreads(Heap*);
  43         ~MachineThreads();
  44
  45         void gatherConservativeRoots(ConservativeRoots&, JITStubRoutineSet&, CodeBlockSet&, void* stackCurrent, RegisterState& registers);
  46
  47         JS_EXPORT_PRIVATE void makeUsableFromMultipleThreads();
  48         JS_EXPORT_PRIVATE void addCurrentThread(); // Only needs to be called by clients that can use the same heap from multiple threads.
  49
  50     private:
  51         void gatherFromCurrentThread(ConservativeRoots&, JITStubRoutineSet&, CodeBlockSet&, void* stackCurrent, RegisterState& registers);
  52
  53         class Thread;
  54
  55         static void removeThread(void*);
  56         void removeCurrentThread();
  57
  58         void gatherFromOtherThread(ConservativeRoots&, Thread*, JITStubRoutineSet&, CodeBlockSet&);
  59
  60         Mutex m_registeredThreadsMutex;
  61         Thread* m_registeredThreads;
  62         WTF::ThreadSpecificKey m_threadSpecific;
  63 #if !ASSERT_DISABLED
  64         Heap* m_heap;
  65 #endif
  66     };
  67
  68 } // namespace JSC
  69
  70 #if COMPILER(GCC)
  71 #define REGISTER_BUFFER_ALIGNMENT __attribute__ ((aligned (sizeof(void*))))
  72 #else
  73 #define REGISTER_BUFFER_ALIGNMENT
  74 #endif
  75
  76 // ALLOCATE_AND_GET_REGISTER_STATE() is a macro so that it is always "inlined" even in debug builds.
  77 #if COMPILER(MSVC)
  78 #pragma warning(push)
  79 #pragma warning(disable: 4611)
  80 #define ALLOCATE_AND_GET_REGISTER_STATE(registers) \
  81     MachineThreads::RegisterState registers REGISTER_BUFFER_ALIGNMENT; \
  82     setjmp(registers)
  83 #pragma warning(pop)
  84 #else
  85 #define ALLOCATE_AND_GET_REGISTER_STATE(registers) \
  86     MachineThreads::RegisterState registers REGISTER_BUFFER_ALIGNMENT; \
  87     setjmp(registers)
  88 #endif
  89
  90 #endif // MachineThreads_h