utils.cpp

// Copyright 2018 The clvk authors.
//
// 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 "utils.hpp"
#include <array>
#include <cstdio>
#include <cstdlib>

#ifdef __APPLE__
#include <unistd.h>
#endif

#ifdef WIN32
#include <Windows.h>
#include <io.h>
#endif

#if !defined(WIN32) && !defined(__APPLE__)
#include <pthread.h>
#endif

char* cvk_mkdtemp(std::string& tmpl) {
#ifdef WIN32
    if (_mktemp_s(&tmpl.front(), tmpl.size() + 1) != 0) {
        return nullptr;
    }

    if (!CreateDirectory(tmpl.c_str(), nullptr)) {
        return nullptr;
    }

    return &tmpl.front();
#else
    return mkdtemp(&tmpl.front());
#endif
}

int cvk_exec(const std::string& cmd, std::string* output) {
#ifdef WIN32
#define popen _popen
#define pclose _pclose
#endif
    cvk_info("About to run \"%s\"", cmd.c_str());

    std::array<char, 512> buffer;
    std::string out;
    std::string cmd_with_err = cmd + " 2>&1";
    FILE* pipe = popen(cmd_with_err.c_str(), "r");

    if (pipe == nullptr) {
        return -1;
    }

    while (fgets(buffer.data(), buffer.size(), pipe) != nullptr) {
        out += buffer.data();
    }

    if (output != nullptr) {
        *output = std::move(out);
    }

    int ret = pclose(pipe);

    cvk_info("Return code was: %d", ret);

    return ret;
#ifdef WIN32
#undef popen
#undef pclose
#endif
}

void cvk_set_current_thread_name_if_supported(const std::string& name) {
#if !defined(WIN32) && !defined(__APPLE__)
    pthread_setname_np(pthread_self(), name.c_str());
#endif
}