add basic websocket support

client/websocket.c

@@ -0,0 +1,57 @@
+#include <stdlib.h>
+
+#include "curl/curl.h"
+#include "curl/websockets.h"
+
+#include "types.h"
+
+extern CURLM* multi_handle;
+
+struct WSResult* recv_from_websocket(CURL* http_handle, int buffer_size) {
+  const struct curl_ws_frame* ws_meta;
+  char* buffer = malloc(buffer_size);
+  size_t result_len;
+  CURLcode res = curl_ws_recv(http_handle, buffer, buffer_size, &result_len, &ws_meta);
+
+  struct WSResult* result = malloc(sizeof(struct WSResult));
+  result->buffer_size = result_len;
+  result->buffer = buffer;
+  result->res = (int) res;
+  result->closed = (ws_meta->flags & CURLWS_CLOSE);
+
+  return result;
+}
+
+int send_to_websocket(CURL* http_handle, const char* data, int data_len) {
+  size_t sent;
+  CURLcode res = curl_ws_send(http_handle, data, data_len, &sent, 0, CURLWS_BINARY);
+  return (int) res;
+}
+
+void close_websocket(CURL* http_handle) {
+  size_t sent;
+  curl_ws_send(http_handle, "", 0, &sent, 0, CURLWS_CLOSE);
+}
+
+//allow the main code to automatically clean up this websocket
+void cleanup_websocket(CURL* http_handle) {
+  struct RequestInfo *request_info;
+  curl_easy_getinfo(http_handle, CURLINFO_PRIVATE, &request_info);
+
+  curl_multi_remove_handle(multi_handle, http_handle);
+  curl_easy_cleanup(http_handle);
+  free(request_info);
+}
+
+int get_result_size (const struct WSResult* result) {
+  return result->buffer_size;
+}
+char* get_result_buffer (const struct WSResult* result) {
+  return result->buffer;
+}
+int get_result_code (const struct WSResult* result) {
+  return result->res;
+}
+int get_result_closed (const struct WSResult* result) {
+  return result->closed;
+}