Пример #1
0
HttpRequest::HttpRequest(
  Method method,
  const yield::uri::Uri& uri,
  YO_NEW_REF Object* body,
  uint8_t http_version
)
  : HttpMessage<HttpRequest>(body, http_version),
    method(method),
    uri(uri) {
  get_header().put(method.get_name(), method.get_name_len());

  get_header().put(' ');

  iovec uri_path;
  uri.get_path(uri_path);
  get_header().put(uri_path);

  if (http_version == 0) {
    get_header().put(" HTTP/1.0\r\n", 11);
  } else {
    get_header().put(" HTTP/1.1\r\n", 11);
  }

  set_fields_offset(static_cast<uint16_t>(get_header().size()));

  if (uri.has_host()) {
    iovec uri_host;
    uri.get_host(uri_host);
    if (uri.get_port() == 80) {
      set_field("Host", 4, uri_host);
    } else {
      const char* uri_port_p
      = static_cast<char*>(uri_host.iov_base) + uri_host.iov_len;
      if (
        uri_port_p >= static_cast<char*>(uri_path.iov_base) - 6
        &&
        uri_port_p < uri_path.iov_base
        &&
        *uri_port_p == ':'
      ) {
        const char* uri_port_ps = uri_port_p;
        uri_port_p++;
        while (uri_port_p < uri_path.iov_base && isdigit(*uri_port_p)) {
          uri_port_p++;
        }
        uri_host.iov_len += uri_port_p - uri_port_ps;

        set_field("Host", 4, uri_host);
      } else {
        std::ostringstream host;
        host.write(
          static_cast<char*>(uri_host.iov_base),
          uri_host.iov_len
        );
        host << ':';
        host << uri.get_port();

        set_field("Host", 4, host.str());
      }
    }
  }
}