xpcom/tests/TestStorageStream.cpp

   1 /* ***** BEGIN LICENSE BLOCK *****
   2  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   3  *
   4  * The contents of this file are subject to the Mozilla Public License Version
   5  * 1.1 (the "License"); you may not use this file except in compliance with
   6  * the License. You may obtain a copy of the License at
   7  * http://www.mozilla.org/MPL/
   8  *
   9  * Software distributed under the License is distributed on an "AS IS" basis,
  10  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  11  * for the specific language governing rights and limitations under the
  12  * License.
  13  *
  14  * The Original Code is Mozilla.
  15  *
  16  * The Initial Developer of the Original Code is IBM Corporation.
  17  * Portions created by IBM Corporation are Copyright (C) 2004
  18  * IBM Corporation. All Rights Reserved.
  19  *
  20  * Contributor(s):
  21  *   Darin Fisher <darin@meer.net>
  22  *
  23  * Alternatively, the contents of this file may be used under the terms of
  24  * either the GNU General Public License Version 2 or later (the "GPL"), or
  25  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  26  * in which case the provisions of the GPL or the LGPL are applicable instead
  27  * of those above. If you wish to allow use of your version of this file only
  28  * under the terms of either the GPL or the LGPL, and not to allow others to
  29  * use your version of this file under the terms of the MPL, indicate your
  30  * decision by deleting the provisions above and replace them with the notice
  31  * and other provisions required by the GPL or the LGPL. If you do not delete
  32  * the provisions above, a recipient may use your version of this file under
  33  * the terms of any one of the MPL, the GPL or the LGPL.
  34  *
  35  * ***** END LICENSE BLOCK ***** */
  36
  37 #include <stdlib.h>
  38 #include "nsIStorageStream.h"
  39 #include "nsIInputStream.h"
  40 #include "nsIOutputStream.h"
  41 #include "nsCOMPtr.h"
  42
  43 int main()
  44 {
  45   char kData[4096];
  46   memset(kData, 0, sizeof(kData));
  47
  48   nsresult rv;
  49   nsCOMPtr<nsIStorageStream> stor;
  50
  51   rv = NS_NewStorageStream(4096, PR_UINT32_MAX, getter_AddRefs(stor));
  52   if (NS_FAILED(rv))
  53     return -1;
  54
  55   nsCOMPtr<nsIOutputStream> out;
  56   rv = stor->GetOutputStream(0, getter_AddRefs(out));
  57   if (NS_FAILED(rv))
  58     return -1;
  59
  60   PRUint32 n;
  61
  62   rv = out->Write(kData, sizeof(kData), &n);
  63   if (NS_FAILED(rv))
  64     return -1;
  65
  66   rv = out->Write(kData, sizeof(kData), &n);
  67   if (NS_FAILED(rv))
  68     return -1;
  69
  70   rv = out->Close();
  71   if (NS_FAILED(rv))
  72     return -1;
  73
  74   out = nsnull;
  75
  76   nsCOMPtr<nsIInputStream> in;
  77   rv = stor->NewInputStream(0, getter_AddRefs(in));
  78   if (NS_FAILED(rv))
  79     return -1;
  80
  81   char buf[4096];
  82
  83   // consume contents of input stream
  84   do {
  85     rv = in->Read(buf, sizeof(buf), &n);
  86     if (NS_FAILED(rv))
  87       return -1;
  88   } while (n != 0);
  89
  90   rv = in->Close();
  91   if (NS_FAILED(rv))
  92     return -1;
  93   in = nsnull;
  94
  95   // now, write 3 more full 4k segments + 11 bytes, starting at 8192
  96   // total written equals 20491 bytes
  97
  98   rv = stor->GetOutputStream(8192, getter_AddRefs(out));
  99   if (NS_FAILED(rv))
 100     return -1;
 101
 102   rv = out->Write(kData, sizeof(kData), &n);
 103   if (NS_FAILED(rv))
 104     return -1;
 105
 106   rv = out->Write(kData, sizeof(kData), &n);
 107   if (NS_FAILED(rv))
 108     return -1;
 109
 110   rv = out->Write(kData, sizeof(kData), &n);
 111   if (NS_FAILED(rv))
 112     return -1;
 113
 114   rv = out->Write(kData, 11, &n);
 115   if (NS_FAILED(rv))
 116     return -1;
 117
 118   rv = out->Close();
 119   if (NS_FAILED(rv))
 120     return -1;
 121
 122   out = nsnull;
 123
 124   // now, read all
 125   rv = stor->NewInputStream(0, getter_AddRefs(in));
 126   if (NS_FAILED(rv))
 127     return -1;
 128
 129   // consume contents of input stream
 130   do {
 131     rv = in->Read(buf, sizeof(buf), &n);
 132     if (NS_FAILED(rv))
 133       return -1;
 134   } while (n != 0);
 135
 136   rv = in->Close();
 137   if (NS_FAILED(rv))
 138     return -1;
 139   in = nsnull;
 140
 141   return 0;
 142 }