From ed7050667ed5da4b245d77020ddd4b3981936c7f Mon Sep 17 00:00:00 2001
From: Christian Stade-Schuldt <tafkasorg@yahoo.de>
Date: Thu, 4 Jun 2015 22:08:29 +0200
Subject: [PATCH] add fritzbox helper

---
 fritzbox_helper.py | 94 ++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 94 insertions(+)
 create mode 100755 fritzbox_helper.py

diff --git a/fritzbox_helper.py b/fritzbox_helper.py
new file mode 100755
index 0000000..08e8c25
--- /dev/null
+++ b/fritzbox_helper.py
@@ -0,0 +1,94 @@
+#!/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
\ No newline at end of file