Method wait_request

tee/cocoon/FwdProxy.cpp:120–177

  }

  td::Status wait_request() {
    auto buf = pipe_.input_buffer().clone();
    TRY_RESULT(header, read(buf, 4));
    td::uint8 version = header[0];
    td::uint8 cmd = header[1];
    td::uint8 address_type = header[3];
    if (version != 0x5) {
      return td::Status::Error(PSLICE() << "Invalid socks5 request version " << version << ", expected 0x5");
    }
    if (cmd != 0x1) {
      return td::Status::Error(PSLICE() << "Unsupported socks5 command " << cmd);
    }
    std::string host;
    if (address_type == 1) {  // ipv4
      TRY_RESULT(ipv4, read(buf, 4));
      host = PSTRING() << int(td::uint8(ipv4[0])) << "." << int(td::uint8(ipv4[1])) << "." << int(td::uint8(ipv4[2]))
                       << "." << int(td::uint8(ipv4[3]));
    } else if (address_type == 3) {  // host
      return td::Status::Error("TODO: support host destination in socks5");
    } else if (address_type == 4) {  // ipv6
      return td::Status::Error("TODO: support ipv6 destination in socks5");
    } else {
      return td::Status::Error(PSLICE() << "Unsupported address type=" << address_type);
    }
    TRY_RESULT(port_raw, read(buf, 2));
    pipe_.input_buffer() = std::move(buf);

    td::int32 port = td::uint8(port_raw[0]) * 256 + td::uint8(port_raw[1]);
    TRY_STATUS(destination_.init_host_port(host, port));
    LOG(INFO) << "Connect " << destination_;
    TRY_RESULT(socket, td::SocketFd::open(destination_));
    td::IPAddress bound_address;
    TRY_STATUS(bound_address.init_socket_address(socket));
    auto bound_port = bound_address.get_port();

    td::string response;
    response += '\x05';
    response += '\x00';
    response += '\x00';
    if (bound_address.is_ipv4()) {
      response += '\x01';
      auto ipv4 = ntohl(bound_address.get_ipv4());
      response += static_cast<char>(ipv4 & 255);
      response += static_cast<char>((ipv4 >> 8) & 255);
      response += static_cast<char>((ipv4 >> 16) & 255);
      response += static_cast<char>((ipv4 >> 24) & 255);
    } else {
      response += '\x04';
      response += bound_address.get_ipv6();
    }
    response += static_cast<char>((bound_port >> 8) & 255);
    response += static_cast<char>(bound_port & 255);

    pipe_.output_buffer().append(response);
    dest_pipe_ = make_socket_pipe(std::move(socket));
    state_ = State::Done;
    return td::Status::OK();
  }