[BusyBox-cvs] busybox/libbb copy_file.c, 1.25, 1.26 copyfd.c, 1.6, 1.7 print_file.c, 1.7, 1.8

[Glenn McGrath]

Update of /var/cvs/busybox/libbb
In directory winder:/tmp/cvs-serv3728/libbb

Modified Files:
	copy_file.c copyfd.c print_file.c 
Log Message:
As we no longer use function pointers for read in common archiving code 
archive_xread can be replaced with bb_full_read, and archive_copy_file 
with bb_copyfd*
bb_copyfd is split into two functions bb_copyfd_size and bb_copyfd_eof, 
they share a common backend.


Index: copyfd.c
===================================================================
RCS file: /var/cvs/busybox/libbb/copyfd.c,v
retrieving revision 1.6
retrieving revision 1.7
diff -u -d -r1.6 -r1.7
--- copyfd.c	14 Jul 2003 21:20:55 -0000	1.6
+++ copyfd.c	21 Nov 2003 22:24:56 -0000	1.7
@@ -29,44 +29,55 @@
 #define BUFSIZ 4096
 #endif
 
-/* If chunksize is 0 copy until EOF */
-extern int bb_copyfd(int fd1, int fd2, const off_t chunksize)
+/* If size is 0 copy until EOF */
+extern size_t bb_full_fd_action(int src_fd, int dst_fd, const size_t size, ssize_t (*action)(int fd, const void *, size_t))
 {
-	ssize_t nread;
-	size_t size;
-	off_t remaining;
+	size_t read_total = 0;
 	RESERVE_CONFIG_BUFFER(buffer,BUFSIZ);
 
-	remaining = size = BUFSIZ;
-	if (chunksize) {
-		remaining = chunksize;
-	}
+	while ((size == 0) || (read_total < size)) {
+		size_t read_try;
+		ssize_t read_actual;
 
-	do {
-		if (size > remaining) {
-			size = remaining;
+ 		if ((size == 0) || (size - read_total > BUFSIZ)) {
+			read_try = BUFSIZ;
+		} else {
+			read_try = size - read_total;
 		}
 
-		if ((nread = safe_read(fd1, buffer, size)) > 0) {
-			if (bb_full_write(fd2, buffer, nread) < 0) {
+		read_actual = safe_read(src_fd, buffer, read_try);
+		if (read_actual > 0) {
+			if (action && (action(dst_fd, buffer, (size_t) read_actual) != read_actual)) {
 				bb_perror_msg(bb_msg_write_error);	/* match Read error below */
 				break;
 			}
-			if (chunksize && ((remaining -= nread) == 0)) {
-				return 0;
-			}
-		} else if (!nread) {
-			if (chunksize) {
+		}
+		else if (read_actual == 0) {
+			if (size) {
 				bb_error_msg("Unable to read all data");
-				break;
 			}
-			return 0;
-		} else {				/* nread < 0 */
-			bb_perror_msg("Read error");	/* match bb_msg_write_error above */
+			break;
+		} else {
+			/* read_actual < 0 */
+			bb_perror_msg("Read error");
 			break;
 		}
 
-	} while (1);
+		read_total += read_actual;
+	}
 
-	return -1;
+	RELEASE_CONFIG_BUFFER(buffer);
+
+	return(read_total);
+}
+
+
+extern int bb_copyfd_size(int fd1, int fd2, const off_t size)
+{
+	return(bb_full_fd_action(fd1, fd2, size, bb_full_write));
+}
+
+extern int bb_copyfd_eof(int fd1, int fd2)
+{
+	return(bb_full_fd_action(fd1, fd2, 0, bb_full_write));
 }

Index: copy_file.c
===================================================================
RCS file: /var/cvs/busybox/libbb/copy_file.c,v
retrieving revision 1.25
retrieving revision 1.26
diff -u -d -r1.25 -r1.26
--- copy_file.c	26 May 2003 14:06:00 -0000	1.25
+++ copy_file.c	21 Nov 2003 22:24:56 -0000	1.26
@@ -181,7 +181,7 @@
 			}
 		}
 
-		if (bb_copyfd(fileno(sfp), fileno(dfp), 0) == -1)
+		if (bb_copyfd_eof(fileno(sfp), fileno(dfp)) == -1)
 			status = -1;
 
 		if (fclose(dfp) < 0) {

Index: print_file.c
===================================================================
RCS file: /var/cvs/busybox/libbb/print_file.c,v
retrieving revision 1.7
retrieving revision 1.8
diff -u -d -r1.7 -r1.8
--- print_file.c	14 Jul 2003 21:20:56 -0000	1.7
+++ print_file.c	21 Nov 2003 22:24:56 -0000	1.8
@@ -28,7 +28,7 @@
 	bb_xfflush_stdout();
 	/* Note: Do not use STDOUT_FILENO here, as this is a lib routine
 	 *       and the calling code may have reassigned stdout. */
-	if (bb_copyfd(fileno(file), fileno(stdout), 0) == -1) {
+	if (bb_copyfd_eof(fileno(file), fileno(stdout)) == -1) {
 		/* bb_copyfd outputs any needed messages, so just die. */
 		exit(bb_default_error_retval);
 	}