[Perl] Eigen WHOIS-server bouwen - Softwareontwikkeling

donderdag 24 juli 2003 23:18

Acties:

Topicstarter

Een dik jaar geleden voor intern gebruik een whois-servertje geschreven in Perl (code: zie verder).
Uiteindelijk is het ding nooit in gebruik geraakt, maar nu willen we het toch gaan gebruiken. Probleem is: start je dit vanuit Linux shell, dan stopt de server als je uitlogt...

De vraag is nu: hoe krijg je dit als lopend proces uit de shell, en hoe herstart je bij fouten ?
D.m.v. wat zoeken op Google ben ik op het "forken" van processen gekomen, maar daar loop ik vast... Iemand die me op het juiste pad kan zetten ?

De huidige code:

code:

#!/usr/bin/perl

#########################################################################
## Settings
$bufferlen=20000;   ## Buffer length
$password="xxxx";   ## Server shutdown password

$dbname = "xxxxx"; # the database's name
$dbusername = "xxxxx"; # the database's username
$dbpassword = "xxxxx"; # the database's password

#########################################################################
use IO::Socket;
use IO::Select;
use DBI;

$SIG{PIPE}='IGNORE';
$serversock=new IO::Socket::INET(Listen=>1,LocalPort=>43);
$childsock=new IO::Select($serversock);
$/="\0";

#########################################################################
## Connect to the database
$dbh = DBI->connect("DBI:mysql:database=$dbname", "$dbusername", "$dbpassword") || die "Error: Couldn't open database !\n";

#########################################################################
## Start the server
&serve;

$bytestransferred=0;
$numrequests=0;

#########################################################################
## Server
sub serve {
    %in_str = ();   ## Incoming data
    %in_sock = ();  ## Incoming sockets
    %ack = ();  ## Incoming acks
    $num_sock = -1; ## Socket count
    $stat_started = localtime;  ## Server start time

    print "Server started ($stat_started)\n";

    ## Keep switching sockets to listen to
    while(@sock=$childsock->can_read) {
        foreach(@sock) {
            ## New connection
            if($_==$serversock) {
                $num_sock++;

                $newclient=$serversock->accept;
                $childsock->add($newclient);

                $in_str{$newclient} = "";
                $in_sock{$newclient} = $num_sock;
            } else {
                ## Get incoming data
                my $R=sysread($_, $i, $bufferlen);

                ## Incoming socket still active ?
                ## If not -> disconnect
                if ($R==0) {
                    $T=syswrite($_, '', $bufferlen);
                    if ($T==undef) {
                        &disconnect_sock($_);
                    }
                } else {
                    ## Add incoming data to socket string
                    if (length($i)<=2) {
                        if (($i ne "\n") && ($i ne "\r") && ($i ne "\r\n") && ($i ne "\n\r")) {
                            $in_str{$_} = $in_str{$_}."$i";
                        } else {
                            if ($in_str{$_} ne "") {
                                &parse_in($_,$in_str{$_});
                                $in_str{$_}="";
                            }
                        }
                    } else {
                        &parse_in($_,$i);
                        $in_str{$_}="";
                    }
                }
            }
        }
    }
}


#########################################################################
## Send message to socket
sub send {
    my($sock,$message) = @_;

    $bytestransferred += length($message);

    if ($message !~ /\0$/) {
        $message.="\0";
    }
    syswrite($sock, $message,$bufferlen);
}

#########################################################################
## Broadcast message
sub broadcast {
    my($message) = @_;
    foreach $C ($childsock->handles) {
        &send($C,$message);
    }
}

#########################################################################
## Parser
sub parse_in {
    my($sock,$data) = @_;

    $bytestransferred += length($data);

    ## Action to do:
    if ($data =~ /^exit/i) {        ## Disconnect client
        &disconnect_sock($sock);
    } elsif ($data =~ /^stop $password/i) { ## Stop server
        &stop_server();
    } else {
        &whois($sock,$data);
        &disconnect_sock($sock);
    }
}

#########################################################################
## Disconnect
sub disconnect_sock {
    my($sock) = @_;

    $childsock->remove($sock);
    $in_sock{$sock} = undef;
}

#########################################################################
## Stop server
sub stop_server {
    &broadcast("Server stopped");
    $stat_stop = localtime;
    print "Server stopped ($stat_stop)\n";
    print "Transferred $bytestransferred bytes\n";
    print "$numrequests requests served\n";

    # Close the DB connection
    $dbh->disconnect || die "Error: Couldn't close database !\n";

    exit(1);
}

#########################################################################
## Whois
sub whois {
    my($sock,$domain) = @_;

    $numrequests++;

    &send($sock,"WHOIS v1.0");
    &send($sock,"");

    $query = "
        SELECT `domain`
        FROM `whois`
        WHERE `domain` = '$domain'
    ";
    
    my $sth=$dbh->prepare($query); # prepare the database for the SQL query
    $sth->execute; # execute the SQL statement

    my ($db_domain) = $sth->fetchrow_array;

    if ($db_domain eq $domain) {
        $status = "REGISTERED";
    } else {
        $status = "FREE";
    }

    &send($sock,"$domain: $status");
}

donderdag 24 juli 2003 23:33

Acties:

Soultaker

maartenba schreef op 24 July 2003 @ 23:18:
Probleem is: start je dit vanuit Linux shell, dan stopt de server als je uitlogt...

Niet als je 'm met nohup start. (Zie: "man nohup")

De vraag is nu: hoe krijg je dit als lopend proces uit de shell, en hoe herstart je bij fouten?

In principe heet dat "daemonizen". Google maar eens op Perl en daemonize. In feite ben je met een simpele fork al klaar, al is 't in Linux gebruikelijk om dubbel te forken (ik kan de preciese procedure en redenen wel nazoeken, als je geïnteresseerd bent), om bepaalde afhankelijkheden van de terminal kwijt te raken. Ik weet niet meer exact wat het nut was.

Wat betreft het levend houden van je programma, kun je twee processen laten draaien. Een proces start een child proces dat het echte werk doet en als de child afsluit, start de parent proces een nieuwe child op. Kijk voor zulke functionaliteit naar de SIGCHLD signal en de Perl signal manual pages ("man perlsig").

Overigens hoef je dit allemaal niet zelf te schrijven. Je kunt ook bestaande utillities gebruiken, zoals bijvoorbeeld inetd ("man inetd"), die specifiek toegespitst is op het runnen van internet services, of iets als Dan Bernstein's supervise utillity uit de daemontools collectie.

edit:
Je merkt wel dat de moraal van dit verhaal is dat je heel veel informatie uit Google en de manual pages van de standaard UNIX (of Linux) utillities kunt halen!

[ Voor 11% gewijzigd door Soultaker op 24-07-2003 23:35 ]

donderdag 24 juli 2003 23:36

Acties:

maartenba

Topicstarter

Soultaker schreef op 24 juli 2003 @ 23:33:
Je merkt wel dat de moraal van dit verhaal is dat je heel veel informatie uit Google en de manual pages van de standaard UNIX (of Linux) utillities kunt halen!

Je moet het aleen weten vinden

Eerst maar eens de inetd-aanpak proberen, deze lijkt me de makkelijkste oplossing, die waarschijnlijk even goed werkt als andere oplossingen...

donderdag 24 juli 2003 23:44

Acties:

Soultaker

maartenba schreef op 24 July 2003 @ 23:36:
Je moet het aleen weten vinden Eerst maar eens de inetd-aanpak proberen, deze lijkt me de makkelijkste oplossing, die waarschijnlijk even goed werkt als andere oplossingen...

Ik zou zelf zeggen dat het gebruik van supervise het makkelijkst is, zeker als je je script niet kunt detachen (daemonizen). Voor het automatisch starten bij het opstarten is inetd natuurlijk weer handig (anders moet je daar een rc-scriptje voor maken).

vrijdag 25 juli 2003 00:05

Acties: