Landed raptor modular netlib stuff

git-svn-id: svn://10.0.0.236/trunk@2225 18797224-902f-48f8-a5cc-f745e15eee43

mozilla/network/module/nsHttpUrl.cpp

@@ -0,0 +1,279 @@
+/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*-
+ *
+ * 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.
+ */
+
+#include "nsAgg.h"
+#include "nsIProtocolConnection.h"
+#include "nsIHttpUrl.h"
+#include "nsIPostToServer.h"
+
+ #include "nsINetService.h"  /* XXX: NS_FALSE */
+#include "nsNetStream.h"
+#include "net.h"
+
+#include "prmem.h"
+#include "plstr.h"
+
+static NS_DEFINE_IID(kIOutputStreamIID,  NS_IOUTPUTSTREAM_IID);
+
+
+class nsHttpUrlImpl : public nsIProtocolConnection,
+                      public nsIPostToServer,
+                      public nsIHttpUrl
+{
+public:
+    typedef enum {
+        Send_None,
+        Send_File,
+        Send_Data,
+        Send_DataFromFile,
+    } SendType;
+
+public:
+    nsHttpUrlImpl(nsISupports* outer);
+
+    /* Aggregated nsISupports interface... */
+    NS_DECL_AGGREGATED
+
+    /* nsIProtocolConnection interface... */
+    NS_IMETHOD InitializeURLInfo(URL_Struct_ *URL_s);
+
+    /* nsIPostToServer interface... */
+    NS_IMETHOD  SendFile(const char *aFile);
+    NS_IMETHOD  SendData(const char *aBuffer, PRInt32 aLength);
+    NS_IMETHOD  SendDataFromFile(const char *aFile);
+
+    /* nsIHttpUrl interface... */
+
+
+static nsISupports* NewHttpUrlImpl(nsISupports *outer);
+
+protected:
+    virtual ~nsHttpUrlImpl();
+
+    nsresult PostFile(const char *aFile);
+
+    SendType m_PostType;
+    char *m_PostBuffer;
+    PRInt32 m_PostBufferLength;
+};
+
+
+
+nsHttpUrlImpl::nsHttpUrlImpl(nsISupports* outer)
+{
+    NS_INIT_AGGREGATED(outer);
+
+    m_PostType = Send_None;
+    m_PostBuffer = nsnull;
+    m_PostBufferLength = 0;
+}
+
+nsHttpUrlImpl::~nsHttpUrlImpl()
+{
+    if (nsnull != m_PostBuffer) PR_Free(m_PostBuffer);
+}
+
+
+
+NS_IMPL_AGGREGATED(nsHttpUrlImpl)
+
+nsresult nsHttpUrlImpl::AggregatedQueryInterface(const nsIID &aIID, 
+                                             void** aInstancePtr)
+{
+    if (NULL == aInstancePtr) {
+        return NS_ERROR_NULL_POINTER;
+    }
+
+    static NS_DEFINE_IID(kISupportsIID,            NS_ISUPPORTS_IID);
+    static NS_DEFINE_IID(kIProtocolConnectionIID,  NS_IPROTOCOLCONNECTION_IID);
+    static NS_DEFINE_IID(kIPostToServerIID,        NS_IPOSTTOSERVER_IID);
+    static NS_DEFINE_IID(kIHttpUrlIID,             NS_IHTTPURL_IID);
+    if (aIID.Equals(kIProtocolConnectionIID)) {
+        *aInstancePtr = (void*) ((nsIProtocolConnection*)this);
+        AddRef();
+        return NS_OK;
+    }
+    if (aIID.Equals(kIHttpUrlIID)) {
+        *aInstancePtr = (void*) ((nsIHttpUrl*)this);
+        AddRef();
+        return NS_OK;
+    }
+    if (aIID.Equals(kIPostToServerIID)) {
+        *aInstancePtr = (void*) ((nsIPostToServer*)this);
+        AddRef();
+        return NS_OK;
+    }
+    if (aIID.Equals(kISupportsIID)) {
+        *aInstancePtr = (void*) ((nsISupports *)&fAggregated);
+        AddRef();
+        return NS_OK;
+    }
+
+    return NS_NOINTERFACE;
+}
+
+
+NS_METHOD nsHttpUrlImpl::InitializeURLInfo(URL_Struct *URL_s)
+{
+    nsresult result = NS_OK;
+
+    if (Send_None != m_PostType) {
+        /* Free any existing POST data hanging off the URL_Struct */
+        if (nsnull != URL_s->post_data) {
+            PR_Free(URL_s->post_data);
+        }
+
+        /*
+         * Store the new POST data into the URL_Struct
+         */
+        URL_s->post_data         = m_PostBuffer;
+        URL_s->post_data_size    = m_PostBufferLength;
+        URL_s->post_data_is_file = (Send_Data == m_PostType) ? FALSE : TRUE;
+        /*
+         * Is the request a POST or PUT
+         */
+        URL_s->method = (Send_File == m_PostType) ? URL_PUT_METHOD : URL_POST_METHOD;
+
+        /* Reset the local post state... */
+        m_PostType = Send_None;
+        m_PostBuffer = nsnull;  /* The URL_Struct owns the memory now... */
+        m_PostBufferLength = 0;
+    }
+
+    return result;
+}
+
+
+NS_METHOD  nsHttpUrlImpl::SendFile(const char *aFile)
+{
+    nsresult result;
+
+    result = PostFile(aFile);
+    if (NS_OK == result) {
+        m_PostType = Send_File;
+    }
+    return result;
+}
+
+
+NS_METHOD  nsHttpUrlImpl::SendDataFromFile(const char *aFile)
+{
+    nsresult result;
+
+    result = PostFile(aFile);
+    if (NS_OK == result) {
+        m_PostType = Send_DataFromFile;
+    }
+    return result;
+}
+
+
+NS_METHOD  nsHttpUrlImpl::SendData(const char *aBuffer, PRInt32 aLength)
+{
+    nsresult result = NS_OK;
+
+    /* Deal with error conditions... */
+    if (nsnull == aBuffer) {
+        result = NS_ERROR_ILLEGAL_VALUE;
+        goto done;
+    } 
+    else if (Send_None != m_PostType) {
+        result = NS_IPOSTTOSERVER_ALREADY_POSTING;
+        goto done;
+    }
+
+    /* Copy the post data... */
+    m_PostBuffer = (char *)PR_Malloc(aLength);
+    if (nsnull == m_PostBuffer) {
+        result = NS_ERROR_OUT_OF_MEMORY;
+        goto done;
+    }
+    memcpy(m_PostBuffer, aBuffer, aLength);
+    m_PostBufferLength = aLength;
+    m_PostType = Send_Data;
+
+done:
+    return result;
+}
+
+
+nsresult nsHttpUrlImpl::PostFile(const char *aFile)
+{
+    nsresult result = NS_OK;
+
+    /* Deal with error conditions... */
+    if (nsnull == aFile) {
+        result = NS_ERROR_ILLEGAL_VALUE;
+        goto done;
+    } 
+    else if (Send_None != m_PostType) {
+        result = NS_IPOSTTOSERVER_ALREADY_POSTING;
+        goto done;
+    }
+
+    /* Copy the post data... */
+    m_PostBuffer = PL_strdup(aFile);
+
+    if (nsnull == m_PostBuffer) {
+        result = NS_ERROR_OUT_OF_MEMORY;
+        goto done;
+    }
+    m_PostBufferLength = PL_strlen(aFile);
+
+done:
+    return result;
+}
+
+
+nsISupports* nsHttpUrlImpl::NewHttpUrlImpl(nsISupports* aOuter)
+{
+    nsHttpUrlImpl* it;
+    nsISupports *result;
+
+    it = new nsHttpUrlImpl(aOuter);
+    if (nsnull != it) {
+        if (nsnull != aOuter) {
+            result = &it->fAggregated;
+        } else {
+            result = (nsIProtocolConnection *)it;
+        }
+
+        result->AddRef();
+    }
+
+    return result;
+}
+
+
+
+extern "C" NS_NET nsresult NS_NewHttpUrl(nsISupports** aInstancePtrResult,
+                                         nsISupports* aOuter)
+{
+    NS_PRECONDITION(nsnull != aInstancePtrResult, "null ptr");
+    if (nsnull == aInstancePtrResult) {
+        return NS_ERROR_NULL_POINTER;
+    }
+
+    *aInstancePtrResult = nsHttpUrlImpl::NewHttpUrlImpl(aOuter);
+    if (nsnull == *aInstancePtrResult) {
+        return NS_ERROR_OUT_OF_MEMORY;
+    }
+
+    return NS_OK;
+}
+