1
0
mirror of https://github.com/irssi/irssi.git synced 2024-09-29 04:45:57 -04:00

Add server-fuzz to fe-fuzz

Based on test-796.c in the tests directory.
This commit is contained in:
Joseph Bisch 2018-08-15 11:11:09 -04:00
parent 22ce54231e
commit 4f5f62ac63
2 changed files with 190 additions and 3 deletions

View File

@ -1,6 +1,6 @@
SUBDIRS = irc fe-common SUBDIRS = irc fe-common
bin_PROGRAMS = irssi-fuzz bin_PROGRAMS = irssi-fuzz server-fuzz
# Force link with CXX for libfuzzer support # Force link with CXX for libfuzzer support
CCLD=$(CXX) $(CXXFLAGS) CCLD=$(CXX) $(CXXFLAGS)
@ -21,7 +21,18 @@ irssi_fuzz_LDADD = \
irssi_fuzz_SOURCES = \ irssi_fuzz_SOURCES = \
irssi.c \ irssi.c \
$(top_srcdir)/src/fe-text/module-formats.c ../fe-text/module-formats.c
server_fuzz_DEPENDENCIES = @COMMON_LIBS@
server_fuzz_LDADD = \
@COMMON_LIBS@ \
@PROG_LIBS@ \
$(FUZZER_LIBS)
server_fuzz_SOURCES = \
server.c \
../fe-text/module-formats.c
noinst_HEADERS = \ noinst_HEADERS = \
$(top_srcdir)/src/fe-text/module-formats.h ../fe-text/module-formats.h

176
src/fe-fuzz/server.c Normal file
View File

@ -0,0 +1,176 @@
/*
server.c : irssi
Copyright (C) 2018 Joseph Bisch
This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
*/
#include "module.h"
#include "modules-load.h"
#include "module-formats.h"
#include "levels.h"
#include "themes.h"
#include "core.h"
#include "fe-common-core.h"
#include "args.h"
#include "printtext.h"
#include "misc.h"
#include "servers-setup.h"
#include <stddef.h>
#include <stdint.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "irc.h"
#include "irc-servers.h"
#include "irc-channels.h"
/* irc.c */
void irc_init(void);
void irc_deinit(void);
/* irc-session.c */
void irc_session_init(void);
void irc_session_deinit(void);
/* fe-common-irc.c */
void fe_common_irc_init(void);
void fe_common_irc_deinit(void);
SERVER_REC *server;
void event_connected(IRC_SERVER_REC *server, const char *data, const char *from)
{
char *params, *nick;
g_return_if_fail(server != NULL);
params = event_get_params(data, 1, &nick);
if (g_strcmp0(server->nick, nick) != 0) {
/* nick changed unexpectedly .. connected via proxy, etc. */
g_free(server->nick);
server->nick = g_strdup(nick);
}
/* set the server address */
g_free(server->real_address);
server->real_address = from == NULL ?
g_strdup(server->connrec->address) : /* shouldn't happen.. */
g_strdup(from);
/* last welcome message found - commands can be sent to server now. */
server->connected = 1;
server->real_connect_time = time(NULL);
/* let the queue send now that we are identified */
g_get_current_time(&server->wait_cmd);
if (server->connrec->usermode != NULL) {
/* Send the user mode, before the autosendcmd.
* Do not pass this through cmd_mode because it
* is not known whether the resulting MODE message
* (if any) is the initial umode or a reply to this.
*/
irc_send_cmdv(server, "MODE %s %s", server->nick,
server->connrec->usermode);
g_free_not_null(server->wanted_usermode);
server->wanted_usermode = g_strdup(server->connrec->usermode);
}
signal_emit("event connected", 1, server);
g_free(params);
}
void irc_server_init_bare_minimum(IRC_SERVER_REC *server) {
server->isupport = g_hash_table_new((GHashFunc) g_istr_hash,
(GCompareFunc) g_istr_equal);
/* set the standards */
g_hash_table_insert(server->isupport, g_strdup("CHANMODES"), g_strdup("beI,k,l,imnpst"));
g_hash_table_insert(server->isupport, g_strdup("PREFIX"), g_strdup("(ohv)@%+"));
}
void test_server() {
//SERVER_REC *server; /* = g_new0(IRC_SERVER_REC, 1); */
CHAT_PROTOCOL_REC *proto;
SERVER_CONNECT_REC *conn;
proto = chat_protocol_find("IRC");
conn = server_create_conn(proto->id, "localhost", 0, "", "", "user");
server = proto->server_init_connect(conn);
server->session_reconnect = TRUE;
server->tag = g_strdup("testserver");
/* we skip some initialisations that would try to send data */
/* irc_servers_deinit(); */
irc_session_deinit();
irc_irc_deinit();
server_connect_finished(server);
/* make up for the skipped session init */
irc_server_init_bare_minimum(IRC_SERVER(server));
irc_irc_init();
irc_session_init();
/* irc_servers_init(); */
server_connect_unref(conn);
}
int LLVMFuzzerInitialize(int *argc, char ***argv) {
core_register_options();
fe_common_core_register_options();
/* no args */
args_execute(0, NULL);
core_preinit((*argv)[0]);
core_init();
irc_init();
fe_common_core_init();
fe_common_irc_init();
signal_add("event 001", (SIGNAL_FUNC) event_connected);
module_register("core", "fe-fuzz");
return 0;
}
int LLVMFuzzerTestOneInput(const uint8_t* data, size_t size) {
if (size < 1) return 0;
test_server();
gboolean prefixedChoice = (gboolean)*data;
gchar *copy = g_strndup((const gchar *)data+1, size-1);
gchar **lines = g_strsplit(copy, "\r\n", -1);
gchar **head = lines;
for (; *lines != NULL; lines++) {
gchar *prefixedLine;
if (prefixedChoice) {
prefixedLine = g_strdup_printf(":user %s\n", *lines);
} else {
prefixedLine = g_strdup_printf("%s\n", *lines);
}
signal_emit("server incoming", 2, server, prefixedLine);
g_free(prefixedLine);
}
g_strfreev(head);
g_free(copy);
server_unref(server);
return 0;
}