task.cpp

#include "task.hpp"

#include "logger.hpp"
#include "utils.hpp"
#include "config.hpp"

#include "pstream.h"

#include <thread>
#include <ctime>

namespace remexec {

using namespace std;
using namespace redi;

Task::Task(string _name, vector<string> _args, string _binary, string _tmpath, size_t _timeout) :
		name(_name),
		binary(_binary),
		tmpath(_tmpath),
		timeout(_timeout)
{
	args.push_back(_name);
	args.insert(args.end(), _args.begin(), _args.end());
}

Task::~Task()
{
}

bool Task::run(ostream &out, ostream &err){
	string wd = getcwd();
	chdir(tmpath);

	pstream p(binary, args, pstreams::pstdout | pstreams::pstderr | pstreams::newpg);

	chdir(wd);

	if (p.good()){
		char buf[32];
		size_t n;

		constexpr int interval = 100;
		size_t times = 0;
		bool done[2] {false, false};

		while (!done[0] && !done[1]){
			if (timeout > 0 && times * interval >= timeout){
				Log::warn("Task time limit exceed");
				p.rdbuf()->killpg(SIGKILL);
				break;
			}

			for (int i = 0; i < 2; ++i){
				if (!done[i]){
					while ((n = (i == 0 ? p.out() : p.err()).readsome(buf, sizeof(buf))) > 0){
						(i == 0 ? out : err).write(buf, n);
					}
					(i == 0 ? out : err).flush();

					if (p.eof()){
						done[i] = true;
						p.clear();
					}
				}
			}

			usleep(interval * 1000); //100ms
			++times;
		}

		p.close();

		return done[0] && done[1];
	}

	return false;
}

}