source: libcf/trunk/src/cf_thread.c@ 65

/**
 * @file    cf_thread.c
 * @author  myusgun <myusgun@gmail.com>
 * @version 0.1
 */
#include "cf_thread.h"
#include "cf_local.h"
#include "cf_error.h"

#include <stdlib.h>

#if defined(_WIN32) || defined(_WIN64)
# include <windows.h>
# include <process.h>
# define THREAD_TYPE    HANDLE
# define MUTEX_TYPE     HANDLE
# define THREAD_RETURN  unsigned long   /**< 스레드 워커 함수 반환 형 */
# define THREAD_CALL    __stdcall
#else // #if defined(_WIN32) || defined(_WIN64)
# include <pthread.h>
# define THREAD_TYPE    pthread_t
# define MUTEX_TYPE     pthread_mutex_t
# define THREAD_RETURN  void *          /**< 스레드 워커 함수 반환 형 */
# define THREAD_CALL
#endif // #if defined(_WIN32) || defined(_WIN64)

#define ASSERT_THREAD(__h)              \
    if (__h == NULL)                    \
        return CF_ERROR_THREAD_INVALID_ARGS

#define ASSERT_MUTEX(__h)               \
    if (__h == NULL)                    \
        return CF_ERROR_MUTEX_INVALID_ARGS

typedef THREAD_RETURN (THREAD_CALL * THREAD_WORKER) (void *);

static int
CF_Thread_Local_Close (void * ctx)
{
#if defined(_WIN32) || defined(_WIN64)
    CloseHandle (ctx);
#else
    free (ctx);
#endif

    return CF_OK;
}

/**
 * 스레드를 생성
 *
 * @return 성공 시, CF_OK; 실패 시, 오류 코드
 *
 * @param threadID  스레드 아이디 주소
 * @param callback  스레드 워커 함수 이름
 * @param arg       스레드 함수로 전달할 인자
 */
int
CF_Thread_Create (CF_Thread             * threadID,
                  CF_Thread_Function    callback,
                  void                  * arg)
{
    int result = 0;

    THREAD_WORKER f = (THREAD_WORKER) callback;

#if defined(_WIN32) || defined(_WIN64)

    *threadID = (THREAD_TYPE) CreateThread (NULL, 0, f, arg, 0, NULL);
    if (*threadID == NULL)
        return CF_ERROR_THREAD_CREATE;
#else
    *threadID = (THREAD_TYPE *) calloc (sizeof (THREAD_TYPE), 1);
    if (*threadID == NULL)
        return CF_ERROR_THREAD_CREATE;

    result = pthread_create ((THREAD_TYPE *) *threadID, NULL, f, arg);
    if (result < 0)
        return CF_ERROR_THREAD_CREATE;
#endif

    return CF_OK;
}

/**
 * 스레드 아이디를 해제
 *
 * @return 성공 시, CF_OK; 실패 시, 오류 코드
 *
 * @param threadID 스레드 아이디 주소
 *
 * @remark 스레드 아이디를 해제하는 것이며 워커 스레드가 종료되지 않음
 */
int
CF_Thread_Release (CF_Thread * threadID)
{
    ASSERT_THREAD (*threadID);

    CF_Thread_Local_Close (*threadID);

    return CF_OK;
}

/**
 * 스레드가 종료될 때 까지 대기
 *
 * @return CF_OK 반환
 *
 * @param threadID 스레드 아이디 주소
 */
int
CF_Thread_Join (CF_Thread * threadID)
{
    ASSERT_THREAD (*threadID);

#if defined(_WIN32) || defined(_WIN64)
    WaitForSingleObject ((THREAD_TYPE) *threadID, INFINITE);
#else
    char status[16] = {0x00,};
    pthread_join (*((THREAD_TYPE *) *threadID), (void **)status);
#endif

    return CF_OK;
}

/**
 * 뮤텍스 생성
 *
 * @return 성공 시, CF_OK; 실패 시, 오류 코드
 *
 * @param mutex 뮤텍스 아이디 주소
 *
 * @see CF_Thread_Create
 */
int
CF_Mutex_Create (CF_Mutex * mutex)
{
    int result = 0;

#if defined(_WIN32) || defined(_WIN64)
    *mutex = (MUTEX_TYPE) CreateMutexA (NULL, FALSE, NULL);
    if (*mutex == NULL)
        return CF_ERROR_MUTEX_CREATE;
#else
    *mutex = (MUTEX_TYPE *) calloc (sizeof (MUTEX_TYPE), 1);
    if (*mutex == NULL)
        return CF_ERROR_MUTEX_CREATE;

    result = pthread_mutex_init (*mutex, NULL);
    if (result < 0)
        return CF_ERROR_MUTEX_CREATE;
#endif

    return CF_OK;
}

/**
 * 뮤텍스 해제
 *
 * @return 성공 시, CF_OK; 실패 시, 오류 코드
 *
 * @param mutex 뮤텍스 아이디 주소
 */
int
CF_Mutex_Destory (CF_Mutex * mutex)
{
    ASSERT_MUTEX (*mutex);

    CF_Thread_Local_Close (*mutex);

    return CF_OK;
}

/**
 * 뮤텍스 잠금
 *
 * @return 성공 시, CF_OK; 실패 시, 오류 코드
 *
 * @param mutex 뮤텍스 아이디 주소
 */
int
CF_Mutex_Lock (CF_Mutex * mutex)
{
    ASSERT_MUTEX (*mutex);

#if defined(_WIN32) || defined(_WIN64)
    WaitForSingleObject ((MUTEX_TYPE) *mutex, INFINITE);
#else
    pthread_mutex_lock ((MUTEX_TYPE *) *mutex);
#endif

    return CF_OK;
}

/**
 * 뮤텍스 잠금 해제
 *
 * @return 성공 시, CF_OK; 실패 시, 오류 코드
 *
 * @param mutex 뮤텍스 아이디 주소
 */
int
CF_Mutex_Unlock (CF_Mutex * mutex)
{
    ASSERT_MUTEX (*mutex);

#if defined(_WIN32) || defined(_WIN64)
    ReleaseMutex (*mutex);
#else
    pthread_mutex_unlock ((MUTEX_TYPE *) *mutex);
#endif

    return CF_OK;
}