Annotation of parser3/src/main/pa_exec.C, revision 1.48.2.11

1.1       paf         1: /** @file
                      2:        Parser: program executing for different OS-es.
                      3: 
1.48.2.5  paf         4:        Copyright(c) 2000,2001-2003 ArtLebedev Group(http://www.artlebedev.com)
1.25      paf         5:        Author: Alexandr Petrosian <paf@design.ru> (http://paf.design.ru)
1.1       paf         6: 
1.18      paf         7:        @todo setrlimit
1.1       paf         8: */
1.38      paf         9: 
1.48.2.11! paf        10: static const char* IDENT_EXEC_C="$Date: 2003/02/04 12:08:57 $";
1.1       paf        11: 
                     12: #include "pa_config_includes.h"
                     13: 
1.19      paf        14: #include "pa_exec.h"
                     15: #include "pa_exception.h"
                     16: #include "pa_common.h"
                     17: 
1.1       paf        18: #ifdef WIN32
                     19: #      include <windows.h>
                     20: #else
                     21: #      include <signal.h>
                     22: #      include <sys/types.h>
                     23: #      include <sys/wait.h>
                     24: #endif
                     25: 
                     26: #ifdef WIN32
                     27: 
                     28: /// this func from http://www.ccas.ru/~posp/popov/spawn.htm
1.36      paf        29: static DWORD CreateHiddenConsoleProcess(LPCTSTR szCmdLine,
1.1       paf        30:                                                                                char *szEnv,
                     31:                                         PROCESS_INFORMATION* ppi, 
                     32:                                         LPHANDLE phInWrite,
                     33:                                         LPHANDLE phOutRead,
1.9       parser     34:                                         LPHANDLE phErrRead)
1.1       paf        35: {
1.36      paf        36:     DWORD result=0;
                     37:        BOOL fCreated;
1.1       paf        38:     STARTUPINFO si;
                     39:     SECURITY_ATTRIBUTES sa={0};
                     40:     HANDLE hInRead;
                     41:     HANDLE hOutWrite;
                     42:     HANDLE hErrWrite;
                     43: 
                     44:     // Create pipes
                     45:     // initialize security attributes for handle inheritance (for WinNT)
1.9       parser     46:     sa.nLength=sizeof(sa);
                     47:     sa.bInheritHandle=TRUE;
                     48:     sa.lpSecurityDescriptor=NULL;
1.1       paf        49: 
                     50:     // create STDIN pipe
1.9       parser     51:     if(!CreatePipe(&hInRead, phInWrite, &sa, 0))
1.1       paf        52:         goto error;
                     53: 
                     54:     // create STDOUT pipe
1.9       parser     55:     if(!CreatePipe(phOutRead, &hOutWrite, &sa, 0))
1.1       paf        56:         goto error;
                     57: 
                     58:     // create STDERR pipe
1.9       parser     59:     if(!CreatePipe(phErrRead, &hErrWrite, &sa, 0))
1.1       paf        60:         goto error;
                     61: 
                     62:     // process startup information
1.9       parser     63:     memset(&si, 0, sizeof(si));
                     64:     si.cb=sizeof(si); 
                     65:     si.dwFlags=STARTF_USESHOWWINDOW | STARTF_USESTDHANDLES;
1.1       paf        66:     // child process' console must be hidden for Win95 compatibility
1.9       parser     67:     si.wShowWindow=SW_HIDE;
1.1       paf        68:     // assign "other" sides of pipes
1.9       parser     69:     si.hStdInput=hInRead;
                     70:     si.hStdOutput=hOutWrite;
                     71:     si.hStdError=hErrWrite;
1.1       paf        72: 
1.31      paf        73:        // calculating script's directory
                     74:        char dir[MAX_STRING];
                     75:        strncpy(dir, szCmdLine, MAX_STRING-1); dir[MAX_STRING-1]=0;
1.33      paf        76:        lsplit(dir,' '); // trim arguments
1.31      paf        77:        rsplit(dir,'/'); rsplit(dir,'\\'); // trim filename
                     78:        chdir(dir);
                     79: 
1.1       paf        80:     // Create a child process (suspended)
1.9       parser     81:     fCreated=CreateProcess(NULL,
1.26      paf        82:                               (LPTSTR)szCmdLine,
1.1       paf        83:                               NULL,
                     84:                               NULL,
                     85:                               TRUE,
1.30      paf        86:                               CREATE_NO_WINDOW,
1.1       paf        87:                               szEnv,
1.31      paf        88:                               dir,
1.1       paf        89:                               &si,
1.9       parser     90:                               ppi);
1.36      paf        91:        if(!fCreated)
                     92:                result=GetLastError();
1.1       paf        93: 
1.9       parser     94:     CloseHandle(hInRead);
                     95:     CloseHandle(hOutWrite);
                     96:     CloseHandle(hErrWrite);
1.1       paf        97: 
1.9       parser     98:     if(!fCreated)
1.1       paf        99:         goto error;
                    100: 
1.36      paf       101:     return result;
1.1       paf       102: 
                    103: error:
1.36      paf       104:        if(!result/*yet*/)
                    105:                result=GetLastError(); // get it
                    106: 
1.9       parser    107:     CloseHandle(*phInWrite);
                    108:     CloseHandle(*phOutRead);
                    109:     CloseHandle(*phErrRead);
1.1       paf       110: 
1.36      paf       111:     return result;
1.1       paf       112: }
                    113: 
1.48.2.5  paf       114: static void read_pipe(Pool& pool, StringPtr result, HANDLE hOutRead, const char* file_spec, 
1.35      paf       115:                                          String::Untaint_lang lang){
1.1       paf       116:        while(true) {
1.48.2.2  paf       117:                char *buf=new(pool) char[MAX_STRING];
1.1       paf       118:                unsigned long size;
1.30      paf       119:                if(!ReadFile(hOutRead, buf, MAX_STRING, &size, NULL) || !size) 
1.1       paf       120:                        break;
1.48.2.2  paf       121:                result->APPEND(buf, size, lang, file_spec, 0);
1.1       paf       122:     }
                    123: }
                    124: 
1.48.2.11! paf       125: static CharPtr buildCommand(CharPtr file_spec_cstr, ArrayString& argv) {
1.48.2.2  paf       126:        CharPtr result=file_spec_cstr;
1.4       paf       127:        if(FILE *f=fopen(file_spec_cstr, "r")) {
1.48.2.2  paf       128:                try {
1.1       paf       129:                char buf[MAX_STRING];
                    130:                size_t size=fread(buf, 1, MAX_STRING-1, f);
                    131:                if(size>2) {
                    132:                        buf[size]=0;
                    133:                        if(strncmp(buf, "#!", 2)==0) {
1.48.2.5  paf       134:                                const char* begin=buf+2;
1.4       paf       135:                                if(*begin==' ') // alx: were an old magic for some linux-es
                    136:                                        begin++;
                    137:                                if(char *end=strchr(begin, '\n')) {
1.48.2.2  paf       138:                                        String string(begin, end-begin);
1.1       paf       139:                                        string << " " << file_spec_cstr;
1.8       parser    140:                                        result=string.cstr();
1.1       paf       141:                                }
                    142:                        }
                    143:                }
1.48.2.2  paf       144:                } catch(...) {
                    145:                        fclose(f);
                    146:                        rethrow;
                    147:                }
1.1       paf       148:                fclose(f);
                    149:        }
1.48.2.2  paf       150:        { // appending argv
                    151:                String string(result);
                    152:                for(int i=0; i<argv.count(); i++) {
                    153:                        string << " ";
                    154:                        string << *argv[i];
1.8       parser    155:                }
                    156: 
1.48.2.2  paf       157:                result=string.cstr();
1.8       parser    158:        }
                    159: 
                    160:        return result;
1.1       paf       161: }
                    162: 
                    163: #else
                    164: 
1.48.2.5  paf       165: static pid_t execve_piped(const char* file_spec_cstr, 
1.27      paf       166:                        char * const argv[], char * const env[],
                    167:                        int *pipe_in, int *pipe_out, int *pipe_err) {
1.46      paf       168:        pid_t pid;
1.1       paf       169:        int in_fds[2];
                    170:        int out_fds[2];
                    171:        int err_fds[2];
                    172:        int save_errno;
                    173:        
                    174:        if(pipe_in && pipe(in_fds)<0) {
                    175:                save_errno=errno;
                    176:                errno=save_errno;
                    177:                return 0;
                    178:        }
                    179:        
                    180:        if(pipe_out && pipe(out_fds)<0) {
                    181:                save_errno=errno;
                    182:                if(pipe_in) {
                    183:                        close(in_fds[0]); close(in_fds[1]);
                    184:                }
                    185:                errno=save_errno;
                    186:                return 0;
                    187:        }
                    188:        
                    189:        if(pipe_err && pipe(err_fds)<0) {
                    190:                save_errno=errno;
                    191:                if(pipe_in) {
                    192:                        close(in_fds[0]); close(in_fds[1]);
                    193:                }
                    194:                if(pipe_out) {
                    195:                        close(out_fds[0]); close(out_fds[1]);
                    196:                }
                    197:                errno=save_errno;
                    198:                return 0;
                    199:        }
                    200:        
                    201:        if((pid=fork())<0) {
                    202:                save_errno=errno;
                    203:                if(pipe_in) {
                    204:                        close(in_fds[0]); close(in_fds[1]);
                    205:                }
                    206:                if(pipe_out) {
                    207:                        close(out_fds[0]); close(out_fds[1]);
                    208:                }
                    209:                if(pipe_err) {
                    210:                        close(err_fds[0]); close(err_fds[1]);
                    211:                }
                    212:                errno=save_errno;
1.46      paf       213:                return -1;
1.1       paf       214:        }
                    215:        
                    216:        if(!pid) {
                    217:                /* Child process */
                    218:                
                    219:                if(pipe_out) {
                    220:                        close(out_fds[0]);
                    221:                        dup2(out_fds[1], STDOUT_FILENO);
                    222:                        close(out_fds[1]);
                    223:                }
                    224:                
                    225:                if(pipe_in) {
                    226:                        close(in_fds[1]);
                    227:                        dup2(in_fds[0], STDIN_FILENO);
                    228:                        close(in_fds[0]);
                    229:                }
                    230:                
                    231:                if(pipe_err) {
                    232:                        close(err_fds[0]);
                    233:                        dup2(err_fds[1], STDERR_FILENO);
                    234:                        close(err_fds[1]);
                    235:                }
                    236:                
1.42      paf       237:                /* grabbed this from Apache source: */
                    238:                /*  HP-UX SIGCHLD fix goes here, if someone will remind me what it is... */
1.1       paf       239:                signal(SIGCHLD, SIG_DFL);       /* Was that it? */
                    240:        
1.31      paf       241:                // chdir to script's directory
                    242:                char dir[MAX_STRING];
                    243:                strncpy(dir, file_spec_cstr, MAX_STRING-1); dir[MAX_STRING-1]=0;
                    244:                rsplit(dir,'/'); // trim filename
                    245:                chdir(dir);
                    246: 
                    247:                //  execute
                    248:                execve(file_spec_cstr, argv, env);
1.1       paf       249:                exit(-errno);
                    250:        }
                    251:        
                    252:        /* Parent process */
                    253:        
                    254:        if(pipe_out) {
                    255:                close(out_fds[1]);
                    256:                *pipe_out=out_fds[0];
                    257:        }
                    258:        
                    259:        if(pipe_in) {
                    260:                close(in_fds[0]);
                    261:                *pipe_in=in_fds[1];
                    262:        }
                    263:        
                    264:        if(pipe_err) {
                    265:                close(err_fds[1]);
                    266:                *pipe_err=err_fds[0];
                    267:        }
                    268: 
                    269:        return pid;
                    270: }
                    271: 
                    272: static int get_exit_status(int pid) {
                    273:        int status;
                    274:        if(!waitpid(pid, &status, 0))
                    275:                return -1;
                    276:        return WIFEXITED(status) ? 
                    277:                WEXITSTATUS(status) : -2;
                    278: }
                    279: 
1.48.2.8  paf       280: static void read_pipe(Pool& pool, StringPtr result, int file, const char* file_spec, String::Untaint_lang lang){
1.1       paf       281:        while(true) {
1.48.2.8  paf       282:                char *buf=new(pool) char[MAX_STRING];
1.40      paf       283:                ssize_t size=read(file, buf, MAX_STRING);
                    284:                if(size<=0)
1.1       paf       285:                        break;
1.48.2.8  paf       286:                result->APPEND(buf, size, lang, file_spec, 0);
1.1       paf       287:     }
                    288: }
                    289: 
                    290: #endif
                    291: 
1.48.2.8  paf       292: #ifndef DOXYGEN
                    293: struct Append_env_pair_info {
1.1       paf       294: #ifdef WIN32
1.48.2.9  paf       295:        String* string;
1.1       paf       296: #else
1.48.2.8  paf       297:        Pool *pool;
                    298:        char **env_ref;
1.48.2.2  paf       299: #endif
1.48.2.8  paf       300: };
                    301: #endif
                    302: ///@test maybe here and at argv construction --- cstr(String::UL_UNSPECIFIED
                    303: static void append_env_pair(HashStringString::key_type key, HashStringString::value_type value,
                    304:                Append_env_pair_info *info) {
1.48.2.2  paf       305: #ifdef WIN32
1.48.2.8  paf       306:        *info->string << key << "=" << value;
                    307:        info->string->APPEND_AS_IS("", 1, 0, 0); // zero byte
1.48.2.2  paf       308: #else
                    309:        String string;
1.48.2.1  paf       310:        string << key << "=" << value;
1.1       paf       311: 
1.48.2.8  paf       312:        *(info->env_ref++)=string.cstr(*info->pool, String::UL_UNSPECIFIED);
1.1       paf       313: #endif
                    314: }
1.31      paf       315: 
1.48.2.2  paf       316: /// @todonow unix part to smart_ptr
                    317: int pa_exec(Pool& pool,
                    318:                        bool forced_allow, 
1.48.2.4  paf       319:                        StringPtr file_spec, 
1.48.2.2  paf       320:                        HashStringString& env, 
1.48.2.11! paf       321:                        ArrayString& argv, 
1.48.2.4  paf       322:                        StringPtr in, StringPtr out, StringPtr err) {
1.28      paf       323: #ifdef NO_PA_EXECS
1.32      paf       324:        if(!forced_allow)
1.34      paf       325:                throw Exception("parser.runtime",
1.48.2.2  paf       326:                        file_spec,
1.32      paf       327:                        "parser execs are disabled [recompile parser without --disable-execs configure option]");
                    328: #endif
1.28      paf       329: 
1.48.2.8  paf       330:        char* file_spec_cstr=file_spec->cstr(pool, String::UL_FILE_SPEC); 
1.1       paf       331: #ifdef WIN32
                    332: 
                    333:        PROCESS_INFORMATION pi; 
                    334:        HANDLE hInWrite, hOutRead, hErrRead;
1.48.2.9  paf       335:        CharPtr cmd=buildCommand(file_spec->cstr(String::UL_FILE_SPEC), argv);
1.48.2.2  paf       336:        CharPtr env_cstr(0);
                    337:        { // env
                    338:                String string;
1.48.2.9  paf       339:                Append_env_pair_info info={&string};
                    340:                env.for_each(append_env_pair, &info);
                    341:                env_cstr=info.string->cstr(String::UL_UNSPECIFIED);
1.1       paf       342:        }
1.36      paf       343:        if(DWORD error=CreateHiddenConsoleProcess(cmd, env_cstr, &pi, &hInWrite, &hOutRead, &hErrRead)) {
                    344:                char szErrorDesc[MAX_STRING];
1.48.2.6  paf       345:                const char *param="the file you tried to run";
1.36      paf       346:                size_t error_size=FormatMessage(
                    347:                        FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_ARGUMENT_ARRAY , NULL, error,
                    348:                        MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),
1.48.2.7  paf       349:                        szErrorDesc, sizeof(szErrorDesc), (va_list *)&param);
1.36      paf       350:                if(error_size>3) // ".\r\n"
                    351:                        szErrorDesc[error_size-3]=0;
                    352:             
                    353:                throw Exception(0,
1.48.2.2  paf       354:                        file_spec,
1.36      paf       355:                        "exec failed - %s (%ld). Consider adding shbang line (#!x:\\interpreter\\command line)", 
                    356:                                error_size?szErrorDesc:"<unknown>", (long)error);
                    357:        } else {
1.48.2.2  paf       358:                CharPtr in_cstr=in->cstr();
1.5       paf       359:                DWORD written_size;
1.48.2.2  paf       360:                WriteFile(hInWrite, in_cstr, in->size(), &written_size, NULL);
1.5       paf       361:                // EOF for stupid text reads
                    362:                // normally they should read CONTENT_LENGTH bytes,
                    363:                // without this char
                    364:                WriteFile(hInWrite, "\x1A", 1, &written_size, NULL);
1.9       parser    365:                CloseHandle(hInWrite);
1.48.2.2  paf       366:                read_pipe(pool, out, hOutRead, file_spec_cstr, String::UL_AS_IS);
1.9       parser    367:                CloseHandle(hOutRead);
1.48.2.2  paf       368:                read_pipe(pool, err, hErrRead, file_spec_cstr, String::UL_TAINTED);             
1.9       parser    369:                CloseHandle(hErrRead);
1.1       paf       370: /*     
                    371: from http://www.apache.org/websrc/cvsweb.cgi/apache-1.3/src/main/util_script.c?rev=1.151&content-type=text/vnd.viewcvs-markup
                    372: 
                    373:        * We must close the handles to the new process and its main thread
                    374:     * to prevent handle and memory leaks.
                    375: */     
                    376:                CloseHandle(pi.hProcess);
                    377:         CloseHandle(pi.hThread);
                    378:        }
                    379: 
                    380: #else
                    381: 
                    382:        int pipe_write, pipe_read, pipe_err;
1.31      paf       383: 
1.47      paf       384: #ifdef PA_SAFE_MODE
1.32      paf       385:        if(!forced_allow) {
                    386:                struct stat finfo;
                    387:                if(stat(file_spec_cstr, &finfo)!=0)
1.34      paf       388:                        throw Exception("file.missing", 
1.48.2.8  paf       389:                                        file_spec, 
1.32      paf       390:                                        "stat failed: %s (%d), actual filename '%s'", 
                    391:                                                strerror(errno), errno, file_spec_cstr);
1.31      paf       392: 
1.47      paf       393:                if(finfo.st_uid/*foreign?*/!=geteuid()
                    394:                        && finfo.st_gid/*foreign?*/!=getegid())
                    395:                        throw Exception("parser.runtime", 
1.48.2.8  paf       396:                                file_spec, 
1.47      paf       397:                                "parser is in safe mode: executing files of foreign group and user disabled [recompile parser with --disable-safe-mode configure option], actual filename '%s'", 
1.32      paf       398:                                        file_spec_cstr);
                    399:        }
1.31      paf       400: #endif
                    401: 
1.48.2.8  paf       402:        char * argv_cstrs[1+10+1]={file_spec_cstr, 0};
                    403:        const int argv_size=argv.count();
                    404:        const int argv_max=sizeof(argv_cstrs)/sizeof(argv_cstrs[0])-1-1;
                    405:        if(argv_size>argv_max)
                    406:                throw Exception("parser.runtime",
                    407:                        file_spec,
                    408:                        "too many arguments (%d > max %d)", argv_size, argv_max);
                    409:        for(int i=0; i<argv_size; i++)
                    410:                argv_cstrs[1+i]=argv[i]->cstr(pool);
                    411:        argv_cstrs[1+argv_size]=0;
                    412: 
                    413:        char **env_cstrs=new(pool) char *[env.count()+1/*0*/];
                    414:        {
                    415:                Append_env_pair_info info={&pool, env_cstrs}; 
                    416:                env.for_each(append_env_pair, &info);
                    417:                *info.env_ref=0;
1.1       paf       418:        }
1.31      paf       419: 
1.46      paf       420:        pid_t pid=execve_piped(
1.1       paf       421:                file_spec_cstr,
1.27      paf       422:                argv_cstrs, env_cstrs,
1.22      paf       423:                &pipe_write, &pipe_read, &pipe_err);
1.46      paf       424:        if(pid>0) {
1.21      paf       425:                // in child
1.48.2.8  paf       426:                if(in && in->size()) {// there is some in data
                    427:                        CharPtr in_cstr=in->cstr();
                    428:                        write(pipe_write, in_cstr, in->size());
                    429:                }
1.1       paf       430:                close(pipe_write);
1.48.2.8  paf       431:                read_pipe(pool, out, pipe_read, file_spec_cstr, String::UL_AS_IS);
1.5       paf       432:                close(pipe_read);
1.48.2.8  paf       433:                read_pipe(pool, err, pipe_err, file_spec_cstr, String::UL_TAINTED);
1.1       paf       434:                close(pipe_err);
                    435: 
                    436:                return get_exit_status(pid); // negative may mean "-errno[execl()]"
                    437:        } else 
1.34      paf       438:                throw Exception(0,
1.48.2.8  paf       439:                        file_spec,
1.46      paf       440:                        "%s error: %s (%d)", pid<0?"fork":"pipe", strerror(errno), errno); 
1.1       paf       441: #endif
                    442: 
                    443:        return 0;
                    444: }

E-mail: