lib: Enhance poll_funcs_tevent for multiple tevent_contexts

[mat/samba.git] / source3 / lib / unix_msg / test_drain.c

diff --git a/source3/lib/unix_msg/test_drain.c b/source3/lib/unix_msg/test_drain.c
index 6fe8c188367912de5a6396091c3d41fe6b80c304..c2568b6646b2632e829dce3205c35e12c2a625d6 100644 (file)
--- a/source3/lib/unix_msg/test_drain.c
+++ b/source3/lib/unix_msg/test_drain.c
@@ -16,7 +16,7 @@ static void recv_cb(struct unix_msg_ctx *ctx,
 
 int main(int argc, const char *argv[])
 {
-       struct poll_funcs funcs;
+       struct poll_funcs *funcs;
        const char *sock;
        struct unix_msg_ctx *ctx;
        struct tevent_context *ev;
@@ -37,10 +37,13 @@ int main(int argc, const char *argv[])
                perror("tevent_context_init failed");
                return 1;
        }
-       poll_funcs_init_tevent(&funcs, ev);
+       funcs = poll_funcs_init_tevent(ev);
+       if (funcs == NULL) {
+               fprintf(stderr, "poll_funcs_init_tevent failed\n");
+               return 1;
+       }
 
-       ret = unix_msg_init(sock, &funcs, 256, 1,
-                           recv_cb, &state, &ctx);
+       ret = unix_msg_init(sock, funcs, 256, 1, recv_cb, &state, &ctx);
        if (ret != 0) {
                fprintf(stderr, "unix_msg_init failed: %s\n",
                        strerror(ret));