Ошибка расписания при вызове многопоточного C FFI с функцией обратного вызова Haskell - PullRequest
Новая
       12

Ошибка расписания при вызове многопоточного C FFI с функцией обратного вызова Haskell

7 голосов
/ 17 января 2012

Ниже приведен код FFI на Haskell / C, который выдает ошибку расписания во время выполнения (GHC 7.0.3, Mac OS 10.7, x86_64). Я искал объяснение ошибки, но не нашел ничего подходящего.

код C (mt.c): 

#include <pthread.h>
#include <stdio.h>

typedef void(*FunctionPtr)(int);

/* This is our thread function.  It is like main(), but for a thread*/
void *threadFunc(void *arg)
{
  FunctionPtr fn;
  fn = (FunctionPtr) arg;
  fn(1); //call haskell function with a CInt argument to see if it works
}

void create_threads(FunctionPtr* fp, int numThreads )
{
  pthread_t pth[numThreads];  // array of pthreads
  int t;
  for (t=0; t < numThreads;){
    pthread_create(&pth[t],NULL,threadFunc,*(fp + t));
    t++;
  }

  printf("main waiting for all threads to terminate...\n");
  for (t=0; t < numThreads;t++){
    pthread_join(pth[t],NULL);
  }
}

Код Haskell (t.hs) - он вызывает create_threads в mt.c выше с Storable Vector из FunPtr для функции Haskell f (после применения первых трех аргументов к f): 

{-# LANGUAGE BangPatterns #-}
import Control.Concurrent (forkIO, threadDelay, MVar, newEmptyMVar, putMVar, takeMVar)
import qualified Data.Vector.Storable.Mutable as MSV
import qualified Data.Vector.Storable as SV
import Control.Monad.Primitive (PrimState)
import Control.Monad (mapM, forM_)
import Foreign.Ptr (Ptr, FunPtr)
import Foreign.C.Types (CInt)


type Length = CInt

-- | f is a function that is called back by create_threads in mt.c
f :: MVar Int -> MSV.MVector (PrimState IO) CInt -> Length -> CInt -> IO ()
f m v l x = do
              !i <- takeMVar m
              case (i< fromIntegral l) of
                True -> MSV.unsafeWrite v i x >> print x >> putMVar m (i+1) 
                False -> return () -- overflow

-- a "wrapper" import gives us a converter for converting a Haskell function to a foreign function pointer
foreign import ccall "wrapper"
  wrap :: (CInt -> IO()) -> IO (FunPtr (CInt -> IO()))

foreign import ccall safe "create_threads"
  createThreads :: Ptr (FunPtr (CInt -> IO())) -> CInt -> IO()

main = do
  let threads = [1..4]
  m <- mapM (\x -> newEmptyMVar) $ threads
  -- intialize mvars with 0
  forM_ m $ \x -> putMVar x 0
  let l = 10
  -- intialize vectors of length 10 that will be filled by function f
  v <- mapM (\x -> MSV.new l) threads
  -- create a list of function pointers to partial function - the partial function is obtained by applying first three arguments to         function f
  lf <- mapM (\(x,y) -> wrap (f x y (fromIntegral l))) $ zip m v
  -- convert above function list to a storable vector of function pointers
  let fv = SV.fromList lf
  -- call createThreads with storable vector of function pointers, and number of threads - createThreads will spawn threads which will use  function pointers for callback
  SV.unsafeWith fv $ \x -> createThreads x (fromIntegral $ length threads)

Пожалуйста, игнорируйте небезопасные части в коде - моя цель здесь состоит в том, чтобы протестировать обратный вызов, используя Haskell FFI с многопоточным C-кодом. Когда я его компилирую и запускаю, я получаю сообщение об ошибке ниже: 

$ ghc -O2 t.hs mt.c -lpthread
[1 of 1] Compiling Main             ( t.hs, t.o )
Linking t ...
$ ./t
main waiting for all threads to terminate...
t: schedule: re-entered unsafely.
   Perhaps a 'foreign import unsafe' should be 'safe'?
$ uname -a
Darwin desktop.local 11.2.0 Darwin Kernel Version 11.2.0: Tue Aug  9 20:54:00 PDT 2011; root:xnu-1699.24.8~1/RELEASE_X86_64 x86_64
$ ghc --version
The Glorious Glasgow Haskell Compilation System, version 7.0.3

Ошибка расписания происходит, только если у меня есть потоки C, которые вызывают функцию haskell f. Я полагаю, что в моем коде больше ошибок, чем в одной из библиотек или GHC. Итак, я хотел бы сначала проверить наличие указателей на причину ошибки.

1 Ответ

4 голосов
/ 17 января 2012

В этом случае ошибка schedule произошла из-за того, что код haskell был скомпилирован без опции -threaded.

Код haskell вызывает функцию C create_threads, которая порождает несколько потоков для функции threadFunc C. threadFunc вызывает функцию Haskell f. Таким образом, хотя код на Haskell компилируется без опции -threaded, он все равно приводит к тому, что несколько потоков C выполняют f.

GHC Runtime Scheduler обнаружил этот упущение при выполнении нескольких потоков без многопоточного выполнения и пометил его как ошибку. Это намного лучше, чем загадочный сбой во время выполнения. Я понял упущение, когда проверил код rts / schedule.c в базе кодов GHC и увидел комментарий ниже. Это сообщило мне о том, что threaded время выполнения не включено: 

// Check whether we have re-entered the RTS from Haskell without
 // going via suspendThread()/resumeThread (i.e. a 'safe' foreign
 // call).
...