update submodules for GHC.HetMet.GArrow -> Control.GArrow renaming
[ghc-hetmet.git] / rts / Adjustor.c
index e021487..607c0b8 100644 (file)
@@ -38,15 +38,25 @@ Haskell side.
 
 #include "PosixSource.h"
 #include "Rts.h"
-#include "RtsExternal.h"
+
 #include "RtsUtils.h"
-#include <stdlib.h>
+#include "Stable.h"
 
 #if defined(USE_LIBFFI_FOR_ADJUSTORS)
-
 #include "ffi.h"
 #include <string.h>
+#endif
 
+#if defined(i386_HOST_ARCH) && defined(darwin_HOST_OS)
+extern void adjustorCode(void);
+#elif defined(powerpc_HOST_ARCH) || defined(powerpc64_HOST_ARCH)
+// from AdjustorAsm.s
+// not declared as a function so that AIX-style
+// fundescs can never get in the way.
+extern void *adjustorCode;
+#endif
+
+#if defined(USE_LIBFFI_FOR_ADJUSTORS)
 void
 freeHaskellFunctionPtr(void* ptr)
 {
@@ -101,7 +111,7 @@ createAdjustor (int cconv,
         arg_types[i] = char_to_ffi_type(typeString[i+1]);
     }
     switch (cconv) {
-#ifdef mingw32_TARGET_OS
+#ifdef mingw32_HOST_OS
     case 0: /* stdcall */
         abi = FFI_STDCALL;
         break;
@@ -210,12 +220,12 @@ stgAllocStable(size_t size_in_bytes, StgStablePtr *stable)
   nat data_size_in_words, total_size_in_words;
   
   /* round up to a whole number of words */
-  data_size_in_words  = (size_in_bytes + sizeof(W_) + 1) / sizeof(W_);
+  data_size_in_words  = ROUNDUP_BYTES_TO_WDS(size_in_bytes);
   total_size_in_words = sizeofW(StgArrWords) + data_size_in_words;
   
   /* allocate and fill it in */
   arr = (StgArrWords *)allocate(total_size_in_words);
-  SET_ARR_HDR(arr, &stg_ARR_WORDS_info, CCCS, data_size_in_words);
+  SET_ARR_HDR(arr, &stg_ARR_WORDS_info, CCCS, size_in_bytes);
  
   /* obtain a stable ptr */
   *stable = getStablePtr((StgPtr)arr);
@@ -294,7 +304,7 @@ typedef struct AdjustorStub {
 } AdjustorStub;
 #endif
 
-#if defined(darwin_HOST_OS) || defined(powerpc_HOST_ARCH) || defined(powerpc64_HOST_ARCH)
+#if (defined(i386_HOST_ARCH) && defined(darwin_HOST_OS)) || defined(powerpc_HOST_ARCH) || defined(powerpc64_HOST_ARCH)
 static int totalArgumentSize(char *typeString)
 {
     int sz = 0;
@@ -423,7 +433,6 @@ createAdjustor(int cconv, StgStablePtr hptr,
         AdjustorStub *adjustorStub = allocateExec(sizeof(AdjustorStub),&code);
         adjustor = adjustorStub;
 
-        extern void adjustorCode(void);
         int sz = totalArgumentSize(typeString);
         
         adjustorStub->call[0] = 0xe8;
@@ -509,6 +518,7 @@ createAdjustor(int cconv, StgStablePtr hptr,
     {  
        int i = 0;
        char *c;
+       StgWord8 *adj_code;
 
        // determine whether we have 6 or more integer arguments,
        // and therefore need to flush one to the stack.
@@ -519,34 +529,36 @@ createAdjustor(int cconv, StgStablePtr hptr,
 
        if (i < 6) {
            adjustor = allocateExec(0x30,&code);
-
-           *(StgInt32 *)adjustor        = 0x49c1894d;
-           *(StgInt32 *)(adjustor+0x4)  = 0x8948c889;
-           *(StgInt32 *)(adjustor+0x8)  = 0xf28948d1;
-           *(StgInt32 *)(adjustor+0xc)  = 0x48fe8948;
-           *(StgInt32 *)(adjustor+0x10) = 0x000a3d8b;
-           *(StgInt32 *)(adjustor+0x14) = 0x25ff0000;
-           *(StgInt32 *)(adjustor+0x18) = 0x0000000c;
-           *(StgInt64 *)(adjustor+0x20) = (StgInt64)hptr;
-           *(StgInt64 *)(adjustor+0x28) = (StgInt64)wptr;
+            adj_code = (StgWord8*)adjustor;
+
+           *(StgInt32 *)adj_code        = 0x49c1894d;
+           *(StgInt32 *)(adj_code+0x4)  = 0x8948c889;
+           *(StgInt32 *)(adj_code+0x8)  = 0xf28948d1;
+           *(StgInt32 *)(adj_code+0xc)  = 0x48fe8948;
+           *(StgInt32 *)(adj_code+0x10) = 0x000a3d8b;
+           *(StgInt32 *)(adj_code+0x14) = 0x25ff0000;
+           *(StgInt32 *)(adj_code+0x18) = 0x0000000c;
+           *(StgInt64 *)(adj_code+0x20) = (StgInt64)hptr;
+           *(StgInt64 *)(adj_code+0x28) = (StgInt64)wptr;
        }
        else
        {
            adjustor = allocateExec(0x40,&code);
-
-           *(StgInt32 *)adjustor        = 0x35ff5141;
-           *(StgInt32 *)(adjustor+0x4)  = 0x00000020;
-           *(StgInt32 *)(adjustor+0x8)  = 0x49c1894d;
-           *(StgInt32 *)(adjustor+0xc)  = 0x8948c889;
-           *(StgInt32 *)(adjustor+0x10) = 0xf28948d1;
-           *(StgInt32 *)(adjustor+0x14) = 0x48fe8948;
-           *(StgInt32 *)(adjustor+0x18) = 0x00123d8b;
-           *(StgInt32 *)(adjustor+0x1c) = 0x25ff0000;
-           *(StgInt32 *)(adjustor+0x20) = 0x00000014;
+            adj_code = (StgWord8*)adjustor;
+
+           *(StgInt32 *)adj_code        = 0x35ff5141;
+           *(StgInt32 *)(adj_code+0x4)  = 0x00000020;
+           *(StgInt32 *)(adj_code+0x8)  = 0x49c1894d;
+           *(StgInt32 *)(adj_code+0xc)  = 0x8948c889;
+           *(StgInt32 *)(adj_code+0x10) = 0xf28948d1;
+           *(StgInt32 *)(adj_code+0x14) = 0x48fe8948;
+           *(StgInt32 *)(adj_code+0x18) = 0x00123d8b;
+           *(StgInt32 *)(adj_code+0x1c) = 0x25ff0000;
+           *(StgInt32 *)(adj_code+0x20) = 0x00000014;
            
-           *(StgInt64 *)(adjustor+0x28) = (StgInt64)obscure_ccall_ret_code;
-           *(StgInt64 *)(adjustor+0x30) = (StgInt64)hptr;
-           *(StgInt64 *)(adjustor+0x38) = (StgInt64)wptr;
+           *(StgInt64 *)(adj_code+0x28) = (StgInt64)obscure_ccall_ret_code;
+           *(StgInt64 *)(adj_code+0x30) = (StgInt64)hptr;
+           *(StgInt64 *)(adj_code+0x38) = (StgInt64)wptr;
        }
     }
 #elif defined(sparc_HOST_ARCH)
@@ -912,11 +924,6 @@ TODO: Depending on how much allocation overhead stgMallocBytes uses for
         AdjustorStub *adjustorStub;
         int sz = 0, extra_sz, total_sz;
 
-            // from AdjustorAsm.s
-            // not declared as a function so that AIX-style
-            // fundescs can never get in the way.
-        extern void *adjustorCode;
-        
 #ifdef FUNDESCS
         adjustorStub = stgMallocBytes(sizeof(AdjustorStub), "createAdjustor");
 #else
@@ -1121,9 +1128,9 @@ if ( *(unsigned char*)ptr != 0xe8 ) {
  freeStablePtr(((AdjustorStub*)ptr)->hptr);
 #elif defined(x86_64_HOST_ARCH)
  if ( *(StgWord16 *)ptr == 0x894d ) {
-     freeStablePtr(*(StgStablePtr*)(ptr+0x20));
+     freeStablePtr(*(StgStablePtr*)((StgWord8*)ptr+0x20));
  } else if ( *(StgWord16 *)ptr == 0x5141 ) {
-     freeStablePtr(*(StgStablePtr*)(ptr+0x30));
+     freeStablePtr(*(StgStablePtr*)((StgWord8*)ptr+0x30));
  } else {
    errorBelch("freeHaskellFunctionPtr: not for me, guv! %p\n", ptr);
    return;
@@ -1151,7 +1158,6 @@ if ( *(unsigned char*)ptr != 0xe8 ) {
  }
  freeStablePtr(((StgStablePtr*)ptr)[1]);
 #elif defined(powerpc_HOST_ARCH) || defined(powerpc64_HOST_ARCH)
- extern void* adjustorCode;
  if ( ((AdjustorStub*)ptr)->code != (StgFunPtr) &adjustorCode ) {
    errorBelch("freeHaskellFunctionPtr: not for me, guv! %p\n", ptr);
    return;