44 lines
1.5 KiB
C++
44 lines
1.5 KiB
C++
#include "bidirectional_stream_copy.hh"
|
|
|
|
#include <cstdlib>
|
|
#include <cstring>
|
|
#include <iostream>
|
|
|
|
using namespace std;
|
|
|
|
void show_usage(const char *argv0) {
|
|
cerr << "Usage: " << argv0 << " [-l] <host> <port>\n\n"
|
|
<< " -l specifies listen mode; <host>:<port> is the listening address." << endl;
|
|
}
|
|
|
|
int main(int argc, char **argv) {
|
|
try {
|
|
bool server_mode = false;
|
|
if (argc < 3 || ((server_mode = (strncmp("-l", argv[1], 3) == 0)) && argc < 4)) {
|
|
show_usage(argv[0]);
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
// in client mode, connect; in server mode, accept exactly one connection
|
|
auto socket = [&] {
|
|
if (server_mode) {
|
|
TCPSocket listening_socket; // create a TCP socket
|
|
listening_socket.set_reuseaddr(); // reuse the server's address as soon as the program quits
|
|
listening_socket.bind({argv[2], argv[3]}); // bind to specified address
|
|
listening_socket.listen(); // mark the socket as listening for incoming connections
|
|
return listening_socket.accept(); // accept exactly one connection
|
|
}
|
|
TCPSocket connecting_socket;
|
|
connecting_socket.connect({argv[1], argv[2]});
|
|
return connecting_socket;
|
|
}();
|
|
|
|
bidirectional_stream_copy(socket);
|
|
} catch (const exception &e) {
|
|
cerr << "Exception: " << e.what() << endl;
|
|
return EXIT_FAILURE;
|
|
}
|
|
|
|
return EXIT_SUCCESS;
|
|
}
|