android/android-emugl/shared/emugl/common/testing/test_thread.h

// Copyright (C) 2014 The Android Open Source Project
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#ifndef EMUGL_COMMON_TESTING_TEST_THREAD_H
#define EMUGL_COMMON_TESTING_TEST_THREAD_H

#ifdef _WIN32
#  define WIN32_LEAN_AND_MEAN 1
#  include <windows.h>
#else
#  include <pthread.h>
#endif

namespace emugl {

// Very basic platform thread wrapper that only uses a tiny stack.
// This shall only be used during unit testing, and allows test code
// to not depend on the implementation of emugl::Thread.
class TestThread {
public:
    // Main thread function type.
    typedef void* (ThreadFunction)(void* param);

    // Constructor actually launches a new platform thread.
    TestThread(ThreadFunction* func, void* funcParam) {
#ifdef _WIN32
        mThread = CreateThread(NULL,
                            16384,
                            (DWORD WINAPI (*)(void*))func,
                            funcParam,
                            0,
                            NULL);
#else
        pthread_attr_t attr;
        pthread_attr_init(&attr);
        pthread_attr_setstacksize(&attr, 16384);
        pthread_create(&mThread,  &attr, func, funcParam);
        pthread_attr_destroy(&attr);
#endif
    }

    ~TestThread() {
#ifdef _WIN32
        CloseHandle(mThread);
#endif
    }

    void join() {
#ifdef _WIN32
        WaitForSingleObject(mThread, INFINITE);
#else
        void* ret = NULL;
        pthread_join(mThread, &ret);
#endif
    }

private:
#ifdef _WIN32
    HANDLE mThread;
#else
    pthread_t mThread;
#endif
};

}  // namespace emugl

#endif  // EMUGL_COMMON_TESTING_TEST_THREAD_H