FGFS-4.1/OpenSceneGraph
4
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added support for Win32 broadcast and recience over lan. Changes sent in

Browse Source 
by Michael Gronager.
This commit is contained in:
Robert Osfield
2002-07-04 18:41:13 +00:00
parent 5014a4f809
commit 0dcd5cb017
5 changed files with 156 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files

99
src/Demos/osgcluster/broadcaster.cpp
View File

@@ -1,8 +1,9 @@
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/ioctl.h>
#include <sys/types.h>
#if !defined (WIN32)
#include <sys/ioctl.h>
#include <sys/uio.h>
#include <sys/socket.h>
#include <netinet/in.h>
@@ -11,9 +12,12 @@
#include <sys/time.h>
#include <net/if.h>
#include <netdb.h>
#endif
#include <string.h>
#if defined(__linux)
#include <unistd.h>
# include <linux/sockios.h>
#elif defined(__FreeBSD__)
# include <sys/sockio.h>
@@ -25,6 +29,9 @@
# include <sys/sockio.h>
#elif defined (__DARWIN_OSX__) // added
# include <sys/sockio.h> // added
#elif defined (WIN32) // added
# include <winsock.h> // added
# include <stdio.h> // added
#else
# error Teach me how to build on this system
#endif
@@ -43,50 +50,82 @@ Broadcaster::Broadcaster( void )
Broadcaster::~Broadcaster( void )
{
#if defined (WIN32)
closesocket( _so);
#else
close( _so );
#endif
}
bool Broadcaster::init( void )
{
#if defined (WIN32)
WORD version = MAKEWORD(1,1);
WSADATA wsaData;
// First, we start up Winsock
WSAStartup(version, &wsaData);
#endif
if( _port == 0 )
{
fprintf( stderr, "Broadcaster::init() - port not defined\n" );
return false;
fprintf( stderr, "Broadcaster::init() - port not defined\n" );
return false;
}
if( (_so = socket( AF_INET, SOCK_DGRAM, 0 )) < 0 )
{
perror( "socket" );
return false;
perror( "Socket" );
return false;
}
#if defined (WIN32)
const BOOL on = TRUE;
#else
int on = 1;
#endif
#if defined (WIN32)
setsockopt( _so, SOL_SOCKET, SO_REUSEADDR, (const char *) &on, sizeof(int));
#else
setsockopt( _so, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
#endif
saddr.sin_family = AF_INET;
saddr.sin_port = htons( _port );
if( _address == 0 )
{
struct ifreq ifr;
setsockopt( _so, SOL_SOCKET, SO_BROADCAST, &on, sizeof(on));
#if defined (__linux)
strcpy( ifr.ifr_name, "eth0" );
#elif defined(__sun)
strcpy( ifr.ifr_name, "hme0" );
#if defined (WIN32)
setsockopt( _so, SOL_SOCKET, SO_BROADCAST, (const char *) &on, sizeof(int));
#else
strcpy( ifr.ifr_name, "ef0" );
setsockopt( _so, SOL_SOCKET, SO_BROADCAST, &on, sizeof(on));
#endif
if( (ioctl( _so, SIOCGIFBRDADDR, &ifr)) < 0 )
{
perror( "Broadcaster::init() Cannot get Broadcast Address" );
return false;
}
saddr.sin_addr.s_addr = (
((sockaddr_in *)&ifr.ifr_broadaddr)->sin_addr.s_addr);
}
else
{
saddr.sin_addr.s_addr = _address;
#if !defined (WIN32)
struct ifreq ifr;
#endif
#if defined (__linux)
strcpy( ifr.ifr_name, "eth0" );
#elif defined(__sun)
strcpy( ifr.ifr_name, "hme0" );
#elif !defined (WIN32)
strcpy( ifr.ifr_name, "ef0" );
#endif
#if defined (WIN32) // get the server address
saddr.sin_addr.s_addr = htonl(INADDR_BROADCAST);
}
#else
if( (ioctl( _so, SIOCGIFBRDADDR, &ifr)) < 0 )
{
perror( "Broadcaster::init() Cannot get Broadcast Address" );
return false;
}
saddr.sin_addr.s_addr = (((sockaddr_in *)&ifr.ifr_broadaddr)->sin_addr.s_addr);
}
else
{
saddr.sin_addr.s_addr = _address;
}
#endif
#define _VERBOSE 1
#ifdef _VERBOSE
unsigned char *ptr = (unsigned char *)&saddr.sin_addr.s_addr;
printf( "Broadcast address : %u.%u.%u.%u\n", ptr[0], ptr[1], ptr[2], ptr[3] );
@@ -121,7 +160,7 @@ void Broadcaster::setBuffer( void *buffer, const unsigned int size )
void Broadcaster::sync( void )
{
_initialized || init();
if(!_initialized) init();
if( _buffer == 0L )
{
@@ -129,9 +168,15 @@ void Broadcaster::sync( void )
return;
}
#if defined (WIN32)
unsigned int size = sizeof( SOCKADDR_IN );
sendto( _so, (const char *)_buffer, _buffer_size, 0, (struct sockaddr *)&saddr, size );
int err = WSAGetLastError ();
int *dum = (int*) _buffer;
#else
unsigned int size = sizeof( struct sockaddr_in );
sendto( _so, (const void *)_buffer, _buffer_size,
0, (struct sockaddr *)&saddr, size );
sendto( _so, (const void *)_buffer, _buffer_size, 0, (struct sockaddr *)&saddr, size );
#endif
}

16
src/Demos/osgcluster/broadcaster.h
View File

@@ -7,7 +7,9 @@
// Class definition for broadcasting a buffer to a LAN
//
#include <netinet/in.h>
#ifndef WIN32
#include <netinet/in.h>
#endif
class Broadcaster
{
@@ -34,12 +36,20 @@ class Broadcaster
bool init( void );
private :
#if defined (WIN32)
SOCKET _so;
#else
int _so;
#endif
bool _initialized;
short _port;
void *_buffer;
unsigned int _buffer_size;
struct sockaddr_in saddr;
unsigned long _address;
#if defined (WIN32)
SOCKADDR_IN saddr;
#else
struct sockaddr_in saddr;
#endif
unsigned long _address;
};
#endif

4
src/Demos/osgcluster/osgcluster.cpp
View File

@@ -17,6 +17,10 @@
#include <osg/Quat>
#if defined (WIN32)
#include <winsock.h>
#endif
#include "receiver.h"
#include "broadcaster.h"

60
src/Demos/osgcluster/receiver.cpp
View File

@@ -1,13 +1,17 @@
#include <stdio.h>
#include <unistd.h>
#include <fcntl.h>
#include <sys/types.h>
#if defined (WIN32)
#include <winsock.h>
#else
#include <unistd.h>
#include <sys/uio.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <netdb.h>
#include <arpa/inet.h>
#include <sys/time.h>
#endif
#include <string.h>
@@ -22,34 +26,54 @@ Receiver::Receiver( void )
Receiver::~Receiver( void )
{
#if defined (WIN32)
closesocket( _so);
#else
close( _so );
#endif
}
bool Receiver::init( void )
{
#ifdef WIN32
WORD version = MAKEWORD(1,1);
WSADATA wsaData;
// First, we start up Winsock
WSAStartup(version, &wsaData);
#endif
if( _port == 0 )
{
fprintf( stderr, "Receiver::init() - port not defined\n" );
return false;
fprintf( stderr, "Receiver::init() - port not defined\n" );
return false;
}
if( (_so = socket( AF_INET, SOCK_DGRAM, 0 )) < 0 )
{
perror( "socket" );
return false;
perror( "Socket" );
return false;
}
#if defined (WIN32)
// const BOOL on = TRUE;
// setsockopt( _so, SOL_SOCKET, SO_REUSEADDR, (const char*) &on, sizeof(int));
#else
int on = 1;
setsockopt( _so, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
#endif
struct sockaddr_in saddr;
// struct sockaddr_in saddr;
saddr.sin_family = AF_INET;
saddr.sin_port = htons( _port );
#if defined (WIN32)
saddr.sin_addr.s_addr = htonl(INADDR_ANY);
#else
saddr.sin_addr.s_addr = 0;
#endif
if( bind( _so, (struct sockaddr *)&saddr, sizeof( saddr )) < 0 )
{
perror( "bind" );
return false;
return false;
}
_initialized = true;
@@ -70,7 +94,7 @@ void Receiver::setBuffer( void *buffer, const unsigned int size )
void Receiver::sync( void )
{
_initialized || init();
if(!_initialized) init();
if( _buffer == 0L )
{
@@ -93,13 +117,29 @@ void Receiver::sync( void )
tv.tv_sec = 0;
tv.tv_usec = 0;
#if defined (WIN32)
// saddr.sin_port = htons( _port );
recvfrom( _so, (char *)_buffer, _buffer_size, 0, (sockaddr*)&saddr, &size );
// recvfrom(sock_Receive, szMessage, 256, 0, (sockaddr*)&addr_Cli, &clilen)
int err = WSAGetLastError ();
int *dum = (int*) _buffer;
while( select( _so+1, &fdset, 0L, 0L, &tv ) )
{
if( FD_ISSET( _so, &fdset ) )
{
recvfrom( _so, (char *)_buffer, _buffer_size, 0, (sockaddr*)&saddr, &size );
}
}
#else
recvfrom( _so, (caddr_t)_buffer, _buffer_size, 0, 0, &size );
while( select( _so+1, &fdset, 0L, 0L, &tv ) )
{
if( FD_ISSET( _so, &fdset ) )
{
{
recvfrom( _so, (caddr_t)_buffer, _buffer_size, 0, 0, &size );
}
}
}
#endif
}

22
src/Demos/osgcluster/receiver.h
View File

@@ -8,6 +8,9 @@
// Class definition for the recipient of a broadcasted message
//
#ifndef WIN32
#include <netinet/in.h>
#endif
class Receiver
{
@@ -30,10 +33,19 @@ class Receiver
bool init( void );
private :
int _so;
bool _initialized;
short _port;
void *_buffer;
unsigned int _buffer_size;
#if defined (WIN32)
SOCKET _so;
#else
int _so;
#endif
#if defined (WIN32)
SOCKADDR_IN saddr;
#else
struct sockaddr_in saddr;
#endif
bool _initialized;
short _port;
void *_buffer;
unsigned int _buffer_size;
};
#endif