fpicker/source/win32/filepicker/getfilenamewrapper.cxx

   1 /*************************************************************************
   2  *
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * Copyright 2008 by Sun Microsystems, Inc.
   6  *
   7  * OpenOffice.org - a multi-platform office productivity suite
   8  *
   9  * $RCSfile: getfilenamewrapper.cxx,v $
  10  * $Revision: 1.10 $
  11  *
  12  * This file is part of OpenOffice.org.
  13  *
  14  * OpenOffice.org is free software: you can redistribute it and/or modify
  15  * it under the terms of the GNU Lesser General Public License version 3
  16  * only, as published by the Free Software Foundation.
  17  *
  18  * OpenOffice.org is distributed in the hope that it will be useful,
  19  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  20  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  21  * GNU Lesser General Public License version 3 for more details
  22  * (a copy is included in the LICENSE file that accompanied this code).
  23  *
  24  * You should have received a copy of the GNU Lesser General Public License
  25  * version 3 along with OpenOffice.org.  If not, see
  26  * <http://www.openoffice.org/license.html>
  27  * for a copy of the LGPLv3 License.
  28  *
  29  ************************************************************************/
  30
  31 // MARKER(update_precomp.py): autogen include statement, do not remove
  32 #include "precompiled_fpicker.hxx"
  33
  34 //------------------------------------------------------------------------
  35 // includes
  36 //------------------------------------------------------------------------
  37
  38 #include <stdio.h>
  39 #include <osl/diagnose.h>
  40 #include "getfilenamewrapper.hxx"
  41
  42 #if defined _MSC_VER
  43 #pragma warning(push, 1)
  44 #endif
  45 #include <objbase.h>
  46 #include <process.h>
  47 #if defined _MSC_VER
  48 #pragma warning(pop)
  49 #endif
  50
  51 namespace /* private */
  52 {
  53
  54     //-----------------------------------------------
  55     // This class prevents changing of the working
  56     // directory.
  57     //-----------------------------------------------
  58     class CurDirGuard
  59     {
  60         BOOL m_bValid;
  61         wchar_t* m_pBuffer;
  62         DWORD m_nBufLen;
  63
  64     public:
  65         CurDirGuard()
  66         : m_bValid( FALSE )
  67         , m_pBuffer( NULL )
  68         , m_nBufLen( 0 )
  69         {
  70             m_nBufLen = GetCurrentDirectoryW( 0, NULL );
  71             if ( m_nBufLen )
  72             {
  73                 m_pBuffer = new wchar_t[m_nBufLen];
  74                 m_bValid = ( GetCurrentDirectoryW( m_nBufLen, m_pBuffer ) == ( m_nBufLen - 1 ) );
  75             }
  76         }
  77
  78         ~CurDirGuard()
  79         {
  80             BOOL bDirSet = FALSE;
  81
  82             if ( m_pBuffer )
  83             {
  84                 if ( m_bValid )
  85                 {
  86                     if ( m_nBufLen - 1 > MAX_PATH )
  87                     {
  88                         if ( (LONG32)GetVersion() < 0 )
  89                         {
  90                             // this is Win 98/ME branch, such a long path can not be set
  91                             // use the system path as fallback later
  92                         }
  93                         else
  94                         {
  95                             DWORD nNewLen = m_nBufLen + 8;
  96                             wchar_t* pNewBuffer = new wchar_t[nNewLen];
  97                             if ( m_nBufLen > 3 && m_pBuffer[0] == (wchar_t)'\\' && m_pBuffer[1] == (wchar_t)'\\' )
  98                             {
  99                                 if ( m_pBuffer[2] == (wchar_t)'?' )
 100                                     _snwprintf( pNewBuffer, nNewLen, L"%s", m_pBuffer );
 101                                 else
 102                                     _snwprintf( pNewBuffer, nNewLen, L"\\\\?\\UNC\\%s", m_pBuffer+2 );
 103                             }
 104                             else
 105                                 _snwprintf( pNewBuffer, nNewLen, L"\\\\?\\%s", m_pBuffer );
 106                             bDirSet = SetCurrentDirectoryW( pNewBuffer );
 107
 108                             delete [] pNewBuffer;
 109                         }
 110                     }
 111                     else
 112                         bDirSet = SetCurrentDirectoryW( m_pBuffer );
 113                 }
 114
 115                 delete [] m_pBuffer;
 116                 m_pBuffer = NULL;
 117             }
 118
 119             if ( !bDirSet )
 120             {
 121                 // the fallback solution
 122                 wchar_t pPath[MAX_PATH+1];
 123                 if ( GetWindowsDirectoryW( pPath, MAX_PATH+1 ) <= MAX_PATH )
 124                 {
 125                     SetCurrentDirectoryW( pPath );
 126                 }
 127                 else
 128                 {
 129                     // the system path is also too long?!!
 130                 }
 131             }
 132         }
 133     };
 134
 135     //-----------------------------------------------
 136     //
 137     //-----------------------------------------------
 138
 139     struct GetFileNameParam
 140     {
 141         GetFileNameParam(bool bOpen, LPOPENFILENAME lpofn) :
 142             m_bOpen(bOpen),
 143             m_lpofn(lpofn),
 144             m_bRet(false),
 145             m_ExtErr(0)
 146         {}
 147
 148         bool                    m_bOpen;
 149         LPOPENFILENAME  m_lpofn;
 150         bool                    m_bRet;
 151         int                             m_ExtErr;
 152     };
 153
 154     //-----------------------------------------------
 155     //
 156     //-----------------------------------------------
 157
 158     unsigned __stdcall ThreadProc(void* pParam)
 159     {
 160         CurDirGuard aGuard;
 161
 162         GetFileNameParam* lpgfnp =
 163             reinterpret_cast<GetFileNameParam*>(pParam);
 164
 165         HRESULT hr = OleInitialize( NULL );
 166
 167         if (lpgfnp->m_bOpen)
 168             lpgfnp->m_bRet = GetOpenFileName(lpgfnp->m_lpofn);
 169         else
 170             lpgfnp->m_bRet = GetSaveFileName(lpgfnp->m_lpofn);
 171
 172         lpgfnp->m_ExtErr = CommDlgExtendedError();
 173
 174         if ( SUCCEEDED( hr ) )
 175             OleUninitialize();
 176
 177         return 0;
 178     }
 179
 180     //-----------------------------------------------
 181     // exceutes GetOpenFileName/GetSaveFileName in
 182     // a separat thread
 183     //-----------------------------------------------
 184
 185     bool ThreadExecGetFileName(LPOPENFILENAME lpofn, bool bOpen, /*out*/ int& ExtErr)
 186     {
 187         GetFileNameParam gfnp(bOpen,lpofn);
 188         unsigned         id;
 189
 190         HANDLE hThread = reinterpret_cast<HANDLE>(
 191             _beginthreadex(0, 0, ThreadProc, &gfnp, 0, &id));
 192
 193         OSL_POSTCOND(hThread, "could not create STA thread");
 194
 195         WaitForSingleObject(hThread, INFINITE);
 196         CloseHandle(hThread);
 197
 198         ExtErr = gfnp.m_ExtErr;
 199
 200         return gfnp.m_bRet;
 201     }
 202
 203     //-----------------------------------------------
 204     // This function returns true if the calling
 205     // thread belongs to a Multithreaded Appartment
 206     // (MTA)
 207     //-----------------------------------------------
 208
 209     bool IsMTA()
 210     {
 211         HRESULT hr = CoInitialize(NULL);
 212
 213         if (RPC_E_CHANGED_MODE == hr)
 214             return true;
 215
 216         if(SUCCEEDED(hr))
 217             CoUninitialize();
 218
 219         return false;
 220     }
 221
 222 } // namespace private
 223
 224
 225 //-----------------------------------------------
 226 //
 227 //-----------------------------------------------
 228
 229 CGetFileNameWrapper::CGetFileNameWrapper() :
 230     m_ExtendedDialogError(0)
 231 {
 232 }
 233
 234 //-----------------------------------------------
 235 //
 236 //-----------------------------------------------
 237
 238 bool CGetFileNameWrapper::getOpenFileName(LPOPENFILENAME lpofn)
 239 {
 240     OSL_PRECOND(lpofn,"invalid parameter");
 241
 242     bool bRet = false;
 243
 244     if (IsMTA())
 245     {
 246         bRet = ThreadExecGetFileName(
 247             lpofn, true, m_ExtendedDialogError);
 248     }
 249     else
 250     {
 251         CurDirGuard aGuard;
 252
 253         HRESULT hr = OleInitialize( NULL );
 254
 255         bRet = GetOpenFileName(lpofn);
 256         m_ExtendedDialogError = CommDlgExtendedError();
 257
 258         if ( SUCCEEDED( hr ) )
 259             OleUninitialize();
 260     }
 261
 262     return bRet;
 263 }
 264
 265 //-----------------------------------------------
 266 //
 267 //-----------------------------------------------
 268
 269 bool CGetFileNameWrapper::getSaveFileName(LPOPENFILENAME lpofn)
 270 {
 271     OSL_PRECOND(lpofn,"invalid parameter");
 272
 273     bool bRet = false;
 274
 275     if (IsMTA())
 276     {
 277         bRet = ThreadExecGetFileName(
 278             lpofn, false, m_ExtendedDialogError);
 279     }
 280     else
 281     {
 282         CurDirGuard aGuard;
 283
 284         bRet = GetSaveFileName(lpofn);
 285         m_ExtendedDialogError = CommDlgExtendedError();
 286     }
 287
 288     return bRet;
 289 }
 290
 291 //-----------------------------------------------
 292 //
 293 //-----------------------------------------------
 294
 295 int CGetFileNameWrapper::commDlgExtendedError( )
 296 {
 297     return m_ExtendedDialogError;
 298 }
 299