Test for webrtc sockets.

4 files changed, 237 insertions, 1 deletions

diff --git a/tests/runner.py b/tests/runner.py
index 2f508dfc..af671c11 100755
--- a/tests/runner.py
+++ b/tests/runner.py
@@ -771,4 +771,4 @@ an individual test with
   else:
     testRunner = unittest.TextTestRunner(verbosity=2)
     for suite in suites:
-      testRunner.run(suite)
+      testRunner.run(suite)
\ No newline at end of file
diff --git a/tests/sockets/webrtc_host.c b/tests/sockets/webrtc_host.c
new file mode 100644
index 00000000..410184ad
--- /dev/null
+++ b/tests/sockets/webrtc_host.c
@@ -0,0 +1,89 @@
+#include <errno.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/ioctl.h>
+#include <assert.h>
+#if EMSCRIPTEN
+#include <emscripten.h>
+#endif
+
+#define EXPECTED_BYTES 5
+#define BUFLEN 16
+
+int result = 0;
+int sock;
+char buf[BUFLEN];
+char expected[] = "emscripten";
+struct sockaddr_in si_host,
+                   si_peer;
+struct iovec iov[1];                   
+struct msghdr hdr;
+int done = 0;
+
+void iter(void* arg) {
+  int n;
+  n = recvmsg(sock, &hdr, 0);
+
+  if(0 < n) {
+    done = 1;
+    fprintf(stderr, "received %d bytes: %s", n, (char*)hdr.msg_iov[0].iov_base);
+
+    shutdown(sock, SHUT_RDWR);
+    close(sock);
+
+#if EMSCRIPTEN
+    int result = 1;
+    // REPORT_RESULT();
+    exit(EXIT_SUCCESS);
+    emscripten_cancel_main_loop();
+#endif
+  } else if(EWOULDBLOCK != errno) {
+    perror("recvmsg failed");
+    exit(EXIT_FAILURE);
+    emscripten_cancel_main_loop();
+  }
+}
+
+int main(void)
+{
+  memset(&si_host, 0, sizeof(struct sockaddr_in));
+  memset(&si_peer, 0, sizeof(struct sockaddr_in));
+
+  si_host.sin_family = AF_INET;
+  si_host.sin_port = htons(8991);
+  si_host.sin_addr.s_addr = htonl(INADDR_ANY);
+
+  if(-1 == (sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP))) {
+    perror("cannot create host socket");
+    exit(EXIT_FAILURE);
+  }
+
+  if(-1 == bind(sock, (struct sockaddr*)&si_host, sizeof(struct sockaddr))) {
+    perror("cannot bind host socket");
+    exit(EXIT_FAILURE);
+  }
+  
+  iov[0].iov_base = buf;
+  iov[0].iov_len = sizeof(buf);
+  
+  memset (&hdr, 0, sizeof (struct msghdr));
+
+  hdr.msg_name = &si_peer;
+  hdr.msg_namelen = sizeof(struct sockaddr_in);
+  hdr.msg_iov = iov;
+  hdr.msg_iovlen = 1;
+
+#if EMSCRIPTEN
+  emscripten_set_main_loop(iter, 0, 0);
+#else
+  while (!done) iter(NULL);
+#endif
+
+  return EXIT_SUCCESS;
+}
\ No newline at end of file
diff --git a/tests/sockets/webrtc_peer.c b/tests/sockets/webrtc_peer.c
new file mode 100644
index 00000000..d24979e7
--- /dev/null
+++ b/tests/sockets/webrtc_peer.c
@@ -0,0 +1,81 @@
+#include <errno.h>
+#include <sys/types.h>
+#include <sys/socket.h>
+#include <netinet/in.h>
+#include <arpa/inet.h>
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include <unistd.h>
+#include <sys/ioctl.h>
+#include <assert.h>
+#if EMSCRIPTEN
+#include <emscripten.h>
+#endif
+
+#define EXPECTED_BYTES 5
+#define BUFLEN 16
+#define HOST_ADDR "10.0.0.1"
+
+int result = 0;
+int sock;
+char buf[16] = "emscripten";
+struct sockaddr_in si_host;
+struct iovec iov[1];
+struct msghdr hdr;
+int done = 0;
+
+void iter(void* arg) {
+  int n;
+  n = sendmsg(sock, &hdr, 0);
+
+  if(0 < n) {
+    done = 1;
+    fprintf(stderr, "sent %d bytes: %s", n, (char*)hdr.msg_iov[0].iov_base);
+
+    shutdown(sock, SHUT_RDWR);
+    close(sock);
+
+    exit(EXIT_SUCCESS);
+    emscripten_cancel_main_loop();
+  } else if(EWOULDBLOCK != errno) {
+    perror("sendmsg failed");
+    exit(EXIT_FAILURE);
+    emscripten_cancel_main_loop();
+  }
+}
+
+int main(void)
+{  
+  memset(&si_host, 0, sizeof(struct sockaddr_in));
+
+  si_host.sin_family = AF_INET;
+  si_host.sin_port = htons(8991);
+  if(0 == inet_pton(AF_INET, HOST_ADDR, &si_host.sin_addr)) {
+    perror("inet_aton failed");
+    exit(EXIT_FAILURE);
+  }
+
+  if(-1 == (sock = socket(PF_INET, SOCK_DGRAM, IPPROTO_UDP))) {
+    perror("cannot create socket");
+    exit(EXIT_FAILURE);
+  }
+
+  iov[0].iov_base = buf;
+  iov[0].iov_len = sizeof(buf);
+  
+  memset (&hdr, 0, sizeof (struct msghdr));
+
+  hdr.msg_name = &si_host;
+  hdr.msg_namelen = sizeof(struct sockaddr_in);
+  hdr.msg_iov = iov;
+  hdr.msg_iovlen = 1;
+
+#if EMSCRIPTEN
+  emscripten_set_main_loop(iter, 0, 0);
+#else
+  while (!done) iter(NULL);
+#endif
+
+  return EXIT_SUCCESS;
+}
\ No newline at end of file
diff --git a/tests/test_sockets.py b/tests/test_sockets.py
index 03593674..70679b15 100644
--- a/tests/test_sockets.py
+++ b/tests/test_sockets.py
@@ -198,3 +198,69 @@ class sockets(BrowserCore):
   #   ]:
   #     with harness:
   #       self.btest(os.path.join('sockets', 'test_enet_client.c'), expected='0', args=['-DSOCKK=9011'] + enet)
+
+  def test_webrtc(self):
+    host_src = 'webrtc_host.c'
+    peer_src = 'webrtc_peer.c'
+
+    host_outfile = 'host.html'
+    peer_outfile = 'peer.html'
+
+    host_filepath = path_from_root('tests', 'sockets', host_src)      
+    temp_host_filepath = os.path.join(self.get_dir(), os.path.basename(host_src))
+    with open(host_filepath) as f: host_src = f.read()
+    with open(temp_host_filepath, 'w') as f: f.write(self.with_report_result(host_src))
+
+    peer_filepath = path_from_root('tests', 'sockets', peer_src)      
+    temp_peer_filepath = os.path.join(self.get_dir(), os.path.basename(peer_src))
+    with open(peer_filepath) as f: peer_src = f.read()
+    with open(temp_peer_filepath, 'w') as f: f.write(self.with_report_result(peer_src))
+
+    open(os.path.join(self.get_dir(), 'host_pre.js'), 'w').write('''
+      var Module = {
+        webrtc: {
+          broker: 'https://mdsw.ch:8080',
+          session: undefined,
+          onpeer: function(peer, route) {
+            window.open('http://localhost:8888/peer.html?' + route);
+            //iframe = document.createElement("IFRAME");
+            //iframe.setAttribute("src", "http://localhost:8888/peer.html?" + route);
+            //iframe.style.display = "none";
+            //document.body.appendChild(iframe);
+            peer.listen();
+          },
+          onconnect: function(peer) {
+          },
+          ondisconnect: function(peer) {
+          },
+          onerror: function(error) {
+            console.error(error);
+          }
+        }
+      };
+    ''')
+
+    open(os.path.join(self.get_dir(), 'peer_pre.js'), 'w').write('''
+      var Module = {
+        webrtc: {
+          broker: 'https://mdsw.ch:8080',
+          session: window.location.toString().split('?')[1],
+          onpeer: function(peer, route) {
+            peer.connect(Module['webrtc']['session']);
+          },
+          onconnect: function(peer) {
+          },
+          ondisconnect: function(peer) {
+          },
+          onerror: function(error) {
+            console.error(error);
+          }
+        }
+      };
+    ''')
+
+    Popen([PYTHON, EMCC, temp_host_filepath, '-o', host_outfile] + ['-s', 'GL_TESTING=1', '--pre-js', 'host_pre.js', '-s', 'SOCKET_WEBRTC=1', '-s', 'SOCKET_DEBUG=1']).communicate()
+    Popen([PYTHON, EMCC, temp_peer_filepath, '-o', peer_outfile] + ['-s', 'GL_TESTING=1', '--pre-js', 'peer_pre.js', '-s', 'SOCKET_WEBRTC=1', '-s', 'SOCKET_DEBUG=1']).communicate()
+
+    expected = '1'
+    self.run_browser(host_outfile, '.', ['/report_result?' + e for e in expected])
\ No newline at end of file