mirror of
https://dev.iopsys.eu/bbf/bbfdm.git
synced 2025-12-10 07:44:39 +01:00
remove bbf non-related directory
This commit is contained in:
Sukru Senli
parent
7d98d8d01a
commit
236481c981
1 changed files with 0 additions and 338 deletions
338
ubus/ubus.py
338
ubus/ubus.py
|
|
@ -1,338 +0,0 @@
|
||||||
#!/usr/bin/python
|
|
||||||
|
|
||||||
# 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 2 of the License, or
|
|
||||||
# (at your option) any later version.
|
|
||||||
#
|
|
||||||
# Copyright (C) 2019 iopsys Software Solutions AB
|
|
||||||
# Author: Amin Ben Ramdhane <amin.benramdhane@pivasoftware.com>
|
|
||||||
|
|
||||||
import os, re, sys, time, json, getopt, getpass
|
|
||||||
from collections import OrderedDict
|
|
||||||
from pexpect import pxssh
|
|
||||||
from shutil import copyfile
|
|
||||||
|
|
||||||
def removefile( filename ):
|
|
||||||
try:
|
|
||||||
os.remove(filename)
|
|
||||||
except OSError:
|
|
||||||
pass
|
|
||||||
|
|
||||||
def securemkdir( folder ):
|
|
||||||
try:
|
|
||||||
os.mkdir(folder)
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
|
|
||||||
def printopenobj( obj ):
|
|
||||||
fp = open('./.ubus_tmp', 'a')
|
|
||||||
print >> fp, "\"%s\" : {" % obj
|
|
||||||
fp.close()
|
|
||||||
|
|
||||||
def printopenfile():
|
|
||||||
fp = open('./.ubus_tmp', 'a')
|
|
||||||
print >> fp, "{"
|
|
||||||
fp.close()
|
|
||||||
|
|
||||||
def printclosefile():
|
|
||||||
fp = open('./.ubus_tmp', 'a')
|
|
||||||
print >> fp, "}"
|
|
||||||
fp.close()
|
|
||||||
|
|
||||||
def removelastline():
|
|
||||||
file_r = open("./.ubus_tmp")
|
|
||||||
lines = file_r.readlines()
|
|
||||||
lines = lines[:-1]
|
|
||||||
file_r.close()
|
|
||||||
file_w = open("./.ubus_tmp",'w')
|
|
||||||
file_w.writelines(lines)
|
|
||||||
file_w.close()
|
|
||||||
printclosefile ()
|
|
||||||
|
|
||||||
def copy_data_in_file( data):
|
|
||||||
file_r = open("./.tmp.json", 'a')
|
|
||||||
print >> file_r, "%s" % data
|
|
||||||
file_r.close()
|
|
||||||
|
|
||||||
def replace_data_in_file( data_in, data_out ):
|
|
||||||
file_r = open("./.ubus_tmp", "rt")
|
|
||||||
file_w = open("./.ubus_tmp_1", "wt")
|
|
||||||
text = ''.join(file_r).replace(data_in, data_out)
|
|
||||||
file_w.write(text)
|
|
||||||
file_r.close()
|
|
||||||
file_w.close()
|
|
||||||
copyfile("./.ubus_tmp_1", "./.ubus_tmp")
|
|
||||||
removefile("./.ubus_tmp_1")
|
|
||||||
|
|
||||||
def updatejsontmpfile():
|
|
||||||
replace_data_in_file ("}\n", "},\n")
|
|
||||||
replace_data_in_file ("},\n},", "}\n},")
|
|
||||||
replace_data_in_file ("}\n},\n},", "}\n}\n},")
|
|
||||||
replace_data_in_file ("}\n},\n}\n},", "}\n}\n}\n},")
|
|
||||||
replace_data_in_file ("}\n},\n}\n}\n},", "}\n}\n}\n}\n},")
|
|
||||||
replace_data_in_file ("}\n}\n}\n},\n}\n},", "}\n}\n}\n}\n}\n},")
|
|
||||||
replace_data_in_file ("}\n}\n}\n}\n}\n}\n},", "}\n}\n}\n}\n}\n}\n},")
|
|
||||||
replace_data_in_file ("}\n}\n}\n},\n}\n}\n}\n},", "}\n}\n}\n}\n}\n}\n}\n}")
|
|
||||||
|
|
||||||
def get_type_field( data ):
|
|
||||||
if isinstance(data,bool):
|
|
||||||
return "boolean"
|
|
||||||
elif isinstance(data,(float, int)):
|
|
||||||
return "integer"
|
|
||||||
elif isinstance(data,list):
|
|
||||||
return "array"
|
|
||||||
else:
|
|
||||||
return "string"
|
|
||||||
|
|
||||||
def read_json( data, parse ):
|
|
||||||
for x in data:
|
|
||||||
if isinstance(data[x],dict):
|
|
||||||
if parse == 0:
|
|
||||||
read_json (data[x], 1)
|
|
||||||
break
|
|
||||||
else:
|
|
||||||
printopenobj(x)
|
|
||||||
read_json (data[x], 1)
|
|
||||||
printclosefile()
|
|
||||||
elif isinstance(data[x],list):
|
|
||||||
for element in data[x]:
|
|
||||||
if isinstance(element,dict):
|
|
||||||
printopenobj(x)
|
|
||||||
read_json (element, 1)
|
|
||||||
printclosefile()
|
|
||||||
else:
|
|
||||||
type_field = get_type_field(element)
|
|
||||||
printopenobj(x)
|
|
||||||
fp = open('./.ubus_tmp', 'a')
|
|
||||||
print >> fp, "\"description\": \"TODO\","
|
|
||||||
print >> fp, "\"list\": \"true\","
|
|
||||||
print >> fp, "\"type\": \"%s\"" % type_field
|
|
||||||
fp.close()
|
|
||||||
printclosefile()
|
|
||||||
break
|
|
||||||
else:
|
|
||||||
type_field = get_type_field(data[x])
|
|
||||||
printopenobj(x)
|
|
||||||
fp = open('./.ubus_tmp', 'a')
|
|
||||||
print >> fp, "\"description\": \"TODO\","
|
|
||||||
print >> fp, "\"type\": \"%s\"" % type_field
|
|
||||||
fp.close()
|
|
||||||
printclosefile()
|
|
||||||
|
|
||||||
|
|
||||||
def generate_output( data, parse ):
|
|
||||||
removefile("./.tmp.json")
|
|
||||||
copy_data_in_file(data)
|
|
||||||
|
|
||||||
file = open("./.tmp.json")
|
|
||||||
data = json.loads(file.read(), object_pairs_hook=OrderedDict)
|
|
||||||
file.close()
|
|
||||||
|
|
||||||
read_json (data, parse)
|
|
||||||
|
|
||||||
def generate_in_out_put( root_object, obj, line ):
|
|
||||||
fp = open('./.ubus_tmp', 'a')
|
|
||||||
print >> fp, "\"description\": \"TODO\","
|
|
||||||
in_args = re.search('{(.+?)}', line)
|
|
||||||
if in_args:
|
|
||||||
found = in_args.group(1)
|
|
||||||
lst = found.split(",")
|
|
||||||
stringcount = len(lst)
|
|
||||||
print >> fp, "\"input\": {"
|
|
||||||
for idx, value in enumerate(lst):
|
|
||||||
in_arg = value.split(":")
|
|
||||||
print >> fp, "%s: {" % in_arg[0]
|
|
||||||
print >> fp, "\"description\": \"TODO\","
|
|
||||||
print >> fp, "\"type\": %s" % in_arg[1].lower()
|
|
||||||
print >> fp, "}"
|
|
||||||
print >> fp, "}"
|
|
||||||
else:
|
|
||||||
print >> fp, "\"input\": {},"
|
|
||||||
|
|
||||||
if root_object == "system" and obj == "reboot" or \
|
|
||||||
root_object == "juci.system" and (obj == "reboot" or obj == "defaultreset" or obj == "report") or \
|
|
||||||
root_object == "router.dropbear" and obj == "get_ssh_keys" or \
|
|
||||||
root_object == "network.wireless" or \
|
|
||||||
root_object == "network.device" and (obj == "set_alias" or obj == "set_state") or \
|
|
||||||
root_object == "session" and obj == "list" or \
|
|
||||||
root_object == "network.interface" and (obj == "up" or obj == "down" or obj == "renew" or obj == "prepare" or obj == "notify_proto" or obj == "remove" or obj == "set_data"):
|
|
||||||
print >> fp, "\"output\": {}"
|
|
||||||
fp.close()
|
|
||||||
else:
|
|
||||||
if ((root_object == "router.wireless" or root_object == "wifix") and obj == "status") or \
|
|
||||||
((root_object == "router.wireless" or root_object == "wifix") and obj == "assoclist") or \
|
|
||||||
((root_object == "router.wireless" or root_object == "wifix") and obj == "stas") or \
|
|
||||||
((root_object == "router.wireless" or root_object == "wifix") and obj == "stations"):
|
|
||||||
command1 = "uci get wireless.@wifi-iface[0].device"
|
|
||||||
s.sendline (command1)
|
|
||||||
s.prompt()
|
|
||||||
device = s.before.split('\n',1)[1];
|
|
||||||
device_name = device.translate(None, ' \n\t\r')
|
|
||||||
command = "ubus call %s %s \'{\"vif\":\"%s\"}\'" % (root_object, obj, device_name)
|
|
||||||
elif root_object == "network.interface":
|
|
||||||
command = "ubus call %s.lan %s" % (root_object, obj)
|
|
||||||
else:
|
|
||||||
command = "ubus call %s %s" % (root_object, obj)
|
|
||||||
s.sendline (command)
|
|
||||||
s.prompt()
|
|
||||||
data = s.before.split('\n',1)[1];
|
|
||||||
print "****************************************************************************"
|
|
||||||
print "command = %s" % command
|
|
||||||
print "data = %s" % data
|
|
||||||
print "****************************************************************************"
|
|
||||||
parse = 1
|
|
||||||
if root_object == "network.device" and obj == "status" or \
|
|
||||||
root_object == "router.graph" and obj == "iface_traffic" or \
|
|
||||||
root_object == "router.port" and obj == "status" or \
|
|
||||||
root_object == "router.network" and obj == "clients" or \
|
|
||||||
root_object == "router.network" and obj == "dump" or \
|
|
||||||
(root_object == "router.wireless" or root_object == "wifix") and obj == "temperature" or \
|
|
||||||
(root_object == "router.wireless" or root_object == "wifix") and obj == "stas" or \
|
|
||||||
(root_object == "router.wireless" or root_object == "wifix") and obj == "radios":
|
|
||||||
parse = 0
|
|
||||||
if ("Command failed" in data.split("\n", 1)[0]) or (data == ""):
|
|
||||||
print >> fp, "\"output\": {}"
|
|
||||||
fp.close()
|
|
||||||
else:
|
|
||||||
print >> fp, "\"output\": {"
|
|
||||||
fp.close()
|
|
||||||
generate_output(data, parse)
|
|
||||||
printclosefile()
|
|
||||||
|
|
||||||
def convert_object( line ):
|
|
||||||
found = ""
|
|
||||||
m = re.search('\'(.+?)\'', line)
|
|
||||||
if m:
|
|
||||||
found = m.group(1)
|
|
||||||
if "network.interface." in found:
|
|
||||||
return found
|
|
||||||
else:
|
|
||||||
printopenobj(found)
|
|
||||||
|
|
||||||
return found
|
|
||||||
|
|
||||||
def convert_param( root_object, line ):
|
|
||||||
m = re.search('\"(.+?)\":{', line)
|
|
||||||
if m:
|
|
||||||
obj = m.group(1)
|
|
||||||
printopenobj(obj)
|
|
||||||
generate_in_out_put(root_object, obj, line)
|
|
||||||
printclosefile()
|
|
||||||
|
|
||||||
def removetmpfiles():
|
|
||||||
removefile("./.ubus_tmp")
|
|
||||||
removefile("./.tmp.json")
|
|
||||||
|
|
||||||
def generatejsonfromdata( data, gendir ):
|
|
||||||
first_object = 0
|
|
||||||
root_object = ""
|
|
||||||
|
|
||||||
securemkdir(gendir)
|
|
||||||
removetmpfiles()
|
|
||||||
printopenfile()
|
|
||||||
|
|
||||||
lines = data.split ('\n')
|
|
||||||
for line in lines :
|
|
||||||
if "@" in line:
|
|
||||||
if first_object != 0:
|
|
||||||
printclosefile()
|
|
||||||
root_object = convert_object(line)
|
|
||||||
if "network.interface." in root_object:
|
|
||||||
first_object = 0
|
|
||||||
else:
|
|
||||||
first_object = 1
|
|
||||||
continue
|
|
||||||
else:
|
|
||||||
if "network.interface." in root_object:
|
|
||||||
continue
|
|
||||||
convert_param(root_object, line)
|
|
||||||
|
|
||||||
printclosefile()
|
|
||||||
printclosefile()
|
|
||||||
updatejsontmpfile ()
|
|
||||||
removelastline ()
|
|
||||||
|
|
||||||
file_r = open("./.ubus_tmp", "r")
|
|
||||||
obj = json.load(file_r, object_pairs_hook=OrderedDict)
|
|
||||||
dump = json.dumps(obj, indent=2)
|
|
||||||
file_r.close()
|
|
||||||
|
|
||||||
try:
|
|
||||||
json_file = open(gendir + "/ubus.json", "a")
|
|
||||||
print >> json_file, "%s" % dump
|
|
||||||
json_file.close()
|
|
||||||
except:
|
|
||||||
pass
|
|
||||||
|
|
||||||
removetmpfiles()
|
|
||||||
|
|
||||||
def printusage():
|
|
||||||
print "Usage:"
|
|
||||||
print ""
|
|
||||||
print sys.argv[0] + " -h{hostname} -u{user} -p{password}"
|
|
||||||
print "Example:"
|
|
||||||
print sys.argv[0] + " -h 192.168.1.1 -u root -p 10pD3v"
|
|
||||||
print ""
|
|
||||||
print "Or you can run the tool directly using " + sys.argv[0]
|
|
||||||
print "then enter your Hostname, Username and Password"
|
|
||||||
print "Example:"
|
|
||||||
print sys.argv[0]
|
|
||||||
print "Enter your Hostname: 192.168.1.1"
|
|
||||||
print "Enter your Username: root"
|
|
||||||
print "Enter your Password: 10pD3v"
|
|
||||||
|
|
||||||
### main ###
|
|
||||||
try:
|
|
||||||
optlist, args = getopt.getopt(sys.argv[1:], 'h:u:p:', ['help','?'])
|
|
||||||
except Exception, e:
|
|
||||||
print str(e)
|
|
||||||
printusage()
|
|
||||||
exit(1)
|
|
||||||
|
|
||||||
options = dict(optlist)
|
|
||||||
if len(args) > 1:
|
|
||||||
printusage()
|
|
||||||
exit(1)
|
|
||||||
|
|
||||||
if [elem for elem in options if elem in ['-?','--?','--help']]:
|
|
||||||
printusage()
|
|
||||||
exit(1)
|
|
||||||
|
|
||||||
if '-h' in options:
|
|
||||||
host = options['-h']
|
|
||||||
else:
|
|
||||||
host = raw_input('Enter your Hostname: ')
|
|
||||||
if '-u' in options:
|
|
||||||
user = options['-u']
|
|
||||||
else:
|
|
||||||
user = raw_input('Enter your Username: ')
|
|
||||||
if '-p' in options:
|
|
||||||
password = options['-p']
|
|
||||||
else:
|
|
||||||
password = getpass.getpass('Enter your Password: ')
|
|
||||||
|
|
||||||
## Login via SSH
|
|
||||||
try:
|
|
||||||
s = pxssh.pxssh()
|
|
||||||
s.login (host, user, password)
|
|
||||||
print "SSH session login successful"
|
|
||||||
command = "ubus -v list"
|
|
||||||
s.sendline (command)
|
|
||||||
s.prompt()
|
|
||||||
data = s.before.split("\n",1)[1];
|
|
||||||
gendir = "source_" + time.strftime("%Y-%m-%d_%H-%M-%S")
|
|
||||||
print "Start Generation of JSON file"
|
|
||||||
print "Please wait..."
|
|
||||||
generatejsonfromdata(data, gendir)
|
|
||||||
s.logout()
|
|
||||||
|
|
||||||
except pxssh.ExceptionPxssh as e:
|
|
||||||
print("SSH session failed on login.")
|
|
||||||
print(e)
|
|
||||||
|
|
||||||
if (os.path.isdir(gendir)):
|
|
||||||
print "JSON file generated under \"./%s\"" % gendir
|
|
||||||
else:
|
|
||||||
print "No JSON file generated!"
|
|
||||||
|
|
||||||
Loading…
Add table
Reference in a new issue