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())
