Merge pull request #9 from briancroom/prevent-sigpipe

Prevent crashing due to SIGPIPE
kylef-archive · Dec 16, 2015 · faebd06 · faebd06
2 parents 5579c42 + d8ead8e
commit faebd06
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/Sources/Socket.swift b/Sources/Socket.swift
@@ -60,6 +60,12 @@ class Socket {
     guard setsockopt(descriptor, SOL_SOCKET, SO_REUSEADDR, &value, socklen_t(sizeof(Int32))) != -1 else {
       throw SocketError(function: "setsockopt()")
     }
+
+#if !os(Linux)
+    guard setsockopt(descriptor, SOL_SOCKET, SO_NOSIGPIPE, &value, socklen_t(sizeof(Int32))) != -1 else {
+        throw SocketError(function: "setsockopt()")
+    }
+#endif
   }
 
   init(descriptor: Descriptor) {
@@ -103,7 +109,12 @@ class Socket {
 
   func send(output: String) {
     output.withCString { bytes in
-      system_send(descriptor, bytes, Int(strlen(bytes)), 0)
+#if os(Linux)
+    let flags = Int32(MSG_NOSIGNAL)
+#else
+    let flags = Int32(0)
+#endif
+      system_send(descriptor, bytes, Int(strlen(bytes)), flags)
     }
   }