def BaseHomePage(): if request.method == "GET": return render_template("Homepage.html", out_of_date=AppManager.check_software_version(), version_number=AppManager.get_software_verision_number()) elif request.method == "POST": return jsonify(AppManager.get_all_user_campaigns(current_user.user_email))
def inject_dict_for_all_campaign_templates(): if 'cid' in g: cid = g.get('cid') campaign_name = AppManager.campaign_get_campaign_name_from_cid(cid) if cid is not None: return dict(campaign=campaign_name, cid=cid) return dict()
def global_settings_page(): if request.method == "POST": # -- Add user returns a dict with action/result/reason keys. result = UsrMgmt.add_new_user(request.form, current_user.user_email) return jsonify(result) logs = AppManager.get_application_logs(current_user.user_email) return render_template("settings/GlobalSettings.html", logs=logs)
def create_new_campaign(): if request.method == "POST": success_bool, success_msg = AppManager.campaign_create_campaign(current_user.user_email, request.form) if success_bool is True: return redirect(url_for('BaseHomePage')) else: return render_template('CreateCampaign.html', error=success_msg) return render_template('CreateCampaign.html')
def login(): if request.method == "POST": if 'email' in request.form and 'password' in request.form and request.form['email'] is not None and request.form['password'] is not None: user_object = UsrMgmt.user_login(request.form['email'], request.form['password']) if user_object is False: return redirect(url_for("BaseHomePage", error="Incorrect Username/Password")) if user_object.first_logon == 1: login_user(user_object) return redirect(url_for("BaseHomePage")) else: guid = UsrMgmt.get_first_logon_guid(request.form['email']) # return render_template("auth/PasswordResetPage.html",guid=guid) return redirect(url_for("PasswordReset", guid=guid)) return render_template("auth/LoginPage.html", fudge_version=AppManager.get_software_verision_number(), fudge_version_name=AppManager.get_software_verision_name())
def global_settings_page(): if request.method == "POST": # -- Add user returns a dict with action/result/reason keys. result = UsrMgmt.add_new_user(request.form, current_user.user_email) return jsonify(result) # Getting server & user logs, reversing for newest first. logs = AppManager.get_application_logs(current_user.user_email) logs.reverse() user_list = UsrMgmt.get_users_state(current_user.user_email) # Removing our own user as to not present a self-disabling option. for index, user in enumerate(user_list): if user['user_email'] == current_user.user_email: del user_list[index] return render_template("settings/GlobalSettings.html", logs=logs, users=user_list)
def get_user_campaigns(): current_user.user_email = "admin" return jsonify(AppManager.get_all_user_campaigns(current_user.user_email))
from Implant.Implant import ImplantSingleton from ServerApp.modules.UserManagement import UserManagementController from ServerApp.modules.StagerGeneration import StagerGeneration from ServerApp.modules.ImplantManagement import ImplantManagement from ServerApp.modules.ApplicationManager import AppManager from ServerApp.modules.ExportManager import CampaignExportManager from NetworkProfiles.NetworkProfileManager import NetworkProfileManager from NetworkProfiles.NetworkListenerManagement import NetworkListenerManagement Listener = NetworkListenerManagement.instance Imp = ImplantSingleton.instance UsrMgmt = UserManagementController() ImpMgmt = ImplantManagement() StagerGen = StagerGeneration() AppManager = AppManager() ExpoManager = CampaignExportManager() NetProfMng = NetworkProfileManager() app = Flask(__name__) app.config.from_object(__name__) app.config['SECRET_KEY'] = str(uuid.uuid4()) login = LoginManager(app) login.init_app(app) # -- Context Processors --# @app.context_processor def inject_dict_for_all_auth_templates(): # -- Returns the list of Campaigns the authenticated user has at least read access to if current_user.is_authenticated:
def BaseHomePage(): return render_template("Homepage.html", out_of_date=AppManager.check_software_version(), version_number=AppManager.get_software_verision_number())