gecko-dev/xpcom/tests/TestStorageStream.cpp

111 lines
2.2 KiB
C++
Raw Normal View History

2012-05-21 11:12:37 +00:00
/* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
2004-09-29 16:48:10 +00:00
2004-09-29 20:24:53 +00:00
#include <stdlib.h>
2004-09-29 16:48:10 +00:00
#include "nsIStorageStream.h"
#include "nsIInputStream.h"
#include "nsIOutputStream.h"
#include "nsCOMPtr.h"
int main()
{
2004-09-29 20:24:53 +00:00
char kData[4096];
memset(kData, 0, sizeof(kData));
2004-09-29 16:48:10 +00:00
nsresult rv;
nsCOMPtr<nsIStorageStream> stor;
rv = NS_NewStorageStream(4096, UINT32_MAX, getter_AddRefs(stor));
2004-09-29 16:48:10 +00:00
if (NS_FAILED(rv))
return -1;
nsCOMPtr<nsIOutputStream> out;
rv = stor->GetOutputStream(0, getter_AddRefs(out));
if (NS_FAILED(rv))
return -1;
uint32_t n;
2004-09-29 16:48:10 +00:00
rv = out->Write(kData, sizeof(kData), &n);
if (NS_FAILED(rv))
return -1;
rv = out->Write(kData, sizeof(kData), &n);
if (NS_FAILED(rv))
return -1;
rv = out->Close();
if (NS_FAILED(rv))
return -1;
out = nullptr;
2004-09-29 16:48:10 +00:00
nsCOMPtr<nsIInputStream> in;
rv = stor->NewInputStream(0, getter_AddRefs(in));
if (NS_FAILED(rv))
return -1;
char buf[4096];
// consume contents of input stream
do {
rv = in->Read(buf, sizeof(buf), &n);
if (NS_FAILED(rv))
return -1;
} while (n != 0);
rv = in->Close();
if (NS_FAILED(rv))
return -1;
in = nullptr;
2004-09-29 16:48:10 +00:00
// now, write 3 more full 4k segments + 11 bytes, starting at 8192
// total written equals 20491 bytes
rv = stor->GetOutputStream(8192, getter_AddRefs(out));
if (NS_FAILED(rv))
return -1;
rv = out->Write(kData, sizeof(kData), &n);
if (NS_FAILED(rv))
return -1;
rv = out->Write(kData, sizeof(kData), &n);
if (NS_FAILED(rv))
return -1;
rv = out->Write(kData, sizeof(kData), &n);
if (NS_FAILED(rv))
return -1;
rv = out->Write(kData, 11, &n);
if (NS_FAILED(rv))
return -1;
rv = out->Close();
if (NS_FAILED(rv))
return -1;
out = nullptr;
2004-09-29 16:48:10 +00:00
// now, read all
rv = stor->NewInputStream(0, getter_AddRefs(in));
if (NS_FAILED(rv))
return -1;
// consume contents of input stream
do {
rv = in->Read(buf, sizeof(buf), &n);
if (NS_FAILED(rv))
return -1;
} while (n != 0);
rv = in->Close();
if (NS_FAILED(rv))
return -1;
in = nullptr;
2004-09-29 16:48:10 +00:00
return 0;
}