diff options
| -rw-r--r-- | usr.sbin/diskpart/diskpart.c | 7 |
1 files changed, 6 insertions, 1 deletions
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); |
