gtkradiant/radiant/watchbsp.cpp

621 lines
19 KiB
C++
Raw Normal View History

/*
Copyright (c) 2001, Loki software, inc.
All rights reserved.
Redistribution and use in source and binary forms, with or without modification,
are permitted provided that the following conditions are met:
Redistributions of source code must retain the above copyright notice, this list
of conditions and the following disclaimer.
Redistributions in binary form must reproduce the above copyright notice, this
list of conditions and the following disclaimer in the documentation and/or
other materials provided with the distribution.
Neither the name of Loki software nor the names of its contributors may be used
to endorse or promote products derived from this software without specific prior
written permission.
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS''
AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY
DIRECT,INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
*/
//-----------------------------------------------------------------------------
//
// DESCRIPTION:
// monitoring window for running BSP processes (and possibly various other stuff)
#include "stdafx.h"
#include "watchbsp.h"
#include "feedback.h"
#ifdef _WIN32
#include <winsock2.h>
#endif
#if defined (__linux__) || defined (__APPLE__)
#include <sys/time.h>
#define SOCKET_ERROR -1
#endif
#ifdef __APPLE__
#include <unistd.h>
#endif
#include <assert.h>
// Static functions for the SAX callbacks -------------------------------------------------------
// utility for saxStartElement below
static void abortStream(message_info_t *data)
{
g_pParentWnd->GetWatchBSP()->Reset();
// tell there has been an error
if (g_pParentWnd->GetWatchBSP()->HasBSPPlugin ())
g_BSPFrontendTable.m_pfnEndListen(2);
// yeah this doesn't look good.. but it's needed so that everything will be ignored until the stream goes out
data->ignore_depth = -1;
data->recurse++;
}
#include "stream_version.h"
static void saxStartElement(message_info_t *data, const xmlChar *name, const xmlChar **attrs)
{
if (data->ignore_depth == 0)
{
if (data->bGeometry)
// we have a handler
{
data->pGeometry->saxStartElement (data, name, attrs);
}
else
{
if (strcmp ((char *)name, "q3map_feedback") == 0)
{
// check the correct version
// old q3map don't send a version attribute
// the ones we support .. send Q3MAP_STREAM_VERSION
if (!attrs[0] || !attrs[1] || (strcmp((char*)attrs[0],"version") != 0))
{
Sys_FPrintf(SYS_ERR, "No stream version given in the feedback stream, this is an old q3map version.\n"
"Please turn off monitored compiling if you still wish to use this q3map executable\n");
abortStream(data);
return;
}
else if (strcmp((char*)attrs[1],Q3MAP_STREAM_VERSION) != 0)
{
Sys_FPrintf(SYS_ERR,
"This version of Radiant reads version %s debug streams, I got an incoming connection with version %s\n"
"Please make sure your versions of Radiant and q3map are matching.\n", Q3MAP_STREAM_VERSION, (char*)attrs[1]);
abortStream(data);
return;
}
}
// we don't treat locally
else if (strcmp ((char *)name, "message") == 0)
{
data->msg_level = atoi ((char *)attrs[1]);
}
else if (strcmp ((char *)name, "polyline") == 0)
// polyline has a particular status .. right now we only use it for leakfile ..
{
data->bGeometry = true;
data->pGeometry = &g_pointfile;
data->pGeometry->saxStartElement( data, name, attrs );
}
else if (strcmp ((char *)name, "select") == 0)
{
CSelectMsg *pSelect = new CSelectMsg();
data->bGeometry = true;
data->pGeometry = pSelect;
data->pGeometry->saxStartElement( data, name, attrs );
}
else if (strcmp ((char *)name, "pointmsg") == 0)
{
CPointMsg *pPoint = new CPointMsg();
data->bGeometry = true;
data->pGeometry = pPoint;
data->pGeometry->saxStartElement( data, name, attrs );
}
else if (strcmp ((char *)name, "windingmsg") == 0)
{
CWindingMsg *pWinding = new CWindingMsg();
data->bGeometry = true;
data->pGeometry = pWinding;
data->pGeometry->saxStartElement( data, name, attrs );
}
else
{
Sys_FPrintf (SYS_WRN, "WARNING: ignoring unrecognized node in XML stream (%s)\n", name);
// we don't recognize this node, jump over it
// (NOTE: the ignore mechanism is a bit screwed, only works when starting an ignore at the highest level)
data->ignore_depth = data->recurse;
}
}
}
data->recurse++;
}
static void saxEndElement(message_info_t *data, const xmlChar *name) {
data->recurse--;
// we are out of an ignored chunk
if ( data->recurse == data->ignore_depth ) {
data->ignore_depth = 0;
return;
}
if ( data->bGeometry ) {
data->pGeometry->saxEndElement( data, name );
// we add the object to the debug window
if ( !data->bGeometry ) {
g_DbgDlg.Push( data->pGeometry );
}
}
if ( data->recurse == data->stop_depth ) {
#ifdef _DEBUG
Sys_Printf ("Received error msg .. shutting down..\n");
#endif
g_pParentWnd->GetWatchBSP()->Reset();
// tell there has been an error
if ( g_pParentWnd->GetWatchBSP()->HasBSPPlugin() ) {
g_BSPFrontendTable.m_pfnEndListen( 2 );
}
return;
}
}
static void saxCharacters(message_info_t *data, const xmlChar *ch, int len)
{
if (data->bGeometry)
{
data->pGeometry->saxCharacters (data, ch, len);
}
else
{
if (data->ignore_depth != 0)
return;
// output the message using the level
char buf[1024];
memcpy( buf, ch, len );
buf[len] = '\0';
Sys_FPrintf (data->msg_level, "%s", buf);
// if this message has error level flag, we mark the depth to stop the compilation when we get out
// we don't set the msg level if we don't stop on leak
if (data->msg_level == 3)
{
data->stop_depth = data->recurse-1;
}
}
}
static void saxComment(void *ctx, const xmlChar *msg)
{
Sys_Printf("XML comment: %s\n", msg);
}
static void saxWarning(void *ctx, const char *msg, ...)
{
char saxMsgBuffer[4096];
va_list args;
va_start(args, msg);
vsprintf (saxMsgBuffer, msg, args);
va_end(args);
Sys_FPrintf(SYS_WRN, "XML warning: %s\n", saxMsgBuffer);
}
static void saxError(void *ctx, const char *msg, ...)
{
char saxMsgBuffer[4096];
va_list args;
va_start(args, msg);
vsprintf (saxMsgBuffer, msg, args);
va_end(args);
Sys_FPrintf(SYS_ERR, "XML error: %s\n", saxMsgBuffer);
}
static void saxFatal(void *ctx, const char *msg, ...)
{
char buffer[4096];
va_list args;
va_start(args, msg);
vsprintf (buffer, msg, args);
va_end(args);
Sys_FPrintf(SYS_ERR, "XML fatal error: %s\n", buffer);
}
static xmlSAXHandler saxParser = {
0, /* internalSubset */
0, /* isStandalone */
0, /* hasInternalSubset */
0, /* hasExternalSubset */
0, /* resolveEntity */
0, /* getEntity */
0, /* entityDecl */
0, /* notationDecl */
0, /* attributeDecl */
0, /* elementDecl */
0, /* unparsedEntityDecl */
0, /* setDocumentLocator */
0, /* startDocument */
0, /* endDocument */
(startElementSAXFunc)saxStartElement, /* startElement */
(endElementSAXFunc)saxEndElement, /* endElement */
0, /* reference */
(charactersSAXFunc)saxCharacters, /* characters */
0, /* ignorableWhitespace */
0, /* processingInstruction */
(commentSAXFunc)saxComment, /* comment */
(warningSAXFunc)saxWarning, /* warning */
(errorSAXFunc)saxError, /* error */
(fatalErrorSAXFunc)saxFatal, /* fatalError */
};
// ------------------------------------------------------------------------------------------------
CWatchBSP::~CWatchBSP()
{
Reset();
if (m_sBSPName)
{
delete[] m_sBSPName;
m_sBSPName = NULL;
}
Net_Shutdown();
}
void CWatchBSP::Reset()
{
if (m_pInSocket)
{
Net_Disconnect(m_pInSocket);
m_pInSocket = NULL;
}
if (m_pListenSocket)
{
Net_Disconnect(m_pListenSocket);
m_pListenSocket = NULL;
}
if (m_xmlInputBuffer)
{
xmlFreeParserInputBuffer (m_xmlInputBuffer);
m_xmlInputBuffer = NULL;
}
m_eState = EIdle;
}
bool CWatchBSP::SetupListening()
{
#ifdef _DEBUG
if (m_pListenSocket)
{
Sys_Printf("ERROR: m_pListenSocket != NULL in CWatchBSP::SetupListening\n");
return false;
}
#endif
Sys_Printf("Setting up\n");
if( !Net_Setup() )
return false;
m_pListenSocket = Net_ListenSocket(39000);
if (m_pListenSocket == NULL)
return false;
Sys_Printf("Listening...\n");
return true;
}
void CWatchBSP::DoEBeginStep() {
Reset();
if ( !SetupListening() ) {
CString msg;
msg = "Failed to get a listening socket on port 39000.\nTry running with BSP monitoring disabled if you can't fix this.\n";
Sys_Printf( msg );
gtk_MessageBox( g_pParentWnd->m_pWidget, msg, "BSP monitoring", MB_OK | MB_ICONERROR );
return;
}
// set the timer for timeouts and step cancellation
g_timer_reset( m_pTimer );
g_timer_start( m_pTimer );
if ( !m_bBSPPlugin ) {
Sys_Printf( "=== running BSP command ===\n%s\n", g_ptr_array_index( m_pCmd, m_iCurrentStep ) );
if ( !Q_Exec( NULL, (char *)g_ptr_array_index( m_pCmd, m_iCurrentStep ), NULL, true ) ) {
CString msg;
msg = "Failed to execute the following command: ";
msg += (char *)g_ptr_array_index( m_pCmd, m_iCurrentStep );
msg += "\nCheck that the file exists and that you don't run out of system resources.\n";
Sys_Printf( msg );
gtk_MessageBox( g_pParentWnd->m_pWidget, msg, "BSP monitoring", MB_OK | MB_ICONERROR );
return;
}
// re-initialise the debug window
if ( m_iCurrentStep == 0 ) {
g_DbgDlg.Init();
}
}
m_eState = EBeginStep;
}
void CWatchBSP::RoutineProcessing()
{
// used for select()
#ifdef _WIN32
TIMEVAL tout = { 0, 0 };
#endif
#if defined (__linux__) || defined (__APPLE__)
timeval tout;
tout.tv_sec = 0;
tout.tv_usec = 0;
#endif
switch (m_eState)
{
case EBeginStep:
// timeout: if we don't get an incoming connection fast enough, go back to idle
if ( g_timer_elapsed( m_pTimer, NULL ) > g_PrefsDlg.m_iTimeout )
{
gtk_MessageBox(g_pParentWnd->m_pWidget, "The connection timed out, assuming the BSP process failed\nMake sure you are using a networked version of Q3Map?\nOtherwise you need to disable BSP Monitoring in prefs.", "BSP process monitoring", MB_OK );
Reset();
if (m_bBSPPlugin)
{
// status == 1 : didn't get the connection
g_BSPFrontendTable.m_pfnEndListen(1);
}
return;
}
#ifdef _DEBUG
// some debug checks
if (!m_pListenSocket)
{
Sys_Printf("ERROR: m_pListenSocket == NULL in CWatchBSP::RoutineProcessing EBeginStep state\n");
return;
}
#endif
// we are not connected yet, accept any incoming connection
m_pInSocket = Net_Accept(m_pListenSocket);
if (m_pInSocket)
{
Sys_Printf("Connected.\n");
// prepare the message info struct for diving in
memset (&m_message_info, 0, sizeof(message_info_s));
// a dumb flag to make sure we init the push parser context when first getting a msg
m_bNeedCtxtInit = true;
m_eState = EWatching;
}
break;
case EWatching:
#ifdef _DEBUG
// some debug checks
if (!m_pInSocket)
{
Sys_Printf("ERROR: m_pInSocket == NULL in CWatchBSP::RoutineProcessing EWatching state\n");
return;
}
#endif
// select() will identify if the socket needs an update
// if the socket is identified that means there's either a message or the connection has been closed/reset/terminated
fd_set readfds;
int ret;
FD_ZERO(&readfds);
FD_SET(((unsigned int)m_pInSocket->socket), &readfds);
// from select man page:
// n is the highest-numbered descriptor in any of the three sets, plus 1
// (no use on windows)
ret = select( m_pInSocket->socket + 1, &readfds, NULL, NULL, &tout );
if (ret == SOCKET_ERROR)
{
Sys_Printf("WARNING: SOCKET_ERROR in CWatchBSP::RoutineProcessing\n");
Sys_Printf("Terminating the connection.\n");
Reset();
return;
}
#ifdef _DEBUG
if (ret == -1)
{
// we are non-blocking?? we should never get timeout errors
Sys_Printf("WARNING: unexpected timeout expired in CWatchBSP::Processing\n");
Sys_Printf("Terminating the connection.\n");
Reset();
return;
}
#endif
if (ret == 1)
{
// the socket has been identified, there's something (message or disconnection)
// see if there's anything in input
ret = Net_Receive( m_pInSocket, &msg );
if (ret > 0)
{
// unsigned int size = msg.size; //++timo just a check
strcpy (m_xmlBuf, NMSG_ReadString (&msg));
if (m_bNeedCtxtInit)
{
m_xmlParserCtxt = NULL;
m_xmlParserCtxt = xmlCreatePushParserCtxt (&saxParser, &m_message_info, m_xmlBuf, strlen(m_xmlBuf), NULL);
if (m_xmlParserCtxt == NULL)
{
Sys_FPrintf (SYS_ERR, "Failed to create the XML parser (incoming stream began with: %s)\n", m_xmlBuf);
Reset();
}
m_bNeedCtxtInit = false;
}
else
{
xmlParseChunk (m_xmlParserCtxt, m_xmlBuf, strlen(m_xmlBuf), 0);
}
}
else
{
// error or connection closed/reset
// NOTE: if we get an error down the XML stream we don't reach here
Net_Disconnect( m_pInSocket );
m_pInSocket = NULL;
Sys_Printf("Connection closed.\n");
if (m_bBSPPlugin)
{
Reset();
// let the BSP plugin know that the job is done
g_BSPFrontendTable.m_pfnEndListen(0);
return;
}
// move to next step or finish
m_iCurrentStep++;
if (m_iCurrentStep < m_pCmd->len )
{
DoEBeginStep();
}
else
{
// release the GPtrArray and the strings
if (m_pCmd != NULL)
{
for (m_iCurrentStep=0; m_iCurrentStep < m_pCmd->len; m_iCurrentStep++ )
{
delete[] (char *)g_ptr_array_index( m_pCmd, m_iCurrentStep );
}
g_ptr_array_free( m_pCmd, false );
}
m_pCmd = NULL;
// launch the engine .. OMG
if (g_PrefsDlg.m_bRunQuake)
{
// do we enter sleep mode before?
if (g_PrefsDlg.m_bDoSleep)
{
Sys_Printf("Going into sleep mode..\n");
g_pParentWnd->OnSleep();
}
Sys_Printf("Running engine...\n");
Str cmd;
// build the command line
cmd = g_pGameDescription->mEnginePath.GetBuffer();
// this is game dependant
if (!strcmp(ValueForKey(g_qeglobals.d_project_entity, "gamemode"),"mp"))
{
// MP
cmd += g_pGameDescription->mMultiplayerEngine.GetBuffer();
}
else
{
// SP
cmd += g_pGameDescription->mEngine.GetBuffer();
}
#ifdef _WIN32
// NOTE: we are using unix pathnames and CreateProcess doesn't like / in the program path
// FIXME: This isn't true anymore, doesn't it?
FindReplace( cmd, "/", "\\" );
#endif
Str cmdline;
if ( g_pGameDescription->quake2 )
{
cmdline = ". +exec radiant.cfg +map ";
cmdline += m_sBSPName;
}
else
{
cmdline = "+set sv_pure 0 ";
// TTimo: a check for vm_* but that's all fine
//cmdline = "+set sv_pure 0 +set vm_ui 0 +set vm_cgame 0 +set vm_game 0 ";
if (*ValueForKey(g_qeglobals.d_project_entity, "gamename") != '\0')
{
cmdline += "+set fs_game ";
cmdline += ValueForKey(g_qeglobals.d_project_entity, "gamename");
cmdline += " ";
}
//!\todo Read the start-map args from a config file.
if (g_pGameDescription->mGameFile == "wolf.game")
{
if (!strcmp(ValueForKey(g_qeglobals.d_project_entity, "gamemode"),"mp"))
{
// MP
cmdline += "+devmap ";
cmdline += m_sBSPName;
}
else
{
// SP
cmdline += "+set nextmap \"spdevmap ";
cmdline += m_sBSPName;
cmdline += "\"";
}
}
else
{
cmdline += "+devmap ";
cmdline += m_sBSPName;
}
}
Sys_Printf("%s %s\n", cmd.GetBuffer(), cmdline.GetBuffer());
// execute now
if (!Q_Exec(cmd.GetBuffer(), (char *)cmdline.GetBuffer(), g_pGameDescription->mEnginePath.GetBuffer(), false))
{
CString msg;
msg = "Failed to execute the following command: ";
msg += cmd; msg += cmdline;
Sys_Printf(msg);
gtk_MessageBox(g_pParentWnd->m_pWidget, msg, "BSP monitoring", MB_OK | MB_ICONERROR );
}
}
Reset();
}
}
}
break;
default:
break;
}
}
void CWatchBSP::DoMonitoringLoop( GPtrArray *pCmd, char *sBSPName )
{
if (m_sBSPName)
{
delete[] m_sBSPName;
}
m_sBSPName = sBSPName;
if (m_eState != EIdle)
{
Sys_Printf("WatchBSP got a monitoring request while not idling...\n");
// prompt the user, should we cancel the current process and go ahead?
if (gtk_MessageBox(g_pParentWnd->m_pWidget, "I am already monitoring a BSP process.\nDo you want me to override and start a new compilation?",
"BSP process monitoring", MB_YESNO ) == IDYES)
{
// disconnect and set EIdle state
Reset();
}
}
m_pCmd = pCmd;
m_iCurrentStep = 0;
DoEBeginStep();
}
void CWatchBSP::ExternalListen()
{
m_bBSPPlugin = true;
DoEBeginStep ();
}
// the part of the watchbsp interface we export to plugins
// NOTE: in the long run, the whole watchbsp.cpp interface needs to go out and be handled at the BSP plugin level
// for now we provide something really basic and limited, the essential is to have something that works fine and fast (for 1.1 final)
void WINAPI QERApp_Listen()
{
// open the listening socket
g_pParentWnd->GetWatchBSP()->ExternalListen();
}