internals: added ntux_dprintf(), a signal-resilient dprintf implementation.

2 files changed, 69 insertions, 0 deletions

diff --git a/src/internal/ntux_dprintf_impl.c b/src/internal/ntux_dprintf_impl.c
new file mode 100644
index 0000000..1f87eb6
--- /dev/null
+++ b/src/internal/ntux_dprintf_impl.c
@@ -0,0 +1,63 @@
+#include <ntapi/ntapi.h>
+#include <psxabi/sys_sysapi.h>
+#include <psxabi/sys_errno.h>
+
+#include <ntux/ntux.h>
+#include "ntux_driver_impl.h"
+#include "ntux_nolibc_impl.h"
+#include "ntux_errinfo_impl.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <unistd.h>
+
+int ntux_dprintf(int fd, const char * fmt, ...)
+{
+	int	ret;
+	int	cnt;
+	int	size;
+	va_list	ap;
+	char *	ch;
+	char *	buf;
+	char	chbuf[2048];
+
+	va_start(ap,fmt);
+
+	size = sizeof(chbuf);
+	buf  = ((cnt = ntapi->vsnprintf(chbuf,size,fmt,ap)) < size)
+		? chbuf : calloc(1, cnt + 1);
+
+	va_end(ap);
+
+	if (buf == chbuf) {
+		(void)0;
+
+	} else if (buf) {
+		va_start(ap,fmt);
+		ntapi->vsprintf(buf,fmt,ap);
+		va_end(ap);
+
+	} else {
+		return -1;
+	}
+
+	ret = 0;
+	ch  = buf;
+
+	for (; cnt && ret>=0; ) {
+		ret = __sys_write(fd,ch,cnt);
+
+		while ((ret < 0) && (ret == -EINTR))
+			ret = __sys_write(fd,ch,cnt);
+
+		ch  += ret;
+		cnt -= ret;
+	}
+
+	ret = (ret < 0) ? -1 : ch - buf;
+
+	if (buf != chbuf)
+		free(buf);
+
+	return ret;
+}
diff --git a/src/internal/ntux_dprintf_impl.h b/src/internal/ntux_dprintf_impl.h
new file mode 100644
index 0000000..4296bd0
--- /dev/null
+++ b/src/internal/ntux_dprintf_impl.h
@@ -0,0 +1,6 @@
+#ifndef NTUX_DPRINTF_IMPL_H
+#define NTUX_DPRINTF_IMPL_H
+
+int ntux_dprintf(int fd, const char * fmt, ...);
+
+#endif