Added example.

This example program can connect to remote host given in command-line via SSL and retrieves given path using SPDY protocol. To build this program, OpenSSL + NPN support is required.
2012-01-27 01:32:05 +09:00 · 2012-01-27 01:32:05 +09:00 · 8eea63aa30
parent 21e165f1f8
commit 8eea63aa30
1 changed files with 357 additions and 0 deletions
--- a/examples/spdycl.cc
+++ b/examples/spdycl.cc
@ -0,0 +1,357 @@
+/*
+ * Spdylay - SPDY Library
+ *
+ * Copyright (c) 2012 Tatsuhiro Tsujikawa
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the
+ * "Software"), to deal in the Software without restriction, including
+ * without limitation the rights to use, copy, modify, merge, publish,
+ * distribute, sublicense, and/or sell copies of the Software, and to
+ * permit persons to whom the Software is furnished to do so, subject to
+ * the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be
+ * included in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+ * EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+ * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+ * NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+ * LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+ * OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+ * WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ */
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netdb.h>
+#include <unistd.h>
+#include <fcntl.h>
+#include <sys/epoll.h>
+#include <netinet/in.h>
+#include <netinet/tcp.h>
+#include <signal.h>
+
+#include <cassert>
+#include <cstdio>
+#include <cerrno>
+#include <cstdlib>
+#include <cstring>
+#include <string>
+#include <iostream>
+#include <string>
+#include <set>
+#include <iomanip>
+#include <fstream>
+
+#include <openssl/ssl.h>
+#include <openssl/err.h>
+#include <spdylay/spdylay.h>
+
+int connect_to(const char *host, const char *service)
+{
+  struct addrinfo hints;
+  int fd = -1;
+  int r;
+  memset(&hints, 0, sizeof(struct addrinfo));
+  hints.ai_family = AF_UNSPEC;
+  hints.ai_socktype = SOCK_STREAM;
+  struct addrinfo *res;
+  r = getaddrinfo(host, service, &hints, &res);
+  if(r != 0) {
+    std::cerr << "getaddrinfo: " << gai_strerror(r) << std::endl;
+    return -1;
+  }
+  for(struct addrinfo *rp = res; rp; rp = rp->ai_next) {
+    fd = socket(rp->ai_family, rp->ai_socktype, rp->ai_protocol);
+    if(fd == -1) {
+      continue;
+    }
+    while((r = connect(fd, rp->ai_addr, rp->ai_addrlen)) == -1 &&
+          errno == EINTR);
+    if(r == 0) {
+      break;
+    }
+    close(fd);
+    fd = -1;
+  }
+  freeaddrinfo(res);
+  return fd;
+}
+
+int make_non_block(int fd)
+{
+  int flags, r;
+  while((flags = fcntl(fd, F_GETFL, 0)) == -1 && errno == EINTR);
+  if(flags == -1) {
+    return -1;
+  }
+  while((r = fcntl(fd, F_SETFL, flags | O_NONBLOCK)) == -1 && errno == EINTR);
+  if(r == -1) {
+    return -1;
+  }
+  return 0;
+}
+
+class SpdylayClient {
+public:
+  SpdylayClient(int fd, SSL *ssl,
+                const spdylay_session_callbacks *callbacks)
+    : fd_(fd), ssl_(ssl), want_write_(false)
+  {
+    spdylay_session_client_new(&session_, callbacks, this);
+  }
+  ~SpdylayClient()
+  {
+    spdylay_session_del(session_);
+    SSL_shutdown(ssl_);
+    shutdown(fd_, SHUT_WR);
+    close(fd_);
+    SSL_free(ssl_);
+  }
+  int on_read_event()
+  {
+    return spdylay_session_recv(session_);
+  }
+  int on_write_event()
+  {
+    return spdylay_session_send(session_);
+  }
+  ssize_t send_data(const uint8_t *data, size_t len, int flags)
+  {
+    ssize_t r;
+    r = SSL_write(ssl_, data, len);
+    return r;
+  }
+  ssize_t recv_data(uint8_t *data, size_t len, int flags)
+  {
+    ssize_t r;
+    want_write_ = false;
+    r = SSL_read(ssl_, data, len);
+    if(r < 0) {
+      if(SSL_get_error(ssl_, r) == SSL_ERROR_WANT_WRITE) {
+        want_write_ = true;
+      }
+    }
+    return r;
+  }
+  bool want_read()
+  {
+    return spdylay_session_want_read(session_);
+  }
+  bool want_write()
+  {
+    return spdylay_session_want_write(session_) || want_write_;
+  }
+  int fd() const
+  {
+    return fd_;
+  }
+  int req_submit(const char *path)
+  {
+    return spdylay_req_submit(session_, path);
+  }
+  bool would_block(int r)
+  {
+    int e = SSL_get_error(ssl_, r);
+    return e == SSL_ERROR_WANT_WRITE || e == SSL_ERROR_WANT_READ;
+  }
+private:
+  int fd_;
+  SSL *ssl_;
+  spdylay_session *session_;
+  bool want_write_;
+};
+
+ssize_t send_callback(spdylay_session *session,
+                      const uint8_t *data, size_t len, int flags,
+                      void *user_data)
+{
+  SpdylayClient *sc = (SpdylayClient*)user_data;
+  ssize_t r = sc->send_data(data, len, flags);
+  if(r < 0) {
+    if(sc->would_block(r)) {
+      r = SPDYLAY_ERR_WOULDBLOCK;
+    } else {
+      r = SPDYLAY_ERR_CALLBACK_FAILURE;
+    }
+  }
+  return r;
+}
+
+ssize_t recv_callback(spdylay_session *session,
+                      uint8_t *data, size_t len, int flags, void *user_data)
+{
+  SpdylayClient *sc = (SpdylayClient*)user_data;
+  ssize_t r = sc->recv_data(data, len, flags);
+  if(r < 0) {
+    if(sc->would_block(r)) {
+      r = SPDYLAY_ERR_WOULDBLOCK;
+    } else {
+      r = SPDYLAY_ERR_CALLBACK_FAILURE;
+    }
+  } else if(r == 0) {
+    r = SPDYLAY_ERR_CALLBACK_FAILURE;
+  }
+  return r;
+}
+
+void ctl_epollev(int epollfd, int op, SpdylayClient& sc)
+{
+  epoll_event ev;
+  memset(&ev, 0, sizeof(ev));
+  if(sc.want_read()) {
+    ev.events |= EPOLLIN;
+  }
+  if(sc.want_write()) {
+    ev.events |= EPOLLOUT;
+  }
+  if(epoll_ctl(epollfd, op, sc.fd(), &ev) == -1) {
+    perror("epoll_ctl");
+    exit(EXIT_FAILURE);
+  }
+}
+
+int select_next_proto_cb(SSL* ssl,
+                         unsigned char **out, unsigned char *outlen,
+                         const unsigned char *in, unsigned int inlen,
+                         void *arg)
+{
+  *out = (unsigned char*)in+1;
+  *outlen = in[0];
+  std::cout << "NPN select next proto: server offers:" << std::endl;
+  for(unsigned int i = 0; i < inlen; i += in[i]+1) {
+    std::cout << "* " << std::string(&in[i+1], &in[i+1]+in[i]) << std::endl;
+    if(in[i] == 6 && memcmp(&in[i+1], "spdy/2", in[i]) == 0) {
+      *out = (unsigned char*)in+i+1;
+      *outlen = in[i];
+    }
+  }
+  return SSL_TLSEXT_ERR_OK;
+
+
+  int status = SSL_select_next_proto(out, outlen, in, inlen,
+                                     (const unsigned char*)"spdy/2", 6);
+  switch(status) {
+  case OPENSSL_NPN_UNSUPPORTED:
+    std::cerr << "npn unsupported" << std::endl;
+    break;
+  case OPENSSL_NPN_NEGOTIATED:
+    std::cout << "npn negotiated" << std::endl;
+    break;
+  case OPENSSL_NPN_NO_OVERLAP:
+    std::cout << "npn no overlap" << std::endl;
+    break;
+  default:
+    std::cout << "not reached?" << std::endl;
+  }
+  return SSL_TLSEXT_ERR_OK;
+}
+
+int communicate(const char *host, const char *service, const char *path,
+                const spdylay_session_callbacks *callbacks)
+{
+  int r;
+  int fd = connect_to(host, service);
+  if(fd == -1) {
+    std::cerr << "Could not connect to the host" << std::endl;
+    return -1;
+  }
+  SSL_CTX *ssl_ctx;
+  ssl_ctx = SSL_CTX_new(SSLv23_client_method());
+  if(!ssl_ctx) {
+    std::cerr << ERR_error_string(ERR_get_error(), 0) << std::endl;
+    return -1;
+  }
+  /* Disable SSLv2 and enable all workarounds for buggy servers */
+  SSL_CTX_set_options(ssl_ctx, SSL_OP_ALL|SSL_OP_NO_SSLv2);
+  SSL_CTX_set_mode(ssl_ctx, SSL_MODE_AUTO_RETRY);
+  SSL_CTX_set_mode(ssl_ctx, SSL_MODE_RELEASE_BUFFERS);
+  SSL_CTX_set_next_proto_select_cb(ssl_ctx, select_next_proto_cb, 0);
+
+  SSL *ssl = SSL_new(ssl_ctx);
+  if(!ssl) {
+    std::cerr << ERR_error_string(ERR_get_error(), 0) << std::endl;
+    return -1;
+  }
+
+  if(SSL_set_fd(ssl, fd) == 0) {
+    std::cerr << ERR_error_string(ERR_get_error(), 0) << std::endl;
+    return -1;
+  }
+  r = SSL_connect(ssl);
+  if(r <= 0) {
+    std::cerr << ERR_error_string(ERR_get_error(), 0) << std::endl;
+    return -1;
+  }
+
+  make_non_block(fd);
+  SpdylayClient sc(fd, ssl, callbacks);
+  int epollfd = epoll_create(1);
+  if(epollfd == -1) {
+    perror("epoll_create");
+    return -1;
+  }
+
+  sc.req_submit(path);
+
+  ctl_epollev(epollfd, EPOLL_CTL_ADD, sc);
+  static const size_t MAX_EVENTS = 1;
+  epoll_event events[MAX_EVENTS];
+  bool ok = true;
+  while(1) {
+    int nfds = epoll_wait(epollfd, events, MAX_EVENTS, -1);
+    if(nfds == -1) {
+      perror("epoll_wait");
+      return -1;
+    }
+    for(int n = 0; n < nfds; ++n) {
+      if(((events[n].events & EPOLLIN) && sc.on_read_event() != 0) ||
+         ((events[n].events & EPOLLOUT) && sc.on_write_event() != 0)) {
+        ok = false;
+        std::cout << "Fatal" << std::endl;
+        break;
+      }
+      if((events[n].events & EPOLLHUP) || (events[n].events & EPOLLERR)) {
+        std::cout << "HUP" << std::endl;
+        ok = false;
+        break;
+      }
+    }
+    if(!ok) {
+      break;
+    }
+    ctl_epollev(epollfd, EPOLL_CTL_MOD, sc);
+  }
+
+
+  return ok ? 0 : -1;
+}
+
+int run(const char *host, const char *service, const char *path)
+{
+  spdylay_session_callbacks callbacks = {
+    send_callback,
+    recv_callback,
+  };
+  return communicate(host, service, path, &callbacks);
+}
+
+int main(int argc, char **argv)
+{
+  if(argc < 2) {
+    std::cerr << "Usage: " << argv[0] << " HOST PORT PATH" << std::endl;
+    exit(EXIT_FAILURE);
+  }
+  struct sigaction act;
+  memset(&act, 0, sizeof(struct sigaction));
+  act.sa_handler = SIG_IGN;
+  sigaction(SIGPIPE, &act, 0);
+  const char *host = argv[1];
+  const char *service = argv[2];
+  const char *path = argv[3];
+  SSL_load_error_strings();
+  SSL_library_init();
+  run(host, service, path);
+  return 0;
+}