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