blob: ca477bf6157e7f0ec1e056a3d3b6193e276f7f59 [file] [log] [blame]
/*
* Copyright (C) 2016 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.
*/
#include <condition_variable>
#include <chrono>
#include <functional>
#include <mutex>
#include <thread>
#include <hidl/Status.h>
namespace android {
namespace lshal {
static constexpr std::chrono::milliseconds IPC_CALL_WAIT{500};
class BackgroundTaskState {
public:
BackgroundTaskState(std::function<void(void)> &&func)
: mFunc(std::forward<decltype(func)>(func)) {}
void notify() {
std::unique_lock<std::mutex> lock(mMutex);
mFinished = true;
lock.unlock();
mCondVar.notify_all();
}
template<class C, class D>
bool wait(std::chrono::time_point<C, D> end) {
std::unique_lock<std::mutex> lock(mMutex);
mCondVar.wait_until(lock, end, [this](){ return this->mFinished; });
return mFinished;
}
void operator()() {
mFunc();
}
private:
std::mutex mMutex;
std::condition_variable mCondVar;
bool mFinished = false;
std::function<void(void)> mFunc;
};
void *callAndNotify(void *data) {
BackgroundTaskState &state = *static_cast<BackgroundTaskState *>(data);
state();
state.notify();
return NULL;
}
template<class R, class P>
bool timeout(std::chrono::duration<R, P> delay, std::function<void(void)> &&func) {
auto now = std::chrono::system_clock::now();
BackgroundTaskState state{std::forward<decltype(func)>(func)};
pthread_t thread;
if (pthread_create(&thread, NULL, callAndNotify, &state)) {
std::cerr << "FATAL: could not create background thread." << std::endl;
return false;
}
bool success = state.wait(now + delay);
if (!success) {
pthread_kill(thread, SIGINT);
}
pthread_join(thread, NULL);
return success;
}
template<class Function, class I, class... Args>
typename std::result_of<Function(I *, Args...)>::type
timeoutIPC(const sp<I> &interfaceObject, Function &&func, Args &&... args) {
using ::android::hardware::Status;
typename std::result_of<Function(I *, Args...)>::type ret{Status::ok()};
auto boundFunc = std::bind(std::forward<Function>(func),
interfaceObject.get(), std::forward<Args>(args)...);
bool success = timeout(IPC_CALL_WAIT, [&ret, &boundFunc] {
ret = std::move(boundFunc());
});
if (!success) {
return Status::fromStatusT(TIMED_OUT);
}
return ret;
}
} // namespace lshal
} // namespace android