1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
|
#############################################################################
# Copyright (c) 2007-2015 Balabit
#
# This program is free software; you can redistribute it and/or modify it
# under the terms of the GNU General Public License version 2 as published
# by the Free Software Foundation, 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 St, Fifth Floor, Boston, MA 02110-1301 USA
#
# As an additional exemption you are allowed to compile & link against the
# OpenSSL libraries as published by the OpenSSL project. See the file
# COPYING for details.
#
#############################################################################
from globals import *
from log import *
from messagegen import *
from messagecheck import *
from control import flush_files, stop_syslogng
config = """@version: %(syslog_ng_version)s
options { ts_format(iso); chain_hostnames(no); keep_hostname(yes); threaded(yes); };
source s_int { internal(); };
source s_tcp { tcp(port(%(port_number)d)); };
destination d_sql {
sql(type(sqlite3) database("%(current_dir)s/test-sql.db") host(dummy) port(1234) username(dummy) password(dummy)
table("logs")
null("@NULL@")
columns("date datetime", "host", "program", "pid", "msg", "dummy int default 5678")
values("$DATE", "$HOST", "$PROGRAM", "${PID:-@NULL@}", "$MSG", default)
indexes("date", "host", "program")
flags(explicit-commits)
flush-lines(25) flush_timeout(100));
};
log { source(s_tcp); destination(d_sql); };
""" % locals()
def check_env():
if not has_module('afsql'):
print('afsql module is not available, skipping SQL test')
return False
paths=('/opt/syslog-ng/bin', '/usr/bin', '/usr/local/bin')
found=False
for pth in paths:
if os.path.isfile(os.path.join(pth, 'sqlite3')):
found = True
if not found:
print_user("no sqlite3 tool, skipping SQL test\nSearched: %s\n" % ':'.join(paths))
return False
soext='.so'
if re.match('hp-ux', sys.platform) and not re.match('ia64', os.uname()[4]):
soext='.sl'
found = False
paths = (os.environ.get('dbd_dir', ''), '/usr/local/lib/dbd', '/usr/lib/dbd',
'/usr/lib64/dbd/', '/opt/syslog-ng/lib/dbd', '/usr/lib/x86_64-linux-gnu/dbd')
for pth in paths:
if pth and os.path.isfile('%s/libdbdsqlite3%s' % (pth, soext)):
found = True
break
if not found:
print_user('No sqlite3 backend for libdbi. Skipping SQL test.\nSearched: %s\n' % ':'.join(paths))
return False
print('sqlite3 found, proceeding to SQL tests')
return True
def test_sql():
messages = (
'sql1',
'sql2'
)
s = SocketSender(AF_INET, ('localhost', port_number), dgram=0)
expected = []
for msg in messages:
expected.extend(s.sendMessages(msg, pri=7))
print_user("Waiting for 10 seconds until syslog-ng writes all records to the SQL table")
time.sleep(10)
stopped = stop_syslogng()
time.sleep(5)
return stopped and check_sql_expected("%s/test-sql.db" % current_dir, "logs", expected, settle_time=5, syslog_prefix="Sep 7 10:43:21 bzorp prog 12345")
|