poe-manager.py

#!/usr/bin/env python3

import math
import argparse
import requests
from random import random
from time import sleep
from bs4 import BeautifulSoup


def main(args):
    s = requests.Session()
    # unfortunately, the ssl ciphers used by zyxel are very old and deprecated, thus plaintext HTTP
    url = "http://{}/cgi-bin/dispatcher.cgi".format(args.host)
    s.verify = False

    login_data = {
        "username": args.user,
        "password": encode(args.pwd),
        "login": 'true',
    }
    # print("Logging in...")
    login_step1 = s.post(url, data=login_data)
    login_check_data = {
        "authId": login_step1.text.strip(),
        "login_chk": 'true',
    }
    # implicitly wait for login to occur
    sleep(1)
    login_step2 = s.post(url, data=login_check_data)
    if 'OK' not in login_step2.text:
        raise Exception("Login failed: %s" % login_step2.text)

    # print("Login successful, parsing cookie.")
    ret = s.get(url, params={"cmd": 773})
    if ret.ok:
        soup = BeautifulSoup(ret.content, 'html.parser')
    else:
        raise Exception("Failed to fetch the state of PoE port %s."
                        "Got response: %s" % (args.port, ret.text))
    if args.state is None:
        table = soup.select("table")[2]
        data = []
        # https://stackoverflow.com/a/23377804
        for row in table.find_all('tr'):
            cols = row.find_all('td')
            cols = [ele.text.strip() for ele in cols]
            data.append([ele for ele in cols if ele])

        ret = []
        for entry in data[3:]:
            if entry:
                entr = {}
                for i, item in enumerate(entry[:-1]):
                    entr[data[0][i]] = item
                ret.append(entr)
        if args.port:
            if args.verbose:
                output = ret[args.port - 1]
            else:
                output = ret[args.port - 1].get("State")
        else:
            output = ret
        print(output)
    else:
        xssid_content = soup.find('input', {'name': 'XSSID'}).get('value')
        print("Executing command: Turn %s PoE Port %s." %
              ('on' if args.state else 'off', args.port))
        command_data = {
            "XSSID": xssid_content,
            "portlist": args.port,
            "state": args.state,
            "portPriority": 2,
            "portPowerMode": 3,
            "portRangeDetection": 0,
            "portLimitMode": 0,
            "poeTimeRange": 20,
            "cmd": 775,
            "sysSubmit": "Apply"
        }
        ret = s.post(url, data=command_data)
        if 'window.location.replace' in ret.text:
            print("Command executed successfully!")
        else:
            raise Exception("Failed to execute command: %s" % ret.text)


def encode(_input):
    # The python representation of the JS function with the same name.
    # This could be improved further, but I can't be bothered.
    password = ""
    possible = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789"
    _len = lenn = len(_input)
    i = 1
    while i <= (321 - _len):
        if 0 == i % 5 and _len > 0:
            _len -= 1
            password += _input[_len]
        elif i == 123:
            if lenn < 10:
                password += "0"
            else:
                password += str(math.floor(lenn / 10))
        elif i == 289:
            password += str(lenn % 10)
        else:
            password += possible[math.floor(random() * len(possible))]
        i += 1
    return password


if __name__ == "__main__":
    parser = argparse.ArgumentParser(
        description='Manage the PoE ports of a Zyxel GS1900-10HP switch.')
    parser.add_argument('--host', '-H', dest='host',
                        required=True, help='The hostname of the switch.')
    parser.add_argument('--user', '-U', dest='user',
                        required=True, help='An administrative user.')
    parser.add_argument('--password', '-P', dest='pwd',
                        required=True, help='Password of the admin user.')
    parser.add_argument('--port', '-p', dest='port', type=int, required=True,
                        help='The port number. When querying information, 0 means all ports.')
    parser.add_argument('--state', '-s', dest='state', type=int,
                        choices=[0, 1],
                        help='Turn the port on (1) or off (0). To query the state, '
                             'rather than set it, omit this parameter.')
    parser.add_argument('--verbose', '-V', dest='verbose', action="store_true",
                        help='Return detailed information when querying the specified port state.')
    main(parser.parse_args())