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 Zip Writer Component.
15 : *
16 : * The Initial Developer of the Original Code is
17 : * Dave Townsend <dtownsend@oxymoronical.com>.
18 : *
19 : * Portions created by the Initial Developer are Copyright (C) 2007
20 : * the Initial Developer. All Rights Reserved.
21 : *
22 : * Contributor(s):
23 : * Mook <mook.moz+random.code@gmail.com>
24 : *
25 : * Alternatively, the contents of this file may be used under the terms of
26 : * either the GNU General Public License Version 2 or later (the "GPL"), or
27 : * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
28 : * in which case the provisions of the GPL or the LGPL are applicable instead
29 : * of those above. If you wish to allow use of your version of this file only
30 : * under the terms of either the GPL or the LGPL, and not to allow others to
31 : * use your version of this file under the terms of the MPL, indicate your
32 : * decision by deleting the provisions above and replace them with the notice
33 : * and other provisions required by the GPL or the LGPL. If you do not delete
34 : * the provisions above, a recipient may use your version of this file under
35 : * the terms of any one of the MPL, the GPL or the LGPL.
36 : *
37 : * ***** END LICENSE BLOCK *****
38 : */
39 :
40 : #include "nscore.h"
41 : #include "nsIInputStream.h"
42 : #include "nsIOutputStream.h"
43 :
44 : /*
45 : * Fully reads the required amount of data. Keeps reading until all the
46 : * data is retrieved or an error is hit.
47 : */
48 161755 : NS_HIDDEN_(nsresult) ZW_ReadData(nsIInputStream *aStream, char *aBuffer, PRUint32 aCount)
49 : {
50 485265 : while (aCount > 0) {
51 : PRUint32 read;
52 161755 : nsresult rv = aStream->Read(aBuffer, aCount, &read);
53 161755 : NS_ENSURE_SUCCESS(rv, rv);
54 161755 : aCount -= read;
55 161755 : aBuffer += read;
56 : // If we hit EOF before reading the data we need then throw.
57 161755 : if (read == 0 && aCount > 0)
58 0 : return NS_ERROR_FAILURE;
59 : }
60 :
61 161755 : return NS_OK;
62 : }
63 :
64 : /*
65 : * Fully writes the required amount of data. Keeps writing until all the
66 : * data is written or an error is hit.
67 : */
68 59219 : NS_HIDDEN_(nsresult) ZW_WriteData(nsIOutputStream *aStream, const char *aBuffer,
69 : PRUint32 aCount)
70 : {
71 171590 : while (aCount > 0) {
72 : PRUint32 written;
73 53152 : nsresult rv = aStream->Write(aBuffer, aCount, &written);
74 53152 : NS_ENSURE_SUCCESS(rv, rv);
75 53152 : if (written <= 0)
76 0 : return NS_ERROR_FAILURE;
77 53152 : aCount -= written;
78 53152 : aBuffer += written;
79 : }
80 :
81 59219 : return NS_OK;
82 : }
|