def idToCountry(self, countryId):
"""\
Convert country id to country representation.
"""
if 'code3' == self.cf_country_print_mode:
return GeoIP.id_to_country_code3(countryId)
elif 'name' == self.cf_country_print_mode:
return GeoIP.id_to_country_name(countryId)
else: # 'code' (default)
return GeoIP.id_to_country_code(countryId)
def onStartup(self):
"""\
Create a new GeoIP object and register callback functions.
:rtype : object
"""
# get the admin plugin so we can register commands, issue kicks and such
self._adminPlugin = self.console.getPlugin('admin')
if not self._adminPlugin:
# something is wrong, can't start without admin plugin
self.error('Could not find admin plugin')
return False
# register our commands
if 'commands' in self.config.sections():
for cmd in self.config.options('commands'):
level = self.config.get('commands', cmd)
sp = cmd.split('-')
alias = None
if len(sp) == 2:
cmd, alias = sp
func = self.getCmd(cmd)
if func:
self._adminPlugin.registerCommand(self, cmd, level, func, alias)
self.gi = GeoIP.open(self.cf_geoipdat_path, GeoIP.GEOIP_STANDARD)
if self.console.gameName in self._frostBiteGameNames:
self.registerEvent(b3.events.EVT_PUNKBUSTER_NEW_CONNECTION)
else:
self.registerEvent(b3.events.EVT_CLIENT_AUTH)
self.debug('Started')
def onPlayerConnect(self, client):
"""\
Examine players country and allow/deny to connect.
"""
self.debug(
'Checking player: %s, name: %s, ip: %s, level: %s' % (client.cid, client.name, client.ip, client.maxLevel))
countryId = self.gi.id_by_addr(str(client.ip))
countryCode = GeoIP.id_to_country_code(countryId)
country = self.idToCountry(countryId)
self.debug('Country: %s' % country)
if self.isAllowConnect(countryCode, client):
if (0 < len(self.cf_allow_message) and (
not self.isMessageExcludeFrom(countryCode))) and client.guid and self.console.upTime() > 300:
message = self.getMessage('cf_allow_message', {'name': client.name, 'country': country})
self.console.say(message)
pass # do nothing
else:
if 0 < len(self.cf_deny_message) and (not self.isMessageExcludeFrom(countryCode)):
message = self.getMessage('cf_deny_message', {'name': client.name, 'country': country})
self.console.say(message)
client.kick(silent=True)
self.debug('Checking done.')
def cmd_cfcountry(self, data, client, cmd=None):
"""\
<player> - What country is this player from?
"""
# this will split the player name and the message (oops, no message...)
_input = self._adminPlugin.parseUserCmd(data)
if _input:
# input[0] is the player id
sclient = self._adminPlugin.findClientPrompt(_input[0], client)
if not sclient:
# a player matchin the name was not found, a list of closest matches will be displayed
# we can exit here and the user will retry with a more specific player
return False
else:
client.message('^7Invalid or missing data, try !help cfcountry')
return False
# are we still here? Let's get the country
countryId = self.gi.id_by_addr(str(sclient.ip))
countryCode = GeoIP.id_to_country_code(countryId)
country = self.idToCountry(countryId)
cmd.sayLoudOrPM(client, '^1%s^7 is connecting from ^1%s (%s)^7' % (sclient.name, country, countryCode))
return True
