vmstat: -p handles /dev/ and does not overflow #319523 #330969

2005-10-30 06:03:28 +00:00
parent 73030f7346
commit 36524ac0fd
2 changed files with 7 additions and 4 deletions
--- a/vmstat.c
+++ b/vmstat.c
@@ -608,10 +608,12 @@ int main(int argc, char *argv[]) {
      case 'p':
        statMode |= PARTITIONSTAT;
 	if (argv[1]){
-	           ++argv;
-		   sprintf(partition, "%s", *argv);
-	 }else{fprintf(stderr, "-p requires an argument\n");
-               exit(EXIT_FAILURE);
+	  char *cp = *++argv;
+	  if(!memcmp(cp,"/dev/",5)) cp += 5;
+	  snprintf(partition, sizeof partition, "%s", cp);
+	}else{
+	  fprintf(stderr, "-p requires an argument\n");
+          exit(EXIT_FAILURE);
 	}
        break;
      case 'S':