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