qubes-linux-utils/qrexec-lib/copy-file.c

#include <unistd.h>
#include <ioall.h>
#include "libqubes-rpc-filecopy.h"
#include "crc32.h"

notify_progress_t *notify_progress_func = NULL;
void register_notify_progress(notify_progress_t *func)
{
	notify_progress_func = func;
}

int copy_file(int outfd, int infd, long long size, unsigned long *crc32)
{
	char buf[4096];
	long long written = 0;
	int ret;
	int count;
	while (written < size) {
		if (size - written > (int)sizeof(buf))
			count = sizeof buf;
		else
			count = size - written;
		ret = read(infd, buf, count);
		if (!ret)
			return COPY_FILE_READ_EOF;
		if (ret < 0)
			return COPY_FILE_READ_ERROR;
		/* acumulate crc32 if requested */
		if (crc32)
			*crc32 = Crc32_ComputeBuf(*crc32, buf, ret);
		if (!write_all(outfd, buf, ret))
			return COPY_FILE_WRITE_ERROR;
		if (notify_progress_func != NULL)
			notify_progress_func(ret, 0);
		written += ret;
	}
	return COPY_FILE_OK;
}

const char * copy_file_status_to_str(int status)
{
	switch (status) {
		case COPY_FILE_OK: return "OK";
		case COPY_FILE_READ_EOF: return "Unexpected end of data while reading";
		case COPY_FILE_READ_ERROR: return "Error reading";
		case COPY_FILE_WRITE_ERROR: return "Error writing";
		default: return "????????";
	}
}