cpp/stutterfuzz
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Start of resolution

Browse Source
This commit is contained in:
Ian Gulliver
2016-03-04 16:30:55 -08:00
parent b70a1e09f2
commit 43e73cd959
1 changed files with 35 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
stutterfuzz.c
View File

@@ -2,6 +2,7 @@
#include <dirent.h> #include <dirent.h>
#include <fcntl.h> #include <fcntl.h>
#include <getopt.h> #include <getopt.h>
#include <netdb.h>
#include <stdbool.h> #include <stdbool.h>
#include <stdio.h> #include <stdio.h>
#include <stdint.h> #include <stdint.h>
@@ -19,6 +20,8 @@
static struct { static struct {
char *blob_dir; char *blob_dir;
char *node;
char *service;
uint32_t num_conns; uint32_t num_conns;
uint32_t cycle_ms; uint32_t cycle_ms;
} config = { } config = {
@@ -35,6 +38,10 @@ struct file {
struct conn { struct conn {
int fd; int fd;
enum {
CONN_CONNECTING,
CONN_SENDING,
} state;
struct file *file; struct file *file;
size_t offset; size_t offset;
struct list_head conn_list; struct list_head conn_list;
@@ -48,6 +55,8 @@ static uint64_t rounds = 0, open_conns = 0;
static bool parse_opts(int argc, char *argv[]) { static bool parse_opts(int argc, char *argv[]) {
static struct option long_options[] = { static struct option long_options[] = {
{"blob-dir", required_argument, 0, 'b'}, {"blob-dir", required_argument, 0, 'b'},
{"host", required_argument, 0, 'h'},
{"port", required_argument, 0, 'p'},
{"cycle-ms", required_argument, 0, 'c'}, {"cycle-ms", required_argument, 0, 'c'},
{"num-conns", required_argument, 0, 'n'}, {"num-conns", required_argument, 0, 'n'},
{0, 0, 0, 0 }, {0, 0, 0, 0 },
@@ -60,6 +69,14 @@ static bool parse_opts(int argc, char *argv[]) {
config.blob_dir = optarg; config.blob_dir = optarg;
break; break;
case 'h':
config.node = optarg;
break;
case 'p':
config.service = optarg;
break;
case 'c': case 'c':
config.cycle_ms = (uint32_t) strtoul(optarg, NULL, 10); config.cycle_ms = (uint32_t) strtoul(optarg, NULL, 10);
break; break;
@@ -78,7 +95,9 @@ static bool parse_opts(int argc, char *argv[]) {
return false; return false;
} }
if (!config.blob_dir) { if (!config.blob_dir ||
!config.node ||
!config.service) {
return false; return false;
} }
@@ -171,11 +190,25 @@ static size_t conn_get_split(struct conn *conn) {
} }
static void conn_new() { static void conn_new() {
static struct addrinfo *addrs = NULL;
if (!addrs) {
struct addrinfo hints = {
.ai_flags = AI_V4MAPPED | AI_ADDRCONFIG,
.ai_family = AF_UNSPEC,
.ai_socktype = SOCK_STREAM,
};
assert(!getaddrinfo(config.node, config.service, &hints, &addrs));
}
struct conn *conn = malloc(sizeof(*conn)); struct conn *conn = malloc(sizeof(*conn));
assert(conn); assert(conn);
conn->fd = socket(AF_INET6, SOCK_STREAM | SOCK_NONBLOCK, 0); conn->fd = socket(AF_INET6, SOCK_STREAM | SOCK_NONBLOCK, 0);
assert(conn->fd >= 0); assert(conn->fd >= 0);
// TODO: start connection // TODO: consider sending data here
char buf[1];
sendto(conn->fd, buf, 0, MSG_DONTWAIT | MSG_FASTOPEN, addrs[0].ai_addr, addrs[0].ai_addrlen);
conn->state = CONN_CONNECTING;
conn->file = file_next(); conn->file = file_next();
conn->offset = 0; conn->offset = 0;
list_add(&conn->conn_list, &conn_head); list_add(&conn->conn_list, &conn_head);