user2476761254
Published © GPL3+

Network Monitoring with AWS IoT

Use Raspberry PI as network monitor box, always connected with AWS IoT/SNS to monitor network status or help power up/down device remotely.

BeginnerProtip8,692
Network Monitoring with AWS IoT

Things used in this project

Hardware components

Raspberry Pi 1 Model B+
Raspberry Pi 1 Model B+
×1

Software apps and online services

AWS IoT
Amazon Web Services AWS IoT
AWS SNS
Amazon Web Services AWS SNS

Story

Read more

Schematics

System design

Code

Raspberry Pi AWS IoT Client Code

Python
The code is based on Python MQTT Paho library (https://pypi.python.org/pypi/paho-mqtt/1.1)
It will monitor the network speed (reported by router based on SNMP), publish it if certain threshold reached.
It also subscribe and listen to our request, for example "status" report, "powerup"/"powerdown" specific PC, "scan" the network for active workstation, or report the speed.
Network Monitoring with AWS IoT
Copyright (C) 2016 Louis Tam

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License
along with this program.  If not, see <http://www.gnu.org/licenses/>.

import json
import ssl
import time
import subprocess
import os
import paho.mqtt.client as mqtt

devicecert = "/root/key/a1a97c3bfe-certificate.pem.crt"
devicekey = "/root/key/a1a97c3bfe-private.pem.key"
servercert = "/root/key/root-CA.crt"
server = "A3UTD0DHSM47I.iot.us-west-2.amazonaws.com"
topic="$aws/things/netmon1/shadow/update"
mac = 'YOUR PC MAC ADDRESS'
qos=0
retain=False

def powerup():
    print ("send magic packet to "+mac)
    subprocess.Popen(["wakeonlan", mac])
    time.sleep(1)
    subprocess.Popen(["wakeonlan", mac])
    time.sleep(1)
    subprocess.Popen(["wakeonlan", mac])
    #Assume target is waken
    return 1

def powerdown():
    print ("Power down PC")
    p = subprocess.Popen(["arp","-a"],stdout=subprocess.PIPE,stderr=subprocess.PIPE)
    output, errors = p.communicate()
    output=output.decode()
    output=output.splitlines()
    
    for i in range(len(output)):
        if mac in str(output[i]):
            a = output[i].split('(', 1 )
            a = a[1].split(')',1)
            ip = a[0]
    subprocess.Popen(["net", "rpc", "shutdown", "-I", ip, "-U", "user%pass"]) #For Windows PC, configure the actual username/password here
    return 1

def scan():
    print ("Scan subnet now")
    subprocess.Popen(["fping", "-c", "1", "-g", "192.168.1.0/24"]) #Configure the actual subnet of network
    time.sleep(20)
    p = subprocess.Popen(['arp','-a'],stdout=subprocess.PIPE,stderr=subprocess.PIPE)
    output, errors = p.communicate()
    output=output.decode()
    print (output)
    output=output.splitlines()
    print (len(output))
    return "Active IP: " + str(len(output))

def checkspeed(index):
    print ("check network traffic")
    p = subprocess.Popen(["snmpwalk","-v2c","-c","public","192.168.1.1",".1.3.6.1.2.1.2.2.1.10"],stdout=subprocess.PIPE,stderr=subprocess.PIPE)
    output, errors = p.communicate()
    time.sleep(1)
    output=output.decode()
    output=output.splitlines()
    a = [0] * len(output)
    for i in range(len(output)):
        a[i] = output[i].split(' = Counter32: ', 1 )

    time.sleep(5)
    p = subprocess.Popen(["snmpwalk","-v2c","-c","public","192.168.1.1",".1.3.6.1.2.1.2.2.1.10"],stdout=subprocess.PIPE,stderr=subprocess.PIPE)
    output, errors = p.communicate()
    time.sleep(1)
    output=output.decode()
    output=output.splitlines()
    b = [0] * len(output)
    c = [0] * len(output)
    for i in range(len(output)):
        b[i] = output[i].split(' = Counter32: ', 1 )
    for i in range(len(a)):
        c[i] = int((((int(b[i][1])-int(a[i][1]))*8*100)/(5*100)))
        print (c[i])
    return c
    
def monspeed():
    c = checkspeed(-1)
    if c[2] > 2000000: # customize the interface/speed to trigger the warning
        awsmsg = {'state':{ 'reported': {'warning': 'speed' , 'result': c}}}
        payload = json.dumps(awsmsg)
        print (awsmsg)
        client.publish(topic,payload,qos,retain)

def status():
    print ("report status")
    p = subprocess.Popen(["uptime"],stdout=subprocess.PIPE,stderr=subprocess.PIPE)
    output, errors = p.communicate()
    output=output.decode()
    output=output.split(',', 1 )
    return output[0]

def on_message(client, userdata, msg):
    print("Message: " + msg.topic + " Payload:" + str(msg.payload))
    if "timestamp" in str(msg.payload):
        return;
    if "request" not in str(msg.payload):
        return;
    print ("Process message")
    if "powerup" in str(msg.payload):
        ret = powerup()
        awsmsg = {'state':{ 'reported': {'action': 'powerup' , 'result': ret}}}
    if "powerdown" in str(msg.payload):
        ret = powerdown()
        awsmsg = {'state':{ 'reported': {'action': 'powerdown' , 'result': ret}}}
    if "status" in str(msg.payload):
        ret = status()
        awsmsg = {'state':{ 'reported': {'action': 'status' , 'result': ret}}}
    if "scan" in str(msg.payload):
        ret = scan()
        awsmsg = {'state':{ 'reported': {'action': 'scan' , 'result': ret}}}
    if "speed" in str(msg.payload):
        c = checkspeed(-1)
        awsmsg = {'state':{ 'reported': {'action': 'scan' , 'result': c}}}
    payload = json.dumps(awsmsg)
    print (awsmsg)
    client.publish(topic, payload, qos, retain)

def on_log(client, userdata, level, msg):
    print("Log: " + msg)

def on_connect(client, userdata, flags, rc):
    client.subscribe("$aws/things/netmon1/#")

client = mqtt.Client("netmon1")
client.on_connect = on_connect
client.on_message = on_message
client.on_log = on_log
client.tls_set(servercert, devicecert, devicekey, ssl.CERT_REQUIRED, ssl.PROTOCOL_SSLv23)
client.connect(server, 8883, 60)

client.loop_start()

awsmsg = {'state':{ 'reported': {'powerup': 1 }}} #update on power up
payload = json.dumps(awsmsg)
client.publish(topic, payload ,qos, retain)

while True:
    monspeed()
    time.sleep(60)

Credits

user2476761254

user2476761254

1 project • 8 followers

Comments

Add projectSign up / Login