ssl server support, websocket upgrades, and net changes

1 files changed, 225 insertions, 0 deletions

diff --git a/src/net/websocket.c b/src/net/websocket.c
new file mode 100644
index 0000000..af37f32
--- /dev/null
+++ b/src/net/websocket.c
@@ -0,0 +1,225 @@
+#include "websocket.h"
+#include "common.h"
+#include "util.h"
+#include "../lua.h"
+#include "../hash/sha01.h"
+#include "../encode/base64.h"
+#include <limits.h>
+#include <ctype.h>
+
+//why?????
+const char* magic = "258EAFA5-E914-47DA-95CA-C5AB0DC85B11";
+
+//calloc or zero {out}
+void hex_decode(char* in, char* out, uint64_t* outlen){
+  uint64_t len = 0;
+  for(int i = 0; in[i] != '\0'; i++, len++){
+    int value = 0;
+    char c = in[i];
+    if(c >= '0' && c <= '9')
+      value = c - '0';
+    else if(c >= 'A' && c <= 'F')
+      value = 10 + (c - 'A');
+    else if(c >= 'a' && c <= 'f')
+      value = 10 + (c - 'a');
+
+    out[(i/2)] += value << (((i + 1) % 2) * 4);
+    //out[i/2] += strtoul((char[]){in[i], in[i+1]}, 0, 16);
+  }
+
+  *outlen = len/2;
+}
+
+struct ws_frame_info ws_frame_decode(char* buffer){
+  struct ws_frame_info frame_info = {.fin = (buffer[0] >> 7) & 1, .rsv1 = (buffer[0] >> 6) & 1,
+    .rsv2 = (buffer[0] >> 5) & 1, .rsv3 = (buffer[0] >> 4) & 1, .opcode = buffer[0] & 0b1111,
+    .mask = (buffer[1] >> 7) & 1, .payload = buffer[1] & 0b1111111};
+  return frame_info;
+}
+
+void ws_frame_build(struct ws_frame_info frame, uint64_t extended, char* mkey, str* out){
+  uint8_t d = (frame.fin << 7) | (frame.rsv1 << 6) | (frame.rsv2 << 5) | (frame.rsv3 << 4) | frame.opcode;
+  str_pushc(out, d);
+  frame.payload = extended;
+  if(extended > 125) frame.payload = 126;
+  if(extended > UINT16_MAX) frame.payload = 127;
+  d = (frame.mask << 7) | frame.payload;
+  str_pushc(out, d);
+
+  if(frame.payload == 126){
+    uint16_t sh = extended;
+    for(int i = 2; i != 0; i--)
+      str_pushc(out, ((uint8_t*)&sh)[i - 1]);
+  } else if(frame.payload == 127){
+    for(int i = 8; i != 0; i--)
+      str_pushc(out, ((uint8_t*)&extended)[i - 1]);
+  }
+
+  if(frame.mask){
+    str_pushl(out, mkey, 4);
+  }
+}
+
+#define BUFFER_LEN 4096
+
+int ws_read(struct net_data* data, struct ws_frame_info* frame_info){
+  char buffer[BUFFER_LEN] = {0};
+  int total_len = 0;
+  int len;
+
+  for(; (len = net_ctx_read(data, buffer + total_len, 2 - total_len)) > 0;){ 
+    total_len += len;
+    if(total_len >= 2) break;
+  }
+
+  if(len < 0 || total_len <= 0) return -1;
+
+  uint64_t payload = 0;
+  *frame_info = ws_frame_decode(buffer);
+  memset(buffer, 0, total_len);
+  total_len = 0;
+
+  if(frame_info->payload <= 125) payload = frame_info->payload;
+  else if(frame_info->payload == 126) {
+    for(; (len = net_ctx_read(data, buffer + total_len, 2 - total_len)) > 0;){ 
+      total_len += len;
+      if(total_len >= 2) break;
+    }
+
+    if(len < 0) return -1;
+
+    payload = (buffer[0] & 0xff) << 8 | (buffer[1] & 0xff);
+  } else {
+    for(; (len = net_ctx_read(data, buffer + total_len, 8 - total_len)) > 0;){
+      total_len += len;
+      if(total_len >= 8) break;
+    }
+
+    if(len < 0) return -1;
+
+    payload = ((uint64_t)buffer[0] & 0xff) << 56 | ((uint64_t)buffer[1] & 0xff) << 48 | ((uint64_t)buffer[2] & 0xff) << 40 |
+      ((uint64_t)buffer[3] & 0xff) << 32 | (buffer[4] & 0xff) << 24 | (buffer[5] & 0xff) << 16 | (buffer[6] & 0xff) << 8 | (buffer[7] & 0xff);
+  }
+
+  total_len = 0;
+  uint8_t mask[4] = {0};
+  if(frame_info->mask){
+    for(; (len = net_ctx_read(data, buffer + total_len, 4 - total_len)) > 0;){
+      total_len += len;
+      if(total_len >= 4) break;
+    }
+    mask[0] = buffer[0];
+    mask[1] = buffer[1];
+    mask[2] = buffer[2];
+    mask[3] = buffer[3];
+  }
+
+  uint64_t i = 0;
+  memset(buffer, 0, BUFFER_LEN);
+  for(; data->buffer->len != payload && (len = net_ctx_read(data, buffer, lesser(payload - data->buffer->len, BUFFER_LEN))) > 0;){ 
+    if(frame_info->mask){
+      for(int z = 0; z != len; z++,i++)
+        buffer[z] ^= mask[i % 4];
+    }
+    str_pushl(data->buffer, buffer, len);
+    memset(buffer, 0, len);
+  }
+
+  if(len < 0) return -1;
+
+  return 1;
+}
+
+int l_ws_read(lua_State* L){
+  lua_getfield(L, 1, "_ws");
+  struct net_data* data = lua_touserdata(L, -1);
+  struct ws_frame_info frame = {};
+  int c = ws_read(data, &frame);
+  if(c == -1){
+    luaI_error(L, -1, "SSL_read error");
+    str_clear(data->buffer);
+  }
+
+  lua_newtable(L);
+  int idx = lua_gettop(L);
+  luaI_tsetsl(L, idx, "content", data->buffer->c, data->buffer->len);
+  luaI_tseti(L, idx, "opcode", frame.opcode);
+
+  str_clear(data->buffer);
+  return 1;
+}
+
+int l_ws_write(lua_State* L){
+  lua_getfield(L, 1, "_ws");
+  struct net_data* data = lua_touserdata(L, -1);
+  struct ws_frame_info frame = {};
+  frame.rsv1 = 0;
+  frame.fin = 1;
+  frame.mask = 0;
+  frame.opcode = 0b0001;
+
+  uint64_t len;
+  const char* s = lua_tolstring(L, 2, &len);
+
+  str* f = str_init("");
+  ws_frame_build(frame, len, NULL, f);
+  str_pushl(f, s, len);
+  write(data->sock, f->c, f->len);
+  str_free(f);
+
+  return 0;
+}
+
+int l_websocket_upgrade(lua_State* L){
+  int res_idx = 1;
+  int req_idx = 2;
+  
+  lua_getfield(L, res_idx, "_");
+  struct net_data* ctx = lua_touserdata(L, -1);
+
+  lua_getfield(L, req_idx, "sec-websocket-key");
+  const char* wskey = luaL_checklstring(L, -1, NULL);
+  str* newkey = str_init(wskey);
+  str_push(newkey, magic);
+  char* sha = calloc(1512, sizeof * sha);
+  printf("%s\n", newkey->c);
+  sha1((uint8_t*)newkey->c, newkey->len, sha);
+  printf("%s\n", sha);
+  char* bin = calloc(512, sizeof * bin);
+  uint64_t len;
+  hex_decode(sha, bin, &len);
+  char* b64 = calloc(512 * 3, sizeof * b64);
+  en_base64(bin, len, b64);
+
+  char* upgrade = calloc(8192, sizeof * upgrade);
+  sprintf(upgrade, "HTTP/1.1 101 Switching Protocols\r\n"
+                   "upgrade: websocket\r\n"
+                   "connection: upgrade\r\n"
+                   "sec-websocket-accept: %s\r\n"
+                   "\r\n", b64);
+
+  printf("%s\n", upgrade);
+  net_ctx_write(ctx, upgrade, strlen(upgrade));
+  free(upgrade);
+  free(sha);
+  free(bin);
+  free(b64);
+  str_free(newkey);
+
+  struct net_data *data = calloc(1, sizeof * data);
+  data->sock = ctx->sock;
+  data->ssl = ctx->ssl;
+  data->ctx = ctx->ctx;
+  data->buffer = str_init("");
+
+  luaI_tsetnil(L, req_idx, "roll");
+  luaI_tsetlud(L, res_idx, "_ws", data);
+#warning "missing ws commands"
+  luaI_tsetcf(L, res_idx, "send", l_ws_write);
+  //luaI_tsetcf(L, res_idx, "write", );
+  //luaI_tsetcf(L, res_idx, "sendfile", );
+  luaI_tsetcf(L, res_idx, "read", l_ws_read);
+  //luaI_tsetcf(L, res_idx, "close", );
+
+  return 1;
+}