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 mozilla.org code.
15 : *
16 : * The Initial Developer of the Original Code is Mozilla Foundation
17 : * Portions created by the Initial Developer are Copyright (C) 2010
18 : * the Initial Developer. All Rights Reserved.
19 : *
20 : * Contributor(s):
21 : * Honza Bambas <honzab@firemni.cz>
22 : *
23 : * Alternatively, the contents of this file may be used under the terms of
24 : * either the GNU General Public License Version 2 or later (the "GPL"), or
25 : * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
26 : * in which case the provisions of the GPL or the LGPL are applicable instead
27 : * of those above. If you wish to allow use of your version of this file only
28 : * under the terms of either the GPL or the LGPL, and not to allow others to
29 : * use your version of this file under the terms of the MPL, indicate your
30 : * decision by deleting the provisions above and replace them with the notice
31 : * and other provisions required by the GPL or the LGPL. If you do not delete
32 : * the provisions above, a recipient may use your version of this file under
33 : * the terms of any one of the MPL, the GPL or the LGPL.
34 : *
35 : * ***** END LICENSE BLOCK ***** */
36 :
37 : #include "RedirectChannelRegistrar.h"
38 :
39 : namespace mozilla {
40 : namespace net {
41 :
42 : template<class KeyClass, class T>
43 : bool
44 : RedirectChannelRegistrar::nsCOMPtrHashtable<KeyClass,T>::Get(KeyType aKey, T** retVal) const
45 : {
46 0 : typename base_type::EntryType* ent = this->GetEntry(aKey);
47 :
48 0 : if (ent) {
49 0 : if (retVal)
50 0 : NS_IF_ADDREF(*retVal = ent->mData);
51 :
52 0 : return true;
53 : }
54 :
55 0 : if (retVal)
56 0 : *retVal = nsnull;
57 :
58 0 : return false;
59 : }
60 :
61 0 : NS_IMPL_ISUPPORTS1(RedirectChannelRegistrar, nsIRedirectChannelRegistrar)
62 :
63 0 : RedirectChannelRegistrar::RedirectChannelRegistrar()
64 0 : : mId(1)
65 : {
66 0 : mRealChannels.Init(64);
67 0 : mParentChannels.Init(64);
68 0 : }
69 :
70 : NS_IMETHODIMP
71 0 : RedirectChannelRegistrar::RegisterChannel(nsIChannel *channel,
72 : PRUint32 *_retval NS_OUTPARAM)
73 : {
74 0 : mRealChannels.Put(mId, channel);
75 0 : *_retval = mId;
76 :
77 0 : ++mId;
78 :
79 : // Ensure we always provide positive ids
80 0 : if (!mId)
81 0 : mId = 1;
82 :
83 0 : return NS_OK;
84 : }
85 :
86 : NS_IMETHODIMP
87 0 : RedirectChannelRegistrar::GetRegisteredChannel(PRUint32 id,
88 : nsIChannel **_retval NS_OUTPARAM)
89 : {
90 0 : if (!mRealChannels.Get(id, _retval))
91 0 : return NS_ERROR_NOT_AVAILABLE;
92 :
93 0 : return NS_OK;
94 : }
95 :
96 : NS_IMETHODIMP
97 0 : RedirectChannelRegistrar::LinkChannels(PRUint32 id,
98 : nsIParentChannel *channel,
99 : nsIChannel** _retval)
100 : {
101 0 : if (!mRealChannels.Get(id, _retval))
102 0 : return NS_ERROR_NOT_AVAILABLE;
103 :
104 0 : mParentChannels.Put(id, channel);
105 0 : return NS_OK;
106 : }
107 :
108 : NS_IMETHODIMP
109 0 : RedirectChannelRegistrar::GetParentChannel(PRUint32 id,
110 : nsIParentChannel **_retval NS_OUTPARAM)
111 : {
112 0 : if (!mParentChannels.Get(id, _retval))
113 0 : return NS_ERROR_NOT_AVAILABLE;
114 :
115 0 : return NS_OK;
116 : }
117 :
118 : NS_IMETHODIMP
119 0 : RedirectChannelRegistrar::DeregisterChannels(PRUint32 id)
120 : {
121 0 : mRealChannels.Remove(id);
122 0 : mParentChannels.Remove(id);
123 0 : return NS_OK;
124 : }
125 :
126 : }
127 : }
|