Line data Source code
1 : /* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
2 : /*
3 : * This file is part of the LibreOffice project.
4 : *
5 : * This Source Code Form is subject to the terms of the Mozilla Public
6 : * License, v. 2.0. If a copy of the MPL was not distributed with this
7 : * file, You can obtain one at http://mozilla.org/MPL/2.0/.
8 : *
9 : * This file incorporates work covered by the following license notice:
10 : *
11 : * Licensed to the Apache Software Foundation (ASF) under one or more
12 : * contributor license agreements. See the NOTICE file distributed
13 : * with this work for additional information regarding copyright
14 : * ownership. The ASF licenses this file to you under the Apache
15 : * License, Version 2.0 (the "License"); you may not use this file
16 : * except in compliance with the License. You may obtain a copy of
17 : * the License at http://www.apache.org/licenses/LICENSE-2.0 .
18 : */
19 :
20 : #include "hsqldb/HStorageMap.hxx"
21 : #include <comphelper/types.hxx>
22 : #include <com/sun/star/embed/XTransactionBroadcaster.hpp>
23 : #include <com/sun/star/embed/XTransactedObject.hpp>
24 : #include <com/sun/star/embed/ElementModes.hpp>
25 : #include <com/sun/star/lang/DisposedException.hpp>
26 : #include "diagnose_ex.h"
27 : #include <osl/diagnose.h>
28 : #include <osl/thread.h>
29 :
30 : #include <o3tl/compat_functional.hxx>
31 :
32 :
33 : namespace connectivity
34 : {
35 :
36 : namespace hsqldb
37 : {
38 :
39 : using namespace ::com::sun::star::uno;
40 : using namespace ::com::sun::star::lang;
41 : using namespace ::com::sun::star::embed;
42 : using namespace ::com::sun::star::io;
43 :
44 151 : StreamHelper::StreamHelper(const Reference< XStream>& _xStream)
45 151 : : m_xStream(_xStream)
46 : {
47 151 : }
48 :
49 302 : StreamHelper::~StreamHelper()
50 : {
51 : try
52 : {
53 151 : m_xStream.clear();
54 151 : m_xSeek.clear();
55 151 : if ( m_xInputStream.is() )
56 : {
57 35 : m_xInputStream->closeInput();
58 35 : m_xInputStream.clear();
59 : }
60 : // this is done implicity by the closing of the input stream
61 116 : else if ( m_xOutputStream.is() )
62 : {
63 116 : m_xOutputStream->closeOutput();
64 : try
65 : {
66 114 : ::comphelper::disposeComponent(m_xOutputStream);
67 : }
68 0 : catch(const DisposedException&)
69 : {
70 : }
71 0 : catch(const Exception&)
72 : {
73 : OSL_FAIL("Could not dispose OutputStream");
74 : }
75 114 : m_xOutputStream.clear();
76 : }
77 : }
78 2 : catch(const Exception&)
79 : {
80 : OSL_FAIL("Exception caught!");
81 : }
82 151 : }
83 :
84 214 : Reference< XInputStream> StreamHelper::getInputStream()
85 : {
86 214 : if ( !m_xInputStream.is() )
87 35 : m_xInputStream = m_xStream->getInputStream();
88 214 : return m_xInputStream;
89 : }
90 :
91 1126 : Reference< XOutputStream> StreamHelper::getOutputStream()
92 : {
93 1126 : if ( !m_xOutputStream.is() )
94 125 : m_xOutputStream = m_xStream->getOutputStream();
95 1126 : return m_xOutputStream;
96 : }
97 :
98 781 : Reference< XSeekable> StreamHelper::getSeek()
99 : {
100 781 : if ( !m_xSeek.is() )
101 22 : m_xSeek.set(m_xStream,UNO_QUERY);
102 781 : return m_xSeek;
103 : }
104 :
105 3167 : TStorages& lcl_getStorageMap()
106 : {
107 3167 : static TStorages s_aMap;
108 3167 : return s_aMap;
109 : }
110 :
111 9 : OUString lcl_getNextCount()
112 : {
113 : static sal_Int32 s_nCount = 0;
114 9 : return OUString::number(s_nCount++);
115 : }
116 :
117 3060 : OUString StorageContainer::removeURLPrefix(const OUString& _sURL,const OUString& _sFileURL)
118 : {
119 3060 : return _sURL.copy(_sFileURL.getLength()+1);
120 : }
121 :
122 381 : OUString StorageContainer::removeOldURLPrefix(const OUString& _sURL)
123 : {
124 381 : OUString sRet = _sURL;
125 : #if defined(WNT)
126 : sal_Int32 nIndex = sRet.lastIndexOf('\\');
127 : #else
128 381 : sal_Int32 nIndex = sRet.lastIndexOf('/');
129 : #endif
130 381 : if ( nIndex != -1 )
131 : {
132 381 : sRet = _sURL.copy(nIndex+1);
133 : }
134 381 : return sRet;
135 :
136 : }
137 : /*****************************************************************************/
138 : /* convert jstring to rtl_uString */
139 :
140 6033 : OUString StorageContainer::jstring2ustring(JNIEnv * env, jstring jstr)
141 : {
142 6033 : if (env->ExceptionCheck())
143 : {
144 0 : env->ExceptionClear();
145 : OSL_FAIL("ExceptionClear");
146 : }
147 6033 : OUString aStr;
148 6033 : if ( jstr )
149 : {
150 6033 : jboolean bCopy(sal_True);
151 6033 : const jchar* pChar = env->GetStringChars(jstr,&bCopy);
152 6033 : jsize len = env->GetStringLength(jstr);
153 6033 : aStr = OUString(pChar,len);
154 :
155 6033 : if(bCopy)
156 6033 : env->ReleaseStringChars(jstr,pChar);
157 : }
158 :
159 6033 : if (env->ExceptionCheck())
160 : {
161 0 : env->ExceptionClear();
162 : OSL_FAIL("ExceptionClear");
163 : }
164 6033 : return aStr;
165 : }
166 :
167 :
168 9 : OUString StorageContainer::registerStorage(const Reference< XStorage>& _xStorage,const OUString& _sURL)
169 : {
170 : OSL_ENSURE(_xStorage.is(),"Storage is NULL!");
171 9 : TStorages& rMap = lcl_getStorageMap();
172 : // check if the storage is already in our map
173 : TStorages::iterator aFind = ::std::find_if(rMap.begin(),rMap.end(),
174 : ::o3tl::compose1(
175 : ::std::bind2nd(::std::equal_to<Reference<XStorage> >(),_xStorage)
176 18 : ,::o3tl::compose1(::o3tl::select1st<TStorageURLPair>(),::o3tl::compose1(::o3tl::select1st<TStorages::mapped_type>(),::o3tl::select2nd<TStorages::value_type>())))
177 9 : );
178 9 : if ( aFind == rMap.end() )
179 : {
180 9 : aFind = rMap.insert(TStorages::value_type(lcl_getNextCount(),TStorages::mapped_type(TStorageURLPair(_xStorage,_sURL),TStreamMap()))).first;
181 : }
182 :
183 9 : return aFind->first;
184 : }
185 :
186 729 : TStorages::mapped_type StorageContainer::getRegisteredStorage(const OUString& _sKey)
187 : {
188 729 : TStorages::mapped_type aRet;
189 729 : TStorages& rMap = lcl_getStorageMap();
190 729 : TStorages::iterator aFind = rMap.find(_sKey);
191 : OSL_ENSURE(aFind != rMap.end(),"Storage could not be found in list!");
192 729 : if ( aFind != rMap.end() )
193 729 : aRet = aFind->second;
194 :
195 729 : return aRet;
196 : }
197 :
198 24 : OUString StorageContainer::getRegisteredKey(const Reference< XStorage>& _xStorage)
199 : {
200 24 : OUString sKey;
201 : OSL_ENSURE(_xStorage.is(),"Storage is NULL!");
202 24 : TStorages& rMap = lcl_getStorageMap();
203 : // check if the storage is already in our map
204 : TStorages::iterator aFind = ::std::find_if(rMap.begin(),rMap.end(),
205 : ::o3tl::compose1(
206 : ::std::bind2nd(::std::equal_to<Reference<XStorage> >(),_xStorage)
207 48 : ,::o3tl::compose1(::o3tl::select1st<TStorageURLPair>(),::o3tl::compose1(::o3tl::select1st<TStorages::mapped_type>(),::o3tl::select2nd<TStorages::value_type>())))
208 24 : );
209 24 : if ( aFind != rMap.end() )
210 6 : sKey = aFind->first;
211 24 : return sKey;
212 : }
213 :
214 9 : void StorageContainer::revokeStorage(const OUString& _sKey,const Reference<XTransactionListener>& _xListener)
215 : {
216 9 : TStorages& rMap = lcl_getStorageMap();
217 9 : TStorages::iterator aFind = rMap.find(_sKey);
218 9 : if ( aFind != rMap.end() )
219 : {
220 : try
221 : {
222 9 : if ( _xListener.is() )
223 : {
224 0 : Reference<XTransactionBroadcaster> xBroad(aFind->second.first.first,UNO_QUERY);
225 0 : if ( xBroad.is() )
226 0 : xBroad->removeTransactionListener(_xListener);
227 0 : Reference<XTransactedObject> xTrans(aFind->second.first.first,UNO_QUERY);
228 0 : if ( xTrans.is() )
229 0 : xTrans->commit();
230 : }
231 : }
232 0 : catch(const Exception&)
233 : {
234 : }
235 9 : rMap.erase(aFind);
236 : }
237 9 : }
238 :
239 152 : TStreamMap::mapped_type StorageContainer::registerStream(JNIEnv * env,jstring name, jstring key,sal_Int32 _nMode)
240 : {
241 152 : TStreamMap::mapped_type pHelper;
242 152 : TStorages& rMap = lcl_getStorageMap();
243 304 : OUString sKey = jstring2ustring(env,key);
244 152 : TStorages::iterator aFind = rMap.find(sKey);
245 : OSL_ENSURE(aFind != rMap.end(),"Storage could not be found in list!");
246 152 : if ( aFind != rMap.end() )
247 : {
248 152 : TStorages::mapped_type aStoragePair = StorageContainer::getRegisteredStorage(sKey);
249 : OSL_ENSURE(aStoragePair.first.first.is(),"No Storage available!");
250 152 : if ( aStoragePair.first.first.is() )
251 : {
252 152 : OUString sOrgName = StorageContainer::jstring2ustring(env,name);
253 304 : OUString sName = removeURLPrefix(sOrgName,aStoragePair.first.second);
254 152 : TStreamMap::iterator aStreamFind = aFind->second.second.find(sName);
255 : OSL_ENSURE( aStreamFind == aFind->second.second.end(),"A Stream was already registered for this object!");
256 152 : if ( aStreamFind != aFind->second.second.end() )
257 : {
258 0 : pHelper = aStreamFind->second;
259 : }
260 : else
261 : {
262 : try
263 : {
264 : try
265 : {
266 152 : pHelper.reset(new StreamHelper(aStoragePair.first.first->openStreamElement(sName,_nMode)));
267 : }
268 1 : catch(const Exception&)
269 : {
270 1 : OUString sStrippedName = removeOldURLPrefix(sOrgName);
271 :
272 1 : if ( ((_nMode & ElementModes::WRITE) != ElementModes::WRITE ) )
273 : {
274 0 : bool bIsStream = true;
275 : try
276 : {
277 0 : bIsStream = aStoragePair.first.first->isStreamElement(sStrippedName);
278 : }
279 0 : catch(const Exception&)
280 : {
281 0 : bIsStream = false;
282 : }
283 0 : if ( !bIsStream )
284 0 : return pHelper; // readonly file without data stream
285 : }
286 1 : pHelper.reset( new StreamHelper(aStoragePair.first.first->openStreamElement( sStrippedName, _nMode ) ) );
287 0 : }
288 151 : aFind->second.second.insert(TStreamMap::value_type(sName,pHelper));
289 : }
290 1 : catch(const Exception& e)
291 : {
292 : #if OSL_DEBUG_LEVEL > 0
293 : OString sMessage( "[HSQLDB-SDBC] caught an exception while opening a stream\n" );
294 : sMessage += "Name: ";
295 : sMessage += OString( sName.getStr(), sName.getLength(), osl_getThreadTextEncoding() );
296 : sMessage += "\nMode: 0x";
297 : if ( _nMode < 16 )
298 : sMessage += "0";
299 : sMessage += OString::number( _nMode, 16 ).toAsciiUpperCase();
300 : OSL_FAIL( sMessage.getStr() );
301 : #endif
302 1 : StorageContainer::throwJavaException(e,env);
303 : }
304 152 : }
305 152 : }
306 : }
307 152 : return pHelper;
308 : }
309 :
310 149 : void StorageContainer::revokeStream( JNIEnv * env,jstring name, jstring key)
311 : {
312 149 : TStorages& rMap = lcl_getStorageMap();
313 149 : TStorages::iterator aFind = rMap.find(jstring2ustring(env,key));
314 : OSL_ENSURE(aFind != rMap.end(),"Storage could not be found in list!");
315 149 : if ( aFind != rMap.end() )
316 149 : aFind->second.second.erase(removeURLPrefix(jstring2ustring(env,name),aFind->second.first.second));
317 149 : }
318 :
319 2095 : TStreamMap::mapped_type StorageContainer::getRegisteredStream( JNIEnv * env,jstring name, jstring key)
320 : {
321 2095 : TStreamMap::mapped_type pRet;
322 2095 : TStorages& rMap = lcl_getStorageMap();
323 2095 : TStorages::iterator aFind = rMap.find(jstring2ustring(env,key));
324 : OSL_ENSURE(aFind != rMap.end(),"Storage could not be found in list!");
325 2095 : if ( aFind != rMap.end() )
326 : {
327 2095 : TStreamMap::iterator aStreamFind = aFind->second.second.find(removeURLPrefix(jstring2ustring(env,name),aFind->second.first.second));
328 2095 : if ( aStreamFind != aFind->second.second.end() )
329 2095 : pRet = aStreamFind->second;
330 : }
331 :
332 2095 : return pRet;
333 : }
334 :
335 3 : void StorageContainer::throwJavaException(const Exception& _aException,JNIEnv * env)
336 : {
337 3 : if (env->ExceptionCheck())
338 0 : env->ExceptionClear();
339 3 : OString cstr( OUStringToOString(_aException.Message, RTL_TEXTENCODING_JAVA_UTF8 ) );
340 : OSL_TRACE( __FILE__": forwarding Exception: %s", cstr.getStr() );
341 3 : env->ThrowNew(env->FindClass("java/io/IOException"), cstr.getStr());
342 3 : }
343 :
344 : } // namespace hsqldb
345 :
346 :
347 : }
348 : // namespace connectivity
349 :
350 :
351 : /* vim:set shiftwidth=4 softtabstop=4 expandtab: */
|