lldb/test/API/api/multithreaded/common.h

#ifndef LLDB_TEST_API_COMMON_H
#define LLDB_TEST_API_COMMON_H

#include <chrono>
#include <condition_variable>
#include <cstdlib>
#include <exception>
#include <iostream>
#include <mutex>
#include <queue>
#include <string>

#ifdef _WIN32
#include <direct.h>
#include <io.h>
#else
#include <unistd.h>
#endif

/// Simple exception class with a message
struct Exception : public std::exception
{
  std::string s;
  Exception(std::string ss) : s(ss) {}
  virtual ~Exception() throw () { }
  const char* what() const throw() { return s.c_str(); }
};

// Synchronized data structure for listener to send events through
template<typename T>
class multithreaded_queue {
  std::condition_variable m_condition;
  std::mutex m_mutex;
  std::queue<T> m_data;

public:

  void push(T e) {
    {
      std::lock_guard<std::mutex> lock(m_mutex);
      m_data.push(e);
    }
    m_condition.notify_all();
  }

  T pop(bool &success) {
    std::unique_lock<std::mutex> lock(m_mutex);
    if (!m_condition.wait_for(lock, std::chrono::seconds(5),
                              [&] { return !m_data.empty(); })) {
      success = false;
      return T();
    }

    T ret = m_data.front();
    m_data.pop();
    success = true;
    return ret;
  }
};

/// Returns the current working directory.
///
/// The platform accessors all return a malloc'd buffer, so the result is
/// copied into a std::string and freed here.
inline std::string get_working_dir() {
#if defined(__APPLE__) || defined(__FreeBSD__) || defined(__NetBSD__) ||       \
    defined(__OpenBSD__)
  char *dir = getwd(0);
#elif defined(_WIN32)
  char *dir = _getcwd(0, 0);
#else
  char *dir = get_current_dir_name();
#endif
  if (!dir)
    return std::string();
  std::string result(dir);
  free(dir);
  return result;
}

#endif // LLDB_TEST_API_COMMON_H