CurlClientMain.cpp File Reference

#include <folly/portability/GFlags.h>
#include <folly/io/async/EventBase.h>
#include <folly/io/async/SSLContext.h>
#include <folly/SocketAddress.h>
#include <proxygen/lib/http/HTTPConnector.h>
#include "CurlClient.h"

Go to the source code of this file.

Functions
	DEFINE_string (http_method,"GET","HTTP method to use. GET or POST are supported")
	DEFINE_string (url,"https://github.com/facebook/proxygen","URL to perform the HTTP method against")
	DEFINE_string (input_filename,"","Filename to read from for POST requests")
	DEFINE_int32 (http_client_connect_timeout, 1000,"connect timeout in milliseconds")
	DEFINE_string (ca_path,"/etc/ssl/certs/ca-certificates.crt","Path to trusted CA file")
	DEFINE_string (cert_path,"","Path to client certificate file")
	DEFINE_string (key_path,"","Path to client private key file")
	DEFINE_string (next_protos,"h2,h2-14,spdy/3.1,spdy/3,http/1.1","Next protocol string for NPN/ALPN")
	DEFINE_string (plaintext_proto,"","plaintext protocol")
	DEFINE_int32 (recv_window, 65536,"Flow control receive window for h2/spdy")
	DEFINE_bool (h2c, true,"Attempt HTTP/1.1 -> HTTP/2 upgrade")
	DEFINE_string (headers,"","List of N=V headers separated by ,")
	DEFINE_string (proxy,"","HTTP proxy URL")
int	main (int argc, char *argv[])

Function Documentation

DEFINE_bool	(	h2c	,
		true	,
		"Attempt HTTP/1.1 -> HTTP/2 upgrade"
	)

DEFINE_int32	(	http_client_connect_timeout	,
		1000	,
		"connect timeout in milliseconds"
	)

DEFINE_int32	(	recv_window	,
		65536	,
		"Flow control receive window for h2/spdy"
	)

DEFINE_string	(	http_method	,
		"GET"	,
		"HTTP method to use. GET or POST are supported"
	)

DEFINE_string	(	url	,
		"https://github.com/facebook/proxygen"	,
		"URL to perform the HTTP method against"
	)

DEFINE_string	(	input_filename	,
		""	,
		"Filename to read from for POST requests"
	)

DEFINE_string	(	ca_path	,
		"/etc/ssl/certs/ca-certificates.crt"	,
		"Path to trusted CA file"
	)

DEFINE_string	(	cert_path	,
		""	,
		"Path to client certificate file"
	)

DEFINE_string	(	key_path	,
		""	,
		"Path to client private key file"
	)

DEFINE_string	(	next_protos	,
		"	h2,
		h2-	14,
		spdy/3.	1,
		spdy/	3,
		http/1.1"	,
		"Next protocol string for NPN/ALPN"
	)

DEFINE_string	(	plaintext_proto	,
		""	,
		"plaintext protocol"
	)

DEFINE_string	(	headers	,
		""	,
		"List of	N = V headers separated by,
		"
	)

DEFINE_string	(	proxy	,
		""	,
		"HTTP proxy URL"
	)

int main	(	int	argc,
		char *	argv[]
	)

Definition at line 50 of file CurlClientMain.cpp.

References addr, folly::AsyncSocket::anyAddress(), proxygen::ERROR, f, proxygen::URL::getHost(), proxygen::URL::getPort(), folly::INFO, proxygen::URL::isSecure(), folly::EventBase::loop(), CurlService::CurlClient::parseHeaders(), CurlService::CurlClient::setFlowControlSettings(), proxygen::HTTPConnector::setPlaintextProtocol(), and proxygen::stringToMethod().

                                 {
  gflags::ParseCommandLineFlags(&argc, &argv, true);
  google::InitGoogleLogging(argv[0]);
  google::InstallFailureSignalHandler();

  EventBase evb;
  URL url(FLAGS_url);
  URL proxy(FLAGS_proxy);

  if (FLAGS_http_method != "GET" && FLAGS_http_method != "POST") {
    LOG(ERROR) << "http_method must be either GET or POST";
    return EXIT_FAILURE;
  }

  HTTPMethod httpMethod = *stringToMethod(FLAGS_http_method);
  if (httpMethod == HTTPMethod::POST) {
    try {
      File f(FLAGS_input_filename);
      (void)f;
    } catch (const std::system_error& se) {
      LOG(ERROR) << "Couldn't open file for POST method";
      LOG(ERROR) << se.what();
      return EXIT_FAILURE;
    }
  }

  HTTPHeaders headers = CurlClient::parseHeaders(FLAGS_headers);

  CurlClient curlClient(&evb,
                        httpMethod,
                        url,
                        FLAGS_proxy.empty() ? nullptr : &proxy,
                        headers,
                        FLAGS_input_filename,
                        FLAGS_h2c);
  curlClient.setFlowControlSettings(FLAGS_recv_window);

  SocketAddress addr;
  if (!FLAGS_proxy.empty()) {
    addr = SocketAddress(proxy.getHost(), proxy.getPort(), true);
  } else {
    addr = SocketAddress(url.getHost(), url.getPort(), true);
  }
  LOG(INFO) << "Trying to connect to " << addr;

  // Note: HHWheelTimer is a large object and should be created at most
  // once per thread
  HHWheelTimer::UniquePtr timer{HHWheelTimer::newTimer(
      &evb,
      std::chrono::milliseconds(HHWheelTimer::DEFAULT_TICK_INTERVAL),
      AsyncTimeout::InternalEnum::NORMAL,
      std::chrono::milliseconds(5000))};
  HTTPConnector connector(&curlClient, timer.get());
  if (!FLAGS_plaintext_proto.empty()) {
    connector.setPlaintextProtocol(FLAGS_plaintext_proto);
  }
  static const AsyncSocket::OptionMap opts{{{SOL_SOCKET, SO_REUSEADDR}, 1}};

  if (url.isSecure()) {
    curlClient.initializeSsl(
        FLAGS_ca_path, FLAGS_next_protos, FLAGS_cert_path, FLAGS_key_path);
    connector.connectSSL(
        &evb,
        addr,
        curlClient.getSSLContext(),
        nullptr,
        std::chrono::milliseconds(FLAGS_http_client_connect_timeout),
        opts,
        folly::AsyncSocket::anyAddress(),
        curlClient.getServerName());
  } else {
    connector.connect(&evb, addr,
        std::chrono::milliseconds(FLAGS_http_client_connect_timeout), opts);
  }

  evb.loop();

  return EXIT_SUCCESS;
}