LRN: Added-the-ability-to-substitute-scheduler-select.

3 files changed, 40 insertions, 63 deletions

diff --git a/src/util/disk.h b/src/util/disk.h
index f287834b8..fee43abb8 100644
--- a/src/util/disk.h
+++ b/src/util/disk.h
@@ -22,49 +22,14 @@
  * @file util/disk.h
  * @brief Internal DISK related helper functions
  * @author Nils Durner
- */ 
-  

+ */
 #ifndef GNUNET_DISK_H_
 #define GNUNET_DISK_H_
   
 #include "gnunet_disk_lib.h"
-  

-/**
- * Handle used to access files (and pipes).  
- */ 
-struct GNUNET_DISK_FileHandle 
-{
-  
-#ifdef MINGW
-  /**
-   * File handle under W32.
-   */ 
-  HANDLE h;
-
-  /**
-   * Type
-   */
-  enum {GNUNET_DISK_FILE, GNUNET_PIPE} type;
 
-  /**
-   * Structure for overlapped reading (for pipes)
-   */
-  OVERLAPPED *oOverlapRead;
-
-  /**
-   * Structure for overlapped writing (for pipes)
-   */
-  OVERLAPPED *oOverlapWrite;
-#else
-
-  /**
-   * File handle on other OSes.
-   */ 
-  int fd;
-   
-#endif                          /* 
 */
 };
-



+
 /**
  * Retrieve OS file handle
  *
@@ -75,6 +40,8 @@ struct GNUNET_DISK_FileHandle
  * @return GNUNET_OK on success, GNUNET_SYSERR otherwise
  */ 
 int GNUNET_DISK_internal_file_handle_ (const struct GNUNET_DISK_FileHandle
-                                       *fh, 
void *dst, 
size_t dst_len);
-

+                                       *fh, 
+                                       void *dst, 
+                                       size_t dst_len);
+
 #endif  /* GNUNET_DISK_H_ */
diff --git a/src/util/network.c b/src/util/network.c
index 98eeb7b0c..6ce3df1c2 100644
--- a/src/util/network.c
+++ b/src/util/network.c
@@ -65,28 +65,6 @@ struct GNUNET_NETWORK_Handle
 };
 
 
-struct GNUNET_NETWORK_FDSet
-{
-
-  /**
-   * Maximum number of any socket socket descriptor in the set (plus one)
-   */
-  int nsds;
-
-  /**
-   * Bitset with the descriptors.
-   */
-  fd_set sds;
-
-#ifdef WINDOWS
-  /**
-   * Linked list of handles
-   */
-  struct GNUNET_CONTAINER_SList *handles;
-#endif
-
-};
-
 #ifndef FD_COPY
 #define FD_COPY(s, d) (memcpy ((d), (s), sizeof (fd_set)))
 #endif
@@ -1620,5 +1598,4 @@ GNUNET_NETWORK_socket_select (struct GNUNET_NETWORK_FDSet *rfds,
     return 0;
 }
 
-
 /* end of network.c */
diff --git a/src/util/scheduler.c b/src/util/scheduler.c
index dd0ce6054..692c4f0b7 100644
--- a/src/util/scheduler.c
+++ b/src/util/scheduler.c
@@ -246,6 +246,39 @@ static enum GNUNET_SCHEDULER_Priority max_priority_added;
 static int current_lifeness;
 
 /**
+ * Function to use as a select() in the scheduler.
+ * Defaults to GNUNET_NETWORK_socket_select ()
+ */
+GNUNET_SCHEDULER_select scheduler_select = GNUNET_NETWORK_socket_select;
+
+/**
+ * Sets the select function to use in the scheduler (scheduler_select).
+ *
+ * @param new_select new select function to use
+ * @return previously used select function
+ */
+GNUNET_SCHEDULER_select
+GNUNET_SCHEDULER_set_select (GNUNET_SCHEDULER_select new_select)
+{
+  GNUNET_SCHEDULER_select old_select = scheduler_select;
+  scheduler_select = new_select;
+  if (scheduler_select == NULL)
+    scheduler_select = GNUNET_NETWORK_socket_select;
+  return old_select;
+}
+
+/**
+ * Gets the select function currently used in the scheduler.
+ *
+ * @return currently used select function
+ */
+GNUNET_SCHEDULER_select
+GNUNET_SCHEDULER_get_select ()
+{
+  return scheduler_select;
+}
+
+/**
  * Check that the given priority is legal (and return it).
  *
  * @param p priority value to check
@@ -806,7 +839,7 @@ GNUNET_SCHEDULER_run (GNUNET_SCHEDULER_Task task, void *task_cls)
           /* no blocking, more work already ready! */
           timeout = GNUNET_TIME_UNIT_ZERO;
         }
-      ret = GNUNET_NETWORK_socket_select (rs, ws, NULL, timeout);
+      ret = scheduler_select (rs, ws, NULL, timeout);
       if (ret == GNUNET_SYSERR)
         {
           if (errno == EINTR)