quickfix_ruby_ud 2.0.7-x86_64-linux

data/ext/quickfix/ThreadedSSLSocketConnection.cpp

@@ -0,0 +1,364 @@
+/* ====================================================================
+ * Copyright (c) 1998-2006 Ralf S. Engelschall. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following
+ *    disclaimer in the documentation and/or other materials
+ *    provided with the distribution.
+ *
+ * 3. All advertising materials mentioning features or use of this
+ *    software must display the following acknowledgment:
+ *    "This product includes software developed by
+ *     Ralf S. Engelschall <rse@engelschall.com> for use in the
+ *     mod_ssl project (http://www.modssl.org/)."
+ *
+ * 4. The names "mod_ssl" must not be used to endorse or promote
+ *    products derived from this software without prior written
+ *    permission. For written permission, please contact
+ *    rse@engelschall.com.
+ *
+ * 5. Products derived from this software may not be called "mod_ssl"
+ *    nor may "mod_ssl" appear in their names without prior
+ *    written permission of Ralf S. Engelschall.
+ *
+ * 6. Redistributions of any form whatsoever must retain the following
+ *    acknowledgment:
+ *    "This product includes software developed by
+ *     Ralf S. Engelschall <rse@engelschall.com> for use in the
+ *     mod_ssl project (http://www.modssl.org/)."
+ *
+ * THIS SOFTWARE IS PROVIDED BY RALF S. ENGELSCHALL ``AS IS'' AND ANY
+ * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL RALF S. ENGELSCHALL OR
+ * HIS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
+ * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
+ * OF THE POSSIBILITY OF SUCH DAMAGE.
+ * ====================================================================
+ */
+
+/* ====================================================================
+ * Copyright (c) 1995-1999 Ben Laurie. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in
+ *    the documentation and/or other materials provided with the
+ *    distribution.
+ *
+ * 3. All advertising materials mentioning features or use of this
+ *    software must display the following acknowledgment:
+ *    "This product includes software developed by Ben Laurie
+ *    for use in the Apache-SSL HTTP server project."
+ *
+ * 4. The name "Apache-SSL Server" must not be used to
+ *    endorse or promote products derived from this software without
+ *    prior written permission.
+ *
+ * 5. Redistributions of any form whatsoever must retain the following
+ *    acknowledgment:
+ *    "This product includes software developed by Ben Laurie
+ *    for use in the Apache-SSL HTTP server project."
+ *
+ * THIS SOFTWARE IS PROVIDED BY BEN LAURIE ``AS IS'' AND ANY
+ * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL BEN LAURIE OR
+ * HIS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
+ * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
+ * OF THE POSSIBILITY OF SUCH DAMAGE.
+ * ====================================================================
+ */
+
+/****************************************************************************
+** Copyright (c) 2001-2014
+**
+** This file is part of the QuickFIX FIX Engine
+**
+** This file may be distributed under the terms of the quickfixengine.org
+** license as defined by quickfixengine.org and appearing in the file
+** LICENSE included in the packaging of this file.
+**
+** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
+** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
+**
+** See http://www.quickfixengine.org/LICENSE for licensing information.
+**
+** Contact ask@quickfixengine.org if any conditions of this licensing are
+** not clear to you.
+**
+****************************************************************************/
+
+#ifdef _MSC_VER
+#include "stdafx.h"
+#else
+#include "config.h"
+#endif
+
+#if (HAVE_SSL > 0)
+
+#include "Session.h"
+#include "ThreadedSSLSocketAcceptor.h"
+#include "ThreadedSSLSocketConnection.h"
+#include "ThreadedSSLSocketInitiator.h"
+#include "Utility.h"
+
+namespace FIX {
+ThreadedSSLSocketConnection::ThreadedSSLSocketConnection(socket_handle socket, SSL *ssl, Sessions sessions, Log *pLog)
+    : m_socket(socket),
+      m_ssl(ssl),
+      m_pLog(pLog),
+      m_sessions(sessions),
+      m_pSession(0),
+      m_disconnect(false) {
+  FD_ZERO(&m_fds);
+  FD_SET(m_socket, &m_fds);
+}
+
+ThreadedSSLSocketConnection::ThreadedSSLSocketConnection(
+    const SessionID &sessionID,
+    socket_handle socket,
+    SSL *ssl,
+    const std::string &address,
+    short port,
+    Log *pLog)
+    : m_socket(socket),
+      m_ssl(ssl),
+      m_address(address),
+      m_port(port),
+      m_pLog(pLog),
+      m_pSession(Session::lookupSession(sessionID)),
+      m_disconnect(false) {
+  FD_ZERO(&m_fds);
+  FD_SET(m_socket, &m_fds);
+  if (m_pSession) {
+    m_pSession->setResponder(this);
+  }
+}
+
+ThreadedSSLSocketConnection::~ThreadedSSLSocketConnection() {
+  if (m_pSession) {
+    m_pSession->setResponder(0);
+    Session::unregisterSession(m_pSession->getSessionID());
+  }
+}
+
+bool ThreadedSSLSocketConnection::send(const std::string &message) {
+  int totalSent = 0;
+
+  while (totalSent < (int)message.length()) {
+    errno = 0;
+    int errCodeSSL = 0;
+    int sent = 0;
+    ERR_clear_error();
+
+    // Cannot do concurrent SSL write and read as ssl context has to be
+    // protected.
+    {
+      Locker locker(m_mutex);
+
+      sent = SSL_write(m_ssl, message.c_str() + totalSent, message.length() - totalSent);
+      if (sent <= 0) {
+        errCodeSSL = SSL_get_error(m_ssl, sent);
+      }
+    }
+
+    if (sent <= 0) {
+      if ((errCodeSSL == SSL_ERROR_WANT_READ) || (errCodeSSL == SSL_ERROR_WANT_WRITE)) {
+        errno = EINTR;
+        sent = 0;
+      } else {
+        std::string error = socket_error();
+
+        m_pSession->getLog()->onEvent("SSL send error <" + IntConvertor::convert(errCodeSSL) + "> " + error);
+
+        return false;
+      }
+    }
+
+    totalSent += sent;
+  }
+
+  return true;
+}
+
+bool ThreadedSSLSocketConnection::connect() { return socket_connect(getSocket(), m_address.c_str(), m_port) >= 0; }
+
+void ThreadedSSLSocketConnection::disconnect() {
+  m_disconnect = true;
+  ssl_socket_close(m_socket, m_ssl);
+}
+
+bool ThreadedSSLSocketConnection::read() {
+  struct timeval timeout = {1, 0};
+  fd_set readset = m_fds;
+
+  try {
+    // Wait for input (1 second timeout)
+    int result = select(1 + m_socket, &readset, 0, 0, &timeout);
+
+    if (result > 0) // Something to read
+    {
+      bool pending = false;
+
+      do {
+        pending = false;
+        errno = 0;
+        int size = 0;
+        int errCodeSSL = 0;
+        ERR_clear_error();
+
+        // Cannot do concurrent SSL write and read as ssl context has to be
+        // protected.
+        {
+          Locker locker(m_mutex);
+
+          size = SSL_read(m_ssl, m_buffer, sizeof(m_buffer));
+          if (size <= 0) {
+            errCodeSSL = SSL_get_error(m_ssl, size);
+          } else if (SSL_pending(m_ssl) > 0) {
+            pending = true;
+          }
+        }
+
+        if (size <= 0) {
+          if ((errCodeSSL == SSL_ERROR_WANT_READ) || (errCodeSSL == SSL_ERROR_WANT_WRITE)) {
+            errno = EINTR;
+            size = 0;
+
+            return true;
+          } else {
+            std::string error = socket_error();
+
+            if (m_pSession) {
+              m_pSession->getLog()->onEvent("SSL read error <" + IntConvertor::convert(errCodeSSL) + "> " + error);
+            } else {
+              std::cerr << UtcTimeStampConvertor::convert(UtcTimeStamp::now()) << "SSL read error <"
+                        << IntConvertor::convert(errCodeSSL) << "> " << error << std::endl;
+            }
+
+            throw SocketRecvFailed(size);
+          }
+        }
+
+        m_parser.addToStream(m_buffer, size);
+      } while (pending);
+    } else if (result == 0 && m_pSession) // Timeout
+    {
+      m_pSession->next(UtcTimeStamp::now());
+    } else if (result < 0) // Error
+    {
+      throw SocketRecvFailed(result);
+    }
+
+    processStream();
+
+    if (m_disconnect) {
+      return false;
+    }
+
+    return true;
+  } catch (SocketRecvFailed &e) {
+    if (m_disconnect) {
+      return false;
+    }
+
+    if (m_pSession) {
+      m_pSession->getLog()->onEvent(e.what());
+      m_pSession->disconnect();
+    } else {
+      disconnect();
+    }
+
+    return false;
+  }
+}
+
+bool ThreadedSSLSocketConnection::readMessage(std::string &message) EXCEPT(SocketRecvFailed) {
+  try {
+    return m_parser.readFixMessage(message);
+  } catch (MessageParseError &) {}
+  return true;
+}
+
+void ThreadedSSLSocketConnection::processStream() {
+  std::string message;
+  while (readMessage(message)) {
+    if (!m_pSession) {
+      if (!setSession(message)) {
+        disconnect();
+        continue;
+      }
+    }
+    try {
+      m_pSession->next(message, UtcTimeStamp::now());
+    } catch (InvalidMessage &) {
+      if (!m_pSession->isLoggedOn()) {
+        disconnect();
+        return;
+      }
+    }
+  }
+}
+
+bool ThreadedSSLSocketConnection::setSession(const std::string &message) {
+  m_pSession = Session::lookupSession(message, true);
+  if (!m_pSession) {
+    if (m_pLog) {
+      m_pLog->onEvent("Session not found for incoming message: " + message);
+      m_pLog->onIncoming(message);
+    }
+    return false;
+  }
+
+  SessionID sessionID = m_pSession->getSessionID();
+  m_pSession = 0;
+
+  // see if the session frees up within 5 seconds
+  for (int i = 1; i <= 5; i++) {
+    if (!Session::isSessionRegistered(sessionID)) {
+      m_pSession = Session::registerSession(sessionID);
+    }
+    if (m_pSession) {
+      break;
+    }
+    process_sleep(1);
+  }
+
+  if (!m_pSession) {
+    return false;
+  }
+  if (m_sessions.find(m_pSession->getSessionID()) == m_sessions.end()) {
+    return false;
+  }
+
+  m_pSession->setResponder(this);
+  return true;
+}
+
+} // namespace FIX
+
+#endif

data/ext/quickfix/ThreadedSSLSocketConnection.h

@@ -0,0 +1,191 @@
+/* ====================================================================
+ * Copyright (c) 1998-2006 Ralf S. Engelschall. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following
+ *    disclaimer in the documentation and/or other materials
+ *    provided with the distribution.
+ *
+ * 3. All advertising materials mentioning features or use of this
+ *    software must display the following acknowledgment:
+ *    "This product includes software developed by
+ *     Ralf S. Engelschall <rse@engelschall.com> for use in the
+ *     mod_ssl project (http://www.modssl.org/)."
+ *
+ * 4. The names "mod_ssl" must not be used to endorse or promote
+ *    products derived from this software without prior written
+ *    permission. For written permission, please contact
+ *    rse@engelschall.com.
+ *
+ * 5. Products derived from this software may not be called "mod_ssl"
+ *    nor may "mod_ssl" appear in their names without prior
+ *    written permission of Ralf S. Engelschall.
+ *
+ * 6. Redistributions of any form whatsoever must retain the following
+ *    acknowledgment:
+ *    "This product includes software developed by
+ *     Ralf S. Engelschall <rse@engelschall.com> for use in the
+ *     mod_ssl project (http://www.modssl.org/)."
+ *
+ * THIS SOFTWARE IS PROVIDED BY RALF S. ENGELSCHALL ``AS IS'' AND ANY
+ * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL RALF S. ENGELSCHALL OR
+ * HIS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
+ * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
+ * OF THE POSSIBILITY OF SUCH DAMAGE.
+ * ====================================================================
+ */
+
+/* ====================================================================
+ * Copyright (c) 1995-1999 Ben Laurie. All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in
+ *    the documentation and/or other materials provided with the
+ *    distribution.
+ *
+ * 3. All advertising materials mentioning features or use of this
+ *    software must display the following acknowledgment:
+ *    "This product includes software developed by Ben Laurie
+ *    for use in the Apache-SSL HTTP server project."
+ *
+ * 4. The name "Apache-SSL Server" must not be used to
+ *    endorse or promote products derived from this software without
+ *    prior written permission.
+ *
+ * 5. Redistributions of any form whatsoever must retain the following
+ *    acknowledgment:
+ *    "This product includes software developed by Ben Laurie
+ *    for use in the Apache-SSL HTTP server project."
+ *
+ * THIS SOFTWARE IS PROVIDED BY BEN LAURIE ``AS IS'' AND ANY
+ * EXPRESSED OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
+ * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL BEN LAURIE OR
+ * HIS CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
+ * SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
+ * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT,
+ * STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
+ * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED
+ * OF THE POSSIBILITY OF SUCH DAMAGE.
+ * ====================================================================
+ */
+
+/* -*- C++ -*- */
+
+/****************************************************************************
+** Copyright (c) 2001-2014
+**
+** This file is part of the QuickFIX FIX Engine
+**
+** This file may be distributed under the terms of the quickfixengine.org
+** license as defined by quickfixengine.org and appearing in the file
+** LICENSE included in the packaging of this file.
+**
+** This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
+** WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
+**
+** See http://www.quickfixengine.org/LICENSE for licensing information.
+**
+** Contact ask@quickfixengine.org if any conditions of this licensing are
+** not clear to you.
+**
+****************************************************************************/
+
+#ifndef FIX_THREADEDSSLSOCKETCONNECTION_H
+#define FIX_THREADEDSSLSOCKETCONNECTION_H
+
+#if (HAVE_SSL > 0)
+
+#ifdef _MSC_VER
+#pragma warning(disable : 4503 4355 4786 4290)
+#endif
+
+#include "Mutex.h"
+#include "Parser.h"
+#include "Responder.h"
+#include "SessionID.h"
+#include "UtilitySSL.h"
+#include <map>
+#include <set>
+
+namespace FIX {
+class ThreadedSSLSocketAcceptor;
+class ThreadedSSLSocketInitiator;
+class Session;
+class Application;
+class Log;
+
+/// Encapsulates a socket file descriptor (multi-threaded).
+class ThreadedSSLSocketConnection : Responder {
+public:
+  typedef std::set<SessionID> Sessions;
+
+  ThreadedSSLSocketConnection(socket_handle s, SSL *ssl, Sessions sessions, Log *pLog);
+  ThreadedSSLSocketConnection(
+      const SessionID &,
+      socket_handle s,
+      SSL *ssl,
+      const std::string &address,
+      short port,
+      Log *pLog);
+  virtual ~ThreadedSSLSocketConnection();
+
+  Session *getSession() const { return m_pSession; }
+  socket_handle getSocket() const { return m_socket; }
+  bool connect();
+  void disconnect();
+  bool read();
+  SSL *sslObject() { return m_ssl; }
+
+private:
+  typedef std::pair<socket_handle, SSL *> SocketKey;
+
+  bool readMessage(std::string &msg) EXCEPT(SocketRecvFailed);
+  void processStream();
+  bool send(const std::string &);
+  bool setSession(const std::string &msg);
+
+  socket_handle m_socket;
+  SSL *m_ssl;
+  char m_buffer[BUFSIZ];
+
+  std::string m_address;
+  int m_port;
+
+  Log *m_pLog;
+  Parser m_parser;
+  Sessions m_sessions;
+  Session *m_pSession;
+  bool m_disconnect;
+  fd_set m_fds;
+
+  Mutex m_mutex;
+};
+} // namespace FIX
+
+#endif // FIX_THREADEDSOCKETCONNECTION_H
+
+#endif