/*++

   Copyright    (c)    1995    Microsoft Corporation

   Module  Name :

      ftps.cxx

   Abstract:

      This file defines the ftp server stress client

   Author:

       Murali R. Krishnan    ( MuraliK )     25-July-1995

   Environment:

       Win32 - uses Wininet extensions

   Project:

       FTP Server DLL

   Functions Exported:



   Revision History:
      MuraliK   05-Nov-1995    Added support for GetFile tests

--*/


/************************************************************
 *     Include Headers
 ************************************************************/
# include <windows.h>
# include <wininet.h>
# include <stdio.h>
# include <stdlib.h>
# include <iostream.h>
# include <winsock2.h>

#define DEFAULT_TRACE_FLAGS     (DEBUG_ITERATION)

# include <dbgutil.h>


# define DEFAULT_NUMBER_OF_THREADS        (10)
# define DEFAULT_NUMBER_OF_ITERATIONS     (100)

# define MAX_BUFFER_SIZE    64000  // 64K approx

#ifndef _NO_TRACING_
#include <initguid.h>
DEFINE_GUID(IisFtptestGuid, 
0x784d8914, 0xaa8c, 0x11d2, 0x92, 0x5e, 0x00, 0xc0, 0x4f, 0x72, 0xd9, 0x0e);
#else
DECLARE_DEBUG_VARIABLE();
#endif
DECLARE_DEBUG_PRINTS_OBJECT();



static const char PSZ_APPLICATION_NAME[] = "murali's stresser";
static char * g_lpszServerAddress;

BOOL
GenUsageMessage( int argc, char * argv[]);

// Tests the raw connectivity to server.
BOOL TestConnections( int argc, char * argv[]);

// Tests the raw get file from server.
BOOL TestGetFile( int argc, char * argv[]);


//
//  The following DefineAllCommands() defines a template for all commands.
//  Format: CmdCodeName     CommandName         Function Pointer   Comments
//
//  To add addditional test commands, add just another line to the list
//  Dont touch any macros below, they are all automatically generated.
//  Always the first entry should be usage function.
//

#define  DefineAllCommands()    \
 Cmd( CmdUsage,             "usage",                GenUsageMessage,   \
        " Commands Available" )                                        \
 Cmd( CmdConnections,       "conn",                TestConnections,    \
        " Raw Connections to server" )                                 \
 Cmd( CmdGetFile,           "get",                 TestGetFile,        \
        " Simple Get File (/readme.txt) from server" )                 \


// Define command codes

# define Cmd( CmdCode, CmdName, CmdFunc, CmdComments)       CmdCode,

typedef enum  _CmdCodes {
    DefineAllCommands()
    maxCmdCode
} CmdCodes;

#undef Cmd

// Define the functions and array of mappings

// General command function type
typedef BOOL ( * CMDFUNC)( int argc, char * argv[]);

typedef  struct _CmdStruct {
    CmdCodes    cmdCode;
    char *      pszCmdName;
    CMDFUNC     cmdFunc;
    char *      pszCmdComments;
} CmdStruct;


// Define Prototypes of command functions
# define Cmd( CmdCode, CmdName, CmdFunc, CmdComments)    \
    BOOL CmdFunc(int argc, char * argv[]);

// Cause an expansion to generate prototypes
// DefineAllCommands()
// Automatic generation causes a problem when we have NULL in Function ptrs :(
// Let the user explicitly define the prototypes

#undef Cmd

//
// Define the global array of commands
//

# define Cmd( CmdCode, CmdName, CmdFunc, CmdComments)        \
    { CmdCode, CmdName, CmdFunc, CmdComments},

static CmdStruct   g_cmds[] = {

    DefineAllCommands()
    { maxCmdCode, NULL, NULL}       // sentinel command
};

#undef Cmd



/************************************************************
 *    Functions
 ************************************************************/

BOOL
GenUsageMessage( int argc, char * argv[])
{
    CmdStruct * pCmd;

    printf( " Usage:\n %s <server-name/address> <cmd name> <cmd arguments>\n",
            argv[0]);
    for( pCmd = g_cmds; pCmd != NULL && pCmd->cmdCode != maxCmdCode; pCmd++) {
        printf( "\t%s\t%s\n", pCmd->pszCmdName, pCmd->pszCmdComments);
    }

    return ( TRUE);
} // GenUsageMessage()



static
CmdStruct * DecodeCommand( char * pszCmd)
{
    CmdStruct * pCmd;
    if ( pszCmd != NULL) {

        for( pCmd = g_cmds;
             pCmd != NULL && pCmd->cmdCode != maxCmdCode; pCmd++) {

            if ( _stricmp( pszCmd, pCmd->pszCmdName) == 0) {
                 return ( pCmd);
            }
        } // for
    }

    return ( &g_cmds[0]);      // No match found, return usage message
} // DecodeCommand()


/************************************************************
 *    Functions
 ************************************************************/


inline
VOID
GenUserName( OUT LPSTR pszBuffer, IN LPCSTR pszPrefix, IN DWORD iteration)
{
    // assume sufficient buffer space

    // Format of response:   UserPrefix@pProcessId.tThreadId.Iteration

    sprintf( pszBuffer, "%s@p%d.t%d.%d",
            pszPrefix,
            GetCurrentProcessId(),
            GetCurrentThreadId(),
            iteration);

    return;

} // GenUserName()



DWORD
TestConnectionsInOneThread( IN CHAR * pszServer, IN DWORD nIterations)
{
    HINTERNET hinet;
    HINTERNET hFtp;
    DWORD NumSuccess = 0;
    CHAR  rgchBuffer[300];
    CHAR  rgchUserName[100];
    DWORD cbBuffer;
    DWORD dwError;
    DWORD iter;
    BOOL  fReturn2;


    hinet = InternetOpen( PSZ_APPLICATION_NAME, 0, NULL, 0, 0);

    IF_DEBUG( ENTRY) {

        DBGPRINTF(( DBG_CONTEXT, "InternetOpen()==> %08x\n", hinet));
    }

    if ( hinet == NULL) {

        return (0);
    }

    for( iter = 0 ; iter < nIterations; iter++) {

        IF_DEBUG( ITERATION) {

            DBGPRINTF(( DBG_CONTEXT, "Iteration = %u\n", iter));
        }

        GenUserName( rgchUserName, "conn", iter);

        hFtp = InternetConnect( hinet, pszServer, 0, "anonymous",
                               rgchUserName,
                               INTERNET_SERVICE_FTP, 0, NULL);
        IF_DEBUG( ENTRY) {

            DBGPRINTF(( DBG_CONTEXT, "    InternetConnect()==> %08x\n",
                       hFtp));
        }

        if ( hFtp == NULL) {

            continue;
        }

        IF_DEBUG( ENTRY) {

            DBGPRINTF(( DBG_CONTEXT, "       InternetGetLastResponse()\n"));
        }
        cbBuffer = sizeof(rgchBuffer) - 1;
        if ( InternetGetLastResponseInfo( &dwError, rgchBuffer, &cbBuffer)) {

            IF_DEBUG( RESPONSE) {

                rgchBuffer[cbBuffer] = '\0';
                cout << " ErrorCode = " << dwError
                  << "\tResponse = " << rgchBuffer;
            }
        }

        IF_DEBUG( ENTRY) {

            DBGPRINTF(( DBG_CONTEXT, "    InternetCloseHandle(%08x)\n",
                       hFtp));
        }

        if ( InternetCloseHandle(hFtp)) {
            NumSuccess++;
        }
    } // for()

    IF_DEBUG( ENTRY) {

        DBGPRINTF(( DBG_CONTEXT, "InternetCloseHandle(%08x)\n",
                   hinet));
    }

    fReturn2 = InternetCloseHandle(hinet);

    return ( NumSuccess);

} // TestConnectionsInOneThread()




BOOL TestConnections( int argc, char * argv[])
/*++
  This function routinely establishes and throws away connections.
  It does not do any other work.
  This is used for testing logon and quit sequences of FTP server.

  Arguments:
     argc  count of arguments
     argv  arguments
      argv[0] = "conn"  -- name of this test function
      argv[1] = # of thread to use for execution
      argv[2] = # of iterations for each thread.

--*/
{
    DWORD NumThreads = DEFAULT_NUMBER_OF_THREADS;
    DWORD NumIterations = DEFAULT_NUMBER_OF_ITERATIONS;
    DWORD NumSuccesses;


    if (argc >= 2 && argv[1] != NULL) {

        NumThreads = atoi(argv[1]);
    }

    if (argc >= 3 && argv[2] != NULL) {

        NumIterations = atoi(argv[2]);
    }

    // We will implement multithreading later on.

    // for now just one thread is used. ( current thread)
    NumSuccesses = TestConnectionsInOneThread(g_lpszServerAddress,
                                              NumIterations);

    cout << " Tested for Iterations : " << NumIterations;
    cout << "   Successes = " << NumSuccesses;
    cout << "   Failures  = " << (NumIterations - NumSuccesses) << endl;

    return ( NumSuccesses == NumIterations);

} // TestConnections()




DWORD
ReadFtpFile( IN HINTERNET hFtp, IN LPCSTR pszFileName, IN DWORD sTimeout)
{
    HINTERNET hFtpFile;
    CHAR  chBuffer[MAX_BUFFER_SIZE];
    DWORD dwBytesRead = 0;
    DWORD dwError = NO_ERROR;
    BOOL  fRead;

    if ( hFtp == NULL) {

        return ( ERROR_INVALID_PARAMETER);
    }

    hFtpFile = FtpOpenFile( hFtp, pszFileName, GENERIC_READ,
                           FTP_TRANSFER_TYPE_ASCII, 0);

    IF_DEBUG( ENTRY) {

        DBGPRINTF(( DBG_CONTEXT, "\t\tFtpOpenFile(%s) ==> %08x\n",
                   pszFileName, hFtpFile));
    }

    if ( hFtpFile == NULL) {

        return ( GetLastError());
    }

    //===================================
    //
    // Start copying the file from the
    // server to the client
    //
    //====================================


    if ( sTimeout != 0) {

        //
        // sleep causes data time out to occur.
        // If there is a non-zero timeout then we will have
        //  problems with data sockets timing out on server.
        //

        cout << " Requested " << pszFileName <<
          "  Sleeping for " << sTimeout << " seconds" << endl;
        Sleep( sTimeout * 1000);
    }


    do {

        dwError = NO_ERROR;

        // read and discard the data.
        fRead = InternetReadFile(
                                 hFtpFile,
                                 (LPVOID) chBuffer,
                                 (DWORD) MAX_BUFFER_SIZE,
                                 (LPDWORD) &dwBytesRead);

        if ( !fRead) {

            dwError = GetLastError();
            DBGPRINTF(( DBG_CONTEXT, " InternetReadFile(%s) failed."
                       " Error=%d\n",
                       pszFileName, dwError));
        }

        IF_DEBUG( ENTRY) {

            DBGPRINTF(( DBG_CONTEXT, " InternetReadFile(%08x, %s) ==> %d"
                       " (Err=%d)\n",
                       hFtpFile, pszFileName, fRead, dwError));
        }


    } while (fRead && dwBytesRead > 0);

    //
    // Close the handle for read file
    //

    if (!InternetCloseHandle(hFtpFile)) {

        if ( dwError != NO_ERROR) {

            DBGERROR(( DBG_CONTEXT, " Double Errors are occuring Old=%d\n",
                       dwError));
        }

        dwError =  ( GetLastError());
    }

    return ( dwError);

} // ReadFtpFile()



DWORD
TestGetFileInOneThread(IN CHAR *   pszServer,
                       IN LPCSTR   pszFileName,
                       IN DWORD    nIterations,
                       IN DWORD    sTimeout)
{
    HINTERNET hinet;
    HINTERNET hFtp;
    DWORD NumSuccess = 0;
    CHAR  rgchBuffer[300];
    CHAR  rgchUserName[100];
    DWORD cbBuffer;
    DWORD dwError;
    DWORD iter;
    BOOL  fReturn2;


    hinet = InternetOpen( PSZ_APPLICATION_NAME, 0, NULL, 0, 0);

    IF_DEBUG( ENTRY) {

        DBGPRINTF(( DBG_CONTEXT, "InternetOpen()==> %08x\n", hinet));
    }

    if ( hinet == NULL) {

        return (0);
    }


    for( iter = 0 ; iter < nIterations; iter++) {

        DWORD dwReadError = NO_ERROR;


        IF_DEBUG( ITERATION) {

            DBGPRINTF(( DBG_CONTEXT, "Iteration = %u\n", iter));
        }

        GenUserName( rgchUserName, "getFile", iter);
        hFtp = InternetConnect( hinet, pszServer, 0, "anonymous", rgchUserName,
                               INTERNET_SERVICE_FTP, 0, NULL);
        IF_DEBUG( ENTRY) {

            DBGPRINTF(( DBG_CONTEXT, "    InternetConnect()==> %08x\n",
                       hFtp));
        }

        if ( hFtp == NULL) {

            continue;
        }

        dwReadError = ReadFtpFile( hFtp, pszFileName, sTimeout);

        if ( dwReadError != NO_ERROR) {

            cout << " Read File failed: Error = " << dwReadError << endl;
        }

        IF_DEBUG( ENTRY) {

            DBGPRINTF(( DBG_CONTEXT, "       InternetGetLastResponse()\n"));
        }

        cbBuffer = sizeof(rgchBuffer) - 1;
        if ( InternetGetLastResponseInfo( &dwError, rgchBuffer, &cbBuffer)) {

            IF_DEBUG( RESPONSE) {

                rgchBuffer[cbBuffer] = '\0';
                cout << " ErrorCode = " << dwError
                  << "\tResponse = " << rgchBuffer << endl;
            }
        }

        IF_DEBUG( ENTRY) {

            DBGPRINTF(( DBG_CONTEXT, "    InternetCloseHandle(%08x)\n",
                       hFtp));
        }

        if ( InternetCloseHandle(hFtp) && dwReadError == NO_ERROR) {
            NumSuccess++;
        }
    } // for()

    IF_DEBUG( ENTRY) {

        DBGPRINTF(( DBG_CONTEXT, "InternetCloseHandle(%08x)\n",
                   hinet));
    }

    fReturn2 = InternetCloseHandle(hinet);

    return ( NumSuccess);

} // TestGetFileInOneThread()




BOOL TestGetFile( int argc, char * argv[])
/*++
  This function routinely establishes and throws away connections.
  It does not do any other work.
  This is used for testing logon and quit sequences of FTP server.

  Arguments:
     argc  count of arguments
     argv  arguments
      argv[0] = "get"  -- name of this test function
      argv[1] = # of thread to use for execution
      argv[2] = # of iterations for each thread.
      argv[3] = Name of file/path to get from server.
       (default = /readme.txt)
      argv[4] = Sleep time for enabling test of data transfer timeouts
        Units = seconds
        ( default = 0 ==> do not sleep).

--*/
{
    DWORD NumThreads = DEFAULT_NUMBER_OF_THREADS;
    DWORD NumIterations = DEFAULT_NUMBER_OF_ITERATIONS;
    DWORD NumSuccesses;
    LPSTR pszFileToGet = "/readme.txt";
    DWORD sTimeout = 0;
    int   i;

    i = argc;  //  "i" is the running counter.
    switch ( argc) {

      case 5:
        --i;
        DBG_ASSERT( argv[i] != NULL);
        sTimeout = atoi(argv[i]);

        // Fall Through

      case 4:
        --i;
        DBG_ASSERT( argv[i] != NULL);
        pszFileToGet = argv[i];

        // Fall Through

      case 3:
        --i;
        DBG_ASSERT( argv[i] != NULL);
        NumIterations = atoi(argv[i]);

        // Fall through

      case 2:
        --i;
        DBG_ASSERT( argv[i] != NULL);
        NumThreads = atoi( argv[i]);

        // Fall through

      default:
      case 1: case 0:
        break;
    } // switch

    // We will implement multithreading later on.

    // for now just one thread is used. ( current thread)
    NumSuccesses = TestGetFileInOneThread(g_lpszServerAddress,
                                          pszFileToGet,
                                          NumIterations,
                                          sTimeout);

    cout << " Tested GetFile for Iterations: " << NumIterations << endl;
    cout << "   Timeout for Sleep = " << sTimeout << " seconds" << endl;
    cout << "   Successes = " << NumSuccesses;
    cout << "   Failures  = " << (NumIterations - NumSuccesses) << endl;

    return ( NumSuccesses == NumIterations);

} // TestGetFile()



int __cdecl
main( int argc, char * argv[])
{

    DWORD err = NO_ERROR;
    char ** ppszArgv;       // arguments for command functions
    int     cArgs;           // arg count for command functions
    char *  pszCmdName;
    CmdStruct  * pCmd;
    CMDFUNC pCmdFunc = NULL;

#ifndef _NO_TRACING_
    CREATE_DEBUG_PRINT_OBJECT( "ftpstress", IisFtptestGuid);
    CREATE_INITIALIZE_DEBUG();
#else
    CREATE_DEBUG_PRINT_OBJECT( "ftpstress");
    SET_DEBUG_FLAGS( DEBUG_ITERATION);
#endif

    if ( argc < 3 || argv[1] == NULL ) {

      // Insufficient arguments
       GenUsageMessage( argc, argv);
       return ( 1);
    }

    pszCmdName = argv[2];
    if (( pCmd = DecodeCommand( pszCmdName)) == NULL
        || pCmd->cmdFunc == NULL) {
        printf( "Internal Error: Invalid Command %s\n", pszCmdName);
        GenUsageMessage( argc, argv);
        return ( 1);
    }

    g_lpszServerAddress = argv[1];   // get server address
    cArgs = argc - 2;
    ppszArgv = argv + 2;

    if ( !(*pCmd->cmdFunc)( cArgs, ppszArgv)) {     // call the test function

        // Test function failed.
        printf( "Command %s failed. Error = %d\n", pszCmdName, GetLastError());
        return ( 1);
    }

    printf( " Command %s succeeded\n", pszCmdName);

    DELETE_DEBUG_PRINT_OBJECT();

    return ( 0);        // success
} // main()




/************************ End of File ***********************/