Norm Jacobs
2010-06-10 3bb28515cc09646f3843577291517a9be3f05d90
commit | author | age
f1b729 1 #!/usr/perl5/bin/perl
NJ 2 #
3 # CDDL HEADER START
4 #
5 # The contents of this file are subject to the terms of the
6 # Common Development and Distribution License (the "License").
7 # You may not use this file except in compliance with the License.
8 #
9 # You can obtain a copy of the license at usr/src/OPENSOLARIS.LICENSE
10 # or http://www.opensolaris.org/os/licensing.
11 # See the License for the specific language governing permissions
12 # and limitations under the License.
13 #
14 # When distributing Covered Code, include this CDDL HEADER in each
15 # file and include the License file at usr/src/OPENSOLARIS.LICENSE.
16 # If applicable, add the following below this CDDL HEADER, with the
17 # fields enclosed by brackets "[]" replaced with your own identifying
18 # information: Portions Copyright [yyyy] [name of copyright owner]
19 #
20 # CDDL HEADER END
21 #
22 # Copyright (c) 2010, Oracle and/or it's affiliates.  All rights reserved.
23 #
24 #
25 # build-watch.pl - a simple utility to watch a process and it's children under
26 #            dtrace and process the results for dependency information.
27 #
28 # Ex:
29 #    $ export WS_TOP=/top/of/your/workspace
30 #    $ cd $(WS_TOP)/components/{component}
31 #    $ $(WS_TOP)/tools/build-watch.pl -c "gmake install"
32 #
33
34 use File::Temp qw/tempfile/;
35 use Getopt::Long;
36
37 my $verbose = 0;
38
39 my @ignore = (
40     '^[^/\.].*',        # ignore paths that don't begin with / or .    
41     '^/dev/',        # ignore devices
42     '^/etc/',        # ignore config files
43     '^/proc/',        # ignore /proc
44     '^/tmp/',        # ignore temp files
45     '^/var/',        # ignore more temp/volatile files
46     '^/usr/lib/locale/',    # ignore locale support
47     '^/usr/share/lib/make/',    # ignore make bits
48     '^/usr/share/lib/zoneinfo/',    # ignore timezone info
49     '/SUNWspro/',        # ignore compiler bits
50     '/sunstudio12.1/',    # ignore more compiler bits
51     '^/ws/',        # nothing in /ws can be interesting
52     '^\.[/\.]{0,1}$'    # ignore ., .., and ./
53 );
54
55 sub match
56 {
57     local($string, @expressions) = @_;
58
59     foreach (@expressions) {
60         ($string =~ m{$_}) && return (1);
61     }
62     return (0);
63 }
64
65 sub process_dtrace_results
66 {
67     local ($filename) = @_;
68     my (%tools, %files, $fh) = ();
69
70     open($fh, "<$filename") || die "open($filename): $!\n";
71     while (<$fh>) {
72         if (/^TOOL:\s+(\S+) = (\S+)$/) {
73             $tools{$2} = $1;
74         } elsif ((/^FILE:\s+(\S+)\s*$/) && (match($1, @ignore) == 0) &&
75              (-e $1)) {
76             $files{$1} = $1;
77         }
78     }
79     close($fh);
80
81     return (\%tools, \%files);
82 }
83
84 sub generate_package_requirements
85 {
86     local (*tools, *files) = @_;
87     my ($count, %pkgs, @search_strings, $search_string) = (0);
88
89     # create a set of search strings to query the package DB
90     foreach (sort (keys %tools, keys %files)) {
91         if ($count++ % 100 == 0) {
92             defined($search_string) && \
93                 push(@search_strings, $search_string);
94             $search_string = $_;
95         } else {
96             $search_string .= " OR $_";
97         }
98     }
99     push(@search_strings, $search_string);
100
101     # walk through the search strings and query the package DB
102     foreach (@search_strings) {
103         my $IFH;
104
105         open($IFH, "pkg search -l -H -o path,pkg.name '$_'|");
106         while (<$IFH>) {
107                    (/^(\S+)\s+(\S+)$/) && ($pkgs{$1} = $2);
108         }
109         close($IFH);
110     }
111
112     return (\%pkgs);
113 }
114
115 #
116 # Main execution begins here
117 #
118 GetOptions("c|command=s" => \$cmd, "i|input-file=s" => \@file,
119        "v|verbose" => \$verbose);
120
121 if (defined($cmd)) {
122     $file = (tempfile(UNLINK => 1))[1];
123
124     if (!defined($ENV{'WS_TOP'})) {
125         print("WS_TOP must be set in the calling environment\n");
126         exit(1);
127     }
128     ($verbose == 1) && print("*** Executing '$cmd' under dtrace...\n");
129     system($ENV{'WS_TOP'}."/tools/build-watch.d", "-o", $file, "-c", $cmd);
130 }
131
132 ($verbose == 1) && printf("*** Processing results...\n");
133 my ($tools, $files) = process_dtrace_results($file);
134
135 ($verbose == 1) && printf("*** Generating package requirements...\n");
136 my ($pkgs) = generate_package_requirements($tools, $files);
137
138 if (defined($tools)) {
3bb285 139     print "\n";
NJ 140     print "REQUIRED_TOOL +=\t$_\n" for (sort keys %$tools);
f1b729 141 }
NJ 142
143 if (defined($files)) {
3bb285 144     print "\n";
NJ 145     print "REQUIRED_FILE +=\t$_\n" for (sort keys %$files);
f1b729 146 }
NJ 147
148 if (defined($pkgs)) {
149     @unique{values %$pkgs} = ();
3bb285 150     print "\n";
NJ 151     print "REQUIRED_PKG +=\t$_\n" for (sort keys %unique);
f1b729 152 }
NJ 153
154 exit(0);