From 80e691eb7fe6da094ac1cc2f552ad92046f1a847 Mon Sep 17 00:00:00 2001
From: kevlo <kevlo@FreeBSD.org>
Date: Tue, 6 Nov 2007 10:35:50 +0000
Subject: Check return value for setenv()

---
 usr.bin/du/du.c | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

(limited to 'usr.bin/du')

diff --git a/usr.bin/du/du.c b/usr.bin/du/du.c
index 73583e3..17da719 100644
--- a/usr.bin/du/du.c
+++ b/usr.bin/du/du.c
@@ -140,16 +140,20 @@ main(int argc, char *argv[])
 				cflag = 1;
 				break;
 			case 'h':
-				setenv("BLOCKSIZE", "512", 1);
+				if (setenv("BLOCKSIZE", "512", 1) == -1)
+					warnx(
+					    "setenv: cannot set BLOCKSIZE=512");
 				hflag = 1;
 				break;
 			case 'k':
 				hflag = 0;
-				setenv("BLOCKSIZE", "1024", 1);
+				if (setenv("BLOCKSIZE", "1024", 1) == -1)
+					warnx("setenv: cannot set BLOCKSIZE=1024");
 				break;
 			case 'm':
 				hflag = 0;
-				setenv("BLOCKSIZE", "1048576", 1);
+				if (setenv("BLOCKSIZE", "1048576", 1) == -1)
+					warnx("setenv: cannot set BLOCKSIZE=1048576");
 				break;
 			case 'n':
 				nodumpflag = 1;
-- 
cgit v1.1