minetestbot-modules/serverup.py

84 lines
2.9 KiB
Python
Executable File

#!/usr/bin/env python
"""
serverup.py - Minetest-Server Ping Module
Copyright 2012, sfan5
"""
import socket, time
def serverup(phenny, input):
arg = input.group(2)
if not arg:
return phenny.reply("Give me a Server Address")
if not '.' in arg:
return phenny.reply("Invalid Address")
if ':' in arg:
return phenny.reply("Syntax changed, please use 'example.org 1337' instead of 'example.org:1337'")
if ' ' in arg:
address = arg.split(' ')[0]
port = arg.split(' ')[1]
if '-' in port or ',' in port:
ports = []
ports_ = port.split(',')
for p in ports_:
if '-' in p:
if len(p.split('-')) != 2:
return phenny.reply("Invalid Port List")
else:
try:
a = int(p.split('-')[0])
except:
return phenny.reply("Invalid Port: %s" % p.split('-')[0])
try:
b = int(p.split('-')[1]) + 1
except:
return phenny.reply("Invalid Port: %s" % p.split('-')[1])
for i in range(a, b):
ports.append(i)
else:
try:
ports.append(int(p))
except:
return phenny.reply("Invalid Port: %s" % p)
else:
try:
ports = [int(port)]
except:
return phenny.reply("Invalid Port: %s" % port)
else:
address = arg
ports = [30000]
if len(ports) != 1 and input.sender.startswith('#') and not (input.admin or input.owner):
return phenny.reply("To check multiple Ports please use Query")
if len(ports) > 6 and not (input.admin or input.owner): # Owner and Admins of the Bot can bypass the Limit
return phenny.reply("Too many Ports specified")
for port in ports:
repres = address + ':' + str(port)
try:
start = time.time()
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
sock.settimeout(2.5)
buf = "\x4f\x45\x74\x03\x00\x00\x00\x01"
sock.sendto(buf, (address, port))
data, addr = sock.recvfrom(1000)
if data:
peer_id = data[12:14]
buf = "\x4f\x45\x74\x03" + peer_id + "\x00\x00\x03"
sock.sendto(buf, (address, port))
sock.close()
end = time.time()
t = (end-start)*1000
phenny.say("%s is up (%dms)" % (repres,t))
else:
phenny.say("%s seems to be down " % repres)
except:
phenny.say("%s seems to be down " % repres)
serverup.commands = ['up']
serverup.thread = True
if __name__ == '__main__':
print(__doc__)