diff options
Diffstat (limited to 'source3/torture')
-rw-r--r-- | source3/torture/proto.h | 1 | ||||
-rw-r--r-- | source3/torture/test_msg.c | 131 | ||||
-rw-r--r-- | source3/torture/torture.c | 1 |
3 files changed, 133 insertions, 0 deletions
diff --git a/source3/torture/proto.h b/source3/torture/proto.h index 6e2a00450f..b6da0f477c 100644 --- a/source3/torture/proto.h +++ b/source3/torture/proto.h @@ -105,5 +105,6 @@ bool run_local_sprintf_append(int dummy); bool run_cleanup1(int dummy); bool run_cleanup2(int dummy); bool run_ctdb_conn(int dummy); +bool run_msg_test(int dummy); #endif /* __TORTURE_H__ */ diff --git a/source3/torture/test_msg.c b/source3/torture/test_msg.c new file mode 100644 index 0000000000..88b07e742c --- /dev/null +++ b/source3/torture/test_msg.c @@ -0,0 +1,131 @@ +/* + Unix SMB/CIFS implementation. + Test msg_stream API + Copyright (C) Volker Lendecke 2012 + + 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 3 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, see <http://www.gnu.org/licenses/>. +*/ + +#include "includes.h" +#include "torture/proto.h" +#include "lib/util/tevent_unix.h" +#include "msg_channel.h" + +struct msg_test_state { + struct tevent_context *ev; + struct messaging_context *msg; + struct msg_channel *channel; +}; + +static void msg_test_got_channel(struct tevent_req *subreq); +static void msg_test_got_msg(struct tevent_req *subreq); + +static struct tevent_req *msg_test_send(TALLOC_CTX *mem_ctx, + struct tevent_context *ev) +{ + struct tevent_req *req, *subreq; + struct msg_test_state *state; + + req = tevent_req_create(mem_ctx, &state, struct msg_test_state); + if (req == NULL) { + return NULL; + } + state->ev = ev; + + state->msg = messaging_init(state, state->ev); + if (tevent_req_nomem(state->msg, req)) { + return tevent_req_post(req, ev); + } + subreq = msg_channel_init_send(state, state->ev, state->msg, MSG_PING); + if (tevent_req_nomem(subreq, req)) { + return tevent_req_post(req, ev); + } + tevent_req_set_callback(subreq, msg_test_got_channel, req); + return req; +} + +static void msg_test_got_channel(struct tevent_req *subreq) +{ + struct tevent_req *req = tevent_req_callback_data( + subreq, struct tevent_req); + struct msg_test_state *state = tevent_req_data( + req, struct msg_test_state); + int ret; + + ret = msg_channel_init_recv(subreq, state, &state->channel); + TALLOC_FREE(subreq); + if (tevent_req_error(req, ret)) { + return; + } + subreq = msg_read_send(state, state->ev, state->channel); + if (tevent_req_nomem(subreq, req)) { + return; + } + tevent_req_set_callback(subreq, msg_test_got_msg, req); +} + +static void msg_test_got_msg(struct tevent_req *subreq) +{ + struct tevent_req *req = tevent_req_callback_data( + subreq, struct tevent_req); + struct msg_test_state *state = tevent_req_data( + req, struct msg_test_state); + struct messaging_rec *msg; + int ret; + + ret = msg_read_recv(subreq, state, &msg); + TALLOC_FREE(subreq); + if (tevent_req_error(req, ret)) { + return; + } + tevent_req_done(req); +} + +static int msg_test_recv(struct tevent_req *req) +{ + int err; + + if (tevent_req_is_unix_error(req, &err)) { + return err; + } + return 0; +} + +bool run_msg_test(int dummy) +{ + struct tevent_context *ev; + struct tevent_req *req; + int ret; + + ev = tevent_context_init(talloc_tos()); + if (ev == NULL) { + fprintf(stderr, "tevent_context_init failed\n"); + return false; + } + req = msg_test_send(ev, ev); + if (req == NULL) { + fprintf(stderr, "msg_test_send failed\n"); + return false; + } + if (!tevent_req_poll(req, ev)) { + fprintf(stderr, "tevent_req_poll failed\n"); + return false; + } + ret = msg_test_recv(req); + TALLOC_FREE(req); + printf("msg_test_recv returned %s\n", + ret ? strerror(ret) : "success"); + TALLOC_FREE(ev); + return (ret == 0); +} diff --git a/source3/torture/torture.c b/source3/torture/torture.c index 0c421b5342..ae0d6d4c08 100644 --- a/source3/torture/torture.c +++ b/source3/torture/torture.c @@ -8918,6 +8918,7 @@ static struct { { "LOCAL-GENCACHE", run_local_gencache, 0}, { "LOCAL-TALLOC-DICT", run_local_talloc_dict, 0}, { "LOCAL-CTDB-CONN", run_ctdb_conn, 0}, + { "LOCAL-MSG", run_msg_test, 0}, { "LOCAL-BASE64", run_local_base64, 0}, { "LOCAL-RBTREE", run_local_rbtree, 0}, { "LOCAL-MEMCACHE", run_local_memcache, 0}, |