/// <summary> /// Registers a device described by the message. /// </summary> /// <param name="message">The provisioning message.</param> /// <param name="cancellationToken">The cancellation token.</param> /// <returns>The registration result.</returns> public override async Task <DeviceRegistrationResult> RegisterAsync( ProvisioningTransportRegisterMessage message, CancellationToken cancellationToken) { if (Logging.IsEnabled) { Logging.Enter(this, $"{nameof(ProvisioningTransportHandlerAmqp)}.{nameof(RegisterAsync)}"); } if (message == null) { throw new ArgumentNullException(nameof(message)); } cancellationToken.ThrowIfCancellationRequested(); try { AmqpAuthStrategy authStrategy; if (message.Security is SecurityProviderTpm) { authStrategy = new AmqpAuthStrategyTpm((SecurityProviderTpm)message.Security); } else if (message.Security is SecurityProviderX509) { authStrategy = new AmqpAuthStrategyX509((SecurityProviderX509)message.Security); } else if (message.Security is SecurityProviderSymmetricKey) { authStrategy = new AmqpAuthStrategySymmetricKey((SecurityProviderSymmetricKey)message.Security); } else { throw new NotSupportedException( $"{nameof(message.Security)} must be of type {nameof(SecurityProviderTpm)}, " + $"{nameof(SecurityProviderX509)} or {nameof(SecurityProviderSymmetricKey)}"); } if (Logging.IsEnabled) { Logging.Associate(authStrategy, this); } bool useWebSocket = (FallbackType == TransportFallbackType.WebSocketOnly); var builder = new UriBuilder() { Scheme = useWebSocket ? WebSocketConstants.Scheme : AmqpConstants.SchemeAmqps, Host = message.GlobalDeviceEndpoint, Port = Port, }; string registrationId = message.Security.GetRegistrationID(); string linkEndpoint = $"{message.IdScope}/registrations/{registrationId}"; using AmqpClientConnection connection = authStrategy.CreateConnection(builder.Uri, message.IdScope); await authStrategy.OpenConnectionAsync(connection, s_timeoutConstant, useWebSocket, Proxy, RemoteCertificateValidationCallback).ConfigureAwait(false); cancellationToken.ThrowIfCancellationRequested(); await CreateLinksAsync( connection, linkEndpoint, message.ProductInfo).ConfigureAwait(false); cancellationToken.ThrowIfCancellationRequested(); string correlationId = Guid.NewGuid().ToString(); DeviceRegistration deviceRegistration = (message.Payload != null && message.Payload.Length > 0) ? new DeviceRegistration { Payload = new JRaw(message.Payload) } : null; RegistrationOperationStatus operation = await RegisterDeviceAsync(connection, correlationId, deviceRegistration).ConfigureAwait(false); // Poll with operationId until registration complete. int attempts = 0; string operationId = operation.OperationId; // Poll with operationId until registration complete. while (string.CompareOrdinal(operation.Status, RegistrationOperationStatus.OperationStatusAssigning) == 0 || string.CompareOrdinal(operation.Status, RegistrationOperationStatus.OperationStatusUnassigned) == 0) { cancellationToken.ThrowIfCancellationRequested(); await Task.Delay( operation.RetryAfter ?? RetryJitter.GenerateDelayWithJitterForRetry(s_defaultOperationPoolingInterval), cancellationToken).ConfigureAwait(false); try { operation = await OperationStatusLookupAsync( connection, operationId, correlationId).ConfigureAwait(false); } catch (ProvisioningTransportException e) when(e.ErrorDetails is ProvisioningErrorDetailsAmqp amqp && e.IsTransient) { operation.RetryAfter = amqp.RetryAfter; } attempts++; } if (string.CompareOrdinal(operation.Status, RegistrationOperationStatus.OperationStatusAssigned) == 0) { authStrategy.SaveCredentials(operation); } await connection.CloseAsync(s_timeoutConstant).ConfigureAwait(false); return(ConvertToProvisioningRegistrationResult(operation.RegistrationState)); } catch (Exception ex) when(!(ex is ProvisioningTransportException)) { if (Logging.IsEnabled) { Logging.Error( this, $"{nameof(ProvisioningTransportHandlerAmqp)} threw exception {ex}", nameof(RegisterAsync)); } throw new ProvisioningTransportException($"AMQP transport exception", ex, true); } finally { if (Logging.IsEnabled) { Logging.Exit(this, $"{nameof(ProvisioningTransportHandlerAmqp)}.{nameof(RegisterAsync)}"); } } }
public override async Task <DeviceRegistrationResult> RegisterAsync( ProvisioningTransportRegisterMessage message, CancellationToken cancellationToken) { if (Logging.IsEnabled) { Logging.Enter(this, $"{nameof(ProvisioningTransportHandlerAmqp)}.{nameof(RegisterAsync)}"); } cancellationToken.ThrowIfCancellationRequested(); try { AmqpAuthStrategy authStrategy; if (message.Security is SecurityClientHsmTpm) { authStrategy = new AmqpAuthStrategyTpm((SecurityClientHsmTpm)message.Security); } else if (message.Security is SecurityClientHsmX509) { authStrategy = new AmqpAuthStrategyX509((SecurityClientHsmX509)message.Security); } else { throw new NotSupportedException( $"{nameof(message.Security)} must be of type {nameof(SecurityClientHsmTpm)} " + $"or {nameof(SecurityClientHsmX509)}"); } if (Logging.IsEnabled) { Logging.Associate(authStrategy, this); } bool useWebSocket = (FallbackType == TransportFallbackType.WebSocketOnly); var builder = new UriBuilder() { Scheme = useWebSocket ? WebSocketConstants.Scheme : AmqpConstants.SchemeAmqps, Host = message.GlobalDeviceEndpoint, Port = useWebSocket ? WebSocketConstants.Port : AmqpConstants.DefaultSecurePort }; string registrationId = message.Security.GetRegistrationID(); string linkEndpoint = $"{message.IdScope}/registrations/{registrationId}"; AmqpClientConnection connection = authStrategy.CreateConnection(builder.Uri, linkEndpoint); await authStrategy.OpenConnectionAsync(connection, TimeoutConstant, useWebSocket).ConfigureAwait(false); cancellationToken.ThrowIfCancellationRequested(); await CreateLinksAsync(connection, linkEndpoint, message.ProductInfo).ConfigureAwait(false); cancellationToken.ThrowIfCancellationRequested(); string correlationId = new Guid().ToString(); RegistrationOperationStatus operation = await RegisterDeviceAsync(connection, correlationId).ConfigureAwait(false); // Poll with operationId until registration complete. int attempts = 0; string operationId = operation.OperationId; // Poll with operationId until registration complete. while (string.CompareOrdinal(operation.Status, RegistrationOperationStatus.OperationStatusAssigning) == 0 || string.CompareOrdinal(operation.Status, RegistrationOperationStatus.OperationStatusUnassigned) == 0) { cancellationToken.ThrowIfCancellationRequested(); await Task.Delay( operation.RetryAfter ?? DefaultOperationPoolingIntervalMilliseconds).ConfigureAwait(false); cancellationToken.ThrowIfCancellationRequested(); operation = await OperationStatusLookupAsync( connection, operationId, correlationId).ConfigureAwait(false); attempts++; } if (string.CompareOrdinal(operation.Status, RegistrationOperationStatus.OperationStatusAssigned) == 0) { authStrategy.SaveCredentials(operation); } return(ConvertToProvisioningRegistrationResult(operation.RegistrationStatus)); } // TODO: Catch only expected exceptions from Amqp. catch (Exception ex) { if (Logging.IsEnabled) { Logging.Error( this, $"{nameof(ProvisioningTransportHandlerAmqp)} threw exception {ex}", nameof(RegisterAsync)); } // TODO: Extract trackingId from the exception. throw new ProvisioningTransportException($"AMQP transport exception", true, "", ex); } finally { if (Logging.IsEnabled) { Logging.Exit(this, $"{nameof(ProvisioningTransportHandlerAmqp)}.{nameof(RegisterAsync)}"); } } }