summaryrefslogtreecommitdiff
path: root/scripts/vyatta-vtysh.pl
blob: 3c291a34cdb2add511b8cda5106533a3259e2d1b (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
#!/usr/bin/perl -w
#
# Module: vyatta-vtysh.pl
# 
# **** License ****
# Version: VPL 1.0
# 
# The contents of this file are subject to the Vyatta Public License
# Version 1.0 ("License"); you may not use this file except in
# compliance with the License. You may obtain a copy of the License at
# http://www.vyatta.com/vpl
# 
# Software distributed under the License is distributed on an "AS IS"
# basis, WITHOUT WARRANTY OF ANY KIND, either express or implied. See
# the License for the specific language governing rights and limitations
# under the License.
# 
# This code was originally developed by Vyatta, Inc.
# Portions created by Vyatta are Copyright (C) 2007 Vyatta, Inc.
# All Rights Reserved.
# 
# Author: Stig Thormodsrud
# Date: December 2007
# Description: Wrapper script between vyatta cfg templates and vtysh
# 
# **** End License ****
#

use warnings;
use strict;
use POSIX;

my $vtysh   = '/usr/bin/vtysh';
my $logfile = '/tmp/vtysh.log';

my $ignore_error = 0;

sub log_it {
    my ($cmdline) = @_;

    if (substr($cmdline,0,2) eq "-c" and $ignore_error) {
	$cmdline = "-n $cmdline";
    }
    my $timestamp = strftime("%Y%m%d-%H:%M.%S", localtime);
    my $user = $ENV{'USER'};
    $user = "boot" if !defined $user;
    my $rc = open(my $fh, ">>", $logfile);
    if (!defined $rc) {
	print "Can't open $logfile: [$!]\n";
	return;
    }
    print $fh "$timestamp:$user [$cmdline]\n";
    close($fh);
}

sub parse_cmdline {
    my $cmdline;

    foreach my $arg (@ARGV) {
	if (substr($arg, 0, 2) eq "-n") {
	    $ignore_error = 1;
	    next;
	}
	if (substr($arg,0, 2) eq "-c") {
	    #
	    # This script expects a space between the -c and the command,
	    # but try to handle it anyway
	    #
	    if ($arg ne "-c") {
		my $tmp = substr($arg,2);
		$cmdline .= "-c \"$tmp\" ";
		next;
	    } 
	} else {
	    $arg = " \"$arg\" ";
	}
	$cmdline .= $arg;
    }
    return $cmdline
}

#
# Send the config to quagga
#
# Note: Quagga never exits with an error code, but it does print an
#       error message to stdout.  So if there is output, print it and
#       exit with an error.  Certain error messages we dont' care about
#       such as issuing a "no <foo>" when foo doesn't exist.  In those
#       cases it is up to the template file whether or not to fail on 
#       the error code.
#
sub send_cmds_to_quagga {
    my ($cmdline) = @_;

    my $output = `$vtysh $cmdline`;
    if (defined $output && $output ne "") {
	if ($ignore_error) {
	    log_it("Ignoring: $output");
	    return 0;
	}
	log_it("Error: $output");
	print "%$output\n";
	return 1;
    }
    return 0;
}


#
# main
#
my ($cmdline, $rc);
$cmdline = parse_cmdline();
log_it($cmdline);
$rc = send_cmds_to_quagga($cmdline);
exit $rc

#end of file