]> git.donarmstrong.com Git - infobot.git/blob - src/Modules/BZFlag.pl
use read to get buffering
[infobot.git] / src / Modules / BZFlag.pl
1 #!/usr/bin/perl
2 #
3 # BZFlag
4 # Copyright (c) 1993 - 2002 Tim Riker
5 #
6 # This package is free software;  you can redistribute it and/or
7 # modify it under the terms of the license found in the file
8 # named LICENSE that should have accompanied this file.
9 #
10 # THIS PACKAGE IS PROVIDED ``AS IS'' AND WITHOUT ANY EXPRESS OR
11 # IMPLIED WARRANTIES, INCLUDING, WITHOUT LIMITATION, THE IMPLIED
12 # WARRANTIES OF MERCHANTIBILITY AND FITNESS FOR A PARTICULAR PURPOSE.
13
14 package BZFlag;
15 use strict;
16
17 my $no_BZFlag;
18
19 BEGIN {
20         $no_BZFlag = 0;
21         eval "use Socket";
22         eval "use LWP::UserAgent";
23         $no_BZFlag++ if ($@);
24 }
25
26 sub BZFlag::BZFlag {
27         my ($message) = @_;
28   my ($retval);
29         if ($no_BZFlag) {
30                 &main::status("BZFlag module requires Socket.");
31                 return 'BZFlag module not active';
32         }
33         if ($message =~ /^bzfquery\s+([^:]*)(?::([0-9]*))?$/xi) {
34                 $retval = &query($1,$2);
35         } elsif ($message =~ /^bzflist$/xi) {
36                 $retval = &list();
37   } else {
38                 $retval = "BZFlag: unhandled command \"$message\"";
39         }
40         &::performStrictReply($retval);
41 }
42
43 sub BZFlag::list {
44         my ($response);
45         my $ua = new LWP::UserAgent;
46
47         $ua->timeout(5);
48
49         my $req = HTTP::Request->new('GET', 'http://list.bzflag.org:5156/');
50         my $res = $ua->request($req);
51   my %servers;
52   my $totalServers = 0;
53   my $totalPlayers = 0;
54         for my $line (split("\n",$res->content)) {
55                 my ($serverport, $version, $flags, $ip, $comments) = split(" ",$line,5);
56                 # not "(A4)18" to handle old dumb perl
57                 my ($style,$maxPlayers,$maxShots,
58                                 $rogueSize,$redSize,$greenSize,$blueSize,$purpleSize,
59                                 $rogueMax,$redMax,$greenMax,$blueMax,$purpleMax,
60                                 $shakeWins,$shakeTimeout,
61                                 $maxPlayerScore,$maxTeamScore,$maxTime) =
62                                 unpack("A4A4A4A4A4A4A4A4A4A4A4A4A4A4A4A4A4A4", $flags);
63                 my $playerSize = hex($rogueSize) + hex($redSize) + hex($greenSize)
64                                 + hex($blueSize) + hex($purpleSize);
65           $servers{$serverport} = $playerSize;
66           $totalServers += 1;
67           $totalPlayers += $playerSize;
68         }
69   $response .= "s=$totalServers p=$totalPlayers";
70   foreach my $key (sort {$servers{$b} <=> $servers{$a}} (keys(%servers))) {
71                 if ($servers{$key} > 0) {
72                         $response .= " $key($servers{$key})";
73                 }
74   }
75         &::performStrictReply($response);
76         return;
77 }
78
79 sub BZFlag::querytext {
80         my ($servernameport) = @_;
81         my ($servername,$port) = split(":",$servernameport);
82         if ($no_BZFlag) {
83                 &main::status("BZFlag module requires Socket.");
84                 return 'BZFlag module not active';
85         }
86         #my @teamName = ("Rogue", "Red", "Green", "Blue", "Purple");
87         my @teamName = ("X", "R", "G", "B", "P");
88         my ($message, $server, $response);
89         $port = 5155 unless $port;
90
91         # socket define
92         my $sockaddr = 'S n a4 x8';
93
94         # port to port number
95         my ($name,$aliases,$proto) = getprotobyname('tcp');
96         ($name,$aliases,$port)  = getservbyname($port,'tcp') unless $port =~ /^\d+$/;
97
98         # get server address
99         my ($type,$len,$serveraddr);
100         ($name,$aliases,$type,$len,$serveraddr) = gethostbyname($servername);
101         $server = pack($sockaddr, AF_INET, $port, $serveraddr);
102
103         # connect
104         return 'socket() error' unless socket(S1, AF_INET, SOCK_STREAM, $proto);
105         return "could not connect to $servername:$port" unless connect(S1, $server);
106
107         # don't buffer
108         select(S1); $| = 1; select(STDOUT);
109
110         # get hello
111         my $buffer;
112         return 'read error' unless read(S1, $buffer, 10) == 10;
113
114         # parse reply
115         my ($magic,$major,$minor,$revision);
116         ($magic,$major,$minor,$revision,$port) = unpack("a4 a1 a2 a1 n", $buffer);
117
118         # quit if version isn't valid
119         return 'not a bzflag server' if ($magic ne "BZFS");
120         return 'incompatible version' if ($major < 1);
121         return 'incompatible version' if ($major == 1 && $minor < 7);
122         return 'incompatible version' if ($major == 1 && $minor == 7 && $revision eq "b");
123
124         # quit if rejected
125         return 'rejected by server' if ($port == 0);
126
127         # reconnect on new port
128         $server = pack($sockaddr, AF_INET, $port, $serveraddr);
129         return 'socket() error on reconnect' unless socket(S, AF_INET, SOCK_STREAM, $proto);
130         return "could not reconnect to $servername:$port" unless connect(S, $server);
131         select(S); $| = 1; select(STDOUT);
132
133         # close first socket
134         close(S1);
135
136         # send game request
137         print S pack("n2", 0, 0x7167);
138
139         # get reply
140         return 'server read error' unless read(S, $buffer, 40) == 40;
141         my ($infolen,$infocode,$style,$maxPlayers,$maxShots,
142                 $rogueSize,$redSize,$greenSize,$blueSize,$purpleSize,
143                 $rogueMax,$redMax,$greenMax,$blueMax,$purpleMax,
144                 $shakeWins,$shakeTimeout,
145                 $maxPlayerScore,$maxTeamScore,$maxTime) = unpack("n20", $buffer);
146         return 'bad server data' unless $infocode == 0x7167;
147
148         # send players request
149         print S pack("n2", 0, 0x7170);
150
151         # get number of teams and players we'll be receiving
152         return 'count read error' unless read(S, $buffer, 8) == 8;
153         my ($countlen,$countcode,$numTeams,$numPlayers) = unpack("n4", $buffer);
154         return 'bad count data' unless $countcode == 0x7170;
155
156         # get the teams
157         for (1..$numTeams) {
158                 return 'team read error' unless read(S, $buffer, 14) == 14;
159                 my ($teamlen,$teamcode,$team,$size,$aSize,$won,$lost) = unpack("n7", $buffer);
160                 return 'bad team data' unless $teamcode == 0x7475;
161                 if ($size > 0) {
162                         my $score = $won - $lost;
163                         $response .= "$teamName[$team]:$score($won-$lost) ";
164                 }
165         }
166
167         # get the players
168         for (1..$numPlayers) {
169                 last unless read(S, $buffer, 180) == 180;
170                 my ($playerlen,$playercode,$pAddr,$pPort,$pNum,$type,$team,$won,$lost,$sign,$email) =
171                                 unpack("n2Nn2 n4A32A128", $buffer);
172                 return 'bad player data' unless $playercode == 0x6170;
173                 my $score = $won - $lost;
174                 $response .= " $sign($teamName[$team]";
175                 $response .= ":$email" if ($email);
176                 $response .= ")$score($won-$lost)";
177         }
178         $response .= "No Players" if ($numPlayers <= 1);
179
180         # close socket
181         close(S);
182
183         return $response;
184 }
185
186 sub BZFlag::query {
187         my ($servernameport) = @_;
188         &::performStrictReply(&querytext($servernameport));
189   return;
190 }
191
192 1;
193 # vim: ts=2 sw=2