initial commit

examples/chat_server/chat.cpp

@@ -0,0 +1,30 @@
+#include "chat.hpp"
+
+using websocketchat::chat_handler;
+
+
+bool chat_handler::validate(websocketpp::session_ptr client) {
+	// We only know about the chat resource
+	if (client->get_request() != "/chat") {
+		client->set_http_error(404);
+		return false;
+	}
+	
+	// Require specific origin example
+	if (client->get_header("Sec-WebSocket-Origin") != "http://zaphoyd.com") {
+		client->set_http_error(403);
+		return false;
+	}
+	
+	return true;
+}
+
+
+void connect(session_ptr client) {
+	std::cout << "client " << client << " joined the lobby." << std::endl;
+	m_connections.insert(client);
+
+	// send user list to all clients
+	// send signon message from server to all other clients.
+
+}

examples/chat_server/chat.hpp

@@ -0,0 +1,85 @@
+#ifndef CHAT_HPP
+#define CHAT_HPP
+
+// com.zaphoyd.websocketpp.chat protocol
+// 
+// client messages:
+// msg [UTF8 text]
+// 
+// server messages:
+// {"type":"msg","sender":"<sender>","value":"<msg>" }
+// {"type":"participants","value":[<participant>,...]}
+
+#include <boost/shared_ptr.hpp>
+
+#include <websocketpp/websocket_connection_handler.hpp>
+
+#include <set>
+#include <map>
+#include <string>
+#include <vector>
+
+namespace websocketchat {
+
+class chat_handler : public websocketpp::connection_handler {
+public:
+	chat_handler() {}
+	virtual ~chat_handler() {}
+	
+	bool validate(websocketpp::session_ptr client); 
+	
+	// add new connection to the lobby
+	void connect(session_ptr client) {
+		std::cout << "client " << client << " joined the lobby." << std::endl;
+		m_connections.insert(client);
+
+		// send user list to all clients
+		// send signon message from server to all other clients.
+
+	}
+		
+	// someone disconnected from the lobby, remove them
+	void disconnect(websocketpp::session_ptr client,const std::string &reason) {
+		std::set<session_ptr>::iterator it = m_connections.find(client);
+		
+		if (it == m_connections.end()) {
+			// this client has already disconnected, we can ignore this.
+			// this happens during certain types of disconnect where there is a
+			// deliberate "soft" disconnection preceeding the "hard" socket read
+			// fail or disconnect ack message.
+			return;
+		}
+		
+		std::cout << "client " << client << " left the lobby." << std::endl;
+		
+		m_connections.erase(it);
+
+		// send signoff message from server to all clients
+		// send user list to all remaining clients
+	}
+	
+	void message(websocketpp::session_ptr client,const std::string &msg) {
+		std::cout << "message from client " << client << ": " << msg << std::endl;
+		
+		// create JSON message to send based on msg
+
+		for (std::set<session_ptr>::iterator it = m_connections.begin();
+												it != m_connections.end(); it++) {
+			(*it)->send(msg);
+		}
+	}
+	
+	// lobby will ignore binary messages
+	void message(websocketpp::session_ptr client,
+		const std::vector<unsigned char> &data) {}
+private:
+	std::string serialize_state();
+
+	// list of outstanding connections
+	std::set<session_ptr> m_connections;
+};
+
+typedef boost::shared_ptr<chat_handler> chat_handler_ptr;
+
+}
+#endif // CHAT_HPP

examples/chat_server/chat_server.cpp

@@ -0,0 +1,41 @@
+#include "chat.hpp"
+
+#include <websocketpp/websocket_server.hpp>
+#include <boost/asio.hpp>
+
+#include <iostream>
+
+using boost::asio::ip::tcp;
+
+int main(int argc, char* argv[]) {
+	std::string host = "localhost:5000";
+	short port = 5000;
+	
+	if (argc == 3) {
+		port = atoi(argv[2]);
+		
+		std::stringstream temp;
+		temp << argv[1] << ":" << port;
+		
+		host = temp.str();
+	}
+	
+	websocketchat::chat_handler_ptr chat_handler(new websocketchat::chat_handler());
+	
+	try {
+		boost::asio::io_service io_service;
+		tcp::endpoint endpoint(tcp::v6(), port);
+		
+		websocketpp::server_ptr server(
+			new websocketpp::server(io_service,endpoint,host,chat_handler)
+		);
+		
+		std::cout << "Starting chat server on " << host << std::endl;
+		
+		io_service.run();
+	} catch (std::exception& e) {
+		std::cerr << "Exception: " << e.what() << std::endl;
+	}
+	
+	return 0;
+}