getSocketType

swhitty · Nov 21, 2024 · 24cf9d2 · 24cf9d2
1 parent cca899e
commit 24cf9d2
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 0 deletions.
diff --git a/FlyingSocks/Sources/Socket.swift b/FlyingSocks/Sources/Socket.swift
@@ -99,6 +99,12 @@ public struct Socket: Sendable, Hashable {
         }
     }
 
+    public var socketType: SocketType {
+        get throws {
+            try SocketType(rawValue: getValue(for: .socketType))
+        }
+    }
+
     public var flags: Flags {
         get throws {
             let flags = Socket.fcntl(file.rawValue, F_GETFL)
@@ -567,6 +573,11 @@ public extension SocketOption where Self == BoolSocketOption {
 }
 
 public extension SocketOption where Self == Int32SocketOption {
+
+    static var socketType: Self {
+        Int32SocketOption(name: SO_TYPE)
+    }
+
     static var sendBufferSize: Self {
         Int32SocketOption(name: SO_SNDBUF)
     }

diff --git a/FlyingSocks/Tests/SocketTests.swift b/FlyingSocks/Tests/SocketTests.swift
@@ -50,6 +50,41 @@ struct SocketTests {
         #expect(SocketType.datagram.rawValue == Socket.datagram)
     }
 
+    @Test
+    func getSocketType_stream() throws {
+        #expect(
+            try Socket(domain: AF_INET).socketType == .stream
+        )
+        #expect(
+            try Socket(domain: AF_INET6).socketType == .stream
+        )
+        #expect(
+            try Socket(domain: AF_UNIX).socketType == .stream
+        )
+        #expect(
+            try Socket(domain: AF_INET, type: .stream).socketType == .stream
+        )
+        #expect(
+            try Socket(domain: AF_INET, type: .stream).socketType == .stream
+        )
+        #expect(
+            try Socket(domain: AF_INET, type: .stream).socketType == .stream
+        )
+    }
+
+    @Test
+    func getSocketType_datagram() throws {
+        #expect(
+            try Socket(domain: AF_INET, type: .datagram).socketType == .datagram
+        )
+        #expect(
+            try Socket(domain: AF_INET, type: .datagram).socketType == .datagram
+        )
+        #expect(
+            try Socket(domain: AF_INET, type: .datagram).socketType == .datagram
+        )
+    }
+
     @Test
     func socketEvents() {
         let events: Set<Socket.Event> = [.read, .write]