michael@0: /*
michael@0:  * Copyright (c) 2002-2003 ActiveState Corp.
michael@0:  * Author: Trent Mick (TrentM@ActiveState.com)
michael@0:  * 
michael@0:  * Permission is hereby granted, free of charge, to any person obtaining a
michael@0:  * copy of this software and associated documentation files (the
michael@0:  * "Software"), to deal in the Software without restriction, including
michael@0:  * without limitation the rights to use, copy, modify, merge, publish,
michael@0:  * distribute, sublicense, and/or sell copies of the Software, and to
michael@0:  * permit persons to whom the Software is furnished to do so, subject to
michael@0:  * the following conditions:
michael@0:  * 
michael@0:  * The above copyright notice and this permission notice shall be included
michael@0:  * in all copies or substantial portions of the Software.
michael@0:  * 
michael@0:  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
michael@0:  * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
michael@0:  * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.
michael@0:  * IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY
michael@0:  * CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT,
michael@0:  * TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE
michael@0:  * SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
michael@0:  */
michael@0: 
michael@0: /* Console launch executable.  
michael@0:  * 
michael@0:  * This program exists solely to launch:
michael@0:  * 		python <installdir>/<exename>.py <argv>
michael@0:  * on Windows. "<exename>.py" must be in the same directory.
michael@0:  *
michael@0:  * Rationale:
michael@0:  *    - On some Windows flavours .py *can* be put on the PATHEXT to be
michael@0:  *      able to find "<exename>.py" if it is on the PATH. This is fine
michael@0:  *      until you need shell redirection to work. It does NOT for
michael@0:  *      extensions to PATHEXT.  Redirection *does* work for "python
michael@0:  *      <script>.py" so we will try to do that.
michael@0:  */
michael@0: 
michael@0: #ifdef WIN32
michael@0:     #include <windows.h>
michael@0:     #include <process.h>
michael@0:     #include <direct.h>
michael@0:     #include <shlwapi.h>
michael@0: #else /* linux */
michael@0:     #include <unistd.h>
michael@0: #endif /* WIN32 */
michael@0: #include <sys/stat.h>
michael@0: #include <errno.h>
michael@0: #include <stdio.h>
michael@0: #include <stdlib.h>
michael@0: #include <string.h>
michael@0: #include <stdarg.h>
michael@0: 
michael@0: //---- constants
michael@0: 
michael@0: #define BUF_LENGTH 2048
michael@0: #define MAX_PYTHON_ARGS 50
michael@0: #define MAX_FILES 50
michael@0: #define MAXPATHLEN 1024
michael@0: #ifdef WIN32
michael@0:     #define SEP '\\'
michael@0:     #define ALTSEP '/'
michael@0:     // path list element separator
michael@0:     #define DELIM ';'
michael@0: #else /* linux */
michael@0:     #define SEP '/'
michael@0:     // path list element separator
michael@0:     #define DELIM ':'
michael@0: #endif
michael@0: 
michael@0: #ifdef WIN32
michael@0:     #define spawnvp _spawnvp
michael@0:     #define snprintf _snprintf
michael@0:     #define vsnprintf _vsnprintf
michael@0:     //NOTE: this is for the stat *call* and the stat *struct*
michael@0:     #define stat _stat
michael@0: #endif
michael@0: 
michael@0: 
michael@0: //---- globals
michael@0: 
michael@0: char* programName = NULL;
michael@0: char* programPath = NULL;
michael@0: #ifndef WIN32 /* i.e. linux */
michael@0:     extern char **environ;   // the user environment
michael@0: #endif /* linux */
michael@0: 
michael@0: //---- error logging functions
michael@0: 
michael@0: void _LogError(const char* format ...)
michael@0: {
michael@0:     va_list ap;
michael@0:     va_start(ap, format);
michael@0: #if defined(WIN32) && defined(_WINDOWS)
michael@0:     // put up a MessageBox
michael@0:     char caption[BUF_LENGTH+1];
michael@0:     snprintf(caption, BUF_LENGTH, "Error in %s", programName);
michael@0:     char msg[BUF_LENGTH+1];
michael@0:     vsnprintf(msg, BUF_LENGTH, format, ap);
michael@0:     va_end(ap);
michael@0:     MessageBox(NULL, msg, caption, MB_OK | MB_ICONEXCLAMATION);
michael@0: #else
michael@0:     fprintf(stderr, "%s: error: ", programName);
michael@0:     vfprintf(stderr, format, ap);
michael@0:     va_end(ap);
michael@0: #endif /* WIN32 && _WINDOWS */
michael@0: }
michael@0: 
michael@0: 
michael@0: void _LogWarning(const char* format ...)
michael@0: {
michael@0:     va_list ap;
michael@0:     va_start(ap, format);
michael@0: #if defined(WIN32) && defined(_WINDOWS)
michael@0:     // put up a MessageBox
michael@0:     char caption[BUF_LENGTH+1];
michael@0:     snprintf(caption, BUF_LENGTH, "Warning in %s", programName);
michael@0:     char msg[BUF_LENGTH+1];
michael@0:     vsnprintf(msg, BUF_LENGTH, format, ap);
michael@0:     va_end(ap);
michael@0:     MessageBox(NULL, msg, caption, MB_OK | MB_ICONWARNING);
michael@0: #else
michael@0:     fprintf(stderr, "%s: warning: ", programName);
michael@0:     vfprintf(stderr, format, ap);
michael@0:     va_end(ap);
michael@0: #endif /* WIN32 && _WINDOWS */
michael@0: }
michael@0: 
michael@0: 
michael@0: 
michael@0: //---- utilities functions
michael@0: 
michael@0: /* _IsDir: Is the given dirname an existing directory */
michael@0: static int _IsDir(char *dirname)
michael@0: {
michael@0: #ifdef WIN32
michael@0:     DWORD dwAttrib;
michael@0:     dwAttrib = GetFileAttributes(dirname);
michael@0:     if (dwAttrib == -1) {
michael@0:         return 0;
michael@0:     }
michael@0:     if (dwAttrib & FILE_ATTRIBUTE_DIRECTORY) {
michael@0:         return 1;
michael@0:     }
michael@0:     return 0;
michael@0: #else /* i.e. linux */
michael@0:     struct stat buf;
michael@0:     if (stat(dirname, &buf) != 0)
michael@0:         return 0;
michael@0:     if (!S_ISDIR(buf.st_mode))
michael@0:         return 0;
michael@0:     return 1;
michael@0: #endif
michael@0: }
michael@0: 
michael@0: 
michael@0: /* _IsLink: Is the given filename a symbolic link */
michael@0: static int _IsLink(char *filename)
michael@0: {
michael@0: #ifdef WIN32
michael@0:     return 0;
michael@0: #else /* i.e. linux */
michael@0:     struct stat buf;
michael@0:     if (lstat(filename, &buf) != 0)
michael@0:         return 0;
michael@0:     if (!S_ISLNK(buf.st_mode))
michael@0:         return 0;
michael@0:     return 1;
michael@0: #endif
michael@0: }
michael@0: 
michael@0: 
michael@0: /* Is executable file
michael@0:  * On Linux: check 'x' permission. On Windows: just check existence.
michael@0:  */
michael@0: static int _IsExecutableFile(char *filename)
michael@0: {
michael@0: #ifdef WIN32
michael@0:     return (int)PathFileExists(filename);
michael@0: #else /* i.e. linux */
michael@0:     struct stat buf;
michael@0:     if (stat(filename, &buf) != 0)
michael@0:         return 0;
michael@0:     if (!S_ISREG(buf.st_mode))
michael@0:         return 0;
michael@0:     if ((buf.st_mode & 0111) == 0)
michael@0:         return 0;
michael@0:     return 1;
michael@0: #endif /* WIN32 */
michael@0: }
michael@0: 
michael@0: 
michael@0: /* _GetProgramPath: Determine the absolute path to the given program name.
michael@0:  *
michael@0:  *      Takes into account the current working directory, etc.
michael@0:  *      The implementations require the global 'programName' to be set.
michael@0:  */
michael@0: #ifdef WIN32
michael@0:     static char* _GetProgramPath(void)
michael@0:     {
michael@0:         //XXX this is ugly but I didn't want to use malloc, no reason
michael@0:         static char progPath[MAXPATHLEN+1];
michael@0:         // get absolute path to module
michael@0:         if (!GetModuleFileName(NULL, progPath, MAXPATHLEN)) {
michael@0:             _LogError("could not get absolute program name from "\
michael@0:                 "GetModuleFileName\n");
michael@0:             exit(1);
michael@0:         }
michael@0:         // just need dirname
michael@0:         for (char* p = progPath+strlen(progPath);
michael@0:              *p != SEP && *p != ALTSEP;
michael@0:              --p)
michael@0:         {
michael@0:             *p = '\0';
michael@0:         }
michael@0:         *p = '\0';  // remove the trailing SEP as well
michael@0: 
michael@0:         return progPath;
michael@0:     }
michael@0: #else
michael@0: 
michael@0:     /* _JoinPath requires that any buffer argument passed to it has at
michael@0:        least MAXPATHLEN + 1 bytes allocated.  If this requirement is met,
michael@0:        it guarantees that it will never overflow the buffer.  If stuff
michael@0:        is too long, buffer will contain a truncated copy of stuff.
michael@0:     */
michael@0:     static void
michael@0:     _JoinPath(char *buffer, char *stuff)
michael@0:     {
michael@0:         size_t n, k;
michael@0:         if (stuff[0] == SEP)
michael@0:             n = 0;
michael@0:         else {
michael@0:             n = strlen(buffer);
michael@0:             if (n > 0 && buffer[n-1] != SEP && n < MAXPATHLEN)
michael@0:                 buffer[n++] = SEP;
michael@0:         }
michael@0:         k = strlen(stuff);
michael@0:         if (n + k > MAXPATHLEN)
michael@0:             k = MAXPATHLEN - n;
michael@0:         strncpy(buffer+n, stuff, k);
michael@0:         buffer[n+k] = '\0';
michael@0:     }
michael@0: 
michael@0: 
michael@0:     static char*
michael@0:     _GetProgramPath(void)
michael@0:     {
michael@0:         /* XXX this routine does *no* error checking */
michael@0:         char* path = getenv("PATH");
michael@0:         static char progPath[MAXPATHLEN+1];
michael@0: 
michael@0:         /* If there is no slash in the argv0 path, then we have to
michael@0:          * assume the program is on the user's $PATH, since there's no
michael@0:          * other way to find a directory to start the search from.  If
michael@0:          * $PATH isn't exported, you lose.
michael@0:          */
michael@0:         if (strchr(programName, SEP)) {
michael@0:             strncpy(progPath, programName, MAXPATHLEN);
michael@0:         }
michael@0:         else if (path) {
michael@0:             int bufspace = MAXPATHLEN;
michael@0:             while (1) {
michael@0:                 char *delim = strchr(path, DELIM);
michael@0: 
michael@0:                 if (delim) {
michael@0:                     size_t len = delim - path;
michael@0:                     if (len > bufspace) {
michael@0:                         len = bufspace;
michael@0:                     }
michael@0:                     strncpy(progPath, path, len);
michael@0:                     *(progPath + len) = '\0';
michael@0:                     bufspace -= len;
michael@0:                 }
michael@0:                 else {
michael@0:                     strncpy(progPath, path, bufspace);
michael@0:                 }
michael@0: 
michael@0:                 _JoinPath(progPath, programName);
michael@0:                 if (_IsExecutableFile(progPath)) {
michael@0:                     break;
michael@0:                 }
michael@0: 
michael@0:                 if (!delim) {
michael@0:                     progPath[0] = '\0';
michael@0:                     break;
michael@0:                 }
michael@0:                 path = delim + 1;
michael@0:             }
michael@0:         }
michael@0:         else {
michael@0:             progPath[0] = '\0';
michael@0:         }
michael@0: 
michael@0:         // now we have to resolve a string of possible symlinks
michael@0:         //   - we'll just handle the simple case of a single level of
michael@0:         //     indirection
michael@0:         //
michael@0:         // XXX note this does not handle multiple levels of symlinks
michael@0:         //     here is pseudo-code for that (please implement it :):
michael@0:         // while 1:
michael@0:         //     if islink(progPath):
michael@0:         //         linkText = readlink(progPath)
michael@0:         //         if isabsolute(linkText):
michael@0:         //             progPath = os.path.join(dirname(progPath), linkText)
michael@0:         //         else:
michael@0:         //             progPath = linkText
michael@0:         //     else:
michael@0:         //         break
michael@0:         if (_IsLink(progPath)) {
michael@0:             char newProgPath[MAXPATHLEN+1];
michael@0:             readlink(progPath, newProgPath, MAXPATHLEN);
michael@0:             strncpy(progPath, newProgPath, MAXPATHLEN);
michael@0:         }
michael@0: 
michael@0:         
michael@0:         // prefix with the current working directory if the path is
michael@0:         // relative to conform with the Windows version of this
michael@0:         if (strlen(progPath) != 0 && progPath[0] != SEP) {
michael@0:             char cwd[MAXPATHLEN+1];
michael@0:             char tmp[MAXPATHLEN+1];
michael@0:             //XXX should check for failure retvals
michael@0:             getcwd(cwd, MAXPATHLEN);
michael@0:             snprintf(tmp, MAXPATHLEN, "%s%c%s", cwd, SEP, progPath);
michael@0:             strncpy(progPath, tmp, MAXPATHLEN);
michael@0:         }
michael@0:         
michael@0:         // 'progPath' now contains the full path to the program *and* the program
michael@0:         // name. The latter is not desire.
michael@0:         char* pLetter = progPath + strlen(progPath);
michael@0:         for (;pLetter != progPath && *pLetter != SEP; --pLetter) {
michael@0:             /* do nothing */
michael@0:         }
michael@0:         *pLetter = '\0';
michael@0: 
michael@0:         return progPath;
michael@0:     }
michael@0: #endif  /* WIN32 */
michael@0: 
michael@0: 
michael@0: //---- mainline
michael@0: 
michael@0: int main(int argc, char** argv)
michael@0: {
michael@0:     programName = argv[0];
michael@0:     programPath = _GetProgramPath();
michael@0: 
michael@0:     // Determine the extension-less program basename.
michael@0:     // XXX Will not always handle app names with '.' in them (other than
michael@0:     //     the '.' for the extension.
michael@0:     char programNameNoExt[MAXPATHLEN+1];
michael@0:     char *pStart, *pEnd;
michael@0:     pStart = pEnd = programName + strlen(programName) - 1;
michael@0:     while (pStart != programName && *(pStart-1) != SEP) {
michael@0: 	pStart--;
michael@0:     }
michael@0:     while (1) {
michael@0: 	if (pEnd == pStart) {
michael@0:             pEnd = programName + strlen(programName) - 1;
michael@0: 	    break;
michael@0: 	}
michael@0: 	pEnd--;
michael@0: 	if (*(pEnd+1) == '.') {
michael@0: 	    break;
michael@0: 	}
michael@0:     }
michael@0:     strncpy(programNameNoExt, pStart, pEnd-pStart+1);
michael@0:     *(programNameNoExt+(pEnd-pStart+1)) = '\0';
michael@0: 
michael@0:     // determine the full path to "<exename>.py"
michael@0:     char pyFile[MAXPATHLEN+1];
michael@0:     snprintf(pyFile, MAXPATHLEN, "%s%c%s.py", programPath, SEP,
michael@0: 	     programNameNoExt);
michael@0:     
michael@0:     // Build the argument array for launching.
michael@0:     char* pythonArgs[MAX_PYTHON_ARGS+1];
michael@0:     int nPythonArgs = 0;
michael@0:     pythonArgs[nPythonArgs++] = "python";
michael@0:     pythonArgs[nPythonArgs++] = "-tt";
michael@0:     pythonArgs[nPythonArgs++] = pyFile;
michael@0:     for (int i = 1; i < argc; ++i) {
michael@0:         pythonArgs[nPythonArgs++] = argv[i];
michael@0:     }
michael@0:     pythonArgs[nPythonArgs++] = NULL;
michael@0: 
michael@0:     return _spawnvp(_P_WAIT, pythonArgs[0], pythonArgs);
michael@0: }
michael@0: 
michael@0: 
michael@0: //---- mainline for win32 subsystem:windows app
michael@0: #ifdef WIN32
michael@0:     int WINAPI WinMain(
michael@0:         HINSTANCE hInstance,      /* handle to current instance */
michael@0:         HINSTANCE hPrevInstance,  /* handle to previous instance */
michael@0:         LPSTR lpCmdLine,          /* pointer to command line */
michael@0:         int nCmdShow              /* show state of window */
michael@0:     )
michael@0:     {
michael@0:         return main(__argc, __argv);
michael@0:     }
michael@0: #endif
michael@0: