talksik/uWebSockets
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update uSocketsMock.c to the 0.2.0 interfaces

Browse Source
This commit is contained in:
Alex Hultman
2019-06-11 19:31:58 +02:00
parent 2714ffe8d6
commit 28e0f24f2f
1 changed files with 50 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fuzzing/uSocketsMock.c
+50 -50
View File
@@ -1,89 +1,89 @@
/* uSockets is entierly opaque so we can use the real header straight up */ /* uSockets is entierly opaque so we can use the real header straight up */
#include "../uSockets/src/libusockets_new.h" #include "../uSockets/src/libusockets.h"
#include <stdio.h> #include <stdio.h>
#include <stdlib.h> #include <stdlib.h>
struct us_loop { struct us_loop_t {
/* We only support one listen socket */ /* We only support one listen socket */
struct us_listen_socket *listen_socket; struct us_listen_socket_t *listen_socket;
}; };
struct us_loop *us_create_loop(int default_hint, void (*wakeup_cb)(struct us_loop *loop), void (*pre_cb)(struct us_loop *loop), void (*post_cb)(struct us_loop *loop), unsigned int ext_size) { 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) {
return (struct us_loop *) malloc(sizeof(struct us_loop) + ext_size); return (struct us_loop_t *) malloc(sizeof(struct us_loop_t) + ext_size);
} }
void us_loop_free(struct us_loop *loop) { void us_loop_free(struct us_loop_t *loop) {
free(loop); free(loop);
} }
void *us_loop_ext(struct us_loop *loop) { void *us_loop_ext(struct us_loop_t *loop) {
return loop + 1; return loop + 1;
} }
void us_loop_run(struct us_loop *loop) { void us_loop_run(struct us_loop_t *loop) {
} }
struct us_socket_context { struct us_socket_context_t {
struct us_loop *loop; struct us_loop_t *loop;
struct us_socket *(*on_open)(struct us_socket *s, int is_client, char *ip, int ip_length); struct us_socket_t *(*on_open)(struct us_socket_t *s, int is_client, char *ip, int ip_length);
struct us_socket *(*on_close)(struct us_socket *s); struct us_socket_t *(*on_close)(struct us_socket_t *s);
struct us_socket *(*on_data)(struct us_socket *s, char *data, int length); struct us_socket_t *(*on_data)(struct us_socket_t *s, char *data, int length);
struct us_socket *(*on_writable)(struct us_socket *s); struct us_socket_t *(*on_writable)(struct us_socket_t *s);
struct us_socket *(*on_timeout)(struct us_socket *s); struct us_socket_t *(*on_timeout)(struct us_socket_t *s);
struct us_socket *(*on_end)(struct us_socket *s); struct us_socket_t *(*on_end)(struct us_socket_t *s);
}; };
struct us_socket_context *us_create_socket_context(struct us_loop *loop, int ext_size) { struct us_socket_context_t *us_create_socket_context(int ssl, struct us_loop_t *loop, int ext_size, struct us_socket_context_options_t options) {
struct us_socket_context *socket_context = (struct us_socket_context *) malloc(sizeof(struct us_socket_context) + ext_size); struct us_socket_context_t *socket_context = (struct us_socket_context_t *) malloc(sizeof(struct us_socket_context_t) + ext_size);
socket_context->loop = loop; socket_context->loop = loop;
return socket_context; return socket_context;
} }
void us_socket_context_free(struct us_socket_context *context) { void us_socket_context_free(int ssl, struct us_socket_context_t *context) {
free(context); free(context);
} }
void us_socket_context_on_open(struct us_socket_context *context, struct us_socket *(*on_open)(struct us_socket *s, int is_client, char *ip, int ip_length)) { void us_socket_context_on_open(int ssl, struct us_socket_context_t *context, struct us_socket_t *(*on_open)(struct us_socket_t *s, int is_client, char *ip, int ip_length)) {
context->on_open = on_open; context->on_open = on_open;
} }
void us_socket_context_on_close(struct us_socket_context *context, struct us_socket *(*on_close)(struct us_socket *s)) { void us_socket_context_on_close(int ssl, struct us_socket_context_t *context, struct us_socket_t *(*on_close)(struct us_socket_t *s)) {
context->on_close = on_close; context->on_close = on_close;
} }
void us_socket_context_on_data(struct us_socket_context *context, struct us_socket *(*on_data)(struct us_socket *s, char *data, int length)) { void us_socket_context_on_data(int ssl, struct us_socket_context_t *context, struct us_socket_t *(*on_data)(struct us_socket_t *s, char *data, int length)) {
context->on_data = on_data; context->on_data = on_data;
} }
void us_socket_context_on_writable(struct us_socket_context *context, struct us_socket *(*on_writable)(struct us_socket *s)) { void us_socket_context_on_writable(int ssl, struct us_socket_context_t *context, struct us_socket_t *(*on_writable)(struct us_socket_t *s)) {
context->on_writable = on_writable; context->on_writable = on_writable;
} }
void us_socket_context_on_timeout(struct us_socket_context *context, struct us_socket *(*on_timeout)(struct us_socket *s)) { void us_socket_context_on_timeout(int ssl, struct us_socket_context_t *context, struct us_socket_t *(*on_timeout)(struct us_socket_t *s)) {
context->on_timeout = on_timeout; context->on_timeout = on_timeout;
} }
void us_socket_context_on_end(struct us_socket_context *context, struct us_socket *(*on_end)(struct us_socket *s)) { void us_socket_context_on_end(int ssl, struct us_socket_context_t *context, struct us_socket_t *(*on_end)(struct us_socket_t *s)) {
context->on_end = on_end; context->on_end = on_end;
} }
void *us_socket_context_ext(struct us_socket_context *context) { void *us_socket_context_ext(int ssl, struct us_socket_context_t *context) {
return context + 1; return context + 1;
} }
struct us_listen_socket { struct us_listen_socket_t {
int socket_ext_size; int socket_ext_size;
struct us_socket_context *context; struct us_socket_context_t *context;
}; };
struct us_listen_socket *us_socket_context_listen(struct us_socket_context *context, const char *host, int port, int options, int socket_ext_size) { struct us_listen_socket_t *us_socket_context_listen(int ssl, struct us_socket_context_t *context, const char *host, int port, int options, int socket_ext_size) {
struct us_listen_socket *listen_socket = (struct us_listen_socket *) malloc(sizeof(struct us_listen_socket)); struct us_listen_socket_t *listen_socket = (struct us_listen_socket_t *) malloc(sizeof(struct us_listen_socket_t));
listen_socket->socket_ext_size = socket_ext_size; listen_socket->socket_ext_size = socket_ext_size;
listen_socket->context = context; listen_socket->context = context;
@@ -93,83 +93,83 @@ struct us_listen_socket *us_socket_context_listen(struct us_socket_context *cont
return listen_socket; return listen_socket;
} }
void us_listen_socket_close(struct us_listen_socket *ls) { void us_listen_socket_close(int ssl, struct us_listen_socket_t *ls) {
free(ls); free(ls);
} }
struct us_socket { struct us_socket_t {
struct us_socket_context *context; struct us_socket_context_t *context;
}; };
/* For ubsan? */ /* For ubsan? */
struct us_new_socket_t { struct us_new_socket_t {
struct us_socket_context *context; struct us_socket_context_t *context;
}; };
struct us_socket *us_socket_context_connect(struct us_socket_context *context, const char *host, int port, int options, int socket_ext_size) { struct us_socket_t *us_socket_context_connect(int ssl, struct us_socket_context_t *context, const char *host, int port, int options, int socket_ext_size) {
printf("us_socket_context_connect\n"); printf("us_socket_context_connect\n");
} }
struct us_loop *us_socket_context_loop(struct us_socket_context *context) { struct us_loop_t *us_socket_context_loop(int ssl, struct us_socket_context_t *context) {
return context->loop; return context->loop;
} }
struct us_socket *us_socket_context_adopt_socket(struct us_socket_context *context, struct us_socket *s, int ext_size) { struct us_socket_t *us_socket_context_adopt_socket(int ssl, struct us_socket_context_t *context, struct us_socket_t *s, int ext_size) {
printf("us_socket_context_adopt_socket\n"); printf("us_socket_context_adopt_socket\n");
} }
struct us_socket_context *us_create_child_socket_context(struct us_socket_context *context, int context_ext_size) { struct us_socket_context_t *us_create_child_socket_context(int ssl, struct us_socket_context_t *context, int context_ext_size) {
printf("us_create_child_socket_context\n"); printf("us_create_child_socket_context\n");
} }
int us_socket_write(struct us_socket *s, const char *data, int length, int msg_more) { int us_socket_write(int ssl, struct us_socket_t *s, const char *data, int length, int msg_more) {
} }
void us_socket_timeout(struct us_socket *s, unsigned int seconds) { void us_socket_timeout(int ssl, struct us_socket_t *s, unsigned int seconds) {
} }
void *us_socket_ext(struct us_socket *s) { void *us_socket_ext(int ssl, struct us_socket_t *s) {
return s + 1; return s + 1;
} }
struct us_socket_context *us_socket_get_context(struct us_socket *s) { struct us_socket_context_t *us_socket_context(int ssl, struct us_socket_t *s) {
return s->context; return s->context;
} }
void us_socket_flush(struct us_socket *s) { void us_socket_flush(int ssl, struct us_socket_t *s) {
} }
void us_socket_shutdown(struct us_socket *s) { void us_socket_shutdown(int ssl, struct us_socket_t *s) {
} }
int us_socket_is_shut_down(struct us_socket *s) { int us_socket_is_shut_down(int ssl, struct us_socket_t *s) {
return 0; return 0;
} }
int us_socket_is_closed(struct us_socket *s) { int us_socket_is_closed(int ssl, struct us_socket_t *s) {
return 0; return 0;
} }
struct us_socket *us_socket_close(struct us_socket *s) { struct us_socket_t *us_socket_close(int ssl, struct us_socket_t *s) {
return s; return s;
} }
void us_socket_remote_address(struct us_socket *s, char *buf, int *length) { void us_socket_remote_address(int ssl, struct us_socket_t *s, char *buf, int *length) {
printf("us_socket_remote_address\n"); printf("us_socket_remote_address\n");
} }
/* We expose this function to let fuzz targets push data to uSockets */ /* We expose this function to let fuzz targets push data to uSockets */
void us_loop_read_mocked_data(struct us_loop *loop, char *data, unsigned int size) { void us_loop_read_mocked_data(struct us_loop_t *loop, char *data, unsigned int size) {
/* We have one listen socket */ /* We have one listen socket */
int socket_ext_size = loop->listen_socket->socket_ext_size; int socket_ext_size = loop->listen_socket->socket_ext_size;
/* Create a socket with information from the listen socket */ /* Create a socket with information from the listen socket */
struct us_socket *s = (struct us_socket *) malloc(sizeof(struct us_socket) + socket_ext_size); struct us_socket_t *s = (struct us_socket_t *) malloc(sizeof(struct us_socket_t) + socket_ext_size);
s->context = loop->listen_socket->context; s->context = loop->listen_socket->context;
/* Emit open event */ /* Emit open event */