source: ogAgent-Git/ogcore-mock.py @ 9d65966

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/<cucu>', 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/<cucu>', 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('''<!DOCTYPE html><html>not found</html>''')

@app.errorhandler(500)
def _internal_server_error(e):
    return render_template_string('''<!DOCTYPE html><html>err</html>''')

@app.errorhandler(Exception)
def _exception(e):
    print(e)
    return render_template_string('''<!DOCTYPE html><html>exception</html>''')

if __name__ == '__main__':
    app.run(host = '192.168.1.249', port = 443, debug=True)