Squashed 'third_party/seasocks/' content from commit 016dc60
Change-Id: I195fa5bfd0c0e3cc66fbbefcc7b5170bafcf7a36
git-subtree-dir: third_party/seasocks
git-subtree-split: 016dc60b247e0d1d563aea6d22a9075e6884ab9f
diff --git a/src/app/c/ws_test.cpp b/src/app/c/ws_test.cpp
new file mode 100644
index 0000000..2a8823a
--- /dev/null
+++ b/src/app/c/ws_test.cpp
@@ -0,0 +1,112 @@
+// Copyright (c) 2013, Matt Godbolt
+// All rights reserved.
+//
+// Redistribution and use in source and binary forms, with or without
+// modification, are permitted provided that the following conditions are met:
+//
+// Redistributions of source code must retain the above copyright notice, this
+// list of conditions and the following disclaimer.
+//
+// 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.
+//
+// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+// AND ANY EXPRESS 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 THE COPYRIGHT HOLDER OR 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.
+
+// An extraordinarily simple test which presents a web page with some buttons.
+// Clicking on the numbered button increments the number, which is visible to
+// other connected clients. WebSockets are used to do this: by the rather
+// suspicious means of sending raw JavaScript commands to be executed on other
+// clients.
+
+#include "seasocks/PrintfLogger.h"
+#include "seasocks/Server.h"
+#include "seasocks/StringUtil.h"
+#include "seasocks/WebSocket.h"
+#include "seasocks/util/Json.h"
+
+#include <cstring>
+#include <iostream>
+#include <memory>
+#include <set>
+#include <sstream>
+#include <string>
+
+using namespace seasocks;
+using namespace std;
+
+class MyHandler: public WebSocket::Handler {
+public:
+ MyHandler(Server* server) : _server(server), _currentValue(0) {
+ setValue(1);
+ }
+
+ virtual void onConnect(WebSocket* connection) {
+ _connections.insert(connection);
+ connection->send(_currentSetValue.c_str());
+ cout << "Connected: " << connection->getRequestUri()
+ << " : " << formatAddress(connection->getRemoteAddress())
+ << endl;
+ cout << "Credentials: " << *(connection->credentials()) << endl;
+ }
+
+ virtual void onData(WebSocket* connection, const char* data) {
+ if (0 == strcmp("die", data)) {
+ _server->terminate();
+ return;
+ }
+ if (0 == strcmp("close", data)) {
+ cout << "Closing.." << endl;
+ connection->close();
+ cout << "Closed." << endl;
+ return;
+ }
+
+ int value = atoi(data) + 1;
+ if (value > _currentValue) {
+ setValue(value);
+ for (auto connection : _connections) {
+ connection->send(_currentSetValue.c_str());
+ }
+ }
+ }
+
+ virtual void onDisconnect(WebSocket* connection) {
+ _connections.erase(connection);
+ cout << "Disconnected: " << connection->getRequestUri()
+ << " : " << formatAddress(connection->getRemoteAddress())
+ << endl;
+ }
+
+private:
+ set<WebSocket*> _connections;
+ Server* _server;
+ int _currentValue;
+ string _currentSetValue;
+
+ void setValue(int value) {
+ _currentValue = value;
+ _currentSetValue = makeExecString("set", _currentValue);
+ }
+};
+
+int main(int argc, const char* argv[]) {
+ shared_ptr<Logger> logger(new PrintfLogger(Logger::DEBUG));
+
+ Server server(logger);
+
+ shared_ptr<MyHandler> handler(new MyHandler(&server));
+ server.addWebSocketHandler("/ws", handler);
+ server.serve("src/ws_test_web", 9090);
+ return 0;
+}