summaryrefslogtreecommitdiffstats
path: root/scripts/generic/print_calling_makefile.pl
blob: 8f19f10ae85b09bc4b8a7cc646bec5574ac2c9fc (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
#!/usr/bin/env perl
# Copyright (C) 2017 The Qt Company Ltd.
# SPDX-License-Identifier: LicenseRef-Qt-Commercial OR GPL-3.0-only WITH Qt-GPL-exception-1.0

package QtQA::App::PrintCallingMakefile;

use strict;
use warnings;

=head1 NAME

print_calling_makefile - print the name of the currently executing makefile

=head1 SYNOPSIS

  # Within "some-makefile":
  first:
          perl print_calling_makefile.pl

  # then, at command prompt:
  $ nmake -f some-makefile
  some-makefile

When called as a child process of nmake or jom, prints the name of the
currently processed makefile to standard output, and exits. The name of
the makefile is printed as passed on the command-line, so it may be an
absolute or relative path.

This script is a workaround for nmake's lack of any equivalent to GNU make's
$(MAKEFILE_LIST) variable.  For example, in a GNU makefile, where one would
write something like:

  echo Current makefile is $(firstword $(MAKEFILE_LIST))

In a makefile for nmake, one may write:

  for /f "usebackq tokens=*" %%a in (`perl print_calling_makefile.pl`) do echo Current makefile is %%a

This script only works on Windows.

=cut

use Data::Dumper;
use English qw( -no_match_vars );
use Text::ParseWords qw(shellwords);
use Win32::Process::Info qw(WMI);
use Win32;

# Returns 1 if the given $process looks like it refers to jom/nmake.
sub looks_like_make
{
    my ($process) = @_;
    return ($process->{ ExecutablePath } =~ m{\b(?:jom|nmake)\.exe}i);
}

# From the given process $info (arrayref), find and return the process
# with the given $pid, or die.
sub extract_process
{
    my ($info, $pid) = @_;

    my (@process) = grep { $_->{ ProcessId } == $pid } @{ $info };
    if (@process > 1) {
        die "error: multiple processes with ProcessId $pid";
    }
    if (@process == 0) {
        die "error: cannot find process with ProcessId $pid";
    }

    return $process[0];
}

# Given an nmake/jom $command line, extracts and returns the Makefile
# argument, or dies.
sub extract_makefile_from_command
{
    my ($command) = @_;

    my $makefile;

    # We use shellwords() to parse the command-line; this is following
    # Bourne-style rules, which may be incompatible with the way nmake
    # parses its own command line in some respects.  In practice, for
    # the relatively basic command lines we expect to see, this should
    # be acceptable.
    my @words = shellwords( $command );
    my @argv = @words;

    while (my $arg = shift @argv) {
        # All known forms are:
        #
        #   [-f] [Makefile]
        #   [/f] [Makefile]
        #   [-F] [Makefile]
        #   [/F] [Makefile]
        #   [-fMakefile]
        #   [/fMakefile]
        #   [-FMakefile]
        #   [/FMakefile]
        #
        if ($arg =~ m{\A [-/][fF] (.*) \z}xms) {
            if ($1) {
                # makefile is in this argument
                $makefile = $1;
            } else {
                # makefile is in next argument
                # (or, if this is the last argument, we'll die later)
                $makefile = shift @argv;
            }
            last;
        }
    }

    if (!$makefile) {
        local $LIST_SEPARATOR = '] [';
        die "Can't extract makefile from command line $command\n"
           ."Parsed as: [@words]\n";
    }

    return $makefile;
}

# Main function.
sub run
{
    my $pid = Win32::GetCurrentProcessId();

    # $info will contain info for _all_ processes (at least those we have permission to view).
    # It would also be possible to do a new query per process; a basic benchmark showed no
    # significant difference between the two.
    my $info = Win32::Process::Info->new()->GetProcInfo();
    $info || die;

    # $process_tree holds the known process tree, used purely for debugging when something
    # goes wrong.
    my ($process_tree, $process);

    eval {
        $process = extract_process( $info, $pid );

        while ($process) {
            last if looks_like_make( $process );
            $process_tree = { %{$process}, _child => $process_tree };
            $pid = $process->{ ParentProcessId } || die Dumper( $process ).' ... has no ParentProcessId';
            $process = extract_process( $info, $pid );
        }

        if (!$process) {
            die 'error: could not find any calling nmake/jom';
        }
    };

    if (my $error = $@) {
        die "$error\nProcess tree: ".Dumper( $process_tree )."\n";
    }

    my $command = $process->{ CommandLine } || die Dumper( $process ).' ... has no CommandLine';
    my $makefile = extract_makefile_from_command( $command );
    print "$makefile\n";

    return;
}

run unless caller;
1;