netwerk/streamconv/test/Converters.cpp

   1 #include "Converters.h"
   2 #include "nsIStringStream.h"
   3 #include "nsCOMPtr.h"
   4 #include "nsComponentManagerUtils.h"
   5
   6 #include <stdio.h>
   7
   8 //////////////////////////////////////////////////
   9 // TestConverter
  10 //////////////////////////////////////////////////
  11
  12 NS_IMPL_ISUPPORTS3(TestConverter,
  13                    nsIStreamConverter,
  14                    nsIStreamListener,
  15                    nsIRequestObserver)
  16
  17 TestConverter::TestConverter() {
  18 }
  19
  20 // Convert aFromStream (of type aFromType), to _retval (nsIInputStream of type aToType).
  21 // This Convert method simply converts the stream byte-by-byte, to the first character
  22 // in the aToType "string".
  23 NS_IMETHODIMP
  24 TestConverter::Convert(nsIInputStream *aFromStream,
  25                        const char *aFromType,
  26                        const char *aToType,
  27                        nsISupports *ctxt,
  28                        nsIInputStream **_retval) {
  29     char buf[1024+1];
  30     PRUint32 read;
  31     nsresult rv = aFromStream->Read(buf, 1024, &read);
  32     if (NS_FAILED(rv) || read == 0) return rv;
  33
  34     // verify that the data we're converting matches the from type
  35     // if it doesn't then we're being handed the wrong data.
  36     char fromChar = *aFromType;
  37
  38     if (fromChar != buf[0]) {
  39         printf("We're receiving %c, but are supposed to have %c.\n", buf[0], fromChar);
  40         return NS_ERROR_FAILURE;
  41     }
  42
  43
  44     // Get the first character
  45     char toChar = *aToType;
  46
  47     for (PRUint32 i = 0; i < read; i++)
  48         buf[i] = toChar;
  49
  50     buf[read] = '\0';
  51
  52     nsCOMPtr<nsIStringInputStream> str
  53       (do_CreateInstance("@mozilla.org/io/string-input-stream;1", &rv));
  54     NS_ENSURE_SUCCESS(rv, rv);
  55
  56     rv = str->SetData(buf, read);
  57     NS_ENSURE_SUCCESS(rv, rv);
  58
  59     NS_ADDREF(*_retval = str);
  60     return NS_OK;
  61 }
  62
  63 /* This method initializes any internal state before the stream converter
  64  * begins asynchronous conversion */
  65 NS_IMETHODIMP
  66 TestConverter::AsyncConvertData(const char *aFromType,
  67                                 const char *aToType,
  68                                 nsIStreamListener *aListener,
  69                                 nsISupports *ctxt) {
  70     NS_ASSERTION(aListener, "null listener");
  71
  72     mListener = aListener;
  73
  74     // based on these types, setup internal state to handle the appropriate conversion.
  75     fromType = aFromType;
  76     toType = aToType;
  77
  78     return NS_OK;
  79 }
  80
  81 // nsIStreamListener method
  82 /* This method handles asyncronous conversion of data. */
  83 NS_IMETHODIMP
  84 TestConverter::OnDataAvailable(nsIRequest* request,
  85                                nsISupports *ctxt,
  86                                nsIInputStream *inStr,
  87                                PRUint32 sourceOffset,
  88                                PRUint32 count) {
  89     nsresult rv;
  90     nsCOMPtr<nsIInputStream> convertedStream;
  91     // just make a syncronous call to the Convert() method.
  92     // Anything can happen here, I just happen to be using the sync call to
  93     // do the actual conversion.
  94     rv = Convert(inStr, fromType.get(), toType.get(), ctxt, getter_AddRefs(convertedStream));
  95     if (NS_FAILED(rv)) return rv;
  96
  97     PRUint32 len;
  98     convertedStream->Available(&len);
  99     return mListener->OnDataAvailable(request, ctxt, convertedStream, sourceOffset, len);
 100 }
 101
 102 // nsIRequestObserver methods
 103 /* These methods just pass through directly to the mListener */
 104 NS_IMETHODIMP
 105 TestConverter::OnStartRequest(nsIRequest* request, nsISupports *ctxt) {
 106     return mListener->OnStartRequest(request, ctxt);
 107 }
 108
 109 NS_IMETHODIMP
 110 TestConverter::OnStopRequest(nsIRequest* request, nsISupports *ctxt,
 111                              nsresult aStatus) {
 112     return mListener->OnStopRequest(request, ctxt, aStatus);
 113 }
 114
 115
 116 ////////////////////////////////////////////////////////////////////////
 117 // TestConverterFactory
 118 ////////////////////////////////////////////////////////////////////////
 119 TestConverterFactory::TestConverterFactory(const nsCID &aClass,
 120                                    const char* className,
 121                                    const char* contractID)
 122     : mClassID(aClass), mClassName(className), mContractID(contractID)
 123 {
 124 }
 125
 126 TestConverterFactory::~TestConverterFactory()
 127 {
 128 }
 129
 130 NS_IMPL_ISUPPORTS1(TestConverterFactory, nsIFactory)
 131
 132 NS_IMETHODIMP
 133 TestConverterFactory::CreateInstance(nsISupports *aOuter,
 134                                  const nsIID &aIID,
 135                                  void **aResult)
 136 {
 137     if (! aResult)
 138         return NS_ERROR_NULL_POINTER;
 139
 140     if (aOuter)
 141         return NS_ERROR_NO_AGGREGATION;
 142
 143     *aResult = nsnull;
 144
 145     nsresult rv = NS_OK;
 146
 147     nsISupports *inst = nsnull;
 148     if (mClassID.Equals(kTestConverterCID)) {
 149         TestConverter *conv = new TestConverter();
 150         if (!conv) return NS_ERROR_OUT_OF_MEMORY;
 151         conv->QueryInterface(NS_GET_IID(nsISupports), (void**)&inst);
 152     }
 153     else {
 154         return NS_ERROR_NO_INTERFACE;
 155     }
 156
 157     if (!inst)
 158         return NS_ERROR_OUT_OF_MEMORY;
 159     NS_ADDREF(inst);
 160     *aResult = inst;
 161     NS_RELEASE(inst);
 162     return rv;
 163 }
 164
 165 nsresult TestConverterFactory::LockFactory(PRBool aLock)
 166 {
 167     // Not implemented in simplest case.
 168     return NS_OK;
 169 }
 170 ////////////////////////////////////////////////////////////////////////
 171 // TestConverterFactory END
 172 ////////////////////////////////////////////////////////////////////////
 173