1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
|
/*
* This file and its contents are supplied under the terms of the
* Common Development and Distribution License ("CDDL"), version 1.0.
* You may only use this file in accordance with the terms of version
* 1.0 of the CDDL.
*
* A full copy of the text of the CDDL should have accompanied this
* source. A copy of the CDDL is also available via the Internet at
* http://www.illumos.org/license/CDDL.
*/
/*
* Copyright 2017 Sebastian Wiedenroth
*/
/*
* Test for MSG_NOSIGNAL flag.
*/
#include <unistd.h>
#include <signal.h>
#include <stdio.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <arpa/inet.h>
#include <errno.h>
#include <assert.h>
/* allow the test to build and fail */
#ifndef MSG_NOSIGNAL
#define MSG_NOSIGNAL 0
#endif
volatile sig_atomic_t sigcount = 0;
void
sigpipe_h(int sig0)
{
sigcount++;
signal(SIGPIPE, sigpipe_h);
}
int
main()
{
signal(SIGPIPE, sigpipe_h);
int len = 0;
const char *msg = "hello illumos";
struct sockaddr_in sin;
sin.sin_family = AF_INET;
sin.sin_addr.s_addr = inet_addr("127.0.0.1");
sin.sin_port = htons(4242);
int s = socket(PF_INET, SOCK_STREAM, 0);
int c = socket(PF_INET, SOCK_STREAM, 0);
assert(s >= 0 && c >= 0);
assert(bind(s, (struct sockaddr *)&sin, sizeof (sin)) >= 0);
assert(listen(s, 3) >= 0);
assert(connect(c, (struct sockaddr *)&sin, sizeof (sin)) >= 0);
assert(close(s) == 0);
assert(MSG_NOSIGNAL > 0);
assert(sigcount == 0);
/* test failure with signal */
len = send(c, msg, strlen(msg), 0);
assert(len == -1 && errno == EPIPE);
sleep(1);
assert(sigcount == 1);
/* test failure without signal */
len = send(c, msg, strlen(msg), MSG_NOSIGNAL);
assert(len == -1 && errno == EPIPE);
sleep(1);
assert(sigcount == 1);
assert(close(c) == 0);
return (0);
}
|
