blob: c00c69b9171a216046682d8681bfd51fb6d4d4d4 [file] [log] [blame]
Kamil Rytarowskicb77f0d2017-05-07 23:25:26 +02001#!/usr/bin/env perl
Greg Kroah-Hartmanb2441312017-11-01 15:07:57 +01002# SPDX-License-Identifier: GPL-2.0
H. Peter Anvin12b31562007-03-12 12:16:30 -07003#
4# Clean a text file -- or directory of text files -- of stealth whitespace.
5# WARNING: this can be a highly destructive operation. Use with caution.
6#
7
Kamil Rytarowskicb77f0d2017-05-07 23:25:26 +02008use warnings;
H. Peter Anvin12b31562007-03-12 12:16:30 -07009use bytes;
10use File::Basename;
11
H. Peter Anvincb3ed5b2007-05-25 17:58:26 -070012# Default options
13$max_width = 79;
14
H. Peter Anvin12b31562007-03-12 12:16:30 -070015# Clean up space-tab sequences, either by removing spaces or
16# replacing them with tabs.
17sub clean_space_tabs($)
18{
19 no bytes; # Tab alignment depends on characters
20
21 my($li) = @_;
22 my($lo) = '';
23 my $pos = 0;
24 my $nsp = 0;
25 my($i, $c);
26
27 for ($i = 0; $i < length($li); $i++) {
28 $c = substr($li, $i, 1);
29 if ($c eq "\t") {
30 my $npos = ($pos+$nsp+8) & ~7;
31 my $ntab = ($npos >> 3) - ($pos >> 3);
32 $lo .= "\t" x $ntab;
33 $pos = $npos;
34 $nsp = 0;
35 } elsif ($c eq "\n" || $c eq "\r") {
36 $lo .= " " x $nsp;
37 $pos += $nsp;
38 $nsp = 0;
39 $lo .= $c;
40 $pos = 0;
41 } elsif ($c eq " ") {
42 $nsp++;
43 } else {
44 $lo .= " " x $nsp;
45 $pos += $nsp;
46 $nsp = 0;
47 $lo .= $c;
48 $pos++;
49 }
50 }
51 $lo .= " " x $nsp;
52 return $lo;
53}
54
H. Peter Anvincb3ed5b2007-05-25 17:58:26 -070055# Compute the visual width of a string
56sub strwidth($) {
57 no bytes; # Tab alignment depends on characters
58
59 my($li) = @_;
60 my($c, $i);
61 my $pos = 0;
62 my $mlen = 0;
63
64 for ($i = 0; $i < length($li); $i++) {
65 $c = substr($li,$i,1);
66 if ($c eq "\t") {
67 $pos = ($pos+8) & ~7;
68 } elsif ($c eq "\n") {
69 $mlen = $pos if ($pos > $mlen);
70 $pos = 0;
71 } else {
72 $pos++;
73 }
74 }
75
76 $mlen = $pos if ($pos > $mlen);
77 return $mlen;
78}
79
H. Peter Anvin12b31562007-03-12 12:16:30 -070080$name = basename($0);
81
H. Peter Anvincb3ed5b2007-05-25 17:58:26 -070082@files = ();
83
84while (defined($a = shift(@ARGV))) {
85 if ($a =~ /^-/) {
86 if ($a eq '-width' || $a eq '-w') {
87 $max_width = shift(@ARGV)+0;
88 } else {
89 print STDERR "Usage: $name [-width #] files...\n";
90 exit 1;
91 }
92 } else {
93 push(@files, $a);
94 }
95}
96
97foreach $f ( @files ) {
H. Peter Anvin12b31562007-03-12 12:16:30 -070098 print STDERR "$name: $f\n";
99
100 if (! -f $f) {
101 print STDERR "$f: not a file\n";
102 next;
103 }
104
105 if (!open(FILE, '+<', $f)) {
106 print STDERR "$name: Cannot open file: $f: $!\n";
107 next;
108 }
109
110 binmode FILE;
111
112 # First, verify that it is not a binary file; consider any file
113 # with a zero byte to be a binary file. Is there any better, or
114 # additional, heuristic that should be applied?
115 $is_binary = 0;
116
117 while (read(FILE, $data, 65536) > 0) {
118 if ($data =~ /\0/) {
119 $is_binary = 1;
120 last;
121 }
122 }
123
124 if ($is_binary) {
125 print STDERR "$name: $f: binary file\n";
126 next;
127 }
128
129 seek(FILE, 0, 0);
130
131 $in_bytes = 0;
132 $out_bytes = 0;
133 $blank_bytes = 0;
134
135 @blanks = ();
136 @lines = ();
H. Peter Anvincb3ed5b2007-05-25 17:58:26 -0700137 $lineno = 0;
H. Peter Anvin12b31562007-03-12 12:16:30 -0700138
139 while ( defined($line = <FILE>) ) {
H. Peter Anvincb3ed5b2007-05-25 17:58:26 -0700140 $lineno++;
H. Peter Anvin12b31562007-03-12 12:16:30 -0700141 $in_bytes += length($line);
142 $line =~ s/[ \t\r]*$//; # Remove trailing spaces
143 $line = clean_space_tabs($line);
144
145 if ( $line eq "\n" ) {
146 push(@blanks, $line);
147 $blank_bytes += length($line);
148 } else {
149 push(@lines, @blanks);
150 $out_bytes += $blank_bytes;
151 push(@lines, $line);
152 $out_bytes += length($line);
153 @blanks = ();
154 $blank_bytes = 0;
155 }
H. Peter Anvincb3ed5b2007-05-25 17:58:26 -0700156
157 $l_width = strwidth($line);
158 if ($max_width && $l_width > $max_width) {
159 print STDERR
160 "$f:$lineno: line exceeds $max_width characters ($l_width)\n";
161 }
H. Peter Anvin12b31562007-03-12 12:16:30 -0700162 }
163
164 # Any blanks at the end of the file are discarded
165
166 if ($in_bytes != $out_bytes) {
167 # Only write to the file if changed
168 seek(FILE, 0, 0);
169 print FILE @lines;
170
171 if ( !defined($where = tell(FILE)) ||
172 !truncate(FILE, $where) ) {
173 die "$name: Failed to truncate modified file: $f: $!\n";
174 }
175 }
176
177 close(FILE);
178}