1 files changed, 78 insertions, 0 deletions
diff --git a/src/kernel/mem/virt.c b/src/kernel/mem/virt.c
new file mode 100644
index 0000000..2990668
--- /dev/null
+++ b/src/kernel/mem/virt.c
@@ -0,0 +1,78 @@
+#include <kernel/arch/generic.h>
+#include <kernel/mem/virt.h>
+#include <kernel/util.h>
+
+void virt_iter_new(
+		struct virt_iter *iter, user_ptr virt, size_t length,
+		struct pagedir *pages, bool user, bool writeable)
+{
+	iter->frag       = 0;
+	iter->frag_len   = 0;
+	iter->prior      = 0;
+	iter->error      = false;
+	iter->_virt      = virt;
+	iter->_remaining = length;
+	iter->_pages     = pages;
+	iter->_user      = user;
+	iter->_writeable = writeable;
+}
+
+bool virt_iter_next(struct virt_iter *iter) {
+	/* note: While i'm pretty sure that this should always work, this
+	 * was only tested in cases where the pages were consecutive both in
+	 * virtual and physical memory, which might not always be the case.
+	 * TODO test this */
+
+	user_ptr virt = iter->_virt;
+	size_t partial = iter->_remaining;
+	iter->prior   += iter->frag_len;
+	if (partial <= 0) return false;
+
+	// don't read past the page
+	if ((virt & PAGE_MASK) + partial > PAGE_SIZE)
+		partial = PAGE_SIZE - (virt & PAGE_MASK);
+
+	iter->frag = pagedir_virt2phys(iter->_pages,
+			iter->_virt, iter->_user, iter->_writeable);
+
+	if (iter->frag == 0) {
+		iter->error = true;
+		return false;
+	}
+
+	iter->frag_len    = partial;
+	iter->_remaining -= partial;
+	iter->_virt      += partial;
+	return true;
+}
+
+bool virt_user_cpy(
+		struct pagedir *dest_pages,       user_ptr dest,
+		struct pagedir  *src_pages, const user_ptr src, size_t length)
+{
+	struct virt_iter dest_iter, src_iter;
+	size_t min;
+
+	virt_iter_new(&dest_iter, dest, length, dest_pages, true, true);
+	virt_iter_new( &src_iter, src, length,  src_pages, true, false);
+	dest_iter.frag_len = 0;
+	src_iter.frag_len  = 0;
+
+	for (;;) {
+		if (dest_iter.frag_len <= 0)
+			if (!virt_iter_next(&dest_iter)) break;
+		if ( src_iter.frag_len <= 0)
+			if (!virt_iter_next( &src_iter)) break;
+
+		min = src_iter.frag_len < dest_iter.frag_len
+		    ? src_iter.frag_len : dest_iter.frag_len;
+		memcpy(dest_iter.frag, src_iter.frag, min);
+
+		dest_iter.frag_len -= min;
+		dest_iter.frag     += min;
+		src_iter.frag_len  -= min;
+		src_iter.frag      += min;
+	}
+
+	return !(dest_iter.error || src_iter.error);
+}