1
0
mirror of https://github.com/Tafkas/fritzbox-munin.git synced 2023-10-10 13:36:55 +02:00
fritzbox-munin/fritzbox_helper.py

94 lines
3.2 KiB
Python
Raw Normal View History

2015-06-04 22:08:29 +02:00
#!/usr/bin/env python
"""
fritzbox_helper - A munin plugin for Linux to monitor AVM Fritzbox
Copyright (C) 2015 Christian Stade-Schuldt
Author: Christian Stade-Schuldt
Like Munin, this plugin is licensed under the GNU GPL v2 license
http://www.opensource.org/licenses/GPL-2.0
If you've put your gunicorn pid somewhere other than the
default /var/run/gunicorn.pid, you can add a section like
this to your munin-node's plugin configuration:
[fritzbox_*]
env.fritzbox_ip [ip address of the fritzbox]
env.password [fritzbox password]
This plugin supports the following munin configuration parameters:
#%# family=auto contrib
#%# capabilities=autoconf
"""
import hashlib
import httplib
import os
import re
import sys
from xml.dom import minidom
from io import StringIO, BytesIO
USER_AGENT = "Mozilla/5.0 (U; Windows NT 5.1; rv:5.0) Gecko/20100101 Firefox/5.0"
def get_sid(server,password,port=80):
"""Obtains the sid after login into the fritzbox"""
conn = httplib.HTTPConnection(server+':'+str(port))
headers = { "Accept" : "application/xml",
"Content-Type" : "text/plain",
"User-Agent" : USER_AGENT}
initialPage='/login_sid.lua'
conn.request("GET", initialPage, '', headers)
response = conn.getresponse()
data = response.read()
if response.status != 200:
print "%s %s" % (response.status, response.reason)
sys.exit(0)
else:
theXml = minidom.parseString(data)
sidInfo = theXml.getElementsByTagName('SID')
sid=sidInfo[0].firstChild.data
if sid == "0000000000000000":
challengeInfo = theXml.getElementsByTagName('Challenge')
challenge=challengeInfo[0].firstChild.data
challenge_bf = (challenge + '-' + password).decode('iso-8859-1').encode('utf-16le')
m = hashlib.md5()
m.update(challenge_bf)
response_bf = challenge + '-' + m.hexdigest().lower()
else:
return sid
headers = { "Accept" : "text/html,application/xhtml+xml,application/xml",
"Content-Type" : "application/x-www-form-urlencoded",
"User-Agent" : USER_AGENT}
loginPage="/login_sid.lua?&response=" + response_bf
conn.request("GET", loginPage, '', headers)
response = conn.getresponse()
data = response.read()
if response.status != 200:
print "%s %s" % (response.status, response.reason)
sys.exit(0)
else:
sid = re.search('<SID>(.*?)</SID>', data).group(1)
if sid == "0000000000000000":
print "ERROR - No SID received because of invalid password"
sys.exit(0)
return sid
def get_page(server, sid, page, port=80):
"""Fetches a page from the Fritzbox and returns its content"""
conn = httplib.HTTPConnection(server+':'+str(port))
headers = { "Accept" : "application/xml",
"Content-Type" : "text/plain",
"User-Agent" : USER_AGENT}
pageWithSid=page+"?sid="+sid
conn.request("GET", pageWithSid, '', headers)
response = conn.getresponse()
data = response.read()
if response.status != 200:
print "%s %s" % (response.status, response.reason)
print data
sys.exit(0)
else:
return data