1 files changed, 124 insertions, 0 deletions
diff --git a/cpp/src/qpid/client/SessionHandler.cpp b/cpp/src/qpid/client/SessionHandler.cpp
new file mode 100644
index 0000000000..3885ac437a
--- /dev/null
+++ b/cpp/src/qpid/client/SessionHandler.cpp
@@ -0,0 +1,124 @@
+/*
+ *
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ * 
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ * 
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ *
+ */
+
+#include "SessionHandler.h"
+#include "qpid/framing/amqp_framing.h"
+#include "qpid/framing/all_method_bodies.h"
+
+using namespace qpid::client;
+using namespace qpid::framing;
+using namespace boost;
+
+SessionHandler::SessionHandler() : StateManager(CLOSED), id(0) {}
+
+void SessionHandler::incoming(AMQFrame& frame)
+{
+    AMQBody* body = frame.getBody();
+    if (getState() == OPEN) {
+        SessionClosedBody* closeBody=
+            dynamic_cast<SessionClosedBody*>(body->getMethod());
+        if (closeBody) {
+            setState(CLOSED_BY_PEER);
+            code = closeBody->getReplyCode();
+            text = closeBody->getReplyText();
+            if (onClose) {
+                onClose(closeBody->getReplyCode(), closeBody->getReplyText());
+            }
+        } else {
+            try {
+                in(frame);
+            }catch(ChannelException& e){
+                closed(e.code, e.toString());
+            }
+        }
+    } else {
+        if (body->getMethod()) 
+            handleMethod(body->getMethod());
+        else
+            throw ConnectionException(504, "Channel not open for content.");
+    }
+}
+
+void SessionHandler::outgoing(AMQFrame& frame)
+{
+    if (getState() == OPEN) {
+        frame.setChannel(id);
+        out(frame);
+    } else if (getState() == CLOSED) {
+        throw Exception(QPID_MSG("Channel not open, can't send " << frame));
+    } else if (getState() == CLOSED_BY_PEER) {
+        throw ChannelException(code, text);
+    }
+}
+
+void SessionHandler::open(uint16_t _id)
+{
+    id = _id;
+
+    setState(OPENING);
+    AMQFrame f(id, SessionOpenBody(version));
+    out(f);
+
+    std::set<int> states;
+    states.insert(OPEN);
+    states.insert(CLOSED_BY_PEER);
+    waitFor(states);
+    if (getState() != OPEN) {
+        throw Exception("Failed to open channel.");
+    }
+}
+
+void SessionHandler::close()
+{
+    setState(CLOSING);
+    AMQFrame f(id, SessionCloseBody(version));
+    out(f);
+    waitFor(CLOSED);
+}
+
+void SessionHandler::closed(uint16_t code, const std::string& msg)
+{
+    setState(CLOSED);
+    AMQFrame f(id, SessionClosedBody(version, code, msg));
+    out(f);
+}
+
+void SessionHandler::handleMethod(AMQMethodBody* method)
+{
+    switch (getState()) {
+      case OPENING:
+        if (method->isA<SessionAttachedBody>()) {
+            setState(OPEN);
+        } else {
+            throw ConnectionException(504, "Channel not opened.");
+        }
+        break;
+      case CLOSING:
+        if (method->isA<SessionClosedBody>()) {
+            setState(CLOSED);
+        } //else just ignore it
+        break;
+      case CLOSED:
+        throw ConnectionException(504, "Channel is closed.");
+      default:
+        throw Exception("Unexpected state encountered in SessionHandler!");
+    }
+}