Blame view

examples/tcp/tcpecho.cpp 1.51 KB
48b4c725   Peter M. Groen   Setting up Socket-pp
1
2
  #include <iostream>
  #include <string>
18a2dbfb   Peter M. Groen   Fixed paths
3
4
  #include "socket-cpp/tcp_connector.h"
  #include "socket-cpp/version.h"
48b4c725   Peter M. Groen   Setting up Socket-pp
5
6
7
8
9
10
11
  
  using namespace std;
  using namespace std::chrono;
  
  int main(int argc, char* argv[])
  {
  	cout << "Sample TCP echo client for 'sockpp' "
18a2dbfb   Peter M. Groen   Fixed paths
12
  		<< osdev::components::socket-cpp::SOCKPP_VERSION << '\n' << endl;
48b4c725   Peter M. Groen   Setting up Socket-pp
13
14
15
16
  
  	string host = (argc > 1) ? argv[1] : "localhost";
  	in_port_t port = (argc > 2) ? atoi(argv[2]) : 12345;
  
18a2dbfb   Peter M. Groen   Fixed paths
17
  	osdev::components::socket-cpp::socket_initializer sockInit;
48b4c725   Peter M. Groen   Setting up Socket-pp
18
19
20
21
  
  	// Implicitly creates an inet_address from {host,port}
  	// and then tries the connection.
  
18a2dbfb   Peter M. Groen   Fixed paths
22
  	osdev::components::socket-cpp::tcp_connector conn({host, port});
48b4c725   Peter M. Groen   Setting up Socket-pp
23
24
  	if (!conn) {
  		cerr << "Error connecting to server at "
18a2dbfb   Peter M. Groen   Fixed paths
25
  			<< osdev::components::socket-cpp::inet_address(host, port)
48b4c725   Peter M. Groen   Setting up Socket-pp
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
  			<< "\n\t" << conn.last_error_str() << endl;
  		return 1;
  	}
  
  	cout << "Created a connection from " << conn.address() << endl;
  
      // Set a timeout for the responses
      if (!conn.read_timeout(seconds(5))) {
          cerr << "Error setting timeout on TCP stream: "
                  << conn.last_error_str() << endl;
      }
  
  	string s, sret;
  	while (getline(cin, s) && !s.empty()) {
  		if (conn.write(s) != ssize_t(s.length())) {
  			cerr << "Error writing to the TCP stream: "
  				<< conn.last_error_str() << endl;
  			break;
  		}
  
  		sret.resize(s.length());
  		ssize_t n = conn.read_n(&sret[0], s.length());
  
  		if (n != ssize_t(s.length())) {
  			cerr << "Error reading from TCP stream: "
  				<< conn.last_error_str() << endl;
  			break;
  		}
  
  		cout << sret << endl;
  	}
  
  	return (!conn) ? 1 : 0;
  }