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
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#include <vespa/vespalib/net/socket_spec.h>
#include <vespa/vespalib/net/server_socket.h>
#include <vespa/vespalib/net/socket.h>
#include <vespa/vespalib/util/stringfmt.h>
#include <vespa/vespalib/util/host_name.h>
#include <thread>
#include <functional>
#include <chrono>
using namespace vespalib;
vespalib::string read_msg(SocketHandle &socket) {
vespalib::string msg;
for (;;) {
char c;
ssize_t ret = socket.read(&c, 1);
if (ret != 1) {
fprintf(stderr, "error during read message\n");
return msg;
}
if (c == '\n') {
return msg;
}
msg.append(c);
}
}
void write_msg(SocketHandle &socket, const vespalib::string &msg) {
for (size_t i = 0; i < msg.size(); ++i) {
ssize_t ret = socket.write(&msg[i], 1);
if (ret != 1) {
fprintf(stderr, "error during write message\n");
return;
}
}
}
int main(int argc, char **argv) {
if (argc != 3) {
fprintf(stderr, "usage: %s <host> <port>\n", argv[0]);
return 1;
}
vespalib::string host(argv[1]);
int port = atoi(argv[2]);
fprintf(stderr, "running socket test client at host %s\n", HostName::get().c_str());
fprintf(stderr, "trying to connect to host %s at port %d\n", host.c_str(), port);
auto list = SocketAddress::resolve(port, host.c_str());
if (list.size() > 0) {
fprintf(stderr, "all remote addresses:\n");
for (const auto &addr: list) {
fprintf(stderr, " %s\n", addr.spec().c_str());
}
}
SocketHandle socket = SocketSpec::from_host_port(host, port).client_address().connect();
if (!socket.valid()) {
fprintf(stderr, "connect failed\n");
return 1;
}
fprintf(stderr, "connected to: %s (local address: %s)\n",
SocketAddress::peer_address(socket.get()).spec().c_str(),
SocketAddress::address_of(socket.get()).spec().c_str());
write_msg(socket, "hello from client\n");
fprintf(stderr, "message from server: '%s'\n", read_msg(socket).c_str());
return 0;
}
|
