diff options
author | Ralph Boehme <slow@samba.org> | 2015-10-18 22:23:20 +0200 |
---|---|---|
committer | Jeremy Allison <jra@samba.org> | 2015-10-20 20:22:22 +0200 |
commit | aa96c75346a9bad60471a206d65c7b7049b9ca83 (patch) | |
tree | 172ffde32db714e9e7efb5a2ec5624df5a4c4a49 /lib/async_req | |
parent | 6c81ecc795d729fc049d42b4df6deff520edd81a (diff) | |
download | samba-aa96c75346a9bad60471a206d65c7b7049b9ca83.tar.gz |
selftest: add a test for async_connect_send()
Bug: https://bugzilla.samba.org/show_bug.cgi?id=11564
Signed-off-by: Ralph Boehme <slow@samba.org>
Reviewed-by: Jeremy Allison <jra@samba.org>
Diffstat (limited to 'lib/async_req')
-rw-r--r-- | lib/async_req/async_connect_send_test.c | 130 | ||||
-rw-r--r-- | lib/async_req/wscript_build | 4 |
2 files changed, 134 insertions, 0 deletions
diff --git a/lib/async_req/async_connect_send_test.c b/lib/async_req/async_connect_send_test.c new file mode 100644 index 00000000000..e61205653bb --- /dev/null +++ b/lib/async_req/async_connect_send_test.c @@ -0,0 +1,130 @@ +/* + * Test async connect + * Copyright (C) Ralph Boehme 2015 + * + * 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 "replace.h" +#include "lib/tevent/tevent.h" +#include "lib/async_req/async_sock.h" +#include <stdio.h> +#include <string.h> +#include <stdlib.h> +#include <errno.h> +#include <sys/types.h> +#include <sys/wait.h> + +int main(int argc, const char *argv[]) +{ + int result, listen_sock, status, exit_status; + uint16_t port; + struct sockaddr_in addr = { 0 }; + pid_t pid; + + listen_sock = socket(PF_INET, SOCK_STREAM, 0); + if (listen_sock == -1) { + perror("socket() failed"); + exit(1); + } + + addr.sin_family = AF_INET; + addr.sin_addr.s_addr = inet_addr("127.0.0.1"); + + for (port = 1024; port < UINT16_MAX; port++) { + addr.sin_port = htons(port); + result = bind(listen_sock, (struct sockaddr *)&addr, sizeof(addr)); + if (result == 0) { + break; + } + } + + if (port == UINT16_MAX) { + printf("Huh, no free port?\n"); + return 1; + } + + result = listen(listen_sock, 1); + if (result == -1) { + perror("listen() failed"); + close(listen_sock); + return 1; + } + + pid = fork(); + if (pid == -1) { + perror("fork"); + return 1; + } + + if (pid == 0) { + struct tevent_context *ev; + struct tevent_req *req; + int fd; + + ev = tevent_context_init(NULL); + if (ev == NULL) { + fprintf(stderr, "tevent_context_init failed\n"); + return 1; + } + + fd = socket(PF_INET, SOCK_STREAM, 0); + if (fd == -1) { + perror("socket"); + return 1; + } + + memset(&addr, sizeof(addr), 0); + addr.sin_family = AF_INET; + addr.sin_port = htons(port); + addr.sin_addr.s_addr = inet_addr("127.0.0.1"); + + req = async_connect_send(ev, ev, fd, + (struct sockaddr *)&addr, + sizeof(struct sockaddr_in), + NULL, NULL, NULL); + + if (!tevent_req_poll(req, ev)) { + perror("tevent_req_poll() failed"); + return 1; + } + + status = 0; + result = async_connect_recv(req, &status); + if (result != 0) { + return status; + } + return 0; + } + + result = waitpid(pid, &status, 0); + if (result == -1) { + perror("waitpid"); + return 1; + } + + if (!WIFEXITED(status)) { + printf("child status: %d\n", status); + return 2; + } + + exit_status = WEXITSTATUS(status); + printf("test done: status=%d\n", exit_status); + + if (exit_status != 0) { + return exit_status; + } + + return 0; +} diff --git a/lib/async_req/wscript_build b/lib/async_req/wscript_build index e8af569b4f9..9c2522376db 100644 --- a/lib/async_req/wscript_build +++ b/lib/async_req/wscript_build @@ -7,3 +7,7 @@ bld.SAMBA_SUBSYSTEM('LIBASYNC_REQ', deps='tevent-util socket-blocking' ) +bld.SAMBA_BINARY('async_connect_send_test', + source='async_connect_send_test.c', + deps='LIBASYNC_REQ' +) |