]> git.donarmstrong.com Git - infobot.git/blob - src/Modules/BZFlag.pl
show player totals for network versions
[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 use Fcntl qw(F_GETFL F_SETFL O_NONBLOCK);
17
18 my $no_BZFlag;
19
20 BEGIN {
21         $no_BZFlag = 0;
22         eval "use Socket";
23         eval "use LWP::UserAgent";
24         $no_BZFlag++ if ($@);
25 }
26
27 sub BZFlag {
28         my ($message) = @_;
29         my ($retval);
30         if ($no_BZFlag) {
31                 &main::status("BZFlag module requires Socket.");
32                 return 'BZFlag module not active';
33         }
34         if ($message =~ /^bzfquery\s+([^:]*)(?::([0-9]*))?$/xi) {
35                 $retval = &query($1,$2);
36         } elsif ($message =~ /^bzflist$/xi) {
37                 $retval = &list();
38         } else {
39                 $retval = "BZFlag: unhandled command \"$message\"";
40         }
41         &::performStrictReply($retval);
42 }
43
44 sub list {
45         my ($response);
46         my $ua = new LWP::UserAgent;
47         $ua->proxy('http', $::param{'httpProxy'}) if (&::IsParam("httpProxy"));
48
49         $ua->timeout(5);
50
51         my $req = HTTP::Request->new('GET', 'http://db.bzflag.org/db/?action=LIST');
52         my $res = $ua->request($req);
53         my %servers;
54         my $totalServers = 0;
55         for my $line (split("\n",$res->content)) {
56                 my ($serverport, $version, $flags, $ip, $comments) = split(" ",$line,5);
57                 # not "(A4)18" to handle old dumb perl
58                 my ($style, $maxShots, $shakeWins, $shakeTimeout, $maxPlayerScore, $maxTeamScore, $maxTime,
59                                 $maxPlayers, $rogueSize, $rogueMax, $redSize, $redMax, $greenSize, $greenMax,
60                                 $blueSize, $blueMax, $purpleSize, $purpleMax, $observerSize, $observerMax) =
61                                 unpack("A4A4A4A4A4A4A4A2A2A2A2A2A2A2A2A2A2A2A2A2", $flags);
62                 my $playerSize = hex($rogueSize) + hex($redSize) + hex($greenSize)
63                                 + hex($blueSize) + hex($purpleSize) + hex($observerSize);
64                 $servers{$serverport} = $playerSize;
65           $servers{$version} += $playerSize;
66                 $servers{'PLAYERS'} += $playerSize;
67                 $totalServers += 1;
68         }
69         $response .= "s=$totalServers";
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 list17 {
80         my ($response);
81         my $ua = new LWP::UserAgent;
82         $ua->proxy('http', $::param{'httpProxy'}) if (&::IsParam("httpProxy"));
83
84         $ua->timeout(5);
85
86         my $req = HTTP::Request->new('GET', 'http://list.bzflag.org:5156/');
87         my $res = $ua->request($req);
88         my %servers;
89         my $totalServers = 0;
90         my $totalPlayers = 0;
91         for my $line (split("\n",$res->content)) {
92                 my ($serverport, $version, $flags, $ip, $comments) = split(" ",$line,5);
93                 # not "(A4)18" to handle old dumb perl
94                 my ($style,$maxPlayers,$maxShots,
95                                 $rogueSize,$redSize,$greenSize,$blueSize,$purpleSize,
96                                 $rogueMax,$redMax,$greenMax,$blueMax,$purpleMax,
97                                 $shakeWins,$shakeTimeout,
98                                 $maxPlayerScore,$maxTeamScore,$maxTime) =
99                                 unpack("A4A4A4A4A4A4A4A4A4A4A4A4A4A4A4A4A4A4", $flags);
100                 my $playerSize = hex($rogueSize) + hex($redSize) + hex($greenSize)
101                                 + hex($blueSize) + hex($purpleSize);
102                 $servers{$serverport} = $playerSize;
103                 $totalServers += 1;
104                 $totalPlayers += $playerSize;
105         }
106         $response .= "s=$totalServers p=$totalPlayers";
107         foreach my $key (sort {$servers{$b} <=> $servers{$a}} (keys(%servers))) {
108                 if ($servers{$key} > 0) {
109                         $response .= " $key($servers{$key})";
110                 }
111         }
112         &::performStrictReply($response);
113         return;
114 }
115
116 sub querytext {
117         my ($servernameport) = @_;
118         my ($servername,$port) = split(":",$servernameport);
119         if ($no_BZFlag) {
120                 &main::status("BZFlag module requires Socket.");
121                 return 'BZFlag module not active';
122         }
123         #my @teamName = ("Rogue", "Red", "Green", "Blue", "Purple", "Observer", "Rabbit");
124         my @teamName = ("X", "R", "G", "B", "P", "O", "K");
125         my ($message, $server, $response);
126         $port = 5154 unless $port;
127
128         # socket define
129         my $sockaddr = 'S n a4 x8';
130
131         # port to port number
132         my ($name,$aliases,$proto) = getprotobyname('tcp');
133         ($name,$aliases,$port)  = getservbyname($port,'tcp') unless $port =~ /^\d+$/;
134
135         # get server address
136         my ($type,$len,$serveraddr);
137         ($name,$aliases,$type,$len,$serveraddr) = gethostbyname($servername);
138         $server = pack($sockaddr, AF_INET, $port, $serveraddr);
139
140         # connect
141         # TODO wrap this with a 5 second alarm()
142         return 'socket() error' unless socket(S1, AF_INET, SOCK_STREAM, $proto);
143         return "could not connect to $servername:$port" unless connect(S1, $server);
144
145         # don't buffer
146         select(S1); $| = 1; select(STDOUT);
147
148         # get hello
149         my $buffer;
150         return 'read error' unless read(S1, $buffer, 8) == 8;
151
152         # parse reply
153         my ($magic,$major,$minor,$something,$revision) = unpack("a4 a1 a1 a1 a1", $buffer);
154   my ($version) = $magic . $major . $minor . $something . $revision;
155
156         # quit if version isn't valid
157         return 'not a bzflag server' if ($magic ne "BZFS");
158         # check version
159         if ($version eq "BZFS0026") {
160                 # 1.11.x handled here
161                 return 'read error' unless read(S1, $buffer, 1) == 1;
162                 my ($id) = unpack("C", $buffer);
163     return "rejected by server" if ($id == 255);
164
165                 # send game request
166                 print S1 pack("n2", 0, 0x7167);
167
168                 # get reply
169                 return 'server read error' unless read(S1, $buffer, 46) == 46;
170     my ($infolen,$infocode,$style,$maxPlayers,$maxShots,
171                         $rogueSize,$redSize,$greenSize,$blueSize,$purpleSize,$observerSize,
172                         $rogueMax,$redMax,$greenMax,$blueMax,$purpleMax,$observerMax,
173                         $shakeWins,$shakeTimeout,
174                         $maxPlayerScore,$maxTeamScore,$maxTime,$timeElapsed) = unpack("n23", $buffer);
175                 return "bad server data $infocode" unless $infocode == 0x7167;
176
177                 # send players request
178                 print S1 pack("n2", 0, 0x7170);
179
180                 # get number of teams and players we'll be receiving
181                 return 'count read error' unless read(S1, $buffer, 8) == 8;
182                 my ($countlen,$countcode,$numTeams,$numPlayers) = unpack("n4", $buffer);
183
184                 # get the teams
185                 return 'bad count data' unless $countcode == 0x7170;
186                 return 'count read error' unless read(S1, $buffer, 5) == 5;
187                 ($countlen,$countcode,$numTeams) = unpack("n n C", $buffer);
188                 for (1..$numTeams) {
189                         return 'team read error' unless read(S1, $buffer, 8) == 8;
190                         my ($team,$size,$won,$lost) = unpack("n4", $buffer);
191                         if ($size > 0) {
192                                 my $score = $won - $lost;
193                                 $response .= "$teamName[$team]:$score($won-$lost) ";
194                         }
195                 }
196
197                 # get the players
198                 for (1..$numPlayers) {
199                         last unless read(S1, $buffer, 175) == 175;
200                         my ($playerlen,$playercode,$pID,$type,$team,$won,$lost,$tks,$sign,$email) =
201                                         unpack("n2Cn5A32A128", $buffer);
202                         #my ($playerlen,$playercode,$pAddr,$pPort,$pNum,$type,$team,$won,$lost,$sign,$email) =
203                         #               unpack("n2Nn2 n4A32A128", $buffer);
204                         return 'bad player data' unless $playercode == 0x6170;
205                         my $score = $won - $lost;
206                         $response .= " $sign($teamName[$team]";
207                         $response .= ":$email" if ($email);
208                         $response .= ")$score($won-$lost)";
209                 }
210                 $response .= "No Players" if ($numPlayers < 1);
211
212                 # close socket
213         } elsif ($major == 1 && $minor == 9) {
214                 # 1.10.x handled here
215                 $revision = $something * 10 + $revision;
216                 return 'read error' unless read(S1, $buffer, 1) == 1;
217                 my ($id) = unpack("C", $buffer);
218
219                 # send game request
220                 print S1 pack("n2", 0, 0x7167);
221
222                 # FIXME the packets are wrong from here down
223                 # get reply
224                 return 'server read error' unless read(S1, $buffer, 40) == 40;
225                 my ($infolen,$infocode,$style,$maxPlayers,$maxShots,
226                         $rogueSize,$redSize,$greenSize,$blueSize,$purpleSize,
227                         $rogueMax,$redMax,$greenMax,$blueMax,$purpleMax,
228                         $shakeWins,$shakeTimeout,
229                         $maxPlayerScore,$maxTeamScore,$maxTime) = unpack("n20", $buffer);
230                 return 'bad server data' unless $infocode == 0x7167;
231
232                 # send players request
233                 print S1 pack("n2", 0, 0x7170);
234
235                 # get number of teams and players we'll be receiving
236                 return 'count read error' unless read(S1, $buffer, 8) == 8;
237                 my ($countlen,$countcode,$numTeams,$numPlayers) = unpack("n4", $buffer);
238
239                 # get the teams
240                 return 'bad count data' unless $countcode == 0x7170;
241                 return 'count read error' unless read(S1, $buffer, 5) == 5;
242                 ($countlen,$countcode,$numTeams) = unpack("n n C", $buffer);
243                 for (1..$numTeams) {
244                         return 'team read error' unless read(S1, $buffer, 8) == 8;
245                         my ($team,$size,$won,$lost) = unpack("n4", $buffer);
246                         if ($size > 0) {
247                                 my $score = $won - $lost;
248                                 $response .= "$teamName[$team]:$score($won-$lost) ";
249                         }
250                 }
251
252                 # get the players
253                 for (1..$numPlayers) {
254                         last unless read(S1, $buffer, 175) == 175;
255                         my ($playerlen,$playercode,$pID,$type,$team,$won,$lost,$tks,$sign,$email) =
256                                         unpack("n2Cn5A32A128", $buffer);
257                         #my ($playerlen,$playercode,$pAddr,$pPort,$pNum,$type,$team,$won,$lost,$sign,$email) =
258                         #               unpack("n2Nn2 n4A32A128", $buffer);
259                         return 'bad player data' unless $playercode == 0x6170;
260                         my $score = $won - $lost;
261                         $response .= " $sign($teamName[$team]";
262                         $response .= ":$email" if ($email);
263                         $response .= ")$score($won-$lost)";
264                 }
265                 $response .= "No Players" if ($numPlayers < 1);
266
267                 # close socket
268                 close(S1);
269         } elsif ($major == 1 && $minor == 0 && $something == 7) {
270     # 1.7* versions handled here
271                 # old servers send a reconnect port number
272                 return 'read error' unless read(S1, $buffer, 2) == 2;
273                 my ($reconnect) = unpack("n", $buffer);
274                 $minor = $minor * 10 + $something;
275                 # quit if rejected
276                 return 'rejected by server' if ($reconnect == 0);
277
278                 # reconnect on new port
279                 $server = pack($sockaddr, AF_INET, $reconnect, $serveraddr);
280                 return 'socket() error on reconnect' unless socket(S, AF_INET, SOCK_STREAM, $proto);
281                 return "could not reconnect to $servername:$reconnect" unless connect(S, $server);
282                 select(S); $| = 1; select(STDOUT);
283
284                 # close first socket
285                 close(S1);
286
287                 # send game request
288                 print S pack("n2", 0, 0x7167);
289
290                 # get reply
291                 return 'server read error' unless read(S, $buffer, 40) == 40;
292                 my ($infolen,$infocode,$style,$maxPlayers,$maxShots,
293                         $rogueSize,$redSize,$greenSize,$blueSize,$purpleSize,
294                         $rogueMax,$redMax,$greenMax,$blueMax,$purpleMax,
295                         $shakeWins,$shakeTimeout,
296                         $maxPlayerScore,$maxTeamScore,$maxTime) = unpack("n20", $buffer);
297                 return 'bad server data' unless $infocode == 0x7167;
298
299                 # send players request
300                 print S pack("n2", 0, 0x7170);
301
302                 # get number of teams and players we'll be receiving
303                 return 'count read error' unless read(S, $buffer, 8) == 8;
304                 my ($countlen,$countcode,$numTeams,$numPlayers) = unpack("n4", $buffer);
305                 return 'bad count data' unless $countcode == 0x7170;
306
307                 # get the teams
308                 for (1..$numTeams) {
309                         return 'team read error' unless read(S, $buffer, 14) == 14;
310                         my ($teamlen,$teamcode,$team,$size,$aSize,$won,$lost) = unpack("n7", $buffer);
311                         return 'bad team data' unless $teamcode == 0x7475;
312                         if ($size > 0) {
313                                 my $score = $won - $lost;
314                                 $response .= "$teamName[$team]:$score($won-$lost) ";
315                         }
316                 }
317
318                 # get the players
319                 for (1..$numPlayers) {
320                         last unless read(S, $buffer, 180) == 180;
321                         my ($playerlen,$playercode,$pAddr,$pPort,$pNum,$type,$team,$won,$lost,$sign,$email) =
322                                         unpack("n2Nn2 n4A32A128", $buffer);
323                         return 'bad player data' unless $playercode == 0x6170;
324                         my $score = $won - $lost;
325                         $response .= " $sign($teamName[$team]";
326                         $response .= ":$email" if ($email);
327                         $response .= ")$score($won-$lost)";
328                 }
329                 $response .= "No Players" if ($numPlayers <= 1);
330
331                 # close socket
332                 close(S);
333         } else {
334                 $response = "incompatible version: $version";
335         }
336
337         return $response;
338 }
339
340 sub query {
341         my ($servernameport) = @_;
342         &::performStrictReply(&querytext($servernameport));
343         return;
344 }
345
346 1;
347 # vim: ts=2 sw=2