[wxWidgets.git] / tests / streams / largefile.cpp

///////////////////////////////////////////////////////////////////////////////
// Name:        tests/streams/largefile.cpp
// Purpose:     Tests for large file support
// Author:      Mike Wetherell
// RCS-ID:      $Id$
// Copyright:   (c) 2004 Mike Wetherell
// Licence:     wxWindows licence
///////////////////////////////////////////////////////////////////////////////

//
// We're interested in what happens around offsets 0x7fffffff and 0xffffffff
// so the test writes a small chunk of test data just before and just after
// these offsets, then reads them back.
//
// The tests can be run with:
//
//  test --verbose largeFile
//
// On systems supporting sparse files they will also be registered in the
// Streams subsuite so that they run by default.
//

// For compilers that support precompilation, includes "wx/wx.h".
#include "testprec.h"

#ifdef __BORLANDC__
    #pragma hdrstop
#endif

// for all others, include the necessary headers
#ifndef WX_PRECOMP
    #include "wx/wx.h"
#endif

#include "wx/filename.h"
#include "wx/wfstream.h"

#ifdef __WXMSW__
    #include "wx/msw/wrapwin.h"
    #ifdef __VISUALC__
        // 'nonstandard extension used : nameless struct/union' occurs inside
        // winioctl.h
        #pragma warning(disable:4201)
    #endif
    #include <winioctl.h>
    #ifdef __VISUALC__
        #pragma warning(default:4201)
    #endif
#endif

#ifdef __VISUALC__
    #define fileno _fileno
#endif

using std::auto_ptr;


///////////////////////////////////////////////////////////////////////////////
// Helpers

bool IsFAT(const wxString& path);
void MakeSparse(const wxString& path, int fd);


///////////////////////////////////////////////////////////////////////////////
// Base class for the test cases - common code

class LargeFileTest : public CppUnit::TestCase
{
public:
    LargeFileTest(std::string name) : CppUnit::TestCase(name) { }
    virtual ~LargeFileTest() { }

protected:
    void runTest();

    virtual wxInputStream *MakeInStream(const wxString& name) const = 0;
    virtual wxOutputStream *MakeOutStream(const wxString& name) const = 0;
    virtual bool HasLFS() const = 0;
};

void LargeFileTest::runTest()
{
    // self deleting temp file
    struct TmpFile {
        TmpFile() : m_name(wxFileName::CreateTempFileName(wxT("wxlfs-"))) { }
        ~TmpFile() { if (!m_name.empty()) wxRemoveFile(m_name); }
        wxString m_name;
    } tmpfile;

    CPPUNIT_ASSERT(!tmpfile.m_name.empty());

    bool haveLFS = true;
    bool fourGigLimit = false;

    if (!HasLFS()) {
        haveLFS = false;
        wxString n(getName().c_str(), *wxConvCurrent);
        wxLogInfo(n + wxT(": No large file support, testing up to 2GB only"));
    }
    else if (IsFAT(tmpfile.m_name)) {
        fourGigLimit = true;
        wxString n(getName().c_str(), *wxConvCurrent);
        wxLogInfo(n + wxT(": FAT volumes are limited to 4GB files"));
    }

    // size of the test blocks
    const size_t size = 0x40;

    // the test blocks
    char upto2Gig[size];
    char past2Gig[size];
    char upto4Gig[size];
    char past4Gig[size];
    memset(upto2Gig, 'A', size);
    memset(past2Gig, 'B', size);
    memset(upto4Gig, 'X', size);
    memset(past4Gig, 'Y', size);

    wxFileOffset pos;

    // write a large file
    {
        auto_ptr<wxOutputStream> out(MakeOutStream(tmpfile.m_name));

        // write 'A's at [ 0x7fffffbf, 0x7fffffff [
        pos = 0x7fffffff - size;
        CPPUNIT_ASSERT(out->SeekO(pos) == pos);
        CPPUNIT_ASSERT(out->Write(upto2Gig, size).LastWrite() == size);
        pos += size;

        if (haveLFS) {
            // write 'B's at [ 0x7fffffff, 0x8000003f [
            CPPUNIT_ASSERT(out->Write(past2Gig, size).LastWrite() == size);
            pos += size;
            CPPUNIT_ASSERT(out->TellO() == pos);

            // write 'X's at [ 0xffffffbf, 0xffffffff [
            pos = 0xffffffff - size;
            CPPUNIT_ASSERT(out->SeekO(pos) == pos);
            CPPUNIT_ASSERT(out->Write(upto4Gig, size).LastWrite() == size);
            pos += size;

            if (!fourGigLimit) {
                // write 'Y's at [ 0xffffffff, 0x10000003f [
                CPPUNIT_ASSERT(out->Write(past4Gig, size).LastWrite() == size);
                pos += size;
            }
        }

        // check the file seems to be the right length
        CPPUNIT_ASSERT(out->TellO() == pos);
        CPPUNIT_ASSERT(out->GetLength() == pos);
    }

    // read the large file back
    {
        auto_ptr<wxInputStream> in(MakeInStream(tmpfile.m_name));
        char buf[size];

        if (haveLFS) {
            CPPUNIT_ASSERT(in->GetLength() == pos);
            pos = 0xffffffff;

            if (!fourGigLimit) {
                CPPUNIT_ASSERT(in->GetLength() > pos);

                // read back the 'Y's at [ 0xffffffff, 0x10000003f [
                CPPUNIT_ASSERT(in->SeekI(pos) == pos);
                CPPUNIT_ASSERT(in->Read(buf, size).LastRead() == size);
                CPPUNIT_ASSERT(memcmp(buf, past4Gig, size) == 0);

                CPPUNIT_ASSERT(in->TellI() == in->GetLength());
            }

            // read back the 'X's at [ 0xffffffbf, 0xffffffff [
            pos -= size;
            CPPUNIT_ASSERT(in->SeekI(pos) == pos);
            CPPUNIT_ASSERT(in->Read(buf, size).LastRead() == size);
            CPPUNIT_ASSERT(memcmp(buf, upto4Gig, size) == 0);
            pos += size;
            CPPUNIT_ASSERT(in->TellI() == pos);

            // read back the 'B's at [ 0x7fffffff, 0x8000003f [
            pos = 0x7fffffff;
            CPPUNIT_ASSERT(in->SeekI(pos) == pos);
            CPPUNIT_ASSERT(in->Read(buf, size).LastRead() == size);
            CPPUNIT_ASSERT(memcmp(buf, past2Gig, size) == 0);
        }
        else {
            CPPUNIT_ASSERT(in->GetLength() == 0x7fffffff);
            pos = 0x7fffffff;
        }

        // read back the 'A's at [ 0x7fffffbf, 0x7fffffff [
        pos -= size;
        CPPUNIT_ASSERT(in->SeekI(pos) == pos);
        CPPUNIT_ASSERT(in->Read(buf, size).LastRead() == size);
        CPPUNIT_ASSERT(memcmp(buf, upto2Gig, size) == 0);
        pos += size;
        CPPUNIT_ASSERT(in->TellI() == pos);
    }
}


///////////////////////////////////////////////////////////////////////////////
// wxFile test case

class LargeFileTest_wxFile : public LargeFileTest
{
public:
    LargeFileTest_wxFile() : LargeFileTest("wxFile streams") { }

protected:
    wxInputStream *MakeInStream(const wxString& name) const;
    wxOutputStream *MakeOutStream(const wxString& name) const;
    bool HasLFS() const { return (wxFileOffset)0xffffffff > 0; }
};

wxInputStream *LargeFileTest_wxFile::MakeInStream(const wxString& name) const
{
    auto_ptr<wxFileInputStream> in(new wxFileInputStream(name));
    CPPUNIT_ASSERT(in->Ok());
    return in.release();
}

wxOutputStream *LargeFileTest_wxFile::MakeOutStream(const wxString& name) const
{
    wxFile file(name, wxFile::write);
    CPPUNIT_ASSERT(file.IsOpened());
    int fd = file.fd();
    file.Detach();
    MakeSparse(name, fd);
    return new wxFileOutputStream(fd);
}


///////////////////////////////////////////////////////////////////////////////
// wxFFile test case

class LargeFileTest_wxFFile : public LargeFileTest
{
public:
    LargeFileTest_wxFFile() : LargeFileTest("wxFFile streams") { }

protected:
    wxInputStream *MakeInStream(const wxString& name) const;
    wxOutputStream *MakeOutStream(const wxString& name) const;
    bool HasLFS() const;
};

wxInputStream *LargeFileTest_wxFFile::MakeInStream(const wxString& name) const
{
    auto_ptr<wxFFileInputStream> in(new wxFFileInputStream(name));
    CPPUNIT_ASSERT(in->Ok());
    return in.release();
}

wxOutputStream *LargeFileTest_wxFFile::MakeOutStream(const wxString& name) const
{
    wxFFile file(name, wxT("w"));
    CPPUNIT_ASSERT(file.IsOpened());
    FILE *fp = file.fp();
    file.Detach();
    MakeSparse(name, fileno(fp));
    return new wxFFileOutputStream(fp);
}

bool LargeFileTest_wxFFile::HasLFS() const
{
#ifdef wxHAS_LARGE_FFILES
    return true;
#else
    return false;
#endif
}


///////////////////////////////////////////////////////////////////////////////
// The suite

class largeFile : public CppUnit::TestSuite
{
public:
    largeFile() : CppUnit::TestSuite("largeFile") { }

    static CppUnit::Test *suite();
};

CppUnit::Test *largeFile::suite()
{
    largeFile *suite = new largeFile;

    suite->addTest(new LargeFileTest_wxFile);
    suite->addTest(new LargeFileTest_wxFFile);

    return suite;
}


///////////////////////////////////////////////////////////////////////////////
// Implement the helpers
//
// Ideally these tests will be part of the default suite so that regressions
// are picked up. However this is only possible when sparse files are
// supported otherwise the tests require too much disk space.

#ifdef __WXMSW__

#ifndef FILE_SUPPORTS_SPARSE_FILES
#define FILE_SUPPORTS_SPARSE_FILES 0x00000040
#endif

#ifndef FSCTL_SET_SPARSE

#   ifndef FILE_SPECIAL_ACCESS
#       define FILE_SPECIAL_ACCESS FILE_ANY_ACCESS
#   endif
#   define FSCTL_SET_SPARSE CTL_CODE(FILE_DEVICE_FILE_SYSTEM, 49, \
                                     METHOD_BUFFERED, FILE_SPECIAL_ACCESS)
#endif

static DWORD  volumeFlags;
static wxChar volumeType[64];
static bool   volumeInfoInit;

void GetVolumeInfo(const wxString& path)
{
    // extract the volume 'C:\' or '\\tooter\share\' from the path
    wxString vol;

    if (path.substr(1, 2) == wxT(":\\")) {
        vol = path.substr(0, 3);
    } else {
        if (path.substr(0, 2) == wxT("\\\\")) {
            size_t i = path.find(wxT('\\'), 2);

            if (i != wxString::npos && i > 2) {
                size_t j = path.find(wxT('\\'), ++i);

                if (j != i)
                    vol = path.substr(0, j) + wxT("\\");
            }
        }
    }

    // NULL means the current volume
    const wxChar *pVol = vol.empty() ? (const wxChar *)NULL
                                     : vol.c_str();

    if (!::GetVolumeInformation(pVol, NULL, 0, NULL, NULL,
                                &volumeFlags,
                                volumeType,
                                WXSIZEOF(volumeType)))
    {
        wxLogSysError(wxT("GetVolumeInformation() failed"));
    }

    volumeInfoInit = true;
}

bool IsFAT(const wxString& path)
{
    if (!volumeInfoInit)
        GetVolumeInfo(path);
    return wxString(volumeType).Upper().find(wxT("FAT")) != wxString::npos;
}

void MakeSparse(const wxString& path, int fd)
{
    DWORD cb;

    if (!volumeInfoInit)
        GetVolumeInfo(path);

    if ((volumeFlags & FILE_SUPPORTS_SPARSE_FILES) != 0)
        if (!::DeviceIoControl((HANDLE)_get_osfhandle(fd),
                               FSCTL_SET_SPARSE,
                               NULL, 0, NULL, 0, &cb, NULL))
            volumeFlags &= ~FILE_SUPPORTS_SPARSE_FILES;
}

// return the suite if sparse files are supported, otherwise return NULL
//
CppUnit::Test* GetlargeFileSuite()
{
    if (!volumeInfoInit) {
        wxString path;
        {
            wxFile file;
            path = wxFileName::CreateTempFileName(wxT("wxlfs-"), &file);
            MakeSparse(path, file.fd());
        }
        wxRemoveFile(path);
    }

    if ((volumeFlags & FILE_SUPPORTS_SPARSE_FILES) != 0)
        return largeFile::suite();
    else
        return NULL;
}

#else // __WXMSW__

bool IsFAT(const wxString& WXUNUSED(path)) { return false; }
void MakeSparse(const wxString& WXUNUSED(path), int WXUNUSED(fd)) { }

// return the suite if sparse files are supported, otherwise return NULL
//
CppUnit::Test* GetlargeFileSuite()
{
    wxString path;
    struct stat st1, st2;
    memset(&st1, 0, sizeof(st1));
    memset(&st2, 0, sizeof(st2));

    {
        wxFile file;
        path = wxFileName::CreateTempFileName(wxT("wxlfs-"), &file);

        fstat(file.fd(), &st1);
        file.Seek(st1.st_blksize);
        file.Write("x", 1);
        fstat(file.fd(), &st1);

        file.Seek(0);
        file.Write("x", 1);
        fstat(file.fd(), &st2);
    }

    wxRemoveFile(path);

    if (st1.st_blocks != st2.st_blocks)
        return largeFile::suite();
    else
        return NULL;
}

#endif // __WXMSW__

CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(largeFile, "largeFile");
CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(largeFile, "Streams.largeFile");
Commit	Line	Data
48714f74 RN	1	///////////////////////////////////////////////////////////////////////////////
	2	// Name: tests/streams/largefile.cpp
	3	// Purpose: Tests for large file support
	4	// Author: Mike Wetherell
	5	// RCS-ID: $Id$
	6	// Copyright: (c) 2004 Mike Wetherell
526954c5	7	// Licence: wxWindows licence
48714f74 RN	8	///////////////////////////////////////////////////////////////////////////////
	9
	10	//
086d7f31	11	// We're interested in what happens around offsets 0x7fffffff and 0xffffffff
48714f74 RN	12	// so the test writes a small chunk of test data just before and just after
	13	// these offsets, then reads them back.
	14	//
	15	// The tests can be run with:
	16	//
	17	// test --verbose largeFile
	18	//
9e477451 MW	19	// On systems supporting sparse files they will also be registered in the
9e477451 MW	20	// Streams subsuite so that they run by default.
48714f74 RN	21	//
	22
	23	// For compilers that support precompilation, includes "wx/wx.h".
	24	#include "testprec.h"
	25
	26	#ifdef __BORLANDC__
	27	#pragma hdrstop
	28	#endif
	29
	30	// for all others, include the necessary headers
	31	#ifndef WX_PRECOMP
	32	#include "wx/wx.h"
	33	#endif
	34
	35	#include "wx/filename.h"
	36	#include "wx/wfstream.h"
	37
	38	#ifdef __WXMSW__
845e4f9b VZ	39	#include "wx/msw/wrapwin.h"
	40	#ifdef __VISUALC__
	41	// 'nonstandard extension used : nameless struct/union' occurs inside
	42	// winioctl.h
	43	#pragma warning(disable:4201)
	44	#endif
	45	#include <winioctl.h>
	46	#ifdef __VISUALC__
	47	#pragma warning(default:4201)
	48	#endif
48714f74 RN	49	#endif
48714f74 RN	50
d4b17832 VZ	51	#ifdef __VISUALC__
	52	#define fileno _fileno
	53	#endif
	54
48714f74 RN	55	using std::auto_ptr;
	56
	57
	58	///////////////////////////////////////////////////////////////////////////////
	59	// Helpers
	60
	61	bool IsFAT(const wxString& path);
	62	void MakeSparse(const wxString& path, int fd);
	63
	64
	65	///////////////////////////////////////////////////////////////////////////////
	66	// Base class for the test cases - common code
	67
	68	class LargeFileTest : public CppUnit::TestCase
	69	{
	70	public:
	71	LargeFileTest(std::string name) : CppUnit::TestCase(name) { }
	72	virtual ~LargeFileTest() { }
	73
	74	protected:
	75	void runTest();
	76
	77	virtual wxInputStream *MakeInStream(const wxString& name) const = 0;
	78	virtual wxOutputStream *MakeOutStream(const wxString& name) const = 0;
	79	virtual bool HasLFS() const = 0;
	80	};
	81
	82	void LargeFileTest::runTest()
	83	{
	84	// self deleting temp file
	85	struct TmpFile {
9a83f860	86	TmpFile() : m_name(wxFileName::CreateTempFileName(wxT("wxlfs-"))) { }
48714f74 RN	87	~TmpFile() { if (!m_name.empty()) wxRemoveFile(m_name); }
	88	wxString m_name;
	89	} tmpfile;
	90
	91	CPPUNIT_ASSERT(!tmpfile.m_name.empty());
	92
	93	bool haveLFS = true;
	94	bool fourGigLimit = false;
	95
	96	if (!HasLFS()) {
	97	haveLFS = false;
	98	wxString n(getName().c_str(), *wxConvCurrent);
9a83f860	99	wxLogInfo(n + wxT(": No large file support, testing up to 2GB only"));
48714f74 RN	100	}
	101	else if (IsFAT(tmpfile.m_name)) {
	102	fourGigLimit = true;
	103	wxString n(getName().c_str(), *wxConvCurrent);
9a83f860	104	wxLogInfo(n + wxT(": FAT volumes are limited to 4GB files"));
48714f74 RN	105	}
	106
	107	// size of the test blocks
	108	const size_t size = 0x40;
	109
	110	// the test blocks
	111	char upto2Gig[size];
	112	char past2Gig[size];
	113	char upto4Gig[size];
	114	char past4Gig[size];
	115	memset(upto2Gig, 'A', size);
	116	memset(past2Gig, 'B', size);
	117	memset(upto4Gig, 'X', size);
	118	memset(past4Gig, 'Y', size);
	119
	120	wxFileOffset pos;
	121
	122	// write a large file
	123	{
	124	auto_ptr<wxOutputStream> out(MakeOutStream(tmpfile.m_name));
43b2d5e7	125
48714f74 RN	126	// write 'A's at [ 0x7fffffbf, 0x7fffffff [
	127	pos = 0x7fffffff - size;
	128	CPPUNIT_ASSERT(out->SeekO(pos) == pos);
	129	CPPUNIT_ASSERT(out->Write(upto2Gig, size).LastWrite() == size);
	130	pos += size;
	131
	132	if (haveLFS) {
	133	// write 'B's at [ 0x7fffffff, 0x8000003f [
	134	CPPUNIT_ASSERT(out->Write(past2Gig, size).LastWrite() == size);
	135	pos += size;
	136	CPPUNIT_ASSERT(out->TellO() == pos);
	137
	138	// write 'X's at [ 0xffffffbf, 0xffffffff [
	139	pos = 0xffffffff - size;
	140	CPPUNIT_ASSERT(out->SeekO(pos) == pos);
	141	CPPUNIT_ASSERT(out->Write(upto4Gig, size).LastWrite() == size);
	142	pos += size;
	143
	144	if (!fourGigLimit) {
	145	// write 'Y's at [ 0xffffffff, 0x10000003f [
	146	CPPUNIT_ASSERT(out->Write(past4Gig, size).LastWrite() == size);
	147	pos += size;
	148	}
	149	}
	150
	151	// check the file seems to be the right length
	152	CPPUNIT_ASSERT(out->TellO() == pos);
	153	CPPUNIT_ASSERT(out->GetLength() == pos);
	154	}
	155
	156	// read the large file back
	157	{
	158	auto_ptr<wxInputStream> in(MakeInStream(tmpfile.m_name));
	159	char buf[size];
	160
	161	if (haveLFS) {
	162	CPPUNIT_ASSERT(in->GetLength() == pos);
	163	pos = 0xffffffff;
	164
	165	if (!fourGigLimit) {
	166	CPPUNIT_ASSERT(in->GetLength() > pos);
	167
	168	// read back the 'Y's at [ 0xffffffff, 0x10000003f [
	169	CPPUNIT_ASSERT(in->SeekI(pos) == pos);
	170	CPPUNIT_ASSERT(in->Read(buf, size).LastRead() == size);
	171	CPPUNIT_ASSERT(memcmp(buf, past4Gig, size) == 0);
	172
	173	CPPUNIT_ASSERT(in->TellI() == in->GetLength());
	174	}
	175
	176	// read back the 'X's at [ 0xffffffbf, 0xffffffff [
	177	pos -= size;
	178	CPPUNIT_ASSERT(in->SeekI(pos) == pos);
	179	CPPUNIT_ASSERT(in->Read(buf, size).LastRead() == size);
	180	CPPUNIT_ASSERT(memcmp(buf, upto4Gig, size) == 0);
	181	pos += size;
	182	CPPUNIT_ASSERT(in->TellI() == pos);
	183
	184	// read back the 'B's at [ 0x7fffffff, 0x8000003f [
	185	pos = 0x7fffffff;
	186	CPPUNIT_ASSERT(in->SeekI(pos) == pos);
	187	CPPUNIT_ASSERT(in->Read(buf, size).LastRead() == size);
	188	CPPUNIT_ASSERT(memcmp(buf, past2Gig, size) == 0);
	189	}
190	else {
191	CPPUNIT_ASSERT(in->GetLength() == 0x7fffffff);
192	pos = 0x7fffffff;
193	}
194
195	// read back the 'A's at [ 0x7fffffbf, 0x7fffffff [
196	pos -= size;
197	CPPUNIT_ASSERT(in->SeekI(pos) == pos);
198	CPPUNIT_ASSERT(in->Read(buf, size).LastRead() == size);
199	CPPUNIT_ASSERT(memcmp(buf, upto2Gig, size) == 0);
200	pos += size;
201	CPPUNIT_ASSERT(in->TellI() == pos);
202	}
203	}
204
205
206	///////////////////////////////////////////////////////////////////////////////
207	// wxFile test case
208
209	class LargeFileTest_wxFile : public LargeFileTest
210	{
211	public:
212	LargeFileTest_wxFile() : LargeFileTest("wxFile streams") { }
213
214	protected:
215	wxInputStream *MakeInStream(const wxString& name) const;
216	wxOutputStream *MakeOutStream(const wxString& name) const;
217	bool HasLFS() const { return (wxFileOffset)0xffffffff > 0; }
218	};
219
220	wxInputStream *LargeFileTest_wxFile::MakeInStream(const wxString& name) const
221	{
222	auto_ptr<wxFileInputStream> in(new wxFileInputStream(name));
223	CPPUNIT_ASSERT(in->Ok());
224	return in.release();
225	}
226
227	wxOutputStream *LargeFileTest_wxFile::MakeOutStream(const wxString& name) const
228	{
229	wxFile file(name, wxFile::write);
230	CPPUNIT_ASSERT(file.IsOpened());
231	int fd = file.fd();
232	file.Detach();
233	MakeSparse(name, fd);
234	return new wxFileOutputStream(fd);
235	}
236
237
238	///////////////////////////////////////////////////////////////////////////////
239	// wxFFile test case
240
241	class LargeFileTest_wxFFile : public LargeFileTest
242	{
243	public:
244	LargeFileTest_wxFFile() : LargeFileTest("wxFFile streams") { }
245
246	protected:
247	wxInputStream *MakeInStream(const wxString& name) const;
248	wxOutputStream *MakeOutStream(const wxString& name) const;
249	bool HasLFS() const;
250	};
251
252	wxInputStream *LargeFileTest_wxFFile::MakeInStream(const wxString& name) const
253	{
254	auto_ptr<wxFFileInputStream> in(new wxFFileInputStream(name));
255	CPPUNIT_ASSERT(in->Ok());
256	return in.release();
257	}
258
259	wxOutputStream *LargeFileTest_wxFFile::MakeOutStream(const wxString& name) const
260	{
9a83f860	261	wxFFile file(name, wxT("w"));
48714f74 RN	262	CPPUNIT_ASSERT(file.IsOpened());
	263	FILE *fp = file.fp();
	264	file.Detach();
	265	MakeSparse(name, fileno(fp));
	266	return new wxFFileOutputStream(fp);
	267	}
	268
	269	bool LargeFileTest_wxFFile::HasLFS() const
	270	{
224d978f MW	271	#ifdef wxHAS_LARGE_FFILES
224d978f MW	272	return true;
48714f74 RN	273	#else
	274	return false;
	275	#endif
	276	}
	277
	278
	279	///////////////////////////////////////////////////////////////////////////////
	280	// The suite
	281
	282	class largeFile : public CppUnit::TestSuite
	283	{
	284	public:
	285	largeFile() : CppUnit::TestSuite("largeFile") { }
	286
	287	static CppUnit::Test *suite();
	288	};
	289
	290	CppUnit::Test *largeFile::suite()
	291	{
	292	largeFile *suite = new largeFile;
	293
	294	suite->addTest(new LargeFileTest_wxFile);
	295	suite->addTest(new LargeFileTest_wxFFile);
	296
	297	return suite;
	298	}
	299
	300
	301	///////////////////////////////////////////////////////////////////////////////
	302	// Implement the helpers
	303	//
	304	// Ideally these tests will be part of the default suite so that regressions
	305	// are picked up. However this is only possible when sparse files are
	306	// supported otherwise the tests require too much disk space.
48714f74 RN	307
	308	#ifdef __WXMSW__
	309
	310	#ifndef FILE_SUPPORTS_SPARSE_FILES
	311	#define FILE_SUPPORTS_SPARSE_FILES 0x00000040
	312	#endif
	313
	314	#ifndef FSCTL_SET_SPARSE
	315
43b2d5e7	316	# ifndef FILE_SPECIAL_ACCESS
48714f74 RN	317	# define FILE_SPECIAL_ACCESS FILE_ANY_ACCESS
	318	# endif
	319	# define FSCTL_SET_SPARSE CTL_CODE(FILE_DEVICE_FILE_SYSTEM, 49, \
	320	METHOD_BUFFERED, FILE_SPECIAL_ACCESS)
	321	#endif
	322
	323	static DWORD volumeFlags;
	324	static wxChar volumeType[64];
	325	static bool volumeInfoInit;
	326
	327	void GetVolumeInfo(const wxString& path)
	328	{
	329	// extract the volume 'C:\' or '\\tooter\share\' from the path
	330	wxString vol;
43b2d5e7	331
9a83f860	332	if (path.substr(1, 2) == wxT(":\\")) {
48714f74 RN	333	vol = path.substr(0, 3);
48714f74 RN	334	} else {
9a83f860 VZ	335	if (path.substr(0, 2) == wxT("\\\\")) {
9a83f860 VZ	336	size_t i = path.find(wxT('\\'), 2);
48714f74 RN	337
48714f74 RN	338	if (i != wxString::npos && i > 2) {
9a83f860	339	size_t j = path.find(wxT('\\'), ++i);
48714f74 RN	340
48714f74 RN	341	if (j != i)
9a83f860	342	vol = path.substr(0, j) + wxT("\\");
48714f74 RN	343	}
	344	}
	345	}
	346
	347	// NULL means the current volume
90adb904	348	const wxChar pVol = vol.empty() ? (const wxChar )NULL
48eba629	349	: vol.c_str();
48714f74 RN	350
48714f74 RN	351	if (!::GetVolumeInformation(pVol, NULL, 0, NULL, NULL,
43b2d5e7	352	&volumeFlags,
48714f74 RN	353	volumeType,
48714f74 RN	354	WXSIZEOF(volumeType)))
43b2d5e7	355	{
9a83f860	356	wxLogSysError(wxT("GetVolumeInformation() failed"));
43b2d5e7	357	}
48714f74 RN	358
	359	volumeInfoInit = true;
	360	}
	361
	362	bool IsFAT(const wxString& path)
	363	{
	364	if (!volumeInfoInit)
	365	GetVolumeInfo(path);
9a83f860	366	return wxString(volumeType).Upper().find(wxT("FAT")) != wxString::npos;
48714f74 RN	367	}
	368
	369	void MakeSparse(const wxString& path, int fd)
	370	{
	371	DWORD cb;
	372
	373	if (!volumeInfoInit)
	374	GetVolumeInfo(path);
43b2d5e7	375
48714f74 RN	376	if ((volumeFlags & FILE_SUPPORTS_SPARSE_FILES) != 0)
	377	if (!::DeviceIoControl((HANDLE)_get_osfhandle(fd),
	378	FSCTL_SET_SPARSE,
	379	NULL, 0, NULL, 0, &cb, NULL))
086d7f31	380	volumeFlags &= ~FILE_SUPPORTS_SPARSE_FILES;
48714f74 RN	381	}
48714f74 RN	382
9e477451 MW	383	// return the suite if sparse files are supported, otherwise return NULL
9e477451 MW	384	//
48714f74 RN	385	CppUnit::Test* GetlargeFileSuite()
	386	{
	387	if (!volumeInfoInit) {
9e477451 MW	388	wxString path;
	389	{
	390	wxFile file;
	391	path = wxFileName::CreateTempFileName(wxT("wxlfs-"), &file);
	392	MakeSparse(path, file.fd());
	393	}
48714f74 RN	394	wxRemoveFile(path);
	395	}
	396
	397	if ((volumeFlags & FILE_SUPPORTS_SPARSE_FILES) != 0)
	398	return largeFile::suite();
	399	else
	400	return NULL;
	401	}
	402
	403	#else // __WXMSW__
	404
	405	bool IsFAT(const wxString& WXUNUSED(path)) { return false; }
	406	void MakeSparse(const wxString& WXUNUSED(path), int WXUNUSED(fd)) { }
	407
9e477451 MW	408	// return the suite if sparse files are supported, otherwise return NULL
	409	//
	410	CppUnit::Test* GetlargeFileSuite()
	411	{
	412	wxString path;
	413	struct stat st1, st2;
	414	memset(&st1, 0, sizeof(st1));
	415	memset(&st2, 0, sizeof(st2));
	416
	417	{
	418	wxFile file;
	419	path = wxFileName::CreateTempFileName(wxT("wxlfs-"), &file);
	420
	421	fstat(file.fd(), &st1);
	422	file.Seek(st1.st_blksize);
	423	file.Write("x", 1);
	424	fstat(file.fd(), &st1);
	425
	426	file.Seek(0);
	427	file.Write("x", 1);
	428	fstat(file.fd(), &st2);
	429	}
	430
	431	wxRemoveFile(path);
	432
	433	if (st1.st_blocks != st2.st_blocks)
	434	return largeFile::suite();
	435	else
	436	return NULL;
	437	}
48714f74 RN	438
	439	#endif // __WXMSW__
	440
	441	CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(largeFile, "largeFile");
	442	CPPUNIT_TEST_SUITE_NAMED_REGISTRATION(largeFile, "Streams.largeFile");