From fe6144793fe7d2a4d674204f675d2e48a2fe06d2 Mon Sep 17 00:00:00 2001
From: sin <sin@2f30.org>
Date: Mon, 20 Jan 2014 11:25:57 +0000
Subject: [PATCH] Check mmap() return value and unmap at the end

---
 tr.c | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/tr.c b/tr.c
index 8cca430..a4ddef9 100644
--- a/tr.c
+++ b/tr.c
@@ -124,6 +124,8 @@ main(int argc, char *argv[])
 
 	mappings = (wchar_t *)mmap(NULL, 0x110000 * sizeof(wchar_t),
 				   PROT_READ|PROT_WRITE, MAP_ANON|MAP_PRIVATE, -1, 0);
+	if (mappings == MAP_FAILED)
+		eprintf("mmap:");
 
 	ARGBEGIN {
 	default:
@@ -147,5 +149,7 @@ main(int argc, char *argv[])
 	if(ferror(stdin))
 		eprintf("<stdin>: read error:");
 
+	munmap(mappings, 0x110000 * sizeof(wchar_t));
+
 	return EXIT_SUCCESS;
 }