1 : /* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
2 : /* ***** BEGIN LICENSE BLOCK *****
3 : * Version: MPL 1.1/GPL 2.0/LGPL 2.1
4 : *
5 : * The contents of this file are subject to the Mozilla Public License Version
6 : * 1.1 (the "License"); you may not use this file except in compliance with
7 : * the License. You may obtain a copy of the License at
8 : * http://www.mozilla.org/MPL/
9 : *
10 : * Software distributed under the License is distributed on an "AS IS" basis,
11 : * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
12 : * for the specific language governing rights and limitations under the
13 : * License.
14 : *
15 : * The Original Code is mozilla.org code.
16 : *
17 : * The Initial Developer of the Original Code is Google Inc.
18 : * Portions created by the Initial Developer are Copyright (C) 2005
19 : * the Initial Developer. All Rights Reserved.
20 : *
21 : * Contributor(s):
22 : * Darin Fisher <darin@meer.net>
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 "nsInputStreamChannel.h"
39 :
40 : //-----------------------------------------------------------------------------
41 : // nsInputStreamChannel
42 :
43 : nsresult
44 4 : nsInputStreamChannel::OpenContentStream(bool async, nsIInputStream **result,
45 : nsIChannel** channel)
46 : {
47 4 : NS_ENSURE_TRUE(mContentStream, NS_ERROR_NOT_INITIALIZED);
48 :
49 : // If content length is unknown, then we must guess. In this case, we assume
50 : // the stream can tell us. If the stream is a pipe, then this will not work.
51 :
52 4 : PRInt64 len = ContentLength64();
53 4 : if (len < 0) {
54 : PRUint32 avail;
55 4 : nsresult rv = mContentStream->Available(&avail);
56 4 : if (rv == NS_BASE_STREAM_CLOSED) {
57 : // This just means there's nothing in the stream
58 0 : avail = 0;
59 4 : } else if (NS_FAILED(rv)) {
60 0 : return rv;
61 : }
62 4 : SetContentLength64(avail);
63 : }
64 :
65 4 : EnableSynthesizedProgressEvents(true);
66 :
67 4 : NS_ADDREF(*result = mContentStream);
68 4 : return NS_OK;
69 : }
70 :
71 : //-----------------------------------------------------------------------------
72 : // nsInputStreamChannel::nsISupports
73 :
74 123119 : NS_IMPL_ISUPPORTS_INHERITED1(nsInputStreamChannel,
75 : nsBaseChannel,
76 : nsIInputStreamChannel)
77 :
78 : //-----------------------------------------------------------------------------
79 : // nsInputStreamChannel::nsIInputStreamChannel
80 :
81 : NS_IMETHODIMP
82 3146 : nsInputStreamChannel::SetURI(nsIURI *uri)
83 : {
84 3146 : NS_ENSURE_TRUE(!URI(), NS_ERROR_ALREADY_INITIALIZED);
85 3146 : nsBaseChannel::SetURI(uri);
86 3146 : return NS_OK;
87 : }
88 :
89 : NS_IMETHODIMP
90 0 : nsInputStreamChannel::GetContentStream(nsIInputStream **stream)
91 : {
92 0 : NS_IF_ADDREF(*stream = mContentStream);
93 0 : return NS_OK;
94 : }
95 :
96 : NS_IMETHODIMP
97 3146 : nsInputStreamChannel::SetContentStream(nsIInputStream *stream)
98 : {
99 3146 : NS_ENSURE_TRUE(!mContentStream, NS_ERROR_ALREADY_INITIALIZED);
100 3146 : mContentStream = stream;
101 3146 : return NS_OK;
102 : }
|