Require address verification after one successful verification
This commit is contained in:
@@ -0,0 +1,23 @@
|
||||
"""Add address verification required field
|
||||
|
||||
Revision ID: d6af394ec1ab
|
||||
Revises: 00ac5d537063
|
||||
Create Date: 2021-07-10
|
||||
"""
|
||||
from alembic import op
|
||||
import sqlalchemy as sa
|
||||
from sqlalchemy.sql.expression import text
|
||||
|
||||
|
||||
revision = 'd6af394ec1ab'
|
||||
down_revision = '00ac5d537063'
|
||||
branch_labels = None
|
||||
depends_on = None
|
||||
|
||||
|
||||
def upgrade():
|
||||
op.add_column('server', sa.Column('address_verification_required', sa.Boolean(), nullable=False, server_default=text('0')))
|
||||
|
||||
|
||||
def downgrade():
|
||||
op.drop_column('server', 'address_verification_required')
|
||||
@@ -102,6 +102,8 @@ class Server(db.Model):
|
||||
rollback_enabled = db.Column(db.Boolean, nullable=False)
|
||||
can_see_far_names = db.Column(db.Boolean, nullable=False)
|
||||
|
||||
address_verification_required = db.Column(db.Boolean, nullable=False, default=False)
|
||||
|
||||
@staticmethod
|
||||
def find_from_json(obj):
|
||||
try:
|
||||
@@ -194,6 +196,9 @@ class Server(db.Model):
|
||||
|
||||
self.last_update = now
|
||||
|
||||
if obj["address_verified"]:
|
||||
self.address_verification_required = True
|
||||
|
||||
def as_json(self):
|
||||
obj = {
|
||||
"address": self.address,
|
||||
|
||||
@@ -92,13 +92,15 @@ def announce():
|
||||
if addr_info is None:
|
||||
return f"Failed to resolve server address {obj['address']!r}.", 400
|
||||
|
||||
valid = False
|
||||
if "world_uuid" not in obj:
|
||||
valid = verify_announce(addr_info, obj["address"], obj["ip"])
|
||||
|
||||
if not valid:
|
||||
if not valid and server and server.address_verification_required:
|
||||
return render_template("address_verification_failed.txt",
|
||||
announce_ip=announce_ip,
|
||||
valid_addresses=[data[4][0] for data in addr_info]), 400
|
||||
obj["address_verified"] = valid
|
||||
|
||||
obj["addr_info"] = addr_info
|
||||
|
||||
|
||||
Reference in New Issue
Block a user