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;
+}

examples/echo_server/Makefile

@@ -0,0 +1,23 @@
+CFLAGS = -O2
+LDFLAGS = 
+
+CXX		?= c++
+SHARED  ?= "1"
+
+ifeq ($(SHARED), 1)
+	LDFLAGS := $(LDFLAGS) -lboost_system -lwebsocketpp
+else
+	LDFLAGS := $(LDFLAGS) -lboost_system ../../libwebsocketpp.a
+endif
+
+echo_server: echo_server.o echo.o
+	$(CXX) $(CFLAGS) $^ -o $@ $(LDFLAGS)
+
+%.o: %.cpp
+	$(CXX) -c $(CFLAGS) -o $@ $^
+
+# cleanup by removing generated files
+#
+.PHONY:		clean
+clean:
+		rm -f *.o echo_server

examples/echo_server/echo.cpp

@@ -0,0 +1,16 @@
+#include "echo.hpp"
+
+using websocketecho::echo_handler;
+
+bool echo_handler::validate(websocketpp::session_ptr client) {
+	return true;
+}
+
+void echo_handler::message(websocketpp::session_ptr client, const std::string &msg) {
+	client->send(msg);
+}
+
+void echo_handler::message(websocketpp::session_ptr client,
+	const std::vector<unsigned char> &data) {
+	client->send(data);
+}

examples/echo_server/echo.hpp

@@ -0,0 +1,35 @@
+#ifndef ECHO_HANDLER_HPP
+#define ECHO_HANDLER_HPP
+
+#include "../../src/websocket_connection_handler.hpp"
+#include <boost/shared_ptr.hpp>
+
+#include <string>
+#include <vector>
+
+namespace websocketecho {
+
+class echo_handler : public websocketpp::connection_handler {
+public:
+	echo_handler() {}
+	virtual ~echo_handler() {}
+	
+	// The echo server allows all domains is protocol free.
+	bool validate(websocketpp::session_ptr client);
+	
+	// an echo server is stateless. The handler has no need to keep track of connected
+	// clients.
+	void connect(websocketpp::session_ptr client) {}
+	void disconnect(websocketpp::session_ptr client,const std::string &reason) {}
+	
+	// both text and binary messages are echoed back to the sending client.
+	void message(websocketpp::session_ptr client,const std::string &msg);
+	void message(websocketpp::session_ptr client,
+		const std::vector<unsigned char> &data);
+};
+
+typedef boost::shared_ptr<echo_handler> echo_handler_ptr;
+
+}
+
+#endif // ECHO_HANDLER_HPP

examples/echo_server/echo_client.html

@@ -0,0 +1,94 @@
+<!doctype html>
+<html>
+<head>
+</head>
+<body>
+
+<script type="text/javascript">
+var ws;
+var url;
+
+function connect() {
+	url = document.getElementById("server_url").value;
+	console.log(url);
+	
+	if ("WebSocket" in window) {
+		ws = new WebSocket(url);
+	} else if ("MozWebSocket" in window) {
+		ws = new WebSocket(url);
+	} else {
+		document.getElementById("messages").innerHTML += "This Browser does not support WebSockets<br />";
+		return;
+	}
+	ws.onopen = function(e) {
+		document.getElementById("messages").innerHTML += "Client: A connection to "+ws.URL+" has been opened.<br />";
+		
+		document.getElementById("server_url").disabled = true;
+		document.getElementById("toggle_connect").innerHTML = "Disconnect";
+	};
+	
+	ws.onerror = function(e) {
+		document.getElementById("messages").innerHTML += "Client: An error occured, see console log for more details.<br />";
+		console.log(e);
+	};
+	
+	ws.onclose = function(e) {
+		document.getElementById("messages").innerHTML += "Client: The connection to "+url+" was closed.<br />";
+	};
+	
+	ws.onmessage = function(e) {
+		document.getElementById("messages").innerHTML += "Server: "+e.data+"<br />";
+	};
+}
+
+function disconnect() {
+	ws.close();
+	document.getElementById("server_url").disabled = false;
+	document.getElementById("toggle_connect").innerHTML = "Connect";
+}
+
+function toggle_connect() {
+	if (document.getElementById("server_url").disabled === false) {
+		connect();
+	} else {
+		disconnect();
+	}
+}
+
+function send() {
+	if (ws === undefined || ws.readyState != 1) {
+		document.getElementById("messages").innerHTML += "Client: Websocket is not avaliable for writing<br />";
+		return;
+	}
+	
+	ws.send(document.getElementById("msg").value);
+	document.getElementById("msg").value = "";
+}
+
+</script>
+
+<style>
+body,html {
+	margin: 0px;
+	padding: 0px;
+}
+#controls {
+	float:right;
+	background-color: #999;
+}
+
+</style>
+
+<div id="controls">
+	<div id="server">
+	<input type="text" name="server_url" id="server_url" value="ws://localhost:5000" />
+	<button id="toggle_connect" onclick="toggle_connect();">Connect</button>
+	</div>
+
+	<div id="message_input"><input type="text" name="msg" id="msg" value="Hello World!" />
+	<button onclick="send();">Send</button></div>
+</div>
+<div id="messages"></div>
+
+</body>
+</html>

examples/echo_server/echo_server.cpp

@@ -0,0 +1,43 @@
+#include "echo.hpp"
+
+#include "../../src/websocketpp.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) {
+		// TODO: input validation?
+		port = atoi(argv[2]);
+
+		
+		std::stringstream temp;
+		temp << argv[1] << ":" << port;
+		
+		host = temp.str();
+	}
+	
+	websocketecho::echo_handler_ptr echo_handler(new websocketecho::echo_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,echo_handler)
+		);
+		
+		std::cout << "Starting echo server on " << host << std::endl;
+		
+		io_service.run();
+	} catch (std::exception& e) {
+		std::cerr << "Exception: " << e.what() << std::endl;
+	}
+	
+	return 0;
+}