Blame view

3rdparty/websocketpp-0.8.2/examples/subprotocol_server/subprotocol_server.cpp 1.2 KB
73ef4ff3   Hu Chunming   提交三方库
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
  #include <iostream>
  
  #include <websocketpp/config/asio_no_tls.hpp>
  #include <websocketpp/server.hpp>
  
  typedef websocketpp::server<websocketpp::config::asio> server;
  
  using websocketpp::connection_hdl;
  using websocketpp::lib::placeholders::_1;
  using websocketpp::lib::placeholders::_2;
  using websocketpp::lib::bind;
  using websocketpp::lib::ref;
  
  
  bool validate(server & s, connection_hdl hdl) {
      server::connection_ptr con = s.get_con_from_hdl(hdl);
  
      std::cout << "Cache-Control: " << con->get_request_header("Cache-Control") << std::endl;
  
      const std::vector<std::string> & subp_requests = con->get_requested_subprotocols();
      std::vector<std::string>::const_iterator it;
  
      for (it = subp_requests.begin(); it != subp_requests.end(); ++it) {
          std::cout << "Requested: " << *it << std::endl;
      }
  
      if (subp_requests.size() > 0) {
          con->select_subprotocol(subp_requests[0]);
      }
  
      return true;
  }
  
  int main() {
      try {
          server s;
  
          s.set_validate_handler(bind(&validate,ref(s),::_1));
  
          s.init_asio();
          s.listen(9005);
          s.start_accept();
  
          s.run();
      } catch (websocketpp::exception const & e) {
          std::cout << e.what() << std::endl;
      }
  }