sal/systools/win32/uwinapi/CommandLineToArgvW.cpp

   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: CommandLineToArgvW.cpp,v $
  10  * $Revision: 1.6 $
  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 #if defined(_MSC_VER) && (_MSC_VER >= 1400)
  31 #pragma warning(disable:4740)
  32 #endif
  33
  34 #include "macros.h"
  35
  36 #ifdef __cplusplus
  37 #define local inline
  38 #else
  39 #define local static
  40 #endif
  41
  42 local LPCWSTR SkipBlanks( LPCWSTR lpScan )
  43 {
  44     while ( ' ' == *lpScan || '\t' == *lpScan )
  45         lpScan++;
  46
  47     return lpScan;
  48 }
  49
  50
  51 local LPCWSTR SkipArgument( LPCWSTR lpScan )
  52 {
  53     BOOL        fQuoted = FALSE;
  54     LPCWSTR     lpArgEnd = NULL;
  55
  56     do
  57     {
  58         switch ( *lpScan )
  59         {
  60         case ' ':
  61         case '\t':
  62             if ( fQuoted )
  63                 lpScan++;
  64             else
  65                 lpArgEnd = lpScan;
  66             break;
  67         case '\"':
  68             lpScan++;
  69             fQuoted = !fQuoted;
  70             break;
  71         case '\0':
  72             lpArgEnd = lpScan;
  73             break;
  74         default:
  75             lpScan++;
  76             break;
  77         }
  78     } while( *lpScan && !lpArgEnd );
  79
  80     return lpScan;
  81 }
  82
  83
  84 IMPLEMENT_THUNK( shell32, WINDOWS, LPWSTR *, WINAPI, CommandLineToArgvW, ( LPCWSTR lpCmdLineW, int *pNumArgs ) )
  85 {
  86     LPWSTR      *lpArgvW = NULL;
  87
  88     if ( !lpCmdLineW || !*lpCmdLineW )
  89     {
  90         CHAR    szFileName[MAX_PATH];
  91
  92         DWORD   dwResult = GetModuleFileNameA( NULL, szFileName, MAX_PATH );
  93
  94         if ( dwResult && dwResult < MAX_PATH )
  95         {
  96             int cchNeeded = MultiByteToWideChar( CP_ACP, 0, szFileName, -1, NULL, 0 );
  97
  98             lpArgvW = (LPWSTR *)GlobalAlloc( 0, cchNeeded * sizeof(WCHAR) + sizeof(LPWSTR) );
  99
 100             if ( lpArgvW )
 101             {
 102                 lpArgvW[0] = (LPWSTR)(lpArgvW + 1);
 103
 104                 MultiByteToWideChar( CP_ACP, 0, szFileName, -1, lpArgvW[0], cchNeeded );
 105                 *pNumArgs = 1;
 106             }
 107             else
 108                 SetLastError( ERROR_OUTOFMEMORY );
 109         }
 110     }
 111     else
 112     {
 113         LPCWSTR lpScan = lpCmdLineW;
 114         int             nTokens = 0;
 115         int             cchNeeded = 0;
 116
 117         // Count arguments and required size
 118
 119         while ( *lpScan )
 120         {
 121             lpScan = SkipBlanks( lpScan );
 122             if ( *lpScan )
 123             {
 124                 LPCWSTR lpArgEnd = SkipArgument( lpScan );
 125
 126                 nTokens++;
 127                 cchNeeded += lpArgEnd - lpScan + 1;
 128                 lpScan = lpArgEnd;
 129             }
 130         }
 131
 132         // Allocate space for one additional NULL pointer to terminate list
 133
 134         lpArgvW = (LPWSTR *)GlobalAlloc( 0, sizeof(LPWSTR) * (nTokens + 1) + sizeof(WCHAR) * cchNeeded );
 135
 136         if ( lpArgvW )
 137         {
 138             // Collect arguments
 139
 140             LPWSTR      lpDestination = (LPWSTR)&lpArgvW[nTokens + 1];
 141
 142             lpScan = lpCmdLineW;
 143             nTokens = 0;
 144
 145             while ( *lpScan )
 146             {
 147                 lpScan = SkipBlanks( lpScan );
 148                 if ( *lpScan )
 149                 {
 150                     LPCWSTR     lpArgEnd = SkipArgument( lpScan );
 151
 152                     lpArgvW[nTokens++] = lpDestination;
 153
 154                     while ( lpScan < lpArgEnd )
 155                     {
 156                         if ( '\"' != *lpScan )
 157                             *lpDestination++ = *lpScan;
 158
 159                         lpScan++;
 160                     }
 161                     *lpDestination++ = 0;
 162                 }
 163             }
 164
 165             lpArgvW[nTokens] = NULL;
 166
 167             *pNumArgs = nTokens;
 168         }
 169         else
 170             SetLastError( ERROR_OUTOFMEMORY );
 171
 172     }
 173
 174     return lpArgvW;
 175 }