Microsoft MVP성태의 닷넷 이야기
닷넷: 2198. C# - Reflection을 이용한 ClientWebSocket의 Ping 호출 [링크 복사], [링크+제목 복사],
조회: 10689
글쓴 사람
정성태 (techsharer at outlook.com)
홈페이지
첨부 파일
(연관된 글이 1개 있습니다.)
(시리즈 글이 4개 있습니다.)
닷넷: 2188. C# - ASP.NET Core SignalR로 구현하는 채팅 서비스 예제
; https://www.sysnet.pe.kr/2/0/13504

닷넷: 2189. C# - WebSocket 클라이언트를 닷넷으로 구현하는 예제 (System.Net.WebSockets)
; https://www.sysnet.pe.kr/2/0/13505

닷넷: 2197. C# - ClientWebSocket의 Ping, Pong 처리
; https://www.sysnet.pe.kr/2/0/13518

닷넷: 2198. C# - Reflection을 이용한 ClientWebSocket의 Ping 호출
; https://www.sysnet.pe.kr/2/0/13519




C# - Reflection을 이용한 ClientWebSocket의 Ping 호출

지난 글에 Ping/Pong이 ClientWebSocket에 의해 어떻게 처리되는지 살펴봤습니다.

C# - ClientWebSocket의 Ping, Pong 처리
; https://www.sysnet.pe.kr/2/0/13518

언급했듯이, (서버 측 nodejs는 Ping을 전송하고 Pong을 받는데) C#의 ClientWebSocket은 무조건 Pong을 전송하는 식입니다. 그렇다면 혹시 Ping을 전송하는 방법은 없을까요?

이미 설명한 것처럼, 모든 처리는 ManagedWebSocket 내부에서 자동으로 수행되는데다 ClientWebSocket이 호출하는 Send 메서드는 내부적으로 Binary와 Text 유형만 보낼 수 있도록 고정했기 때문에,

public override ValueTask SendAsync(ReadOnlyMemory<byte> buffer, WebSocketMessageType messageType, WebSocketMessageFlags messageFlags, CancellationToken cancellationToken)
{
    // ...[생략]...
    MessageOpcode opcode;

    if (_lastSendWasFragment)
    {
        if (_lastSendHadDisableCompression != disableCompression)
        {
            throw new ArgumentException(SR.net_WebSockets_Argument_MessageFlagsHasDifferentCompressionOptions, nameof(messageFlags));
        }
        opcode = MessageOpcode.Continuation;
    }
    else
    {
        opcode = messageType == WebSocketMessageType.Binary ? MessageOpcode.Binary : MessageOpcode.Text;
    }

    ValueTask t = SendFrameAsync(opcode, endOfMessage, disableCompression, buffer, cancellationToken);
    // ...[생략]...

    return t;
}

딱히 제어를 할 수 있는 여지가 없습니다. 정 원한다면, 가령 테스트를 위해서 필요하다면 Reflection을 이용해 처리하는 수밖에 없습니다.

그다지 어렵지 않으니 간단하게 코드로 만들어 볼까요? ^^ 우선 SocketHandle과 ManagedWebSocket을 차례로 구하는 것부터 시작해야 합니다.

private static object? GetSocketHandle(ClientWebSocket ws)
{
    Type type = ws.GetType();
    FieldInfo? fi = type.GetField("_innerWebSocket", BindingFlags.NonPublic | BindingFlags.Instance);
    if (fi == null)
    {
        return null;
    }

    return fi.GetValue(ws);
}

private static WebSocket? GetManagedSocket(ClientWebSocket ws)
{
    var webSocketHandle = GetSocketHandle(ws);
    if (webSocketHandle == null)
    {
        return null;
    }

    Type type = webSocketHandle.GetType();
    return type.GetProperty("WebSocket", BindingFlags.Public | BindingFlags.Instance)?.GetValue(webSocketHandle)
        as WebSocket;
}

이후 남은 작업은, ManagedWebSocket이 구현한 Pong 코드를 참조해,

private async ValueTask HandleReceivedPingPongAsync(MessageHeader header, CancellationToken cancellationToken)
{
    // ...[생략]...

    // If this was a ping, send back a pong response.
    if (header.Opcode == MessageOpcode.Ping)
    {
        // ...[생략]...
        await SendFrameAsync(
            MessageOpcode.Pong,)
            endOfMessage: true,
            disableCompression: true,
            _receiveBuffer.Slice(_receiveBufferOffset, (int)header.PayloadLength),
            cancellationToken).ConfigureAwait(false);
    }

    // ...[생략]...
}

우리도 저 SendFrameAsync 메서드를 이용해 MessageOpcode.Ping (0x9)에 해당하는 값을 전송하도록 맞춰주기만 하면 됩니다.

private static ValueTask SendPingFrameAsync(WebSocket managedSocket)
{
    Type type = managedSocket.GetType();
    MethodInfo? targetMi = null;

    foreach (MethodInfo mi in type.GetMethods(BindingFlags.NonPublic | BindingFlags.Instance))
    {
        if (mi.Name == "SendFrameAsync" && mi.GetParameters().Length == 5)
        {
            targetMi = mi;
            break;
        }
    }

    if (targetMi == null)
    {
        throw new NullReferenceException("SendFrameAsync not found");
    }

    /*
    private enum MessageOpcode : byte
    {
        Continuation = 0x0,
        Text = 0x1,
        Binary = 0x2,
        Close = 0x8,
        Ping = 0x9,
        Pong = 0xA
    }
    */

    object? objValue = targetMi.Invoke(managedSocket, new object?[] { (byte)0x9, true, true, ReadOnlyMemory<byte>.Empty, CancellationToken.None });
    if (objValue == null)
    {
        throw new NullReferenceException("SendFrameAsync returned null");
    }

    return (ValueTask)objValue;
}

private static void SendKeepAliveFrame(WebSocket managedSocket)
{
    ValueTask t = SendPingFrameAsync(managedSocket);
    if (t.IsCompletedSuccessfully)
    {
        t.GetAwaiter().GetResult();
    }
    else
    {
        // "Observe" any exception, ignoring it to prevent the unobserved exception event from being raised.
        t.AsTask().ContinueWith(static p => { _ = p.Exception; },
            CancellationToken.None,
            TaskContinuationOptions.OnlyOnFaulted | TaskContinuationOptions.ExecuteSynchronously,
            TaskScheduler.Default);
    }
}

끝이군요. ^^ 이제 위에서 구현한 SendKeepAliveFrame 메서드를 ClientWebSocket에 적용해 다음과 같이 일부러 Ping을 전송하는 코드를 만들 수 있습니다.

string url = "ws://localhost:18000/";

var connectTimeout = new CancellationTokenSource();
connectTimeout.CancelAfter(2000);

System.Net.WebSockets.ClientWebSocket ws = new System.Net.WebSockets.ClientWebSocket();
ws.Options.KeepAliveInterval = TimeSpan.FromSeconds(0); // Pong 전송 코드 수행을 막기 위해 일부러 0으로 설정

await ws.ConnectAsync(new Uri(url), connectTimeout.Token);

if (ws.State != System.Net.WebSockets.WebSocketState.Open)
{
    Console.WriteLine($"Failed to connect: {url}");
    return;
}

_ = Task.Run(() =>
{
    WebSocket? managedSocket = GetManagedSocket(ws);
    if (managedSocket == null)
    {
        return;
    }

    while (true)
    {
        SendKeepAliveFrame(managedSocket);
        Thread.Sleep(5000);
    }
});

끝입니다. 잘 동작하는지 확인을 위해 nodejs 서버 코드에 로그를 남겨 보면,

var { WebSocketServer } = require('ws')

const sockserver = new WebSocketServer({ port: 18000 })

sockserver.on('connection', ws => {
    interval_id = 0;

    console.log('New client connected!')

    ws.send('connection established')

    ws.on('ping', () => {
        console.log('ping received from client');
    });

    // ...[생략]...
})

정상적으로 "ping received ..." 메시지가 출력되는 것을 볼 수 있습니다. 당연히 nodejs는 저렇게 ping을 수신하면 응답으로 pong을 다시 클라이언트로 전송하게 될 것입니다. 아쉽게도 수신 여부를 ClientWebSocket으로 알 수는 없지만, 비주얼 스튜디오의 닷넷 소스코드 디버깅 기능을 이용하면,

비주얼 스튜디오 2022를 이용한 (소스 코드가 없는) 닷넷 모듈 디버깅 - "외부 원본(External Sources)"
; https://www.sysnet.pe.kr/2/0/13109

ReceiveAsyncPrivate 메서드 내부의 MessageOpcode.Pong 조건절에 BP를 설정해 확인해 볼 수는 있습니다.

[AsyncMethodBuilder(typeof(PoolingAsyncValueTaskMethodBuilder<>))]
private async ValueTask ReceiveAsyncPrivate(Memory payloadBuffer, CancellationToken cancellationToken)
{
    // ...[생략]...

    while (true) // in case we get control frames that should be ignored from the user's perspective
    {
        // ...[생략]...
        await _stream.ReadAsync(Memory.Empty, cancellationToken).ConfigureAwait(false);

        // ...[생략]...
        string? headerErrorMessage = TryParseMessageHeaderFromReceiveBuffer(out header);

        // 아래의 코드에 Breakpoint 설정
        if (header.Opcode == MessageOpcode.Ping || header.Opcode == MessageOpcode.Pong)
        {
            await HandleReceivedPingPongAsync(header, cancellationToken).ConfigureAwait(false);
            continue;
        }

        // ...[생략]...
    }

    // ...[생략]...
}

(첨부 파일은 이 글의 예제 코드를 포함합니다.)




[이 글에 대해서 여러분들과 의견을 공유하고 싶습니다. 틀리거나 미흡한 부분 또는 의문 사항이 있으시면 언제든 댓글 남겨주십시오.]

[연관 글]






[최초 등록일: ]
[최종 수정일: 1/10/2024]

Creative Commons License
이 저작물은 크리에이티브 커먼즈 코리아 저작자표시-비영리-변경금지 2.0 대한민국 라이센스에 따라 이용하실 수 있습니다.
by SeongTae Jeong, mailto:techsharer at outlook.com

비밀번호

댓글 작성자
 




... 106  107  108  109  110  111  112  [113]  114  115  116  117  118  119  120  ...
NoWriterDateCnt.TitleFile(s)
11133정성태1/20/201724831.NET Framework: 635. C# 개발자를 위한 Win32 DLL export 함수의 호출 규약 (2) - x86 환경의 __fastcall [1]파일 다운로드1
11132정성태1/19/201736659.NET Framework: 634. C# 개발자를 위한 Win32 DLL export 함수의 호출 규약 (1) - x86 환경에서의 __cdecl, __stdcall에 대한 Name mangling [1]파일 다운로드1
11131정성태1/13/201725472.NET Framework: 633. C# - IL 코드 분석을 위한 팁 [2]
11130정성태1/11/201726081.NET Framework: 632. x86 실행 환경에서 SECURITY_ATTRIBUTES 구조체를 CreateEvent에 전달할 때 예외 발생파일 다운로드1
11129정성태1/11/201730210.NET Framework: 631. async/await에 대한 "There Is No Thread" 글의 부가 설명 [9]파일 다운로드1
11128정성태1/9/201723993.NET Framework: 630. C# - Interlocked.CompareExchange 사용 예제 [3]파일 다운로드1
11127정성태1/8/201724444기타: 63. (개발자를 위한) Visual Studio의 "with MSDN" 라이선스 설명
11126정성태1/7/201729144기타: 62. Edge 웹 브라우저의 즐겨찾기(Favorites)를 편집/백업/복원하는 방법 [1]파일 다운로드1
11125정성태1/7/201725978개발 환경 구성: 310. IIS - appcmd.exe를 이용해 특정 페이지에 클라이언트 측 인증서를 제출하도록 설정하는 방법
11124정성태1/4/201729379개발 환경 구성: 309. 3년짜리 유효 기간을 제공하는 StartSSL [2]
11123정성태1/3/201724804.NET Framework: 629. .NET Core의 dotnet.exe CLI 명령어 확장 방법 [1]
11122정성태1/3/201723483.NET Framework: 628. TransactionScope에 사용자 정의 트랜잭션을 참여시키는 방법 [2]파일 다운로드1
11121정성태1/1/201721431개발 환경 구성: 308. "ASP.NET Core Web Application (.NET Core)"와 "ASP.NET Core Web Application (.NET Framework)" 차이점
11120정성태12/25/201628522개발 환경 구성: 307. ASP.NET Core Web Application을 IIS에서 호스팅하는 방법
11119정성태12/23/201649940개발 환경 구성: 306. Visual Studio Code에서 Python 개발 환경 구성 [2]
11118정성태12/22/201637823오류 유형: 374. Python 64비트 설치 시 0x80070659 오류 발생 [3]
11117정성태12/21/201623801웹: 35. nopCommerce 예제 사이트 구성 방법
11116정성태12/21/201625820디버깅 기술: 84. NopCommerce의 Autofac 부하(CPU, Memory) [2]
11115정성태12/21/201628838Windows: 133. 윈도우 서버 2016에서 플래시가 동작하지 않는 경우 [2]
11114정성태12/19/201638934Windows: 132. 역슬래시(backslash) 문자가 왜 통화 표기 문자(한글인 경우 "\")로 보일까요? [2]
11113정성태12/6/201621712오류 유형: 373. ICOMAdminCatalog::GetCollection에서 CO_E_ISOLEVELMISMATCH(0x8004E02F) 오류 발생파일 다운로드1
11112정성태11/23/201626877오류 유형: 372. MySQL 서비스가 올라오지 않는 경우 - Error 1067
11111정성태11/23/201635503.NET Framework: 627. C++로 만든 DLL을 C#에서 사용하기 [2]
11110정성태11/17/201623016.NET Framework: 626. Commit 메모리가 낮은 상황에서도 메모리 부족(Out-of-memory) 예외 발생 [2]
11109정성태11/17/201623008.NET Framework: 625. ASP.NET에서 System.Web.HttpApplication 인스턴스는 다중으로 생성됩니다.
11108정성태11/13/201622065.NET Framework: 624. WPF - Line 요소를 Canvas에 위치시켰을 때 흐림(blur) 현상파일 다운로드1
... 106  107  108  109  110  111  112  [113]  114  115  116  117  118  119  120  ...