82 lines
2.9 KiB
Python
Executable File
82 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python
|
|
"""
|
|
serverup.py - Minetest server ping module
|
|
Copyright 2014, sfan5
|
|
Licensed under GNU General Public License v2.0
|
|
"""
|
|
|
|
import socket, time
|
|
|
|
def serverup(phenny, input):
|
|
arg = input.group(2)
|
|
if not arg:
|
|
return phenny.reply("give me an address (and port if you want)")
|
|
if not '.' in arg:
|
|
return phenny.reply("invalid address")
|
|
if ':' in arg:
|
|
return phenny.reply("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 private chat")
|
|
if len(ports) > 6 and not (input.admin or input.owner): # owner and admins of the bot can bypass the limit
|
|
return phenny.reply("ow, too many ports!")
|
|
for port in ports:
|
|
repres = address + ':' + str(port)
|
|
try:
|
|
sock = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
|
|
sock.settimeout(2.5)
|
|
buf = b"\x4f\x45\x74\x03\x00\x00\x00\x01"
|
|
sock.sendto(buf, (address, port))
|
|
start = time.time()
|
|
data, addr = sock.recvfrom(1000)
|
|
if data:
|
|
end = time.time()
|
|
peer_id = data[12:14]
|
|
buf = b"\x4f\x45\x74\x03" + peer_id + b"\x00\x00\x03"
|
|
sock.sendto(buf, (address, port))
|
|
sock.close()
|
|
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']
|
|
|
|
if __name__ == '__main__':
|
|
print(__doc__)
|