test/count: replace use of gdb with a LD_PRELOAD shim

There is a certain amount of boilerplate to pass the call on the
original function, so abstract it out as a C preprocessor macro, plus
some extra includes in notmuch-test.h
This commit is contained in:
David Bremner 2021-10-24 22:15:14 -03:00
parent 1643c0459a
commit 9397e7e8eb
2 changed files with 42 additions and 12 deletions

View file

@ -102,22 +102,25 @@ output=$(sed 's/^\(A Xapian exception [^:]*\):.*$/\1/' OUTPUT)
test_expect_equal "${output}" "A Xapian exception occurred opening database"
restore_database
cat <<EOF > count-files.gdb
set breakpoint pending on
set logging file count-files-gdb.log
set logging on
break count_files
commands
shell cp /dev/null ${MAIL_DIR}/.notmuch/xapian/postlist.*
continue
end
run
make_shim qsm-shim<<EOF
#include <notmuch-test.h>
WRAP_DLFUNC (notmuch_status_t, notmuch_query_search_messages, (notmuch_query_t *query, notmuch_messages_t **messages))
/* XXX WARNING THIS CORRUPTS THE DATABASE */
int fd = open ("target_postlist", O_WRONLY|O_TRUNC);
if (fd < 0)
exit (8);
close (fd);
return notmuch_query_search_messages_orig(query, messages);
}
EOF
backup_database
test_begin_subtest "error message from query_search_messages"
${TEST_GDB} --batch-silent --return-child-result -x count-files.gdb \
--args notmuch count --output=files '*' 2>OUTPUT 1>/dev/null
ln -s ${MAIL_DIR}/.notmuch/xapian/postlist.* target_postlist
notmuch_with_shim qsm-shim count --output=files '*' 2>OUTPUT 1>/dev/null
cat <<EOF > EXPECTED
notmuch count: A Xapian exception occurred
A Xapian exception occurred performing query

View file

@ -1,9 +1,17 @@
#ifndef _NOTMUCH_TEST_H
#define _NOTMUCH_TEST_H
#ifndef _GNU_SOURCE
#define _GNU_SOURCE
#endif
#include <assert.h>
#include <dlfcn.h>
#include <fcntl.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <talloc.h>
#include <unistd.h>
@ -20,4 +28,23 @@ expect0 (int line, notmuch_status_t ret)
}
#define EXPECT0(v) expect0 (__LINE__, v);
inline static void *
dlsym_next (const char *symbol)
{
void *sym = dlsym (RTLD_NEXT, symbol);
char *str = dlerror ();
if (str != NULL) {
fprintf (stderr, "finding symbol '%s' failed: %s", symbol, str);
exit (77);
}
return sym;
}
#define WRAP_DLFUNC(_rtype, _func, _args) \
_rtype _func _args; \
_rtype _func _args { \
static _rtype (*_func##_orig) _args = NULL; \
if (! _func##_orig ) *(void **) (&_func##_orig) = dlsym_next (#_func);
#endif