+++ /dev/null
-#ifdef _MSC_VER
-#pragma warning ( disable : 4786 )
-#endif//_MSC_VER
-
-#include <list>
-#include <string>
-#include <sstream>
-#include <malloc.h>
-extern "C" {
- typedef unsigned short u_short;
-#include <stdio.h>
-#include <oskittcp.h>
-#include <windows.h>
-#ifndef _MSC_VER
-#include <winsock2.h>
-#endif//_MSC_VER
-};
-
-unsigned char hwaddr[6] = { 0x08, 0x00, 0x20, 0x0b, 0xb7, 0xbb };
-
-#undef malloc
-#undef free
-
-unsigned long TCP_IPIdentification;
-
-#define MAX_DG_SIZE 0x10000
-#define TI_DbgPrint(x,y) printf y
-
-std::list<std::string> output_packets;
-
-typedef struct _CONNECTION_ENDPOINT {
- OSK_UINT State;
-} CONNECTION_ENDPOINT, *PCONNECTION_ENDPOINT;
-
-extern "C" int is_stack_ptr ( const void* p )
-{
- MEMORY_BASIC_INFORMATION mbi1, mbi2;
- VirtualQuery ( p, &mbi1, sizeof(mbi1) );
- VirtualQuery ( _alloca(1), &mbi2, sizeof(mbi2) );
- return mbi1.AllocationBase == mbi2.AllocationBase;
-}
-
-int TCPSocketState(void *ClientData,
- void *WhichSocket,
- void *WhichConnection,
- OSK_UINT NewState ) {
- PCONNECTION_ENDPOINT Connection = (PCONNECTION_ENDPOINT)WhichConnection;
- //PLIST_ENTRY Entry;
-
- TI_DbgPrint(MID_TRACE,("Called: NewState %x\n", NewState));
-
- if( !Connection ) {
- TI_DbgPrint(MID_TRACE,("Socket closing.\n"));
- return 0;
- }
-
- if( (NewState & SEL_CONNECT) &&
- !(Connection->State & SEL_CONNECT) ) {
- } else if( (NewState & SEL_READ) || (NewState & SEL_FIN) ) {
- }
-
- return 0;
-}
-
-#define STRINGIFY(x) #x
-
-char hdr[14] = { 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
- 0x08, 0x00 };
-
-int TCPPacketSend(void *ClientData, OSK_PCHAR data, OSK_UINT len ) {
- output_packets.push_back( std::string( hdr, 14 ) +
- std::string( (char *)data, (int)len ) );
- return 0;
-}
-
-struct ifaddr *TCPFindInterface( void *ClientData,
- OSK_UINT AddrType,
- OSK_UINT FindType,
- struct sockaddr *ReqAddr ) {
- static struct sockaddr_in ifa = { AF_INET }, nm = { AF_INET };
- static struct ifaddr a = {
- (struct sockaddr *)&ifa,
- NULL,
- (struct sockaddr *)&nm,
- 0,
- 0,
- 1,
- 1500
- };
- ifa.sin_addr.s_addr = inet_addr( "10.10.2.115" );
- nm.sin_addr.s_addr = inet_addr( "255.255.255.0" );
- return &a;
-}
-
-void *TCPMalloc( void *ClientData,
- OSK_UINT Bytes, OSK_PCHAR File, OSK_UINT Line ) {
- void *v = malloc( Bytes );
- fprintf( stderr, "(%s:%d) malloc( %d ) => %x\n", File, Line, Bytes, v );
- return v;
-}
-
-void TCPFree( void *ClientData,
- void *data, OSK_PCHAR File, OSK_UINT Line ) {
- fprintf( stderr, "(%s:%d) free( %x )\n", File, Line, data );
- free( data );
-}
-
-int TCPSleep( void *ClientData, void *token, int priority, char *msg,
- int tmio ) {
-#if 0
- PSLEEPING_THREAD SleepingThread;
-
- TI_DbgPrint(MID_TRACE,
- ("Called TSLEEP: tok = %x, pri = %d, wmesg = %s, tmio = %x\n",
- token, priority, msg, tmio));
-
- SleepingThread = ExAllocatePool( NonPagedPool, sizeof( *SleepingThread ) );
- if( SleepingThread ) {
- KeInitializeEvent( &SleepingThread->Event, NotificationEvent, FALSE );
- SleepingThread->SleepToken = token;
-
- ExAcquireFastMutex( &SleepingThreadsLock );
- InsertTailList( &SleepingThreadsList, &SleepingThread->Entry );
- ExReleaseFastMutex( &SleepingThreadsLock );
-
- TI_DbgPrint(MID_TRACE,("Waiting on %x\n", token));
- KeWaitForSingleObject( &SleepingThread->Event,
- WrSuspended,
- KernelMode,
- TRUE,
- NULL );
-
- ExAcquireFastMutex( &SleepingThreadsLock );
- RemoveEntryList( &SleepingThread->Entry );
- ExReleaseFastMutex( &SleepingThreadsLock );
-
- ExFreePool( SleepingThread );
- }
- TI_DbgPrint(MID_TRACE,("Waiting finished: %x\n", token));
-#endif
- return 0;
-}
-
-void TCPWakeup( void *ClientData, void *token ) {
-#if 0
- PLIST_ENTRY Entry;
- PSLEEPING_THREAD SleepingThread;
-
- ExAcquireFastMutex( &SleepingThreadsLock );
- Entry = SleepingThreadsList.Flink;
- while( Entry != &SleepingThreadsList ) {
- SleepingThread = CONTAINING_RECORD(Entry, SLEEPING_THREAD, Entry);
- TI_DbgPrint(MID_TRACE,("Sleeper @ %x\n", SleepingThread));
- if( SleepingThread->SleepToken == token ) {
- TI_DbgPrint(MID_TRACE,("Setting event to wake %x\n", token));
- KeSetEvent( &SleepingThread->Event, IO_NETWORK_INCREMENT, FALSE );
- }
- Entry = Entry->Flink;
- }
- ExReleaseFastMutex( &SleepingThreadsLock );
-#endif
-}
-
-OSKITTCP_EVENT_HANDLERS EventHandlers = {
- NULL,
- TCPSocketState,
- TCPPacketSend,
- TCPFindInterface,
- TCPMalloc,
- TCPFree,
- TCPSleep,
- TCPWakeup
-};
-
-void display_row( char *data, int off, int len ) {
- int i;
-
- printf( "%08x:", off );
- for( i = off; i < len && i < off + 16; i++ ) {
- printf( " %02x", data[i] & 0xff );
- }
-
- printf( " -- " );
-
- for( i = off; i < len && i < off + 16; i++ ) {
- printf( "%c", (data[i] >= ' ') ? data[i] : '.' );
- }
-
- printf( "\n" );
-}
-
-int main( int argc, char **argv ) {
- int asock = INVALID_SOCKET, selret, dgrecv, fromsize, err, port = 5001;
- char datagram[MAX_DG_SIZE];
- void *conn = 0;
- struct fd_set readf;
- struct timeval tv;
- struct sockaddr_in addr_from = { AF_INET }, addr_to = { AF_INET };
- std::list<std::string>::iterator i;
- WSADATA wsadata;
-
- WSAStartup( 0x101, &wsadata );
-
- if( argc > 1 ) port = atoi(argv[1]);
-
- RegisterOskitTCPEventHandlers( &EventHandlers );
- InitOskitTCP();
-
- asock = socket( AF_INET, SOCK_DGRAM, 0 );
-
- addr_from.sin_port = htons( port );
-
- if( bind( asock, (struct sockaddr *)&addr_from, sizeof( addr_from ) ) ) {
- printf( "Bind error\n" );
- return 0;
- }
-
- addr_to.sin_port = htons( port & (~1) );
- addr_to.sin_addr.s_addr = inet_addr("127.0.0.1");
-
- while( true ) {
- FD_ZERO( &readf );
- FD_SET( asock, &readf );
- tv.tv_sec = 0;
- tv.tv_usec = 10000;
- selret = select( asock + 1, &readf, NULL, NULL, &tv );
-
- if( FD_ISSET( asock, &readf ) ) {
- fromsize = sizeof( addr_from );
- dgrecv = recvfrom( asock, datagram, sizeof(datagram), 0,
- (struct sockaddr *)&addr_from, &fromsize );
-
- if( datagram[0] == 'C' && datagram[1] == 'M' &&
- datagram[2] == 'D' && datagram[3] == ' ' ) {
- int theport, bytes, /*recvret,*/ off, bytin;
- struct sockaddr_in nam;
- std::string faddr, word;
- std::istringstream
- cmdin( std::string( datagram + 4, dgrecv - 4 ) );
-
- cmdin >> word;
-
- if( word == "socket" ) {
- cmdin >> faddr >> theport;
-
- nam.sin_family = AF_INET;
- nam.sin_addr.s_addr = inet_addr(faddr.c_str());
- nam.sin_port = htons(theport);
-
- if( (err = OskitTCPSocket( NULL, &conn, AF_INET,
- SOCK_STREAM, 0 )) != 0 ) {
- fprintf( stderr, "OskitTCPSocket: error %d\n", err );
- }
-
- if( (err = OskitTCPConnect( conn, NULL, &nam,
- sizeof(nam) )) != 0 ) {
- fprintf( stderr, "OskitTCPConnect: error %d\n", err );
- } else {
- printf( "Socket created\n" );
- }
- }
-
- /* The rest of the commands apply only to an open socket */
- if( !conn ) continue;
-
- if( word == "recv" ) {
- cmdin >> bytes;
-
- if( (err = OskitTCPRecv( conn, (OSK_PCHAR)datagram,
- sizeof(datagram),
- (unsigned int *)&bytin, 0 )) != 0 ) {
- fprintf( stderr, "OskitTCPRecv: error %d\n", err );
- } else {
- for( off = 0; off < bytin; off += 16 ) {
- display_row( datagram, off, bytin );
- }
- printf( "\n" );
- }
- } else if ( word == "type" ) {
- std::string therest = &cmdin.str()[word.size()];
- char* p = &therest[0];
- p += strspn ( p, " \t" );
- char* src = p;
- char* dst = p;
- while ( *src )
- {
- char c = *src++;
- if ( c == '\r' || c == '\n' ) break;
- if ( c == '\\' )
- {
- c = *src++;
- switch ( c )
- {
- case 'b': c = '\b'; break;
- case 'n': c = '\n'; break;
- case 'r': c = '\r'; break;
- case 't': c = '\t'; break;
- case 'v': c = '\v'; break;
- }
- }
- *dst++ = c;
- }
- *dst = '\0';
- if ( (err = OskitTCPSend ( conn, (OSK_PCHAR)p, strlen(p), (OSK_UINT*)&bytin, 0 ))
- != 0 ) {
- fprintf ( stderr, "OskitTCPConnect: error %d\n", err );
- } else {
- printf ( "wrote %d bytes\n", bytin );
- }
- } else if( word == "send" ) {
- off = 0;
- while( cmdin >> word ) {
- datagram[off++] =
- atoi( (std::string("0x") + word).c_str() );
- }
-
- if( (err = OskitTCPSend( conn, (OSK_PCHAR)datagram,
- off, (OSK_UINT *)&bytin, 0 ))
- != 0 ) {
- fprintf( stderr, "OskitTCPConnect: error %d\n", err );
- } else {
- printf( "wrote %d bytes\n", bytin );
- }
- } else if( word == "close" ) {
- OskitTCPClose( conn );
- conn = NULL;
- }
- } else if( dgrecv > 14 ) {
- addr_to = addr_from;
-
- if( datagram[12] == 8 && datagram[13] == 6 ) {
- /* Answer arp query */
- char laddr[4];
- /* Mark patch as to the previous sender */
- memcpy( datagram + 32, datagram + 6, 6 );
- memcpy( datagram, datagram + 6, 6 );
- /* Mark packet as from us */
- memcpy( datagram + 22, hwaddr, 6 );
- memcpy( datagram + 6, hwaddr, 6 );
- /* Swap inet addresses */
- memcpy( laddr, datagram + 28, 4 );
- memcpy( datagram + 28, datagram + 38, 4 );
- memcpy( datagram + 38, laddr, 4 );
- /* Set reply opcode */
- datagram[21] = 2;
-
- err = sendto( asock, datagram, dgrecv, 0,
- (struct sockaddr *)&addr_to,
- sizeof(addr_to) );
-
- if( err != 0 )
- printf( "sendto: %d\n", err );
- } else {
- memcpy( hdr, datagram + 6, 6 );
- memcpy( hdr + 6, datagram, 6 );
- memcpy( hdr + 12, datagram + 12, 2 );
- OskitTCPReceiveDatagram
- ( (unsigned char *)datagram + 14,
- dgrecv - 14, 20 );
- }
- }
- }
-
- TimerOskitTCP();
-
- for( i = output_packets.begin(); i != output_packets.end(); i++ ) {
- err = sendto( asock, i->c_str(), i->size(), 0,
- (struct sockaddr *)&addr_to, sizeof(addr_to) );
-
- fprintf( stderr, "** SENDING PACKET %d bytes **\n", i->size() );
-
- if( err != 0 )
- printf( "sendto: %d\n", err );
- }
-
- output_packets.clear();
- }
-}
-