diff --git a/tests/Makefile.am b/tests/Makefile.am index 14f160c5..9d123627 100644 --- a/tests/Makefile.am +++ b/tests/Makefile.am @@ -9,6 +9,7 @@ check_PROGRAMS = runtests \ rb_snprintf_try_append1 \ sasl_abort1 \ send1 \ + send_multiline1 \ serv_connect1 \ substitution1 AM_CFLAGS=$(WARNFLAGS) diff --git a/tests/send_multiline1.c b/tests/send_multiline1.c new file mode 100644 index 00000000..83f176e1 --- /dev/null +++ b/tests/send_multiline1.c @@ -0,0 +1,141 @@ +/* + * Copyright (C) 2020 Ed Kellett + * + * 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 +#include +#include +#include +#include "tap/basic.h" + +#include "ircd_util.h" +#include "client_util.h" + +#include "send.h" +#include "s_serv.h" +#include "monitor.h" +#include "s_conf.h" +#include "hash.h" + +#define MSG "%s:%d (%s)", __FILE__, __LINE__, __FUNCTION__ + +static void sendto_multiline_basic(void) +{ + struct Client *user = make_local_person(); + const char *s; + + /* multiline with no items should do nothing */ + send_multiline_init(user, " ", "foo"); + send_multiline_fini(user, NULL); + is_client_sendq_empty(user, MSG); + + /* 510 = 17 * 30. line the end of an item with the end of the 510-byte data exactly */ + send_multiline_init(user, " ", "prefix78901234567 "); + for (size_t i = 0; i < 29; i++) + send_multiline_item(user, "item567890123456"); + send_multiline_fini(user, NULL); + + s = get_client_sendq(user); + is_int(512, strlen(s), MSG); + is_string("item567890123456\r\n", &s[494], MSG); + is_client_sendq_empty(user, MSG); + + /* just run exactly the same thing again, there's static state */ + send_multiline_init(user, " ", "prefix78901234567 "); + for (size_t i = 0; i < 29; i++) + send_multiline_item(user, "item567890123456"); + send_multiline_fini(user, NULL); + + s = get_client_sendq(user); + is_int(512, strlen(s), MSG); + is_string("item567890123456\r\n", &s[494], MSG); + is_client_sendq_empty(user, MSG); + + /* the same thing again but with one extra character, so we have an item that just won't fit */ + send_multiline_init(user, " ", "prefix789012345678 "); + for (size_t i = 0; i < 29; i++) + send_multiline_item(user, "item567890123456"); + send_multiline_item(user, "bar"); + send_multiline_fini(user, "foo "); + + s = get_client_sendq(user); + is_string("item567890123456\r\n", &s[478], MSG); + is_client_sendq("foo item567890123456 bar\r\n", user, MSG); + + remove_local_person(user); +} + +static void sendto_multiline_extra_space(void) +{ + struct Client *server = make_remote_server_full(&me, "remote.test", "777"); + struct Client *luser = make_local_person(); + struct Client *ruser = make_remote_person(server); + const char *s; + + strcpy(ruser->id, "777000001"); + add_to_id_hash(ruser->id, ruser); + + /* ":me.test foo4567890123 local_test :" -> 22 + 13 = 35 */ + send_multiline_init(luser, " ", ":%s foo4567890123 %s :", + get_id(&me, luser), get_id(luser, luser)); + /* both of these should be noop */ + send_multiline_remote_pad(luser, &me); + send_multiline_remote_pad(luser, luser); + /* so all this should fit on one line */ + for (size_t i = 0; i < 28; i++) + send_multiline_item(luser, "item567890123456"); + send_multiline_fini(luser, NULL); + + s = get_client_sendq(luser); + is_int(512, strlen(s), MSG); + is_string("item567890123456\r\n", &s[494], MSG); + is_client_sendq_empty(luser, MSG); + + /* as above, but remote_test is one longer */ + send_multiline_init(ruser, " ", ":%s foo456789012 %s :", + get_id(&me, ruser), get_id(ruser, ruser)); + /* should add "me.test" - 3 = 4 */ + send_multiline_remote_pad(ruser, &me); + /* should add "remote_test" - 9 = 2 */ + send_multiline_remote_pad(ruser, ruser); + /* so all this should fit on one line */ + for (size_t i = 0; i < 28; i++) + send_multiline_item(ruser, "item567890123456"); + /* and this shouldn't */ + send_multiline_item(ruser, "x"); + send_multiline_fini(ruser, NULL); + + s = get_client_sendq(server); + is_int(506, strlen(s), MSG); + is_string("item567890123456\r\n", &s[488], MSG); + is_client_sendq(":0AA foo456789012 777000001 :x\r\n", server, MSG); +} + +int main(int argc, char *argv[]) +{ + plan_lazy(); + + ircd_util_init(__FILE__); + client_util_init(); + + sendto_multiline_basic(); + sendto_multiline_extra_space(); + + client_util_free(); + ircd_util_free(); + return 0; +} diff --git a/tests/send_multiline1.conf b/tests/send_multiline1.conf new file mode 100644 index 00000000..5490917f --- /dev/null +++ b/tests/send_multiline1.conf @@ -0,0 +1,29 @@ +serverinfo { + sid = "0AA"; + name = "me.test"; + description = "Test server"; + network_name = "Test network"; +}; + +connect "remote.test" { + host = "::1"; + fingerprint = "test"; + class = "default"; +}; + +connect "remote2.test" { + host = "::1"; + fingerprint = "test"; + class = "default"; +}; + +connect "remote3.test" { + host = "::1"; + fingerprint = "test"; + class = "default"; +}; + +privset "admin" { + privs = oper:admin; +}; +