Added write offset

Made offsets movable

2 files changed, 9 insertions, 6 deletions

diff --git a/include/lib/mem.h b/include/lib/mem.h
index 04056f7..646a170 100644
--- a/include/lib/mem.h
+++ b/include/lib/mem.h
@@ -15,7 +15,8 @@ struct MemTab {
 struct RotBuffer {
 	void* base;
 	unsigned int size;
-	unsigned int offset;
+	unsigned int roffset;
+	unsigned int woffset;
 };
 
 void memset(unsigned char* dest, unsigned char value, unsigned int n);
diff --git a/src/lib/mem.c b/src/lib/mem.c
index dc915e6..e51ec32 100644
--- a/src/lib/mem.c
+++ b/src/lib/mem.c
@@ -5,13 +5,15 @@ void memcpyrot(unsigned char* src, struct RotBuffer* rb, unsigned int n)
 {
 	if (n > rb->size)
 		return;
-	unsigned char* addr = rb->base + rb->offset;
+	unsigned char* base = rb->base;
+	unsigned int offset = rb->woffset;
 	for (unsigned int i = 0; i < n; i++) {
-		if (addr >= (unsigned char*)(rb->base + rb->size))
-			addr = rb->base;
-		*addr = src[i];
-		addr++;
+		if (offset >= rb->size)
+			offset = 0;
+		base[offset] = src[i];
+		offset++;
 	}
+	rb->woffset = offset;
 }
 
 void memshow32(unsigned long* addr, unsigned int n)