PCRE2: optimize memory allocations

src/crystal/system/thread.cr

@@ -220,12 +220,16 @@ class Thread
     rescue ex
       @exception = ex
     ensure
+      run_destructors
       Thread.threads.delete(self)
       Fiber.inactive(fiber)
       detach { system_close }
     end
   end
 
+  protected def run_destructors : Nil
+  end
+
   protected def name=(@name : String)
     self.system_name = name
   end

src/crystal/system/thread_local.cr

@@ -0,0 +1,33 @@
+class Thread
+  struct Local(T)
+    # def initialize
+    #   {% raise "T must be a Reference or Pointer" unless T < Reference || T < Pointer %}
+    # end
+
+    # def initialize(&destructor : Proc(T, Nil))
+    #   {% raise "T must be a Reference or Pointer" unless T < Reference || T < Pointer %}
+    # end
+
+    def get : T
+      get? || raise KeyError.new
+    end
+
+    def get(& : -> T) : T
+      get? || set(yield)
+    end
+
+    # def get? : T?
+
+    # def set(value : T) : T
+  end
+end
+
+{% if flag?(:wasi) %}
+  require "./wasi/thread_local"
+{% elsif flag?(:unix) %}
+  require "./unix/thread_local"
+{% elsif flag?(:win32) %}
+  require "./win32/thread_local"
+{% else %}
+  {% raise "Thread not supported" %}
+{% end %}

src/crystal/system/unix/thread_local.cr

@@ -0,0 +1,30 @@
+require "c/pthread"
+
+class Thread
+  struct Local(T)
+    @key : LibC::PthreadKeyT
+
+    def initialize
+      {% raise "T must be a Reference or Pointer" unless T < Reference || T < Pointer %}
+      err = LibC.pthread_key_create(out @key, nil)
+      raise RuntimeError.from_os_error("pthread_key_create", Errno.new(err)) unless err == 0
+    end
+
+    def initialize(&destructor : Proc(T, Nil))
+      {% raise "T must be a Reference or Pointer" unless T < Reference || T < Pointer %}
+      err = LibC.pthread_key_create(out @key, destructor.unsafe_as(Proc(Void*, Nil)))
+      raise RuntimeError.from_os_error("pthread_key_create", Errno.new(err)) unless err == 0
+    end
+
+    def get? : T?
+      pointer = LibC.pthread_getspecific(@key)
+      pointer.as(T) if pointer
+    end
+
+    def set(value : T) : T
+      err = LibC.pthread_setspecific(@key, value.as(Void*))
+      raise RuntimeError.from_os_error("pthread_setspecific", Errno.new(err)) unless err == 0
+      value
+    end
+  end
+end

src/crystal/system/wasi/thread_local.cr

@@ -0,0 +1,21 @@
+class Thread
+  struct Local(T)
+    @value : T?
+
+    def initialize
+      {% raise "T must be a Reference or Pointer" unless T < Reference || T < Pointer %}
+    end
+
+    def initialize(&destructor : T ->)
+      {% raise "T must be a Reference or Pointer" unless T < Reference || T < Pointer %}
+    end
+
+    def get? : T?
+      @value
+    end
+
+    def set(value : T) : T
+      @value = value
+    end
+  end
+end

src/crystal/system/win32/thread_local.cr

@@ -0,0 +1,30 @@
+require "c/fibersapi"
+
+class Thread
+  struct Local(T)
+    @key : LibC::DWORD
+
+    def initialize
+      {% raise "T must be a Reference or Pointer" unless T < Reference || T < Pointer %}
+      @key = LibC.FlsAlloc
+      raise RuntimeError.from_winerror("FlsAlloc: out of indexes") if @key == LibC::FLS_OUT_OF_INDEXES
+    end
+
+    def initialize(&destructor : Proc(T, Nil))
+      {% raise "T must be a Reference or Pointer" unless T < Reference || T < Pointer %}
+      @key = LibC.FlsAlloc(destructor.unsafe_as(Proc(Void*, Nil)))
+      raise RuntimeError.from_winerror("FlsAlloc: out of indexes") if @key == LibC::FLS_OUT_OF_INDEXES
+    end
+
+    def get? : T?
+      pointer = LibC.FlsGetValue(@key)
+      pointer.as(T) if pointer
+    end
+
+    def set(value : T) : T
+      ret = LibC.FlsSetValue(@key, value.as(Void*))
+      raise RuntimeError.from_winerror("FlsSetValue") if ret == 0
+      value
+    end
+  end
+end

src/lib_c/aarch64-android/c/pthread.cr

@@ -26,6 +26,7 @@ lib LibC
   fun pthread_getspecific(__key : PthreadKeyT) : Void*
   fun pthread_join(__pthread : PthreadT, __return_value_ptr : Void**) : Int
   fun pthread_key_create(__key_ptr : PthreadKeyT*, __key_destructor : Void* ->) : Int
+  fun pthread_key_delete(__key_ptr : PthreadKeyT) : Int
 
   fun pthread_mutexattr_destroy(__attr : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_init(__attr : PthreadMutexattrT*) : Int

src/lib_c/aarch64-darwin/c/pthread.cr

@@ -13,9 +13,13 @@ lib LibC
   fun pthread_cond_wait(x0 : PthreadCondT*, x1 : PthreadMutexT*) : Int
   fun pthread_create(x0 : PthreadT*, x1 : PthreadAttrT*, x2 : Void* -> Void*, x3 : Void*) : Int
   fun pthread_detach(x0 : PthreadT) : Int
+  fun pthread_getspecific(PthreadKeyT) : Void*
   fun pthread_get_stackaddr_np(x0 : PthreadT) : Void*
   fun pthread_get_stacksize_np(x0 : PthreadT) : SizeT
   fun pthread_join(x0 : PthreadT, x1 : Void**) : Int
+  alias PthreadKeyDestructor = (Void*) ->
+  fun pthread_key_create(PthreadKeyT*, PthreadKeyDestructor) : Int
+  fun pthread_key_delete(PthreadKeyT) : Int
   fun pthread_mutexattr_destroy(x0 : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_init(x0 : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_settype(x0 : PthreadMutexattrT*, x1 : Int) : Int
@@ -26,4 +30,5 @@ lib LibC
   fun pthread_mutex_unlock(x0 : PthreadMutexT*) : Int
   fun pthread_self : PthreadT
   fun pthread_setname_np(Char*) : Int
+  fun pthread_setspecific(PthreadKeyT, Void*) : Int
 end

src/lib_c/aarch64-darwin/c/sys/types.cr

@@ -40,6 +40,7 @@ lib LibC
   end
 
   type PthreadT = Void*
+  alias PthreadKeyT = UInt
   alias SSizeT = Long
   alias SusecondsT = Int
   alias TimeT = Long

src/lib_c/aarch64-linux-gnu/c/pthread.cr

@@ -25,8 +25,12 @@ lib LibC
   fun pthread_create(newthread : PthreadT*, attr : PthreadAttrT*, start_routine : Void* -> Void*, arg : Void*) : Int
   fun pthread_detach(th : PthreadT) : Int
   fun pthread_getattr_np(thread : PthreadT, attr : PthreadAttrT*) : Int
+  fun pthread_getspecific(PthreadKeyT) : Void*
   fun pthread_equal(thread1 : PthreadT, thread2 : PthreadT) : Int
   fun pthread_join(th : PthreadT, thread_return : Void**) : Int
+  alias PthreadKeyDestructor = (Void*) ->
+  fun pthread_key_create(PthreadKeyT*, PthreadKeyDestructor) : Int
+  fun pthread_key_delete(PthreadKeyT) : Int
   fun pthread_mutexattr_destroy(attr : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_init(attr : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_settype(attr : PthreadMutexattrT*, type : Int) : Int
@@ -37,4 +41,5 @@ lib LibC
   fun pthread_mutex_unlock(mutex : PthreadMutexT*) : Int
   fun pthread_self : PthreadT
   fun pthread_setname_np(PthreadT, Char*) : Int
+  fun pthread_setspecific(PthreadKeyT, Void*) : Int
 end

src/lib_c/aarch64-linux-gnu/c/sys/types.cr

@@ -41,6 +41,7 @@ lib LibC
   end
 
   alias PthreadT = ULong
+  alias PthreadKeyT = UInt
   alias SSizeT = Long
   alias SusecondsT = Long
   alias TimeT = Long

src/lib_c/aarch64-linux-musl/c/pthread.cr

@@ -17,7 +17,11 @@ lib LibC
   fun pthread_create(x0 : PthreadT*, x1 : PthreadAttrT*, x2 : Void* -> Void*, x3 : Void*) : Int
   fun pthread_detach(x0 : PthreadT) : Int
   fun pthread_getattr_np(x0 : PthreadT, x1 : PthreadAttrT*) : Int
+  fun pthread_getspecific(PthreadKeyT) : Void*
   fun pthread_join(x0 : PthreadT, x1 : Void**) : Int
+  alias PthreadKeyDestructor = (Void*) ->
+  fun pthread_key_create(PthreadKeyT*, PthreadKeyDestructor) : Int
+  fun pthread_key_delete(PthreadKeyT) : Int
   fun pthread_mutexattr_destroy(x0 : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_init(x0 : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_settype(x0 : PthreadMutexattrT*, x1 : Int) : Int
@@ -28,4 +32,5 @@ lib LibC
   fun pthread_mutex_unlock(x0 : PthreadMutexT*) : Int
   fun pthread_self : PthreadT
   fun pthread_setname_np(PthreadT, Char*) : Int
+  fun pthread_setspecific(PthreadKeyT, Void*) : Int
 end

src/lib_c/aarch64-linux-musl/c/sys/types.cr

@@ -54,6 +54,7 @@ lib LibC
   end
 
   type PthreadT = Void*
+  alias PthreadKeyT = UInt
   alias SSizeT = Long
   alias SusecondsT = Long
   alias TimeT = Long

src/lib_c/arm-linux-gnueabihf/c/pthread.cr

@@ -25,8 +25,12 @@ lib LibC
   fun pthread_create(newthread : PthreadT*, attr : PthreadAttrT*, start_routine : Void* -> Void*, arg : Void*) : Int
   fun pthread_detach(th : PthreadT) : Int
   fun pthread_getattr_np(thread : PthreadT, attr : PthreadAttrT*) : Int
+  fun pthread_getspecific(PthreadKeyT) : Void*
   fun pthread_equal(thread1 : PthreadT, thread2 : PthreadT) : Int
   fun pthread_join(th : PthreadT, thread_return : Void**) : Int
+  alias PthreadKeyDestructor = (Void*) ->
+  fun pthread_key_create(PthreadKeyT*, PthreadKeyDestructor) : Int
+  fun pthread_key_delete(PthreadKeyT) : Int
   fun pthread_mutexattr_destroy(attr : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_init(attr : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_settype(attr : PthreadMutexattrT*, type : Int) : Int
@@ -37,4 +41,5 @@ lib LibC
   fun pthread_mutex_unlock(mutex : PthreadMutexT*) : Int
   fun pthread_self : PthreadT
   fun pthread_setname_np(PthreadT, Char*) : Int
+  fun pthread_setspecific(PthreadKeyT, Void*) : Int
 end

src/lib_c/arm-linux-gnueabihf/c/sys/types.cr

@@ -54,6 +54,7 @@ lib LibC
   end
 
   alias PthreadT = ULong
+  alias PthreadKeyT = UInt
   alias SSizeT = Int
   alias SusecondsT = Long
   alias TimeT = Long

src/lib_c/i386-linux-gnu/c/pthread.cr

@@ -25,7 +25,11 @@ lib LibC
   fun pthread_create(newthread : PthreadT*, attr : PthreadAttrT*, start_routine : Void* -> Void*, arg : Void*) : Int
   fun pthread_detach(th : PthreadT) : Int
   fun pthread_getattr_np(thread : PthreadT, attr : PthreadAttrT*) : Int
+  fun pthread_getspecific(PthreadKeyT) : Void*
   fun pthread_join(th : PthreadT, thread_return : Void**) : Int
+  alias PthreadKeyDestructor = (Void*) ->
+  fun pthread_key_create(PthreadKeyT*, PthreadKeyDestructor) : Int
+  fun pthread_key_delete(PthreadKeyT) : Int
   fun pthread_mutexattr_destroy(attr : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_init(attr : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_settype(attr : PthreadMutexattrT*, type : Int) : Int
@@ -36,4 +40,5 @@ lib LibC
   fun pthread_mutex_unlock(mutex : PthreadMutexT*) : Int
   fun pthread_self : PthreadT
   fun pthread_setname_np(PthreadT, Char*) : Int
+  fun pthread_setspecific(PthreadKeyT, Void*) : Int
 end

src/lib_c/i386-linux-gnu/c/sys/types.cr

@@ -41,6 +41,7 @@ lib LibC
   end
 
   alias PthreadT = ULong
+  alias PthreadKeyT = UInt
   alias SSizeT = Int
   alias SusecondsT = Long
   alias TimeT = Long

src/lib_c/i386-linux-musl/c/pthread.cr

@@ -17,7 +17,11 @@ lib LibC
   fun pthread_create(x0 : PthreadT*, x1 : PthreadAttrT*, x2 : Void* -> Void*, x3 : Void*) : Int
   fun pthread_detach(x0 : PthreadT) : Int
   fun pthread_getattr_np(x0 : PthreadT, x1 : PthreadAttrT*) : Int
+  fun pthread_getspecific(PthreadKeyT) : Void*
   fun pthread_join(x0 : PthreadT, x1 : Void**) : Int
+  alias PthreadKeyDestructor = (Void*) ->
+  fun pthread_key_create(PthreadKeyT*, PthreadKeyDestructor) : Int
+  fun pthread_key_delete(PthreadKeyT) : Int
   fun pthread_mutexattr_destroy(x0 : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_init(x0 : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_settype(x0 : PthreadMutexattrT*, x1 : Int) : Int
@@ -28,4 +32,5 @@ lib LibC
   fun pthread_mutex_unlock(x0 : PthreadMutexT*) : Int
   fun pthread_self : PthreadT
   fun pthread_setname_np(PthreadT, Char*) : Int
+  fun pthread_setspecific(PthreadKeyT, Void*) : Int
 end

src/lib_c/i386-linux-musl/c/sys/types.cr

@@ -54,6 +54,7 @@ lib LibC
   end
 
   type PthreadT = Void*
+  alias PthreadKeyT = UInt
   alias SSizeT = Int
   alias SusecondsT = Long
   alias TimeT = Long

src/lib_c/x86_64-darwin/c/pthread.cr

@@ -15,7 +15,11 @@ lib LibC
   fun pthread_detach(x0 : PthreadT) : Int
   fun pthread_get_stackaddr_np(x0 : PthreadT) : Void*
   fun pthread_get_stacksize_np(x0 : PthreadT) : SizeT
+  fun pthread_getspecific(PthreadKeyT) : Void*
   fun pthread_join(x0 : PthreadT, x1 : Void**) : Int
+  alias PthreadKeyDestructor = (Void*) ->
+  fun pthread_key_create(PthreadKeyT*, PthreadKeyDestructor) : Int
+  fun pthread_key_delete(PthreadKeyT) : Int
   fun pthread_mutexattr_destroy(x0 : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_init(x0 : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_settype(x0 : PthreadMutexattrT*, x1 : Int) : Int
@@ -26,4 +30,5 @@ lib LibC
   fun pthread_mutex_unlock(x0 : PthreadMutexT*) : Int
   fun pthread_self : PthreadT
   fun pthread_setname_np(Char*) : Int
+  fun pthread_setspecific(PthreadKeyT, Void*) : Int
 end

src/lib_c/x86_64-darwin/c/sys/types.cr

@@ -40,6 +40,7 @@ lib LibC
   end
 
   type PthreadT = Void*
+  alias PthreadKeyT = UInt
   alias SSizeT = Long
   alias SusecondsT = Int
   alias TimeT = Long

src/lib_c/x86_64-dragonfly/c/pthread.cr

@@ -19,7 +19,11 @@ lib LibC
   fun pthread_cond_wait(x0 : PthreadCondT*, x1 : PthreadMutexT*) : Int
   fun pthread_create(x0 : PthreadT*, x1 : PthreadAttrT*, x2 : Void* -> Void*, x3 : Void*) : Int
   fun pthread_detach(x0 : PthreadT) : Int
+  fun pthread_getspecific(PthreadKeyT) : Void*
   fun pthread_join(x0 : PthreadT, x1 : Void**) : Int
+  alias PthreadKeyDestructor = (Void*) ->
+  fun pthread_key_create(PthreadKeyT*, PthreadKeyDestructor) : Int
+  fun pthread_key_delete(PthreadKeyT) : Int
   fun pthread_mutexattr_destroy(x0 : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_init(x0 : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_settype(x0 : PthreadMutexattrT*, x1 : Int) : Int
@@ -30,4 +34,5 @@ lib LibC
   fun pthread_mutex_unlock(x0 : PthreadMutexT*) : Int
   fun pthread_self : PthreadT
   fun pthread_setname_np(PthreadT, Char*) : Int
+  fun pthread_setspecific(PthreadKeyT, Void*) : Int
 end

src/lib_c/x86_64-dragonfly/c/sys/types.cr

@@ -17,6 +17,7 @@ lib LibC
   type PthreadAttrT = Void*
   type PthreadCondT = Void*
   type PthreadCondattrT = Void*
+  alias PthreadKeyT = Int
   type PthreadMutexT = Void*
   type PthreadMutexattrT = Void*
   type PthreadT = Void*

src/lib_c/x86_64-freebsd/c/pthread.cr

@@ -19,7 +19,11 @@ lib LibC
   fun pthread_cond_wait(x0 : PthreadCondT*, x1 : PthreadMutexT*) : Int
   fun pthread_create(x0 : PthreadT*, x1 : PthreadAttrT*, x2 : Void* -> Void*, x3 : Void*) : Int
   fun pthread_detach(x0 : PthreadT) : Int
+  fun pthread_getspecific(PthreadKeyT) : Void*
   fun pthread_join(x0 : PthreadT, x1 : Void**) : Int
+  alias PthreadKeyDestructor = (Void*) ->
+  fun pthread_key_create(PthreadKeyT*, PthreadKeyDestructor) : Int
+  fun pthread_key_delete(PthreadKeyT) : Int
   fun pthread_mutexattr_destroy(x0 : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_init(x0 : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_settype(x0 : PthreadMutexattrT*, x1 : Int) : Int
@@ -30,4 +34,5 @@ lib LibC
   fun pthread_mutex_unlock(x0 : PthreadMutexT*) : Int
   fun pthread_self : PthreadT
   fun pthread_set_name_np(PthreadT, Char*)
+  fun pthread_setspecific(PthreadKeyT, Void*) : Int
 end

src/lib_c/x86_64-freebsd/c/sys/types.cr

@@ -17,6 +17,7 @@ lib LibC
   type PthreadAttrT = Void*
   type PthreadCondT = Void*
   type PthreadCondattrT = Void*
+  alias PthreadKeyT = Int
   type PthreadMutexT = Void*
   type PthreadMutexattrT = Void*
   type PthreadT = Void*

src/lib_c/x86_64-linux-gnu/c/pthread.cr

@@ -25,7 +25,11 @@ lib LibC
   fun pthread_create(newthread : PthreadT*, attr : PthreadAttrT*, start_routine : Void* -> Void*, arg : Void*) : Int
   fun pthread_detach(th : PthreadT) : Int
   fun pthread_getattr_np(thread : PthreadT, attr : PthreadAttrT*) : Int
+  fun pthread_getspecific(PthreadKeyT) : Void*
   fun pthread_join(th : PthreadT, thread_return : Void**) : Int
+  alias PthreadKeyDestructor = (Void*) ->
+  fun pthread_key_create(PthreadKeyT*, PthreadKeyDestructor) : Int
+  fun pthread_key_delete(PthreadKeyT) : Int
   fun pthread_mutexattr_destroy(attr : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_init(attr : PthreadMutexattrT*) : Int
   fun pthread_mutexattr_settype(attr : PthreadMutexattrT*, type : Int) : Int
@@ -36,4 +40,5 @@ lib LibC
   fun pthread_mutex_unlock(mutex : PthreadMutexT*) : Int
   fun pthread_self : PthreadT
   fun pthread_setname_np(PthreadT, Char*) : Int
+  fun pthread_setspecific(PthreadKeyT, Void*) : Int
 end