C#標準ライブラリでWebSocketクライアントを作る

C#でWebSocketクライアントを作りたい場合、websocket-sharpというライブラリがよく使われているが、標準ライブラリだけで賄いたい。

標準でSystem.Net.WebSockets名前空間にClientWebSocketクラスが用意されており、MSDNに使い方が載っている。

Windows 8 のネットワーク接続 - Windows 8 と WebSocket プロトコル

一応通信はできるのだが、あまりにも原始的なのでラッパークラスを作る。

using System;
using System.IO;
using System.Net.WebSockets;
using System.Threading;
using System.Threading.Tasks;

public class WebSocketClient
{
    private const int SendBufferSize = 1024;
    private const int ReceiveBufferSize = 1024;

    private ClientWebSocket Socket;

    public event EventHandler OnOpenHandler;
    public event EventHandler OnCloseHandler;
    public event EventHandler<string> OnTextHandler;
    public event EventHandler<byte[]> OnBinaryHandler;

    public Uri Uri { get; set; }

    public async Task SendTextAsync(string text)
    {
        using var memoryStream = new MemoryStream();
        using var streamWriter = new StreamWriter(memoryStream);

        await streamWriter.WriteAsync(text);
        await streamWriter.FlushAsync();

        memoryStream.Seek(0, SeekOrigin.Begin);

        while (memoryStream.Position < memoryStream.Length)
        {
            var buffer = new byte[SendBufferSize];

            var remaining = memoryStream.Length - memoryStream.Position;

            var bufferSize = SendBufferSize < remaining ? SendBufferSize : (int)remaining;

            await memoryStream.ReadAsync(buffer, 0, buffer.Length);

            await Socket.SendAsync(new ArraySegment<byte>(buffer, 0, bufferSize), WebSocketMessageType.Text, !(memoryStream.Position < memoryStream.Length), CancellationToken.None);
        }
    }

    public async Task SendBinaryAsync(byte[] binary)
    {
        using var memoryStream = new MemoryStream();

        memoryStream.Write(binary, 0, binary.Length);

        memoryStream.Seek(0, SeekOrigin.Begin);

        while (memoryStream.Position < memoryStream.Length)
        {
            var buffer = new byte[SendBufferSize];

            var remaining = memoryStream.Length - memoryStream.Position;

            var bufferSize = SendBufferSize < remaining ? SendBufferSize : (int)remaining;

            await memoryStream.ReadAsync(buffer, 0, buffer.Length);

            await Socket.SendAsync(new ArraySegment<byte>(buffer, 0, bufferSize), WebSocketMessageType.Binary, !(memoryStream.Position < memoryStream.Length), CancellationToken.None);
        }
    }

    public async Task ConnectAsync()
    {
        if (Socket?.State == WebSocketState.Open) return;

        Socket = new ClientWebSocket();

        await Socket.ConnectAsync(Uri, CancellationToken.None);

        OnOpenHandler(this, EventArgs.Empty);
    }

    public async Task CloseAsync()
    {
        if (Socket == null || Socket.State == WebSocketState.Closed) return;

        await Socket.CloseAsync(WebSocketCloseStatus.NormalClosure, "", CancellationToken.None);
    }

    public WebSocketClient(Uri uri)
    {
        Uri = uri;

        OnOpenHandler += async (sender, e) => 
        {
            using var memoryStream = new MemoryStream();
            using var streamReader = new StreamReader(memoryStream);

            WebSocketMessageType messageType = WebSocketMessageType.Text;

            while (Socket.State == WebSocketState.Open)
            {
                var buffer = new byte[ReceiveBufferSize];

                var result = await Socket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);

                if (messageType != result.MessageType)
                {
                    messageType = result.MessageType;

                    await memoryStream.FlushAsync();

                    memoryStream.Seek(0, SeekOrigin.Begin);
                    memoryStream.SetLength(0);
                }

                if (result.MessageType == WebSocketMessageType.Text)
                {
                    await memoryStream.WriteAsync(buffer, 0, result.Count);

                    if (result.EndOfMessage)
                    {
                        await memoryStream.FlushAsync();

                        memoryStream.Seek(0, SeekOrigin.Begin);

                        OnTextHandler(this, streamReader.ReadToEnd());

                        memoryStream.Seek(0, SeekOrigin.Begin);
                        memoryStream.SetLength(0);
                    }
                }
                else if (result.MessageType == WebSocketMessageType.Binary)
                {
                    await memoryStream.WriteAsync(buffer, 0, result.Count);

                    if (result.EndOfMessage)
                    {
                        await memoryStream.FlushAsync();

                        memoryStream.Seek(0, SeekOrigin.Begin);

                        OnBinaryHandler(this, memoryStream.ToArray());

                        memoryStream.Seek(0, SeekOrigin.Begin);
                        memoryStream.SetLength(0);
                    }
                }
                else if (result.MessageType == WebSocketMessageType.Close)
                {
                    OnCloseHandler(this, EventArgs.Empty);
                }
            }
        };
    }

    public WebSocketClient(string uri) : this(new Uri(uri))
    {

    }
}

var ws = new WebSocketClient("ws://localhost:3000");

ws.OnOpenHandler += async (event, e) =>
{
    ws.OnTextHandler += (event, e) =>
    {
        Console.WriteLine(e);
    };
    
    ws.SendText("aaaaaaaaaa");
    
    await ws.CloseAsync();
};

ws.ConnectAsync();

LINQで短くする

using System;
using System.Collections.Generic;
using System.Linq;
using System.Net.WebSockets;
using System.Text;
using System.Threading;
using System.Threading.Tasks;

public class WebSocketClient
{
    private const int SendBufferSize = 1024;
    private const int ReceiveBufferSize = 1024;

    private ClientWebSocket Socket;

    public event EventHandler OnOpenHandler;
    public event EventHandler OnCloseHandler;
    public event EventHandler<string> OnTextHandler;
    public event EventHandler<byte[]> OnBinaryHandler;

    public Uri Uri { get; set; }

    public async Task SendTextAsync(string text)
    {
        var values = Encoding.UTF8.GetBytes(text).Select((value, index) => (value, index)).GroupBy(a => a.index / SendBufferSize).Select(a => a.Select(b => b.value).ToArray()).ToArray();

        foreach (var value in values)
        {
            await Socket.SendAsync(new ArraySegment<byte>(value), WebSocketMessageType.Text, value == values.Last(), CancellationToken.None);
        }
    }

    public async Task SendBinaryAsync(byte[] binary)
    {
        var values = binary.Select((value, index) => (value, index)).GroupBy(a => a.index / SendBufferSize).Select(a => a.Select(b => b.value).ToArray()).ToArray();

        foreach (var value in values)
        {
            await Socket.SendAsync(new ArraySegment<byte>(value), WebSocketMessageType.Binary, value == values.Last(), CancellationToken.None);
        }
    }

    public async Task ConnectAsync()
    {
        if (Socket?.State == WebSocketState.Open) return;

        Socket = new ClientWebSocket();

        await Socket.ConnectAsync(Uri, CancellationToken.None);

        OnOpenHandler(this, EventArgs.Empty);
    }

    public async Task CloseAsync()
    {
        if (Socket == null || Socket.State == WebSocketState.Closed) return;

        await Socket.CloseAsync(WebSocketCloseStatus.NormalClosure, "", CancellationToken.None);
    }

    public WebSocketClient(Uri uri)
    {
        Uri = uri;

        OnOpenHandler += async (sender, e) => 
        {
            WebSocketMessageType messageType = WebSocketMessageType.Text;

            var bytes = new List<byte>();

            while (Socket.State == WebSocketState.Open)
            {
                var buffer = new byte[ReceiveBufferSize];

                var result = await Socket.ReceiveAsync(new ArraySegment<byte>(buffer), CancellationToken.None);

                if (messageType != result.MessageType)
                {
                    messageType = result.MessageType;

                    bytes.Clear();
                }

                if (result.MessageType == WebSocketMessageType.Text)
                {
                    bytes.AddRange(buffer.Take(result.Count));

                    if (result.EndOfMessage)
                    {
                        OnTextHandler(this, Encoding.UTF8.GetString(bytes.ToArray()));

                        bytes.Clear();
                    }
                }
                else if (result.MessageType == WebSocketMessageType.Binary)
                {
                    bytes.AddRange(buffer.Take(result.Count));

                    if (result.EndOfMessage)
                    {
                        OnBinaryHandler(this, bytes.ToArray());

                        bytes.Clear();
                    }
                }
                else if (result.MessageType == WebSocketMessageType.Close)
                {
                    OnCloseHandler(this, EventArgs.Empty);
                }
            }
        };
    }

    public WebSocketClient(string uri) : this(new Uri(uri))
    {

    }
}

コメント

コメントを投稿

このブログの人気の投稿

fontconfigの設定

イメージ
Linuxでのフォント設定はfontconfigによって行われています。 fontconfigは、自動でその文字を表示できるようにフォントを選んでくれるのですが、フォントが複数ある場合自分の思うようにフォントが表示されないことがあります。 今回は、Noto Fontsをシステム標準のフォントに設定します。 一般的にはテーマからフォントの指定などを行うのが一般的だと思いますが、中にはそれに従わないアプリケーションもあります。 例えば、Firefoxではインターフェイスはデスクトップ環境のフォント設定に従うのですが、Webページのほうはフォントが指定されています。 Linuxでは一般的に明朝体にserif,ゴシック体にsans,等幅フォントにmonospaceが初期設定になっています。 fontconfigは、日本語フォントがあるとこれらのフォントを置き換えてくれます。 fontconfigはVLゴシックやipaフォントなどのメジャーなフォントは自動で標準のフォントに設定してくれるようになっていますが、Noto Fontsなどの新しいフォントやマイナーなフォントは自動では設定してくれません。 その、優先順位は/etc/fonts/font.avail/65-nonlatin.conf に記述されており、fontconfigはそれに従います。 この順序は今のところ、日本語フォントだけを抽出すると MSゴシック>梅ゴシック>VLゴシック>IPAMona>IPAゴシック>さざなみゴシック>東風ゴシック というようになっています。 このファイルを直接書き換えても良いのですが、fontconfigのアップデートで書き換えられてしまったり、アップデートに追従できなくなる恐れがあります。 fontconfigの設定を手動で書き換えたい場合、 /etc/fonts/local.conf にその内容を記述することで有効化することができます。フォント設定として、local.conf は最優先されます。 local.conf に、serifはNoto Serif、sansはNoto Sans、monospaceはNoto Mono、と明示的に設定することで、アプリケーションごとに統一したフォントを...
続きを読む

VLCでBlu-rayを再生

先日VLCでのBlu-rayの再生について書きましたが、古いバージョンのAACSにしか対応していませんでした。 VLC media playerでBlu-rayディスクを再生する - ぽぬぽぬ BDのリッピングソフトであるMakeMKVに付属しているlibmmbdがlibaacsと互換性があり、新しいバージョンのAACSにも対応しています。(2016/06/13現在AACS v60に対応)
続きを読む

UEFIのブートオーダーを一時的に変更する

UEFIにはブートマネージャー機能が内臓されている。 UEFI環境ではまず、UEFIのブートマネージャーがOSのブートマネージャーを起動し、OSを起動させている。 UEFIのブートマネージャには、次回起動時のブートオーダーを指定できるBootNextという機能がある。 Windows起動中、別のシステムで再起動したい場合、 設定→更新とセキュリティ→回復→PCの起動をカスタマイズ　から起動したいシステムを指定して再起動することができる。 Windowsを優先的に起動する設定にしており、Linuxを起動したいときにこの段階を踏むのがめんどくさいので、バッチファイルを作ってみる。 Windowsのブートローダーを管理するBCDEditでUEFIもいじれるらしい。 まず、BCDEditでUEFIのブートエントリを表示してみる。コマンドプロンプトを管理者権限で起動し、以下のコマンドを入力。 >bcdedit /enum firmware ファームウェアのブート マネージャー -------------------------------- identifier {fwbootmgr} displayorder {bootmgr} {f745f75a-0195-11e8-84b8-806e6f6e6963} {32a8102e-da80-11e7-8406-ce8e5306cb4c} {f745f759-0195-11e8-84b8-806e6f6e6963} timeout 1 Windows ブート マネージャー -------------------------------- identifier {bootmgr} device partition=\Device\HarddiskVolume2 path \EFI\MICROSOFT\BOOT\BOOTMGFW.EFI description ...
続きを読む