/*++

Copyright (c) 1990-98  Microsoft Corporation All Rights Reserved

Module Name:

    testapp.c

Abstract:   

Author:

    Eliyas Yakub

Environment:

    Win32 console multi-threaded application

Revision History:

--*/
#include <windows.h>
#include <winioctl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "..\sys\sioctl.h"


BOOLEAN
ManageDriver(
    IN LPCTSTR  DriverName,
    IN LPCTSTR  ServiceName,
    IN USHORT   Function
    );

BOOLEAN
SetupDriverName(
    PUCHAR DriverLocation
    );

char OutputBuffer[100];
char InputBuffer[100];

VOID _cdecl main( ULONG argc, PCHAR argv[] )
{
    HANDLE hDevice;
    BOOL bRc;
    ULONG bytesReturned;
    DWORD errNum = 0;
    UCHAR driverLocation[MAX_PATH];


    //
    // open the device
    //
    
    if((hDevice = CreateFile( "\\\\.\\IoctlTest",
            GENERIC_READ | GENERIC_WRITE,
            0,
            NULL,
            CREATE_ALWAYS,
            FILE_ATTRIBUTE_NORMAL,
            NULL)) == INVALID_HANDLE_VALUE) {
                
        errNum = GetLastError();

        if (errNum != ERROR_FILE_NOT_FOUND) {

            printf("CreateFile failed!  ERROR_FILE_NOT_FOUND = %d\n", errNum);

            return ;
        }
       
        //
        // The driver is not started yet so let us the install the driver.
        // First setup full path to driver name.
        //

        if (!SetupDriverName(driverLocation)) {

            return ;
        }
        
        if (!ManageDriver(DRIVER_NAME,
                          driverLocation,
                          DRIVER_FUNC_INSTALL
                          )) {

            printf("Unable to install driver. \n");

            //
            // Error - remove driver.
            //

            ManageDriver(DRIVER_NAME,
                         driverLocation,
                         DRIVER_FUNC_REMOVE
                         );
            
            return;
        }
        
        hDevice = CreateFile( "\\\\.\\IoctlTest",
                GENERIC_READ | GENERIC_WRITE,
                0,
                NULL,
                CREATE_ALWAYS,
                FILE_ATTRIBUTE_NORMAL,
                NULL);

        if ( hDevice == INVALID_HANDLE_VALUE ){
            printf ( "Error: CreatFile Failed : %d\n", GetLastError());
            return;
        }

    }

    //
    // Printing Input & Output buffer pointers and size
    //
    
    printf("InputBuffer Pointer = %s, BufLength = %d\n", InputBuffer, 
                        sizeof(InputBuffer));
    printf("OutputBuffer Pointer = %s BufLength = %d\n", OutputBuffer,
                                sizeof(OutputBuffer));
    //
    // Performing METHOD_BUFFERED
    //
    
    strcpy(InputBuffer,
        "This String is from User Application; using METHOD_BUFFERED");

    printf("\nCalling DeviceIoControl METHOD_BUFFERED:\n");
    
    memset(OutputBuffer, 0, sizeof(OutputBuffer));

    bRc = DeviceIoControl ( hDevice, 
                                        (DWORD) IOCTL_SIOCTL_METHOD_BUFFERED, 
                                        &InputBuffer, 
                                        strlen ( InputBuffer )+1, 
                                        &OutputBuffer,
                                        sizeof( OutputBuffer),
                                        &bytesReturned,
                                        NULL 
                                        );

    if ( !bRc )
    {
        printf ( "Error in DeviceIoControl : %d", GetLastError());
        return;

    }
    printf("    OutBuffer (%d): %s\n", bytesReturned, OutputBuffer);
    
    //
    // Performing METHOD_NIETHER
    //
    
    printf("\nCalling DeviceIoControl METHOD_NEITHER\n");

    strcpy(InputBuffer,
               "This String is from User Application; using METHOD_NEITHER");
    memset(OutputBuffer, 0, sizeof(OutputBuffer));

    bRc = DeviceIoControl ( hDevice, 
                                        (DWORD) IOCTL_SIOCTL_METHOD_NEITHER, 
                                        &InputBuffer, 
                                        strlen ( InputBuffer )+1, 
                                        &OutputBuffer,
                                        sizeof( OutputBuffer),
                                        &bytesReturned,
                                        NULL 
                                        );

    if ( !bRc )
    {
        printf ( "Error in DeviceIoControl : %d\n", GetLastError());
        return;

    }
    
    printf("    OutBuffer (%d): %s\n", bytesReturned, OutputBuffer);
    
    //
    // Performing METHOD_IN_DIRECT
    //
    
    printf("\nCalling DeviceIoControl METHOD_IN_DIRECT\n");

    strcpy(InputBuffer,
               "This String is from User Application; using METHOD_IN_DIRECT");
    strcpy(OutputBuffer,
               "This String is from User Application in OutBuffer; using METHOD_IN_DIRECT");

    bRc = DeviceIoControl ( hDevice, 
                                        (DWORD) IOCTL_SIOCTL_METHOD_IN_DIRECT, 
                                        &InputBuffer, 
                                        strlen ( InputBuffer )+1, 
                                        &OutputBuffer,
                                        sizeof( OutputBuffer),
                                        &bytesReturned,
                                        NULL 
                                        );

    if ( !bRc )
    {
        printf ( "Error in DeviceIoControl : : %d", GetLastError());
        return;
    }

    printf("    Number of bytes transfered from OutBuffer: %d\n", 
                                    bytesReturned);
 
    //
    // Performing METHOD_OUT_DIRECT
    //
    
    printf("\nCalling DeviceIoControl METHOD_OUT_DIRECT\n");
    strcpy(InputBuffer,
               "This String is from User Application; using METHOD_OUT_DIRECT");
    memset(OutputBuffer, 0, sizeof(OutputBuffer));
    bRc = DeviceIoControl ( hDevice, 
                                        (DWORD) IOCTL_SIOCTL_METHOD_OUT_DIRECT, 
                                        &InputBuffer, 
                                        strlen ( InputBuffer )+1, 
                                        &OutputBuffer,
                                        sizeof( OutputBuffer),
                                        &bytesReturned,
                                        NULL 
                                        );

    if ( !bRc )
    {
        printf ( "Error in DeviceIoControl : : %d", GetLastError());
        return;
    }

    printf("    OutBuffer (%d): %s\n", bytesReturned, OutputBuffer);

    CloseHandle ( hDevice );

    //
    // Unload the driver.  Ignore any errors.
    //

    ManageDriver(DRIVER_NAME,
                 driverLocation,
                 DRIVER_FUNC_REMOVE
                 );

    
    //
    // close the handle to the device.
    //

}