[TIL_Carrotww] 69 - 22/12/09

📝Carrotww의 코딩 기록장

🧲 django Channels 인증

🔍 저번 포스팅에 channels 인증에 관한 글을 썼다.
하지만 실제 테스트를 해보니 내가 원하는 사용자로 인증이 되지 않았었다.
지금부터 하나씩 뜯어보려고 한다.

class ChatConsumer(AsyncWebsocketConsumer):
    async def connect(self):
        # 로그인된 사용자인지 확인
        if self.scope['user']:
            pass
        else:
            self.close()
        await self.accept()

나는 위와같이 간단하게 scope에 담긴 user를 확인하여 인증 수행을 테스트 했다.
테스트를 수행하는데 계속 한 사용자의 정보만 들어왔다. 이게 의아하여 scope data찍어봤다.

{'asgi': {'version': '3.0'},
 'client': ['127.0.0.1', 54699],
 'cookies': {'csrftoken': 'FFQG53vwTjHdfsPEkHyF3iu24FHSdq4M27yVrDV8IXeW2fIxzI2gOu1yRckbU5VT',
             'sessionid': 'mccrs5c1rk52kdibw1l469tak3x2elnv'},
 'headers': [(b'host', b'127.0.0.1:8000'),
             (b'connection', b'Upgrade'),
             (b'pragma', b'no-cache'),
             (b'cache-control', b'no-cache'),
             (b'user-agent',
              b'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36'  
              b' (KHTML, like Gecko) Chrome/106.0.5249.207 Whale/3.17.145.18'  
              b' Safari/537.36'),
             (b'upgrade', b'websocket'),
             (b'origin', b'http://127.0.0.1:5500'),
             (b'sec-websocket-version', b'13'),
             (b'accept-encoding', b'gzip, deflate, br'),
             (b'accept-language', b'ko-KR,ko;q=0.9,en-US;q=0.8,en;q=0.7'),     
             (b'cookie',
              b'csrftoken=FFQG53vwTjHdfsPEkHyF3iu24FHSdq4M27yVrDV8IXeW2fIxzI'  
              b'2gOu1yRckbU5VT; sessionid=mccrs5c1rk52kdibw1l469tak3x2elnv'),  
             (b'sec-websocket-key', b'tF4uRtVXNpr55l9KcDoD8w=='),
             (b'sec-websocket-extensions',
              b'permessage-deflate; client_max_window_bits')],
 'path': '/ws/socket-server/1/',
 'path_remaining': '',
 'query_string': b'',
 'raw_path': b'/ws/socket-server/1/',
 'server': ['127.0.0.1', 8000],
 'session': <django.utils.functional.LazyObject object at 0x00000237C83AFD60>, 
 'subprotocols': [],
 'type': 'websocket',
 'url_route': {'args': (), 'kwargs': {'room_id': 1}},
 'user': <channels.auth.UserLazyObject object at 0x00000237C83AFB50>}

보면 'user' 부분이
channels.auth.UserLazyObject 를 사용중인것을 알 수 있었다.
난 channels.auth 에 AuthMiddlesareStack 을 사용하고 있어 해당 함수로 들어가보았다.

class AuthMiddleware(BaseMiddleware):
    """
    Middleware which populates scope["user"] from a Django session.
    Requires SessionMiddleware to function.
    """

    def populate_scope(self, scope):
        # Make sure we have a session
        if "session" not in scope:
            raise ValueError(
                "AuthMiddleware cannot find session in scope. "
                "SessionMiddleware must be above it."
            )
        # Add it to the scope if it's not there already
        if "user" not in scope:
            scope["user"] = UserLazyObject()

    async def resolve_scope(self, scope):
        scope["user"]._wrapped = await get_user(scope)

    async def __call__(self, scope, receive, send):
        scope = dict(scope)
        # Scope injection/mutation per this middleware's needs.
        self.populate_scope(scope)
        # Grab the finalized/resolved scope
        await self.resolve_scope(scope)

        return await super().__call__(scope, receive, send)

scope 함수의 user 데이터는 get_user(scope)를 통해 데이터를 가져오고 있었고 해당 함수는

@database_sync_to_async
def get_user(scope):
    """
    Return the user model instance associated with the given scope.
    If no user is retrieved, return an instance of `AnonymousUser`.
    """
    # postpone model import to avoid ImproperlyConfigured error before Django
    # setup is complete.
    from django.contrib.auth.models import AnonymousUser

    if "session" not in scope:
        raise ValueError(
            "Cannot find session in scope. You should wrap your consumer in "
            "SessionMiddleware."
        )
    session = scope["session"]
    user = None
    try:
        user_id = _get_user_session_key(session)
        backend_path = session[BACKEND_SESSION_KEY]
    except KeyError:
        pass
    else:
        if backend_path in settings.AUTHENTICATION_BACKENDS:
            backend = load_backend(backend_path)
            user = backend.get_user(user_id)
            # Verify the session
            if hasattr(user, "get_session_auth_hash"):
                session_hash = session.get(HASH_SESSION_KEY)
                session_hash_verified = session_hash and constant_time_compare(
                    session_hash, user.get_session_auth_hash()
                )
                if not session_hash_verified:
                    session.flush()
                    user = None
    return user or AnonymousUser()

세션 정보를 통해서 가져오고 있어서 제대로 user 데이터를 못 가져왔던 것이다.
이 부분을 해결하려면 저번 포스팅에서 언급했던 것 처럼 근본적으로 midleware를 내가 커스텀해서 사용하면 된다.