You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.1 KiB
93 lines
2.1 KiB
#include <ccan/io/io.h>
|
|
/* Include the C files directly. */
|
|
#include <ccan/io/poll.c>
|
|
#include <ccan/io/io.c>
|
|
#include <ccan/tap/tap.h>
|
|
#include <sys/wait.h>
|
|
#include <stdio.h>
|
|
|
|
#define PORT "65021"
|
|
|
|
static struct io_listener *l;
|
|
|
|
static struct io_plan *steal_fd(struct io_conn *conn, int *fd)
|
|
{
|
|
io_close_listener(l);
|
|
*fd = io_conn_fd(conn);
|
|
return io_close_taken_fd(conn);
|
|
}
|
|
|
|
static int make_listen_fd(const char *port, struct addrinfo **info)
|
|
{
|
|
int fd, on = 1;
|
|
struct addrinfo *addrinfo, hints;
|
|
|
|
memset(&hints, 0, sizeof(hints));
|
|
hints.ai_family = AF_UNSPEC;
|
|
hints.ai_socktype = SOCK_STREAM;
|
|
hints.ai_flags = AI_PASSIVE;
|
|
hints.ai_protocol = 0;
|
|
|
|
if (getaddrinfo(NULL, port, &hints, &addrinfo) != 0)
|
|
return -1;
|
|
|
|
fd = socket(addrinfo->ai_family, addrinfo->ai_socktype,
|
|
addrinfo->ai_protocol);
|
|
if (fd < 0)
|
|
return -1;
|
|
|
|
setsockopt(fd, SOL_SOCKET, SO_REUSEADDR, &on, sizeof(on));
|
|
if (bind(fd, addrinfo->ai_addr, addrinfo->ai_addrlen) != 0) {
|
|
close(fd);
|
|
return -1;
|
|
}
|
|
if (listen(fd, 1) != 0) {
|
|
close(fd);
|
|
return -1;
|
|
}
|
|
*info = addrinfo;
|
|
return fd;
|
|
}
|
|
|
|
int main(void)
|
|
{
|
|
struct addrinfo *addrinfo = NULL;
|
|
int i, fd, in_fd, status;
|
|
char buf[strlen("hellothere")];
|
|
|
|
/* This is how many tests you plan to run */
|
|
plan_tests(15);
|
|
fd = make_listen_fd(PORT, &addrinfo);
|
|
l = io_new_listener(NULL, fd, steal_fd, &in_fd);
|
|
fflush(stdout);
|
|
if (!fork()) {
|
|
io_close_listener(l);
|
|
fd = socket(addrinfo->ai_family, addrinfo->ai_socktype,
|
|
addrinfo->ai_protocol);
|
|
if (fd < 0)
|
|
exit(1);
|
|
if (connect(fd, addrinfo->ai_addr, addrinfo->ai_addrlen) != 0)
|
|
exit(2);
|
|
signal(SIGPIPE, SIG_IGN);
|
|
for (i = 0; i < strlen("hellothere"); i++) {
|
|
if (write(fd, "hellothere" + i, 1) != 1)
|
|
break;
|
|
}
|
|
close(fd);
|
|
freeaddrinfo(addrinfo);
|
|
exit(0);
|
|
}
|
|
freeaddrinfo(addrinfo);
|
|
ok1(io_loop(NULL, NULL) == NULL);
|
|
|
|
for (i = 0; i < strlen("hellothere"); i++)
|
|
ok1(read(in_fd, buf + i, 1) == 1);
|
|
|
|
ok1(memcmp(buf, "hellothere", sizeof(buf)) == 0);
|
|
ok1(wait(&status));
|
|
ok1(WIFEXITED(status));
|
|
ok1(WEXITSTATUS(status) == 0);
|
|
|
|
/* This exits depending on whether all tests passed */
|
|
return exit_status();
|
|
}
|
|
|