netwerk/base/src/nsBase64Encoder.cpp

   1 /* ***** BEGIN LICENSE BLOCK *****
   2  * Version: MPL 1.1/GPL 2.0/LGPL 2.1
   3  *
   4  * The contents of this file are subject to the Mozilla Public License Version
   5  * 1.1 (the "License"); you may not use this file except in compliance with
   6  * the License. You may obtain a copy of the License at
   7  * http://www.mozilla.org/MPL/
   8  *
   9  * Software distributed under the License is distributed on an "AS IS" basis,
  10  * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
  11  * for the specific language governing rights and limitations under the
  12  * License.
  13  *
  14  * The Original Code is a base64 encoder stream.
  15  *
  16  * The Initial Developer of the Original Code is
  17  * Google Inc.
  18  * Portions created by the Initial Developer are Copyright (C) 2007
  19  * the Initial Developer. All Rights Reserved.
  20  *
  21  * Contributor(s):
  22  *   Christian Biesinger <cbiesinger@web.de> (Initial author)
  23  *
  24  * Alternatively, the contents of this file may be used under the terms of
  25  * either the GNU General Public License Version 2 or later (the "GPL"), or
  26  * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
  27  * in which case the provisions of the GPL or the LGPL are applicable instead
  28  * of those above. If you wish to allow use of your version of this file only
  29  * under the terms of either the GPL or the LGPL, and not to allow others to
  30  * use your version of this file under the terms of the MPL, indicate your
  31  * decision by deleting the provisions above and replace them with the notice
  32  * and other provisions required by the GPL or the LGPL. If you do not delete
  33  * the provisions above, a recipient may use your version of this file under
  34  * the terms of any one of the MPL, the GPL or the LGPL.
  35  *
  36  * ***** END LICENSE BLOCK ***** */
  37
  38 #include "nsBase64Encoder.h"
  39
  40 #include "plbase64.h"
  41 #include "prmem.h"
  42
  43 NS_IMPL_ISUPPORTS1(nsBase64Encoder, nsIOutputStream)
  44
  45 NS_IMETHODIMP
  46 nsBase64Encoder::Close()
  47 {
  48   return NS_OK;
  49 }
  50
  51 NS_IMETHODIMP
  52 nsBase64Encoder::Flush()
  53 {
  54   return NS_OK;
  55 }
  56
  57 NS_IMETHODIMP
  58 nsBase64Encoder::Write(const char* aBuf, PRUint32 aCount, PRUint32* _retval)
  59 {
  60   mData.Append(aBuf, aCount);
  61   *_retval = aCount;
  62   return NS_OK;
  63 }
  64
  65 NS_IMETHODIMP
  66 nsBase64Encoder::WriteFrom(nsIInputStream* aStream, PRUint32 aCount,
  67                            PRUint32* _retval)
  68 {
  69   return NS_ERROR_NOT_IMPLEMENTED;
  70 }
  71
  72 NS_IMETHODIMP
  73 nsBase64Encoder::WriteSegments(nsReadSegmentFun aReader,
  74                                void* aClosure,
  75                                PRUint32 aCount,
  76                                PRUint32* _retval)
  77 {
  78   return NS_ERROR_NOT_IMPLEMENTED;
  79 }
  80
  81 NS_IMETHODIMP
  82 nsBase64Encoder::IsNonBlocking(PRBool* aNonBlocking)
  83 {
  84   *aNonBlocking = PR_FALSE;
  85   return NS_OK;
  86 }
  87
  88 nsresult
  89 nsBase64Encoder::Finish(nsCSubstring& result)
  90 {
  91   char* b64 = PL_Base64Encode(mData.get(), mData.Length(), nsnull);
  92   if (!b64)
  93     return NS_ERROR_OUT_OF_MEMORY;
  94
  95   result.Assign(b64);
  96   PR_Free(b64);
  97   // Free unneeded memory and allow reusing the object
  98   mData.Truncate();
  99   return NS_OK;
 100 }