X-Git-Url: https://git.netwichtig.de/gitweb/?a=blobdiff_plain;f=include%2Fsocketengine.h;h=8e4c3dfc9e3dbf5dca98ae2a86b276b63c15502e;hb=cfa32a6561e0152ebbd7135eaec9f7c794c170b1;hp=b790f6d77596b389bab0862b1b35aa8754ae0797;hpb=808a0a09577009c2d6e494979c2189426b332aef;p=user%2Fhenk%2Fcode%2Finspircd.git

diff --git a/include/socketengine.h b/include/socketengine.h
index b790f6d77..8e4c3dfc9 100644
--- a/include/socketengine.h
+++ b/include/socketengine.h
@@ -20,13 +20,12 @@
  */
 
 
-#ifndef SOCKETENGINE_H
-#define SOCKETENGINE_H
+#pragma once
 
 #include <vector>
 #include <string>
 #include <map>
-#include "inspircd_config.h"
+#include "config.h"
 #include "socket.h"
 #include "base.h"
 
@@ -128,7 +127,7 @@ enum EventMask
 	/** Add a trial write. During the next DispatchEvents invocation, this
 	 * will call HandleEvent with EVENT_WRITE unless writes are known to be
 	 * blocking.
-	 * 
+	 *
 	 * This could be used to group several writes together into a single
 	 * send() syscall, or to ensure that writes are blocking when attempting
 	 * to use FD_WANT_FAST_WRITE.
@@ -137,7 +136,7 @@ enum EventMask
 	/** Assert that writes are known to block. This cancels FD_ADD_TRIAL_WRITE.
 	 * Reset by SE before running EVENT_WRITE
 	 */
-	FD_WRITE_WILL_BLOCK = 0x8000, 
+	FD_WRITE_WILL_BLOCK = 0x8000,
 
 	/** Mask for trial read/trial write */
 	FD_TRIAL_NOTE_MASK = 0x5000
@@ -418,7 +417,7 @@ public:
 	 * @param buf The buffer in which the data that is sent is stored.
 	 * @param len The size of the buffer.
 	 * @param flags A flag value that controls the sending of the data.
-	 * @param to The remote IP address and port.	
+	 * @param to The remote IP address and port.
 	 * @param tolen The size of the to parameter.
 	 * @return This method should return exactly the same values as the system call it emulates.
 	 */
@@ -489,9 +488,24 @@ public:
 	/** Get data transfer statistics, kilobits per second in and out and total.
 	 */
 	void GetStats(float &kbitpersec_in, float &kbitpersec_out, float &kbitpersec_total);
+
+	/** Should we ignore the error in errno?
+	 * Checks EAGAIN and WSAEWOULDBLOCK
+	 */
+	static bool IgnoreError();
 };
 
-SocketEngine* CreateSocketEngine();
+inline bool SocketEngine::IgnoreError()
+{
+	if ((errno == EAGAIN) || (errno == EWOULDBLOCK))
+		return true;
 
+#ifdef _WIN32
+	if (WSAGetLastError() == WSAEWOULDBLOCK)
+		return true;
 #endif
 
+	return false;
+}
+
+SocketEngine* CreateSocketEngine();