loadcpu: Added SNB-optimized spinloop.
[utils.git] / cdtool
CommitLineData
b16b19f3 1#!/usr/bin/perl -w
2
3if(($#ARGV < 0) || ($ARGV[0] eq "help")) {
4 print STDERR "usage:\n";
5 print STDERR "\tcdtool add {FILES|-g GRAFTPOINT}...\n";
6 print STDERR "\tcdtool ls\n";
7 print STDERR "\tcdtool rm NAME\n";
8 print STDERR "\tcdtool szck\n";
9 print STDERR "\tcdtool setgp [GRAFTPOINT]\n";
10 print STDERR "\tcdtool setimg [IMAGE]\n";
11 print STDERR "\tcdtool mkiso\n";
12 exit 1;
13}
14
15if(-r $ENV{"HOME"} . "/.cdtool") {
16 if(open CF, $ENV{"HOME"} . "/.cdtool") {
17 while(<CF>) {
18 next if /^\#/;
19 if(/^(\S+)\s+(.*)$/) {
20 if($1 eq "gp") {
21 $curgraft = $2;
22 } elsif($1 eq "img") {
23 $curimage = $2;
24 } else {
25 print STDERR "cdtool: warning: unknown .cdtool directive: $1\n";
26 }
27 }
28 }
29 close CF;
30 }
31}
32
33sub canonpath {
34 my($path) = @_;
35 $path = "./" . $path unless $path =~ m,^/,;
36 $path =~ s,/$,,;
37 $trailer = "";
38 while(!-d $path) {
39 $path =~ s,/+([^/]*)$,,;
40 $trailer .= "/" . $1;
41 if($path eq "") {
42 die "weird path given: $_[0]";
43 }
44 }
45 if(!open CHILD, "-|") {
46 chdir $path;
47 exec "pwd";
48 die;
49 }
50 chomp($cpath = <CHILD>);
51 close CHILD;
52 return "$cpath$trailer";
53}
54
55sub dirname {
56 my($path) = @_;
57 $path =~ s,/[^/]+/*$,,;
58 return $path;
59}
60
61sub addfile {
62 my($file, $graft) = @_;
63 my($dir, $f, $basename);
64 $basename = $file;
65 $basename =~ s,^.*/,,;
66 if(-f $file) {
67 print "add $file to $graft\n" if $verbose;
68 $img{"$graft$basename"} = $file;
69 } elsif(-d $file) {
70 $dir = canonpath $file;
71 if(opendir DIR, "$dir") {
72 for $f (sort grep !/^\./, readdir DIR) {
73 addfile("$dir/$f", "$graft$basename/");
74 }
75 closedir DIR;
76 } else {
77 print STDERR "cdtool: could not open $file for listing: $!\n";
78 }
79 } else {
80 print STDERR "cdtool: $file: not a regular file or directory\n";
81 }
82}
83
84sub readimage {
85 my($file, $dieonerr) = @_;
86 $dieonerr = 1 unless defined $dieonerr;
87 if(!open(IMG, "<", $file)) {
88 if($dieonerr) {
89 die "$file: $!";
90 } else {
91 return;
92 }
93 }
94 while(<IMG>) {
95 if(/^file (.*):::(.*)$/) {
96 $img{$1} = $2;
97 } elsif(/^name (.*)$/) {
98 $imgname = $1;
99 } elsif(/^rev (\d+)$/) {
100 $rev = $1;
101 } elsif(/^justburned/) {
102 $justburned = 1;
103 }
104 }
105 close IMG;
106}
107
108sub writeimage {
109 my($file) = @_;
110 open(IMG, ">", $file) || die "$file: $!";
111 print IMG "rev $rev\n";
112 print IMG "justburned\n" if $justburned;
113 for $f (sort keys %img) {
114 print IMG ("file " . $f . ":::" . $img{$f} . "\n");
115 }
116 close IMG;
117}
118
119$image = $curimage;
120$imgname = $image;
121$justburned = 0;
122$imgname =~ s,^.*/,,;
123$rev = 1;
124$verbose = 0;
125
126while(($arg = shift @ARGV) =~ /^-(.+)/) {
127 $opts = $1;
128 while($opts =~ s/^(.)//) {
129 if($1 eq "v") {
130 $verbose = 1;
131 }
132 }
133}
134
135$cmd = $arg;
136if($cmd eq "add") {
137 if(!defined $image) {
138 print STDERR "cdtool: no image defined for adding\n";
139 exit 1;
140 }
141
142 readimage $image, 0;
143 $justburned = 0;
144 $graft = $curgraft;
145 $graft .= "/" unless $graft eq "";
146 while(defined($arg = shift @ARGV)) {
147 if($arg eq "-g") {
148 $graft = shift @ARGV;
149 $graft =~ s,/+$,,;
150 $graft .= "/" unless $graft eq "";
151 } else {
152 if(!defined $graft) {
153 print STDERR "cdtool: no graft point defined when adding\n";
154 exit 1;
155 }
156 addfile canonpath($arg), $graft
157 }
158 }
159 writeimage $image;
160} elsif($cmd eq "ls") {
161 readimage $image;
162 for $gf (sort keys %img) {
163 print "$gf\n";
164 }
165} elsif($cmd eq "rm") {
166 die "no name specified" unless defined($file = shift @ARGV);
167 $file = canonpath $file;
168 readimage $image;
169 $found = 0;
170 if(-d $file) {
171 $file .= "/";
172 for $gname (grep substr($img{$_}, 0, length $file) eq $file, keys %img) {
173 delete $img{$gname};
174 $found = 1;
175 }
176 } else {
177 for $gname (grep $img{$_} eq $file, keys %img) {
178 delete $img{$gname};
179 $found = 1;
180 }
181 }
182 if(!$found) {
183 print STDERR "cdtool: $file not found in image\n";
184 exit 1;
185 }
186 $justburned = 0;
187 writeimage $image;
188} elsif($cmd eq "szck") {
189 readimage $image;
190 $size = 0;
191 for $file (values %img) {
192 $size += -s $file;
193 }
194 $size =~ s/(\d{3})/$1,/g;
195 $size =~ s/,$//;
196 print "$size\n";
197} elsif($cmd eq "setgp") {
198 $curgraft = shift @ARGV;
199 $curgraft =~ s,/+$,,;
200} elsif($cmd eq "setimg") {
201 $curimage = shift @ARGV;
bd024d60 202} elsif($cmd eq "imgck") {
203 readimage $image;
204 for $file (values %img) {
205 if(! -r $file) {
206 print "$file\n";
207 }
208 }
b16b19f3 209} elsif($cmd eq "mkiso") {
210 readimage $image;
211 open(ISO, "|-", "mkisofs", "-J", "-R", "-V", "$imgname r$rev", "-graft-points", "-path-list", "-") || die "mkisofs: $!";
212 for $gf (sort keys %img) {
213 print ISO ($gf . "=" . $img{$gf} . "\n");
214 }
215 close ISO;
216 if(!$justburned) {
217 $rev++;
218 $justburned = 1;
219 }
220 writeimage $image;
221} else {
222 print STDERR "cdtool: unknown command: $cmd\n";
223 exit 1;
224}
225
226die if(!open CF, ">" . $ENV{"HOME"} . "/.cdtool");
227if(defined $curgraft) {
228 print CF "gp $curgraft\n";
229}
230if(defined $curimage) {
231 print CF "img $curimage\n";
232}
233close CF;