mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-10-12 04:45:45 +00:00
128f95aa9b
Revising nsIChannel to allow for overlapped i/o. This consists of three parts: 1. Factoring nsIChannel into a protocol specific part, the nsIChannel, and a socket specific, the nsITransport. 2. Derive the nsIChannel from a nsIRequest. 2. Changes the notification system from necko and the URILoader to pass the nsIRequest interface instead of nsIChannel interface. This goal stems from wanting to be able to have active AsyncRead and AsyncWrite operations on nsSocketTransport. This is desired because it would greatly simplify the task of maintaining persistent/reusable socket connections for FTP, HTTP, and Imap (and potentially other protocols). The problem with the existing nsIChannel interface is that it does not allow one to selectively suspend just one of the read or write operations while keeping the other active. r=darin@netscape.com sr=rpotts@netscape.com
392 lines
11 KiB
C++
392 lines
11 KiB
C++
/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*-
|
|
*
|
|
* The contents of this file are subject to the Netscape 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/NPL/
|
|
*
|
|
* 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.org code.
|
|
*
|
|
* The Initial Developer of the Original Code is Netscape
|
|
* Communications Corporation. Portions created by Netscape are
|
|
* Copyright (C) 1998 Netscape Communications Corporation. All
|
|
* Rights Reserved.
|
|
*
|
|
* Contributor(s):
|
|
*/
|
|
#include <stdio.h>
|
|
#include <signal.h>
|
|
|
|
#ifdef WIN32
|
|
#include <windows.h>
|
|
#endif
|
|
#ifdef OS2
|
|
#include <os2.h>
|
|
#endif
|
|
|
|
#include "nspr.h"
|
|
#include "nscore.h"
|
|
#include "nsISocketTransportService.h"
|
|
#include "nsIEventQueueService.h"
|
|
#include "nsIServiceManager.h"
|
|
#include "nsITransport.h"
|
|
#include "nsIRequest.h"
|
|
#include "nsIStreamProvider.h"
|
|
#include "nsIStreamListener.h"
|
|
#include "nsIPipe.h"
|
|
#include "nsIOutputStream.h"
|
|
#include "nsIInputStream.h"
|
|
#include "nsCRT.h"
|
|
#include "nsCOMPtr.h"
|
|
#include "nsIByteArrayInputStream.h"
|
|
|
|
#if defined(PR_LOGGING)
|
|
static PRLogModuleInfo *gTestSocketIOLog;
|
|
#define LOG(args) PR_LOG(gTestSocketIOLog, PR_LOG_DEBUG, args)
|
|
#else
|
|
#define LOG(args)
|
|
#endif
|
|
|
|
static NS_DEFINE_CID(kSocketTransportServiceCID, NS_SOCKETTRANSPORTSERVICE_CID);
|
|
static NS_DEFINE_CID(kEventQueueServiceCID, NS_EVENTQUEUESERVICE_CID);
|
|
static NS_DEFINE_IID(kEventQueueCID, NS_EVENTQUEUE_CID);
|
|
|
|
static PRTime gElapsedTime;
|
|
static int gKeepRunning = 1;
|
|
static nsIEventQueue* gEventQ = nsnull;
|
|
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
// Test Listener
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
|
|
class TestListener : public nsIStreamListener
|
|
{
|
|
public:
|
|
TestListener() { NS_INIT_ISUPPORTS(); }
|
|
virtual ~TestListener() {}
|
|
|
|
NS_DECL_ISUPPORTS
|
|
NS_DECL_NSISTREAMOBSERVER
|
|
NS_DECL_NSISTREAMLISTENER
|
|
};
|
|
|
|
NS_IMPL_ISUPPORTS2(TestListener,
|
|
nsIStreamObserver,
|
|
nsIStreamListener);
|
|
|
|
NS_IMETHODIMP
|
|
TestListener::OnStartRequest(nsIRequest* request, nsISupports* context)
|
|
{
|
|
LOG(("TestListener::OnStartRequest\n"));
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
TestListener::OnDataAvailable(nsIRequest* request,
|
|
nsISupports* context,
|
|
nsIInputStream *aIStream,
|
|
PRUint32 aSourceOffset,
|
|
PRUint32 aLength)
|
|
{
|
|
LOG(("TestListener::OnDataAvailable [offset=%u length=%u]\n",
|
|
aSourceOffset, aLength));
|
|
char buf[1025];
|
|
PRUint32 amt;
|
|
while (1) {
|
|
aIStream->Read(buf, 1024, &amt);
|
|
if (amt == 0)
|
|
break;
|
|
buf[amt] = '\0';
|
|
puts(buf);
|
|
}
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
TestListener::OnStopRequest(nsIRequest* request, nsISupports* context,
|
|
nsresult aStatus, const PRUnichar* aStatusArg)
|
|
{
|
|
LOG(("TestListener::OnStopRequest [aStatus=%x]\n", aStatus));
|
|
gKeepRunning = 0;
|
|
return NS_OK;
|
|
}
|
|
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
// Test Provider
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
|
|
class TestProvider : public nsIStreamProvider
|
|
{
|
|
public:
|
|
TestProvider(char *data);
|
|
virtual ~TestProvider();
|
|
|
|
NS_DECL_ISUPPORTS
|
|
NS_DECL_NSISTREAMOBSERVER
|
|
NS_DECL_NSISTREAMPROVIDER
|
|
|
|
protected:
|
|
nsCOMPtr<nsIByteArrayInputStream> mData;
|
|
};
|
|
|
|
NS_IMPL_ISUPPORTS2(TestProvider,
|
|
nsIStreamProvider,
|
|
nsIStreamObserver)
|
|
|
|
TestProvider::TestProvider(char *data)
|
|
{
|
|
NS_INIT_ISUPPORTS();
|
|
NS_NewByteArrayInputStream(getter_AddRefs(mData), data, strlen(data));
|
|
LOG(("Constructing TestProvider [this=%x]\n", this));
|
|
}
|
|
|
|
TestProvider::~TestProvider()
|
|
{
|
|
LOG(("Destroying TestProvider [this=%x]\n", this));
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
TestProvider::OnStartRequest(nsIRequest* request, nsISupports* context)
|
|
{
|
|
LOG(("TestProvider::OnStartRequest [this=%x]\n", this));
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
TestProvider::OnStopRequest(nsIRequest* request, nsISupports* context,
|
|
nsresult aStatus, const PRUnichar* aStatusArg)
|
|
{
|
|
LOG(("TestProvider::OnStopRequest [status=%x]\n", aStatus));
|
|
|
|
nsCOMPtr<nsIStreamListener> listener = do_QueryInterface(new TestListener());
|
|
|
|
if (NS_SUCCEEDED(aStatus)) {
|
|
nsCOMPtr<nsITransportRequest> treq = do_QueryInterface(request);
|
|
nsCOMPtr<nsITransport> transport;
|
|
treq->GetTransport(getter_AddRefs(transport));
|
|
if (transport) {
|
|
nsCOMPtr<nsIRequest> readRequest;
|
|
transport->AsyncRead(listener, nsnull, 0, 0, 0, getter_AddRefs(readRequest));
|
|
}
|
|
} else
|
|
gKeepRunning = 0;
|
|
|
|
return NS_OK;
|
|
}
|
|
|
|
NS_IMETHODIMP
|
|
TestProvider::OnDataWritable(nsIRequest *request, nsISupports *context,
|
|
nsIOutputStream *output, PRUint32 offset, PRUint32 count)
|
|
{
|
|
LOG(("TestProvider::OnDataWritable [offset=%u, count=%u]\n", offset, count));
|
|
PRUint32 writeCount;
|
|
nsresult rv = output->WriteFrom(mData, count, &writeCount);
|
|
// Zero bytes written on success indicates EOF
|
|
if (NS_SUCCEEDED(rv) && (writeCount == 0))
|
|
return NS_BASE_STREAM_CLOSED;
|
|
return rv;
|
|
}
|
|
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
// Synchronous IO
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
nsresult
|
|
WriteRequest(nsIOutputStream *os, const char *request)
|
|
{
|
|
LOG(("WriteRequest [request=%s]\n", request));
|
|
PRUint32 n;
|
|
return os->Write(request, strlen(request), &n);
|
|
}
|
|
|
|
nsresult
|
|
ReadResponse(nsIInputStream *is)
|
|
{
|
|
PRUint32 bytesRead;
|
|
char buf[2048];
|
|
do {
|
|
is->Read(buf, sizeof(buf), &bytesRead);
|
|
if (bytesRead > 0)
|
|
fwrite(buf, 1, bytesRead, stdout);
|
|
} while (bytesRead > 0);
|
|
return NS_OK;
|
|
}
|
|
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
// Startup...
|
|
//----------------------------------------------------------------------------
|
|
//
|
|
|
|
void
|
|
sighandler(int sig)
|
|
{
|
|
LOG(("got signal: %d\n", sig));
|
|
NS_BREAK();
|
|
}
|
|
|
|
void
|
|
usage(char **argv)
|
|
{
|
|
printf("usage: %s [-sync] <host> <path>\n", argv[0]);
|
|
exit(1);
|
|
}
|
|
|
|
int
|
|
main(int argc, char* argv[])
|
|
{
|
|
nsresult rv;
|
|
|
|
#ifdef XP_OS2_VACPP
|
|
signal(SIGSEGV, (_SigFunc)sighandler);
|
|
#else
|
|
signal(SIGSEGV, sighandler);
|
|
#endif
|
|
|
|
#if defined(PR_LOGGING)
|
|
gTestSocketIOLog = PR_NewLogModule("TestSocketIO");
|
|
#endif
|
|
|
|
if (argc < 3)
|
|
usage(argv);
|
|
|
|
PRIntn i=0;
|
|
PRBool sync = PR_FALSE;
|
|
if (nsCRT::strcasecmp(argv[1], "-sync") == 0) {
|
|
if (argc < 4)
|
|
usage(argv);
|
|
sync = PR_TRUE;
|
|
i = 1;
|
|
}
|
|
|
|
char *hostName = argv[1+i];
|
|
char *fileName = argv[2+i];
|
|
int port = 80;
|
|
|
|
// Create the Event Queue for this thread...
|
|
NS_WITH_SERVICE(nsIEventQueueService, eventQService, kEventQueueServiceCID, &rv);
|
|
if (NS_FAILED(rv)) {
|
|
NS_WARNING("failed to create: event queue service!");
|
|
return rv;
|
|
}
|
|
|
|
rv = eventQService->CreateMonitoredThreadEventQueue();
|
|
if (NS_FAILED(rv)) {
|
|
NS_WARNING("failed to create: thread event queue!");
|
|
return rv;
|
|
}
|
|
|
|
eventQService->GetThreadEventQueue(NS_CURRENT_THREAD, &gEventQ);
|
|
|
|
// Create the Socket transport service...
|
|
NS_WITH_SERVICE(nsISocketTransportService, sts, kSocketTransportServiceCID, &rv);
|
|
if (NS_FAILED(rv)) {
|
|
NS_WARNING("failed to create: socket transport service!");
|
|
return rv;
|
|
}
|
|
|
|
char *buffer = PR_smprintf("GET %s HTTP/1.1" CRLF
|
|
"host: %s" CRLF
|
|
"user-agent: Mozilla/5.0 (X11; N; Linux 2.2.16-22smp i686; en-US; m18) Gecko/20001220" CRLF
|
|
"accept: */*" CRLF
|
|
"accept-language: en" CRLF
|
|
"accept-encoding: gzip,deflate,compress,identity" CRLF
|
|
"keep-alive: 300" CRLF
|
|
"connection: keep-alive" CRLF
|
|
CRLF,
|
|
fileName, hostName);
|
|
LOG(("Request [\n%s]\n", buffer));
|
|
|
|
// Create the socket transport...
|
|
nsCOMPtr<nsITransport> transport;
|
|
rv = sts->CreateTransport(hostName, port, nsnull, -1, 0, 0, getter_AddRefs(transport));
|
|
if (NS_FAILED(rv)) {
|
|
NS_WARNING("failed to create: socket transport!");
|
|
return rv;
|
|
}
|
|
|
|
gElapsedTime = PR_Now();
|
|
|
|
if (!sync) {
|
|
nsCOMPtr<nsIRequest> request;
|
|
rv = transport->AsyncWrite(new TestProvider(buffer), nsnull, 0, 0, 0, getter_AddRefs(request));
|
|
if (NS_FAILED(rv)) {
|
|
NS_WARNING("failed calling: AsyncWrite!");
|
|
return rv;
|
|
}
|
|
|
|
// Enter the message pump to allow the URL load to proceed.
|
|
while ( gKeepRunning ) {
|
|
#ifdef WIN32
|
|
MSG msg;
|
|
if (GetMessage(&msg, NULL, 0, 0)) {
|
|
TranslateMessage(&msg);
|
|
DispatchMessage(&msg);
|
|
}
|
|
else
|
|
gKeepRunning = FALSE;
|
|
#elif XP_MAC
|
|
/* Mac stuff is missing here! */
|
|
#elif XP_OS2
|
|
QMSG qmsg;
|
|
if (WinGetMsg(0, &qmsg, 0, 0, 0))
|
|
WinDispatchMsg(0, &qmsg);
|
|
else
|
|
gKeepRunning = FALSE;
|
|
#else
|
|
PLEvent *gEvent;
|
|
rv = gEventQ->WaitForEvent(&gEvent);
|
|
rv = gEventQ->HandleEvent(gEvent);
|
|
#endif
|
|
}
|
|
}
|
|
else {
|
|
// synchronous write
|
|
{
|
|
nsCOMPtr<nsIOutputStream> os;
|
|
rv = transport->OpenOutputStream(0, 0, 0, getter_AddRefs(os));
|
|
if (NS_FAILED(rv)) {
|
|
LOG(("OpenOutputStream failed [rv=%x]\n", rv));
|
|
return rv;
|
|
}
|
|
rv = WriteRequest(os, buffer);
|
|
if (NS_FAILED(rv)) {
|
|
LOG(("WriteRequest failed [rv=%x]\n", rv));
|
|
return rv;
|
|
}
|
|
}
|
|
// synchronous read
|
|
{
|
|
nsCOMPtr<nsIInputStream> is;
|
|
rv = transport->OpenInputStream(0, 0, 0, getter_AddRefs(is));
|
|
if (NS_FAILED(rv)) {
|
|
LOG(("OpenInputStream failed [rv=%x]\n", rv));
|
|
return rv;
|
|
}
|
|
rv = ReadResponse(is);
|
|
if (NS_FAILED(rv)) {
|
|
LOG(("ReadResponse failed [rv=%x]\n", rv));
|
|
return rv;
|
|
}
|
|
}
|
|
}
|
|
|
|
PRTime endTime;
|
|
endTime = PR_Now();
|
|
LOG(("Elapsed time: %d\n", (PRInt32)(endTime/1000UL - gElapsedTime/1000UL)));
|
|
|
|
sts->Shutdown();
|
|
return 0;
|
|
}
|
|
|