Subversion Repositories wimsdev

/*    Copyright (C) 1998-2003 XIAO, Gang of Universite de Nice - Sophia Antipolis

 *

 *  This program is free software; you can redistribute it and/or modify

 *  it under the terms of the GNU General Public License as published by

 *  the Free Software Foundation; either version 2 of the License, or

 *  (at your option) any later version.

 *

 *  This program is distributed in the hope that it will be useful,

 *  but WITHOUT ANY WARRANTY; without even the implied warranty of

 *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 *  GNU General Public License for more details.

 *

 *  You should have received a copy of the GNU General Public License

 *  along with this program; if not, write to the Free Software

 *  Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.

 */

/* line input / output / translation routines

 * and error routines

 */

#include "wimslogd.h"

int exec_wait;

int execredirected(char *cmdf, char *inf, char *outf, char *errf, char *arg[])

{

  pid_t pid;

  int status, t;

  fflush(NULL);      /* flush all output streams before forking

                      * otherwise they will be doubled */

  pid=fork(); if(pid==-1) return -1;

  if(!pid) {      /* child */

    char buf[4096]; int k;

    if(inf!=NULL) (void)freopen(inf,"r",stdin);

    if(outf!=NULL) (void)freopen(outf,"w",stdout);

    if(errf!=NULL) (void)freopen(errf,"w",stderr);

      /* This is to patch LinuxPPC uid wrapping

       * for scripts */

    t=0; if(strchr(cmdf,'/')) {

      FILE *tf;

      char buf[16];

      tf=fopen(cmdf,"r"); (void)fread(buf,1,10,tf); fclose(tf);

      if(memcmp(buf+1,"ELF",3)!=0) t=1;

    }

    errno=0;

    if(strchr(cmdf,'/')) execve(cmdf,arg,environ);

    else execvp(cmdf,arg);

    snprintf(buf,sizeof(buf),"Failed to execute");

    for(k=0;arg[k];k++) {

      t=strlen(buf);

      snprintf(buf+t,sizeof(buf)-t," %s",arg[k]);

    }

    t=strlen(buf);

    snprintf(buf+t,sizeof(buf)-t,"\n    %s\n",strerror(errno));

    wlogdaccessfile(buf,"a","%s/exec.fail",tmpd);

    exit(127);

  }

  else {      /* parent */

    status=0;

    if(exec_wait) {

      waitpid(pid,&status,0);

      return WEXITSTATUS(status);

    }

    else {

      exec_wait=1; addfork(pid,0); return 0;

    }

  }

}

/* my system(), but with variable parms

 * More secure than system(), and direct fork.

 */

int call_ssh(int wait,char *s,...)

{

  va_list vp;

  char buf[MAX_LINELEN+1];

  char *arg[1024];

  char *inf=NULL, *outf=NULL, *errf=NULL;

  char *cmdf, *p, *p2;

  int i, d;

  va_start(vp,s);

  vsnprintf(buf,sizeof(buf),s,vp);

  va_end(vp);

  p=find_word_start(buf); if(*p==0) return 0;

  cmdf=p;

  for(i=0;*p!=0 && i<1000; p=find_word_start(p2)) {

    switch(*p) {

      case '\'': {

        p++; p2=strchr(p,'\''); if(p2==NULL) p2=p+strlen(p);

        d=0; break;

      }

      case '"': {

        p++; p2=strchr(p,'"'); if(p2==NULL) p2=p+strlen(p);

        d=0; break;

      }

      default: d=1; p2=find_word_end(p); break;

    }

    if(*p2) *p2++=0;

    if(!d) {arg[i++]=p; continue;}

    switch(*p) {

      case '<': inf=++p; break;

      case '>': {

        p++;

        if(*p=='&') {

          merge: p++; errf=outf=p; break;

        }

        else outf=p;

        break;

      }

      case '&': {

        p++; if(*p=='>') goto merge;

        else break;

      }

      case '2': {

        if(*(p+1)=='>') {errf=p+2; break;}

      }

      default: arg[i++]=p; break;

    }

  }

  arg[i]=NULL;

  exec_wait=wait;

  return execredirected(cmdf,inf,outf,errf,arg);

}

/* Read/write to a file with variable parms to print filename */

void wlogdaccessfile(char *content, char *type, char *s,...)

{

  va_list vp;

  char buf[MAX_LINELEN+1];

  FILE *f;

  int l;

  va_start(vp,s);

  vsnprintf(buf,sizeof(buf),s,vp);

  va_end(vp);

  f=fopen(buf,type);

  if(f==NULL) {

    if(*type=='r') content[0]=0;

    return;

  }

  switch(*type) {

    case 'a':

    case 'w': {

      l=strlen(content); fwrite(content,1,l,f); break;

    }

    case 'r': {

      l=fread(content,1,MAX_LINELEN-1,f);

      if(l>0 && l<MAX_LINELEN) content[l]=0;

      else content[0]=0;

      _tolinux(content);

      break;

    }

    default: {

      content[0]=0; break;

    }

  }

  fclose(f);

}

/* system(), but with variable parms

 * Uses sh to execute command.

 */

int call_sh(int wait,char *s,...)

{

  va_list vp;

  char buf[MAX_LINELEN+1];

  char *abuf[8];

  va_start(vp,s);

  vsnprintf(buf,sizeof(buf),s,vp);

  va_end(vp);

  abuf[0]="sh"; abuf[1]="-c"; abuf[2]=buf; abuf[3]=NULL;

  exec_wait=wait;

  return execredirected(abuf[0],NULL,NULL,NULL,abuf);

}

void wimslogd_error(char *msg)

{

  fprintf(stderr,"%s %s\n",nowstr, msg);

}

void debug(char *p,...)

{

  char lbuf[MAX_LINELEN+1];

  char *pp;

  va_list vp;

  snprintf(lbuf,sizeof(lbuf),"%s: ",nowstr);

  pp=lbuf+strlen(lbuf);

  va_start(vp,p);

  vsnprintf(pp,sizeof(lbuf)-(pp-lbuf),p,vp);

  va_end(vp);

  pp=strchr(lbuf,'\n'); if(pp) *pp=0;

  strip_trailing_spaces(lbuf); strcat(lbuf,"\n");

  wlogdaccessfile(lbuf,"a",debugfile);

}