forked from nothingelsematters/synchronous-socket-service
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Client.cpp
102 lines (88 loc) · 2.58 KB
/
Client.cpp
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
//
// Created by Павел Пономарев on 2019-05-19.
//
#include "Client.h"
#include <unistd.h>
#include <sys/socket.h>
#include <netinet/in.h>
bool Client::run(std::string const& argument) {
char serverResponse[BUFFER_SIZE];
int address = parseInt(argument);
int sock = openSocket();
connectSocket(sock, address);
std::string request;
std::cout << "Request: ";
getline(std::cin, request);
ssize_t sent = 0;
ssize_t left = request.size();
while (sent < request.size()) {
ssize_t currentMessage = send(sock, request.data() + sent, left, 0);
if (currentMessage == -1) {
printErrorMessage("sending failure");
if (close(sock) == -1) {
printErrorMessage("Fail to close");
}
exit(EXIT_FAILURE);
} else {
sent += currentMessage;
left -= currentMessage;
}
}
ssize_t received = 0;
std::cout << "Server response:" << std::endl;
while (received < request.size()) {
ssize_t res = recv(sock, serverResponse, BUFFER_SIZE, 0);
if (res == 0) {
break;
}
if (res == -1) {
printErrorMessage("Receiving failure");
if (close(sock) == -1) {
printErrorMessage("Fail to close");
}
exit(EXIT_FAILURE);
}
for (int i = 0; i < res; ++i) {
std::cout << serverResponse[i];
}
received += res;
}
std::cout << std::endl;
if (close(sock) == -1) {
printErrorMessage("Fail to close");
}
return true;
}
int Client::openSocket() {
int sock = socket(AF_INET, SOCK_STREAM, 0);
if (sock < 0) {
return -1;
}
int opt = 1;
if (setsockopt(sock, SOL_SOCKET, SO_REUSEADDR, &opt, sizeof(opt)) < 0) {
return -2;
}
return sock;
}
bool Client::connectSocket(int socket, int address) {
sockaddr_in addr{};
addr.sin_port = htons(address);
addr.sin_addr.s_addr = htonl(INADDR_LOOPBACK);
addr.sin_family = AF_INET;
return (connect(socket, (struct sockaddr*) &addr, sizeof(addr)) >= 0);
}
void printHelpMessage() {
std::cout << "os-net-client <address> -- run client; \n "
"os-net-client help | -help | --help -- print this message.\n";
}
int main(int argc, char** argv) {
if (argc < 2) {
printErrorMessage("incorrect number of arguments", false);
printHelpMessage();
return EXIT_FAILURE;
}
std::string argument(argv[1]);
Client client;
client.run(argument);
return EXIT_SUCCESS;
}