def import_scope(scopetype=""):
if scopetype == "blacklist":
importBlacklist = True
importForm = forms.ImportBlacklistForm()
elif scopetype == "scope":
importBlacklist = False
importForm = forms.ImportScopeForm()
else:
return abort(404)
if importForm.validate_on_submit():
newScopeItems = importForm.scope.data.split("\n")
fail, exist, success = ScopeItem.import_scope_list(
newScopeItems, importBlacklist
)
db.session.commit()
current_app.ScopeManager.update()
if success:
flash(f"{len(success)} targets added to {scopetype}!", "success")
if exist:
flash(f"{len(exist)} targets already existed!", "info")
if fail:
flash(f"{len(fail)} targets failed to import!", "danger")
for item in fail:
flash(f"{item}", "danger")
else:
for field, errors in importForm.errors.items():
for error in errors:
flash(error, "danger")
return redirect(url_for(f"admin.{scopetype}"))
def scope():
scope = ScopeItem.get_scope()
scopeSize = current_app.ScopeManager.get_scope_size()
# if it's zero, let's make sure the ScopeManager is up to date
if scopeSize == 0:
current_app.ScopeManager.update()
scopeSize = current_app.ScopeManager.get_scope_size()
# if it's zero again that's fine, we just had to check
newForm = forms.NewScopeForm()
delForm = forms.ScopeDeleteForm()
editForm = forms.ScopeToggleForm()
importForm = forms.ImportScopeForm()
addTagForm = forms.TagScopeForm()
addTagForm.tagname.choices = [(row.name, row.name) for row in Tag.query.all()]
if newForm.validate_on_submit():
if '/' not in newForm.target.data:
newForm.target.data = newForm.target.data + '/32'
target = ipaddress.ip_network(newForm.target.data, False)
newTarget = ScopeItem(target=target.with_prefixlen, blacklist=False)
db.session.add(newTarget)
db.session.commit()
current_app.ScopeManager.update()
flash('%s added!' % newTarget.target, 'success')
return redirect(url_for('admin.scope'))
return render_template(
"admin/scope.html", scope=scope, scopeSize=scopeSize, delForm=delForm,
editForm=editForm, newForm=newForm, importForm=importForm, addTagForm=addTagForm)
def importScope(scopetype=''):
if scopetype == 'blacklist':
importBlacklist = True
importForm = forms.ImportBlacklistForm()
elif scopetype == 'scope':
importBlacklist = False
importForm = forms.ImportScopeForm()
else:
abort(404)
if importForm.validate_on_submit():
successImports = []
alreadyExists = []
failedImports = []
newScopeItems = importForm.scope.data.split('\n')
for item in newScopeItems:
item = item.strip()
fail, exist, success = ScopeItem.importScope(item, importBlacklist)
failedImports = failedImports + fail
alreadyExists = alreadyExists + exist
successImports = successImports + success
db.session.commit()
current_app.ScopeManager.update()
if len(successImports) > 0:
flash('%s targets added to %s!' % (len(successImports), scopetype), 'success')
if len(alreadyExists) > 0:
flash('%s targets already existed!' % len(alreadyExists), 'info')
if len(failedImports) > 0:
flash('%s targets failed to import!' % len(failedImports), 'danger')
for item in failedImports:
flash('%s' % item, 'danger')
else:
for field, errors in importForm.errors.items():
for error in errors:
flash(error, 'danger')
return redirect(url_for('admin.%s' % scopetype))
def blacklist():
render = {
"scope": ScopeItem.getBlacklist(),
"blacklistSize": current_app.ScopeManager.get_blacklist_size(),
"effectiveScopeSize": current_app.ScopeManager.get_effective_scope_size(),
"newForm": forms.NewScopeForm(),
"delForm": forms.ScopeDeleteForm(),
"editForm": forms.ScopeToggleForm(),
"importForm": forms.ImportScopeForm(),
"addTagForm": forms.TagScopeForm(),
}
render["addTagForm"].tagname.choices = [
(row.name, row.name) for row in Tag.query.all()
]
if render["newForm"].validate_on_submit():
target = ipaddress.ip_network(render["newForm"].target.data, False)
newTarget = ScopeItem(target=target.with_prefixlen, blacklist=True)
db.session.add(newTarget)
db.session.commit()
current_app.ScopeManager.update()
flash(f"{newTarget.target} blacklisted.", "success")
return redirect(url_for("admin.blacklist"))
return render_template("admin/blacklist.html", **render)