Merge pull request #3655 from matrix-org/neilj/disable_hs
Flag to disable HS without disabling federationpull/3669/head
						commit
						0ad98e38d0
					
				| 
						 | 
				
			
			@ -0,0 +1 @@
 | 
			
		|||
Ability to disable client/server Synapse via conf toggle
 | 
			
		||||
| 
						 | 
				
			
			@ -779,6 +779,10 @@ class Auth(object):
 | 
			
		|||
        """Checks if the user should be rejected for some external reason,
 | 
			
		||||
        such as monthly active user limiting or global disable flag
 | 
			
		||||
        """
 | 
			
		||||
        if self.hs.config.hs_disabled:
 | 
			
		||||
            raise AuthError(
 | 
			
		||||
                403, self.hs.config.hs_disabled_message, errcode=Codes.HS_DISABLED
 | 
			
		||||
            )
 | 
			
		||||
        if self.hs.config.limit_usage_by_mau is True:
 | 
			
		||||
            current_mau = yield self.store.get_monthly_active_count()
 | 
			
		||||
            if current_mau >= self.hs.config.max_mau_value:
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -57,6 +57,7 @@ class Codes(object):
 | 
			
		|||
    CONSENT_NOT_GIVEN = "M_CONSENT_NOT_GIVEN"
 | 
			
		||||
    CANNOT_LEAVE_SERVER_NOTICE_ROOM = "M_CANNOT_LEAVE_SERVER_NOTICE_ROOM"
 | 
			
		||||
    MAU_LIMIT_EXCEEDED = "M_MAU_LIMIT_EXCEEDED"
 | 
			
		||||
    HS_DISABLED = "M_HS_DISABLED"
 | 
			
		||||
    UNSUPPORTED_ROOM_VERSION = "M_UNSUPPORTED_ROOM_VERSION"
 | 
			
		||||
    INCOMPATIBLE_ROOM_VERSION = "M_INCOMPATIBLE_ROOM_VERSION"
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -78,6 +78,10 @@ class ServerConfig(Config):
 | 
			
		|||
            "mau_limit_reserved_threepids", []
 | 
			
		||||
        )
 | 
			
		||||
 | 
			
		||||
        # Options to disable HS
 | 
			
		||||
        self.hs_disabled = config.get("hs_disabled", False)
 | 
			
		||||
        self.hs_disabled_message = config.get("hs_disabled_message", "")
 | 
			
		||||
 | 
			
		||||
        # FIXME: federation_domain_whitelist needs sytests
 | 
			
		||||
        self.federation_domain_whitelist = None
 | 
			
		||||
        federation_domain_whitelist = config.get(
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -21,7 +21,7 @@ from twisted.internet import defer
 | 
			
		|||
 | 
			
		||||
import synapse.handlers.auth
 | 
			
		||||
from synapse.api.auth import Auth
 | 
			
		||||
from synapse.api.errors import AuthError
 | 
			
		||||
from synapse.api.errors import AuthError, Codes
 | 
			
		||||
from synapse.types import UserID
 | 
			
		||||
 | 
			
		||||
from tests import unittest
 | 
			
		||||
| 
						 | 
				
			
			@ -469,3 +469,12 @@ class AuthTestCase(unittest.TestCase):
 | 
			
		|||
            return_value=defer.succeed(small_number_of_users)
 | 
			
		||||
        )
 | 
			
		||||
        yield self.auth.check_auth_blocking()
 | 
			
		||||
 | 
			
		||||
    @defer.inlineCallbacks
 | 
			
		||||
    def test_hs_disabled(self):
 | 
			
		||||
        self.hs.config.hs_disabled = True
 | 
			
		||||
        self.hs.config.hs_disabled_message = "Reason for being disabled"
 | 
			
		||||
        with self.assertRaises(AuthError) as e:
 | 
			
		||||
            yield self.auth.check_auth_blocking()
 | 
			
		||||
        self.assertEquals(e.exception.errcode, Codes.HS_DISABLED)
 | 
			
		||||
        self.assertEquals(e.exception.code, 403)
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
| 
						 | 
				
			
			@ -74,6 +74,8 @@ def setup_test_homeserver(name="test", datastore=None, config=None, reactor=None
 | 
			
		|||
        config.media_storage_providers = []
 | 
			
		||||
        config.auto_join_rooms = []
 | 
			
		||||
        config.limit_usage_by_mau = False
 | 
			
		||||
        config.hs_disabled = False
 | 
			
		||||
        config.hs_disabled_message = ""
 | 
			
		||||
        config.max_mau_value = 50
 | 
			
		||||
        config.mau_limits_reserved_threepids = []
 | 
			
		||||
 | 
			
		||||
| 
						 | 
				
			
			
 | 
			
		|||
		Loading…
	
		Reference in New Issue