reactos/rostests/tests/lantst/lantest.cpp

219 lines
4.9 KiB
C++
Raw Normal View History

#include <windows.h>
#include <net/lan.h>
#include <iostream>
#include <string>
#include <ddk/ntddk.h>
#include <rosrtl/string.h>
using std::string;
using std::cin;
using std::cout;
using std::cerr;
using std::endl;
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 );
}
for( ; i < off + 16; i++ )
printf(" ");
printf( " -- " );
for( i = off; i < len && i < off + 16; i++ ) {
printf( "%c", (data[i] >= ' ' && data[i] <= '~') ? data[i] : '.' );
}
printf( "\n" );
}
void display_buffer( char *Packet, int ReadLen ) {
UINT PktLen;
for( PktLen = 0; PktLen < ReadLen; PktLen += 16 )
display_row( Packet, PktLen, ReadLen );
}
int byte_till_end( char *Packet, int PktLen ) {
int byte;
std::string word;
cin >> word;
while( word != "end" ) {
byte = strtoul( (string("0x") + word).c_str(), 0, 0 );
fprintf( stderr, "Byte[%d]: %x\n", PktLen, byte & 0xff );
Packet[PktLen++] = byte;
cin >> word;
}
return PktLen;
}
/* Ethernet types. We swap constants so we can compare values at runtime
without swapping them there */
#define ETYPE_IPv4 WH2N(0x0800)
#define ETYPE_IPv6 WH2N(0x86DD)
#define ETYPE_ARP WH2N(0x0806)
extern "C"
NTSTATUS NTAPI NtCreateFile(
PHANDLE FileHandle,
ACCESS_MASK DesiredAccess,
POBJECT_ATTRIBUTES ObjectAttributes,
PIO_STATUS_BLOCK IoStatusBlock,
PLARGE_INTEGER AllocationSize,
ULONG FileAttributes,
ULONG ShareAccess,
ULONG CreateDisposition,
ULONG CreateOptions,
PVOID EaBuffer,
ULONG EaLength);
int main( int argc, char **argv ) {
string word;
HANDLE LanFile;
OBJECT_ATTRIBUTES Attributes;
UNICODE_STRING LanDevice;
IO_STATUS_BLOCK Iosb;
HANDLE Event;
PFILE_FULL_EA_INFORMATION EaBuffer;
NTSTATUS Status;
DWORD On = 1, PktLen;
CHAR Packet[1600];
PLAN_PACKET_HEADER Hdr = (PLAN_PACKET_HEADER)Packet;
PLAN_ADDRESS Addr = (PLAN_ADDRESS)Packet;
USHORT TypesToListen[] = { ETYPE_IPv4, ETYPE_IPv6, ETYPE_ARP };
UINT EaLength = LAN_EA_INFO_SIZE(sizeof(TypesToListen)/sizeof(USHORT));
Status = NtCreateEvent(&Event,
EVENT_ALL_ACCESS,
NULL,
0,
0 );
RtlInitUnicodeString( &LanDevice, L"\\Device\\Lan" );
InitializeObjectAttributes( &Attributes,
&LanDevice,
OBJ_CASE_INSENSITIVE,
NULL,
NULL );
EaBuffer = (PFILE_FULL_EA_INFORMATION)calloc( EaLength, 1 );
LAN_FILL_EA_INFO(EaBuffer,sizeof(TypesToListen)/sizeof(USHORT),
TypesToListen);
Status = ZwCreateFile( &LanFile,
SYNCHRONIZE | GENERIC_READ | GENERIC_WRITE |
GENERIC_EXECUTE,
&Attributes,
&Iosb,
NULL,
FILE_ATTRIBUTE_NORMAL,
FILE_SHARE_READ | FILE_SHARE_WRITE,
FILE_OPEN_IF,
FILE_SYNCHRONOUS_IO_NONALERT,
EaBuffer,
EaLength );
if( !NT_SUCCESS(Status) ) {
cerr << "Could not open lan device " << Status << "\n";
return 1;
}
Status = DeviceIoControl( LanFile,
IOCTL_IF_BUFFERED_MODE,
&On,
sizeof(On),
0,
0,
&PktLen,
NULL );
if( !Status ) {
cerr << "Could not turn on buffered mode " << Status << "\n";
return 1;
}
while( cin >> word ) {
if( word == "end" ) {
NtClose( LanFile );
return 0;
} else if( word == "enum" ) {
Status = DeviceIoControl( LanFile,
IOCTL_IF_ENUM_ADAPTERS,
NULL,
0,
Packet,
sizeof(Packet),
&PktLen,
NULL );
cout << "EnumAdapters: " << Status << "\n";
if( Status )
display_buffer( Packet, PktLen );
} else if( word == "query" ) {
cin >> PktLen;
Status = DeviceIoControl( LanFile,
IOCTL_IF_ADAPTER_INFO,
&PktLen,
sizeof(PktLen),
Packet,
sizeof(Packet),
&PktLen,
NULL );
cout << "QueryAdapterInfo: " << Status << "\n";
if( Status )
display_buffer( Packet, PktLen );
} else if( word == "send" ) {
cin >> Hdr->Fixed.Adapter
>> Hdr->Fixed.AddressType
>> Hdr->Fixed.AddressLen
>> Hdr->Fixed.PacketType;
Hdr->Fixed.Mdl = NULL;
PktLen = byte_till_end( Packet, Hdr->Address - (PCHAR)Hdr );
Status = NtWriteFile( LanFile,
NULL,
NULL,
NULL,
&Iosb,
Packet,
PktLen,
NULL,
NULL );
cout << "NtWriteFile: " << Status << "\n";
} else if( word == "recv" ) {
ULONG ReadLen;
Status = NtReadFile( LanFile,
Event,
NULL,
NULL,
&Iosb,
Packet,
sizeof(Packet),
NULL,
NULL );
cout << "NtReadFile: " << Status << "\n";
if( Status == STATUS_PENDING ) {
LARGE_INTEGER Timeout = { 0 };
Status = NtWaitForSingleObject( Event, 1, &Timeout );
}
ReadLen = Iosb.Information;
if( Status == STATUS_SUCCESS ) {
cout << "Read " << ReadLen << " bytes\n";
display_buffer( Packet, ReadLen );
}
}
}
return 0;
}