diff --git a/ogcore-mock.py b/ogcore-mock.py new file mode 100644 index 0000000..5982355 --- /dev/null +++ b/ogcore-mock.py @@ -0,0 +1,197 @@ +from flask import Flask, request, jsonify, render_template_string, abort +import os +import logging +import json +import subprocess + +## FLASK_APP=/path/to/ogcore-mock.py FLASK_ENV=development FLASK_RUN_CERT=adhoc sudo --preserve-env flask run --host 192.168.1.249 --port 443 + + +app = Flask(__name__) +logging.basicConfig(level=logging.INFO) + +## agente lin/win/mac + +@app.route('/opengnsys/rest/ogagent/', methods=['POST']) +def og_agent(cucu): + c = request + logging.info(f"{request.get_json()}") + return jsonify({}) + + + +## agente oglive + +@app.route('/opengnsys/rest/__ogAdmClient/InclusionCliente', methods=['POST']) +def inclusion_cliente(): + c = request + logging.info(f"{request.get_json()}") + #procesoInclusionCliente() or { return (jsonify { 'res': 0 }) } + j = request.get_json(force=True) + iph = j['iph'] ## Toma ip + cfg = j['cfg'] ## Toma configuracion + logging.info(f"iph ({iph}) cfg ({cfg})") + + # dbi->query (sprintf "SELECT ordenadores.*,aulas.idaula,centros.idcentro FROM ordenadores INNER JOIN aulas ON aulas.idaula=ordenadores.idaula INNER JOIN centros ON centros.idcentro=aulas.idcentro WHERE ordenadores.ip = '%s'", iph); + # if (!dbi_result_next_row(result)) { log_error ('client does not exist in database') } + # log_debug (sprintf 'Client %s requesting inclusion', iph); + + idordenador = 42 #dbi_result_get_uint(result, "idordenador") + nombreordenador = "hal9000" #dbi_result_get_string(result, "nombreordenador"); + cache = 42 #dbi_result_get_uint(result, "cache"); + idproautoexec = 42 #dbi_result_get_uint(result, "idproautoexec"); + idaula = 42 #dbi_result_get_uint(result, "idaula"); + idcentro = 42 #dbi_result_get_uint(result, "idcentro"); + + # resul = actualizaConfiguracion(dbi, cfg, idordenador); ## Actualiza la configuración del ordenador + # if (!resul) { log_error ('Cannot add client to database') } + # if (!registraCliente(iph)) { log_error ('client table is full') } ## Incluyendo al cliente en la tabla de sokets + + return jsonify({ + 'res': 1, ## int, Confirmación proceso correcto + 'ido': idordenador, ## int + 'npc': nombreordenador, ## string + 'che': cache, ## int + 'exe': idproautoexec, ## int + 'ida': idaula, ## int + 'idc': idcentro ## int + }) + +def _recorreProcedimientos(parametros, fileexe, idp): + #char idprocedimiento[LONPRM]; + #int procedimientoid, lsize; + #const char *msglog, *param; + #dbi_result result; + + #dbi->query (sprintf 'SELECT procedimientoid,parametros FROM procedimientos_acciones WHERE idprocedimiento=%s ORDER BY orden', $idp); + #if (!result) { log_error ('failed to query database'); return 0; } + if 1: #while (dbi_result_next_row(result)) { + procedimientoid = 0 ## dbi_result_get_uint(result, "procedimientoid"); + if (procedimientoid > 0): ## Procedimiento recursivo + if (not _recorreProcedimientos (parametros, fileexe, procedimientoid)): + return 0 + else: + #param = '@'.join (["nfn=EjecutarScript\rscp=uptime", "nfn=EjecutarScript\rscp=cat /proc/uptime"]) ## dbi_result_get_string(result, "parametros"); + param = '@'.join (["nfn=popup\rtitle=my title\rmessage=my message"]) + parametros = '{}@'.format (param) + fileexe.write (parametros) + #} + + return 1 + +@app.route('/opengnsys/rest/__ogAdmClient/AutoexecCliente', methods=['POST']) +def autoexec_client(): + c = request + logging.info(f"{request.get_json()}") + j = request.get_json(force=True) + iph = j['iph'] ## Toma dirección IP del cliente + exe = j['exe'] ## Toma identificador del procedimiento inicial + logging.info(f"iph ({iph}) exe ({exe})") + + fileautoexec = '/tmp/Sautoexec-{}'.format(iph) + logging.info ("fileautoexec ({})".format (fileautoexec)); + try: + fileexe = open (fileautoexec, 'w') + except Exception as e: + logging.error ("cannot create temporary file: {}".format (e)) + return jsonify({}) + + if (_recorreProcedimientos ('', fileexe, exe)): + res = jsonify ({ 'res': 1, 'nfl': fileautoexec }) + else: + res = jsonify ({ 'res': 0 }) + + fileexe.close() + return res + +@app.route('/opengnsys/rest/__ogAdmClient/enviaArchivo', methods=['POST']) +def envia_archivo(): + c = request + logging.info(f"{request.get_json()}") + j = request.get_json(force=True) + nfl = j['nfl'] ## Toma nombre completo del archivo + logging.info(f"nfl ({nfl})") + + return jsonify({'contents': subprocess.run (['cat', nfl], capture_output=True).stdout.decode('utf-8')}) + +def clienteExistente(iph): + ## esto queda totalmente del lado del servidor, no lo implemento en python + return 42 + +def buscaComandos(ido): + #dbi->query (sprintf "SELECT sesion, parametros FROM acciones WHERE idordenador=%s AND estado='%d' ORDER BY idaccion", ido, ACCION_INICIADA); + #dbi_result_next_row(result) ## cogemos solo una fila + + #if not row { return; } + return + + #ids = 42 #dbi_result_get_uint(result, "sesion"); + #param = "nfn=popup\rtitle=my title\rmessage=my message" #dbi_result_get_string(result, "parametros"); + ## convertirlo a json, aqui lo pongo a capon + #return jsonify ({ 'nfn': 'popup', 'title': 'my title', 'message': 'my message', 'ids': ids }) + +@app.route('/opengnsys/rest/__ogAdmClient/ComandosPendientes', methods=['POST']) +def comandos_pendientes(): + c = request + logging.info(f"{request.get_json()}") + j = request.get_json(force=True) + iph = j['iph'] ## Toma dirección IP + ido = j['ido'] ## Toma identificador del ordenador + logging.info(f"iph ({iph}) ido ({ido})") + + idx = clienteExistente(iph) ## Busca índice del cliente + if not idx: + ## que devuelvo?? pongamos un 404... + abort(404, "Client does not exist") + + param = buscaComandos(ido) ## Existen comandos pendientes, buscamos solo uno + if param is None: + return jsonify({'nfn': 'NoComandosPtes'}) + + #strcpy(tbsockets[idx].estado, CLIENTE_OCUPADO); ## esto queda totalmente del lado del servidor, no lo implemento en python + + return jsonify(param) + +@app.route('/opengnsys/rest/__ogAdmClient/DisponibilidadComandos', methods=['POST']) +def disponibilidad_comandos(): + c = request + logging.info(f"{request.get_json()}") + j = request.get_json(force=True) + iph = j['iph'] + tpc = j['tpc'] + logging.info(f"iph ({iph}) tpc ({tpc})") + + idx = clienteExistente(iph) ## Busca índice del cliente + if not idx: + ## que devuelvo?? pongamos un 404... + abort(404, "Client does not exist") + + #strcpy(tbsockets[idx].estado, tpc); ## esto queda totalmente del lado del servidor, no lo implemento en python + + return jsonify({}) + +@app.route('/opengnsys/rest/__ogAdmClient/', methods=['POST']) +def cucu(cucu): + c = request + j = c.get_json(force=True) + logging.info(f"{request.get_json()} {j}") + if 'cucu' not in j: + abort(400, "missing parameter 'cucu'") + + return jsonify({'cucu': j['cucu']}) + +@app.errorhandler(404) +def _page_not_found(e): + return render_template_string('''not found''') + +@app.errorhandler(500) +def _internal_server_error(e): + return render_template_string('''err''') + +@app.errorhandler(Exception) +def _exception(e): + print(e) + return render_template_string('''exception''') + +if __name__ == '__main__': + app.run(host = '192.168.1.249', port = 443, debug=True)