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

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.18.2.7! (paf       10:: static const char* IDENT_EXEC_C="$Date: 2003/03/21 14:49:00 $";
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.18.2.7! (paf      114:: static void read_pipe(String& result, HANDLE hOutRead, String::Language lang){
1.1       paf       115:        while(true) {
1.48.2.18.2.6  (paf      116::               char *buf=new(PointerFreeGC) char[MAX_STRING];
1.1       paf       117:                unsigned long size;
1.30      paf       118:                if(!ReadFile(hOutRead, buf, MAX_STRING, &size, NULL) || !size) 
1.1       paf       119:                        break;
1.48.2.18.2.6  (paf      120::               result.append(buf, size, lang);
                    121::       }
1.1       paf       122: }
                    123: 
1.48.2.18.2.1  (paf      124:: static const char* buildCommand(const char* file_spec_cstr, const ArrayString& argv) {
                    125::       const char* result=file_spec_cstr;
1.4       paf       126:        if(FILE *f=fopen(file_spec_cstr, "r")) {
1.48.2.2  paf       127:                try {
1.1       paf       128:                char buf[MAX_STRING];
                    129:                size_t size=fread(buf, 1, MAX_STRING-1, f);
                    130:                if(size>2) {
                    131:                        buf[size]=0;
                    132:                        if(strncmp(buf, "#!", 2)==0) {
1.48.2.5  paf       133:                                const char* begin=buf+2;
1.4       paf       134:                                if(*begin==' ') // alx: were an old magic for some linux-es
                    135:                                        begin++;
                    136:                                if(char *end=strchr(begin, '\n')) {
1.48.2.2  paf       137:                                        String string(begin, end-begin);
1.1       paf       138:                                        string << " " << file_spec_cstr;
1.8       parser    139:                                        result=string.cstr();
1.1       paf       140:                                }
                    141:                        }
                    142:                }
1.48.2.2  paf       143:                } catch(...) {
                    144:                        fclose(f);
                    145:                        rethrow;
                    146:                }
1.1       paf       147:                fclose(f);
                    148:        }
1.48.2.2  paf       149:        { // appending argv
                    150:                String string(result);
                    151:                for(int i=0; i<argv.count(); i++) {
                    152:                        string << " ";
                    153:                        string << *argv[i];
1.8       parser    154:                }
                    155: 
1.48.2.2  paf       156:                result=string.cstr();
1.8       parser    157:        }
                    158: 
                    159:        return result;
1.1       paf       160: }
                    161: 
                    162: #else
                    163: 
1.48.2.5  paf       164: static pid_t execve_piped(const char* file_spec_cstr, 
1.27      paf       165:                        char * const argv[], char * const env[],
                    166:                        int *pipe_in, int *pipe_out, int *pipe_err) {
1.46      paf       167:        pid_t pid;
1.1       paf       168:        int in_fds[2];
                    169:        int out_fds[2];
                    170:        int err_fds[2];
                    171:        int save_errno;
                    172:        
                    173:        if(pipe_in && pipe(in_fds)<0) {
                    174:                save_errno=errno;
                    175:                errno=save_errno;
                    176:                return 0;
                    177:        }
                    178:        
                    179:        if(pipe_out && pipe(out_fds)<0) {
                    180:                save_errno=errno;
                    181:                if(pipe_in) {
                    182:                        close(in_fds[0]); close(in_fds[1]);
                    183:                }
                    184:                errno=save_errno;
                    185:                return 0;
                    186:        }
                    187:        
                    188:        if(pipe_err && pipe(err_fds)<0) {
                    189:                save_errno=errno;
                    190:                if(pipe_in) {
                    191:                        close(in_fds[0]); close(in_fds[1]);
                    192:                }
                    193:                if(pipe_out) {
                    194:                        close(out_fds[0]); close(out_fds[1]);
                    195:                }
                    196:                errno=save_errno;
                    197:                return 0;
                    198:        }
                    199:        
                    200:        if((pid=fork())<0) {
                    201:                save_errno=errno;
                    202:                if(pipe_in) {
                    203:                        close(in_fds[0]); close(in_fds[1]);
                    204:                }
                    205:                if(pipe_out) {
                    206:                        close(out_fds[0]); close(out_fds[1]);
                    207:                }
                    208:                if(pipe_err) {
                    209:                        close(err_fds[0]); close(err_fds[1]);
                    210:                }
                    211:                errno=save_errno;
1.46      paf       212:                return -1;
1.1       paf       213:        }
                    214:        
                    215:        if(!pid) {
                    216:                /* Child process */
                    217:                
                    218:                if(pipe_out) {
                    219:                        close(out_fds[0]);
                    220:                        dup2(out_fds[1], STDOUT_FILENO);
                    221:                        close(out_fds[1]);
                    222:                }
                    223:                
                    224:                if(pipe_in) {
                    225:                        close(in_fds[1]);
                    226:                        dup2(in_fds[0], STDIN_FILENO);
                    227:                        close(in_fds[0]);
                    228:                }
                    229:                
                    230:                if(pipe_err) {
                    231:                        close(err_fds[0]);
                    232:                        dup2(err_fds[1], STDERR_FILENO);
                    233:                        close(err_fds[1]);
                    234:                }
                    235:                
1.42      paf       236:                /* grabbed this from Apache source: */
                    237:                /*  HP-UX SIGCHLD fix goes here, if someone will remind me what it is... */
1.1       paf       238:                signal(SIGCHLD, SIG_DFL);       /* Was that it? */
                    239:        
1.31      paf       240:                // chdir to script's directory
                    241:                char dir[MAX_STRING];
                    242:                strncpy(dir, file_spec_cstr, MAX_STRING-1); dir[MAX_STRING-1]=0;
                    243:                rsplit(dir,'/'); // trim filename
                    244:                chdir(dir);
                    245: 
                    246:                //  execute
                    247:                execve(file_spec_cstr, argv, env);
1.1       paf       248:                exit(-errno);
                    249:        }
                    250:        
                    251:        /* Parent process */
                    252:        
                    253:        if(pipe_out) {
                    254:                close(out_fds[1]);
                    255:                *pipe_out=out_fds[0];
                    256:        }
                    257:        
                    258:        if(pipe_in) {
                    259:                close(in_fds[0]);
                    260:                *pipe_in=in_fds[1];
                    261:        }
                    262:        
                    263:        if(pipe_err) {
                    264:                close(err_fds[1]);
                    265:                *pipe_err=err_fds[0];
                    266:        }
                    267: 
                    268:        return pid;
                    269: }
                    270: 
                    271: static int get_exit_status(int pid) {
                    272:        int status;
                    273:        if(!waitpid(pid, &status, 0))
                    274:                return -1;
                    275:        return WIFEXITED(status) ? 
                    276:                WEXITSTATUS(status) : -2;
                    277: }
                    278: 
1.48.2.18.2.7! (paf      279:: static void read_pipe(String& result, int file, String::Language lang){
1.1       paf       280:        while(true) {
1.48.2.18.2.7! (paf      281::               char *buf=new(PointerFreeGC) char[MAX_STRING];
1.40      paf       282:                ssize_t size=read(file, buf, MAX_STRING);
                    283:                if(size<=0)
1.1       paf       284:                        break;
1.48.2.18.2.7! (paf      285::               result.append(buf, size, lang);
        !           286::       }
1.1       paf       287: }
                    288: 
                    289: #endif
                    290: 
1.48.2.8  paf       291: #ifndef DOXYGEN
                    292: struct Append_env_pair_info {
1.1       paf       293: #ifdef WIN32
1.48.2.18.2.6  (paf      294::       String& string;
                    295::       Append_env_pair_info(String& astring): string(astring) {}
1.1       paf       296: #else
1.48.2.8  paf       297:        char **env_ref;
1.48.2.2  paf       298: #endif
1.48.2.8  paf       299: };
                    300: #endif
1.48.2.18.2.5  (paf      301:: ///@test maybe here and at argv construction --- cstr(String::L_UNSPECIFIED
1.48.2.8  paf       302: static void append_env_pair(HashStringString::key_type key, HashStringString::value_type value,
                    303:                Append_env_pair_info *info) {
1.48.2.2  paf       304: #ifdef WIN32
1.48.2.18.2.7! (paf      305::       info->string << key << "=" << *value;
        !           306::       info->string.append("", 1, String::L_AS_IS); // zero byte
1.48.2.2  paf       307: #else
                    308:        String string;
1.48.2.18.2.7! (paf      309::       string << key << "=" << *value;
1.1       paf       310: 
1.48.2.18.2.5  (paf      311::       *(info->env_ref++)=string.cstr(*info->String::L_UNSPECIFIED);
1.1       paf       312: #endif
                    313: }
1.31      paf       314: 
1.48.2.2  paf       315: /// @todonow unix part to smart_ptr
1.48.2.18.2.1  (paf      316:: PA_exec_result pa_exec(
1.48.2.2  paf       317:                        bool forced_allow, 
1.48.2.18.2.1  (paf      318::                       const String& file_spec, 
1.48.2.17  paf       319:                        const HashStringString* env, 
1.48.2.13  paf       320:                        const ArrayString& argv, 
1.48.2.12  paf       321:                        String& in) {
                    322:        PA_exec_result result;
                    323: 
1.28      paf       324: #ifdef NO_PA_EXECS
1.32      paf       325:        if(!forced_allow)
1.34      paf       326:                throw Exception("parser.runtime",
1.48.2.18.2.7! (paf      327::                       &file_spec,
1.32      paf       328:                        "parser execs are disabled [recompile parser without --disable-execs configure option]");
                    329: #endif
1.28      paf       330: 
1.48.2.18.2.7! (paf      331::       const char* file_spec_cstr=file_spec.cstr(String::L_FILE_SPEC); 
1.1       paf       332: #ifdef WIN32
                    333: 
                    334:        PROCESS_INFORMATION pi; 
                    335:        HANDLE hInWrite, hOutRead, hErrRead;
1.48.2.18.2.7! (paf      336::       const char* cmd=buildCommand(file_spec.cstr(String::L_FILE_SPEC), argv);
        !           337::       char* env_cstr;
1.48.2.17  paf       338:        if(env) {
1.48.2.2  paf       339:                String string;
1.48.2.18.2.7! (paf      340::               Append_env_pair_info info(string);
1.48.2.17  paf       341:                env->for_each(append_env_pair, &info);
1.48.2.18.2.7! (paf      342::               env_cstr=info.string.cstrm(String::L_UNSPECIFIED);
1.1       paf       343:        }
1.36      paf       344:        if(DWORD error=CreateHiddenConsoleProcess(cmd, env_cstr, &pi, &hInWrite, &hOutRead, &hErrRead)) {
                    345:                char szErrorDesc[MAX_STRING];
1.48.2.18.2.3  (paf      346::               const char* param="the file you tried to run";
1.36      paf       347:                size_t error_size=FormatMessage(
                    348:                        FORMAT_MESSAGE_FROM_SYSTEM|FORMAT_MESSAGE_ARGUMENT_ARRAY , NULL, error,
                    349:                        MAKELANGID(LANG_NEUTRAL, SUBLANG_NEUTRAL),
1.48.2.7  paf       350:                        szErrorDesc, sizeof(szErrorDesc), (va_list *)&param);
1.36      paf       351:                if(error_size>3) // ".\r\n"
                    352:                        szErrorDesc[error_size-3]=0;
                    353:             
                    354:                throw Exception(0,
1.48.2.18.2.7! (paf      355::                       &file_spec,
1.36      paf       356:                        "exec failed - %s (%ld). Consider adding shbang line (#!x:\\interpreter\\command line)", 
                    357:                                error_size?szErrorDesc:"<unknown>", (long)error);
                    358:        } else {
1.48.2.18.2.1  (paf      359::               const char* in_cstr=in.cstr();
1.5       paf       360:                DWORD written_size;
1.48.2.18.2.4  (paf      361::               WriteFile(hInWrite, in_cstr, in.length(), &written_size, NULL);
1.5       paf       362:                // EOF for stupid text reads
                    363:                // normally they should read CONTENT_LENGTH bytes,
                    364:                // without this char
                    365:                WriteFile(hInWrite, "\x1A", 1, &written_size, NULL);
1.9       parser    366:                CloseHandle(hInWrite);
1.48.2.18.2.7! (paf      367::               read_pipe(result.out, hOutRead, String::L_AS_IS);
1.9       parser    368:                CloseHandle(hOutRead);
1.48.2.18.2.7! (paf      369::               read_pipe(result.err, hErrRead, String::L_TAINTED);             
1.9       parser    370:                CloseHandle(hErrRead);
1.1       paf       371: /*     
                    372: 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
                    373: 
                    374:        * We must close the handles to the new process and its main thread
                    375:     * to prevent handle and memory leaks.
                    376: */     
                    377:                CloseHandle(pi.hProcess);
                    378:         CloseHandle(pi.hThread);
                    379:        }
                    380: 
                    381: #else
                    382: 
                    383:        int pipe_write, pipe_read, pipe_err;
1.31      paf       384: 
1.47      paf       385: #ifdef PA_SAFE_MODE
1.32      paf       386:        if(!forced_allow) {
                    387:                struct stat finfo;
                    388:                if(stat(file_spec_cstr, &finfo)!=0)
1.34      paf       389:                        throw Exception("file.missing", 
1.48.2.8  paf       390:                                        file_spec, 
1.32      paf       391:                                        "stat failed: %s (%d), actual filename '%s'", 
                    392:                                                strerror(errno), errno, file_spec_cstr);
1.31      paf       393: 
1.47      paf       394:                if(finfo.st_uid/*foreign?*/!=geteuid()
                    395:                        && finfo.st_gid/*foreign?*/!=getegid())
                    396:                        throw Exception("parser.runtime", 
1.48.2.8  paf       397:                                file_spec, 
1.47      paf       398:                                "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       399:                                        file_spec_cstr);
                    400:        }
1.31      paf       401: #endif
                    402: 
1.48.2.8  paf       403:        char * argv_cstrs[1+10+1]={file_spec_cstr, 0};
                    404:        const int argv_size=argv.count();
                    405:        const int argv_max=sizeof(argv_cstrs)/sizeof(argv_cstrs[0])-1-1;
                    406:        if(argv_size>argv_max)
                    407:                throw Exception("parser.runtime",
                    408:                        file_spec,
                    409:                        "too many arguments (%d > max %d)", argv_size, argv_max);
                    410:        for(int i=0; i<argv_size; i++)
1.48.2.18.2.1  (paf      411::               argv_cstrs[1+i]=argv[i]->cstr();
1.48.2.8  paf       412:        argv_cstrs[1+argv_size]=0;
                    413: 
1.48.2.18  paf       414:        char **env_cstrs;
                    415:        if(env) {
1.48.2.18.2.1  (paf      416::               env_cstrs=new char *[env->count()+1/*0*/];
1.48.2.18.2.2  (paf      417::               Append_env_pair_info info={&env_cstrs}; 
1.48.2.18  paf       418:                env->for_each(append_env_pair, &info);
1.48.2.8  paf       419:                *info.env_ref=0;
1.48.2.18  paf       420:        } else
                    421:                env_cstrs=0;
1.31      paf       422: 
1.46      paf       423:        pid_t pid=execve_piped(
1.1       paf       424:                file_spec_cstr,
1.27      paf       425:                argv_cstrs, env_cstrs,
1.22      paf       426:                &pipe_write, &pipe_read, &pipe_err);
1.46      paf       427:        if(pid>0) {
1.21      paf       428:                // in child
1.48.2.18.2.4  (paf      429::               if(in.length()) {// there is some in data
1.48.2.18.2.1  (paf      430::                       const char* in_cstr=in.cstr();
1.48.2.18.2.4  (paf      431::                       write(pipe_write, in_cstr, in.length());
1.48.2.8  paf       432:                }
1.1       paf       433:                close(pipe_write);
1.48.2.18.2.5  (paf      434::               result.out=read_pipe(pipe_read, file_spec_cstr, String::L_AS_IS);
1.5       paf       435:                close(pipe_read);
1.48.2.18.2.5  (paf      436::               result.err=read_pipe(pipe_err, file_spec_cstr, String::L_TAINTED);
1.1       paf       437:                close(pipe_err);
                    438: 
1.48.2.12  paf       439:                result.status=get_exit_status(pid); // negative may mean "-errno[execl()]"
1.1       paf       440:        } else 
1.34      paf       441:                throw Exception(0,
1.48.2.8  paf       442:                        file_spec,
1.46      paf       443:                        "%s error: %s (%d)", pid<0?"fork":"pipe", strerror(errno), errno); 
1.1       paf       444: #endif
                    445: 
1.48.2.12  paf       446:        return result;
1.1       paf       447: }

E-mail: