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

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

E-mail: