| | | 1 | | // Copyright (c) ZeroC, Inc. |
| | | 2 | | |
| | | 3 | | using System.Collections.Concurrent; |
| | | 4 | | using System.IO.Pipelines; |
| | | 5 | | using System.Net.Security; |
| | | 6 | | |
| | | 7 | | namespace IceRpc.Transports.Coloc.Internal; |
| | | 8 | | |
| | | 9 | | /// <summary>Implements <see cref="IDuplexClientTransport" /> for the coloc transport.</summary> |
| | | 10 | | internal class ColocClientTransport : IDuplexClientTransport |
| | | 11 | | { |
| | | 12 | | /// <inheritdoc/> |
| | 541 | 13 | | public string Name => ColocTransport.Name; |
| | | 14 | | |
| | | 15 | | private readonly ConcurrentDictionary<ServerAddress, ColocListener> _listeners; |
| | | 16 | | private readonly ColocTransportOptions _options; |
| | | 17 | | |
| | | 18 | | /// <inheritdoc/> |
| | | 19 | | public IDuplexConnection CreateConnection( |
| | | 20 | | ServerAddress serverAddress, |
| | | 21 | | DuplexConnectionOptions duplexConnectionOptions, |
| | | 22 | | SslClientAuthenticationOptions? clientAuthenticationOptions) |
| | 514 | 23 | | { |
| | 514 | 24 | | if (clientAuthenticationOptions is not null) |
| | 0 | 25 | | { |
| | 0 | 26 | | throw new NotSupportedException("The Coloc client transport does not support SSL."); |
| | | 27 | | } |
| | | 28 | | |
| | 514 | 29 | | if ((serverAddress.Transport is string transport && transport != ColocTransport.Name) || |
| | 514 | 30 | | !ColocTransport.CheckParams(serverAddress)) |
| | 2 | 31 | | { |
| | 2 | 32 | | throw new ArgumentException( |
| | 2 | 33 | | $"The server address '{serverAddress}' contains parameters that are not valid for the Coloc client trans |
| | 2 | 34 | | nameof(serverAddress)); |
| | | 35 | | } |
| | | 36 | | |
| | 512 | 37 | | serverAddress = serverAddress with { Transport = Name }; |
| | | 38 | | |
| | 512 | 39 | | var localPipe = new Pipe(new PipeOptions( |
| | 512 | 40 | | pool: duplexConnectionOptions.Pool, |
| | 512 | 41 | | minimumSegmentSize: duplexConnectionOptions.MinSegmentSize, |
| | 512 | 42 | | pauseWriterThreshold: _options.PauseWriterThreshold, |
| | 512 | 43 | | resumeWriterThreshold: _options.ResumeWriterThreshold, |
| | 512 | 44 | | useSynchronizationContext: false)); |
| | 512 | 45 | | return new ClientColocConnection(serverAddress, localPipe, ConnectAsync); |
| | | 46 | | |
| | | 47 | | // The client connection connect operation calls this method to queue a connection establishment request with |
| | | 48 | | // the listener. The returned task is completed once the listener accepts the connection establishment request. |
| | | 49 | | Task<PipeReader> ConnectAsync(PipeReader clientPipeReader, CancellationToken cancellationToken) |
| | 483 | 50 | | { |
| | 483 | 51 | | if (_listeners.TryGetValue(serverAddress, out ColocListener? listener) && |
| | 483 | 52 | | listener.TryQueueConnect( |
| | 483 | 53 | | clientPipeReader, |
| | 483 | 54 | | cancellationToken, |
| | 483 | 55 | | out Task<PipeReader>? serverPipeReaderTask)) |
| | 474 | 56 | | { |
| | 474 | 57 | | return serverPipeReaderTask; |
| | | 58 | | } |
| | | 59 | | else |
| | 9 | 60 | | { |
| | 9 | 61 | | throw new IceRpcException(IceRpcError.ConnectionRefused); |
| | | 62 | | } |
| | 474 | 63 | | } |
| | 512 | 64 | | } |
| | | 65 | | |
| | 497 | 66 | | internal ColocClientTransport( |
| | 497 | 67 | | ConcurrentDictionary<ServerAddress, ColocListener> listeners, |
| | 497 | 68 | | ColocTransportOptions options) |
| | 497 | 69 | | { |
| | 497 | 70 | | _listeners = listeners; |
| | 497 | 71 | | _options = options; |
| | 497 | 72 | | } |
| | | 73 | | } |