gecko-dev/netwerk/test/TestSocketTransport.cpp

392 строки
12 KiB
C++

/* ***** BEGIN LICENSE BLOCK *****
* Version: MPL 1.1/GPL 2.0/LGPL 2.1
*
* The contents of this file are subject to the Mozilla Public License Version
* 1.1 (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
* http://www.mozilla.org/MPL/
*
* Software distributed under the License is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
* for the specific language governing rights and limitations under the
* License.
*
* The Original Code is Mozilla.
*
* The Initial Developer of the Original Code is
* Netscape Communications Corporation.
* Portions created by the Initial Developer are Copyright (C) 2002
* the Initial Developer. All Rights Reserved.
*
* Contributor(s):
* Darin Fisher <darin@netscape.com>
*
* Alternatively, the contents of this file may be used under the terms of
* either the GNU General Public License Version 2 or later (the "GPL"), or
* the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
* in which case the provisions of the GPL or the LGPL are applicable instead
* of those above. If you wish to allow use of your version of this file only
* under the terms of either the GPL or the LGPL, and not to allow others to
* use your version of this file under the terms of the MPL, indicate your
* decision by deleting the provisions above and replace them with the notice
* and other provisions required by the GPL or the LGPL. If you do not delete
* the provisions above, a recipient may use your version of this file under
* the terms of any one of the MPL, the GPL or the LGPL.
*
* ***** END LICENSE BLOCK ***** */
#include "nsIComponentRegistrar.h"
#include "nsISocketTransportService.h"
#include "nsISocketTransport.h"
#include "nsIAsyncInputStream.h"
#include "nsIAsyncOutputStream.h"
#include "nsIProgressEventSink.h"
#include "nsIInterfaceRequestor.h"
#include "nsIInterfaceRequestorUtils.h"
#include "nsIProxyObjectManager.h"
#include "nsIRequest.h"
#include "nsIServiceManager.h"
#include "nsIComponentManager.h"
#include "nsCOMPtr.h"
#include "nsMemory.h"
#include "nsString.h"
#include "nsIFileStreams.h"
#include "nsIStreamListener.h"
#include "nsIEventQueueService.h"
#include "nsIEventQueue.h"
#include "nsILocalFile.h"
#include "nsNetUtil.h"
#include "nsAutoLock.h"
#include "prlog.h"
////////////////////////////////////////////////////////////////////////////////
#if defined(PR_LOGGING)
//
// set NSPR_LOG_MODULES=Test:5
//
static PRLogModuleInfo *gTestLog = nsnull;
#endif
#define LOG(args) PR_LOG(gTestLog, PR_LOG_DEBUG, args)
////////////////////////////////////////////////////////////////////////////////
static NS_DEFINE_CID(kSocketTransportServiceCID, NS_SOCKETTRANSPORTSERVICE_CID);
static NS_DEFINE_CID(kEventQueueServiceCID, NS_EVENTQUEUESERVICE_CID);
PRBool gDone = PR_FALSE;
nsIEventQueue* gEventQ = nsnull;
////////////////////////////////////////////////////////////////////////////////
static void *PR_CALLBACK
DoneEvent_Handler(PLEvent *ev)
{
gDone = PR_TRUE;
return nsnull;
}
static void PR_CALLBACK
DoneEvent_Cleanup(PLEvent *ev)
{
delete ev;
}
static void
PostDoneEvent()
{
LOG(("PostDoneEvent\n"));
PLEvent *ev = new PLEvent();
PL_InitEvent(ev, nsnull,
DoneEvent_Handler,
DoneEvent_Cleanup);
gEventQ->PostEvent(ev);
}
////////////////////////////////////////////////////////////////////////////////
class MyHandler : public nsIOutputStreamCallback
, public nsIInputStreamCallback
{
public:
NS_DECL_ISUPPORTS
MyHandler(const char *path,
nsIAsyncInputStream *in,
nsIAsyncOutputStream *out)
: mInput(in)
, mOutput(out)
, mWriteOffset(0)
{
mBuf = NS_LITERAL_CSTRING("GET ")
+ nsDependentCString(path)
+ NS_LITERAL_CSTRING(" HTTP/1.0\r\n\r\n");
}
virtual ~MyHandler() {}
// called on any thread
NS_IMETHOD OnOutputStreamReady(nsIAsyncOutputStream *out)
{
LOG(("OnOutputStreamReady\n"));
nsresult rv;
PRUint32 n, count = mBuf.Length() - mWriteOffset;
rv = out->Write(mBuf.get() + mWriteOffset, count, &n);
LOG((" write returned [rv=%x count=%u]\n", rv, n));
if (NS_FAILED(rv) || (n == 0)) {
if (rv != NS_BASE_STREAM_WOULD_BLOCK) {
LOG((" done writing; starting to read\n"));
mInput->AsyncWait(this, 0, 0, nsnull);
return NS_OK;
}
}
mWriteOffset += n;
return out->AsyncWait(this, 0, 0, nsnull);
}
// called on any thread
NS_IMETHOD OnInputStreamReady(nsIAsyncInputStream *in)
{
LOG(("OnInputStreamReady\n"));
nsresult rv;
PRUint32 n;
char buf[500];
rv = in->Read(buf, sizeof(buf), &n);
LOG((" read returned [rv=%x count=%u]\n", rv, n));
if (NS_FAILED(rv) || (n == 0)) {
if (rv != NS_BASE_STREAM_WOULD_BLOCK) {
PostDoneEvent();
return NS_OK;
}
}
return in->AsyncWait(this, 0, 0, nsnull);
}
private:
nsCOMPtr<nsIAsyncInputStream> mInput;
nsCOMPtr<nsIAsyncOutputStream> mOutput;
nsCString mBuf;
PRUint32 mWriteOffset;
};
NS_IMPL_THREADSAFE_ISUPPORTS2(MyHandler,
nsIOutputStreamCallback,
nsIInputStreamCallback)
////////////////////////////////////////////////////////////////////////////////
/**
* create transport, open streams, and close
*/
static nsresult
RunCloseTest(nsISocketTransportService *sts,
const char *host, int port,
PRUint32 inFlags, PRUint32 outFlags)
{
nsresult rv;
LOG(("RunCloseTest\n"));
nsCOMPtr<nsISocketTransport> transport;
rv = sts->CreateTransport(nsnull, 0,
nsDependentCString(host), port, nsnull,
getter_AddRefs(transport));
if (NS_FAILED(rv)) return rv;
nsCOMPtr<nsIInputStream> in;
rv = transport->OpenInputStream(inFlags, 0, 0, getter_AddRefs(in));
nsCOMPtr<nsIAsyncInputStream> asyncIn = do_QueryInterface(in, &rv);
if (NS_FAILED(rv)) return rv;
nsCOMPtr<nsIOutputStream> out;
rv = transport->OpenOutputStream(outFlags, 0, 0, getter_AddRefs(out));
nsCOMPtr<nsIAsyncOutputStream> asyncOut = do_QueryInterface(out, &rv);
if (NS_FAILED(rv)) return rv;
LOG(("waiting 1 second before closing transport and streams...\n"));
PR_Sleep(PR_SecondsToInterval(1));
// let nsCOMPtr destructors close everything...
return NS_OK;
}
/**
* asynchronously read socket stream
*/
static nsresult
RunTest(nsISocketTransportService *sts,
const char *host, int port, const char *path,
PRUint32 inFlags, PRUint32 outFlags)
{
nsresult rv;
LOG(("RunTest\n"));
nsCOMPtr<nsISocketTransport> transport;
rv = sts->CreateTransport(nsnull, 0,
nsDependentCString(host), port, nsnull,
getter_AddRefs(transport));
if (NS_FAILED(rv)) return rv;
nsCOMPtr<nsIInputStream> in;
rv = transport->OpenInputStream(inFlags, 0, 0, getter_AddRefs(in));
nsCOMPtr<nsIAsyncInputStream> asyncIn = do_QueryInterface(in, &rv);
if (NS_FAILED(rv)) return rv;
nsCOMPtr<nsIOutputStream> out;
rv = transport->OpenOutputStream(outFlags, 0, 0, getter_AddRefs(out));
nsCOMPtr<nsIAsyncOutputStream> asyncOut = do_QueryInterface(out, &rv);
if (NS_FAILED(rv)) return rv;
MyHandler *handler = new MyHandler(path, asyncIn, asyncOut);
if (handler == nsnull)
return NS_ERROR_OUT_OF_MEMORY;
NS_ADDREF(handler);
rv = asyncOut->AsyncWait(handler, 0, 0, nsnull);
if (NS_SUCCEEDED(rv)) {
PLEvent* event;
gDone = PR_FALSE;
while (!gDone) {
rv = gEventQ->WaitForEvent(&event);
if (NS_FAILED(rv)) return rv;
rv = gEventQ->HandleEvent(event);
if (NS_FAILED(rv)) return rv;
}
}
NS_RELEASE(handler);
return NS_OK;
}
////////////////////////////////////////////////////////////////////////////////
int
main(int argc, char* argv[])
{
nsresult rv;
if (argc < 4) {
printf("usage: TestSocketTransport <host> <port> <path>\n");
return -1;
}
{
nsCOMPtr<nsIServiceManager> servMan;
NS_InitXPCOM2(getter_AddRefs(servMan), nsnull, nsnull);
nsCOMPtr<nsIComponentRegistrar> registrar = do_QueryInterface(servMan);
NS_ASSERTION(registrar, "Null nsIComponentRegistrar");
if (registrar)
registrar->AutoRegister(nsnull);
#if defined(PR_LOGGING)
gTestLog = PR_NewLogModule("Test");
#endif
nsCOMPtr<nsIEventQueueService> eventQService =
do_GetService(kEventQueueServiceCID, &rv);
if (NS_FAILED(rv)) return rv;
rv = eventQService->CreateMonitoredThreadEventQueue();
if (NS_FAILED(rv)) return rv;
rv = eventQService->GetThreadEventQueue(NS_CURRENT_THREAD, &gEventQ);
if (NS_FAILED(rv)) return rv;
nsCOMPtr<nsISocketTransportService> sts =
do_GetService(kSocketTransportServiceCID, &rv);
if (NS_FAILED(rv)) return rv;
LOG(("phase 1 tests...\n"));
LOG(("flags = { OPEN_UNBUFFERED, OPEN_UNBUFFERED }\n"));
rv = RunCloseTest(sts, argv[1], atoi(argv[2]),
nsITransport::OPEN_UNBUFFERED,
nsITransport::OPEN_UNBUFFERED);
NS_ASSERTION(NS_SUCCEEDED(rv), "RunCloseTest failed");
LOG(("flags = { OPEN_BUFFERED, OPEN_UNBUFFERED }\n"));
rv = RunCloseTest(sts, argv[1], atoi(argv[2]),
0 /* nsITransport::OPEN_BUFFERED */,
nsITransport::OPEN_UNBUFFERED);
NS_ASSERTION(NS_SUCCEEDED(rv), "RunCloseTest failed");
LOG(("flags = { OPEN_UNBUFFERED, OPEN_BUFFERED }\n"));
rv = RunCloseTest(sts, argv[1], atoi(argv[2]),
nsITransport::OPEN_UNBUFFERED,
0 /*nsITransport::OPEN_BUFFERED */);
NS_ASSERTION(NS_SUCCEEDED(rv), "RunCloseTest failed");
LOG(("flags = { OPEN_BUFFERED, OPEN_BUFFERED }\n"));
rv = RunCloseTest(sts, argv[1], atoi(argv[2]),
0 /*nsITransport::OPEN_BUFFERED */,
0 /*nsITransport::OPEN_BUFFERED */);
NS_ASSERTION(NS_SUCCEEDED(rv), "RunCloseTest failed");
LOG(("calling Shutdown on socket transport service:\n"));
sts->Shutdown();
LOG(("calling Init on socket transport service:\n"));
sts->Init();
LOG(("phase 2 tests...\n"));
LOG(("flags = { OPEN_UNBUFFERED, OPEN_UNBUFFERED }\n"));
rv = RunTest(sts, argv[1], atoi(argv[2]), argv[3],
nsITransport::OPEN_UNBUFFERED,
nsITransport::OPEN_UNBUFFERED);
NS_ASSERTION(NS_SUCCEEDED(rv), "RunTest failed");
LOG(("flags = { OPEN_BUFFERED, OPEN_UNBUFFERED }\n"));
rv = RunTest(sts, argv[1], atoi(argv[2]), argv[3],
0 /* nsITransport::OPEN_BUFFERED */,
nsITransport::OPEN_UNBUFFERED);
NS_ASSERTION(NS_SUCCEEDED(rv), "RunTest failed");
LOG(("flags = { OPEN_UNBUFFERED, OPEN_BUFFERED }\n"));
rv = RunTest(sts, argv[1], atoi(argv[2]), argv[3],
nsITransport::OPEN_UNBUFFERED,
0 /*nsITransport::OPEN_BUFFERED */);
NS_ASSERTION(NS_SUCCEEDED(rv), "RunTest failed");
LOG(("flags = { OPEN_BUFFERED, OPEN_BUFFERED }\n"));
rv = RunTest(sts, argv[1], atoi(argv[2]), argv[3],
0 /*nsITransport::OPEN_BUFFERED */,
0 /*nsITransport::OPEN_BUFFERED */);
NS_ASSERTION(NS_SUCCEEDED(rv), "RunTest failed");
LOG(("waiting 1 second before calling Shutdown...\n"));
PR_Sleep(PR_SecondsToInterval(1));
LOG(("calling Shutdown on socket transport service:\n"));
sts->Shutdown();
NS_RELEASE(gEventQ);
// give background threads a chance to finish whatever work they may
// be doing.
LOG(("waiting 1 second before exiting...\n"));
PR_Sleep(PR_SecondsToInterval(1));
} // this scopes the nsCOMPtrs
// no nsCOMPtrs are allowed to be alive when you call NS_ShutdownXPCOM
rv = NS_ShutdownXPCOM(nsnull);
NS_ASSERTION(NS_SUCCEEDED(rv), "NS_ShutdownXPCOM failed");
return NS_OK;
}