/*
 * Copyright (c) 1988, 1993, 1994
 *	The Regents of the University of California.  All rights reserved.
 *
 * This code is derived from software contributed to Berkeley by
 * David Hitz of Auspex Systems Inc.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice, this list of conditions and the following disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 * 3. Neither the name of the University nor the names of its contributors
 *    may be used to endorse or promote products derived from this software
 *    without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
 * SUCH DAMAGE.
 */

#include <sys/stat.h>
#include <sys/fcntl.h>

#include <unistd.h>
#include <err.h>
#include <string.h>
#include <stdlib.h>
#include <errno.h>
#include <limits.h>
#include <stdio.h>

#ifdef CMDBOX
#define main(argc, argv)	cp_main(argc, argv)
#endif

static void usage(void);
static int copy(char *from, char *to, int dirflag);


static char iobuf[BUFSIZ];

int
main(int argc, char *argv[])
{
	int r, ch, checkch, i, iflag = 0;
	char *target;
	struct stat to_stat, tmp_stat;

	while ((ch = getopt(argc, argv, "i")) != -1)
		switch(ch) {
		case 'i':
			iflag = isatty(fileno(stdin));
			break;
		case '?':
		default:
			usage();
		}
	argc -= optind;
	argv += optind;

	if (argc < 2)
		usage();

	target = argv[--argc];

	r = stat(target, &to_stat);
	if (r == -1 && errno != ENOENT)
		err(1, "%s", target);
	if (r == -1 || !S_ISDIR(to_stat.st_mode)) {
		/*
		 * File to file
		 */
		if (argc > 1) {
			usage();
			exit(1);
		}
		stat(argv[0], &tmp_stat);
		if (!S_ISREG(tmp_stat.st_mode))
			usage();

		/* interactive mode */
		if (r != -1 && iflag) {
			(void)fprintf(stderr, "overwrite %s? ", target);
			checkch = ch = getchar();
			while (ch != '\n' && ch != EOF)
				ch = getchar();
			if (checkch != 'y')
				exit(0);
		}
		r = copy(argv[0], target, 0);
	} else {
		/*
		 * File(s) to directory
		 */
		r = 0;
		for (i = 0; i < argc; i++)
			r = copy(argv[i], target, 1);
	}
	exit(r);
}

static void
usage(void)
{
	fprintf(stderr, "usage: cp [-i] src target\n"
		"       cp [-i] src1 ... srcN directory\n");
	exit(1);
	/* NOTREACHED */
}

static int
copy(char *from, char *to, int dirflag)
{
	char path[PATH_MAX];
	int fold, fnew, n, mode;
	struct stat stbuf;
	char *p;

	if (dirflag) {
		p = strrchr(from, '/');
		p = p ? p + 1 : from;
		strcpy(path, to);
		if (strcmp(to, "/"))
			strcat(path, "/");
		strcat(path, p);
		to = path;
	}

	if ((fold = open(from, 0)) == -1) {
		warn("%s", from);
		return 1;
	}
	fstat(fold, &stbuf);
	mode = stbuf.st_mode;

	if ((fnew = creat(to, mode)) == -1) {
		warn("%s", to);
		close(fold);
		return 1;
	}
	while ((n = read(fold, iobuf, BUFSIZ)) > 0) {
		if (write(fnew, iobuf, n) != n) {
			warn("%s", to);
			close(fold);
			close(fnew);
			return 1;
		}
	}
	close(fold);
	close(fnew);
	return 0;
}