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

1.1       paf         1: /** @file
                      2:        Parser: program executing for different OS-es.
                      3: 
1.48      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.2! paf        10: static const char* IDENT_EXEC_C="$Date: 2003/01/27 17:17:06 $";
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.2! 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.2! paf       125: static CharPtr buildCommand(CharPtr file_spec_cstr, const Array<ConstStringPtr>& argv) {
        !           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.4       paf       134:                                const char *begin=buf+2;
                    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.46      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.35      paf       280: static void read_pipe(String& result, int file, const char *file_spec, String::Untaint_lang lang){
1.1       paf       281:        while(true) {
1.2       paf       282:                char *buf=(char *)result.pool().malloc(MAX_STRING);
1.40      paf       283:                ssize_t size=read(file, buf, MAX_STRING);
                    284:                if(size<=0)
1.1       paf       285:                        break;
1.35      paf       286:                result.APPEND(buf, size, lang, file_spec, 0);
1.1       paf       287:     }
                    288: }
                    289: 
                    290: #endif
                    291: 
1.30      paf       292: ///@test maybe here and at argv construction --- cstr(String::UL_UNSPECIFIED
1.48.2.2! paf       293: static void append_env_pair(HashStringString::key_type key, HashStringString::value_type value, 
1.1       paf       294: #ifdef WIN32
1.48.2.2! paf       295:                                                                String *string
1.1       paf       296: #else
1.48.2.2! paf       297:                                                                struct {
        !           298:                                                                        Pool *pool;
        !           299:                                                                        char ***env
        !           300:                                                                } info
        !           301: #endif
        !           302:                                                        ) {
        !           303: #ifdef WIN32
        !           304:        *string << key << "=" << value;
        !           305:        string->APPEND_AS_IS("", 1, 0, 0); // zero byte
        !           306: #else
        !           307:        String string;
1.48.2.1  paf       308:        string << key << "=" << value;
1.1       paf       309: 
1.48.2.2! paf       310:        **env=string.cstr(*info->pool, String::UL_UNSPECIFIED);  (*env)++;
1.1       paf       311: #endif
                    312: }
1.31      paf       313: 
1.48.2.2! paf       314: /// @todonow unix part to smart_ptr
        !           315: int pa_exec(Pool& pool,
        !           316:                        bool forced_allow, 
        !           317:                        ConstStringPtr file_spec, 
        !           318:                        HashStringString& env, 
        !           319:                        Array<ConstStringPtr>& argv, 
        !           320:                        ConstStringPtr in, StringPtr out, StringPtr err) {
1.28      paf       321: #ifdef NO_PA_EXECS
1.32      paf       322:        if(!forced_allow)
1.34      paf       323:                throw Exception("parser.runtime",
1.48.2.2! paf       324:                        file_spec,
1.32      paf       325:                        "parser execs are disabled [recompile parser without --disable-execs configure option]");
                    326: #endif
1.28      paf       327: 
1.1       paf       328: #ifdef WIN32
                    329: 
                    330:        PROCESS_INFORMATION pi; 
                    331:        HANDLE hInWrite, hOutRead, hErrRead;
1.48.2.2! paf       332:        CharPtr file_spec_cstr=file_spec->cstr(String::UL_FILE_SPEC); 
        !           333:        CharPtr cmd=buildCommand(file_spec_cstr, argv);
        !           334:        CharPtr env_cstr(0);
        !           335:        { // env
        !           336:                String string;
        !           337:                env.for_each(append_env_pair, &string);
1.45      paf       338:                env_cstr=string.cstr(String::UL_UNSPECIFIED);
1.1       paf       339:        }
1.36      paf       340:        if(DWORD error=CreateHiddenConsoleProcess(cmd, env_cstr, &pi, &hInWrite, &hOutRead, &hErrRead)) {
                    341:                char szErrorDesc[MAX_STRING];
                    342:                char *param="the file you tried to run";
                    343:                size_t error_size=FormatMessage(
                    344:                        FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_ARGUMENT_ARRAY , NULL, error,
                    345:                        MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),
                    346:                        szErrorDesc, sizeof(szErrorDesc), &param);
                    347:                if(error_size>3) // ".\r\n"
                    348:                        szErrorDesc[error_size-3]=0;
                    349:             
                    350:                throw Exception(0,
1.48.2.2! paf       351:                        file_spec,
1.36      paf       352:                        "exec failed - %s (%ld). Consider adding shbang line (#!x:\\interpreter\\command line)", 
                    353:                                error_size?szErrorDesc:"<unknown>", (long)error);
                    354:        } else {
1.48.2.2! paf       355:                CharPtr in_cstr=in->cstr();
1.5       paf       356:                DWORD written_size;
1.48.2.2! paf       357:                WriteFile(hInWrite, in_cstr, in->size(), &written_size, NULL);
1.5       paf       358:                // EOF for stupid text reads
                    359:                // normally they should read CONTENT_LENGTH bytes,
                    360:                // without this char
                    361:                WriteFile(hInWrite, "\x1A", 1, &written_size, NULL);
1.9       parser    362:                CloseHandle(hInWrite);
1.48.2.2! paf       363:                read_pipe(pool, out, hOutRead, file_spec_cstr, String::UL_AS_IS);
1.9       parser    364:                CloseHandle(hOutRead);
1.48.2.2! paf       365:                read_pipe(pool, err, hErrRead, file_spec_cstr, String::UL_TAINTED);             
1.9       parser    366:                CloseHandle(hErrRead);
1.1       paf       367: /*     
                    368: 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
                    369: 
                    370:        * We must close the handles to the new process and its main thread
                    371:     * to prevent handle and memory leaks.
                    372: */     
                    373:                CloseHandle(pi.hProcess);
                    374:         CloseHandle(pi.hThread);
                    375:        }
                    376: 
                    377: #else
                    378: 
                    379:        int pipe_write, pipe_read, pipe_err;
1.27      paf       380:        char *file_spec_cstr=file_spec.cstr(String::UL_FILE_SPEC);
1.31      paf       381: 
1.47      paf       382: #ifdef PA_SAFE_MODE
1.32      paf       383:        if(!forced_allow) {
                    384:                struct stat finfo;
                    385:                if(stat(file_spec_cstr, &finfo)!=0)
1.34      paf       386:                        throw Exception("file.missing", 
1.32      paf       387:                                        &file_spec, 
                    388:                                        "stat failed: %s (%d), actual filename '%s'", 
                    389:                                                strerror(errno), errno, file_spec_cstr);
1.31      paf       390: 
1.47      paf       391:                if(finfo.st_uid/*foreign?*/!=geteuid()
                    392:                        && finfo.st_gid/*foreign?*/!=getegid())
                    393:                        throw Exception("parser.runtime", 
                    394:                                &file_spec, 
                    395:                                "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       396:                                        file_spec_cstr);
                    397:        }
1.31      paf       398: #endif
                    399: 
1.27      paf       400:        char *argv_cstrs[1+10+1]={file_spec_cstr, 0};
1.1       paf       401:        if(argv) {
1.20      paf       402:                const int argv_size=argv->size();
1.27      paf       403:                const int argv_max=sizeof(argv_cstrs)/sizeof(argv_cstrs[0])-1-1;
1.20      paf       404:                if(argv_size>argv_max)
1.34      paf       405:                        throw Exception("parser.runtime",
1.20      paf       406:                                &file_spec,
                    407:                                "too many arguments (%d > max %d)", argv_size, argv_max);
                    408:                for(int i=0; i<argv_size; i++)
1.27      paf       409:                        argv_cstrs[1+i]=argv->get_string(i)->cstr();
                    410:                argv_cstrs[1+argv_size]=0;
1.1       paf       411:        }
1.2       paf       412:        char **env_cstrs=0;
1.1       paf       413:        if(env) {
1.2       paf       414:                env_cstrs=
1.1       paf       415:                        (char **)env->pool().malloc(sizeof(char *)*(env->size()+1/*0*/));
1.2       paf       416:                char **env_ref=env_cstrs;
                    417:                env->for_each(append_env_pair, &env_ref);
                    418:                *env_ref=0;
1.1       paf       419:        }
1.31      paf       420: 
1.46      paf       421:        pid_t pid=execve_piped(
1.1       paf       422:                file_spec_cstr,
1.27      paf       423:                argv_cstrs, env_cstrs,
1.22      paf       424:                &pipe_write, &pipe_read, &pipe_err);
1.46      paf       425:        if(pid>0) {
1.21      paf       426:                // in child
1.16      paf       427:                const char *in_cstr=in.cstr();
1.43      paf       428:                if(*in_cstr) // there is some in data
1.41      paf       429:                        write(pipe_write, in_cstr, in.size());
1.1       paf       430:                close(pipe_write);
1.35      paf       431:                read_pipe(out, pipe_read, file_spec_cstr, String::UL_AS_IS);
1.5       paf       432:                close(pipe_read);
1.35      paf       433:                read_pipe(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.2       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: