summaryrefslogtreecommitdiff
path: root/netwerk/test/TestStreamPump.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'netwerk/test/TestStreamPump.cpp')
-rw-r--r--netwerk/test/TestStreamPump.cpp171
1 files changed, 171 insertions, 0 deletions
diff --git a/netwerk/test/TestStreamPump.cpp b/netwerk/test/TestStreamPump.cpp
new file mode 100644
index 0000000000..376b9deb7c
--- /dev/null
+++ b/netwerk/test/TestStreamPump.cpp
@@ -0,0 +1,171 @@
+/* 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/. */
+
+#include "TestCommon.h"
+#include "nsIComponentRegistrar.h"
+#include "nsIStreamTransportService.h"
+#include "nsIAsyncInputStream.h"
+#include "nsIProgressEventSink.h"
+#include "nsIInterfaceRequestor.h"
+#include "nsIInterfaceRequestorUtils.h"
+#include "nsIRequest.h"
+#include "nsIServiceManager.h"
+#include "nsIComponentManager.h"
+#include "nsISeekableStream.h"
+#include "nsCOMPtr.h"
+#include "nsMemory.h"
+#include "nsStringAPI.h"
+#include "nsIFileStreams.h"
+#include "nsIStreamListener.h"
+#include "nsIFile.h"
+#include "nsNetUtil.h"
+#include "nsAutoLock.h"
+#include "mozilla/Logging.h"
+#include "prprf.h"
+#include <algorithm>
+
+////////////////////////////////////////////////////////////////////////////////
+
+//
+// set NSPR_LOG_MODULES=Test:5
+//
+static PRLogModuleInfo *gTestLog = nullptr;
+#define LOG(args) MOZ_LOG(gTestLog, mozilla::LogLevel::Debug, args)
+
+////////////////////////////////////////////////////////////////////////////////
+
+class MyListener : public nsIStreamListener
+{
+public:
+ NS_DECL_ISUPPORTS
+
+ MyListener() {}
+ virtual ~MyListener() {}
+
+ NS_IMETHOD OnStartRequest(nsIRequest *req, nsISupports *ctx)
+ {
+ LOG(("MyListener::OnStartRequest\n"));
+ return NS_OK;
+ }
+
+ NS_IMETHOD OnDataAvailable(nsIRequest *req, nsISupports *ctx,
+ nsIInputStream *stream,
+ uint64_t offset, uint32_t count)
+ {
+ LOG(("MyListener::OnDataAvailable [offset=%llu count=%u]\n", offset, count));
+
+ char buf[500];
+ nsresult rv;
+
+ while (count) {
+ uint32_t n, amt = std::min<uint32_t>(count, sizeof(buf));
+
+ rv = stream->Read(buf, amt, &n);
+ if (NS_FAILED(rv)) {
+ LOG((" read returned 0x%08x\n", rv));
+ return rv;
+ }
+
+ fwrite(buf, n, 1, stdout);
+ printf("\n");
+
+ LOG((" read %u bytes\n", n));
+ count -= n;
+ }
+
+ return NS_OK;
+ }
+
+ NS_IMETHOD OnStopRequest(nsIRequest *req, nsISupports *ctx, nsresult status)
+ {
+ LOG(("MyListener::OnStopRequest [status=%x]\n", status));
+ QuitPumpingEvents();
+ return NS_OK;
+ }
+};
+
+NS_IMPL_ISUPPORTS(MyListener,
+ nsIRequestObserver,
+ nsIStreamListener)
+
+////////////////////////////////////////////////////////////////////////////////
+
+/**
+ * asynchronously copy file.
+ */
+static nsresult
+RunTest(nsIFile *file, int64_t offset, int64_t length)
+{
+ nsresult rv;
+
+ LOG(("RunTest\n"));
+
+ nsCOMPtr<nsIInputStream> stream;
+ rv = NS_NewLocalFileInputStream(getter_AddRefs(stream), file);
+ if (NS_FAILED(rv)) return rv;
+
+ nsCOMPtr<nsIInputStreamPump> pump;
+ rv = NS_NewInputStreamPump(getter_AddRefs(pump), stream, offset, length);
+ if (NS_FAILED(rv)) return rv;
+
+ rv = pump->AsyncRead(new MyListener(), nullptr);
+ if (NS_FAILED(rv)) return rv;
+
+ PumpEvents();
+ return NS_OK;
+}
+
+////////////////////////////////////////////////////////////////////////////////
+
+int
+main(int argc, char* argv[])
+{
+ if (test_common_init(&argc, &argv) != 0)
+ return -1;
+
+ nsresult rv;
+
+ if (argc < 4) {
+ printf("usage: %s <file-to-read> <start-offset> <read-length>\n", argv[0]);
+ return -1;
+ }
+ char* fileName = argv[1];
+ int64_t offset, length;
+ int err = PR_sscanf(argv[2], "%lld", &offset);
+ if (err == -1) {
+ printf("Start offset must be an integer!\n");
+ return 1;
+ }
+ err = PR_sscanf(argv[3], "%lld", &length);
+ if (err == -1) {
+ printf("Length must be an integer!\n");
+ return 1;
+ }
+
+ {
+ nsCOMPtr<nsIServiceManager> servMan;
+ NS_InitXPCOM2(getter_AddRefs(servMan), nullptr, nullptr);
+ nsCOMPtr<nsIComponentRegistrar> registrar = do_QueryInterface(servMan);
+ NS_ASSERTION(registrar, "Null nsIComponentRegistrar");
+ if (registrar)
+ registrar->AutoRegister(nullptr);
+
+ gTestLog = PR_NewLogModule("Test");
+
+ nsCOMPtr<nsIFile> file;
+ rv = NS_NewNativeLocalFile(nsDependentCString(fileName), false, getter_AddRefs(file));
+ if (NS_FAILED(rv)) return rv;
+
+ rv = RunTest(file, offset, length);
+ NS_ASSERTION(NS_SUCCEEDED(rv), "RunTest failed");
+
+ // give background threads a chance to finish whatever work they may
+ // be doing.
+ PR_Sleep(PR_SecondsToInterval(1));
+ } // this scopes the nsCOMPtrs
+ // no nsCOMPtrs are allowed to be alive when you call NS_ShutdownXPCOM
+ rv = NS_ShutdownXPCOM(nullptr);
+ NS_ASSERTION(NS_SUCCEEDED(rv), "NS_ShutdownXPCOM failed");
+ return NS_OK;
+}