mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-07 12:15:51 +00:00
131 lines
4.3 KiB
C++
131 lines
4.3 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.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):
|
|
*/
|
|
|
|
#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__
|