mirror of
https://github.com/dolphin-emu/dolphin.git
synced 2025-07-21 13:20:27 -06:00
Removed parts of PortableSockets that we do not use. If you need more functionality later, resurrect it from old SVN revisions or re-download the source (see top of the header). However, DO NOT use performance counters to do timing, since the counter value can be different for different CPU cores.
git-svn-id: https://dolphin-emu.googlecode.com/svn/trunk@428 8ced0084-cf51-0410-be5f-012b33b47a6e
This commit is contained in:
Maarten ter Huurne
@ -1,550 +0,0 @@
|
|||||||
// Under MIT licence from http://www.mindcontrol.org/~hplus/http-get.html
|
|
||||||
|
|
||||||
#include "PortableSockets.h"
|
|
||||||
|
|
||||||
#if NEED_GETTIMEOFDAY
|
|
||||||
#if defined( _WIN32 )
|
|
||||||
|
|
||||||
#include <sys/types.h>
|
|
||||||
#include <sys/timeb.h>
|
|
||||||
#include <math.h>
|
|
||||||
#include <assert.h>
|
|
||||||
#include <mmsystem.h>
|
|
||||||
|
|
||||||
#pragma comment( lib, "winmm.lib" )
|
|
||||||
//#pragma comment( lib, "ws2_32.lib" )
|
|
||||||
|
|
||||||
|
|
||||||
namespace {
|
|
||||||
|
|
||||||
// This class could be made cheaper using RDTSC for short-term
|
|
||||||
// measurement. But, whatever.
|
|
||||||
class init_gettimeofday
|
|
||||||
{
|
|
||||||
public:
|
|
||||||
init_gettimeofday()
|
|
||||||
{
|
|
||||||
timeBeginPeriod( 2 );
|
|
||||||
__int64 rr;
|
|
||||||
QueryPerformanceFrequency( (LARGE_INTEGER *)&rr );
|
|
||||||
ticksPerSecInv_ = 1.0 / (double)((DWORD)rr & 0xffffffff);
|
|
||||||
int watchdog = 0;
|
|
||||||
again:
|
|
||||||
lastTicks_ = timeGetTime();
|
|
||||||
QueryPerformanceCounter( (LARGE_INTEGER *)&lastRead_ );
|
|
||||||
timeb tb;
|
|
||||||
ftime( &tb );
|
|
||||||
timeOffset_ = tb.time + tb.millitm * 0.001 - lastRead_ * ticksPerSecInv_;
|
|
||||||
lastTime_ = timeOffset_;
|
|
||||||
// make sure it didn't take too long
|
|
||||||
if( watchdog++ < 10 && (timeGetTime() != lastTicks_) ) {
|
|
||||||
goto again;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
~init_gettimeofday()
|
|
||||||
{
|
|
||||||
timeEndPeriod( 2 );
|
|
||||||
}
|
|
||||||
|
|
||||||
void get( timeval * tv )
|
|
||||||
{
|
|
||||||
__int64 nu;
|
|
||||||
int watchdog = 0;
|
|
||||||
again:
|
|
||||||
DWORD m = timeGetTime();
|
|
||||||
QueryPerformanceCounter( (LARGE_INTEGER *)&nu );
|
|
||||||
DWORD n = timeGetTime();
|
|
||||||
// guard against pre-emption
|
|
||||||
if( (watchdog++ < 10) && (n != m) ) {
|
|
||||||
goto again;
|
|
||||||
}
|
|
||||||
double nuTime = nu * ticksPerSecInv_ + timeOffset_;
|
|
||||||
if( (nu - lastRead_) & 0x7fffffff80000000ULL ) {
|
|
||||||
// there's a chance that we're seeing a jump-ahead
|
|
||||||
double adjust = (nuTime - lastTime_ - (n - lastTicks_) * 0.001);
|
|
||||||
if( adjust > 0.1f ) {
|
|
||||||
timeOffset_ -= adjust;
|
|
||||||
nuTime -= adjust;
|
|
||||||
assert( nuTime >= lastTime_ );
|
|
||||||
}
|
|
||||||
}
|
|
||||||
lastRead_ = nu;
|
|
||||||
lastTicks_ = n;
|
|
||||||
lastTime_ = nuTime;
|
|
||||||
tv->tv_sec = (ulong)floor( nuTime );
|
|
||||||
tv->tv_usec = (ulong)(1000000 * (nuTime - tv->tv_sec));
|
|
||||||
}
|
|
||||||
|
|
||||||
double ticksPerSecInv_;
|
|
||||||
double timeOffset_;
|
|
||||||
double lastTime_;
|
|
||||||
__int64 lastRead_;
|
|
||||||
DWORD lastTicks_;
|
|
||||||
};
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
void gettimeofday( timeval * tv, int )
|
|
||||||
{
|
|
||||||
static init_gettimeofday data;
|
|
||||||
data.get( tv );
|
|
||||||
}
|
|
||||||
|
|
||||||
#else
|
|
||||||
#error "don't know how to do this"
|
|
||||||
#endif
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if NEED_WINDOWS_POLL
|
|
||||||
#if defined( WIN32 )
|
|
||||||
|
|
||||||
#include <assert.h>
|
|
||||||
#include <winsock2.h>
|
|
||||||
#include <windows.h>
|
|
||||||
|
|
||||||
// This is somewhat less than ideal -- better would be if we could
|
|
||||||
// abstract pollfd enough that it's non-copying on Windows.
|
|
||||||
int poll( pollfd * iofds, size_t count, int ms )
|
|
||||||
{
|
|
||||||
FD_SET rd, wr, ex;
|
|
||||||
FD_ZERO( &rd );
|
|
||||||
FD_ZERO( &wr );
|
|
||||||
FD_ZERO( &ex );
|
|
||||||
SOCKET m = 0;
|
|
||||||
for( size_t ix = 0; ix < count; ++ix ) {
|
|
||||||
iofds[ix].revents = 0;
|
|
||||||
if( iofds[ix].fd >= m ) {
|
|
||||||
m = iofds[ix].fd + 1;
|
|
||||||
}
|
|
||||||
if( iofds[ix].events & (POLLIN | POLLPRI) ) {
|
|
||||||
assert( rd.fd_count < FD_SETSIZE );
|
|
||||||
rd.fd_array[ rd.fd_count++ ] = iofds[ix].fd;
|
|
||||||
}
|
|
||||||
if( iofds[ix].events & (POLLOUT) ) {
|
|
||||||
assert( wr.fd_count < FD_SETSIZE );
|
|
||||||
wr.fd_array[ wr.fd_count++ ] = iofds[ix].fd;
|
|
||||||
}
|
|
||||||
assert( ex.fd_count < FD_SETSIZE );
|
|
||||||
ex.fd_array[ ex.fd_count++ ] = iofds[ix].fd;
|
|
||||||
}
|
|
||||||
timeval tv;
|
|
||||||
tv.tv_sec = ms/1000;
|
|
||||||
tv.tv_usec = (ms - (tv.tv_sec * 1000)) * 1000;
|
|
||||||
int r = 0;
|
|
||||||
if( m == 0 ) {
|
|
||||||
::Sleep( ms );
|
|
||||||
}
|
|
||||||
else {
|
|
||||||
r = ::select( (int)m, (rd.fd_count ? &rd : 0), (wr.fd_count ? &wr : 0), (ex.fd_count ? &ex : 0), &tv );
|
|
||||||
}
|
|
||||||
if( r < 0 ) {
|
|
||||||
int err = WSAGetLastError();
|
|
||||||
errno = err;
|
|
||||||
return r;
|
|
||||||
}
|
|
||||||
r = 0;
|
|
||||||
for( size_t ix = 0; ix < count; ++ix ) {
|
|
||||||
for( size_t iy = 0; iy < rd.fd_count; ++iy ) {
|
|
||||||
if( rd.fd_array[ iy ] == iofds[ix].fd ) {
|
|
||||||
iofds[ix].revents |= POLLIN;
|
|
||||||
++r;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for( size_t iy = 0; iy < wr.fd_count; ++iy ) {
|
|
||||||
if( wr.fd_array[ iy ] == iofds[ix].fd ) {
|
|
||||||
iofds[ix].revents |= POLLOUT;
|
|
||||||
++r;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
for( size_t iy = 0; iy < ex.fd_count; ++iy ) {
|
|
||||||
if( ex.fd_array[ iy ] == iofds[ix].fd ) {
|
|
||||||
iofds[ix].revents |= POLLERR;
|
|
||||||
++r;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return r;
|
|
||||||
}
|
|
||||||
|
|
||||||
#else
|
|
||||||
#error "don't know how to do this"
|
|
||||||
#endif
|
|
||||||
#endif
|
|
||||||
|
|
||||||
|
|
||||||
#if NEED_FIREWALL_ENABLE
|
|
||||||
#if defined( WIN32 )
|
|
||||||
|
|
||||||
#undef _WIN32_WINNT
|
|
||||||
#define _WIN32_WINNT 0x500
|
|
||||||
|
|
||||||
#include <objbase.h>
|
|
||||||
#include <oleauto.h>
|
|
||||||
//#include <netfw.h>
|
|
||||||
|
|
||||||
/*
|
|
||||||
#define _ASSERT assert
|
|
||||||
|
|
||||||
namespace {
|
|
||||||
|
|
||||||
HRESULT WindowsFirewallInitialize(OUT INetFwProfile** fwProfile)
|
|
||||||
{
|
|
||||||
HRESULT hr = S_OK;
|
|
||||||
INetFwMgr* fwMgr = NULL;
|
|
||||||
INetFwPolicy* fwPolicy = NULL;
|
|
||||||
|
|
||||||
_ASSERT(fwProfile != NULL);
|
|
||||||
|
|
||||||
*fwProfile = NULL;
|
|
||||||
|
|
||||||
// Create an instance of the firewall settings manager.
|
|
||||||
hr = CoCreateInstance(
|
|
||||||
__uuidof(NetFwMgr),
|
|
||||||
NULL,
|
|
||||||
CLSCTX_INPROC_SERVER,
|
|
||||||
__uuidof(INetFwMgr),
|
|
||||||
(void**)&fwMgr
|
|
||||||
);
|
|
||||||
if (FAILED(hr))
|
|
||||||
{
|
|
||||||
printf("CoCreateInstance failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Retrieve the local firewall policy.
|
|
||||||
hr = fwMgr->get_LocalPolicy(&fwPolicy);
|
|
||||||
if (FAILED(hr))
|
|
||||||
{
|
|
||||||
printf("get_LocalPolicy failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Retrieve the firewall profile currently in effect.
|
|
||||||
hr = fwPolicy->get_CurrentProfile(fwProfile);
|
|
||||||
if (FAILED(hr))
|
|
||||||
{
|
|
||||||
printf("get_CurrentProfile failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
error:
|
|
||||||
|
|
||||||
// Release the local firewall policy.
|
|
||||||
if (fwPolicy != NULL)
|
|
||||||
{
|
|
||||||
fwPolicy->Release();
|
|
||||||
}
|
|
||||||
|
|
||||||
// Release the firewall settings manager.
|
|
||||||
if (fwMgr != NULL)
|
|
||||||
{
|
|
||||||
fwMgr->Release();
|
|
||||||
}
|
|
||||||
|
|
||||||
return hr;
|
|
||||||
}
|
|
||||||
|
|
||||||
void WindowsFirewallCleanup(IN INetFwProfile* fwProfile)
|
|
||||||
{
|
|
||||||
// Release the firewall profile.
|
|
||||||
if (fwProfile != NULL)
|
|
||||||
{
|
|
||||||
fwProfile->Release();
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
HRESULT WindowsFirewallAppIsEnabled(
|
|
||||||
IN INetFwProfile* fwProfile,
|
|
||||||
IN const wchar_t* fwProcessImageFileName,
|
|
||||||
OUT BOOL* fwAppEnabled
|
|
||||||
)
|
|
||||||
{
|
|
||||||
HRESULT hr = S_OK;
|
|
||||||
BSTR fwBstrProcessImageFileName = NULL;
|
|
||||||
VARIANT_BOOL fwEnabled;
|
|
||||||
INetFwAuthorizedApplication* fwApp = NULL;
|
|
||||||
INetFwAuthorizedApplications* fwApps = NULL;
|
|
||||||
|
|
||||||
_ASSERT(fwProfile != NULL);
|
|
||||||
_ASSERT(fwProcessImageFileName != NULL);
|
|
||||||
_ASSERT(fwAppEnabled != NULL);
|
|
||||||
|
|
||||||
*fwAppEnabled = FALSE;
|
|
||||||
|
|
||||||
// Retrieve the authorized application collection.
|
|
||||||
hr = fwProfile->get_AuthorizedApplications(&fwApps);
|
|
||||||
if (FAILED(hr))
|
|
||||||
{
|
|
||||||
printf("get_AuthorizedApplications failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Allocate a BSTR for the process image file name.
|
|
||||||
fwBstrProcessImageFileName = SysAllocString(fwProcessImageFileName);
|
|
||||||
if (SysStringLen(fwBstrProcessImageFileName) == 0)
|
|
||||||
{
|
|
||||||
hr = E_OUTOFMEMORY;
|
|
||||||
printf("SysAllocString failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Attempt to retrieve the authorized application.
|
|
||||||
hr = fwApps->Item(fwBstrProcessImageFileName, &fwApp);
|
|
||||||
if (SUCCEEDED(hr))
|
|
||||||
{
|
|
||||||
// Find out if the authorized application is enabled.
|
|
||||||
hr = fwApp->get_Enabled(&fwEnabled);
|
|
||||||
if (FAILED(hr))
|
|
||||||
{
|
|
||||||
printf("get_Enabled failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (fwEnabled != VARIANT_FALSE)
|
|
||||||
{
|
|
||||||
// The authorized application is enabled.
|
|
||||||
*fwAppEnabled = TRUE;
|
|
||||||
|
|
||||||
printf(
|
|
||||||
"Authorized application %lS is enabled in the firewall.\n",
|
|
||||||
fwProcessImageFileName
|
|
||||||
);
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
printf(
|
|
||||||
"Authorized application %lS is disabled in the firewall.\n",
|
|
||||||
fwProcessImageFileName
|
|
||||||
);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
else
|
|
||||||
{
|
|
||||||
// The authorized application was not in the collection.
|
|
||||||
hr = S_OK;
|
|
||||||
|
|
||||||
printf(
|
|
||||||
"Authorized application %lS is disabled in the firewall.\n",
|
|
||||||
fwProcessImageFileName
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
error:
|
|
||||||
|
|
||||||
// Free the BSTR.
|
|
||||||
SysFreeString(fwBstrProcessImageFileName);
|
|
||||||
|
|
||||||
// Release the authorized application instance.
|
|
||||||
if (fwApp != NULL)
|
|
||||||
{
|
|
||||||
fwApp->Release();
|
|
||||||
}
|
|
||||||
|
|
||||||
// Release the authorized application collection.
|
|
||||||
if (fwApps != NULL)
|
|
||||||
{
|
|
||||||
fwApps->Release();
|
|
||||||
}
|
|
||||||
|
|
||||||
return hr;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
HRESULT WindowsFirewallAddApp(
|
|
||||||
IN INetFwProfile* fwProfile,
|
|
||||||
IN const wchar_t* fwProcessImageFileName,
|
|
||||||
IN const wchar_t* fwName
|
|
||||||
)
|
|
||||||
{
|
|
||||||
HRESULT hr = S_OK;
|
|
||||||
BOOL fwAppEnabled;
|
|
||||||
BSTR fwBstrName = NULL;
|
|
||||||
BSTR fwBstrProcessImageFileName = NULL;
|
|
||||||
INetFwAuthorizedApplication* fwApp = NULL;
|
|
||||||
INetFwAuthorizedApplications* fwApps = NULL;
|
|
||||||
|
|
||||||
_ASSERT(fwProfile != NULL);
|
|
||||||
_ASSERT(fwProcessImageFileName != NULL);
|
|
||||||
_ASSERT(fwName != NULL);
|
|
||||||
|
|
||||||
// First check to see if the application is already authorized.
|
|
||||||
hr = WindowsFirewallAppIsEnabled(
|
|
||||||
fwProfile,
|
|
||||||
fwProcessImageFileName,
|
|
||||||
&fwAppEnabled
|
|
||||||
);
|
|
||||||
if (FAILED(hr))
|
|
||||||
{
|
|
||||||
printf("WindowsFirewallAppIsEnabled failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Only add the application if it isn't already authorized.
|
|
||||||
if (!fwAppEnabled)
|
|
||||||
{
|
|
||||||
// Retrieve the authorized application collection.
|
|
||||||
hr = fwProfile->get_AuthorizedApplications(&fwApps);
|
|
||||||
if (FAILED(hr))
|
|
||||||
{
|
|
||||||
printf("get_AuthorizedApplications failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Create an instance of an authorized application.
|
|
||||||
hr = CoCreateInstance(
|
|
||||||
__uuidof(NetFwAuthorizedApplication),
|
|
||||||
NULL,
|
|
||||||
CLSCTX_INPROC_SERVER,
|
|
||||||
__uuidof(INetFwAuthorizedApplication),
|
|
||||||
(void**)&fwApp
|
|
||||||
);
|
|
||||||
if (FAILED(hr))
|
|
||||||
{
|
|
||||||
printf("CoCreateInstance failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Allocate a BSTR for the process image file name.
|
|
||||||
fwBstrProcessImageFileName = SysAllocString(fwProcessImageFileName);
|
|
||||||
if (SysStringLen(fwBstrProcessImageFileName) == 0)
|
|
||||||
{
|
|
||||||
hr = E_OUTOFMEMORY;
|
|
||||||
printf("SysAllocString failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Set the process image file name.
|
|
||||||
hr = fwApp->put_ProcessImageFileName(fwBstrProcessImageFileName);
|
|
||||||
if (FAILED(hr))
|
|
||||||
{
|
|
||||||
printf("put_ProcessImageFileName failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Allocate a BSTR for the application friendly name.
|
|
||||||
fwBstrName = SysAllocString(fwName);
|
|
||||||
if (SysStringLen(fwBstrName) == 0)
|
|
||||||
{
|
|
||||||
hr = E_OUTOFMEMORY;
|
|
||||||
printf("SysAllocString failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Set the application friendly name.
|
|
||||||
hr = fwApp->put_Name(fwBstrName);
|
|
||||||
if (FAILED(hr))
|
|
||||||
{
|
|
||||||
printf("put_Name failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add the application to the collection.
|
|
||||||
hr = fwApps->Add(fwApp);
|
|
||||||
if (FAILED(hr))
|
|
||||||
{
|
|
||||||
printf("Add failed: 0x%08lx\n", hr);
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
printf(
|
|
||||||
"Authorized application %lS is now enabled in the firewall.\n",
|
|
||||||
fwProcessImageFileName
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
error:
|
|
||||||
|
|
||||||
// Free the BSTRs.
|
|
||||||
SysFreeString(fwBstrName);
|
|
||||||
SysFreeString(fwBstrProcessImageFileName);
|
|
||||||
|
|
||||||
// Release the authorized application instance.
|
|
||||||
if (fwApp != NULL)
|
|
||||||
{
|
|
||||||
fwApp->Release();
|
|
||||||
}
|
|
||||||
|
|
||||||
// Release the authorized application collection.
|
|
||||||
if (fwApps != NULL)
|
|
||||||
{
|
|
||||||
fwApps->Release();
|
|
||||||
}
|
|
||||||
|
|
||||||
return hr;
|
|
||||||
}
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
bool ENABLE_FIREWALL()
|
|
||||||
{
|
|
||||||
BOOL couldEnable = false;
|
|
||||||
HRESULT hr = S_OK;
|
|
||||||
HRESULT comInit = E_FAIL;
|
|
||||||
INetFwProfile* fwProfile = NULL;
|
|
||||||
|
|
||||||
// Initialize COM.
|
|
||||||
#if 1
|
|
||||||
comInit = CoInitialize( 0 );
|
|
||||||
#else
|
|
||||||
comInit = CoInitializeEx(
|
|
||||||
0,
|
|
||||||
COINIT_APARTMENTTHREADED //| COINIT_DISABLE_OLE1DDE
|
|
||||||
);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
// Ignore RPC_E_CHANGED_MODE; this just means that COM has already been
|
|
||||||
// initialized with a different mode. Since we don't care what the mode is,
|
|
||||||
// we'll just use the existing mode.
|
|
||||||
if (comInit != RPC_E_CHANGED_MODE) {
|
|
||||||
hr = comInit;
|
|
||||||
if (FAILED(hr)) {
|
|
||||||
fprintf( stderr, "CoInitializeEx failed: 0x%08lx\n", hr );
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Retrieve the firewall profile currently in effect.
|
|
||||||
hr = WindowsFirewallInitialize(&fwProfile);
|
|
||||||
if (FAILED(hr)) {
|
|
||||||
fprintf( stderr, "WindowsFirewallInitialize failed: 0x%08lx\n", hr );
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
HMODULE hm = GetModuleHandle( 0 );
|
|
||||||
wchar_t path[512];
|
|
||||||
if( !GetModuleFileNameW( hm, path, sizeof(path)/sizeof(wchar_t) ) ) {
|
|
||||||
fprintf( stderr, "GetModuleFileName() failed: 0x%lx\n", GetLastError() );
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
// Add the application to the authorized application collection.
|
|
||||||
hr = WindowsFirewallAddApp(
|
|
||||||
fwProfile,
|
|
||||||
path,
|
|
||||||
L"Introduction Library User"
|
|
||||||
);
|
|
||||||
if (FAILED(hr)) {
|
|
||||||
fprintf( stderr, "WindowsFirewallAddApp failed: 0x%08lx\n", hr );
|
|
||||||
goto error;
|
|
||||||
}
|
|
||||||
|
|
||||||
error:
|
|
||||||
|
|
||||||
WindowsFirewallCleanup(fwProfile);
|
|
||||||
|
|
||||||
if (SUCCEEDED(comInit)) {
|
|
||||||
CoUninitialize();
|
|
||||||
}
|
|
||||||
|
|
||||||
return couldEnable != FALSE;
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
|
|
||||||
#else
|
|
||||||
#error "don't know how to do this"
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#endif
|
|
||||||
@ -1,4 +1,5 @@
|
|||||||
// Under MIT licence from http://www.mindcontrol.org/~hplus/http-get.html
|
// Under MIT licence from http://www.mindcontrol.org/~hplus/http-get.html
|
||||||
|
// We only use a subset of the definitions in the original source.
|
||||||
|
|
||||||
#if !defined( sock_port_h )
|
#if !defined( sock_port_h )
|
||||||
#define sock_port_h
|
#define sock_port_h
|
||||||
@ -6,184 +7,49 @@
|
|||||||
// There are differences between Linux/Berkeley sockets and WinSock2
|
// There are differences between Linux/Berkeley sockets and WinSock2
|
||||||
// This file wraps some of the more common ones (as used by this library).
|
// This file wraps some of the more common ones (as used by this library).
|
||||||
|
|
||||||
#include <stdarg.h>
|
#if defined(_WIN32)
|
||||||
|
|
||||||
#if defined( WIN32 )
|
|
||||||
|
|
||||||
// Windows features go here
|
// Windows features go here
|
||||||
#include <Winsock2.h>
|
|
||||||
#include <stdio.h>
|
|
||||||
|
|
||||||
#if !defined( NEED_SHORT_TYPES )
|
#include <Winsock2.h>
|
||||||
#define NEED_SHORT_TYPES 1
|
|
||||||
#endif
|
|
||||||
#if !defined( NEED_WINDOWS_POLL )
|
|
||||||
#define NEED_WINDOWS_POLL 1
|
|
||||||
#endif
|
|
||||||
#if !defined( NEED_GETTIMEOFDAY )
|
|
||||||
#define NEED_GETTIMEOFDAY 1
|
|
||||||
#endif
|
|
||||||
#if !defined( NEED_SNPRINTF )
|
|
||||||
#define NEED_SNPRINTF 1
|
|
||||||
#endif
|
|
||||||
#if !defined( NEED_HSTRERROR )
|
|
||||||
#define NEED_HSTRERROR 1
|
|
||||||
#endif
|
|
||||||
#if !defined( NEED_GETLASTERROR )
|
|
||||||
#define NEED_GETLASTERROR 1
|
|
||||||
#endif
|
|
||||||
#if !defined( NEED_FIREWALL_ENABLE )
|
|
||||||
#define NEED_FIREWALL_ENABLE 1
|
|
||||||
#endif
|
|
||||||
|
|
||||||
typedef int socklen_t;
|
typedef int socklen_t;
|
||||||
#define MSG_NOSIGNAL 0
|
|
||||||
#define MAKE_SOCKET_NONBLOCKING(x,r) \
|
#define MAKE_SOCKET_NONBLOCKING(x,r) \
|
||||||
do { u_long _x = 1; (r) = ioctlsocket( (x), FIONBIO, &_x ); } while(0)
|
do { u_long _x = 1; (r) = ioctlsocket( (x), FIONBIO, &_x ); } while(0)
|
||||||
#define NONBLOCK_MSG_SEND 0
|
#define NONBLOCK_MSG_SEND 0
|
||||||
#define INIT_SOCKET_LIBRARY() \
|
#define INIT_SOCKET_LIBRARY() \
|
||||||
do { WSADATA wsaData; WSAStartup( MAKEWORD(2,2), &wsaData ); } while(0)
|
do { WSADATA wsaData; WSAStartup( MAKEWORD(2,2), &wsaData ); } while(0)
|
||||||
|
|
||||||
#pragma warning( disable: 4250 )
|
|
||||||
|
|
||||||
#define SIN_ADDR_UINT(x) \
|
|
||||||
((uint&)(x).S_un.S_addr)
|
|
||||||
#define BAD_SOCKET_FD 0xffffffffU
|
#define BAD_SOCKET_FD 0xffffffffU
|
||||||
|
|
||||||
#else
|
#define SOCKET_ERRNO WSAGetLastError()
|
||||||
|
inline bool SOCKET_WOULDBLOCK_ERROR(int e) { return e == WSAEWOULDBLOCK; }
|
||||||
|
inline bool SOCKET_NEED_REOPEN(int e) { return e == WSAECONNRESET; }
|
||||||
|
|
||||||
|
#else
|
||||||
|
// POSIX features go here
|
||||||
|
|
||||||
// Linux features go here
|
|
||||||
#include <sys/types.h>
|
|
||||||
#include <sys/socket.h>
|
#include <sys/socket.h>
|
||||||
#include <netdb.h>
|
#include <netdb.h>
|
||||||
#include <sys/time.h>
|
#include <cerrno>
|
||||||
#include <sys/poll.h>
|
|
||||||
|
|
||||||
#if !defined( NEED_CLOSESOCKET )
|
#if !defined(SOCKET)
|
||||||
#define NEED_CLOSESOCKET 1
|
#define SOCKET int
|
||||||
#endif
|
|
||||||
#if !defined( NEED_IOCTLSOCKET )
|
|
||||||
#define NEED_IOCTLSOCKET 1
|
|
||||||
#endif
|
|
||||||
#if !defined( NEED_SHORT_TYPES )
|
|
||||||
#define NEED_SHORT_TYPES 1
|
|
||||||
#endif
|
|
||||||
#if !defined( NEED_ERRNO )
|
|
||||||
#define NEED_ERRNO 1
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
extern int h_errno;
|
// hack -- I don't make it non-blocking; instead, I pass
|
||||||
// hack -- I don't make it non-blocking; instead, I pass
|
|
||||||
// NONBLOCK_MSG_SEND for each call to sendto()
|
// NONBLOCK_MSG_SEND for each call to sendto()
|
||||||
#define MAKE_SOCKET_NONBLOCKING(x,r) do { (r) = 0; } while(0)
|
#define MAKE_SOCKET_NONBLOCKING(x,r) do { (r) = 0; } while(0)
|
||||||
#define NONBLOCK_MSG_SEND MSG_DONTWAIT
|
#define NONBLOCK_MSG_SEND MSG_DONTWAIT
|
||||||
#define INIT_SOCKET_LIBRARY() do {} while(0)
|
#define INIT_SOCKET_LIBRARY() do {} while(0)
|
||||||
|
|
||||||
#if !defined( SOCKET )
|
|
||||||
#define SOCKET int
|
|
||||||
#endif
|
|
||||||
#define SIN_ADDR_UINT(x) \
|
|
||||||
((uint&)(x).s_addr)
|
|
||||||
#define BAD_SOCKET_FD -1
|
#define BAD_SOCKET_FD -1
|
||||||
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if NEED_CLOSESOCKET
|
|
||||||
#define closesocket close
|
#define closesocket close
|
||||||
#endif
|
|
||||||
|
|
||||||
#if NEED_PROTOENT
|
|
||||||
struct protoent {
|
|
||||||
int p_proto;
|
|
||||||
};
|
|
||||||
#endif
|
|
||||||
#if NEED_GETPROTOBYNAME
|
|
||||||
inline protoent * getprotobyname( char const * s ) {
|
|
||||||
static protoent p;
|
|
||||||
if( !strcmp( s, "udp" ) ) {
|
|
||||||
p.p_proto = IPPROTO_UDP;
|
|
||||||
return &p;
|
|
||||||
}
|
|
||||||
if( !strcmp( s, "tcp" ) ) {
|
|
||||||
p.p_proto = IPPROTO_TCP;
|
|
||||||
return &p;
|
|
||||||
}
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if NEED_SHORT_TYPES
|
|
||||||
typedef unsigned long ulong;
|
|
||||||
typedef unsigned int uint;
|
|
||||||
typedef unsigned short ushort;
|
|
||||||
typedef unsigned char uchar;
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if NEED_WINDOWS_POLL
|
|
||||||
struct pollfd {
|
|
||||||
SOCKET fd;
|
|
||||||
unsigned short events;
|
|
||||||
unsigned short revents;
|
|
||||||
};
|
|
||||||
#define POLLIN 0x1
|
|
||||||
#define POLLPRI 0x2
|
|
||||||
#define POLLOUT 0x4
|
|
||||||
#define POLLERR 0x100
|
|
||||||
#define POLLHUP 0x200
|
|
||||||
#define POLLNVAL 0x8000
|
|
||||||
int poll( pollfd * iofds, size_t count, int ms );
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if NEED_GETTIMEOFDAY
|
|
||||||
// this is not thread safe!
|
|
||||||
void gettimeofday( timeval * otv, int );
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if NEED_SNPRINTF
|
|
||||||
inline int vsnprintf( char * buf, int size, char const * fmt, va_list vl ) {
|
|
||||||
int r = _vsnprintf( buf, size, fmt, vl );
|
|
||||||
if( r < 0 ) {
|
|
||||||
buf[size-1] = 0;
|
|
||||||
r = size-1;
|
|
||||||
}
|
|
||||||
return r;
|
|
||||||
}
|
|
||||||
inline int snprintf( char * buf, int size, char const * fmt, ... ) {
|
|
||||||
va_list vl;
|
|
||||||
va_start( vl, fmt );
|
|
||||||
int r = vsnprintf( buf, size, fmt, vl );
|
|
||||||
va_end( vl );
|
|
||||||
return r;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if NEED_HSTRERROR
|
|
||||||
// NOT thread safe!
|
|
||||||
inline char const * hstrerror( ulong ec ) {
|
|
||||||
static char err[ 128 ];
|
|
||||||
snprintf( err, 128, "host error 0x%lx", ec );
|
|
||||||
return err;
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if NEED_GETLASTERROR
|
|
||||||
#define SOCKET_ERRNO WSAGetLastError()
|
|
||||||
inline bool SOCKET_WOULDBLOCK_ERROR( int e ) { return e == WSAEWOULDBLOCK; }
|
|
||||||
inline bool SOCKET_NEED_REOPEN( int e ) { return e == WSAECONNRESET; }
|
|
||||||
#endif
|
|
||||||
|
|
||||||
#if NEED_ERRNO
|
|
||||||
#include <errno.h>
|
|
||||||
#define SOCKET_ERRNO errno
|
#define SOCKET_ERRNO errno
|
||||||
inline bool SOCKET_WOULDBLOCK_ERROR( int e ) { return e == EWOULDBLOCK; }
|
inline bool SOCKET_WOULDBLOCK_ERROR(int e) { return e == EWOULDBLOCK; }
|
||||||
inline bool SOCKET_NEED_REOPEN( int e ) { return false; }
|
inline bool SOCKET_NEED_REOPEN(int e) { return false; }
|
||||||
#endif
|
|
||||||
|
|
||||||
#if NEED_FIREWALL_ENABLE
|
|
||||||
extern bool ENABLE_FIREWALL();
|
|
||||||
#else
|
|
||||||
#define ENABLE_FIREWALL() true
|
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#endif // sock_port_h
|
#endif // sock_port_h
|
||||||
|
|
||||||
|
|||||||
@ -16,7 +16,6 @@ files = [
|
|||||||
"MemArena.cpp",
|
"MemArena.cpp",
|
||||||
"MemoryUtil.cpp",
|
"MemoryUtil.cpp",
|
||||||
"Plugin.cpp",
|
"Plugin.cpp",
|
||||||
"PortableSockets.cpp",
|
|
||||||
"StringUtil.cpp",
|
"StringUtil.cpp",
|
||||||
"TestFramework.cpp",
|
"TestFramework.cpp",
|
||||||
"Thunk.cpp",
|
"Thunk.cpp",
|
||||||
|
|||||||
Reference in New Issue
Block a user