summaryrefslogtreecommitdiff
path: root/scripts/system/vyatta_update_console.pl
blob: 5045a0dd630de0775e4327c52b70714705f4c7d2 (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
#! /usr/bin/perl

# **** License ****
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License version 2 as
# published by the Free Software Foundation.
#
# This program is distributed in the hope that it will be useful, but
# WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
# General Public License for more details.
#
# This code was originally developed by Vyatta, Inc.
# Portions created by Vyatta are Copyright (C) 2010 Vyatta, Inc.
# All Rights Reserved.
#
# **** End License ****

# Update console configuration in /etc/inittab and grub
# based on Vyatta configuration

use strict;
use warnings;

use lib "/opt/vyatta/share/perl5";
use Vyatta::Config;
use File::Compare;
use File::Copy;

die "$0 expects no arguments\n" if (@ARGV);

# if file is unchanged, do nothing and return false
# otherwise update to new version
sub update {
    my ($old, $new) = @_;

    if ( compare($old, $new) != 0) {
	move($new, $old)
	    or die "Can't move $new to $old";
	return 1;
    } else {
	unlink($new);
	return;
    }
}

my $INITTAB = "/etc/inittab";
my $TMPTAB  = "/tmp/inittab.$$";

sub update_inittab {
    open (my $inittab, '<', $INITTAB)
	or die "Can't open $INITTAB: $!";

    open (my $tmp, '>', $TMPTAB)
	or die "Can't open $TMPTAB: $!";

    # Clone original inittab but remove all references to serial lines
    print {$tmp} grep { ! /^T/ } <$inittab>;
    close $inittab;

    my $config = new Vyatta::Config;
    $config->setLevel("system console device");

    my $id = 0;
    foreach my $tty ($config->listNodes()) {
	my $speed = $config->returnValue("$tty speed");
	$speed = 9600 unless $speed;

	print {$tmp} "T$id:23:respawn:";
	if ($config->exists("$tty modem")) {
	    printf {$tmp} "/sbin/mgetty -x0 -s %d %s\n", $speed, $tty;
	} else {
	    printf {$tmp} "/sbin/getty -L %s %d vt100\n", $tty, $speed;
	}

	# Limit to 0-9 because of limitation of last char in id field
	if (++$id >= 10) {
	    warn "Ignoring $tty only 10 serial devices supported\n";
	    last;
	}
    }
    close $tmp;

    if (update($INITTAB, $TMPTAB)) {
	# This is same as telinit q - it tells init to re-examine inittab
	kill 1, 1;
    }
}

my $GRUBCFG = "/boot/grub/grub.cfg";
my $GRUBTMP = "/tmp/grub.cfg.$$";

# For existing serial line change speed (if necessary)
# Only applys to ttyS0
sub update_grub {
    return unless (-f $GRUBCFG);

    my $config = new Vyatta::Config;
    $config->setLevel("system console device");
    return unless $config->exists("ttyS0");

    my $speed = $config->returnValue("ttyS0 speed");
    $speed = "9600" unless defined($speed);

    open (my $grub, '<', $GRUBCFG)
	or die "Can't open $GRUBCFG: $!";

    open (my $tmp, '>', $GRUBTMP)
	or die "Can't open $GRUBTMP: $!";

    while (<$grub>) {
	if (/^serial / ) {
	    print {$tmp} "serial --unit=0 --speed=$speed\n";
	} elsif (/^(.* console=ttyS0),[0-9]+ (.*)$/) {
	    print {$tmp} "$1,$speed $2\n";
	} else {
	    print {$tmp} $_;
	}
    }
    close $grub;
    close $tmp;

    update($GRUBCFG, $GRUBTMP);
}

update_inittab;
update_grub;

exit 0;