blob: a29bc8af6e61aa45d3ff1fa3ddb425beacf4e688 [file] [log] [blame]
/*-------------------------------------------------------------------------
* drawElements Quality Program Execution Server
* ---------------------------------------------
*
* Copyright 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.
*
*//*!
* \file
* \brief Test Driver.
*//*--------------------------------------------------------------------*/
#include "xsTestDriver.hpp"
#include "deClock.h"
#include <string>
#include <vector>
#include <cstdio>
using std::string;
using std::vector;
#if 0
#define DBG_PRINT(X) printf X
#else
#define DBG_PRINT(X)
#endif
namespace xs
{
TestDriver::TestDriver(xs::TestProcess *testProcess)
: m_state(STATE_NOT_STARTED)
, m_lastExitCode(0)
, m_process(testProcess)
, m_lastProcessDataTime(0)
, m_dataMsgTmpBuf(SEND_RECV_TMP_BUFFER_SIZE)
{
}
TestDriver::~TestDriver(void)
{
reset();
}
void TestDriver::reset(void)
{
m_process->cleanup();
m_state = STATE_NOT_STARTED;
}
void TestDriver::startProcess(const char *name, const char *params, const char *workingDir, const char *caseList)
{
try
{
m_process->start(name, params, workingDir, caseList);
m_state = STATE_PROCESS_STARTED;
}
catch (const TestProcessException &e)
{
printf("Failed to launch test process: %s\n", e.what());
m_state = STATE_PROCESS_LAUNCH_FAILED;
m_lastLaunchFailure = e.what();
}
}
void TestDriver::stopProcess(void)
{
m_process->terminate();
}
bool TestDriver::poll(ByteBuffer &messageBuffer)
{
switch (m_state)
{
case STATE_NOT_STARTED:
return false; // Nothing to report.
case STATE_PROCESS_LAUNCH_FAILED:
DBG_PRINT((" STATE_PROCESS_LAUNCH_FAILED\n"));
if (writeMessage(messageBuffer, ProcessLaunchFailedMessage(m_lastLaunchFailure.c_str())))
{
m_state = STATE_NOT_STARTED;
m_lastLaunchFailure = "";
return true;
}
else
return false;
case STATE_PROCESS_STARTED:
DBG_PRINT((" STATE_PROCESS_STARTED\n"));
if (writeMessage(messageBuffer, ProcessStartedMessage()))
{
m_state = STATE_PROCESS_RUNNING;
return true;
}
else
return false;
case STATE_PROCESS_RUNNING:
{
DBG_PRINT((" STATE_PROCESS_RUNNING\n"));
bool gotProcessData = false;
// Poll log file and info buffer.
gotProcessData = pollLogFile(messageBuffer) || gotProcessData;
gotProcessData = pollInfo(messageBuffer) || gotProcessData;
if (gotProcessData)
return true; // Got IO.
if (!m_process->isRunning())
{
// Process died.
m_state = STATE_READING_DATA;
m_lastExitCode = m_process->getExitCode();
m_lastProcessDataTime = deGetMicroseconds();
return true; // Got state change.
}
return false; // Nothing to report.
}
case STATE_READING_DATA:
{
DBG_PRINT((" STATE_READING_DATA\n"));
bool gotProcessData = false;
// Poll log file and info buffer.
gotProcessData = pollLogFile(messageBuffer) || gotProcessData;
gotProcessData = pollInfo(messageBuffer) || gotProcessData;
if (gotProcessData)
{
// Got data.
m_lastProcessDataTime = deGetMicroseconds();
return true;
}
else if (deGetMicroseconds() - m_lastProcessDataTime > READ_DATA_TIMEOUT * 1000)
{
// Read timeout occurred.
m_state = STATE_PROCESS_FINISHED;
return true; // State change.
}
else
return false; // Still waiting for data.
}
case STATE_PROCESS_FINISHED:
DBG_PRINT((" STATE_PROCESS_FINISHED\n"));
if (writeMessage(messageBuffer, ProcessFinishedMessage(m_lastExitCode)))
{
// Signal TestProcess to clean up any remaining resources.
m_process->cleanup();
m_state = STATE_NOT_STARTED;
m_lastExitCode = 0;
return true;
}
else
return false;
default:
DE_ASSERT(false);
return false;
}
}
bool TestDriver::pollLogFile(ByteBuffer &messageBuffer)
{
return pollBuffer(messageBuffer, MESSAGETYPE_PROCESS_LOG_DATA);
}
bool TestDriver::pollInfo(ByteBuffer &messageBuffer)
{
return pollBuffer(messageBuffer, MESSAGETYPE_INFO);
}
bool TestDriver::pollBuffer(ByteBuffer &messageBuffer, MessageType msgType)
{
const int minBytesAvailable = MESSAGE_HEADER_SIZE + MIN_MSG_PAYLOAD_SIZE;
if (messageBuffer.getNumFree() < minBytesAvailable)
return false; // Not enough space in message buffer.
const int maxMsgSize = de::min((int)m_dataMsgTmpBuf.size(), messageBuffer.getNumFree());
int numRead = 0;
int msgSize = MESSAGE_HEADER_SIZE + 1; // One byte is reserved for terminating 0.
// Fill in data \note Last byte is reserved for 0.
numRead = msgType == MESSAGETYPE_PROCESS_LOG_DATA ?
m_process->readTestLog(&m_dataMsgTmpBuf[MESSAGE_HEADER_SIZE], maxMsgSize - MESSAGE_HEADER_SIZE - 1) :
m_process->readInfoLog(&m_dataMsgTmpBuf[MESSAGE_HEADER_SIZE], maxMsgSize - MESSAGE_HEADER_SIZE - 1);
if (numRead <= 0)
return false; // Didn't get any data.
msgSize += numRead;
// Terminate with 0.
m_dataMsgTmpBuf[msgSize - 1] = 0;
// Write header.
Message::writeHeader(msgType, msgSize, &m_dataMsgTmpBuf[0], MESSAGE_HEADER_SIZE);
// Write to messagebuffer.
messageBuffer.pushFront(&m_dataMsgTmpBuf[0], msgSize);
DBG_PRINT((" wrote %d bytes of %s data\n", msgSize, msgType == MESSAGETYPE_INFO ? "info" : "log"));
return true;
}
bool TestDriver::writeMessage(ByteBuffer &messageBuffer, const Message &message)
{
vector<uint8_t> buf;
message.write(buf);
if (messageBuffer.getNumFree() < (int)buf.size())
return false;
messageBuffer.pushFront(&buf[0], (int)buf.size());
return true;
}
} // namespace xs