gecko-dev/network/module/nsSocketTransport.h

127 строки
4.2 KiB
C++

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*-
*
* The contents of this file are subject to the Netscape Public License
* Version 1.0 (the "NPL"); you may not use this file except in
* compliance with the NPL. You may obtain a copy of the NPL at
* http://www.mozilla.org/NPL/
*
* Software distributed under the NPL is distributed on an "AS IS" basis,
* WITHOUT WARRANTY OF ANY KIND, either express or implied. See the NPL
* for the specific language governing rights and limitations under the
* NPL.
*
* The Initial Developer of this code under the NPL is Netscape
* Communications Corporation. Portions created by Netscape are
* Copyright (C) 1998 Netscape Communications Corporation. All Rights
* Reserved.
*/
#ifndef nsSocketTransport_h__
#define nsSocketTransport_h__
#include "nsIComponentManager.h"
#include "nsITransport.h"
#include "nsIStreamListener.h"
#include "nsIInputStream.h"
#include "nsIOutputStream.h"
#include "nsIEventQueueService.h"
#include "nsAgg.h"
#include "nsString.h"
#define NET_SOCKSTUB_BUF_SIZE 1024
/* forward declaration */
struct URL_Struct_;
class nsSocketTransport : public nsITransport
{
public:
////////////////////////////////////////////////////////////////////////////
// from nsISupports:
NS_DECL_ISUPPORTS
////////////////////////////////////////////////////////////////////////////
// from nsITransport:
NS_IMETHOD GetURL(nsIURI* *result);
NS_IMETHOD SetInputStreamConsumer(nsIStreamListener* aListener);
NS_IMETHOD GetOutputStreamConsumer(nsIStreamListener ** aConsumer);
NS_IMETHOD GetOutputStream(nsIOutputStream ** aOutputStream);
NS_IMETHOD IsTransportOpen(PRBool * aSocketOpen);
NS_IMETHOD Open(nsIURI * aUrl);
////////////////////////////////////////////////////////////////////////////
// from nsIStreamListener:
NS_IMETHOD OnStartRequest(nsIURI* aURL, const char *aContentType);
NS_IMETHOD OnProgress(nsIURI* aURL, PRUint32 aProgress, PRUint32 aProgressMax);
NS_IMETHOD OnStatus(nsIURI* aURL, const PRUnichar* aMsg);
NS_IMETHOD OnStopRequest(nsIURI* aURL, nsresult aStatus, const PRUnichar* aMsg);
NS_IMETHOD GetBindInfo(nsIURI* aURL, nsStreamBindingInfo* aInfo);
NS_IMETHOD OnDataAvailable(nsIURI* aURL, nsIInputStream *aIStream,
PRUint32 aLength);
////////////////////////////////////////////////////////////////////////////
// nsSocketTransport:
// we have two types of connections: socket and file based....
nsSocketTransport(PRUint32 aPortToUse, const char * aHostName);
nsSocketTransport(const char * fileName);
virtual ~nsSocketTransport(void);
NS_IMETHOD GetURLInfo(nsIURI* pURL, URL_Struct_ **aResult);
NS_IMETHOD SetSocketBinding(nsIURI* pURL, URL_Struct_ ** aResult);
// the following routines are called by the sock stub protocol hack....
// we should be able to remove this dependency once we move things to the new
// net lib world...
PRUint32 GetPort() {return m_port;}
const char * GetHostName() { return m_hostName;}
const char * GetFileName() { return m_fileName;}
protected:
// use this function to close the underlying connection....
nsresult CloseCurrentConnection();
// common initialization code..
void Initialize();
// socket specific information...
PRUint32 m_port;
char *m_hostName;
char *m_fileName;
PRBool m_socketIsOpen; // set when we have opened a socket....
PRBool m_isFileConnection;
// the stream we write data from the socket into
nsIInputStream * m_inputStream;
// XXX sock stub hack..need to remember stream with data to be written to the socket
nsIInputStream * m_outStream;
PRUint32 m_outStreamSize;
// the proxied stream listener. Whenever the transport layer
// writes socket date to input stream, it calls OnDataAvailable
// through the inputStreamConsumer any socket specific data
nsIStreamListener *m_inputStreamConsumer;
nsIURI *m_url; // the url we are currently running...
nsString* mData;
PRFileDesc *m_ready_fd;
nsIEventQueueService* mEventQService;
nsIEventQueue* m_evQueue;
char m_buffer[NET_SOCKSTUB_BUF_SIZE];
URL_Struct_ * m_bindedUrlStruct;
};
#endif // nsSocketTransport_h__