From c09392f7bdc8c1f43a34866639bcb773e087f99c Mon Sep 17 00:00:00 2001
From: bel <bel2125@gmail.com>
Date: Thu, 31 Oct 2013 21:53:29 +0100
Subject: [PATCH] Replace trivial websocket example by a more illustrated one

---
 examples/websocket/Makefile           |   2 +-
 examples/websocket/WebSockCallbacks.c | 180 ++++++++++++++++++++++++++
 examples/websocket/WebSockCallbacks.h |  21 +++
 examples/websocket/websock.htm        |  55 ++++++++
 4 files changed, 257 insertions(+), 1 deletion(-)
 create mode 100644 examples/websocket/WebSockCallbacks.c
 create mode 100644 examples/websocket/WebSockCallbacks.h
 create mode 100644 examples/websocket/websock.htm

diff --git a/examples/websocket/Makefile b/examples/websocket/Makefile
index 3e0469e8..ba84c041 100644
--- a/examples/websocket/Makefile
+++ b/examples/websocket/Makefile
@@ -7,7 +7,7 @@
 
 
 PROG = websocket
-SRC = websocket.c
+SRC = WebSockCallbacks.c
 
 TOP = ../..
 CIVETWEB_LIB = libcivetweb.a
diff --git a/examples/websocket/WebSockCallbacks.c b/examples/websocket/WebSockCallbacks.c
new file mode 100644
index 00000000..ef37ce24
--- /dev/null
+++ b/examples/websocket/WebSockCallbacks.c
@@ -0,0 +1,180 @@
+#include <assert.h>
+#include <stdlib.h>
+#include <time.h>
+#include "WebSockCallbacks.h"
+
+#ifdef _WIN32
+#include <Windows.h>
+typedef HANDLE pthread_mutex_t;
+static int pthread_mutex_init(pthread_mutex_t *mutex, void *unused) {
+  unused = NULL;
+  *mutex = CreateMutex(NULL, FALSE, NULL);
+  return *mutex == NULL ? -1 : 0;
+}
+
+static int pthread_mutex_destroy(pthread_mutex_t *mutex) {
+  return CloseHandle(*mutex) == 0 ? -1 : 0;
+}
+
+static int pthread_mutex_lock(pthread_mutex_t *mutex) {
+  return WaitForSingleObject(*mutex, INFINITE) == WAIT_OBJECT_0? 0 : -1;
+}
+
+static int pthread_mutex_unlock(pthread_mutex_t *mutex) {
+  return ReleaseMutex(*mutex) == 0 ? -1 : 0;
+}
+#define mg_sleep(x) Sleep(x)
+#else
+#include <unistd.h>
+#include <pthread.h>
+#define mg_sleep(x) usleep((x) * 1000)
+#endif
+
+
+typedef struct tWebSockInfo {
+  int webSockState;
+  unsigned long initId;
+  struct mg_connection *conn;
+} tWebSockInfo;
+
+static pthread_mutex_t sMutex;
+
+#define MAX_NUM_OF_WEBSOCKS (256)
+static tWebSockInfo *socketList[MAX_NUM_OF_WEBSOCKS];
+
+
+void websocket_ready_handler(struct mg_connection *conn) {
+
+  int i;
+  struct mg_request_info * rq = mg_get_request_info(conn);
+  tWebSockInfo * wsock = malloc(sizeof(tWebSockInfo));
+  assert(wsock);
+  wsock->webSockState = 0;
+  rq->conn_data = wsock;
+
+  pthread_mutex_lock(&sMutex);
+  for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) {
+    if (0==socketList[i]) {
+      socketList[i] = wsock;
+      wsock->conn = conn;
+      wsock->webSockState = 1;
+      break;
+    }
+  }
+  pthread_mutex_unlock(&sMutex);
+}
+
+
+static void websocket_done(tWebSockInfo * wsock) { 
+  int i;
+  if (wsock) {
+    wsock->webSockState = 99;
+    for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) {
+      if (wsock==socketList[i]) {
+        socketList[i] = 0;
+        break;
+      }
+    }
+    free(wsock);
+  }
+}
+
+
+int websocket_data_handler(struct mg_connection *conn, int flags, char *data, size_t data_len) {
+  struct mg_request_info * rq = mg_get_request_info(conn);
+  tWebSockInfo * wsock = (tWebSockInfo*)rq->conn_data;
+  char msg[128];
+  int i;
+
+  pthread_mutex_lock(&sMutex);
+  if (flags==136) {
+    // close websock
+    websocket_done(wsock);
+    rq->conn_data = 0;
+    pthread_mutex_unlock(&sMutex);
+    return 1;
+  }
+  if ((data_len>=5) && (data_len<100) && (flags==129) || (flags==130)) {
+
+    // init command
+    if ((wsock->webSockState==1) && (!memcmp(data,"init ",5))) {
+      char * chk;
+      unsigned long gid;
+      memcpy(msg,data+5,data_len-5);
+      msg[data_len-5]=0;
+      gid = strtoul(msg,&chk,10);
+      wsock->initId = gid;
+      if (gid>0 && chk!=NULL && *chk==0) {
+        wsock->webSockState = 2;
+      }
+      pthread_mutex_unlock(&sMutex);
+      return 1;
+    }
+        
+    // chat message
+    if ((wsock->webSockState==2) && (!memcmp(data,"msg ",4))) {     
+      for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) {
+        if (socketList[i] && (socketList[i]->webSockState==2)) {
+          mg_websocket_write(socketList[i]->conn, WEBSOCKET_OPCODE_TEXT, data, data_len);
+        }
+      }
+      pthread_mutex_unlock(&sMutex);
+      return 1;
+    }
+  }
+
+  // keep alive
+  if ((data_len==4) && !memcmp(data,"ping",4)) {
+    pthread_mutex_unlock(&sMutex);
+    return 1;
+  }
+
+  pthread_mutex_unlock(&sMutex);
+  return 0;
+}
+
+
+void connection_close_handler(struct mg_connection *conn) {   
+  struct mg_request_info * rq = mg_get_request_info(conn);
+  tWebSockInfo * wsock = (tWebSockInfo*)rq->conn_data;
+
+  pthread_mutex_lock(&sMutex);
+  websocket_done(wsock);
+  rq->conn_data = 0;
+  pthread_mutex_unlock(&sMutex);
+}
+
+
+static void * eventMain(void * _ignored) {
+  int i;
+  char msg[256];
+  for (;;) {
+    time_t t = time(0);
+    struct tm * timestr = localtime(&t);
+    sprintf(msg,"title %s",asctime(timestr));
+
+    pthread_mutex_lock(&sMutex);
+    for (i=0;i<MAX_NUM_OF_WEBSOCKS;i++) {
+      if (socketList[i] && (socketList[i]->webSockState==2)) {
+        mg_websocket_write(socketList[i]->conn, WEBSOCKET_OPCODE_TEXT, msg, strlen(msg));
+      }
+    }
+    pthread_mutex_unlock(&sMutex);
+
+    mg_sleep(1000);
+  }
+
+  return _ignored;
+}
+
+
+void websock_init_lib(void) {
+  int ret;
+  ret = pthread_mutex_init(&sMutex, 0);
+  assert(ret==0);
+
+  memset(socketList,0,sizeof(socketList));
+
+  mg_start_thread(eventMain, 0);
+}
+
diff --git a/examples/websocket/WebSockCallbacks.h b/examples/websocket/WebSockCallbacks.h
new file mode 100644
index 00000000..2bd8f1e0
--- /dev/null
+++ b/examples/websocket/WebSockCallbacks.h
@@ -0,0 +1,21 @@
+
+#ifndef WEBSOCKCALLBACKS_H_INCLUDED
+#define WEBSOCKCALLBACKS_H_INCLUDED
+
+#include "civetweb.h"
+
+#ifdef __cplusplus
+extern "C" {
+#endif
+
+void websock_init_lib(void);
+void websocket_ready_handler(struct mg_connection *conn);
+int websocket_data_handler(struct mg_connection *conn, int flags, char *data, size_t data_len);
+void connection_close_handler(struct mg_connection *conn);
+
+
+#ifdef __cplusplus
+}
+#endif
+
+#endif
\ No newline at end of file
diff --git a/examples/websocket/websock.htm b/examples/websocket/websock.htm
new file mode 100644
index 00000000..4ff3a5fa
--- /dev/null
+++ b/examples/websocket/websock.htm
@@ -0,0 +1,55 @@
+<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01//EN" "http://www.w3.org/TR/html4/strict.dtd">
+<html>
+<head>
+  <title>Test</title>
+  <script type='text/javascript' language="javascript">
+  <!--
+  var connection;
+  var keepAlive = false;
+
+  function webSockKeepAlive() {
+    if (keepAlive) {
+      connection.send('ping'); // Send the message 'ping' to the server
+      setTimeout("webSockKeepAlive()", 10000);
+    }
+  }
+
+  function load() {
+    connection = new WebSocket("ws://127.0.0.1/MyWebSock");
+
+    connection.onopen = function () {
+        var send = "init " + Math.round(Math.random()*4294967294+1);
+        console.log('Client: ' + send);
+        connection.send(send);
+        keepAlive = true;
+        webSockKeepAlive();
+      };
+
+    connection.onerror = function (error) {
+        keepAlive = false;
+        connection.close();
+        console.log('WebSocket error: ' + error);
+        alert("WebSocket error");
+      };
+
+    connection.onmessage = function (e) {
+        console.log('Server: ' + e.data);
+        if (e.data.substring(0,5) == "title") {window.document.title = e.data.substring(6);}
+        else if (e.data.substring(0,3) == "msg") {
+          var msgStr = document.getElementById('msg');
+          msgStr.innerHTML = msgStr.innerHTML + e.data.substring(4);
+        }        
+      };
+  }
+  //-->
+  </script>
+
+</head>
+<body onload="load()">
+  <input type="button" onclick="connection.send('msg A');" value="A"></button>
+  <input type="button" onclick="connection.send('msg B');" value="B"></button>
+  <input type="button" onclick="connection.send('msg C');" value="C"></button>
+  <input type="button" onclick="connection.send('msg D');" value="D"></button>
+  <b id="msg"></b>
+</body>
+</html>
\ No newline at end of file