From bd5969fc876a10b18613302db7087ef3c40f18e1 Mon Sep 17 00:00:00 2001
From: Ian Moffett <ian@osmora.org>
Date: Thu, 7 Mar 2024 17:28:00 -0500
Subject: lib: Add mlibc

Signed-off-by: Ian Moffett <ian@osmora.org>
---
 lib/mlibc/tests/ansi/ungetc.c | 74 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 74 insertions(+)
 create mode 100644 lib/mlibc/tests/ansi/ungetc.c

(limited to 'lib/mlibc/tests/ansi/ungetc.c')

diff --git a/lib/mlibc/tests/ansi/ungetc.c b/lib/mlibc/tests/ansi/ungetc.c
new file mode 100644
index 0000000..8773c6d
--- /dev/null
+++ b/lib/mlibc/tests/ansi/ungetc.c
@@ -0,0 +1,74 @@
+#include <stdio.h>
+#include <assert.h>
+
+#ifdef USE_HOST_LIBC
+#define TEST_FILE "ungetc-host-libc.tmp"
+#else
+#define TEST_FILE "ungetc.tmp"
+#endif
+
+void test(int buffering) {
+	FILE *f = fopen(TEST_FILE, "w");
+	for(int c = '0'; c <= '9'; c++) {
+		fputc(c, f);
+	}
+	fclose(f);
+
+	f = fopen(TEST_FILE, "r");
+	if (!buffering) {
+		setbuf(f, NULL);
+	}
+
+	assert(ungetc('x', f) == 'x');
+	assert(fgetc(f) == 'x');
+
+	// Test pushing back the same character
+	for (int c = '0'; c <= '9'; c++) {
+		assert(fgetc(f) == c);
+		assert(ungetc(c, f) == c);
+		assert(fgetc(f) == c);
+	}
+	assert(fgetc(f) == EOF);
+	assert(ungetc(EOF, f) == EOF);
+
+	// Even though the spec does not guarantee it, we should be able to
+	// ungetc more than one character.
+	assert(ungetc('x', f) == 'x');
+	assert(ungetc('y', f) == 'y');
+	assert(fgetc(f) == 'y');
+	assert(fgetc(f) == 'x');
+
+	// Seeking should discard the effects of ungetc.
+	assert(ungetc('x', f) == 'x');
+	rewind(f);
+
+	// Test pushing back a different character
+	for (int c = '0'; c <= '9'; c++) {
+		assert(fgetc(f) == c);
+		assert(ungetc(c - '0' + 'a', f) == c - '0' + 'a');
+		assert(fgetc(f) == c - '0' + 'a');
+	}
+
+#ifndef USE_HOST_LIBC
+	// Too many ungetcs should fail.
+	int eof = 0;
+	for (int i = 0; i < 100; i++) {
+		if (ungetc('x', f) == EOF) {
+			eof = 1;
+			break;
+		}
+	}
+	assert(eof);
+#endif
+
+	fclose(f);
+
+	// TODO: Test with other operations, like fread.
+}
+
+int main() {
+	fprintf(stderr, "with buffering...\n");
+	test(1);
+	fprintf(stderr, "without buffering...\n");
+	test(0);
+}
-- 
cgit v1.2.3