install.c

Go to the documentation of this file.

#include <fcntl.h> 
#include <sys/stat.h>
#include "buffer.h"
#include "stralloc.h"
#include "getln.h"
#include "readwrite.h"
#include "exit.h"
#include "open.h"
#include "error.h"
#include "byte.h"
#include "scan.h"
#include "logmsg.h"
 
#define WHO "install"
 
stralloc target = {0};
char *to;
 
void nomem() { logmsg(WHO,111,FATAL,"out of memory"); }
 
char inbuf[BUFFER_INSIZE];
char outbuf[BUFFER_OUTSIZE];
buffer bi;
buffer bo;
 
void doit(stralloc *line)
{
  char *x;
  unsigned int xlen;
  unsigned int i;
  char *type;
  char *uidstr;
  char *gidstr;
  char *modestr;
  char *mid;
  char *name;
  unsigned long uid;
  unsigned long gid;
  unsigned long mode;
  int fdin;
  int fdout;
 
  x = line->s; xlen = line->len;
 
  type = x;
  i = byte_chr(x,xlen,':'); if (i == xlen) return;
  x[i++] = 0; x += i; xlen -= i;
 
  uidstr = x;
  i = byte_chr(x,xlen,':'); if (i == xlen) return;
  x[i++] = 0; x += i; xlen -= i;
 
  gidstr = x;
  i = byte_chr(x,xlen,':'); if (i == xlen) return;
  x[i++] = 0; x += i; xlen -= i;
 
  modestr = x;
  i = byte_chr(x,xlen,':'); if (i == xlen) return;
  x[i++] = 0; x += i; xlen -= i;
 
  mid = x;
  i = byte_chr(x,xlen,':'); if (i == xlen) return;
  x[i++] = 0; x += i; xlen -= i;
 
  name = x;
  i = byte_chr(x,xlen,':'); if (i == xlen) return;
  x[i++] = 0; x += i; xlen -= i;
 
  if (!stralloc_copys(&target,to)) nomem();
  if (!stralloc_cats(&target,mid)) nomem();
  if (!stralloc_cats(&target,name)) nomem();
  if (!stralloc_0(&target)) nomem();
 
  uid = -1; if (*uidstr) scan_ulong(uidstr,&uid);
  gid = -1; if (*gidstr) scan_ulong(gidstr,&gid);
  scan_8long(modestr,&mode);
 
  switch (*type) {
    case 'd': if (mkdir(target.s,0700) == -1)
                if (errno != EEXIST)
                  logmsg(WHO,111,FATAL,B("unable to mkdir :",target.s));
              break;
    case 'c': fdin = open_read(name);
              if (fdin == -1)
                 logmsg(WHO,111,FATAL,B("unable to read :",name));
               buffer_init(&bi,buffer_unixread,fdin,inbuf,sizeof(inbuf));
               fdout = open_trunc(target.s);
               if (fdout == -1)
                 logmsg(WHO,111,FATAL,B("unable to write :",target.s));
               buffer_init(&bo,buffer_unixwrite,fdout,outbuf,sizeof(outbuf));
 
               switch (buffer_copy(&bo,&bi)) {
                 case -2: logmsg(WHO,111,FATAL,B("unable to read :",name));
                 case -3: logmsg(WHO,111,FATAL,B("unable to write :",target.s));
               }
 
               close(fdin);
               if (buffer_flush(&bo) == -1)
                 logmsg(WHO,111,FATAL,B("unable to write :",target.s));
                if (fsync(fdout) == -1)
                  logmsg(WHO,111,FATAL,B("unable to write :",target.s));
                close(fdout);
                break;
    default:    return;
  }
 
  if (chown(target.s,uid,gid) == -1)
    logmsg(WHO,111,FATAL,B("unable to chown :",target.s));
  if (chmod(target.s,mode) == -1)
    logmsg(WHO,111,FATAL,B("unable to chmod :",target.s));
}
 
char buf[256];
buffer in = BUFFER_INIT(read,0,buf,sizeof(buf));
stralloc line = {0};
 
int main(int argc,char **argv)
{
  int match;
 
  umask(077);
 
  to = argv[1];
  if (!to) logmsg(WHO,100,USAGE,"install dir");
 
  for (;;) {
    if (getln(&in,&line,&match,'\n') == -1)
      logmsg(WHO,111,FATAL,"unable to read input :");
    doit(&line);
    if (!match)
      _exit(0);
  }
 
  return 0;
}