From e4661cdb21a6b64afbb72340f841f45eaefbcdcd Mon Sep 17 00:00:00 2001
From: jkh <jkh@FreeBSD.org>
Date: Fri, 1 Dec 1995 10:26:20 +0000
Subject: Account for newline in fgets().  Whoops! Noticed by:	peter

---
 usr.sbin/diskpart/diskpart.c | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'usr.sbin/diskpart')

diff --git a/usr.sbin/diskpart/diskpart.c b/usr.sbin/diskpart/diskpart.c
index 1c6638d..932b76a 100644
--- a/usr.sbin/diskpart/diskpart.c
+++ b/usr.sbin/diskpart/diskpart.c
@@ -375,8 +375,13 @@ promptfordisk(name)
 		(void) fgets(buf, BUFSIZ, stdin);
 		if (buf[0] == 0)
 			dp->d_type = 1;
-		else
+		else {
+			int len = strlen(buf);
+
+			if (buf[len - 1] == '\n')
+				buf[len - 1] = '\0';
 			dp->d_type = gettype(buf, dktypenames);
+		}
 		if (dp->d_type >= 0)
 			break;
 		fprintf(stderr, "%s: unrecognized controller type\n", buf);
-- 
cgit v1.1