Fuzzing: New broker packet fuzz targets
These initialise the broker and leave it running. This is problematic from the point of view of deterministic testing, but should mean the testing speed is vastly improved.pull/2768/head
parent
5b7036dfcd
commit
0be6d00d66
@ -0,0 +1,143 @@
|
||||
/*
|
||||
Copyright (c) 2023 Cedalo GmbH
|
||||
|
||||
All rights reserved. This program and the accompanying materials
|
||||
are made available under the terms of the Eclipse Public License 2.0
|
||||
and Eclipse Distribution License v1.0 which accompany this distribution.
|
||||
|
||||
The Eclipse Public License is available at
|
||||
https://www.eclipse.org/legal/epl-2.0/
|
||||
and the Eclipse Distribution License is available at
|
||||
http://www.eclipse.org/org/documents/edl-v10.php.
|
||||
|
||||
SPDX-License-Identifier: EPL-2.0 OR BSD-3-Clause
|
||||
|
||||
Contributors:
|
||||
Roger Light - initial implementation and documentation.
|
||||
*/
|
||||
|
||||
#include <arpa/inet.h>
|
||||
#include <errno.h>
|
||||
#include <netinet/in.h>
|
||||
#include <pthread.h>
|
||||
#include <signal.h>
|
||||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <string.h>
|
||||
#include <sys/socket.h>
|
||||
#include <unistd.h>
|
||||
|
||||
#include "broker_fuzz.h"
|
||||
|
||||
#define PORT 1883
|
||||
|
||||
/* The broker fuzz-only main function. */
|
||||
extern "C" int mosquitto_fuzz_main(int argc, char *argv[]);
|
||||
|
||||
void *run_broker(void *args)
|
||||
{
|
||||
char *argv[4];
|
||||
int argc = 4;
|
||||
|
||||
argv[0] = strdup("mosquitto");
|
||||
argv[1] = strdup("-v");
|
||||
argv[2] = strdup("-c");
|
||||
argv[3] = strdup("/tmp/mosquitto.conf");
|
||||
|
||||
mosquitto_fuzz_main(argc, argv);
|
||||
|
||||
for(int i=0; i<argc; i++){
|
||||
free(argv[i]);
|
||||
}
|
||||
|
||||
pthread_exit(NULL);
|
||||
return NULL;
|
||||
}
|
||||
|
||||
|
||||
void recv_timeout(int sock, void *buf, size_t len, int timeout_us)
|
||||
{
|
||||
struct timeval tv = {0, timeout_us};
|
||||
|
||||
setsockopt(sock, SOL_SOCKET, SO_RCVTIMEO, (const char *)&tv, sizeof(tv));
|
||||
(void)recv(sock, buf, len, 0);
|
||||
}
|
||||
|
||||
int connect_retrying(int port)
|
||||
{
|
||||
struct sockaddr_in addr;
|
||||
int sock;
|
||||
int rc;
|
||||
|
||||
memset(&addr, 0, sizeof(addr));
|
||||
addr.sin_family = AF_INET;
|
||||
addr.sin_port = htons(port);
|
||||
addr.sin_addr.s_addr = inet_addr("127.0.0.1");
|
||||
|
||||
sock = socket(AF_INET, SOCK_STREAM, 0);
|
||||
while(1){
|
||||
errno = 0;
|
||||
rc = connect(sock, (struct sockaddr *)&addr, sizeof(addr));
|
||||
if(rc < 0){
|
||||
struct timespec ts;
|
||||
ts.tv_sec = 0;
|
||||
ts.tv_nsec = 10000000; /* 10ms */
|
||||
nanosleep(&ts, NULL);
|
||||
}else{
|
||||
break;
|
||||
}
|
||||
}
|
||||
return sock;
|
||||
}
|
||||
|
||||
static bool initialise(pthread_t *thread)
|
||||
{
|
||||
FILE *fptr;
|
||||
|
||||
signal(SIGPIPE, SIG_IGN);
|
||||
|
||||
fptr = fopen("/tmp/mosquitto.conf", "wb");
|
||||
if(!fptr){
|
||||
printf("FILE %s\n", strerror(errno));
|
||||
abort();
|
||||
}
|
||||
fprintf(fptr, "user root\n");
|
||||
fprintf(fptr, "listener %d\n", PORT);
|
||||
fprintf(fptr, "allow_anonymous true\n");
|
||||
fclose(fptr);
|
||||
|
||||
pthread_create(thread, NULL, run_broker, NULL);
|
||||
|
||||
return true;
|
||||
}
|
||||
|
||||
void deinitialise(pthread_t *thread)
|
||||
{
|
||||
pthread_join(*thread, NULL);
|
||||
unlink("/tmp/mosquitto.conf");
|
||||
}
|
||||
|
||||
|
||||
extern "C" int LLVMFuzzerTestOneInput(const uint8_t *data, size_t size)
|
||||
{
|
||||
struct fuzz_data fuzz;
|
||||
pthread_t thread;
|
||||
static bool initialised = false;
|
||||
|
||||
if(!initialised){
|
||||
initialised = initialise(&thread);
|
||||
}
|
||||
|
||||
if(size < kMinInputLength || size > kMaxInputLength){
|
||||
return 0;
|
||||
}
|
||||
|
||||
memset(&fuzz, 0, sizeof(fuzz));
|
||||
fuzz.port = PORT;
|
||||
fuzz.size = size;
|
||||
fuzz.data = (uint8_t *)data;
|
||||
|
||||
run_client(&fuzz);
|
||||
|
||||
return 0;
|
||||
}
|
Loading…
Reference in New Issue