1*b557fc96SAndrew Rist /**************************************************************
2cdf0e10cSrcweir *
3*b557fc96SAndrew Rist * Licensed to the Apache Software Foundation (ASF) under one
4*b557fc96SAndrew Rist * or more contributor license agreements. See the NOTICE file
5*b557fc96SAndrew Rist * distributed with this work for additional information
6*b557fc96SAndrew Rist * regarding copyright ownership. The ASF licenses this file
7*b557fc96SAndrew Rist * to you under the Apache License, Version 2.0 (the
8*b557fc96SAndrew Rist * "License"); you may not use this file except in compliance
9*b557fc96SAndrew Rist * with the License. You may obtain a copy of the License at
10*b557fc96SAndrew Rist *
11*b557fc96SAndrew Rist * http://www.apache.org/licenses/LICENSE-2.0
12*b557fc96SAndrew Rist *
13*b557fc96SAndrew Rist * Unless required by applicable law or agreed to in writing,
14*b557fc96SAndrew Rist * software distributed under the License is distributed on an
15*b557fc96SAndrew Rist * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16*b557fc96SAndrew Rist * KIND, either express or implied. See the License for the
17*b557fc96SAndrew Rist * specific language governing permissions and limitations
18*b557fc96SAndrew Rist * under the License.
19*b557fc96SAndrew Rist *
20*b557fc96SAndrew Rist *************************************************************/
21*b557fc96SAndrew Rist
22*b557fc96SAndrew Rist
23cdf0e10cSrcweir
24cdf0e10cSrcweir // MARKER(update_precomp.py): autogen include statement, do not remove
25cdf0e10cSrcweir #include "precompiled_fpicker.hxx"
26cdf0e10cSrcweir
27cdf0e10cSrcweir //------------------------------------------------------------------------
28cdf0e10cSrcweir // includes
29cdf0e10cSrcweir //------------------------------------------------------------------------
30cdf0e10cSrcweir
31cdf0e10cSrcweir #include <stdio.h>
32cdf0e10cSrcweir #include <osl/diagnose.h>
33cdf0e10cSrcweir #include "getfilenamewrapper.hxx"
34cdf0e10cSrcweir
35cdf0e10cSrcweir #if defined _MSC_VER
36cdf0e10cSrcweir #pragma warning(push, 1)
37cdf0e10cSrcweir #endif
38cdf0e10cSrcweir #include <objbase.h>
39cdf0e10cSrcweir #include <process.h>
40cdf0e10cSrcweir #if defined _MSC_VER
41cdf0e10cSrcweir #pragma warning(pop)
42cdf0e10cSrcweir #endif
43cdf0e10cSrcweir
44cdf0e10cSrcweir namespace /* private */
45cdf0e10cSrcweir {
46cdf0e10cSrcweir
47cdf0e10cSrcweir //-----------------------------------------------
48cdf0e10cSrcweir // This class prevents changing of the working
49cdf0e10cSrcweir // directory.
50cdf0e10cSrcweir //-----------------------------------------------
51cdf0e10cSrcweir class CurDirGuard
52cdf0e10cSrcweir {
53cdf0e10cSrcweir sal_Bool m_bValid;
54cdf0e10cSrcweir wchar_t* m_pBuffer;
55cdf0e10cSrcweir DWORD m_nBufLen;
56cdf0e10cSrcweir
57cdf0e10cSrcweir public:
CurDirGuard()58cdf0e10cSrcweir CurDirGuard()
59cdf0e10cSrcweir : m_bValid( sal_False )
60cdf0e10cSrcweir , m_pBuffer( NULL )
61cdf0e10cSrcweir , m_nBufLen( 0 )
62cdf0e10cSrcweir {
63cdf0e10cSrcweir m_nBufLen = GetCurrentDirectoryW( 0, NULL );
64cdf0e10cSrcweir if ( m_nBufLen )
65cdf0e10cSrcweir {
66cdf0e10cSrcweir m_pBuffer = new wchar_t[m_nBufLen];
67cdf0e10cSrcweir m_bValid = ( GetCurrentDirectoryW( m_nBufLen, m_pBuffer ) == ( m_nBufLen - 1 ) );
68cdf0e10cSrcweir }
69cdf0e10cSrcweir }
70cdf0e10cSrcweir
~CurDirGuard()71cdf0e10cSrcweir ~CurDirGuard()
72cdf0e10cSrcweir {
73cdf0e10cSrcweir bool bDirSet = false;
74cdf0e10cSrcweir
75cdf0e10cSrcweir if ( m_pBuffer )
76cdf0e10cSrcweir {
77cdf0e10cSrcweir if ( m_bValid )
78cdf0e10cSrcweir {
79cdf0e10cSrcweir if ( m_nBufLen - 1 > MAX_PATH )
80cdf0e10cSrcweir {
81cdf0e10cSrcweir if ( (LONG32)GetVersion() < 0 )
82cdf0e10cSrcweir {
83cdf0e10cSrcweir // this is Win 98/ME branch, such a long path can not be set
84cdf0e10cSrcweir // use the system path as fallback later
85cdf0e10cSrcweir }
86cdf0e10cSrcweir else
87cdf0e10cSrcweir {
88cdf0e10cSrcweir DWORD nNewLen = m_nBufLen + 8;
89cdf0e10cSrcweir wchar_t* pNewBuffer = new wchar_t[nNewLen];
90cdf0e10cSrcweir if ( m_nBufLen > 3 && m_pBuffer[0] == (wchar_t)'\\' && m_pBuffer[1] == (wchar_t)'\\' )
91cdf0e10cSrcweir {
92cdf0e10cSrcweir if ( m_pBuffer[2] == (wchar_t)'?' )
93cdf0e10cSrcweir _snwprintf( pNewBuffer, nNewLen, L"%s", m_pBuffer );
94cdf0e10cSrcweir else
95cdf0e10cSrcweir _snwprintf( pNewBuffer, nNewLen, L"\\\\?\\UNC\\%s", m_pBuffer+2 );
96cdf0e10cSrcweir }
97cdf0e10cSrcweir else
98cdf0e10cSrcweir _snwprintf( pNewBuffer, nNewLen, L"\\\\?\\%s", m_pBuffer );
99cdf0e10cSrcweir bDirSet = SetCurrentDirectoryW( pNewBuffer );
100cdf0e10cSrcweir
101cdf0e10cSrcweir delete [] pNewBuffer;
102cdf0e10cSrcweir }
103cdf0e10cSrcweir }
104cdf0e10cSrcweir else
105cdf0e10cSrcweir bDirSet = SetCurrentDirectoryW( m_pBuffer );
106cdf0e10cSrcweir }
107cdf0e10cSrcweir
108cdf0e10cSrcweir delete [] m_pBuffer;
109cdf0e10cSrcweir m_pBuffer = NULL;
110cdf0e10cSrcweir }
111cdf0e10cSrcweir
112cdf0e10cSrcweir if ( !bDirSet )
113cdf0e10cSrcweir {
114cdf0e10cSrcweir // the fallback solution
115cdf0e10cSrcweir wchar_t pPath[MAX_PATH+1];
116cdf0e10cSrcweir if ( GetWindowsDirectoryW( pPath, MAX_PATH+1 ) <= MAX_PATH )
117cdf0e10cSrcweir {
118cdf0e10cSrcweir SetCurrentDirectoryW( pPath );
119cdf0e10cSrcweir }
120cdf0e10cSrcweir else
121cdf0e10cSrcweir {
122cdf0e10cSrcweir // the system path is also too long?!!
123cdf0e10cSrcweir }
124cdf0e10cSrcweir }
125cdf0e10cSrcweir }
126cdf0e10cSrcweir };
127cdf0e10cSrcweir
128cdf0e10cSrcweir //-----------------------------------------------
129cdf0e10cSrcweir //
130cdf0e10cSrcweir //-----------------------------------------------
131cdf0e10cSrcweir
132cdf0e10cSrcweir struct GetFileNameParam
133cdf0e10cSrcweir {
GetFileNameParam__anon2365831b0111::GetFileNameParam134cdf0e10cSrcweir GetFileNameParam(bool bOpen, LPOPENFILENAME lpofn) :
135cdf0e10cSrcweir m_bOpen(bOpen),
136cdf0e10cSrcweir m_lpofn(lpofn),
137cdf0e10cSrcweir m_bRet(false),
138cdf0e10cSrcweir m_ExtErr(0)
139cdf0e10cSrcweir {}
140cdf0e10cSrcweir
141cdf0e10cSrcweir bool m_bOpen;
142cdf0e10cSrcweir LPOPENFILENAME m_lpofn;
143cdf0e10cSrcweir bool m_bRet;
144cdf0e10cSrcweir int m_ExtErr;
145cdf0e10cSrcweir };
146cdf0e10cSrcweir
147cdf0e10cSrcweir //-----------------------------------------------
148cdf0e10cSrcweir //
149cdf0e10cSrcweir //-----------------------------------------------
150cdf0e10cSrcweir
ThreadProc(void * pParam)151cdf0e10cSrcweir unsigned __stdcall ThreadProc(void* pParam)
152cdf0e10cSrcweir {
153cdf0e10cSrcweir CurDirGuard aGuard;
154cdf0e10cSrcweir
155cdf0e10cSrcweir GetFileNameParam* lpgfnp =
156cdf0e10cSrcweir reinterpret_cast<GetFileNameParam*>(pParam);
157cdf0e10cSrcweir
158cdf0e10cSrcweir HRESULT hr = OleInitialize( NULL );
159cdf0e10cSrcweir
160cdf0e10cSrcweir if (lpgfnp->m_bOpen)
161cdf0e10cSrcweir lpgfnp->m_bRet = GetOpenFileName(lpgfnp->m_lpofn);
162cdf0e10cSrcweir else
163cdf0e10cSrcweir lpgfnp->m_bRet = GetSaveFileName(lpgfnp->m_lpofn);
164cdf0e10cSrcweir
165cdf0e10cSrcweir lpgfnp->m_ExtErr = CommDlgExtendedError();
166cdf0e10cSrcweir
167cdf0e10cSrcweir if ( SUCCEEDED( hr ) )
168cdf0e10cSrcweir OleUninitialize();
169cdf0e10cSrcweir
170cdf0e10cSrcweir return 0;
171cdf0e10cSrcweir }
172cdf0e10cSrcweir
173cdf0e10cSrcweir //-----------------------------------------------
174cdf0e10cSrcweir // exceutes GetOpenFileName/GetSaveFileName in
175cdf0e10cSrcweir // a separat thread
176cdf0e10cSrcweir //-----------------------------------------------
177cdf0e10cSrcweir
ThreadExecGetFileName(LPOPENFILENAME lpofn,bool bOpen,int & ExtErr)178cdf0e10cSrcweir bool ThreadExecGetFileName(LPOPENFILENAME lpofn, bool bOpen, /*out*/ int& ExtErr)
179cdf0e10cSrcweir {
180cdf0e10cSrcweir GetFileNameParam gfnp(bOpen,lpofn);
181cdf0e10cSrcweir unsigned id;
182cdf0e10cSrcweir
183cdf0e10cSrcweir HANDLE hThread = reinterpret_cast<HANDLE>(
184cdf0e10cSrcweir _beginthreadex(0, 0, ThreadProc, &gfnp, 0, &id));
185cdf0e10cSrcweir
186cdf0e10cSrcweir OSL_POSTCOND(hThread, "could not create STA thread");
187cdf0e10cSrcweir
188cdf0e10cSrcweir WaitForSingleObject(hThread, INFINITE);
189cdf0e10cSrcweir CloseHandle(hThread);
190cdf0e10cSrcweir
191cdf0e10cSrcweir ExtErr = gfnp.m_ExtErr;
192cdf0e10cSrcweir
193cdf0e10cSrcweir return gfnp.m_bRet;
194cdf0e10cSrcweir }
195cdf0e10cSrcweir
196cdf0e10cSrcweir //-----------------------------------------------
197cdf0e10cSrcweir // This function returns true if the calling
198cdf0e10cSrcweir // thread belongs to a Multithreaded Appartment
199cdf0e10cSrcweir // (MTA)
200cdf0e10cSrcweir //-----------------------------------------------
201cdf0e10cSrcweir
IsMTA()202cdf0e10cSrcweir bool IsMTA()
203cdf0e10cSrcweir {
204cdf0e10cSrcweir HRESULT hr = CoInitialize(NULL);
205cdf0e10cSrcweir
206cdf0e10cSrcweir if (RPC_E_CHANGED_MODE == hr)
207cdf0e10cSrcweir return true;
208cdf0e10cSrcweir
209cdf0e10cSrcweir if(SUCCEEDED(hr))
210cdf0e10cSrcweir CoUninitialize();
211cdf0e10cSrcweir
212cdf0e10cSrcweir return false;
213cdf0e10cSrcweir }
214cdf0e10cSrcweir
215cdf0e10cSrcweir } // namespace private
216cdf0e10cSrcweir
217cdf0e10cSrcweir
218cdf0e10cSrcweir //-----------------------------------------------
219cdf0e10cSrcweir //
220cdf0e10cSrcweir //-----------------------------------------------
221cdf0e10cSrcweir
CGetFileNameWrapper()222cdf0e10cSrcweir CGetFileNameWrapper::CGetFileNameWrapper() :
223cdf0e10cSrcweir m_ExtendedDialogError(0)
224cdf0e10cSrcweir {
225cdf0e10cSrcweir }
226cdf0e10cSrcweir
227cdf0e10cSrcweir //-----------------------------------------------
228cdf0e10cSrcweir //
229cdf0e10cSrcweir //-----------------------------------------------
230cdf0e10cSrcweir
getOpenFileName(LPOPENFILENAME lpofn)231cdf0e10cSrcweir bool CGetFileNameWrapper::getOpenFileName(LPOPENFILENAME lpofn)
232cdf0e10cSrcweir {
233cdf0e10cSrcweir OSL_PRECOND(lpofn,"invalid parameter");
234cdf0e10cSrcweir
235cdf0e10cSrcweir bool bRet = false;
236cdf0e10cSrcweir
237cdf0e10cSrcweir if (IsMTA())
238cdf0e10cSrcweir {
239cdf0e10cSrcweir bRet = ThreadExecGetFileName(
240cdf0e10cSrcweir lpofn, true, m_ExtendedDialogError);
241cdf0e10cSrcweir }
242cdf0e10cSrcweir else
243cdf0e10cSrcweir {
244cdf0e10cSrcweir CurDirGuard aGuard;
245cdf0e10cSrcweir
246cdf0e10cSrcweir HRESULT hr = OleInitialize( NULL );
247cdf0e10cSrcweir
248cdf0e10cSrcweir bRet = GetOpenFileName(lpofn);
249cdf0e10cSrcweir m_ExtendedDialogError = CommDlgExtendedError();
250cdf0e10cSrcweir
251cdf0e10cSrcweir if ( SUCCEEDED( hr ) )
252cdf0e10cSrcweir OleUninitialize();
253cdf0e10cSrcweir }
254cdf0e10cSrcweir
255cdf0e10cSrcweir return bRet;
256cdf0e10cSrcweir }
257cdf0e10cSrcweir
258cdf0e10cSrcweir //-----------------------------------------------
259cdf0e10cSrcweir //
260cdf0e10cSrcweir //-----------------------------------------------
261cdf0e10cSrcweir
getSaveFileName(LPOPENFILENAME lpofn)262cdf0e10cSrcweir bool CGetFileNameWrapper::getSaveFileName(LPOPENFILENAME lpofn)
263cdf0e10cSrcweir {
264cdf0e10cSrcweir OSL_PRECOND(lpofn,"invalid parameter");
265cdf0e10cSrcweir
266cdf0e10cSrcweir bool bRet = false;
267cdf0e10cSrcweir
268cdf0e10cSrcweir if (IsMTA())
269cdf0e10cSrcweir {
270cdf0e10cSrcweir bRet = ThreadExecGetFileName(
271cdf0e10cSrcweir lpofn, false, m_ExtendedDialogError);
272cdf0e10cSrcweir }
273cdf0e10cSrcweir else
274cdf0e10cSrcweir {
275cdf0e10cSrcweir CurDirGuard aGuard;
276cdf0e10cSrcweir
277cdf0e10cSrcweir bRet = GetSaveFileName(lpofn);
278cdf0e10cSrcweir m_ExtendedDialogError = CommDlgExtendedError();
279cdf0e10cSrcweir }
280cdf0e10cSrcweir
281cdf0e10cSrcweir return bRet;
282cdf0e10cSrcweir }
283cdf0e10cSrcweir
284cdf0e10cSrcweir //-----------------------------------------------
285cdf0e10cSrcweir //
286cdf0e10cSrcweir //-----------------------------------------------
287cdf0e10cSrcweir
commDlgExtendedError()288cdf0e10cSrcweir int CGetFileNameWrapper::commDlgExtendedError( )
289cdf0e10cSrcweir {
290cdf0e10cSrcweir return m_ExtendedDialogError;
291cdf0e10cSrcweir }
292cdf0e10cSrcweir
293