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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
/* $OpenBSD: client-tcp.c,v 1.2 2020/01/17 20:45:50 bluhm Exp $ */
/*
* Copyright (c) 2020 Alexander Bluhm <bluhm@openbsd.org>
*
* Permission to use, copy, modify, and distribute this software for any
* purpose with or without fee is hereby granted, provided that the above
* copyright notice and this permission notice appear in all copies.
*
* THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
* WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
* MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
* ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
* WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
* ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
* OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
*/
#include <sys/types.h>
#include <sys/socket.h>
#include <err.h>
#include <errno.h>
#include <netdb.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include "util.h"
void __dead usage(void);
int connect_socket(const char *, const char *);
void __dead
usage(void)
{
fprintf(stderr, "client [-r rcvmsg] [-s sndmsg] host port\n"
" -E wait for EOF\n"
" -N shutdown write\n"
" -r rcvmsg receive from server and check message\n"
" -s sndmsg send message to server\n");
exit(2);
}
int
main(int argc, char *argv[])
{
const char *host, *port;
struct task todo[100];
size_t tlen = 0;
int ch, s;
while ((ch = getopt(argc, argv, "ENr:s:")) != -1) {
switch (ch) {
case 'E':
case 'N':
case 'r':
case 's':
if (tlen >= sizeof(todo) / sizeof(todo[0]))
errx(1, "too many tasks");
task_enqueue(&todo[tlen], ch, optarg);
tlen++;
break;
default:
usage();
}
}
argc -= optind;
argv += optind;
if (argc == 2) {
host = argv[0];
port = argv[1];
} else {
usage();
}
alarm_timeout();
s = connect_socket(host, port);
print_sockname(s);
print_peername(s);
task_run(s, todo, tlen);
if (close(s) == -1)
err(1, "close");
return 0;
}
int
connect_socket(const char *host, const char *port)
{
struct addrinfo hints, *res, *res0;
int error;
int save_errno;
int s;
const char *cause = NULL;
memset(&hints, 0, sizeof(hints));
hints.ai_family = AF_UNSPEC;
hints.ai_socktype = SOCK_STREAM;
error = getaddrinfo(host, port, &hints, &res0);
if (error)
errx(1, "%s", gai_strerror(error));
s = -1;
for (res = res0; res; res = res->ai_next) {
s = socket(res->ai_family, res->ai_socktype, res->ai_protocol);
if (s == -1) {
cause = "socket";
continue;
}
if (connect(s, res->ai_addr, res->ai_addrlen) == -1) {
cause = "connect";
save_errno = errno;
close(s);
errno = save_errno;
s = -1;
continue;
}
break; /* okay we got one */
}
if (s == -1)
err(1, "%s", cause);
freeaddrinfo(res0);
return s;
}
|
