From: Vsevolod Stakhov Date: Wed, 6 Oct 2021 10:01:21 +0000 (+0100) Subject: [Test] Move tests from lua to C++ X-Git-Tag: 3.1~80 X-Git-Url: https://source.dussan.org/?a=commitdiff_plain;h=26e33fb5c1be1e979366fcb6a76c2b3b0a7528ff;p=rspamd.git [Test] Move tests from lua to C++ --- diff --git a/src/libmime/received.cxx b/src/libmime/received.cxx index 550823a4d..e38c038ae 100644 --- a/src/libmime/received.cxx +++ b/src/libmime/received.cxx @@ -880,6 +880,7 @@ TEST_CASE("parse received") using namespace std::string_view_literals; using map_type = robin_hood::unordered_flat_map; std::vector> cases{ + // Simple received {"from smtp11.mailtrack.pl (smtp11.mailtrack.pl [185.243.30.90])"sv, { {"real_ip", "185.243.30.90"}, @@ -887,7 +888,102 @@ TEST_CASE("parse received") {"real_hostname", "smtp11.mailtrack.pl"}, {"from_hostname", "smtp11.mailtrack.pl"} } - } + }, + // Real Postfix IPv6 received + {"from server.chat-met-vreemden.nl (unknown [IPv6:2a01:7c8:aab6:26d:5054:ff:fed1:1da2])\n" + "\t(using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits))\n" + "\t(Client did not present a certificate)\n" + "\tby mx1.freebsd.org (Postfix) with ESMTPS id CF0171862\n" + "\tfor ; Mon, 6 Jul 2015 09:01:20 +0000 (UTC)\n" + "\t(envelope-from upwest201diana@outlook.com)", + { + {"real_ip", "2a01:7c8:aab6:26d:5054:ff:fed1:1da2"}, + {"from_ip", "2a01:7c8:aab6:26d:5054:ff:fed1:1da2"}, + {"from_hostname", "server.chat-met-vreemden.nl"}, + {"by_hostname", "mx1.freebsd.org"}, + {"for_mbox", ""} + } + }, + // Exim IPv4 received + {"from localhost ([127.0.0.1]:49019 helo=hummus.csx.cam.ac.uk)\n" + " by hummus.csx.cam.ac.uk with esmtp (Exim 4.91-pdpfix1)\n" + " (envelope-from )\n" + " id 1fZ55o-0006DP-3H\n" + " for ; Sat, 30 Jun 2018 02:54:28 +0100", + { + {"from_hostname", "localhost"}, + {"from_ip", "127.0.0.1"}, + {"real_ip", "127.0.0.1"}, + {"for_mbox", ""}, + {"by_hostname", "hummus.csx.cam.ac.uk"}, + } + }, + // Exim IPv6 received + {"from smtp.spodhuis.org ([2a02:898:31:0:48:4558:736d:7470]:38689\n" + " helo=mx.spodhuis.org)\n" + " by hummus.csx.cam.ac.uk with esmtpsa (TLSv1.3:TLS_AES_256_GCM_SHA384:256)\n" + " (Exim 4.91-pdpfix1+cc) (envelope-from )\n" + " id 1fZ55k-0006CO-9M\n" + " for exim-dev@exim.org; Sat, 30 Jun 2018 02:54:24 +0100", + { + {"from_hostname", "smtp.spodhuis.org"}, + {"from_ip", "2a02:898:31:0:48:4558:736d:7470"}, + {"real_ip", "2a02:898:31:0:48:4558:736d:7470"}, + {"for_mbox", "exim-dev@exim.org"}, + {"by_hostname", "hummus.csx.cam.ac.uk"}, + } + }, + // Haraka received + {"from aaa.cn ([1.1.1.1]) by localhost.localdomain (Haraka/2.8.18) with " + "ESMTPA id 349C9C2B-491A-4925-A687-3EF14038C344.1 envelope-from " + "(authenticated bits=0); Tue, 03 Jul 2018 14:18:13 +0200", + { + {"from_hostname", "aaa.cn"}, + {"from_ip", "1.1.1.1"}, + {"real_ip", "1.1.1.1"}, + {"by_hostname", "localhost.localdomain"}, + } + }, + // Invalid by + {"from [192.83.172.101] (HELLO 148.251.238.35) (148.251.238.35) " + "by guovswzqkvry051@sohu.com with gg login " + "by AOL 6.0 for Windows US sub 008 SMTP ; Tue, 03 Jul 2018 09:01:47 -0300", + { + {"from_hostname", "192.83.172.101"}, + {"from_ip", "192.83.172.101"}, + {"real_ip", "192.83.172.101"}, + } + }, + // Invalid hostinfo + {"from example.com ([]) by example.com with ESMTP id 2019091111 ;" + " Thu, 26 Sep 2019 11:19:07 +0200", + { + {"by_hostname", "example.com"}, + {"from_hostname", "example.com"}, + {"real_hostname", "example.com"}, + } + }, + // Different real and announced hostnames + broken crap + {"from 171-29.br (1-1-1-1.z.com.br [1.1.1.1]) by x.com.br (Postfix) " + "with;ESMTP id 44QShF6xj4z1X for ; Thu, 21 Mar 2019 23:45:46 -0300 " + ": ", + { + {"real_ip", "1.1.1.1"}, + {"from_ip", "1.1.1.1"}, + {"from_hostname", "171-29.br"}, + {"real_hostname", "1-1-1-1.z.com.br"}, + {"by_hostname", "x.com.br"}, + } + }, + // Different real and announced ips + no hostname + {"from [127.0.0.1] ([127.0.0.2]) by smtp.gmail.com with ESMTPSA id xxxololo", + { + {"real_ip", "127.0.0.2"}, + {"from_ip", "127.0.0.2"}, + {"from_hostname", "127.0.0.1"}, + {"by_hostname", "smtp.gmail.com"}, + } + }, }; rspamd_mempool_t *pool = rspamd_mempool_new_default("rcvd test", 0); diff --git a/src/libmime/received.hxx b/src/libmime/received.hxx index 98d46770b..de9df52bb 100644 --- a/src/libmime/received.hxx +++ b/src/libmime/received.hxx @@ -114,7 +114,8 @@ struct received_header { : from_hostname(received_char_filter), real_hostname(received_char_filter), real_ip(received_char_filter), - by_hostname(received_char_filter) {} + by_hostname(received_char_filter), + for_mbox() {} /* We have raw C pointers, so copy is explicitly disabled */ received_header(const received_header &other) = delete; received_header(received_header &&other) noexcept { diff --git a/test/lua/unit/received.lua b/test/lua/unit/received.lua deleted file mode 100644 index 34882e1cc..000000000 --- a/test/lua/unit/received.lua +++ /dev/null @@ -1,226 +0,0 @@ --- inet addr tests - -context("Received headers parser", function() - local ffi = require("ffi") - local rspamd_ip = require "rspamd_ip" - - ffi.cdef[[ - struct received_header { - const char *from_hostname; - const char *from_ip; - const char *real_hostname; - const char *real_ip; - const char *by_hostname; - const char *for_mbox; - void *for_addr; - void *addr; - void *hdr; - long timestamp; - int flags; /* See enum rspamd_received_type */ - struct received_header *prev, *next; - }; - struct rspamd_task * rspamd_task_new(struct rspamd_worker *worker, struct rspamd_config *cfg); - int rspamd_smtp_received_parse (struct rspamd_task *task, - const char *data, size_t len, struct received_header *rh); - ]] - - local cases = { - {[[from smtp11.mailtrack.pl (smtp11.mailtrack.pl [185.243.30.90])]], - { - real_ip = '185.243.30.90', - real_hostname = 'smtp11.mailtrack.pl' - }, - }, - {[[from asx121.turbo-inline.com [7.165.23.113] by mx.reskind.net with QMQP; Fri, 08 Feb 2019 06:56:18 -0500]], - { - real_ip = '7.165.23.113', - real_hostname = 'asx121.turbo-inline.com', - } - }, - {[[from server.chat-met-vreemden.nl (unknown [IPv6:2a01:7c8:aab6:26d:5054:ff:fed1:1da2]) - (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) - (Client did not present a certificate) - by mx1.freebsd.org (Postfix) with ESMTPS id CF0171862 - for ; Mon, 6 Jul 2015 09:01:20 +0000 (UTC) - (envelope-from upwest201diana@outlook.com)]], - { - real_ip = '2a01:7c8:aab6:26d:5054:ff:fed1:1da2', - from_hostname = 'server.chat-met-vreemden.nl', - real_hostname = '', - by_hostname = 'mx1.freebsd.org', - }, - }, - {[[from out-9.smtp.github.com (out-9.smtp.github.com [192.30.254.192]) - (using TLSv1.2 with cipher AECDH-AES256-SHA (256/256 bits)) - (No client certificate requested) - by mail.highsecure.ru (Postfix) with ESMTPS id C7B1A30014A - for ; Tue, 3 Jul 2018 14:40:19 +0200 (CEST)]], - { - from_hostname = 'out-9.smtp.github.com', - from_ip = '192.30.254.192', - real_ip = '192.30.254.192', - by_hostname = 'mail.highsecure.ru', - } - }, - {[[from localhost ([127.0.0.1]:49019 helo=hummus.csx.cam.ac.uk) - by hummus.csx.cam.ac.uk with esmtp (Exim 4.91-pdpfix1) - (envelope-from ) - id 1fZ55o-0006DP-3H - for ; Sat, 30 Jun 2018 02:54:28 +0100]], - { - from_hostname = 'localhost', - from_ip = '127.0.0.1', - real_ip = '127.0.0.1', - by_hostname = 'hummus.csx.cam.ac.uk', - } - }, - {[[from smtp.spodhuis.org ([2a02:898:31:0:48:4558:736d:7470]:38689 - helo=mx.spodhuis.org) - by hummus.csx.cam.ac.uk with esmtpsa (TLSv1.3:TLS_AES_256_GCM_SHA384:256) - (Exim 4.91-pdpfix1+cc) (envelope-from ) - id 1fZ55k-0006CO-9M - for exim-dev@exim.org; Sat, 30 Jun 2018 02:54:24 +0100]], - { - from_hostname = 'smtp.spodhuis.org', - from_ip = '2a02:898:31::48:4558:736d:7470', - real_ip = '2a02:898:31::48:4558:736d:7470', - by_hostname = 'hummus.csx.cam.ac.uk', - } - }, - {'from aaa.cn ([1.1.1.1]) by localhost.localdomain (Haraka/2.8.18) with ESMTPA id 349C9C2B-491A-4925-A687-3EF14038C344.1 envelope-from (authenticated bits=0); Tue, 03 Jul 2018 14:18:13 +0200', - { - from_hostname = 'aaa.cn', - from_ip = '1.1.1.1', - real_ip = '1.1.1.1', - } - }, - {'from [192.83.172.101] (HELLO 148.251.238.35) (148.251.238.35) by guovswzqkvry051@sohu.com with gg login by AOL 6.0 for Windows US sub 008 SMTP ; Tue, 03 Jul 2018 09:01:47 -0300', - { - from_ip = '192.83.172.101', - by_hostname = '', - }, - }, - {'from [61.174.163.26] (helo=host) by sc8-sf-list1.sourceforge.net with smtp (Exim 3.31-VA-mm2 #1 (Debian)) id 18t2z0-0001NX-00 for ; Wed, 12 Mar 2003 01:57:10 -0800', - { - from_ip = '61.174.163.26', - by_hostname = 'sc8-sf-list1.sourceforge.net', - }, - }, - {[[from [127.0.0.1] (unknown [65.19.167.131]) - (using TLSv1.2 with cipher ECDHE-RSA-AES256-GCM-SHA384 (256/256 bits)) - (Client did not present a certificate) - by mail01.someotherdomain.org (Postfix) with ESMTPSA id 43tYMW2yKHz50MHS - for ; Mon, 4 Feb 2019 16:39:35 +0000 (GMT)]], - { - from_ip = '65.19.167.131', - real_ip = '65.19.167.131', - by_hostname = 'mail01.someotherdomain.org', - ['for'] = 'user2@somedomain.com', - } - }, - {[[from example.com ([]) by example.com with ESMTP id 2019091111 ; Thu, 26 Sep 2019 11:19:07 +0200]], - { - by_hostname = 'example.com', - }, - }, - {[[from 171-29.br (1-1-1-1.z.com.br [1.1.1.1]) by x.com.br (Postfix) with;ESMTP id 44QShF6xj4z1X for ; Thu, 21 Mar 2019 23:45:46 -0300 : ]], - { - from_hostname = '171-29.br', - real_ip = '1.1.1.1', - by_hostname = 'x.com.br', - } - }, - {[[from [127.0.0.1] ([127.0.0.2]) - by smtp.gmail.com with ESMTPSA id xxxololo]], - { - from_hostname = '127.0.0.1', - real_ip = '127.0.0.2', - by_hostname = 'smtp.gmail.com' - } - }, - } - - local task = ffi.C.rspamd_task_new(nil, nil) - local NULL = ffi.new 'void*' - local function ffi_string(fs) - if fs ~= NULL then return ffi.string(fs) end - return nil - end - local function ip_check(ret) - local sret = ffi_string(ret) - - if not sret then return 'null' end - local ip = rspamd_ip.from_string(sret) - - if not ip then return 'not ip' end - if not ip:is_valid() then return 'unparsed' end - return tostring(ip) - end - - for i,c in ipairs(cases) do - test("Parse received " .. i, function() - local hdr = ffi.new("struct received_header") - c[1] = c[1]:gsub('\n', ' ') -- Replace folding - ffi.C.rspamd_smtp_received_parse(task, c[1], #c[1], hdr) - - for k,v in pairs(c[2]) do - if k == 'from_hostname' then - if #v > 0 then - assert_equal(v, ffi_string(hdr.from_hostname), - string.format('%s: from_hostname: %s, expected: %s', - c[1], ffi_string(hdr.from_hostname), v)) - else - assert_nil(hdr.from_hostname, - string.format('%s: from_hostname: %s, expected: nil', - c[1], ffi_string(hdr.from_hostname))) - end - elseif k == 'from_ip' then - if #v > 0 then - local got_string = ip_check(hdr.from_ip) - local expected_string = tostring(rspamd_ip.from_string(v)) - assert_equal(expected_string, got_string, - string.format('%s: from_ip: %s, expected: %s', - expected_string, got_string, v)) - else - assert_nil(hdr.from_ip, - string.format('%s: from_ip: %s, expected: nil', - c[1], ffi_string(hdr.from_ip))) - end - elseif k == 'real_ip' then - if #v > 0 then - local got_string = ip_check(hdr.real_ip) - local expected_string = tostring(rspamd_ip.from_string(v)) - assert_equal(expected_string, got_string, - string.format('%s: real_ip: %s, expected: %s', - expected_string, got_string, v)) - else - assert_nil(hdr.real_ip, - string.format('%s: real_ip: %s, expected: nil', - c[1], ffi_string(hdr.real_ip))) - end - elseif k == 'by_hostname' then - if #v > 0 then - assert_equal(v, ffi_string(hdr.by_hostname), - string.format('%s: by_hostname: %s, expected: %s', - c[1], ffi_string(hdr.by_hostname), v)) - else - assert_nil(hdr.by_hostname, - string.format('%s: by_hostname: %s, expected: nil', - c[1], ffi_string(hdr.by_hostname))) - end - elseif k == 'for' then - if #v > 0 then - assert_equal(v, ffi_string(hdr['for_mbox']), - string.format('%s: for: %s, expected: %s', - c[1], ffi_string(hdr['for_mbox']), v)) - else - assert_nil(hdr['for_mbox'], - string.format('%s: for: %s, expected: nil', - c[1], ffi_string(hdr['for_mbox']))) - end - end - end - end) - - end -end) \ No newline at end of file