Emit pre/post callbacks while fuzzing

yongju-hong · Jan 11, 2020 · 5d0d821 · 5d0d821
1 parent d6954ba
commit 5d0d821
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/fuzzing/uSocketsMock.c b/fuzzing/uSocketsMock.c
@@ -13,6 +13,10 @@ struct us_loop_t {
 
     /* The list of closed sockets */
     struct us_socket_t *close_list;
+
+    /* Post and pre callbacks */
+    void (*pre_cb)(struct us_loop_t *loop);
+    void (*post_cb)(struct us_loop_t *loop);
 };
 
 struct us_loop_t *us_create_loop(void *hint, void (*wakeup_cb)(struct us_loop_t *loop), void (*pre_cb)(struct us_loop_t *loop), void (*post_cb)(struct us_loop_t *loop), unsigned int ext_size) {
@@ -21,6 +25,9 @@ struct us_loop_t *us_create_loop(void *hint, void (*wakeup_cb)(struct us_loop_t
     loop->listen_socket = 0;
     loop->close_list = 0;
 
+    loop->pre_cb = pre_cb;
+    loop->post_cb = post_cb;
+
     return loop;
 }
 
@@ -215,6 +222,9 @@ void us_loop_read_mocked_data(struct us_loop_t *loop, char *data, unsigned int s
     /* We are unwound so let's free all closed polls here */
 
 
+    /* Call pre_cb */
+    loop->pre_cb(loop);
+
     /* We have one listen socket */
     int socket_ext_size = loop->listen_socket->socket_ext_size;
 
@@ -282,4 +292,7 @@ void us_loop_read_mocked_data(struct us_loop_t *loop, char *data, unsigned int s
 
     /* Free the socket */
     free(s);
+
+    /* Call post_cb */
+    loop->post_cb(loop);
 }