mirror of
https://github.com/moparisthebest/spdylay
synced 2024-12-22 15:48:53 -05:00
358 lines
9.0 KiB
C++
358 lines
9.0 KiB
C++
|
/*
|
||
|
* 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;
|
||
|
}
|