123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251 |
- /*
- * This file is part of the UCB release of Plan 9. It is subject to the license
- * terms in the LICENSE file found in the top-level directory of this
- * distribution and at http://akaros.cs.berkeley.edu/files/Plan9License. No
- * part of the UCB release of Plan 9, including this file, may be copied,
- * modified, propagated, or distributed except according to the terms contained
- * in the LICENSE file.
- */
- #include <u.h>
- #include <libc.h>
- #define DEFB (8*1024)
- #define Nwork 16
- int failed;
- int gflag;
- int uflag;
- int xflag;
- void copy(char *from, char *to, int todir);
- int copy1(int fdf, int fdt, char *from, char *to);
- void worker(int fdf, int fdt, char *from, char *to);
- int64_t nextoff(void);
- void failure(void *, char *note);
- QLock lk;
- int64_t off;
- void
- main(int argc, char *argv[])
- {
- Dir *dirb;
- int todir, i;
- ARGBEGIN {
- case 'g':
- gflag++;
- break;
- case 'u':
- uflag++;
- gflag++;
- break;
- case 'x':
- xflag++;
- break;
- default:
- goto usage;
- } ARGEND
- todir=0;
- if(argc < 2)
- goto usage;
- dirb = dirstat(argv[argc-1]);
- if(dirb!=nil && (dirb->mode&DMDIR))
- todir=1;
- if(argc>2 && !todir){
- fprint(2, "fcp: %s not a directory\n", argv[argc-1]);
- exits("bad usage");
- }
- for(i=0; i<argc-1; i++)
- copy(argv[i], argv[argc-1], todir);
- if(failed)
- exits("errors");
- exits(0);
- usage:
- fprint(2, "usage:\tfcp [-gux] fromfile tofile\n");
- fprint(2, "\tfcp [-x] fromfile ... todir\n");
- exits("usage");
- }
- int
- samefile(Dir *a, char *an, char *bn)
- {
- Dir *b;
- int ret;
- ret = 0;
- b=dirstat(bn);
- if(b != nil)
- if(b->qid.type==a->qid.type)
- if(b->qid.path==a->qid.path)
- if(b->qid.vers==a->qid.vers)
- if(b->dev==a->dev)
- if(b->type==a->type){
- fprint(2, "fcp: %s and %s are the same file\n", an, bn);
- ret = 1;
- }
- free(b);
- return ret;
- }
- void
- copy(char *from, char *to, int todir)
- {
- Dir *dirb, dirt;
- char name[256];
- int fdf, fdt, mode;
- if(todir){
- char *s, *elem;
- elem=s=from;
- while(*s++)
- if(s[-1]=='/')
- elem=s;
- sprint(name, "%s/%s", to, elem);
- to=name;
- }
- if((dirb=dirstat(from))==nil){
- fprint(2,"fcp: can't stat %s: %r\n", from);
- failed = 1;
- return;
- }
- mode = dirb->mode;
- if(mode&DMDIR){
- fprint(2, "fcp: %s is a directory\n", from);
- free(dirb);
- failed = 1;
- return;
- }
- if(samefile(dirb, from, to)){
- free(dirb);
- failed = 1;
- return;
- }
- mode &= 0777;
- fdf=open(from, OREAD);
- if(fdf<0){
- fprint(2, "fcp: can't open %s: %r\n", from);
- free(dirb);
- failed = 1;
- return;
- }
- fdt=create(to, OWRITE, mode);
- if(fdt<0){
- fprint(2, "fcp: can't create %s: %r\n", to);
- close(fdf);
- free(dirb);
- failed = 1;
- return;
- }
- if(copy1(fdf, fdt, from, to)==0 && (xflag || gflag || uflag)){
- nulldir(&dirt);
- if(xflag){
- dirt.mtime = dirb->mtime;
- dirt.mode = dirb->mode;
- }
- if(uflag)
- dirt.uid = dirb->uid;
- if(gflag)
- dirt.gid = dirb->gid;
- if(dirfwstat(fdt, &dirt) < 0)
- fprint(2, "fcp: warning: can't wstat %s: %r\n", to);
- }
- free(dirb);
- close(fdf);
- close(fdt);
- }
- int
- copy1(int fdf, int fdt, char *from, char *to)
- {
- int i, n, rv, pid[Nwork];
- Waitmsg *w;
- n = 0;
- off = 0;
- for(i=0; i<Nwork; i++){
- switch(pid[n] = rfork(RFPROC|RFMEM)){
- case 0:
- notify(failure);
- worker(fdf, fdt, from, to);
- case -1:
- break;
- default:
- n++;
- break;
- }
- }
- if(n == 0){
- fprint(2, "fcp: rfork: %r\n");
- failed = 1;
- return -1;
- }
- rv = 0;
- while((w = wait()) != nil){
- if(w->msg[0]){
- rv = -1;
- failed = 1;
- for(i=0; i<n; i++)
- if(pid[i] > 0)
- postnote(PNPROC, pid[i], "failure");
- }
- free(w);
- }
- return rv;
- }
- void
- worker(int fdf, int fdt, char *from, char *to)
- {
- char buf[DEFB], *bp;
- int32_t len, n;
- int64_t o;
- len = sizeof(buf);
- bp = buf;
- o = nextoff();
- while(n = pread(fdf, bp, len, o)){
- if(n < 0){
- fprint(2, "reading %s at %lld: %r\n", from, o);
- _exits("bad");
- }
- if(pwrite(fdt, buf, n, o) != n){
- fprint(2, "writing %s: %r\n", to);
- _exits("bad");
- }
- bp += n;
- o += n;
- len -= n;
- if(len == 0){
- len = sizeof buf;
- bp = buf;
- o = nextoff();
- }
- }
- _exits(nil);
- }
- int64_t
- nextoff(void)
- {
- int64_t o;
- qlock(&lk);
- o = off;
- off += DEFB;
- qunlock(&lk);
- return o;
- }
- void
- failure(void *v, char *note)
- {
- if(strcmp(note, "failure") == 0)
- _exits(nil);
- noted(NDFLT);
- }
|