VirtualBox

source: vbox/trunk/src/VBox/Devices/GIMDev/DrvUDP.cpp@ 82781

Last change on this file since 82781 was 76553, checked in by vboxsync, 5 years ago

scm --update-copyright-year
  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 8.6 KB
Line 
1/* $Id: DrvUDP.cpp 76553 2019-01-01 01:45:53Z vboxsync $ */
2/** @file
3 * UDP socket stream driver.
4 */
5
6/*
7 * Copyright (C) 2015-2019 Oracle Corporation
8 *
9 * This file is part of VirtualBox Open Source Edition (OSE), as
10 * available from http://www.virtualbox.org. This file is free software;
11 * you can redistribute it and/or modify it under the terms of the GNU
12 * General Public License (GPL) as published by the Free Software
13 * Foundation, in version 2 as it comes in the "COPYING" file of the
14 * VirtualBox OSE distribution. VirtualBox OSE is distributed in the
15 * hope that it will be useful, but WITHOUT ANY WARRANTY of any kind.
16 */
17
18
19/*********************************************************************************************************************************
20* Header Files *
21*********************************************************************************************************************************/
22#define LOG_GROUP LOG_GROUP_DRV_UDP
23#include <VBox/log.h>
24#include <VBox/vmm/pdmdrv.h>
25
26#include "VBoxDD.h"
27
28#include <iprt/socket.h>
29#include <iprt/udp.h>
30#include <iprt/uuid.h>
31
32
33/*********************************************************************************************************************************
34* Defined Constants And Macros *
35*********************************************************************************************************************************/
36/** Converts a pointer to DRVUDP::IStream to a PDRVUDP. */
37#define PDMISTREAM_2_DRVUDP(pInterface) ( (PDRVUDP)((uintptr_t)pInterface - RT_UOFFSETOF(DRVUDP, IStream)) )
38
39
40/*********************************************************************************************************************************
41* Structures and Typedefs *
42*********************************************************************************************************************************/
43/**
44 * UDP driver instance data.
45 *
46 * @implements PDMISTREAM
47 */
48typedef struct DRVUDP
49{
50 /** The stream interface. */
51 PDMISTREAM IStream;
52 /** Pointer to the driver instance. */
53 PPDMDRVINS pDrvIns;
54 /** The server port. */
55 uint16_t uServerPort;
56 /** The server address. */
57 char *pszServerAddress;
58 /** The resolved server address struct. */
59 RTNETADDR ServerAddr;
60 /** The UDP socket. */
61 RTSOCKET hSocket;
62} DRVUDP, *PDRVUDP;
63
64
65/*********************************************************************************************************************************
66* Internal Functions *
67*********************************************************************************************************************************/
68
69
70/** @interface_method_impl{PDMISTREAM,pfnRead} */
71static DECLCALLBACK(int) drvUDPRead(PPDMISTREAM pInterface, void *pvBuf, size_t *pcbRead)
72{
73 int rc = VINF_SUCCESS;
74 PDRVUDP pThis = PDMISTREAM_2_DRVUDP(pInterface);
75 LogFlowFunc(("pvBuf=%p *pcbRead=%#x (%s:%u)\n", pvBuf, *pcbRead, pThis->pszServerAddress, pThis->uServerPort));
76
77 Assert(pvBuf);
78 Assert(pcbRead);
79 if (pThis->hSocket != NIL_RTSOCKET)
80 {
81 size_t cbReallyRead = 0;
82 rc = RTSocketRead(pThis->hSocket, pvBuf, *pcbRead, &cbReallyRead);
83 if (RT_SUCCESS(rc))
84 *pcbRead = cbReallyRead;
85 }
86 else
87 rc = VERR_NET_NOT_SOCKET;
88
89 LogFlowFunc(("*pcbRead=%zu returns %Rrc\n", *pcbRead, rc));
90 return rc;
91}
92
93
94/** @interface_method_impl{PDMISTREAM,pfnWrite} */
95static DECLCALLBACK(int) drvUDPWrite(PPDMISTREAM pInterface, const void *pvBuf, size_t *pcbWrite)
96{
97 int rc = VINF_SUCCESS;
98 PDRVUDP pThis = PDMISTREAM_2_DRVUDP(pInterface);
99 LogFlowFunc(("pvBuf=%p *pcbWrite=%#x (%s:%u)\n", pvBuf, *pcbWrite, pThis->pszServerAddress, pThis->uServerPort));
100
101 Assert(pvBuf);
102 Assert(pcbWrite);
103 if (pThis->hSocket != NIL_RTSOCKET)
104 {
105 size_t cbBuf = *pcbWrite;
106 rc = RTSocketWriteTo(pThis->hSocket, pvBuf, cbBuf, NULL /*pDstAddr*/);
107 if (RT_SUCCESS(rc))
108 *pcbWrite = cbBuf;
109 }
110 else
111 rc = VERR_NET_NOT_SOCKET;
112
113 LogFlowFunc(("*pcbWrite=%zu returns %Rrc\n", *pcbWrite, rc));
114 return rc;
115}
116
117
118/**
119 * @interface_method_impl{PDMIBASE,pfnQueryInterface}
120 */
121static DECLCALLBACK(void *) drvUDPQueryInterface(PPDMIBASE pInterface, const char *pszIID)
122{
123 PPDMDRVINS pDrvIns = PDMIBASE_2_PDMDRV(pInterface);
124 PDRVUDP pThis = PDMINS_2_DATA(pDrvIns, PDRVUDP);
125 PDMIBASE_RETURN_INTERFACE(pszIID, PDMIBASE, &pDrvIns->IBase);
126 PDMIBASE_RETURN_INTERFACE(pszIID, PDMISTREAM, &pThis->IStream);
127 return NULL;
128}
129
130
131/* -=-=-=-=- PDMDRVREG -=-=-=-=- */
132
133/**
134 * Destruct a UDP socket stream driver instance.
135 *
136 * Most VM resources are freed by the VM. This callback is provided so that
137 * any non-VM resources can be freed correctly.
138 *
139 * @param pDrvIns The driver instance data.
140 */
141static DECLCALLBACK(void) drvUDPDestruct(PPDMDRVINS pDrvIns)
142{
143 PDRVUDP pThis = PDMINS_2_DATA(pDrvIns, PDRVUDP);
144 LogFlowFunc(("\n"));
145 PDMDRV_CHECK_VERSIONS_RETURN_VOID(pDrvIns);
146
147 if (pThis->hSocket != NIL_RTSOCKET)
148 {
149 /*
150 * We shutdown the socket here to poke out any blocking socket reads. The caller
151 * on the other thread/s need to ensure that they do -not- invoke drvUDPRead()
152 * or drvUDPWrite() after this.
153 */
154 RTSocketRetain(pThis->hSocket);
155 RTSocketShutdown(pThis->hSocket, true, true);
156 RTSocketClose(pThis->hSocket);
157 pThis->hSocket = NIL_RTSOCKET;
158 LogRel(("DrvUDP#%u: Closed socket to %s:%u\n", pThis->pDrvIns->iInstance, pThis->pszServerAddress, pThis->uServerPort));
159 }
160
161 MMR3HeapFree(pThis->pszServerAddress);
162 pThis->pszServerAddress = NULL;
163}
164
165
166/**
167 * Construct a UDP socket stream driver instance.
168 *
169 * @copydoc FNPDMDRVCONSTRUCT
170 */
171static DECLCALLBACK(int) drvUDPConstruct(PPDMDRVINS pDrvIns, PCFGMNODE pCfg, uint32_t fFlags)
172{
173 RT_NOREF1(fFlags);
174 PDRVUDP pThis = PDMINS_2_DATA(pDrvIns, PDRVUDP);
175 PDMDRV_CHECK_VERSIONS_RETURN(pDrvIns);
176
177 /*
178 * Init the static parts.
179 */
180 pThis->pDrvIns = pDrvIns;
181 /* IBase */
182 pDrvIns->IBase.pfnQueryInterface = drvUDPQueryInterface;
183 /* IStream */
184 pThis->IStream.pfnRead = drvUDPRead;
185 pThis->IStream.pfnWrite = drvUDPWrite;
186
187 /*
188 * Validate and read the configuration.
189 */
190 PDMDRV_VALIDATE_CONFIG_RETURN(pDrvIns, "ServerAddress|ServerPort", "");
191
192 int rc = CFGMR3QueryStringAlloc(pCfg, "ServerAddress", &pThis->pszServerAddress);
193 if (RT_FAILURE(rc))
194 return PDMDrvHlpVMSetError(pDrvIns, rc, RT_SRC_POS,
195 N_("Configuration error: querying \"ServerAddress\" resulted in %Rrc"), rc);
196 rc = CFGMR3QueryU16(pCfg, "ServerPort", &pThis->uServerPort);
197 if (RT_FAILURE(rc))
198 return PDMDrvHlpVMSetError(pDrvIns, rc, RT_SRC_POS,
199 N_("Configuration error: querying \"ServerPort\" resulted in %Rrc"), rc);
200
201 /*
202 * Create the socket and connect.
203 */
204 rc = RTUdpCreateClientSocket(pThis->pszServerAddress, pThis->uServerPort, NULL, &pThis->hSocket);
205 if (RT_SUCCESS(rc))
206 LogRel(("DrvUDP#%u: Connected socket to %s:%u\n",
207 pThis->pDrvIns->iInstance, pThis->pszServerAddress, pThis->uServerPort));
208 else
209 LogRel(("DrvUDP#%u: Failed to create/connect socket to %s:%u rc=%Rrc\n",
210 pThis->pDrvIns->iInstance, pThis->pszServerAddress, pThis->uServerPort, rc));
211 return VINF_SUCCESS;
212}
213
214
215/**
216 * UDP socket driver registration record.
217 */
218const PDMDRVREG g_DrvUDP =
219{
220 /* u32Version */
221 PDM_DRVREG_VERSION,
222 /* szName */
223 "UDP",
224 /* szRCMod */
225 "",
226 /* szR0Mod */
227 "",
228 /* pszDescription */
229 "UDP socket stream driver.",
230 /* fFlags */
231 PDM_DRVREG_FLAGS_HOST_BITS_DEFAULT,
232 /* fClass. */
233 PDM_DRVREG_CLASS_STREAM,
234 /* cMaxInstances */
235 ~0U,
236 /* cbInstance */
237 sizeof(DRVUDP),
238 /* pfnConstruct */
239 drvUDPConstruct,
240 /* pfnDestruct */
241 drvUDPDestruct,
242 /* pfnRelocate */
243 NULL,
244 /* pfnIOCtl */
245 NULL,
246 /* pfnPowerOn */
247 NULL,
248 /* pfnReset */
249 NULL,
250 /* pfnSuspend */
251 NULL,
252 /* pfnResume */
253 NULL,
254 /* pfnAttach */
255 NULL,
256 /* pfnDetach */
257 NULL,
258 /* pfnPowerOff */
259 NULL,
260 /* pfnSoftReset */
261 NULL,
262 /* u32EndVersion */
263 PDM_DRVREG_VERSION
264};
265
Note: See TracBrowser for help on using the repository browser.

© 2023 Oracle
Contact – Privacy policy – Terms of Use