DBGCTcp.cpp revision c98fb3e16fcd571a790eab772c0c66173d225205
/** @file
*
* Debugger TCP backend
*/
/*
* Copyright (C) 2006-2007 innotek GmbH
*
* This file is part of VirtualBox Open Source Edition (OSE), as
* available from http://www.virtualbox.org. This file is free software;
* you can redistribute it and/or modify it under the terms of the GNU
* General Public License as published by the Free Software Foundation,
* in version 2 as it comes in the "COPYING" file of the VirtualBox OSE
* distribution. VirtualBox OSE is distributed in the hope that it will
* be useful, but WITHOUT ANY WARRANTY of any kind.
*/
/*******************************************************************************
* Header Files *
*******************************************************************************/
#include <VBox/dbg.h>
#include <VBox/cfgm.h>
#include <VBox/err.h>
#include <iprt/thread.h>
#include <iprt/tcp.h>
#include <VBox/log.h>
#include <iprt/assert.h>
#include <string.h>
/*******************************************************************************
* Structures and Typedefs *
*******************************************************************************/
/**
* Debug console TCP backend instance data.
*/
typedef struct DBGCTCP
{
/** The I/O backend for the console. */
DBGCBACK Back;
/** The socket of the connection. */
RTSOCKET Sock;
/** Connection status. */
bool fAlive;
} DBGCTCP;
/** Pointer to the instance data of the console TCP backend. */
typedef DBGCTCP *PDBGCTCP;
/** Converts a pointer to DBGCTCP::Back to a pointer to DBGCTCP. */
#define DBGCTCP_BACK2DBGCTCP(pBack) ( (PDBGCTCP)((char *)pBack - RT_OFFSETOF(DBGCTCP, Back)) )
/*******************************************************************************
* Internal Functions *
*******************************************************************************/
static int dbgcTcpConnection(RTSOCKET Sock, void *pvUser);
/**
* Checks if there is input.
*
* @returns true if there is input ready.
* @returns false if there not input ready.
* @param pBack Pointer to the backend structure supplied by
* the backend. The backend can use this to find
* it's instance data.
* @param cMillies Number of milliseconds to wait on input data.
*/
static DECLCALLBACK(bool) dbgcTcpBackInput(PDBGCBACK pBack, uint32_t cMillies)
{
PDBGCTCP pDbgcTcp = DBGCTCP_BACK2DBGCTCP(pBack);
if (!pDbgcTcp->fAlive)
return false;
int rc = RTTcpSelectOne(pDbgcTcp->Sock, cMillies);
if (VBOX_FAILURE(rc) && rc != VERR_TIMEOUT)
pDbgcTcp->fAlive = false;
return rc != VERR_TIMEOUT;
}
/**
* Read input.
*
* @returns VBox status code.
* @param pBack Pointer to the backend structure supplied by
* the backend. The backend can use this to find
* it's instance data.
* @param pvBuf Where to put the bytes we read.
* @param cbBuf Maximum nymber of bytes to read.
* @param pcbRead Where to store the number of bytes actually read.
* If NULL the entire buffer must be filled for a
* successful return.
*/
static DECLCALLBACK(int) dbgcTcpBackRead(PDBGCBACK pBack, void *pvBuf, size_t cbBuf, size_t *pcbRead)
{
PDBGCTCP pDbgcTcp = DBGCTCP_BACK2DBGCTCP(pBack);
if (!pDbgcTcp->fAlive)
return VERR_INVALID_HANDLE;
int rc = RTTcpRead(pDbgcTcp->Sock, pvBuf, cbBuf, pcbRead);
if (VBOX_FAILURE(rc))
pDbgcTcp->fAlive = false;
return rc;
}
/**
* Write (output).
*
* @returns VBox status code.
* @param pBack Pointer to the backend structure supplied by
* the backend. The backend can use this to find
* it's instance data.
* @param pvBuf What to write.
* @param cbBuf Number of bytes to write.
* @param pcbWritten Where to store the number of bytes actually written.
* If NULL the entire buffer must be successfully written.
*/
static DECLCALLBACK(int) dbgcTcpBackWrite(PDBGCBACK pBack, const void *pvBuf, size_t cbBuf, size_t *pcbWritten)
{
PDBGCTCP pDbgcTcp = DBGCTCP_BACK2DBGCTCP(pBack);
if (!pDbgcTcp->fAlive)
return VERR_INVALID_HANDLE;
/*
* convert '\n' to '\r\n' while writing.
*/
int rc = 0;
size_t cbLeft = cbBuf;
while (cbLeft)
{
size_t cb = cbLeft;
/* write newlines */
if (*(const char *)pvBuf == '\n')
{
rc = RTTcpWrite(pDbgcTcp->Sock, "\n\r", 2);
cb = 1;
}
/* write till next newline */
else
{
const char *pszNL = (const char *)memchr(pvBuf, '\n', cbLeft);
if (pszNL)
cb = (uintptr_t)pszNL - (uintptr_t)pvBuf;
rc = RTTcpWrite(pDbgcTcp->Sock, pvBuf, cb);
}
if (VBOX_FAILURE(rc))
{
pDbgcTcp->fAlive = false;
break;
}
/* advance */
cbLeft -= cb;
pvBuf = (const char *)pvBuf + cb;
}
/*
* Set returned value and return.
*/
if (pcbWritten)
*pcbWritten = cbBuf - cbLeft;
return rc;
}
/**
* Serve a TCP Server connection.
*
* @returns VBox status.
* @returns VERR_TCP_SERVER_STOP to terminate the server loop forcing
* the RTTcpCreateServer() call to return.
* @param Sock The socket which the client is connected to.
* The call will close this socket.
* @param pvUser The VM handle.
*/
static int dbgcTcpConnection(RTSOCKET Sock, void *pvUser)
{
LogFlow(("dbgcTcpConnection: connection! Sock=%d pvUser=%p\n", Sock, pvUser));
/*
* Start the console.
*/
DBGCTCP DbgcTcp;
DbgcTcp.Back.pfnInput = dbgcTcpBackInput;
DbgcTcp.Back.pfnRead = dbgcTcpBackRead;
DbgcTcp.Back.pfnWrite = dbgcTcpBackWrite;
DbgcTcp.fAlive = true;
DbgcTcp.Sock = Sock;
int rc = DBGCCreate((PVM)pvUser, &DbgcTcp.Back, 0);
LogFlow(("dbgcTcpConnection: disconnect rc=%Vrc\n", rc));
return rc;
}
/**
* Spawns a new thread with a TCP based debugging console service.
*
* @returns VBox status.
* @param pVM VM handle.
* @param ppvData Where to store a pointer to the instance data.
*/
DBGDECL(int) DBGCTcpCreate(PVM pVM, void **ppvData)
{
/*
* Check what the configuration says.
*/
PCFGMNODE pKey = CFGMR3GetChild(CFGMR3GetRoot(pVM), "DBGC");
bool fEnabled;
int rc = CFGMR3QueryBool(pKey, "Enabled", &fEnabled);
if (rc == VERR_CFGM_VALUE_NOT_FOUND || rc == VERR_CFGM_NO_PARENT)
#if defined(DEBUG) && !defined(__L4ENV__) && !defined(DEBUG_dmik)
fEnabled = true;
#else
fEnabled = false;
#endif
else if (VBOX_FAILURE(rc))
{
AssertMsgFailed(("Configuration error: Querying \"Enabled\" -> %Vrc\n", rc));
return rc;
}
if (!fEnabled)
{
LogFlow(("DBGCTcpCreate: returns VINF_SUCCESS (Disabled)\n"));
return VINF_SUCCESS;
}
/*
* Get the port configuration.
*/
uint32_t u32Port;
rc = CFGMR3QueryU32(pKey, "Port", &u32Port);
if (rc == VERR_CFGM_VALUE_NOT_FOUND || rc == VERR_CFGM_NO_PARENT)
u32Port = 5000;
else if (VBOX_FAILURE(rc))
{
AssertMsgFailed(("Configuration error: Querying \"Port\" -> %Vrc\n", rc));
return rc;
}
/*
* Get the address configuration.
*/
char szAddress[512];
rc = CFGMR3QueryString(pKey, "Address", szAddress, sizeof(szAddress));
if (rc == VERR_CFGM_VALUE_NOT_FOUND || rc == VERR_CFGM_NO_PARENT)
szAddress[0] = '\0';
else if (VBOX_FAILURE(rc))
{
AssertMsgFailed(("Configuration error: Querying \"Address\" -> %Vrc\n", rc));
return rc;
}
/*
* Create the server (separate thread).
*/
PRTTCPSERVER pServer;
rc = RTTcpServerCreate(szAddress, u32Port, RTTHREADTYPE_DEBUGGER, "DBGC", dbgcTcpConnection, pVM, &pServer);
if (VBOX_SUCCESS(rc))
{
LogFlow(("DBGCTcpCreate: Created server on port %d %s\n", u32Port, szAddress));
*ppvData = pServer;
}
LogFlow(("DBGCTcpCreate: returns %Vrc\n", rc));
return rc;
}
/**
* Terminates any running TCP base debugger consolse service.
*
* @returns VBox status.
* @param pVM VM handle.
*/
DBGDECL(int) DBGCTcpTerminate(PVM pVM, void *pvData)
{
/*
* Destroy the server instance if any.
*/
if (pvData)
{
int rc = RTTcpServerDestroy((PRTTCPSERVER)pvData);
AssertRC(rc);
}
return VINF_SUCCESS;
}