// Copyright Joyent, Inc. and other Node contributors.

//

// 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 <node_crypto.h>

#include <v8.h>

#include <node.h>

#include <node_buffer.h>

#include <node_vars.h>

#include <node_root_certs.h>

#include <string.h>

#ifdef _MSC_VER

#define snprintf _snprintf

#define strcasecmp _stricmp

#endif

#include <stdlib.h>

#include <errno.h>

/* Sigh. */

#ifdef _WIN32

# include <windows.h>

#else

# include <pthread.h>

#endif

#if OPENSSL_VERSION_NUMBER >= 0x10000000L

# define OPENSSL_CONST const

#else

# define OPENSSL_CONST

#endif

#define ASSERT_IS_STRING_OR_BUFFER(val) \

if (!val->IsString() && !Buffer::HasInstance(val)) { \

return ThrowException(Exception::TypeError(String::New("Not a string or buffer"))); \

}

static const char PUBLIC_KEY_PFX[] = "-----BEGIN PUBLIC KEY-----";

static const int PUBLIC_KEY_PFX_LEN = sizeof(PUBLIC_KEY_PFX) - 1;

static const char PUBRSA_KEY_PFX[] = "-----BEGIN RSA PUBLIC KEY-----";

static const int PUBRSA_KEY_PFX_LEN = sizeof(PUBRSA_KEY_PFX) - 1;

static const int X509_NAME_FLAGS = ASN1_STRFLGS_ESC_CTRL

| ASN1_STRFLGS_ESC_MSB

| XN_FLAG_SEP_MULTILINE

| XN_FLAG_FN_SN;

#include <node_vars.h>

// We do the following to minimize the detal between v0.6 branch. We want to

// use the variables as they were being used before.

#define on_headers_sym NODE_VAR(on_headers_sym)

#define errno_symbol NODE_VAR(errno_symbol)

#define syscall_symbol NODE_VAR(syscall_symbol)

#define subject_symbol NODE_VAR(subject_symbol)

#define subjectaltname_symbol NODE_VAR(subjectaltname_symbol)

#define modulus_symbol NODE_VAR(modulus_symbol)

#define exponent_symbol NODE_VAR(exponent_symbol)

#define issuer_symbol NODE_VAR(issuer_symbol)

#define valid_from_symbol NODE_VAR(valid_from_symbol)

#define valid_to_symbol NODE_VAR(valid_to_symbol)

#define fingerprint_symbol NODE_VAR(fingerprint_symbol)

#define name_symbol NODE_VAR(name_symbol)

#define version_symbol NODE_VAR(version_symbol)

#define ext_key_usage_symbol NODE_VAR(ext_key_usage_symbol)

#define secure_context_constructor NODE_VAR(secure_context_constructor)

namespace node {

namespace crypto {

static uv_rwlock_t* locks;

using namespace v8;

static unsigned long crypto_id_cb(void) {

#ifdef _WIN32

return (unsigned long) GetCurrentThreadId();

#else /* !_WIN32 */

return (unsigned long) pthread_self();

#endif /* !_WIN32 */

}

static void crypto_lock_init(void) {

int i, n;

n = CRYPTO_num_locks();

locks = new uv_rwlock_t[n];

for (i = 0; i < n; i++)

if (uv_rwlock_init(locks + i))

abort();

}

static void crypto_lock_cb(int mode, int n, const char* file, int line) {

assert((mode & CRYPTO_LOCK) || (mode & CRYPTO_UNLOCK));

assert((mode & CRYPTO_READ) || (mode & CRYPTO_WRITE));

if (mode & CRYPTO_LOCK) {

if (mode & CRYPTO_READ)

uv_rwlock_rdlock(locks + n);

else

uv_rwlock_wrlock(locks + n);

} else {

if (mode & CRYPTO_READ)

uv_rwlock_rdunlock(locks + n);

else

uv_rwlock_wrunlock(locks + n);

}

}

void SecureContext::Initialize(Handle<Object> target) {

HandleScope scope;

Local<FunctionTemplate> t = FunctionTemplate::New(SecureContext::New);

secure_context_constructor = Persistent<FunctionTemplate>::New(t);

t->InstanceTemplate()->SetInternalFieldCount(1);

t->SetClassName(String::NewSymbol("SecureContext"));

NODE_SET_PROTOTYPE_METHOD(t, "init", SecureContext::Init);

NODE_SET_PROTOTYPE_METHOD(t, "setKey", SecureContext::SetKey);

NODE_SET_PROTOTYPE_METHOD(t, "setCert", SecureContext::SetCert);

NODE_SET_PROTOTYPE_METHOD(t, "addCACert", SecureContext::AddCACert);

NODE_SET_PROTOTYPE_METHOD(t, "addCRL", SecureContext::AddCRL);

NODE_SET_PROTOTYPE_METHOD(t, "addRootCerts", SecureContext::AddRootCerts);

NODE_SET_PROTOTYPE_METHOD(t, "setCiphers", SecureContext::SetCiphers);

NODE_SET_PROTOTYPE_METHOD(t, "setOptions", SecureContext::SetOptions);

NODE_SET_PROTOTYPE_METHOD(t, "setSessionIdContext",

SecureContext::SetSessionIdContext);

NODE_SET_PROTOTYPE_METHOD(t, "close", SecureContext::Close);

target->Set(String::NewSymbol("SecureContext"), t->GetFunction());

}

Handle<Value> SecureContext::New(const Arguments& args) {

HandleScope scope;

SecureContext *p = new SecureContext();

p->Wrap(args.Holder());

return args.This();

}

Handle<Value> SecureContext::Init(const Arguments& args) {

HandleScope scope;

SecureContext *sc = ObjectWrap::Unwrap<SecureContext>(args.Holder());

OPENSSL_CONST SSL_METHOD *method = SSLv23_method();

if (args.Length() == 1 && args[0]->IsString()) {

String::Utf8Value sslmethod(args[0]->ToString());

if (strcmp(*sslmethod, "SSLv2_method") == 0) {

#ifndef OPENSSL_NO_SSL2

method = SSLv2_method();

#else

return ThrowException(Exception::Error(String::New("SSLv2 methods disabled")));

#endif

} else if (strcmp(*sslmethod, "SSLv2_server_method") == 0) {

#ifndef OPENSSL_NO_SSL2

method = SSLv2_server_method();

#else

return ThrowException(Exception::Error(String::New("SSLv2 methods disabled")));

#endif

} else if (strcmp(*sslmethod, "SSLv2_client_method") == 0) {

#ifndef OPENSSL_NO_SSL2

method = SSLv2_client_method();

#else

return ThrowException(Exception::Error(String::New("SSLv2 methods disabled")));

#endif

} else if (strcmp(*sslmethod, "SSLv3_method") == 0) {

method = SSLv3_method();

} else if (strcmp(*sslmethod, "SSLv3_server_method") == 0) {

method = SSLv3_server_method();

} else if (strcmp(*sslmethod, "SSLv3_client_method") == 0) {

method = SSLv3_client_method();

} else if (strcmp(*sslmethod, "SSLv23_method") == 0) {

method = SSLv23_method();

} else if (strcmp(*sslmethod, "SSLv23_server_method") == 0) {

method = SSLv23_server_method();

} else if (strcmp(*sslmethod, "SSLv23_client_method") == 0) {

method = SSLv23_client_method();

} else if (strcmp(*sslmethod, "TLSv1_method") == 0) {

method = TLSv1_method();

} else if (strcmp(*sslmethod, "TLSv1_server_method") == 0) {

method = TLSv1_server_method();

} else if (strcmp(*sslmethod, "TLSv1_client_method") == 0) {

method = TLSv1_client_method();

} else {

return ThrowException(Exception::Error(String::New("Unknown method")));

}

}

sc->ctx_ = SSL_CTX_new(method);

// Enable session caching?

SSL_CTX_set_session_cache_mode(sc->ctx_, SSL_SESS_CACHE_SERVER);

// SSL_CTX_set_session_cache_mode(sc->ctx_,SSL_SESS_CACHE_OFF);

sc->ca_store_ = NULL;

return True();

}

// Takes a string or buffer and loads it into a BIO.

// Caller responsible for BIO_free-ing the returned object.

static BIO* LoadBIO (Handle<Value> v) {

BIO *bio = BIO_new(BIO_s_mem());

if (!bio) return NULL;

HandleScope scope;

int r = -1;

if (v->IsString()) {

String::Utf8Value s(v->ToString());

r = BIO_write(bio, *s, s.length());

} else if (Buffer::HasInstance(v)) {

Local<Object> buffer_obj = v->ToObject();

char *buffer_data = Buffer::Data(buffer_obj);

size_t buffer_length = Buffer::Length(buffer_obj);

r = BIO_write(bio, buffer_data, buffer_length);

}

if (r <= 0) {

BIO_free(bio);

return NULL;

}

return bio;

}

// Takes a string or buffer and loads it into an X509

// Caller responsible for X509_free-ing the returned object.

static X509* LoadX509 (Handle<Value> v) {

HandleScope scope; // necessary?

BIO *bio = LoadBIO(v);

if (!bio) return NULL;

X509 * x509 = PEM_read_bio_X509(bio, NULL, NULL, NULL);

if (!x509) {

BIO_free(bio);

return NULL;

}

BIO_free(bio);

return x509;

}

Handle<Value> SecureContext::SetKey(const Arguments& args) {

HandleScope scope;

SecureContext *sc = ObjectWrap::Unwrap<SecureContext>(args.Holder());

unsigned int len = args.Length();

if (len != 1 && len != 2) {

return ThrowException(Exception::TypeError(String::New("Bad parameter")));

}

if (len == 2 && !args[1]->IsString()) {

return ThrowException(Exception::TypeError(String::New("Bad parameter")));

}

BIO *bio = LoadBIO(args[0]);

if (!bio) return False();

String::Utf8Value passphrase(args[1]->ToString());

EVP_PKEY* key = PEM_read_bio_PrivateKey(bio, NULL, NULL,

len == 1 ? NULL : *passphrase);

if (!key) {

BIO_free(bio);

return False();

}

SSL_CTX_use_PrivateKey(sc->ctx_, key);

EVP_PKEY_free(key);

BIO_free(bio);

return True();

}

// Read a file that contains our certificate in "PEM" format,

// possibly followed by a sequence of CA certificates that should be

// sent to the peer in the Certificate message.

//

// Taken from OpenSSL - editted for style.

int SSL_CTX_use_certificate_chain(SSL_CTX *ctx, BIO *in) {

int ret = 0;

X509 *x = NULL;

x = PEM_read_bio_X509_AUX(in, NULL, NULL, NULL);

if (x == NULL) {

SSLerr(SSL_F_SSL_CTX_USE_CERTIFICATE_CHAIN_FILE, ERR_R_PEM_LIB);

goto end;

}

ret = SSL_CTX_use_certificate(ctx, x);

if (ERR_peek_error() != 0) {

// Key/certificate mismatch doesn't imply ret==0 ...

ret = 0;

}

if (ret) {

// If we could set up our certificate, now proceed to

// the CA certificates.

X509 *ca;

int r;

unsigned long err;

if (ctx->extra_certs != NULL) {

sk_X509_pop_free(ctx->extra_certs, X509_free);

ctx->extra_certs = NULL;

}

while ((ca = PEM_read_bio_X509(in, NULL, NULL, NULL))) {

r = SSL_CTX_add_extra_chain_cert(ctx, ca);

if (!r) {

X509_free(ca);

ret = 0;

goto end;

}

// Note that we must not free r if it was successfully

// added to the chain (while we must free the main

// certificate, since its reference count is increased

// by SSL_CTX_use_certificate).

}

// When the while loop ends, it's usually just EOF.

err = ERR_peek_last_error();

if (ERR_GET_LIB(err) == ERR_LIB_PEM &&

ERR_GET_REASON(err) == PEM_R_NO_START_LINE) {

ERR_clear_error();

} else {

// some real error

ret = 0;

}

}

end:

if (x != NULL) X509_free(x);

return ret;

}

Handle<Value> SecureContext::SetCert(const Arguments& args) {

HandleScope scope;

SecureContext *sc = ObjectWrap::Unwrap<SecureContext>(args.Holder());

if (args.Length() != 1) {

return ThrowException(Exception::TypeError(

String::New("Bad parameter")));

}

BIO* bio = LoadBIO(args[0]);

if (!bio) return False();

int rv = SSL_CTX_use_certificate_chain(sc->ctx_, bio);

BIO_free(bio);

if (!rv) {

unsigned long err = ERR_get_error();

if (!err) {

return ThrowException(Exception::Error(

String::New("SSL_CTX_use_certificate_chain")));

}

char string[120];

ERR_error_string_n(err, string, sizeof string);

return ThrowException(Exception::Error(String::New(string)));

}

return True();

}

Handle<Value> SecureContext::AddCACert(const Arguments& args) {

bool newCAStore = false;

HandleScope scope;

SecureContext *sc = ObjectWrap::Unwrap<SecureContext>(args.Holder());

if (args.Length() != 1) {

return ThrowException(Exception::TypeError(String::New("Bad parameter")));

}

if (!sc->ca_store_) {

sc->ca_store_ = X509_STORE_new();

newCAStore = true;

}

X509* x509 = LoadX509(args[0]);

if (!x509) return False();

X509_STORE_add_cert(sc->ca_store_, x509);

SSL_CTX_add_client_CA(sc->ctx_, x509);

X509_free(x509);

if (newCAStore) {

SSL_CTX_set_cert_store(sc->ctx_, sc->ca_store_);

}

return True();

}

Handle<Value> SecureContext::AddCRL(const Arguments& args) {

HandleScope scope;

SecureContext *sc = ObjectWrap::Unwrap<SecureContext>(args.Holder());

if (args.Length() != 1) {

return ThrowException(Exception::TypeError(String::New("Bad parameter")));

}

BIO *bio = LoadBIO(args[0]);

if (!bio) return False();

X509_CRL *x509 = PEM_read_bio_X509_CRL(bio, NULL, NULL, NULL);

if (x509 == NULL) {

BIO_free(bio);

return False();

}

X509_STORE_add_crl(sc->ca_store_, x509);

X509_STORE_set_flags(sc->ca_store_, X509_V_FLAG_CRL_CHECK |

X509_V_FLAG_CRL_CHECK_ALL);

BIO_free(bio);

X509_CRL_free(x509);

return True();

}

Handle<Value> SecureContext::AddRootCerts(const Arguments& args) {

HandleScope scope;

SecureContext *sc = ObjectWrap::Unwrap<SecureContext>(args.Holder());

assert(sc->ca_store_ == NULL);

if (!root_cert_store) {

root_cert_store = X509_STORE_new();

for (int i = 0; root_certs[i]; i++) {

BIO *bp = BIO_new(BIO_s_mem());

if (!BIO_write(bp, root_certs[i], strlen(root_certs[i]))) {

BIO_free(bp);

return False();

}

X509 *x509 = PEM_read_bio_X509(bp, NULL, NULL, NULL);

if (x509 == NULL) {

BIO_free(bp);

return False();

}

X509_STORE_add_cert(root_cert_store, x509);

BIO_free(bp);

X509_free(x509);

}

}

sc->ca_store_ = root_cert_store;

SSL_CTX_set_cert_store(sc->ctx_, sc->ca_store_);

return True();

}

Handle<Value> SecureContext::SetCiphers(const Arguments& args) {

HandleScope scope;

SecureContext *sc = ObjectWrap::Unwrap<SecureContext>(args.Holder());

if (args.Length() != 1 || !args[0]->IsString()) {

return ThrowException(Exception::TypeError(String::New("Bad parameter")));

}

String::Utf8Value ciphers(args[0]->ToString());

SSL_CTX_set_cipher_list(sc->ctx_, *ciphers);

return True();

}

Handle<Value> SecureContext::SetOptions(const Arguments& args) {

HandleScope scope;

SecureContext *sc = ObjectWrap::Unwrap<SecureContext>(args.Holder());

if (args.Length() != 1 || !args[0]->IsUint32()) {

return ThrowException(Exception::TypeError(String::New("Bad parameter")));

}

unsigned int opts = args[0]->Uint32Value();

SSL_CTX_set_options(sc->ctx_, opts);

return True();

}

Handle<Value> SecureContext::SetSessionIdContext(const Arguments& args) {

HandleScope scope;

SecureContext *sc = ObjectWrap::Unwrap<SecureContext>(args.Holder());

if (args.Length() != 1 || !args[0]->IsString()) {

return ThrowException(Exception::TypeError(String::New("Bad parameter")));

}

String::Utf8Value sessionIdContext(args[0]->ToString());

const unsigned char* sid_ctx = (const unsigned char*) *sessionIdContext;

unsigned int sid_ctx_len = sessionIdContext.length();

int r = SSL_CTX_set_session_id_context(sc->ctx_, sid_ctx, sid_ctx_len);

if (r != 1) {

Local<String> message;

BIO* bio;

BUF_MEM* mem;

if ((bio = BIO_new(BIO_s_mem()))) {

ERR_print_errors(bio);

BIO_get_mem_ptr(bio, &mem);

message = String::New(mem->data, mem->length);

BIO_free(bio);

} else {

message = String::New("SSL_CTX_set_session_id_context error");

}

return ThrowException(Exception::TypeError(message));

}

return True();

}

Handle<Value> SecureContext::Close(const Arguments& args) {

HandleScope scope;

SecureContext *sc = ObjectWrap::Unwrap<SecureContext>(args.Holder());

sc->FreeCTXMem();

return False();

}

#ifdef SSL_PRINT_DEBUG

# define DEBUG_PRINT(...) fprintf (stderr, __VA_ARGS__)

#else

# define DEBUG_PRINT(...)

#endif

int Connection::HandleBIOError(BIO *bio, const char* func, int rv) {

if (rv >= 0) return rv;

int retry = BIO_should_retry(bio);

if (BIO_should_write(bio)) {

DEBUG_PRINT("[%p] BIO: %s want write. should retry %d\n", ssl_, func, retry);

return 0;

} else if (BIO_should_read(bio)) {

DEBUG_PRINT("[%p] BIO: %s want read. should retry %d\n", ssl_, func, retry);

return 0;

} else {

static char ssl_error_buf[512];

ERR_error_string_n(rv, ssl_error_buf, sizeof(ssl_error_buf));

HandleScope scope;

Local<Value> e = Exception::Error(String::New(ssl_error_buf));

handle_->Set(String::New("error"), e);

DEBUG_PRINT("[%p] BIO: %s failed: (%d) %s\n", ssl_, func, rv, ssl_error_buf);

return rv;

}

return 0;

}

int Connection::HandleSSLError(const char* func, int rv) {

if (rv >= 0) return rv;

int err = SSL_get_error(ssl_, rv);

if (err == SSL_ERROR_NONE) {

return 0;

} else if (err == SSL_ERROR_WANT_WRITE) {

DEBUG_PRINT("[%p] SSL: %s want write\n", ssl_, func);

return 0;

} else if (err == SSL_ERROR_WANT_READ) {

DEBUG_PRINT("[%p] SSL: %s want read\n", ssl_, func);

return 0;

} else {

HandleScope scope;

BUF_MEM* mem;

BIO *bio;

assert(err == SSL_ERROR_SSL || err == SSL_ERROR_SYSCALL);

// XXX We need to drain the error queue for this thread or else OpenSSL

// has the possibility of blocking connections? This problem is not well

// understood. And we should be somehow propagating these errors up

// into JavaScript. There is no test which demonstrates this problem.

// https://github.com/joyent/node/issues/1719

if ((bio = BIO_new(BIO_s_mem()))) {

ERR_print_errors(bio);

BIO_get_mem_ptr(bio, &mem);

Local<Value> e = Exception::Error(String::New(mem->data, mem->length));

handle_->Set(String::New("error"), e);

BIO_free(bio);

}

return rv;

}

return 0;

}

void Connection::ClearError() {

#ifndef NDEBUG

HandleScope scope;

// We should clear the error in JS-land

assert(handle_->Get(String::New("error"))->BooleanValue() == false);

#endif // NDEBUG

}

void Connection::SetShutdownFlags() {

HandleScope scope;

int flags = SSL_get_shutdown(ssl_);

if (flags & SSL_SENT_SHUTDOWN) {

handle_->Set(String::New("sentShutdown"), True());

}

if (flags & SSL_RECEIVED_SHUTDOWN) {

handle_->Set(String::New("receivedShutdown"), True());

}

}

void Connection::Initialize(Handle<Object> target) {

HandleScope scope;

Local<FunctionTemplate> t = FunctionTemplate::New(Connection::New);

t->InstanceTemplate()->SetInternalFieldCount(1);

t->SetClassName(String::NewSymbol("Connection"));

NODE_SET_PROTOTYPE_METHOD(t, "encIn", Connection::EncIn);

NODE_SET_PROTOTYPE_METHOD(t, "clearOut", Connection::ClearOut);

NODE_SET_PROTOTYPE_METHOD(t, "clearIn", Connection::ClearIn);

NODE_SET_PROTOTYPE_METHOD(t, "encOut", Connection::EncOut);

NODE_SET_PROTOTYPE_METHOD(t, "clearPending", Connection::ClearPending);

NODE_SET_PROTOTYPE_METHOD(t, "encPending", Connection::EncPending);

NODE_SET_PROTOTYPE_METHOD(t, "getPeerCertificate", Connection::GetPeerCertificate);

NODE_SET_PROTOTYPE_METHOD(t, "getSession", Connection::GetSession);

NODE_SET_PROTOTYPE_METHOD(t, "setSession", Connection::SetSession);

NODE_SET_PROTOTYPE_METHOD(t, "isSessionReused", Connection::IsSessionReused);

NODE_SET_PROTOTYPE_METHOD(t, "isInitFinished", Connection::IsInitFinished);

NODE_SET_PROTOTYPE_METHOD(t, "verifyError", Connection::VerifyError);

NODE_SET_PROTOTYPE_METHOD(t, "getCurrentCipher", Connection::GetCurrentCipher);

NODE_SET_PROTOTYPE_METHOD(t, "start", Connection::Start);

NODE_SET_PROTOTYPE_METHOD(t, "shutdown", Connection::Shutdown);

NODE_SET_PROTOTYPE_METHOD(t, "receivedShutdown", Connection::ReceivedShutdown);

NODE_SET_PROTOTYPE_METHOD(t, "close", Connection::Close);

#ifdef OPENSSL_NPN_NEGOTIATED

NODE_SET_PROTOTYPE_METHOD(t, "getNegotiatedProtocol", Connection::GetNegotiatedProto);

NODE_SET_PROTOTYPE_METHOD(t, "setNPNProtocols", Connection::SetNPNProtocols);

#endif

#ifdef SSL_CTRL_SET_TLSEXT_SERVERNAME_CB

NODE_SET_PROTOTYPE_METHOD(t, "getServername", Connection::GetServername);

NODE_SET_PROTOTYPE_METHOD(t, "setSNICallback", Connection::SetSNICallback);

#endif

target->Set(String::NewSymbol("Connection"), t->GetFunction());

}

static int VerifyCallback(int preverify_ok, X509_STORE_CTX *ctx) {

// Quoting SSL_set_verify(3ssl):

//

// The VerifyCallback function is used to control the behaviour when

// the SSL_VERIFY_PEER flag is set. It must be supplied by the

// application and receives two arguments: preverify_ok indicates,

// whether the verification of the certificate in question was passed

// (preverify_ok=1) or not (preverify_ok=0). x509_ctx is a pointer to

// the complete context used for the certificate chain verification.

//

// The certificate chain is checked starting with the deepest nesting

// level (the root CA certificate) and worked upward to the peer's

// certificate. At each level signatures and issuer attributes are

// checked. Whenever a verification error is found, the error number is

// stored in x509_ctx and VerifyCallback is called with preverify_ok=0.

// By applying X509_CTX_store_* functions VerifyCallback can locate the

// certificate in question and perform additional steps (see EXAMPLES).

// If no error is found for a certificate, VerifyCallback is called

// with preverify_ok=1 before advancing to the next level.

//

// The return value of VerifyCallback controls the strategy of the

// further verification process. If VerifyCallback returns 0, the

// verification process is immediately stopped with "verification

// failed" state. If SSL_VERIFY_PEER is set, a verification failure

// alert is sent to the peer and the TLS/SSL handshake is terminated. If

// VerifyCallback returns 1, the verification process is continued. If

// VerifyCallback always returns 1, the TLS/SSL handshake will not be

// terminated with respect to verification failures and the connection

// will be established. The calling process can however retrieve the

// error code of the last verification error using

// SSL_get_verify_result(3) or by maintaining its own error storage

// managed by VerifyCallback.

//

// If no VerifyCallback is specified, the default callback will be

// used. Its return value is identical to preverify_ok, so that any

// verification failure will lead to a termination of the TLS/SSL

// handshake with an alert message, if SSL_VERIFY_PEER is set.

//

// Since we cannot perform I/O quickly enough in this callback, we ignore

// all preverify_ok errors and let the handshake continue. It is

// imparative that the user use Connection::VerifyError after the

// 'secure' callback has been made.

return 1;

}

#ifdef OPENSSL_NPN_NEGOTIATED

int Connection::AdvertiseNextProtoCallback_(SSL *s,

const unsigned char **data,

unsigned int *len,

void *arg) {

Connection *p = static_cast<Connection*>(SSL_get_app_data(s));

if (p->npnProtos_.IsEmpty()) {

// No initialization - no NPN protocols

*data = reinterpret_cast<const unsigned char*>("");

*len = 0;

} else {

*data = reinterpret_cast<const unsigned char*>(Buffer::Data(p->npnProtos_));

*len = Buffer::Length(p->npnProtos_);

}

return SSL_TLSEXT_ERR_OK;

}

int Connection::SelectNextProtoCallback_(SSL *s,

unsigned char **out, unsigned char *outlen,

const unsigned char* in,

unsigned int inlen, void *arg) {

Connection *p = static_cast<Connection*> SSL_get_app_data(s);

// Release old protocol handler if present

if (!p->selectedNPNProto_.IsEmpty()) {

p->selectedNPNProto_.Dispose();

}

if (p->npnProtos_.IsEmpty()) {

// We should at least select one protocol

// If server is using NPN

*out = reinterpret_cast<unsigned char*>(const_cast<char*>("http/1.1"));

*outlen = 8;

// set status unsupported

p->selectedNPNProto_ = Persistent<Value>::New(False());

return SSL_TLSEXT_ERR_OK;

}

const unsigned char* npnProtos =

reinterpret_cast<const unsigned char*>(Buffer::Data(p->npnProtos_));

int status = SSL_select_next_proto(out, outlen, in, inlen, npnProtos,

Buffer::Length(p->npnProtos_));

switch (status) {

case OPENSSL_NPN_UNSUPPORTED:

p->selectedNPNProto_ = Persistent<Value>::New(Null());

break;

case OPENSSL_NPN_NEGOTIATED:

p->selectedNPNProto_ = Persistent<Value>::New(String::New(

reinterpret_cast<const char*>(*out), *outlen

));

break;

case OPENSSL_NPN_NO_OVERLAP:

p->selectedNPNProto_ = Persistent<Value>::New(False());

break;

default:

break;

}

return SSL_TLSEXT_ERR_OK;

}

#endif

#ifdef SSL_CTRL_SET_TLSEXT_SERVERNAME_CB

int Connection::SelectSNIContextCallback_(SSL *s, int *ad, void* arg) {

HandleScope scope;

Connection *p = static_cast<Connection*> SSL_get_app_data(s);

const char* servername = SSL_get_servername(s, TLSEXT_NAMETYPE_host_name);

if (servername) {

if (!p->servername_.IsEmpty()) {

p->servername_.Dispose();

}

p->servername_ = Persistent<String>::New(String::New(servername));

// Call sniCallback_ and use it's return value as context

if (!p->sniCallback_.IsEmpty()) {

if (!p->sniContext_.IsEmpty()) {

p->sniContext_.Dispose();

}

// Get callback init args

Local<Value> argv[1] = {*p->servername_};

Local<Function> callback = *p->sniCallback_;

TryCatch try_catch;

// Call it

Local<Value> ret = callback->Call(Context::GetCurrent()->Global(),

1,

argv);

if (try_catch.HasCaught()) {

FatalException(try_catch);

}

// If ret is SecureContext

if (secure_context_constructor->HasInstance(ret)) {

p->sniContext_ = Persistent<Value>::New(ret);

SecureContext *sc = ObjectWrap::Unwrap<SecureContext>(

Local<Object>::Cast(ret));

SSL_set_SSL_CTX(s, sc->ctx_);

} else {

return SSL_TLSEXT_ERR_NOACK;

}

}

}

return SSL_TLSEXT_ERR_OK;

}

#endif

Handle<Value> Connection::New(const Arguments& args) {

HandleScope scope;

Connection *p = new Connection();

p->Wrap(args.Holder());

if (args.Length() < 1 || !args[0]->IsObject()) {

return ThrowException(Exception::Error(String::New(

"First argument must be a crypto module Credentials")));

}

SecureContext *sc = ObjectWrap::Unwrap<SecureContext>(args[0]->ToObject());

bool is_server = args[1]->BooleanValue();

p->ssl_ = SSL_new(sc->ctx_);

p->bio_read_ = BIO_new(BIO_s_mem());

p->bio_write_ = BIO_new(BIO_s_mem());

SSL_set_app_data(p->ssl_, p);

#ifdef OPENSSL_NPN_NEGOTIATED

if (is_server) {

// Server should advertise NPN protocols

SSL_CTX_set_next_protos_advertised_cb(sc->ctx_,

AdvertiseNextProtoCallback_,

NULL);

} else {

// Client should select protocol from advertised

// If server supports NPN

SSL_CTX_set_next_proto_select_cb(sc->ctx_,

SelectNextProtoCallback_,

NULL);

}

#endif

#ifdef SSL_CTRL_SET_TLSEXT_SERVERNAME_CB

if (is_server) {

SSL_CTX_set_tlsext_servername_callback(sc->ctx_, SelectSNIContextCallback_);

} else {

String::Utf8Value servername(args[2]->ToString());

SSL_set_tlsext_host_name(p->ssl_, *servername);

}

#endif

SSL_set_bio(p->ssl_, p->bio_read_, p->bio_write_);

#ifdef SSL_MODE_RELEASE_BUFFERS

long mode = SSL_get_mode(p->ssl_);

SSL_set_mode(p->ssl_, mode | SSL_MODE_RELEASE_BUFFERS);

#endif

int verify_mode;

if (is_server) {

bool request_cert = args[2]->BooleanValue();

if (!request_cert) {

// Note reject_unauthorized ignored.

verify_mode = SSL_VERIFY_NONE;

} else {

bool reject_unauthorized = args[3]->BooleanValue();

verify_mode = SSL_VERIFY_PEER;

if (reject_unauthorized) verify_mode |= SSL_VERIFY_FAIL_IF_NO_PEER_CERT;

}

} else {

// Note request_cert and reject_unauthorized are ignored for clients.

verify_mode = SSL_VERIFY_NONE;

}

// Always allow a connection. We'll reject in javascript.

SSL_set_verify(p->ssl_, verify_mode, VerifyCallback);

if ((p->is_server_ = is_server)) {

SSL_set_accept_state(p->ssl_);

} else {

SSL_set_connect_state(p->ssl_);

}

return args.This();

}

Handle<Value> Connection::EncIn(const Arguments& args) {

HandleScope scope;

Connection *ss = Connection::Unwrap(args);

if (args.Length() < 3) {

return ThrowException(Exception::TypeError(

String::New("Takes 3 parameters")));

}

if (!Buffer::HasInstance(args[0])) {

return ThrowException(Exception::TypeError(

String::New("Second argument should be a buffer")));

}

Local<Object> buffer_obj = args[0]->ToObject();

char *buffer_data = Buffer::Data(buffer_obj);

size_t buffer_length = Buffer::Length(buffer_obj);

size_t off = args[1]->Int32Value();

if (off >= buffer_length) {

return ThrowException(Exception::Error(

String::New("Offset is out of bounds")));