summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--json.c23
-rw-r--r--lsp.c192
-rw-r--r--main.c15
3 files changed, 183 insertions, 47 deletions
diff --git a/json.c b/json.c
index f514b4f..e63bc64 100644
--- a/json.c
+++ b/json.c
@@ -50,6 +50,7 @@ struct JSONValue {
typedef struct {
char error[64];
+ bool is_text_copied; // if this is true, then json_free will call free on text
const char *text;
// root = values[0]
JSONValue *values;
@@ -362,11 +363,13 @@ static bool json_parse(JSON *json, const char *text) {
arr_addp(json->values); // add root
JSONValue val = {0};
u32 index = 0;
- if (!json_parse_value(json, &index, &val))
+ if (!json_parse_value(json, &index, &val)) {
+ arr_free(json->values);
return false;
-
+ }
SKIP_WHITESPACE;
if (text[index]) {
+ arr_free(json->values);
strbuf_printf(json->error, "extra text after end of root object");
return false;
}
@@ -374,9 +377,25 @@ static bool json_parse(JSON *json, const char *text) {
return true;
}
+// like json_parse, but a copy of text is made, so you can free/overwrite it immediately.
+static bool json_parse_copy(JSON *json, const char *text) {
+ bool success = json_parse(json, str_dup(text));
+ if (success) {
+ json->is_text_copied = true;
+ return true;
+ } else {
+ free((void*)json->text);
+ json->text = NULL;
+ return false;
+ }
+}
+
static void json_free(JSON *json) {
arr_free(json->values);
memset(json, 0, sizeof *json);
+ if (json->is_text_copied) {
+ free((void*)json->text);
+ }
}
static bool json_streq(const JSON *json, const JSONString *string, const char *name) {
diff --git a/lsp.c b/lsp.c
index a204dec..afe8831 100644
--- a/lsp.c
+++ b/lsp.c
@@ -1,8 +1,8 @@
+// @TODO: maximum queue size for requests/responses just in case?
+
typedef enum {
LSP_INITIALIZE,
LSP_OPEN,
- LSP_CHANGE,
- LSP_CLOSE,
} LSPRequestType;
typedef struct {
@@ -22,11 +22,19 @@ typedef struct {
typedef struct {
Process process;
+ JSON *responses;
+ SDL_mutex *responses_mutex;
+ LSPRequest *requests;
+ SDL_mutex *requests_mutex;
+
+ SDL_Thread *communication_thread;
+ SDL_sem *quit_sem;
+ char *received_data; // dynamic array
char error[256];
} LSP;
-static void send_request_content(LSP *lsp, const char *content) {
+static void write_request_content(LSP *lsp, const char *content) {
char header[128];
size_t content_size = strlen(content);
strbuf_printf(header, "Content-Length: %zu\r\n\r\n", content_size);
@@ -66,7 +74,7 @@ static const char *lsp_language_id(Language lang) {
return "text";
}
-static void send_request(LSP *lsp, const LSPRequest *request) {
+static void write_request(LSP *lsp, const LSPRequest *request) {
static unsigned long long id;
++id;
@@ -78,7 +86,7 @@ static void send_request(LSP *lsp, const LSPRequest *request) {
"\"processId\":%d,"
"\"capabilities\":{}"
"}}", id, process_get_id());
- send_request_content(lsp, content);
+ write_request_content(lsp, content);
} break;
case LSP_OPEN: {
const LSPRequestOpen *open = &request->data.open;
@@ -104,7 +112,7 @@ static void send_request(LSP *lsp, const LSPRequest *request) {
free(did_open);
- send_request_content(lsp, did_open);
+ write_request_content(lsp, did_open);
} break;
default:
// @TODO
@@ -112,59 +120,157 @@ static void send_request(LSP *lsp, const LSPRequest *request) {
}
}
-static bool recv_response(LSP *lsp, JSON *json) {
- static char response_data[500000];
- // i think this should always read all the response data.
- long long bytes_read = process_read(&lsp->process, response_data, sizeof response_data);
- if (bytes_read < 0) {
- strbuf_printf(lsp->error, "Read error.");
- return false;
+// figure out if data begins with a complete LSP response.
+static bool has_response(const char *data, size_t data_len, u64 *p_offset, u64 *p_size) {
+ const char *content_length = strstr(data, "Content-Length");
+ if (!content_length) return false;
+ const char *p = content_length + strlen("Content-Length");
+ if (!p[0] || !p[1] || !p[2]) return false;
+ p += 2;
+ size_t size = (size_t)atoll(p);
+ *p_size = size;
+ const char *header_end = strstr(content_length, "\r\n\r\n");
+ if (!header_end) return false;
+ header_end += 4;
+ u64 offset = (u64)(header_end - data);
+ *p_offset = offset;
+ return offset + size <= data_len;
+}
+
+void lsp_send_request(LSP *lsp, const LSPRequest *request) {
+ SDL_LockMutex(lsp->requests_mutex);
+ arr_add(lsp->requests, *request);
+ SDL_UnlockMutex(lsp->requests_mutex);
+}
+
+// receive responses from LSP, up to max_size bytes.
+static void lsp_receive(LSP *lsp, size_t max_size) {
+ size_t received_so_far = arr_len(lsp->received_data);
+ arr_reserve(lsp->received_data, received_so_far + max_size + 1);
+ long long bytes_read = process_read(&lsp->process, lsp->received_data + received_so_far, max_size);
+ if (bytes_read <= 0) {
+ // no data
+ return;
}
- response_data[bytes_read] = '\0';
+ received_so_far += (size_t)bytes_read;
+ // kind of a hack. this is needed because arr_set_len zeroes the data.
+ arr_hdr_(lsp->received_data)->len = (u32)received_so_far;
+ lsp->received_data[received_so_far] = '\0';// null terminate
- char *body_start = strstr(response_data, "\r\n\r\n");
- if (body_start) {
- body_start += 4;
- if (!json_parse(json, body_start)) {
- strbuf_cpy(lsp->error, json->error);
- return false;
- }
- JSONValue result = json_get(json, "result");
+ u64 response_offset=0, response_size=0;
+ while (has_response(lsp->received_data, received_so_far, &response_offset, &response_size)) {
+ char *copy = strn_dup(lsp->received_data + response_offset, response_size);
+ JSON json = {0};
+ json_parse(&json, copy);
+ assert(json.text == copy);
+ json.is_text_copied = true;
+
+ JSONValue result = json_get(&json, "result");
if (result.type == JSON_UNDEFINED) {
// uh oh
- JSONValue error = json_get(json, "error.message");
+ JSONValue error = json_get(&json, "error.message");
if (error.type == JSON_STRING) {
- json_string_get(json, &error.val.string, lsp->error, sizeof lsp->error);
+ json_string_get(&json, &error.val.string, lsp->error, sizeof lsp->error);
} else {
strbuf_printf(lsp->error, "Server error (no message)");
}
+ } else {
+ SDL_LockMutex(lsp->responses_mutex);
+ arr_add(lsp->responses, json);
+ SDL_UnlockMutex(lsp->responses_mutex);
}
- } else {
- strbuf_printf(lsp->error, "No response body.");
- return false;
+
+ size_t leftover_data_len = arr_len(lsp->received_data) - (response_offset + response_size);
+ memmove(lsp->received_data, lsp->received_data + response_offset + response_size,
+ leftover_data_len);
+ arr_set_len(lsp->received_data, leftover_data_len);
+ arr_reserve(lsp->received_data, leftover_data_len + 1);
+ lsp->received_data[leftover_data_len] = '\0';
+ }
+}
+
+static void free_request(LSPRequest *r) {
+ switch (r->type) {
+ case LSP_INITIALIZE:
+ break;
+ case LSP_OPEN: {
+ LSPRequestOpen *open = &r->data.open;
+ free(open->filename);
+ free(open->file_contents);
+ } break;
}
-
- return true;
+}
+
+// Do any necessary communication with the LSP.
+// This writes requests and reads (and parses) responses.
+static int lsp_communication_thread(void *data) {
+ LSP *lsp = data;
+ while (1) {
+ LSPRequest *requests = NULL;
+ SDL_LockMutex(lsp->requests_mutex);
+ while (arr_len(lsp->requests)) {
+ arr_add(requests, lsp->requests[0]);
+ arr_remove(lsp->requests, 0);
+ }
+ SDL_UnlockMutex(lsp->requests_mutex);
+
+ bool quit = false;
+ arr_foreach_ptr(requests, LSPRequest, r) {
+ if (!quit) {
+ // this could slow down lsp_free if there's a gigantic request.
+ // whatever.
+ write_request(lsp, r);
+ }
+ free_request(r);
+
+ if (SDL_SemTryWait(lsp->quit_sem) == 0) {
+ quit = true;
+ // important that we don't break here so all the requests get freed.
+ }
+ }
+
+ arr_free(requests);
+
+ lsp_receive(lsp, (size_t)10<<20);
+ if (quit || SDL_SemWaitTimeout(lsp->quit_sem, 5) == 0)
+ break;
+ }
+ return 0;
}
bool lsp_create(LSP *lsp, const char *analyzer_command) {
ProcessSettings settings = {
.stdin_blocking = true,
- .stdout_blocking = true
+ .stdout_blocking = false
};
process_run_ex(&lsp->process, analyzer_command, &settings);
- char init_request[1024];
- strbuf_printf(init_request,
- "{\"jsonrpc\":\"2.0\",\"id\":1,\"method\":\"initialize\",\"params\":{"
- "\"processId\":%d,"
- "\"capabilities\":{}"
- "}}",
- process_get_id());
- send_request_content(lsp, init_request);
- JSON json = {0};
- if (!recv_response(lsp, &json)) {
- return false;
- }
- json_debug_print(&json);
+ LSPRequest initialize = {
+ .type = LSP_INITIALIZE
+ };
+ lsp_send_request(lsp, &initialize);
+
+ lsp->quit_sem = SDL_CreateSemaphore(0);
+ lsp->responses_mutex = SDL_CreateMutex();
+ lsp->communication_thread = SDL_CreateThread(lsp_communication_thread, "LSP communicate", lsp);
return true;
}
+
+bool lsp_next_response(LSP *lsp, JSON *json) {
+ bool any = false;
+ SDL_LockMutex(lsp->responses_mutex);
+ if (arr_len(lsp->responses)) {
+ *json = lsp->responses[0];
+ arr_remove(lsp->responses, 0);
+ any = true;
+ }
+ SDL_UnlockMutex(lsp->responses_mutex);
+ return any;
+}
+
+void lsp_free(LSP *lsp) {
+ SDL_SemPost(lsp->quit_sem);
+ SDL_WaitThread(lsp->communication_thread, NULL);
+ SDL_DestroySemaphore(lsp->quit_sem);
+ process_kill(&lsp->process);
+ arr_free(lsp->received_data);
+}
diff --git a/main.c b/main.c
index acdbea0..2792bd9 100644
--- a/main.c
+++ b/main.c
@@ -288,8 +288,19 @@ int main(int argc, char **argv) {
// @TODO TEMPORARY
{
LSP lsp={0};
- if (!lsp_create(&lsp, "rust-analyzer"))
- printf("%s\n",lsp.error);
+ if (!lsp_create(&lsp, "rust-analyzer")) {
+ printf("lsp_create: %s\n",lsp.error);
+ exit(1);
+ }
+ while (1) {
+ JSON response = {0};
+ if (lsp_next_response(&lsp, &response)) {
+ json_debug_print(&response);
+ break;
+ }
+ usleep(100000);
+ }
+ lsp_free(&lsp);
exit(0);
}