| H. Peter Anvin | 12b3156 | 2007-03-12 12:16:30 -0700 | [diff] [blame] | 1 | #!/usr/bin/perl -w | 
|  | 2 | # | 
|  | 3 | # Clean a text file -- or directory of text files -- of stealth whitespace. | 
|  | 4 | # WARNING: this can be a highly destructive operation.  Use with caution. | 
|  | 5 | # | 
|  | 6 |  | 
|  | 7 | use bytes; | 
|  | 8 | use File::Basename; | 
|  | 9 |  | 
|  | 10 | # | 
|  | 11 | # Clean up space-tab sequences, either by removing spaces or | 
|  | 12 | # replacing them with tabs. | 
|  | 13 | sub clean_space_tabs($) | 
|  | 14 | { | 
|  | 15 | no bytes;			# Tab alignment depends on characters | 
|  | 16 |  | 
|  | 17 | my($li) = @_; | 
|  | 18 | my($lo) = ''; | 
|  | 19 | my $pos = 0; | 
|  | 20 | my $nsp = 0; | 
|  | 21 | my($i, $c); | 
|  | 22 |  | 
|  | 23 | for ($i = 0; $i < length($li); $i++) { | 
|  | 24 | $c = substr($li, $i, 1); | 
|  | 25 | if ($c eq "\t") { | 
|  | 26 | my $npos = ($pos+$nsp+8) & ~7; | 
|  | 27 | my $ntab = ($npos >> 3) - ($pos >> 3); | 
|  | 28 | $lo .= "\t" x $ntab; | 
|  | 29 | $pos = $npos; | 
|  | 30 | $nsp = 0; | 
|  | 31 | } elsif ($c eq "\n" || $c eq "\r") { | 
|  | 32 | $lo .= " " x $nsp; | 
|  | 33 | $pos += $nsp; | 
|  | 34 | $nsp = 0; | 
|  | 35 | $lo .= $c; | 
|  | 36 | $pos = 0; | 
|  | 37 | } elsif ($c eq " ") { | 
|  | 38 | $nsp++; | 
|  | 39 | } else { | 
|  | 40 | $lo .= " " x $nsp; | 
|  | 41 | $pos += $nsp; | 
|  | 42 | $nsp = 0; | 
|  | 43 | $lo .= $c; | 
|  | 44 | $pos++; | 
|  | 45 | } | 
|  | 46 | } | 
|  | 47 | $lo .= " " x $nsp; | 
|  | 48 | return $lo; | 
|  | 49 | } | 
|  | 50 |  | 
|  | 51 | $name = basename($0); | 
|  | 52 |  | 
|  | 53 | foreach $f ( @ARGV ) { | 
|  | 54 | print STDERR "$name: $f\n"; | 
|  | 55 |  | 
|  | 56 | if (! -f $f) { | 
|  | 57 | print STDERR "$f: not a file\n"; | 
|  | 58 | next; | 
|  | 59 | } | 
|  | 60 |  | 
|  | 61 | if (!open(FILE, '+<', $f)) { | 
|  | 62 | print STDERR "$name: Cannot open file: $f: $!\n"; | 
|  | 63 | next; | 
|  | 64 | } | 
|  | 65 |  | 
|  | 66 | binmode FILE; | 
|  | 67 |  | 
|  | 68 | # First, verify that it is not a binary file; consider any file | 
|  | 69 | # with a zero byte to be a binary file.  Is there any better, or | 
|  | 70 | # additional, heuristic that should be applied? | 
|  | 71 | $is_binary = 0; | 
|  | 72 |  | 
|  | 73 | while (read(FILE, $data, 65536) > 0) { | 
|  | 74 | if ($data =~ /\0/) { | 
|  | 75 | $is_binary = 1; | 
|  | 76 | last; | 
|  | 77 | } | 
|  | 78 | } | 
|  | 79 |  | 
|  | 80 | if ($is_binary) { | 
|  | 81 | print STDERR "$name: $f: binary file\n"; | 
|  | 82 | next; | 
|  | 83 | } | 
|  | 84 |  | 
|  | 85 | seek(FILE, 0, 0); | 
|  | 86 |  | 
|  | 87 | $in_bytes = 0; | 
|  | 88 | $out_bytes = 0; | 
|  | 89 | $blank_bytes = 0; | 
|  | 90 |  | 
|  | 91 | @blanks = (); | 
|  | 92 | @lines  = (); | 
|  | 93 |  | 
|  | 94 | while ( defined($line = <FILE>) ) { | 
|  | 95 | $in_bytes += length($line); | 
|  | 96 | $line =~ s/[ \t\r]*$//;		# Remove trailing spaces | 
|  | 97 | $line = clean_space_tabs($line); | 
|  | 98 |  | 
|  | 99 | if ( $line eq "\n" ) { | 
|  | 100 | push(@blanks, $line); | 
|  | 101 | $blank_bytes += length($line); | 
|  | 102 | } else { | 
|  | 103 | push(@lines, @blanks); | 
|  | 104 | $out_bytes += $blank_bytes; | 
|  | 105 | push(@lines, $line); | 
|  | 106 | $out_bytes += length($line); | 
|  | 107 | @blanks = (); | 
|  | 108 | $blank_bytes = 0; | 
|  | 109 | } | 
|  | 110 | } | 
|  | 111 |  | 
|  | 112 | # Any blanks at the end of the file are discarded | 
|  | 113 |  | 
|  | 114 | if ($in_bytes != $out_bytes) { | 
|  | 115 | # Only write to the file if changed | 
|  | 116 | seek(FILE, 0, 0); | 
|  | 117 | print FILE @lines; | 
|  | 118 |  | 
|  | 119 | if ( !defined($where = tell(FILE)) || | 
|  | 120 | !truncate(FILE, $where) ) { | 
|  | 121 | die "$name: Failed to truncate modified file: $f: $!\n"; | 
|  | 122 | } | 
|  | 123 | } | 
|  | 124 |  | 
|  | 125 | close(FILE); | 
|  | 126 | } |