#!/usr/bin/perl

# Drakwizard

# Copyright (C) 2002, 2003 Mandrakesoft
#
# Authors: Arnaud Desmons <adesmons@mandrakesoft.com>
#          Florent Villard <warly@mandrakesoft.com>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2, or (at your option)
# any later version.
#
# 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.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.

package MDK::Wizard::Dhcp;
use strict;

use common;
use services;
use MDK::Wizard::Varspaceval;
use MDK::Wizard::Wizcommon;

my $wiz = new MDK::Wizard::Wizcommon;

my $wiz_ip_server = $wiz->{net}->itf_get("IPADDR");
my $wiz_tftpserverip = $wiz_ip_server;

my $o = {
	 name => N("DHCP Wizard"),
	 var => {
		 ip1 => '',
		 ip2 => '',
		 pxe => '1',
		 gateway => '',
		 wiz_authoritative => 0,
		 interface => $wiz->{net}->default_itf
		},
	 needed_rpm => [ 'dhcp-server' ],
	 defaultimage => "/usr/share/mcc/themes/default/dhcp_server-mdk.png",
	};

$o->{pages} = {
	       welcome => {
			   name => N("DHCP Wizard") . "\n\n\n" . N("DHCP is a service that automatically assigns networking addresses to your workstations.") . "\n\n\n" . N("This wizard will help you configuring the DHCP services of your server."),
			   no_back => 1,
			   next => 'interface', no_back => 1,
			  },
	       interface => {
			     name => N("Interface the dhcp server must listen to"),
			     pre => sub {
			       if (!$wiz_ip_server) {
				 my $interface = 'eth0';
				 ($wiz_ip_server) = `/sbin/ip addr show dev $interface` =~ /^\s*inet\s+(\d+\.\d+\.\d+\.\d+)/m;
			       }
			       $o->{var}{interface} |= $wiz->{net}->default_itf;
			     },
			     data => [
				      { list => [ keys %{$wiz->{net}{itf}} ],  val => \$o->{var}{interface} },
				     ],
			     next => 'ip_range',
			     no_back => 1,
			    },
	       ip_range => {
			    name => N("Range of addresses used by DHCP") . "\n" . N("Select the range of addresses assigned to the workstations by the DHCP service; unless you have special needs, you can safely accept the proposed values. (ie: 192.168.100.20 192.168.100.40)") . "\n\n" . N("If you want to enable PXE in your dhcp server please check the box (Pre-boot eXecution Environment, a protocol that allows computers to boot through the network)."),
			    pre => sub {
			      #($wiz_ip_server) = `/sbin/ip addr show dev $o->{var}{interface}` =~ /^\s*inet\s+(\d+\.\d+\.\d+\.\d+)/m;
			      ($wiz_ip_server) = $wiz->{net}->{itf}{$o->{var}{interface}}{IPADDR};
			      $wiz_tftpserverip = $wiz_ip_server;
			      my $d = $4 if $wiz_ip_server =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
			      my $s = "$1.$2.$3" if $wiz_ip_server =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
			      $o->{var}{ip1} = compute_range($d, $s);
			      $o->{var}{ip2} = compute_range2($d, $s);
			    },
			    data => [
				     { label => N("Lowest IP address:"),  val => \$o->{var}{ip1} },
				     { label => N("Highest IP address:"), val => \$o->{var}{ip2} },
				     { label => N("Gateway IP address:"), val => \$o->{var}{gateway} },
				     { label => N("Enable PXE:"), type => 'bool', val => \$o->{var}{pxe} },
				    ],
			    next => 'summary',
			    no_back => 1,
			   },
	       dhcp_warning => {
		name => N("Warning") . "\n\n" . N("You are in dhcp, server may not work with your configuration."),
		ignore => 1,
	        next => 'ip_range',
	      },
	      ip_range_error => {
		name => N("Error") . "\n\n" . N("The IP range specified is not correct."),
		ignore => 1,
	        next => 'ip_range',
	      },
	      ip_range_warning => {
		name => N("Warning") . "\n\n" . N("The IP range specified is not in server address range."),
		ignore => 1,
	        next => 'summary',
	      },
	      server_in_range => {
		name => N("Error") . "\n\n" . N("The IP of the server must not be in range."),
		ignore => 1,
	        next => 'ip_range',
	      },
	      summary => {
			  name => N("Configuring the DHCP server") . "\n\n" . N("The wizard collected the following parameters needed to configure your DHCP service:"),
			  pre => sub {
			    $o->{var}{pxeornot} = $o->{var}{pxe} ? N("enabled") : N("disabled");
			  },
			  data => [
				   { label => N("Lowest IP address:"), val_ref => \$o->{var}{ip1} },
				   { label => N("Highest IP address:"), val_ref => \$o->{var}{ip2} },
				   { label => N("Gateway IP address:"), val_ref => \$o->{var}{gateway} },
				   { label => N("Interface:"), val_ref => \$o->{var}{interface} },
				   { label => N("Enable PXE:"), val_ref => \$o->{var}{pxeornot} },
				  ],
			  post => \&do_it,
			  next => 'end',
			 },
	       end => {
		       name => N("Congratulations") . "\n\n" . N("The wizard successfully configured the DHCP services."),
		       end => 1,
		       no_back => 1,
		      },
	       error_end => {
			     name => N("Failed"),
                             data => [ { label => N("Relaunch drakwizard, and try to change some parameters.") } ],
                             no_back => 1,
                             end => 1,
                            },
	      };



sub compute_range {
  my ($d, $s) = @_;
    my $n;
    if ($d <= 64) { $n = "65" }
    elsif ($d <= 128) { $n = "129" }
    else { $n = "1" }
    "$s.$n";
}

sub compute_range2 {
  my ($d, $s) = @_;
    my $n;
    if ($d <= 128) { $n = "254" }
    elsif ($d > 192) { $n = "192" }
    else { $n = "128" }
    "$s.$n";
}

sub check {
    my $check_ip = sub { return $_[0] < 0 || $_[0] > 255 ? 0 : 1 };
    my $r1_trunc = "$1.$2.$3" if $o->{var}{ip1} =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    foreach ($1, $2, $3, $4) { $check_ip->($_) or return 'ip_range_error' }
    my $r2_trunc = "$1.$2.$3" if $o->{var}{ip2} =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    foreach ($1, $2, $3, $4) { $check_ip->($_) or return 'ip_range_error' }
    my $d1 = $4 if $o->{var}{ip1} =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    my $d2 = $4 if $o->{var}{ip2} =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    my $s_trunc = "$1.$2.$3" if $wiz_ip_server =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    my $ds = $4 if $wiz_ip_server =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    if (!$r1_trunc) { standalone::explanations("DHCP wizard : incorrect adress range 1"); return 'ip_range_error' }
    if (!$r2_trunc) { standalone::explanations("DHCP wizard : incorrect adress range 2"); return 'ip_range_error' }
    if ($r1_trunc ne $s_trunc || $r2_trunc ne $s_trunc) {
	standalone::explanations("DHCP wizard : range not in network");
	return 'ip_range_warning';
    }
    if (!$d1 || !$d2 || $d1 > $d2) { standalone::explanations("DHCP wizard : bad range"); return 'ip_range_error' }
    if ($ds >= $d1 && $ds <= $d2) { standalone::explanations("DHCP wizard : server in range"); return 'server_in_range' }
    return 'interface' if keys %{$wiz->{net}{itf}} > 1;
    0
}

sub do_it {
    $::testing and return;
    my $in = 'interactive'->vnew('su', 'DHCP');
    check_starts_on_boot($in, 'dhcpd');
    my $wiz_domain_name = $wiz->{net}->network_get("DOMAINNAME");
    my $wiz_host_name = $wiz->{net}->network_get("HOSTNAME");
    my $wiz_dns = $wiz->{net}->network_get("dnsServer");
    if ($wiz_dns eq '127.0.0.1') {
    	$wiz_dns = $wiz_ip_server
    }
    if (!$o->{var}{gateway}) {
        my $t = `LC_ALL=C /sbin/ip route list scope global`;
	($o->{var}{gateway}) = $t =~ /default via (\d{1,3}\.\d{1,3}\.\d{1,3}\.\d{1,3}) dev/;
    }
    my $wiz_ip_net = "$1.$2.$3.0" if $wiz_ip_server =~ /(\d{1,3})\.(\d{1,3})\.(\d{1,3})\.(\d{1,3})/;
    my $err = check();
    return $err if $err != 0;
    my $wiz_ip_range1 = $o->{var}{ip1};
    my $wiz_ip_range2 = $o->{var}{ip2};
    my $wiz_ip_netmask = $wiz->{net}->itf_get("NETMASK");
    if (!$wiz_ip_netmask) { $wiz_ip_netmask = "255.255.255.0" }
    my $wiz_device = $o->{var}{interface};
# patch to rewrite when got new file about dhcp with INTERFACES value
# currently, I put the device to configure as dhcp server
# in /etc/sysconfig/dhcpd

#[ -f /etc/sysconfig/dhcpd ] && cp -f /etc/sysconfig/dhcpd /var/tmp/wiz_bck/orig/dhcpd
    my $file = "/etc/sysconfig/dhcpd";
    -f $file and MDK::Common::cp_af($file, $file.".orig");
    standalone::explanations("now patching etc/sysconfig/dhcpd");
    if (!`grep INTERFACES $file`) {
	my  $tmp = `/bin/mktemp /tmp/Dhcpconf.XXXXXX`	or die "can't make a temp file: $!";
	open(NEW, "> $tmp")				or die "can't open $tmp: $!";
	open(OLD, "< $file")				or die "can't open default: $!";
	while (<OLD>) {
	    if (m|INTERFACE|) {
		print NEW "# $_";
	    }
	    print NEW $_;
	}
	MDK::Common::append_to_file($tmp, "\n# Added by drakwizard\nINTERFACES=$wiz_device");
	close(OLD);
	close(NEW);
	chomp($tmp);
	system("mv $tmp $file");
    }
    $file = "/etc/dhcpd.conf";
    -f $file and MDK::Common::cp_af($file, $file.".orig");
    output($file, map {
	s|__hname__|$wiz_host_name|g;
	s|__dns__|$wiz_dns|g;
	s|__net__|$wiz_ip_net|g;
	s|__ip__|$wiz_ip_server|g;
	s|__mask__|$wiz_ip_netmask|g;
	s|__rng1__|$wiz_ip_range1|g;
	s|__rng2__|$wiz_ip_range2|g;
	s|__dname__|$wiz_domain_name|g;
	s|__gateway__|$o->{var}{gateway}|g;
	s|__tftpserverip__|$wiz_tftpserverip|g;
	s|__dhcpd_interface__|$wiz_device|g;
	$_;
    } cat_("/usr/share/wizards//dhcp_wizard/scripts/dhcpd.conf.default"));
    standalone::explanations("$file: hname = $wiz_host_name, net = $wiz_ip_net, ip = $wiz_ip_server,
mask = $wiz_ip_netmask, rng1 = $wiz_ip_range1, rng2 = $wiz_ip_range2, dname = $wiz_domain_name");
    MDK::Common::touch("/var/dhcpd/dhcpd.leases");
# modifying webmin config
    $file="/etc/webmin/dhcpd/config";
    if (-f $file) {
	my %mdk = MDK::Wizard::Varspaceval->get($file);
	$mdk{lease_file} = "/var/dhcpd/dhcpd.leases";
	$mdk{interfaces} = $wiz_device;
	standalone::explanations("$file: lease_file = $mdk{lease_file}, interfaces = $mdk{interfaces}");
	MDK::Wizard::Varspaceval->commit($file, \%mdk);
	!$o->{var}{wiz_authoritative} and output($file, map {
	    s|^\s*not\s*authoritative.*|#$&|i;
	    $_
        } cat_("/etc/dhcpd.conf"));
    }

    if ($o->{var}{pxe} == '0') {
      substInFile {
	s/#\s+deny members of "PXE"/deny members of "PXE"/;
      } "/etc/dhcpd.conf";
    }

    $o->{var}{gateway} or substInFile {
      s/#\s+doption routers.*//;
    } "/etc/dhcpd.conf";

    if (services::is_service_running('dhcpd')) {
	services::restart('dhcpd')
    } else {
        services::start('dhcpd')
    }
    10;
    check_started('dhcpd');
}

sub new {
    my ($class) = @_;
    bless $o, $class;
}

1;