faux.eloop: Functions to include and exclude events for fd

faux/eloop.h

@@ -62,6 +62,8 @@ faux_ev_t *faux_eloop_add_sched_periodic_delayed(faux_eloop_t *eloop,
 	const struct timespec *period, unsigned int cycle_num);
 ssize_t faux_eloop_del_sched(faux_eloop_t *eloop, faux_ev_t *ev);
 ssize_t faux_eloop_del_sched_by_id(faux_eloop_t *eloop, int ev_id);
+bool_t faux_eloop_include_fd_event(faux_eloop_t *eloop, int fd, short event);
+bool_t faux_eloop_exclude_fd_event(faux_eloop_t *eloop, int fd, short event);
 
 C_DECL_END
 

faux/eloop/eloop.c

@@ -435,6 +435,7 @@ bool_t faux_eloop_add_fd(faux_eloop_t *eloop, int fd, short events,
 	faux_eloop_fd_t *entry = NULL;
 	faux_list_node_t *new_node = NULL;
 
+	assert(eloop);
 	if (!eloop || (fd < 0))
 		return BOOL_FALSE;
 
@@ -461,6 +462,80 @@ bool_t faux_eloop_add_fd(faux_eloop_t *eloop, int fd, short events,
 }
 
 
+/** @brief Registers additional event for specified fd.
+ *
+ * See poll() for explanation of possible file events ("events" argument).
+ * Suppose some fd was added by faux_eloop_add_fd(). User have specified some
+ * events like POLLIN. Now user wants to track POLLOUT event too. So it's not
+ * necessary to remove fd by faux_eloop_del_fd() and then re-add it with new
+ * event mask. User can include additional events by
+ * faux_eloop_include_fd_event(). Specified event will be added to existent
+ * event mask.
+ *
+ * @param [in] eloop Allocated and initialized event loop object.
+ * @param [in] fd File descriptor to change event mask.
+ * @param [in] events File event to include (like POLLIN, POLLOUT).
+ * @return BOOL_TRUE - success, BOOL_FALSE - error.
+ */
+bool_t faux_eloop_include_fd_event(faux_eloop_t *eloop, int fd, short event)
+{
+	faux_eloop_fd_t *entry = NULL;
+
+	assert(eloop);
+	if (!eloop)
+		return BOOL_FALSE;
+	assert(fd >= 0);
+	if (fd < 0)
+		return BOOL_FALSE;
+
+	entry = (faux_eloop_fd_t *)faux_list_kfind(eloop->fds, &fd);
+	if (!entry)
+		return BOOL_FALSE;
+	entry->events = entry->events | event;
+	faux_pollfd_del_by_fd(eloop->pollfds, fd);
+	faux_pollfd_add(eloop->pollfds, fd, entry->events);
+
+	return BOOL_TRUE;
+}
+
+
+/** @brief Unregisters event for specified fd.
+ *
+ * See poll() for explanation of possible file events ("events" argument).
+ * Suppose some fd was added by faux_eloop_add_fd(). User have specified some
+ * events like POLLIN, POLLOUT. Now user doesn't wants to track one of the
+ * events (POLLOUT for example). So it's not necessary to remove fd by
+ * faux_eloop_del_fd() and then re-add it with new event mask. User can exclude
+ * event by faux_eloop_include_fd_event(). Specified event will be excluded from
+ * existent event mask.
+ *
+ * @param [in] eloop Allocated and initialized event loop object.
+ * @param [in] fd File descriptor to change event mask.
+ * @param [in] events File event to exclude (like POLLIN, POLLOUT).
+ * @return BOOL_TRUE - success, BOOL_FALSE - error.
+ */
+bool_t faux_eloop_exclude_fd_event(faux_eloop_t *eloop, int fd, short event)
+{
+	faux_eloop_fd_t *entry = NULL;
+
+	assert(eloop);
+	if (!eloop)
+		return BOOL_FALSE;
+	assert(fd >= 0);
+	if (fd < 0)
+		return BOOL_FALSE;
+
+	entry = (faux_eloop_fd_t *)faux_list_kfind(eloop->fds, &fd);
+	if (!entry)
+		return BOOL_FALSE;
+	entry->events = entry->events & (~event);
+	faux_pollfd_del_by_fd(eloop->pollfds, fd);
+	faux_pollfd_add(eloop->pollfds, fd, entry->events);
+
+	return BOOL_TRUE;
+}
+
+
 /** @brief Unregisters file descriptor.
  *
  * @param [in] eloop Allocated and initialized event loop object.