sys/src/cmd/cp.c

   1 #include <u.h>
   2 #include <libc.h>
   3
   4 #define DEFB    (8*1024)
   5
   6 int     failed;
   7 int     gflag;
   8 int     uflag;
   9 int     xflag;
  10 void    copy(char *from, char *to, int todir);
  11 int     copy1(int fdf, int fdt, char *from, char *to);
  12
  13 void
  14 main(int argc, char *argv[])
  15 {
  16         Dir *dirb;
  17         int todir, i;
  18
  19         ARGBEGIN {
  20         case 'g':
  21                 gflag++;
  22                 break;
  23         case 'u':
  24                 uflag++;
  25                 gflag++;
  26                 break;
  27         case 'x':
  28                 xflag++;
  29                 break;
  30         default:
  31                 goto usage;
  32         } ARGEND
  33
  34         todir=0;
  35         if(argc < 2)
  36                 goto usage;
  37         dirb = dirstat(argv[argc-1]);
  38         if(dirb!=nil && (dirb->mode&DMDIR))
  39                 todir=1;
  40         if(argc>2 && !todir){
  41                 fprint(2, "cp: %s not a directory\n", argv[argc-1]);
  42                 exits("bad usage");
  43         }
  44         for(i=0; i<argc-1; i++)
  45                 copy(argv[i], argv[argc-1], todir);
  46         if(failed)
  47                 exits("errors");
  48         exits(0);
  49
  50 usage:
  51         fprint(2, "usage:\tcp [-gux] fromfile tofile\n");
  52         fprint(2, "\tcp [-x] fromfile ... todir\n");
  53         exits("usage");
  54 }
  55
  56 int
  57 samefile(Dir *a, char *an, char *bn)
  58 {
  59         Dir *b;
  60         int ret;
  61
  62         ret = 0;
  63         b=dirstat(bn);
  64         if(b != nil)
  65         if(b->qid.type==a->qid.type)
  66         if(b->qid.path==a->qid.path)
  67         if(b->qid.vers==a->qid.vers)
  68         if(b->dev==a->dev)
  69         if(b->type==a->type){
  70                 fprint(2, "cp: %s and %s are the same file\n", an, bn);
  71                 ret = 1;
  72         }
  73         free(b);
  74         return ret;
  75 }
  76
  77 void
  78 copy(char *from, char *to, int todir)
  79 {
  80         Dir *dirb, dirt;
  81         char *name;
  82         int fdf, fdt, mode;
  83
  84         name = nil;
  85         if(todir){
  86                 char *s, *elem;
  87                 elem=s=from;
  88                 while(*s++)
  89                         if(s[-1]=='/')
  90                                 elem=s;
  91                 name = smprint("%s/%s", to, elem);
  92                 to = name;
  93         }
  94
  95         fdf = fdt = -1;
  96         if((dirb=dirstat(from))==nil){
  97                 fprint(2,"cp: can't stat %s: %r\n", from);
  98                 failed = 1;
  99                 goto out;
 100         }
 101         mode = dirb->mode;
 102         if(mode&DMDIR){
 103                 fprint(2, "cp: %s is a directory\n", from);
 104                 failed = 1;
 105                 goto out;
 106         }
 107         if(samefile(dirb, from, to)){
 108                 failed = 1;
 109                 goto out;
 110         }
 111         mode &= 0777;
 112         fdf=open(from, OREAD);
 113         if(fdf<0){
 114                 fprint(2, "cp: can't open %s: %r\n", from);
 115                 failed = 1;
 116                 goto out;
 117         }
 118         fdt=create(to, OWRITE, mode);
 119         if(fdt<0){
 120                 fprint(2, "cp: can't create %s: %r\n", to);
 121                 failed = 1;
 122                 goto out;
 123         }
 124         if(copy1(fdf, fdt, from, to)==0 && (xflag || gflag || uflag)){
 125                 nulldir(&dirt);
 126                 if(xflag){
 127                         dirt.mtime = dirb->mtime;
 128                         dirt.mode = dirb->mode;
 129                 }
 130                 if(uflag)
 131                         dirt.uid = dirb->uid;
 132                 if(gflag)
 133                         dirt.gid = dirb->gid;
 134                 if(dirfwstat(fdt, &dirt) < 0)
 135                         fprint(2, "cp: warning: can't wstat %s: %r\n", to);
 136         }
 137 out:
 138         free(dirb);
 139         if(fdf >= 0)
 140                 close(fdf);
 141         if(fdt >= 0)
 142                 close(fdt);
 143         free(name);
 144 }
 145
 146 int
 147 copy1(int fdf, int fdt, char *from, char *to)
 148 {
 149         char *buf;
 150         long n, n1, rcount;
 151         int rv;
 152         char err[ERRMAX];
 153
 154         buf = malloc(DEFB);
 155         /* clear any residual error */
 156         err[0] = '\0';
 157         errstr(err, ERRMAX);
 158         rv = 0;
 159         for(rcount=0;; rcount++) {
 160                 n = read(fdf, buf, DEFB);
 161                 if(n <= 0)
 162                         break;
 163                 n1 = write(fdt, buf, n);
 164                 if(n1 != n) {
 165                         fprint(2, "cp: error writing %s: %r\n", to);
 166                         failed = 1;
 167                         rv = -1;
 168                         break;
 169                 }
 170         }
 171         if(n < 0) {
 172                 fprint(2, "cp: error reading %s: %r\n", from);
 173                 failed = 1;
 174                 rv = -1;
 175         }
 176         free(buf);
 177         return rv;
 178 }