stpscript2ko.pl

#!/usr/bin/perl
#
# use for to convert script to kernel module.
#   stap -r 2.6.35.14-106.fc14.i686 nettop.stp -m nettop -p4
#       
#       -p NUM Stop  after  pass  NUM.   
#           The passes are numbered 1-5: parse, elaborate, translate,
#           compile, run.  See the PROCESSING section for details.

use strict;
use warnings;
use Getopt::Long;
use File::Basename;
use Term::ANSIColor;

my $DEBUG = 0;
if ($DEBUG) {
    eval q{
        use Smart::Comments;
    };
    die $@ if $@;
}

my $uname;
chomp($uname = `uname -r`); ## $uname
my $myscript; # systemstp script to convert. 
my $mymodule; # kernel module name.
my $guru; # guru mode
my $kerneldir; # kernel source code dir.
my $bufsize; # bufsize (megebyte) for kernel-to-user data transfer. 
my $program = &_my_program();

my $command = 'stap';

my $usage = "
Usage: $program [option] ...

       -g  
            Guru mode. Enable parsing of unsafe expert-level 
            constructs like embedded C.(Not enable default, 
            if script contains embedded C, must enable it.)

       -h, --help 
            Display this help and exit

       -m [modulename], --module [modulename]
            The output module name(default: scriptname.ko).

       -s scriptname, --script scriptname
            A script must be specified.
            The script name want to converted to *.ko.
       
       -r /DIR
            Build for kernel in given build tree. Can also 
            be set with the SYSTEMTAP_RELEASE environment variable.

       -r RELEASE (default)
            Build for kernel in build tree /lib/modules/RELEASE/build. 
            Can also be set with the SYSTEMTAP_RELEASE environment variable.

       -V   Display version information.
       
       -sNUM  
            Use NUM megabyte buffers for kernel-to-user data transfer.  
            On a multiprocessor in bulk mode, this is a per-processor amount.
";

my $ret = GetOptions(
    'script|s=s' => \$myscript,
    'module|m=s' => \$mymodule,
    'help|h'     => \&usage,
    'V'          => \&usage,
    'g'          => \$guru,
    'r=s'        => \$kerneldir,
    'size=i'     => \$bufsize
);

$| =1;

if(! $ret) {
   &usage(); 
}

unless($myscript) {
    &mydie("A script must be specified.");
}

if(! -e $myscript) {
    &mydie("The input script: $myscript not existed.");
}

unless($mymodule) {
    if($myscript =~ /^((\w)+)\.(\w)+/) {
        $mymodule = "$1";
    } else {
        $mymodule = "$myscript";
    }
    
    if($mymodule =~ /[^_0-9a-zA-Z]/) {
        $mymodule =~ s/[^_0-9a-zA-Z]//g;
    }
}
## $mymodule

## the command to finish convertion
if(! $kerneldir) {
    print("\nUse current running kernel: \"$uname\"!\n");
    if($uname =~ /^\'([^']+)\'/) {
        $uname = $1;
    }
    # print "$uname\n";
} else {
    if(-e $kerneldir) {
        ## $kerneldir
        if($kerneldir =~ /((.)+)\/$/) {
            $uname = $1;
        } else {
            $uname = $kerneldir;
        }
        ## $uname
        print("Use kernel build tree: \"$uname\"!\n");
    } else {
        &myprint("The kernel build tree: \"$kerneldir\" not exists!\n");
        print("Use current running kernel: \"$uname\"!\n");
        if($uname =~ /^\'([^']+)\'/) {
            $uname = $1;
        }
    }
}

if($guru) {
    $command = 'stap -g';
}

if($bufsize) {
    $bufsize = "-s$bufsize";
} else {
    $bufsize = '';
}

my $result = `$command $bufsize -r $uname $myscript -m $mymodule -p4 2>&1`;
chomp($result);
## $result
if($? != 0) {
    if($result =~ 'embedded code' and !$guru) {
        &myprint("The script contains embedded C. Make try in enable guru mode use -g.");
        print("Trying enable guru mode.....\n");
        # make try enable guru mode default.
        $command = 'stap -g';
        $result = `$command $bufsize -r $uname $myscript -m $mymodule -p4 2>&1`;
    } 

    if ($result =~ 'no probes') {
        &myprint("Make sure have probes.");
    } elsif ($? != 0) {
        &myprint("$result");
    }
}

if($? == 0) {
    print color("green");
    print "The $myscript successful converted to $result\n";
    print color("reset");
} else {
    &myprint("The $myscript convertion is failed!\n");
}

##----------------------------------------------------
#
sub _my_program {
    require File::Basename;
    return File::Basename::basename( $0 );
}

sub usage {
    print $usage;
    exit;
}

sub mydie {
    print color("red");
    print("@_ \n");
    print color("reset");
    &usage();
}

sub myprint {
    print color("red");
    print("@_ \n");
    print color("reset");
}