46 lines
1.9 KiB
C
46 lines
1.9 KiB
C
/* SIGPIPE остаётся с действием по умолчанию: оба пути STCP обязаны вернуть ошибку закрытого пира. */
|
|
#include "stcp.h"
|
|
#include "../lib/platform_compat.h"
|
|
#include "../lib/mem.h"
|
|
#include <signal.h>
|
|
#include <sys/wait.h>
|
|
#include <stdlib.h>
|
|
|
|
#define CHECK(x) do { if (!(x)) { DEBUG_ERROR(DEBUG_CATEGORY_GENERAL, "FAIL %s:%d: %s", __func__, __LINE__, #x); exit(1); } } while (0)
|
|
|
|
static void check_closed_peer(int buffered) {
|
|
int sockets[2]; CHECK(socketpair(AF_UNIX, SOCK_STREAM, 0, sockets) == 0);
|
|
CHECK(socket_close_wrapper(sockets[1]) == 0);
|
|
struct UASYNC* ua = uasync_create(); CHECK(ua);
|
|
struct stcp_conn conn = {.ua = ua, .sock = sockets[0], .state = STCP_STATE_DATA};
|
|
uint8_t* bytes = u_malloc(16); CHECK(bytes); memset(bytes, 0, 16);
|
|
if (buffered) {
|
|
conn.send_buf = bytes; conn.send_len = 16;
|
|
stcp_write_cb(conn.sock, &conn);
|
|
CHECK(conn.close_err == EPIPE && conn.close_call_soon);
|
|
uasync_drain_immediate(ua);
|
|
CHECK(conn.state == STCP_STATE_CLOSED && !conn.send_buf);
|
|
} else {
|
|
CHECK(stcp_send_raw(&conn, bytes, 16) < 0 && socket_get_error() == EPIPE);
|
|
u_free(bytes); stcp_conn_do_close(&conn, EPIPE);
|
|
}
|
|
uasync_poll(ua, 0); CHECK(ua->timer_alloc_count == ua->timer_free_count);
|
|
uasync_destroy(ua, 0);
|
|
}
|
|
|
|
int main(void) {
|
|
debug_config_init(); debug_set_level(DEBUG_LEVEL_ERROR);
|
|
for (int buffered = 0; buffered < 2; buffered++) {
|
|
pid_t child = fork(); CHECK(child >= 0);
|
|
if (!child) {
|
|
CHECK(signal(SIGPIPE, SIG_DFL) != SIG_ERR);
|
|
size_t baseline = u_get_allocated_count();
|
|
check_closed_peer(buffered); CHECK(u_get_allocated_count() == baseline);
|
|
exit(0);
|
|
}
|
|
int status; CHECK(waitpid(child, &status, 0) == child);
|
|
CHECK(WIFEXITED(status) && WEXITSTATUS(status) == 0);
|
|
}
|
|
return 0;
|
|
}
|