The Tor Browser / file diff

     1.1 --- /dev/null	Thu Jan 01 00:00:00 1970 +0000
     1.2 +++ b/xpcom/tests/TestStorageStream.cpp	Wed Dec 31 06:09:35 2014 +0100
     1.3 @@ -0,0 +1,110 @@
     1.4 +/* This Source Code Form is subject to the terms of the Mozilla Public
     1.5 + * License, v. 2.0. If a copy of the MPL was not distributed with this
     1.6 + * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
     1.7 +
     1.8 +#include <stdlib.h>
     1.9 +#include "nsIStorageStream.h"
    1.10 +#include "nsIInputStream.h"
    1.11 +#include "nsIOutputStream.h"
    1.12 +#include "nsCOMPtr.h"
    1.13 +
    1.14 +int main()
    1.15 +{
    1.16 +  char kData[4096];
    1.17 +  memset(kData, 0, sizeof(kData));
    1.18 +
    1.19 +  nsresult rv;
    1.20 +  nsCOMPtr<nsIStorageStream> stor;
    1.21 +
    1.22 +  rv = NS_NewStorageStream(4096, UINT32_MAX, getter_AddRefs(stor));
    1.23 +  if (NS_FAILED(rv))
    1.24 +    return -1;
    1.25 +
    1.26 +  nsCOMPtr<nsIOutputStream> out;
    1.27 +  rv = stor->GetOutputStream(0, getter_AddRefs(out));
    1.28 +  if (NS_FAILED(rv))
    1.29 +    return -1;
    1.30 +
    1.31 +  uint32_t n;
    1.32 +
    1.33 +  rv = out->Write(kData, sizeof(kData), &n);
    1.34 +  if (NS_FAILED(rv))
    1.35 +    return -1;
    1.36 +
    1.37 +  rv = out->Write(kData, sizeof(kData), &n);
    1.38 +  if (NS_FAILED(rv))
    1.39 +    return -1;
    1.40 +
    1.41 +  rv = out->Close();
    1.42 +  if (NS_FAILED(rv))
    1.43 +    return -1;
    1.44 +
    1.45 +  out = nullptr;
    1.46 +  
    1.47 +  nsCOMPtr<nsIInputStream> in;
    1.48 +  rv = stor->NewInputStream(0, getter_AddRefs(in));
    1.49 +  if (NS_FAILED(rv))
    1.50 +    return -1;
    1.51 +
    1.52 +  char buf[4096];
    1.53 +
    1.54 +  // consume contents of input stream
    1.55 +  do {
    1.56 +    rv = in->Read(buf, sizeof(buf), &n);
    1.57 +    if (NS_FAILED(rv))
    1.58 +      return -1;
    1.59 +  } while (n != 0);
    1.60 +
    1.61 +  rv = in->Close();
    1.62 +  if (NS_FAILED(rv))
    1.63 +    return -1;
    1.64 +  in = nullptr;
    1.65 +
    1.66 +  // now, write 3 more full 4k segments + 11 bytes, starting at 8192
    1.67 +  // total written equals 20491 bytes
    1.68 +
    1.69 +  rv = stor->GetOutputStream(8192, getter_AddRefs(out));
    1.70 +  if (NS_FAILED(rv))
    1.71 +    return -1;
    1.72 +
    1.73 +  rv = out->Write(kData, sizeof(kData), &n);
    1.74 +  if (NS_FAILED(rv))
    1.75 +    return -1;
    1.76 +
    1.77 +  rv = out->Write(kData, sizeof(kData), &n);
    1.78 +  if (NS_FAILED(rv))
    1.79 +    return -1;
    1.80 +
    1.81 +  rv = out->Write(kData, sizeof(kData), &n);
    1.82 +  if (NS_FAILED(rv))
    1.83 +    return -1;
    1.84 +
    1.85 +  rv = out->Write(kData, 11, &n);
    1.86 +  if (NS_FAILED(rv))
    1.87 +    return -1;
    1.88 +
    1.89 +  rv = out->Close();
    1.90 +  if (NS_FAILED(rv))
    1.91 +    return -1;
    1.92 +
    1.93 +  out = nullptr;
    1.94 +
    1.95 +  // now, read all
    1.96 +  rv = stor->NewInputStream(0, getter_AddRefs(in));
    1.97 +  if (NS_FAILED(rv))
    1.98 +    return -1;
    1.99 +
   1.100 +  // consume contents of input stream
   1.101 +  do {
   1.102 +    rv = in->Read(buf, sizeof(buf), &n);
   1.103 +    if (NS_FAILED(rv))
   1.104 +      return -1;
   1.105 +  } while (n != 0);
   1.106 +
   1.107 +  rv = in->Close();
   1.108 +  if (NS_FAILED(rv))
   1.109 +    return -1;
   1.110 +  in = nullptr;
   1.111 +
   1.112 +  return 0;
   1.113 +}