2 files changed, 58 insertions, 24 deletions

diff --git a/coreutils/cp.c b/coreutils/cp.c
index ce63201..1e10f28 100644
--- a/coreutils/cp.c
+++ b/coreutils/cp.c
@@ -42,26 +42,36 @@ static int followLinks = FALSE;
 static int preserveFlag = FALSE;
 static const char *srcName;
 static const char *destName;
-static const char *skipName;
-static int dirFlag = FALSE;
-
+static int destDirFlag = FALSE;
+static int destExistsFlag = FALSE;
+static int srcDirFlag = FALSE;
 
 static int fileAction(const char *fileName, struct stat* statbuf)
 {
     char newdestName[NAME_MAX];
+
     strcpy(newdestName, destName);
-    if (dirFlag==TRUE) {
-	strcat(newdestName, "/");
-	if ( skipName != NULL)
-	    strcat(newdestName, strstr(fileName, skipName));
-	else
-	    strcat(newdestName, srcName);
+    if ( srcDirFlag == TRUE ) {
+	if (recursiveFlag!=TRUE ) {
+	    fprintf(stderr, "cp: %s: omitting directory\n", srcName);
+	    return( TRUE);
+	}
+	strcat(newdestName, strstr(fileName, srcName) + strlen(srcName));
+    } 
+    
+    if (destDirFlag==TRUE && srcDirFlag == FALSE) {
+	if (newdestName[strlen(newdestName)-1] != '/' ) {
+	    strcat(newdestName, "/");
+	}
+	strcat(newdestName, srcName);
     }
+    
     return (copyFile(fileName, newdestName, preserveFlag, followLinks));
 }
 
 extern int cp_main(int argc, char **argv)
 {
+    struct stat statBuf;
 
     if (argc < 3) {
 	usage (cp_usage);
@@ -96,18 +106,20 @@ extern int cp_main(int argc, char **argv)
 
 
     destName = argv[argc - 1];
-    dirFlag = isDirectory(destName);
+    if (stat(destName, &statBuf) >= 0) {
+	destExistsFlag = TRUE;
+	if (S_ISDIR(statBuf.st_mode))
+	    destDirFlag = TRUE;
+    }
 
-    if ((argc > 3) && dirFlag==FALSE) {
+    if ((argc > 3) && destDirFlag==FALSE) {
 	fprintf(stderr, "%s: not a directory\n", destName);
 	exit (FALSE);
     }
 
     while (argc-- > 1) {
 	srcName = *(argv++);
-	skipName = strrchr(srcName, '/');
-	if (skipName) 
-	    skipName++;
+	srcDirFlag = isDirectory(srcName);
 	if (recursiveAction(srcName, recursiveFlag, followLinks, FALSE,
 			       fileAction, fileAction) == FALSE) {
 	    exit( FALSE);
diff --git a/coreutils/mv.c b/coreutils/mv.c
index 2be3961..d0f3461 100644
--- a/coreutils/mv.c
+++ b/coreutils/mv.c
@@ -35,13 +35,25 @@ static const char mv_usage[] = "mv SOURCE DEST\n"
 
 static const char *srcName;
 static const char *destName;
-static const char *skipName;
 static int dirFlag = FALSE;
 
+static int fileAction(const char *fileName, struct stat* statbuf)
+{
+    char newdestName[NAME_MAX];
+
+    fprintf(stderr, "srcName='%s'  destName='%s'\n", srcName, destName);
+    strcpy(newdestName, destName);
+    strcat(newdestName, "/");
+    strcat(newdestName, strstr(fileName, fileName));
+    fprintf(stderr, "newdestName='%s'\n", newdestName); 
+    return (copyFile(fileName, newdestName, TRUE, TRUE));
+}
+
 
 extern int mv_main(int argc, char **argv)
 {
     char newdestName[NAME_MAX];
+    char *skipName;
 
     if (argc < 3) {
 	usage (mv_usage);
@@ -69,16 +81,26 @@ extern int mv_main(int argc, char **argv)
 		strcat(newdestName, strstr(srcName, skipName));
 	    else
 		strcat(newdestName, srcName);
-	    fprintf(stderr, "srcName='%s'\n", srcName);
-	    fprintf(stderr, "skipName='%s'\n", skipName);
-	    fprintf(stderr, "newdestName='%s'\n", newdestName);
-	}
-	if (copyFile(srcName, newdestName, FALSE, FALSE)  == FALSE) {
-	    exit( FALSE);
 	}
-	if (unlink (srcName) < 0) {
-	    perror (srcName);
-	    exit( FALSE);
+	if (isDirectory(srcName)==TRUE && newdestName[strlen(newdestName)] != '/') {
+		strcat(newdestName, "/");
+	    createPath(newdestName, 0777);
+	    fprintf(stderr, "srcName = '%s'\n", srcName);
+	    fprintf(stderr, "newdestName = '%s'\n", newdestName);
+	    if (recursiveAction(srcName, TRUE, TRUE, FALSE,
+				   fileAction, fileAction) == FALSE) 
+	    {
+		exit( FALSE);
+	    }
+	    exit( TRUE);
+	} else {
+	    if (copyFile(srcName, newdestName, FALSE, FALSE)  == FALSE) {
+		exit( FALSE);
+	    }
+	    if (unlink (srcName) < 0) {
+		perror (srcName);
+		exit( FALSE);
+	    }
 	}
     }
     exit( TRUE);