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 36 : NS_IMPL_ISUPPORTS1(nsBase64Encoder, nsIOutputStream)
44 :
45 : NS_IMETHODIMP
46 0 : nsBase64Encoder::Close()
47 : {
48 0 : return NS_OK;
49 : }
50 :
51 : NS_IMETHODIMP
52 0 : nsBase64Encoder::Flush()
53 : {
54 0 : return NS_OK;
55 : }
56 :
57 : NS_IMETHODIMP
58 488 : nsBase64Encoder::Write(const char* aBuf, PRUint32 aCount, PRUint32* _retval)
59 : {
60 488 : mData.Append(aBuf, aCount);
61 488 : *_retval = aCount;
62 488 : return NS_OK;
63 : }
64 :
65 : NS_IMETHODIMP
66 0 : nsBase64Encoder::WriteFrom(nsIInputStream* aStream, PRUint32 aCount,
67 : PRUint32* _retval)
68 : {
69 0 : return NS_ERROR_NOT_IMPLEMENTED;
70 : }
71 :
72 : NS_IMETHODIMP
73 0 : nsBase64Encoder::WriteSegments(nsReadSegmentFun aReader,
74 : void* aClosure,
75 : PRUint32 aCount,
76 : PRUint32* _retval)
77 : {
78 0 : return NS_ERROR_NOT_IMPLEMENTED;
79 : }
80 :
81 : NS_IMETHODIMP
82 0 : nsBase64Encoder::IsNonBlocking(bool* aNonBlocking)
83 : {
84 0 : *aNonBlocking = false;
85 0 : return NS_OK;
86 : }
87 :
88 : nsresult
89 4 : nsBase64Encoder::Finish(nsCSubstring& result)
90 : {
91 4 : char* b64 = PL_Base64Encode(mData.get(), mData.Length(), nsnull);
92 4 : if (!b64)
93 0 : return NS_ERROR_OUT_OF_MEMORY;
94 :
95 4 : result.Assign(b64);
96 4 : PR_Free(b64);
97 : // Free unneeded memory and allow reusing the object
98 4 : mData.Truncate();
99 4 : return NS_OK;
100 : }
|