shorewall_code/New/Shorewall/Common.pm

package Shorewall::Common;
require Exporter;
use File::Basename;
use File::Temp qw/ tempfile tempdir /;

use strict;

our @ISA = qw(Exporter);
our @EXPORT = qw(ALLIPv4

		 warning_message 
		 fatal_error
		 create_temp_object
		 finalize_object
		 emit 
		 emit_unindented
		 save_progress_message
		 save_progress_message_short
		 progress_message
		 progress_message2
		 progress_message3
		 push_indent
		 pop_indent
		 copy
		 copy1
		 
		 @allipv4
		 @rfc1918_networks
		 $line
		 $command
		 $doing
		 $done
		 );
our @EXPORT_OK = ();
our @VERSION = 1.00;

#
# Some IPv4 useful stuff
#
our @allipv4 = ( '0.0.0.0/0' );

use constant { ALLIPv4 => '0.0.0.0/0' };

our @rfc1918_networks = ( "10.0.0.0/24", "172.16.0.0/12", "192.168.0.0/16" );

our $line = '';          # Current config file line

our ( $command, $doing, $done ) = qw/ compile Compiling Compiled/; #describe the current command, it's present progressive, and it's completion.

my $object = 0;          # Object file Handle Reference
my $lastlineblank = 0;   # Avoid extra blank lines in the output
my $indent        = '';
my ( $dir, $file );      # Object's Directory and File
my $tempfile;            # Temporary File Name

#
# Issue a Warning Message
#
sub warning_message 
{
    print STDERR "   WARNING: @_\n";
}

#
# Fatal Error
#
sub fatal_error
{
    print STDERR "   ERROR: @_\n";
    close $object, if $object;
    system "rm -rf $ENV{TMP_DIR}" if $ENV{TMP_DIR};
    die;
}

sub create_temp_object( $ ) {
    my $objectfile = $_[0];
    my $suffix;

    eval {
	( $file, $dir, $suffix ) = fileparse( $objectfile );
	fatal_error "Directory $dir does not exist" unless -d $dir;
	fatal_error "$dir is a Symbolic Link" if -l $dir;
	fatal_error "$objectfile is a Directory" if -d $objectfile;
	fatal_error "$dir is a Symbolic Link" if -l $objectfile;
	fatal_error "$objectfile exists and is not a compiled script" if -e _ && ! -x _;
	( $object, $tempfile ) = tempfile ( 'tempfileXXXX' , DIR => $dir );
    };

    fatal_error "$@" if $@;

    $file = "$dir/$file.$suffix";

}

sub finalize_object() {
    rename $tempfile, $file;
    chmod 0700, $file;
}

#
# Write the argument to the object file (if any) with the current indentation.
# 
# Replaces leading spaces with tabs as appropriate and suppresses consecutive blank lines.
#
sub emit ( $ ) {
    if ( $object ) {
	my $line = $_[0];

	unless ( $line =~ /^\s+$/ ) {
	    $line =~ s/^/$indent/gm if $indent && $line;
	    1 while $line =~ s/^        /\t/m;
	    print $object "$line\n";
	    $lastlineblank = 0;
	} else {
	    print $object '' unless $lastlineblank;
	    $lastlineblank = 1;
	}
    }
}

#
# Write passed message to the object with no indentation.
#

sub emit_unindented( $ ) {
    print $object "$_[0]\n" if $object;
}

#
# Write a progress_message2 command to the output file.
#
sub save_progress_message( $ ) {
    emit "\nprogress_message2 $_[0]\n" if $object;
}

sub progress_message {
    if ( $ENV{VERBOSE} > 1 ) {
	my $ts = '';
	$ts = ( localtime ) . ' ' if $ENV{TIMESTAMP};
	print "${ts}@_\n";
    }
}

sub timestamp() {
    my ($sec, $min, $hr) = ( localtime ) [0,1,2];
    printf '%02d:%02d:%02d ', $hr, $min, $sec;
}

sub progress_message2 {
    if ( $ENV{VERBOSE} > 0 ) {
	timestamp if $ENV{TIMESTAMP};
	print "@_\n";
    }
}

sub progress_message3 {
    if ( $ENV{VERBOSE} >= 0 ) {
	timestamp if $ENV{TIMESTAMP};
	print "@_\n";
    }
}

#
# Push/Pop Indent
#
sub push_indent() {
    $indent = "$indent    ";
}

sub pop_indent() {
    $indent = substr( $indent , 0 , ( length $indent ) - 4 );
}

sub save_progress_message_short( $ ) {
    emit "progress_message $_[0]" if $object;
}

#
# Functions for copying files into the object
#
sub copy( $ ) {
    if ( $object ) {
	my $file = $_[0];
    
	open IF , $file or fatal_error "Unable to open $file: $!";
	    
	while ( my $line = <IF> ) {
	    $line =~ s/^/$indent/ if $indent;
	    print $object $line;
	}
	
	close IF;
    }
}

sub copy1( $ ) {
    if ( $object ) {
	my $file = $_[0];
    
	open IF , $file or fatal_error "Unable to open $file: $!";
	    
	my $do_indent = 1;

	while ( my $line = <IF> ) {
	    if ( $line =~ /^\s+$/ ) {
		print $object "\n";
		$do_indent = 1;
		next;
	    }
	    
	    $line =~ s/^/$indent/ if $indent && $do_indent;
	    print $object $line;
	    $do_indent = ! ( $line =~ /\\$/ );
	}
	
	close IF;
    }
}

1;