Add GetAttributes to ISapServer

huysentruitw · Sep 1, 2022 · c60998d · c60998d
1 parent 0e651de
commit c60998d
Show file tree

Hide file tree

Showing 9 changed files with 103 additions and 5 deletions.
diff --git a/src/SapNwRfc/ISapServer.cs b/src/SapNwRfc/ISapServer.cs
@@ -17,6 +17,12 @@ public interface ISapServer : IDisposable
         /// </summary>
         event EventHandler<SapServerStateChangeEventArgs> StateChange;
 
+        /// <summary>
+        /// Gets the server attributes.
+        /// </summary>
+        /// <returns>The server attributes.</returns>
+        SapServerAttributes GetAttributes();
+
         /// <summary>
         /// Launches the server.
         /// </summary>

diff --git a/src/SapNwRfc/Internal/Interop/RfcInterop.cs b/src/SapNwRfc/Internal/Interop/RfcInterop.cs
@@ -370,6 +370,12 @@ public virtual RfcResultCode LaunchServer(IntPtr rfcHandle, out RfcErrorInfo err
         public virtual RfcResultCode ShutdownServer(IntPtr rfcHandle, uint timeout, out RfcErrorInfo errorInfo)
             => RfcShutdownServer(rfcHandle, timeout, out errorInfo);
 
+        [DllImport(SapNwRfcDllName)]
+        private static extern RfcResultCode RfcGetServerAttributes(IntPtr rfcHandle, out RfcServerAttributes serverAttributes, out RfcErrorInfo errorInfo);
+
+        public virtual RfcResultCode GetServerAttributes(IntPtr rfcHandle, out RfcServerAttributes serverAttributes, out RfcErrorInfo errorInfo)
+            => RfcGetServerAttributes(rfcHandle, out serverAttributes, out errorInfo);
+
         [DllImport(SapNwRfcDllName)]
         private static extern RfcResultCode RfcGetServerContext(IntPtr rfcHandle, out RfcServerContext serverContext, out RfcErrorInfo errorInfo);
 

diff --git a/src/SapNwRfc/Internal/Interop/RfcServerAttributes.cs b/src/SapNwRfc/Internal/Interop/RfcServerAttributes.cs
@@ -0,0 +1,26 @@
+using System;
+using System.Runtime.InteropServices;
+
+namespace SapNwRfc.Internal.Interop
+{
+    [StructLayout(LayoutKind.Sequential, CharSet = CharSet.Unicode)]
+    internal struct RfcServerAttributes
+    {
+        public IntPtr ServerName;
+
+        [MarshalAs(UnmanagedType.I4)]
+        public SapRfcProtocolType Type;
+
+        [MarshalAs(UnmanagedType.U4)]
+        public uint RegistrationCount;
+
+        [MarshalAs(UnmanagedType.I4)]
+        public SapRfcServerState State;
+
+        [MarshalAs(UnmanagedType.U4)]
+        public uint CurrentBusyCount;
+
+        [MarshalAs(UnmanagedType.U4)]
+        public uint PeakBusyCount;
+    }
+}
diff --git a/src/SapNwRfc/Internal/Interop/RfcStateChange.cs b/src/SapNwRfc/Internal/Interop/RfcStateChange.cs
@@ -6,9 +6,9 @@ namespace SapNwRfc.Internal.Interop
     internal struct RfcStateChange
     {
         [MarshalAs(UnmanagedType.I4)]
-        public SapServerState OldState;
+        public SapRfcServerState OldState;
 
         [MarshalAs(UnmanagedType.I4)]
-        public SapServerState NewState;
+        public SapRfcServerState NewState;
     }
 }
diff --git a/src/SapNwRfc/SapRfcProtocolType.cs b/src/SapNwRfc/SapRfcProtocolType.cs
@@ -0,0 +1,19 @@
+namespace SapNwRfc
+{
+    /// <summary>
+    /// Represents the SAP RFC server protocol.
+    /// </summary>
+    public enum SapRfcProtocolType
+    {
+        RFC_UNKOWN,
+        RFC_CLIENT,
+        RFC_STARTED_SERVER,
+        RFC_REGISTERED_SERVER,
+        RFC_MULTI_COUNT_REGISTERED_SERVER,
+        RFC_TCP_SOCKET_CLIENT,
+        RFC_TCP_SOCKET_SERVER,
+        RFC_WEBSOCKET_CLIENT,
+        RFC_WEBSOCKET_SERVER,
+        RFC_PROXY_WEBSOCKET_CLIENT,
+    }
+}
diff --git a/src/SapNwRfc/RfcServerState.cs → src/SapNwRfc/SapRfcServerState.cs b/src/SapNwRfc/RfcServerState.cs → src/SapNwRfc/SapRfcServerState.cs
@@ -3,7 +3,7 @@ namespace SapNwRfc
     /// <summary>
     /// Represents the SAP RFC server states.
     /// </summary>
-    public enum SapServerState
+    public enum SapRfcServerState
     {
         RFC_SERVER_INITIAL,
         RFC_SERVER_STARTING,

diff --git a/src/SapNwRfc/SapServer.cs b/src/SapNwRfc/SapServer.cs
@@ -147,6 +147,19 @@ private void ServerStateChangeListener(IntPtr serverHandle, in RfcStateChange st
             _stateChange?.Invoke(this, new SapServerStateChangeEventArgs(stateChange));
         }
 
+        /// <inheritdoc cref="ISapServer"/>
+        public SapServerAttributes GetAttributes()
+        {
+            RfcResultCode resultCode = _interop.GetServerAttributes(
+                rfcHandle: _rfcServerHandle,
+                serverAttributes: out RfcServerAttributes serverAttributes,
+                errorInfo: out RfcErrorInfo errorInfo);
+
+            resultCode.ThrowOnError(errorInfo);
+
+            return new SapServerAttributes(serverAttributes);
+        }
+
         /// <inheritdoc cref="ISapServer"/>
         public void Launch()
         {

diff --git a/src/SapNwRfc/SapServerAttributes.cs b/src/SapNwRfc/SapServerAttributes.cs
@@ -0,0 +1,28 @@
+using System.Runtime.InteropServices;
+using SapNwRfc.Internal.Interop;
+
+namespace SapNwRfc
+{
+    public sealed class SapServerAttributes
+    {
+        private readonly RfcServerAttributes _serverAttributes;
+
+        internal SapServerAttributes(RfcServerAttributes serverAttributes)
+        {
+            _serverAttributes = serverAttributes;
+            ServerName = Marshal.PtrToStringAuto(serverAttributes.ServerName);
+        }
+
+        public string ServerName { get; }
+
+        public SapRfcProtocolType Type => _serverAttributes.Type;
+
+        public uint RegistrationCount => _serverAttributes.RegistrationCount;
+
+        public SapRfcServerState State => _serverAttributes.State;
+
+        public uint CurrentBusyCount => _serverAttributes.CurrentBusyCount;
+
+        public uint PeakBusyCount => _serverAttributes.PeakBusyCount;
+    }
+}
diff --git a/src/SapNwRfc/SapServerStateChangeEventArgs.cs b/src/SapNwRfc/SapServerStateChangeEventArgs.cs
@@ -17,11 +17,11 @@ internal SapServerStateChangeEventArgs(RfcStateChange stateChange)
         /// <summary>
         /// Gets the old server state.
         /// </summary>
-        public SapServerState OldState { get; }
+        public SapRfcServerState OldState { get; }
 
         /// <summary>
         /// Gets the new server state.
         /// </summary>
-        public SapServerState NewState { get; }
+        public SapRfcServerState NewState { get; }
     }
 }