// Regression test for STL seekg

#include <cstdio>       // I really like formatted output.

#include <cstdarg>

#include <fstream>

#include <vector>

#include<iostream>

#ifndef cstd

#ifdef _MSC_VER

#define cstd

#else

//#define cstd std

#define cstd

using namespace std;

#endif

#endif

#ifndef __SYMBIAN32__

#ifndef USING_STLPORT

#define vsnprintf _vsnprintf

#endif

#endif

class Error : public std::exception

{

 public:

    Error(const char *pszfmt, ...);

    const char *what() const;

 private:

    char mMessage[256];

};

#define countof(x) (sizeof(x)/sizeof((x)[0]))

Error::Error(const char *pszfmt, ...)

{

    cstd::va_list ap;

    va_start(ap, pszfmt);

    cstd::vsnprintf(mMessage, countof(mMessage), pszfmt, ap);

    va_end(ap);

}

const char *Error::what() const

{

    return mMessage;

}

// usage: testseek filename

#ifdef __SYMBIAN32__

int seek_test(int, char**)

#else

int main(int argc, char **argv)

#endif

{

    try

    {

        #ifndef __SYMBIAN32__

         if (argc != 2) {

            throw Error ("Wrong number of arguments (Usage: testseek filename)");

               }

        #endif

         #ifndef __SYMBIAN32__    

        std::ifstream ifs(argv[1]);

         ifs.open(argv[1]);

       #else

       std::ifstream ifs;

         ifs.open("c:\\testframework\\tstdcpp\\seek.txt");

       #endif

        if (!ifs) {

            #ifdef __SYMBIAN32__

               Error  e1("Failed to open file \"%s\"", "c:\\testframework\\tstdcpp\\seek.txt");

                throw e1;

            #else       

            throw Error("Failed to open file \"%s\"", argv[1]);

            #endif

        }

        ifs.exceptions(std::ios::badbit);

        // First pass: read contents of file into character vector

        std::vector<char> charvec;

        cstd::printf("First pass:\n");

        for(;;)

        {

            char ch;

            if (!ifs.get(ch)) {

                break;

            }

            cstd::printf("%c", ch);

            charvec.push_back(ch);

        }

        printf("First pass done\n");

        // Second pass: get positions into pos vector, checking chars

        std::vector<std::ios::pos_type> posvec;

        ifs.clear();    // necessary?

        ifs.seekg(0);

        std::vector<char>::const_iterator char_it;

        printf("Second pass:\n");

        for(char_it = charvec.begin(); char_it != charvec.end(); ++char_it)

        {

            std::ios::pos_type pos = ifs.tellg();

            posvec.push_back(pos);

            char ch;

            if (!ifs.get(ch)) {

                break;

            }

            cstd::printf("%c", ch);

            if (ch != *char_it) {

               #ifdef __SYMBIAN32__

                Error e1("Character mismatch: got '%c', expected '%c'",ch, *char_it);

                 throw e1;

                #else 

                throw Error("Character mismatch: got '%c', expected '%c'",

                            ch, *char_it);

                #endif

            }

        }

        if (char_it != charvec.end()) {

        	  #ifdef __SYMBIAN32__

            Error e1("Unexpected end of file");

            throw e1;

            #else

            throw Error("Unexpected end of file");

            #endif

          }

        printf("Second pass done\n");

        // Third pass: seek to each saved position and read next char

        ifs.clear();    // necessary, but I don't like it

        std::vector<std::ios::pos_type>::const_iterator pos_it;

        for(pos_it = posvec.begin();

            pos_it != posvec.end(); ++pos_it)

        {

            printf ("%d ", (int)*pos_it);

        }

        printf ("\n");

        printf("Third pass:\n");

        for(char_it = charvec.begin(), pos_it = posvec.begin();

            char_it != charvec.end(); ++char_it, ++pos_it)

        {

            ifs.seekg(*pos_it);

            char ch;

            if (!ifs.get(ch)) {

            	#ifdef __SYMBIAN32__

                 Error e1("End of file when expecting '%c'", *char_it);

                 throw e1;

              #else

                throw Error("End of file when expecting '%c'", *char_it);

                #endif

            }

            cstd::printf("%c", ch);

            if (ch != *char_it) {

            	#ifdef __SYMBIAN32__

                 Error e1("Character mismatch: got '%c', expected '%c'",ch, *char_it);

                 throw e1;

              #else

                throw Error("Character mismatch: got '%c', expected '%c'",ch, *char_it);

                #endif

            }

        }

        printf("Third pass done\n");

        printf("Test passed\n");

    }

    catch (std::exception &e)

    {

        cstd::printf("\nCaught exception: %s\n", e.what());

        cstd::printf("Test failed!\n");

        return 1;

    }

    return 0;

}