code: 9ferno

ref: 44ce0097b612a1fefd754065bdf8d9d2e5ef60c8
dir: /utils/cp/cp.c/

#include <lib9.h>

#define	DEFB	(8*1024)

void	copy(char *from, char *to, int todir);
void	copy1(int fdf, int fdt, char *from, char *to);
void	fixbackslash(char *file);

void
main(int argc, char *argv[])
{
	Dir *dirb;
	int todir, i;

	if(argc<3){
		fprint(2, "usage:\tcp fromfile tofile\n");
		fprint(2, "\tcp fromfile ... todir\n");
		exits("usage");
	}

	for(i=0; i<argc; i++)
		fixbackslash(argv[i]);


	todir=0;
	if((dirb = dirstat(argv[argc-1]))!=nil && (dirb->mode&DMDIR))
		todir=1;
	if(argc>3 && !todir){
		fprint(2, "cp: %s not a directory\n", argv[argc-1]);
		exits("bad usage");
	}
	for(i=1; i<argc-1; i++)
		copy(argv[i], argv[argc-1], todir);
	exits(0);
}

void
copy(char *from, char *to, int todir)
{
	Dir *dirb, *dirt;
	char name[256];
	int fdf, fdt;

	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,"cp: can't stat %s: %r\n", from);
		return;
	}
	if(dirb->mode&DMDIR){
		fprint(2, "cp: %s is a directory\n", from);
		return;
	}
	dirb->mode &= 0777;
	if((dirt = dirstat(to))!=nil)
	if(dirb->qid.path==dirt->qid.path && dirb->qid.vers==dirt->qid.vers)
	if(dirb->dev==dirt->dev && dirb->type==dirt->type){
		fprint(2, "cp: %s and %s are the same file\n", from, to);
		return;
	}
	fdf=open(from, OREAD);
	if(fdf<0){
		fprint(2, "cp: can't open %s: %r\n", from);
		return;
	}
	fdt=create(to, OWRITE, dirb->mode);
	if(fdt<0){
		fprint(2, "cp: can't create %s: %r\n", to);
		close(fdf);
		return;
	}
	copy1(fdf, fdt, from, to);
	close(fdf);
	close(fdt);
}

void
copy1(int fdf, int fdt, char *from, char *to)
{
	char *buf;
	long n, n1, rcount;
	char err[ERRMAX];

	buf = malloc(DEFB);
	/* clear any residual error */
	err[0] = '\0';
	errstr(err, ERRMAX);
	for(rcount=0;; rcount++) {
		n = read(fdf, buf, DEFB);
		if(n <= 0)
			break;
		n1 = write(fdt, buf, n);
		if(n1 != n) {
			fprint(2, "cp: error writing %s: %r\n", to);
			break;
		}
	}
	if(n < 0) 
		fprint(2, "cp: error reading %s: %r\n", from);
	free(buf);
}

void
fixbackslash(char *file)
{
	char *p;
	
	for(p=file; *p; p++)
		if(*p == '\\')
			*p = '/';
}