lantest.cpp

Go to the documentation of this file.

#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;
}