#include "winutils.h"

#include <VBox/com/com.h>

#include <VBox/com/listeners.h>

#include <VBox/com/string.h>

#include <VBox/com/Guid.h>

#include <VBox/com/array.h>

#include <VBox/com/ErrorInfo.h>

#include <VBox/com/errorprint.h>

#include <VBox/com/VirtualBox.h>

#include <iprt/net.h>

#include <iprt/initterm.h>

#include <iprt/alloca.h>

#ifndef RT_OS_WINDOWS

# include <arpa/inet.h>

#endif

#include <iprt/err.h>

#include <iprt/time.h>

#include <iprt/timer.h>

#include <iprt/thread.h>

#include <iprt/stream.h>

#include <iprt/path.h>

#include <iprt/param.h>

#include <iprt/pipe.h>

#include <iprt/getopt.h>

#include <iprt/string.h>

#include <iprt/mem.h>

#include <iprt/message.h>

#include <iprt/req.h>

#include <iprt/file.h>

#include <iprt/semaphore.h>

#include <iprt/cpp/utils.h>

#define LOG_GROUP LOG_GROUP_NAT_SERVICE

#include <VBox/log.h>

#include <VBox/sup.h>

#include <VBox/intnet.h>

#include <VBox/intnetinline.h>

#include <VBox/vmm/pdmnetinline.h>

#include <VBox/vmm/vmm.h>

#include <VBox/version.h>

#ifndef RT_OS_WINDOWS

# include <sys/poll.h>

# include <sys/socket.h>

# include <netinet/in.h>

#endif

#include <vector>

#include <string>

#include "../NetLib/VBoxNetLib.h"

#include "../NetLib/VBoxNetBaseService.h"

#include "VBoxLwipCore.h"

extern "C"

{

#include "lwip/opt.h"

#ifdef LWIP_SOCKET

# undef LWIP_SOCKET

# define LWIP_SOCKET 0

#endif

#include "lwip/sys.h"

#include "lwip/stats.h"

#include "lwip/mem.h"

#include "lwip/memp.h"

#include "lwip/pbuf.h"

#include "lwip/netif.h"

#include "lwip/api.h"

#include "lwip/tcp_impl.h"

#include "ipv6/lwip/ethip6.h"

#include "lwip/nd6.h" // for proxy_na_hook

#include "lwip/mld6.h"

#include "lwip/udp.h"

#include "lwip/tcp.h"

#include "lwip/tcpip.h"

#include "lwip/sockets.h"

#include "netif/etharp.h"

#include "proxytest.h"

#include "pxremap.h"

#include "portfwd.h"

}

#include "../NetLib/VBoxPortForwardString.h"

static RTGETOPTDEF g_aGetOptDef[] =

{

{ "--port-forward4", 'p', RTGETOPT_REQ_STRING },

{ "--port-forward6", 'P', RTGETOPT_REQ_STRING }

};

typedef struct NATSEVICEPORTFORWARDRULE

{

PORTFORWARDRULE Pfr;

fwspec FWSpec;

} NATSEVICEPORTFORWARDRULE, *PNATSEVICEPORTFORWARDRULE;

typedef std::vector<NATSEVICEPORTFORWARDRULE> VECNATSERVICEPF;

typedef VECNATSERVICEPF::iterator ITERATORNATSERVICEPF;

typedef VECNATSERVICEPF::const_iterator CITERATORNATSERVICEPF;

struct ip4address2off

{

struct sockaddr_in addr;

uint32_t off;

};

class PortForwardListener;

class VBoxNetLwipNAT: public VBoxNetBaseService

{

friend class PortForwardListener;

public:

VBoxNetLwipNAT();

virtual ~VBoxNetLwipNAT();

void usage(){ /* @todo: should be implemented */ };

int run();

virtual int init(void);

/* @todo: when configuration would be really needed */

virtual int parseOpt(int rc, const RTGETOPTUNION& getOptVal);

private:

struct proxy_options m_ProxyOptions;

struct sockaddr_in m_src4;

struct sockaddr_in6 m_src6;

/**

ip4address2off m_lo2off[10];

uint16_t m_u16Mtu;

netif m_LwipNetIf;

/* Queues */

RTREQQUEUE hReqIntNet;

/* thread where we're waiting for a frames, no semaphores needed */

RTTHREAD hThrIntNetRecv;

/* Our NAT network descriptor in Main */

ComPtr<INATNetwork> net;

STDMETHOD(HandleEvent)(VBoxEventType_T aEventType, IEvent *pEvent);

RTSEMEVENT hSemSVC;

/* Only for debug needs, by default NAT service should load rules from SVC

bool fDontLoadRulesOnStartup;

static void onLwipTcpIpInit(void *arg);

static void onLwipTcpIpFini(void *arg);

static err_t netifInit(netif *pNetif);

static err_t netifLinkoutput(netif *pNetif, pbuf *pBuf);

static int intNetThreadRecv(RTTHREAD, void *);

static void vboxNetLwipNATProcessXmit(void);

VECNATSERVICEPF m_vecPortForwardRule4;

VECNATSERVICEPF m_vecPortForwardRule6;

static int natServicePfRegister(NATSEVICEPORTFORWARDRULE& natServicePf);

static int natServiceProcessRegisteredPf(VECNATSERVICEPF& vecPf);

};

class PortForwardListener

{

public:

PortForwardListener(){}

virtual ~PortForwardListener(){}

HRESULT init(VBoxNetLwipNAT *pNAT) { m_pNAT = pNAT; return S_OK; }

HRESULT init() { return S_OK; }

void uninit(){}

STDMETHOD(HandleEvent)(VBoxEventType_T aEventType, IEvent *pEvent)

{return m_pNAT->HandleEvent(aEventType, pEvent);};

private:

VBoxNetLwipNAT *m_pNAT;

};

typedef ListenerImpl<PortForwardListener, VBoxNetLwipNAT *> PortForwardListenerImpl;

VBOX_LISTENER_DECLARE(PortForwardListenerImpl)

static VBoxNetLwipNAT *g_pLwipNat;

STDMETHODIMP VBoxNetLwipNAT::HandleEvent(VBoxEventType_T aEventType,

IEvent *pEvent)

{

HRESULT hrc = S_OK;

switch (aEventType)

{

case VBoxEventType_OnNATNetworkPortForward:

{

com::Bstr name, strHostAddr, strGuestAddr;

LONG lHostPort, lGuestPort;

BOOL fCreateFW, fIPv6FW;

NATProtocol_T proto = NATProtocol_TCP;

ComPtr<INATNetworkPortForwardEvent> pfEvt = pEvent;

hrc = pfEvt->COMGETTER(Name)(name.asOutParam());

AssertReturn(SUCCEEDED(hrc), hrc);

hrc = pfEvt->COMGETTER(Proto)(&proto);

AssertReturn(SUCCEEDED(hrc), hrc);

hrc = pfEvt->COMGETTER(HostIp)(strHostAddr.asOutParam());

AssertReturn(SUCCEEDED(hrc), hrc);

hrc = pfEvt->COMGETTER(HostPort)(&lHostPort);

AssertReturn(SUCCEEDED(hrc), hrc);

hrc = pfEvt->COMGETTER(GuestIp)(strGuestAddr.asOutParam());

AssertReturn(SUCCEEDED(hrc), hrc);

hrc = pfEvt->COMGETTER(HostPort)(&lGuestPort);

AssertReturn(SUCCEEDED(hrc), hrc);

hrc = pfEvt->COMGETTER(Create)(&fCreateFW);

AssertReturn(SUCCEEDED(hrc), hrc);

hrc = pfEvt->COMGETTER(Ipv6)(&fIPv6FW);

AssertReturn(SUCCEEDED(hrc), hrc);

VECNATSERVICEPF& rules = (fIPv6FW ?

m_vecPortForwardRule6 :

m_vecPortForwardRule4);

NATSEVICEPORTFORWARDRULE r;

RT_ZERO(r);

if (name.length() > RT_ELEMENTS(r.Pfr.aszPfrName))

{

hrc = E_INVALIDARG;

goto port_forward_done;

}

switch (proto)

{

case NATProtocol_TCP:

r.Pfr.iPfrProto = IPPROTO_TCP;

break;

case NATProtocol_UDP:

r.Pfr.iPfrProto = IPPROTO_UDP;

break;

default:

goto port_forward_done;

}

RTStrPrintf(r.Pfr.aszPfrName, RT_ELEMENTS(r.Pfr.aszPfrName),

"%s",

com::Utf8Str(name).c_str());

RTStrPrintf(r.Pfr.aszPfrHostAddr, RT_ELEMENTS(r.Pfr.aszPfrHostAddr),

"%s",

com::Utf8Str(strHostAddr).c_str());

/* XXX: limits should be checked */

r.Pfr.u16PfrHostPort = (uint16_t)lHostPort;

RTStrPrintf(r.Pfr.aszPfrHostAddr, RT_ELEMENTS(r.Pfr.aszPfrGuestAddr),

"%s",

com::Utf8Str(strHostAddr).c_str());

/* XXX: limits should be checked */

r.Pfr.u16PfrGuestPort = (uint16_t)lGuestPort;

if (fCreateFW) /* Addition */

{

rules.push_back(r);

natServicePfRegister(rules.back());

}

else /* Deletion */

{

ITERATORNATSERVICEPF it;

for (it = rules.begin(); it != rules.end(); ++it)

{

/* compare */

NATSEVICEPORTFORWARDRULE& natFw = *it;

if ( natFw.Pfr.iPfrProto == r.Pfr.iPfrProto

&& natFw.Pfr.u16PfrHostPort == r.Pfr.u16PfrHostPort

&& (strncmp(natFw.Pfr.aszPfrHostAddr, r.Pfr.aszPfrHostAddr, INET6_ADDRSTRLEN) == 0)

&& natFw.Pfr.u16PfrGuestPort == r.Pfr.u16PfrGuestPort

&& (strncmp(natFw.Pfr.aszPfrGuestAddr, r.Pfr.aszPfrGuestAddr, INET6_ADDRSTRLEN) == 0))

{

fwspec *pFwCopy = (fwspec *)RTMemAllocZ(sizeof(fwspec));

if (!pFwCopy)

{

break;

}

memcpy(pFwCopy, &natFw.FWSpec, sizeof(fwspec));

/* We shouldn't care about pFwCopy this memory will be freed when

portfwd_rule_del(pFwCopy);

rules.erase(it);

break;

}

} /* loop over vector elements */

} /* condition add or delete */

port_forward_done:

/* clean up strings */

name.setNull();

strHostAddr.setNull();

strGuestAddr.setNull();

break;

}

}

return hrc;

}

void VBoxNetLwipNAT::onLwipTcpIpInit(void* arg)

{

AssertPtrReturnVoid(arg);

VBoxNetLwipNAT *pThis = (VBoxNetLwipNAT *)arg;

HRESULT hrc = com::Initialize();

Assert(!FAILED(hrc));

proxy_arp_hook = pxremap_proxy_arp;

proxy_ip4_divert_hook = pxremap_ip4_divert;

proxy_na_hook = pxremap_proxy_na;

proxy_ip6_divert_hook = pxremap_ip6_divert;

/* lwip thread */

RTNETADDRIPV4 IpNetwork;

IpNetwork.u = g_pLwipNat->m_Ipv4Address.u & g_pLwipNat->m_Ipv4Netmask.u;

ip_addr LwipIpAddr, LwipIpNetMask, LwipIpNetwork;

memcpy(&LwipIpAddr, &g_pLwipNat->m_Ipv4Address, sizeof(ip_addr));

memcpy(&LwipIpNetMask, &g_pLwipNat->m_Ipv4Netmask, sizeof(ip_addr));

memcpy(&LwipIpNetwork, &IpNetwork, sizeof(ip_addr));

netif *pNetif = netif_add(&g_pLwipNat->m_LwipNetIf /* Lwip Interface */,

&LwipIpAddr /* IP address*/,

&LwipIpNetMask /* Network mask */,

&LwipIpAddr /* gateway address, @todo: is self IP acceptable? */,

g_pLwipNat /* state */,

VBoxNetLwipNAT::netifInit /* netif_init_fn */,

lwip_tcpip_input /* netif_input_fn */);

AssertPtrReturnVoid(pNetif);

netif_set_up(pNetif);

netif_set_link_up(pNetif);

/*

for (int i = 0; i <= 1; ++i) {

ip6_addr_t *paddr = netif_ip6_addr(pNetif, i);

ip6_addr_t solicited_node_multicast_address;

ip6_addr_set_solicitednode(&solicited_node_multicast_address,

paddr->addr[3]);

mld6_joingroup(paddr, &solicited_node_multicast_address);

}

/*

{

ip6_addr_t solicited_node_multicast_address;

ip6_addr_set_solicitednode(&solicited_node_multicast_address,

/* last 24 bits of the address */

PP_HTONL(0x00000002));

mld6_netif_joingroup(pNetif, &solicited_node_multicast_address);

}

proxy_init(&g_pLwipNat->m_LwipNetIf, &g_pLwipNat->m_ProxyOptions);

natServiceProcessRegisteredPf(g_pLwipNat->m_vecPortForwardRule4);

natServiceProcessRegisteredPf(g_pLwipNat->m_vecPortForwardRule6);

}

void VBoxNetLwipNAT::onLwipTcpIpFini(void* arg)

{

AssertPtrReturnVoid(arg);

VBoxNetLwipNAT *pThis = (VBoxNetLwipNAT *)arg;

/* XXX: proxy finalization */

netif_set_link_down(&g_pLwipNat->m_LwipNetIf);

netif_set_down(&g_pLwipNat->m_LwipNetIf);

netif_remove(&g_pLwipNat->m_LwipNetIf);

}

err_t VBoxNetLwipNAT::netifInit(netif *pNetif)

{

err_t rcLwip = ERR_OK;

AssertPtrReturn(pNetif, ERR_ARG);

VBoxNetLwipNAT *pNat = static_cast<VBoxNetLwipNAT *>(pNetif->state);

AssertPtrReturn(pNat, ERR_ARG);

LogFlowFunc(("ENTER: pNetif[%c%c%d]\n", pNetif->name[0], pNetif->name[1], pNetif->num));

/* validity */

AssertReturn( pNetif->name[0] == 'N'

&& pNetif->name[1] == 'T', ERR_ARG);

pNetif->hwaddr_len = sizeof(RTMAC);

memcpy(pNetif->hwaddr, &pNat->m_MacAddress, sizeof(RTMAC));

pNat->m_u16Mtu = 1500; // XXX: FIXME

pNetif->mtu = pNat->m_u16Mtu;

pNetif->flags = NETIF_FLAG_BROADCAST

| NETIF_FLAG_ETHARP /* Don't bother driver with ARP and let Lwip resolve ARP handling */

| NETIF_FLAG_ETHERNET; /* Lwip works with ethernet too */

pNetif->linkoutput = netifLinkoutput; /* ether-level-pipe */

pNetif->output = lwip_etharp_output; /* ip-pipe */

pNetif->output_ip6 = ethip6_output;

/* IPv6 link-local address in slot 0 */

netif_create_ip6_linklocal_address(pNetif, /* :from_mac_48bit */ 1);

netif_ip6_addr_set_state(pNetif, 0, IP6_ADDR_PREFERRED); // skip DAD

/*

u8_t nethi = ip4_addr2(&pNetif->ip_addr);

u8_t netlo = ip4_addr3(&pNetif->ip_addr);

ip6_addr_t *paddr = netif_ip6_addr(pNetif, 1);

IP6_ADDR(paddr, 0, 0xFD, 0x17, 0x62, 0x5C);

IP6_ADDR(paddr, 1, 0xF0, 0x37, nethi, netlo);

IP6_ADDR(paddr, 2, 0x00, 0x00, 0x00, 0x00);

IP6_ADDR(paddr, 3, 0x00, 0x00, 0x00, 0x01);

netif_ip6_addr_set_state(pNetif, 1, IP6_ADDR_PREFERRED);

#if LWIP_IPV6_SEND_ROUTER_SOLICIT

pNetif->rs_count = 0;

#endif

LogFlowFunc(("LEAVE: %d\n", rcLwip));

return rcLwip;

}

int VBoxNetLwipNAT::intNetThreadRecv(RTTHREAD, void *)

{

int rc = VINF_SUCCESS;

/* 1. initialization and connection */

HRESULT hrc = com::Initialize();

if (FAILED(hrc))

return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to initialize COM!");

/* Well we're ready */

PINTNETRINGBUF pRingBuf = &g_pLwipNat->m_pIfBuf->Recv;

for (;;)

{

/*

/* 2. waiting for request for */

rc = g_pLwipNat->waitForIntNetEvent(2000);

if (RT_FAILURE(rc))

{

if (rc == VERR_TIMEOUT || rc == VERR_INTERRUPTED)

{

/* do we want interrupt anyone ??? */

continue;

}

LogRel(("VBoxNetNAT: waitForIntNetEvent returned %Rrc\n", rc));

AssertRCReturn(rc,ERR_IF);

}

/*

PCINTNETHDR pHdr;

while ((pHdr = IntNetRingGetNextFrameToRead(pRingBuf)) != NULL)

{

uint8_t const u8Type = pHdr->u8Type;

size_t cbFrame = pHdr->cbFrame;

uint8_t *pu8Frame = NULL;

pbuf *pPbufHdr = NULL;

pbuf *pPbuf = NULL;

switch (u8Type)

{

case INTNETHDR_TYPE_FRAME:

/* @todo:should it be really here?

pPbufHdr = pPbuf = pbuf_alloc(PBUF_RAW, pHdr->cbFrame, PBUF_POOL);

if (!pPbuf)

{

LogRel(("NAT: Can't allocate send buffer cbFrame=%u\n", cbFrame));

break;

}

Assert(pPbufHdr->tot_len == cbFrame);

pu8Frame = (uint8_t *)IntNetHdrGetFramePtr(pHdr, g_pLwipNat->m_pIfBuf);

while(pPbuf)

{

memcpy(pPbuf->payload, pu8Frame, pPbuf->len);

pu8Frame += pPbuf->len;

pPbuf = pPbuf->next;

}

g_pLwipNat->m_LwipNetIf.input(pPbufHdr, &g_pLwipNat->m_LwipNetIf);

AssertReleaseRC(rc);

break;

case INTNETHDR_TYPE_GSO:

{

PCPDMNETWORKGSO pGso = IntNetHdrGetGsoContext(pHdr,

g_pLwipNat->m_pIfBuf);

if (!PDMNetGsoIsValid(pGso, cbFrame,

cbFrame - sizeof(PDMNETWORKGSO)))

break;

cbFrame -= sizeof(PDMNETWORKGSO);

uint8_t abHdrScratch[256];

uint32_t const cSegs = PDMNetGsoCalcSegmentCount(pGso,

cbFrame);

for (size_t iSeg = 0; iSeg < cSegs; iSeg++)

{

uint32_t cbSegFrame;

void *pvSegFrame =

PDMNetGsoCarveSegmentQD(pGso,

(uint8_t *)(pGso + 1),

cbFrame,

abHdrScratch,

iSeg,

cSegs,

&cbSegFrame);

pPbuf = pbuf_alloc(PBUF_RAW, cbSegFrame, PBUF_POOL);

if (!pPbuf)

{

LogRel(("NAT: Can't allocate send buffer cbFrame=%u\n", cbSegFrame));

break;

}

Assert( !pPbuf->next

&& pPbuf->len == cbSegFrame);

memcpy(pPbuf->payload, pvSegFrame, cbSegFrame);

g_pLwipNat->m_LwipNetIf.input(pPbuf, &g_pLwipNat->m_LwipNetIf);

}

}

break;

case INTNETHDR_TYPE_PADDING:

break;

default:

STAM_REL_COUNTER_INC(&g_pLwipNat->m_pIfBuf->cStatBadFrames);

break;

}

IntNetRingSkipFrame(&g_pLwipNat->m_pIfBuf->Recv);

} /* loop */

}

/* 3. deinitilization and termination */

LogFlowFuncLeaveRC(rc);

return rc;

}

void VBoxNetLwipNAT::vboxNetLwipNATProcessXmit()

{

int rc = VINF_SUCCESS;

INTNETIFSENDREQ SendReq;

SendReq.Hdr.u32Magic = SUPVMMR0REQHDR_MAGIC;

SendReq.Hdr.cbReq = sizeof(SendReq);

SendReq.pSession = g_pLwipNat->m_pSession;

SendReq.hIf = g_pLwipNat->m_hIf;

rc = SUPR3CallVMMR0Ex(NIL_RTR0PTR, NIL_VMCPUID, VMMR0_DO_INTNET_IF_SEND, 0, &SendReq.Hdr);

AssertRC(rc);

}

err_t VBoxNetLwipNAT::netifLinkoutput(netif *pNetif, pbuf *pPBuf)

{

int rc = VINF_SUCCESS;

err_t rcLwip = ERR_OK;

AssertPtrReturn(pNetif, ERR_ARG);

AssertPtrReturn(pPBuf, ERR_ARG);

AssertReturn((void *)g_pLwipNat == pNetif->state, ERR_ARG);

LogFlowFunc(("ENTER: pNetif[%c%c%d], pPbuf:%p\n",

pNetif->name[0],

pNetif->name[1],

pNetif->num,

pPBuf));

/*

/* see p.1 */

rc = VINF_SUCCESS;

PINTNETHDR pHdr = NULL;

uint8_t *pu8Frame = NULL;

int offFrame = 0;

int idxSg = 0;

struct pbuf *pPBufPtr = pPBuf;

/* Allocate frame, and pad it if required. */

rc = IntNetRingAllocateFrame(&g_pLwipNat->m_pIfBuf->Send, pPBuf->tot_len, &pHdr, (void **)&pu8Frame);

if (RT_SUCCESS(rc))

{

/* see p. 2 */

while (pPBufPtr)

{

memcpy(&pu8Frame[offFrame], pPBufPtr->payload, pPBufPtr->len);

offFrame += pPBufPtr->len;

pPBufPtr = pPBufPtr->next;

}

}

if (RT_FAILURE(rc))

{

/* Could it be that some frames are still in the ring buffer */

/* 1.c */

AssertMsgFailed(("Debug Me!"));

}

/* Commit - what really this function do */

IntNetRingCommitFrameEx(&g_pLwipNat->m_pIfBuf->Send, pHdr, pPBuf->tot_len);

g_pLwipNat->vboxNetLwipNATProcessXmit();

AssertRCReturn(rc, ERR_IF);

LogFlowFunc(("LEAVE: %d\n", rcLwip));

return rcLwip;

}

VBoxNetLwipNAT::VBoxNetLwipNAT()

{

LogFlowFuncEnter();

m_ProxyOptions.tftp_root = NULL;

m_ProxyOptions.src4 = NULL;

m_ProxyOptions.src6 = NULL;

memset(&m_src4, 0, sizeof(m_src4));

memset(&m_src6, 0, sizeof(m_src6));

m_src4.sin_family = AF_INET;

m_src6.sin6_family = AF_INET6;

#if HAVE_SA_LEN

m_src4.sin_len = sizeof(m_src4);

m_src6.sin6_len = sizeof(m_src6);

#endif

m_LwipNetIf.name[0] = 'N';

m_LwipNetIf.name[1] = 'T';

m_MacAddress.au8[0] = 0x52;

m_MacAddress.au8[1] = 0x54;

m_MacAddress.au8[2] = 0;

m_MacAddress.au8[3] = 0x12;

m_MacAddress.au8[4] = 0x35;

m_MacAddress.au8[5] = 0;

m_Ipv4Address.u = RT_MAKE_U32_FROM_U8( 10, 0, 2, 2); // NB: big-endian

m_Ipv4Netmask.u = RT_H2N_U32_C(0xffffff00);

for(unsigned int i = 0; i < RT_ELEMENTS(g_aGetOptDef); ++i)

m_vecOptionDefs.push_back(&g_aGetOptDef[i]);

m_enmTrunkType = kIntNetTrunkType_SrvNat;

LogFlowFuncLeave();

}

VBoxNetLwipNAT::~VBoxNetLwipNAT()

{

if (m_ProxyOptions.tftp_root != NULL)

{

RTStrFree((char *)m_ProxyOptions.tftp_root);

}

}

int VBoxNetLwipNAT::natServicePfRegister(NATSEVICEPORTFORWARDRULE& natPf)

{

int lrc = 0;

int rc = VINF_SUCCESS;

int socketSpec = SOCK_STREAM;

char *pszHostAddr;

int sockFamily = (natPf.Pfr.fPfrIPv6 ? PF_INET6 : PF_INET);

switch(natPf.Pfr.iPfrProto)

{

case IPPROTO_TCP:

socketSpec = SOCK_STREAM;

break;

case IPPROTO_UDP:

socketSpec = SOCK_DGRAM;

break;

default:

return VERR_IGNORED; /* Ah, just ignore the garbage */

}

pszHostAddr = natPf.Pfr.aszPfrHostAddr;

/* XXX: workaround for inet_pton and an empty ipv4 address

if ( sockFamily == PF_INET

&& pszHostAddr[0] == 0)

pszHostAddr = (char *)"0.0.0.0"; /* XXX: fix it! without type cast */

lrc = fwspec_set(&natPf.FWSpec,

sockFamily,

socketSpec,

pszHostAddr,

natPf.Pfr.u16PfrHostPort,

natPf.Pfr.aszPfrGuestAddr,

natPf.Pfr.u16PfrGuestPort);

AssertReturn(!lrc, VERR_IGNORED);

fwspec *pFwCopy = (fwspec *)RTMemAllocZ(sizeof(fwspec));

AssertPtrReturn(pFwCopy, VERR_IGNORED);

/*

memcpy(pFwCopy, &natPf.FWSpec, sizeof(fwspec));

lrc = portfwd_rule_add(pFwCopy);

AssertReturn(!lrc, VERR_IGNORED);

return VINF_SUCCESS;

}

int VBoxNetLwipNAT::natServiceProcessRegisteredPf(VECNATSERVICEPF& vecRules){

ITERATORNATSERVICEPF it;

for (it = vecRules.begin();

it != vecRules.end(); ++it)

{

int rc = natServicePfRegister((*it));

if (RT_FAILURE(rc))

{

LogRel(("PF: %s is ignored\n", (*it).Pfr.aszPfrName));

continue;

}

}

return VINF_SUCCESS;

}

int VBoxNetLwipNAT::init()

{

com::Bstr bstr;

int rc = VINF_SUCCESS;

LogFlowFuncEnter();

/* virtualbox initialized in super class */

HRESULT hrc;

hrc = virtualbox->FindNATNetworkByName(com::Bstr(m_Network.c_str()).raw(),

net.asOutParam());

AssertComRCReturn(hrc, VERR_NOT_FOUND);

#if !defined(RT_OS_WINDOWS)

/* XXX: Temporaly disabled this code on Windows for further debugging */

ComPtr<IEventSource> pES;

hrc = net->COMGETTER(EventSource)(pES.asOutParam());

AssertComRC(hrc);

ComObjPtr<PortForwardListenerImpl> listener;

hrc = listener.createObject();

AssertComRCReturn(hrc, VERR_INTERNAL_ERROR);

hrc = listener->init(new PortForwardListener(), this);

AssertComRCReturn(hrc, VERR_INTERNAL_ERROR);

com::SafeArray<VBoxEventType_T> events;

events.push_back(VBoxEventType_OnNATNetworkPortForward);

hrc = pES->RegisterListener(listener, ComSafeArrayAsInParam(events), true);

AssertComRCReturn(hrc, VERR_INTERNAL_ERROR);

#endif

com::Bstr bstrSourceIp4Key = com::BstrFmt("NAT/%s/SourceIp4",m_Network.c_str());

com::Bstr bstrSourceIpX;

RTNETADDRIPV4 addr;

hrc = virtualbox->GetExtraData(bstrSourceIp4Key.raw(), bstrSourceIpX.asOutParam());

if (SUCCEEDED(hrc))

{

int rc = RTNetStrToIPv4Addr(com::Utf8Str(bstrSourceIpX).c_str(), &addr);

if (RT_SUCCESS(rc))

{

RT_ZERO(m_src4);

m_src4.sin_addr.s_addr = addr.u;

m_ProxyOptions.src4 = &m_src4;

bstrSourceIpX.setNull();

}

}

if (!fDontLoadRulesOnStartup)

{

/* XXX: extract function and do not duplicate */

com::SafeArray<BSTR> rules;

hrc = net->COMGETTER(PortForwardRules4)(ComSafeArrayAsOutParam(rules));

Assert(SUCCEEDED(hrc));

size_t idxRules = 0;

for (idxRules = 0; idxRules < rules.size(); ++idxRules)

{

Log(("%d-rule: %ls\n", idxRules, rules[idxRules]));

NATSEVICEPORTFORWARDRULE Rule;

RT_ZERO(Rule);

rc = netPfStrToPf(com::Utf8Str(rules[idxRules]).c_str(), 0, &Rule.Pfr);

AssertRC(rc);

m_vecPortForwardRule4.push_back(Rule);

}

rules.setNull();

hrc = net->COMGETTER(PortForwardRules6)(ComSafeArrayAsOutParam(rules));

Assert(SUCCEEDED(hrc));

for (idxRules = 0; idxRules < rules.size(); ++idxRules)

{

Log(("%d-rule: %ls\n", idxRules, rules[idxRules]));

NATSEVICEPORTFORWARDRULE Rule;

netPfStrToPf(com::Utf8Str(rules[idxRules]).c_str(), 1, &Rule.Pfr);

m_vecPortForwardRule6.push_back(Rule);

}

#if 0 /* fetching local mappings */

com::SafeArray<BSTR> strs;

int count_strs;

hrc = net->COMGETTER(LocalMappings)(ComSafeArrayAsOutParam(strs));

if ( SUCCEEDED(hrc)

&& (count_strs = strs.size()))

{

unsigned int j = 0;

int i;

for (i = 0; i < count_strs && j < RT_ELEMENTS(m_lo2off); ++i)

{

char aszAddr[17];

RTNETADDRIPV4 ip4addr;

char *pszTerm;

uint32_t u32Off;

const char *pszLo2Off = com::Utf8Str(strs[i]).c_str();

RT_ZERO(aszAddr);

pszTerm = RTStrStr(pszLo2Off, ";");

if ( !pszTerm

&& (pszTerm - pszLo2Off) >= 17)

continue;

memcpy(aszAddr, pszLo2Off, (pszTerm - pszLo2Off));

rc = RTNetStrToIPv4Addr(aszAddr, &ip4addr);

if (RT_FAILURE(rc))

continue;

u32Off = RTStrToUInt32(pszTerm + 1);

if (u32Off == 0)

continue;

m_lo2off[j].addr.sin_addr.s_addr = ip4addr.u;

m_lo2off[j].off = u32Off;

++j;

}

}

#endif

} /* if (!fDontLoadRulesOnStartup) */

hrc = virtualbox->COMGETTER(HomeFolder)(bstr.asOutParam());

AssertComRCReturn(hrc, VERR_NOT_FOUND);

if (!bstr.isEmpty())

{

com::Utf8Str strTftpRoot(com::Utf8StrFmt("%ls%c%s",

bstr.raw(), RTPATH_DELIMITER, "TFTP"));

char *pszStrTemp; // avoid const char ** vs char **

rc = RTStrUtf8ToCurrentCP(&pszStrTemp, strTftpRoot.c_str());

AssertRC(rc);

m_ProxyOptions.tftp_root = pszStrTemp;

}

/* end of COM initialization */

rc = RTSemEventCreate(&hSemSVC);

AssertRCReturn(rc, rc);

rc = RTReqQueueCreate(&hReqIntNet);

AssertRCReturn(rc, rc);

g_pLwipNat->tryGoOnline();

vboxLwipCoreInitialize(VBoxNetLwipNAT::onLwipTcpIpInit, this);

rc = RTThreadCreate(&g_pLwipNat->hThrIntNetRecv, /* thread handle*/

VBoxNetLwipNAT::intNetThreadRecv, /* routine */

NULL, /* user data */

128 * _1K, /* stack size */

RTTHREADTYPE_IO, /* type */

0, /* flags, @todo: waitable ?*/

"INTNET-RECV");

AssertRCReturn(rc,rc);

LogFlowFuncLeaveRC(rc);

return rc;

}

int VBoxNetLwipNAT::parseOpt(int rc, const RTGETOPTUNION& Val)

{

switch (rc)

{

case 'p':

case 'P':

{

NATSEVICEPORTFORWARDRULE Rule;

VECNATSERVICEPF& rules = (rc == 'P'?

m_vecPortForwardRule6

: m_vecPortForwardRule4);

fDontLoadRulesOnStartup = true;

RT_ZERO(Rule);

int irc = netPfStrToPf(Val.psz, (rc == 'P'), &Rule.Pfr);

rules.push_back(Rule);

return VINF_SUCCESS;

}

default:;

}

return VERR_NOT_FOUND;

}

int VBoxNetLwipNAT::run()

{

while(true)

{

RTSemEventWait(g_pLwipNat->hSemSVC, RT_INDEFINITE_WAIT);

}

vboxLwipCoreFinalize(VBoxNetLwipNAT::onLwipTcpIpFini, this);

/* @todo: clean up of port-forward rules */

m_vecPortForwardRule4.clear();

m_vecPortForwardRule6.clear();

return VINF_SUCCESS;

}

extern "C" DECLEXPORT(int) TrustedMain(int argc, char **argv, char **envp)

{

LogFlowFuncEnter();

#ifdef RT_OS_WINDOWS

WSADATA wsaData;

int err;

err = WSAStartup(MAKEWORD(2,2), &wsaData);

if (err)

{

fprintf(stderr, "wsastartup: failed (%d)\n", err);

return 1;

}

#endif

HRESULT hrc = com::Initialize();

#ifdef VBOX_WITH_XPCOM

if (hrc == NS_ERROR_FILE_ACCESS_DENIED)

{

char szHome[RTPATH_MAX] = "";

com::GetVBoxUserHomeDirectory(szHome, sizeof(szHome));

return RTMsgErrorExit(RTEXITCODE_FAILURE,

"Failed to initialize COM because the global settings directory '%s' is not accessible!", szHome);

}

#endif

if (FAILED(hrc))

return RTMsgErrorExit(RTEXITCODE_FAILURE, "Failed to initialize COM!");

g_pLwipNat = new VBoxNetLwipNAT();

Log2(("NAT: initialization\n"));

int rc = g_pLwipNat->parseArgs(argc - 1, argv + 1);

AssertRC(rc);

if (!rc)

{

g_pLwipNat->init();

g_pLwipNat->run();

}

delete g_pLwipNat;

return 0;

}

#ifndef VBOX_WITH_HARDENING

int main(int argc, char **argv, char **envp)

{

int rc = RTR3InitExe(argc, &argv, RTR3INIT_FLAGS_SUPLIB);

if (RT_FAILURE(rc))

return RTMsgInitFailure(rc);

return TrustedMain(argc, argv, envp);

}

# if defined(RT_OS_WINDOWS)

static LRESULT CALLBACK WindowProc(HWND hwnd,

UINT uMsg,

WPARAM wParam,

LPARAM lParam

)

{

if(uMsg == WM_DESTROY)

{

PostQuitMessage(0);

return 0;

}

return DefWindowProc (hwnd, uMsg, wParam, lParam);

}

static LPCWSTR g_WndClassName = L"VBoxNetNatLwipClass";

static DWORD WINAPI MsgThreadProc(__in LPVOID lpParameter)

{

HWND hwnd = 0;

HINSTANCE hInstance = (HINSTANCE)GetModuleHandle (NULL);

bool bExit = false;

/* Register the Window Class. */

WNDCLASS wc;

wc.style = 0;

wc.lpfnWndProc = WindowProc;

wc.cbClsExtra = 0;

wc.cbWndExtra = sizeof(void *);

wc.hInstance = hInstance;

wc.hIcon = NULL;

wc.hCursor = NULL;

wc.hbrBackground = (HBRUSH)(COLOR_BACKGROUND + 1);

wc.lpszMenuName = NULL;

wc.lpszClassName = g_WndClassName;

ATOM atomWindowClass = RegisterClass(&wc);

if (atomWindowClass != 0)

{

/* Create the window. */

hwnd = CreateWindowEx (WS_EX_TOOLWINDOW | WS_EX_TRANSPARENT | WS_EX_TOPMOST,

g_WndClassName, g_WndClassName,

WS_POPUPWINDOW,

-200, -200, 100, 100, NULL, NULL, hInstance, NULL);

if (hwnd)

{

SetWindowPos(hwnd, HWND_TOPMOST, -200, -200, 0, 0,

SWP_NOACTIVATE | SWP_HIDEWINDOW | SWP_NOCOPYBITS | SWP_NOREDRAW | SWP_NOSIZE);

MSG msg;

while (GetMessage(&msg, NULL, 0, 0))

{

TranslateMessage(&msg);

DispatchMessage(&msg);

}

DestroyWindow (hwnd);

bExit = true;

}

UnregisterClass (g_WndClassName, hInstance);

}

if(bExit)

{

/* no need any accuracy here, in anyway the DHCP server usually gets terminated with TerminateProcess */

exit(0);

}

return 0;

}

int WINAPI WinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, LPSTR lpCmdLine, int nCmdShow)

{

#if 0

NOREF(hInstance); NOREF(hPrevInstance); NOREF(lpCmdLine); NOREF(nCmdShow);

HANDLE hThread = CreateThread(

NULL, /*__in_opt LPSECURITY_ATTRIBUTES lpThreadAttributes, */

0, /*__in SIZE_T dwStackSize, */

MsgThreadProc, /*__in LPTHREAD_START_ROUTINE lpStartAddress,*/

NULL, /*__in_opt LPVOID lpParameter,*/

0, /*__in DWORD dwCreationFlags,*/

NULL /*__out_opt LPDWORD lpThreadId*/

);

if(hThread != NULL)

CloseHandle(hThread);

#endif

return main(__argc, __argv, environ);

}

# endif /* RT_OS_WINDOWS */

#endif /* !VBOX_WITH_HARDENING */