Files
OpenRA/OpenRA.Launcher.Gtk/bridge.c
2011-01-04 12:53:33 +13:00

795 lines
21 KiB
C

/*
* Copyright 2007-2010 The OpenRA Developers (see AUTHORS)
* This file is part of OpenRA, which is free software. It is made
* available to you under the terms of the GNU General Public License
* as published by the Free Software Foundation. For more information,
* see LICENSE.
*/
#include <stdlib.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
#include <webkit/webkit.h>
#include <JavaScriptCore/JavaScript.h>
#include <glib.h>
#include "main.h"
#include "utility.h"
#define JS_STR(str) JSStringCreateWithUTF8CString(str)
#define JS_FUNC(ctx, callback) JSObjectMakeFunctionWithCallback(ctx, NULL, \
callback)
#define JS_TRUE JSValueMakeBoolean(ctx, TRUE)
#define JS_FALSE JSValueMakeBoolean(ctx, FALSE)
#define JS_NULL JSValueMakeNull(ctx)
GString * sanitize_path(gchar const * path)
{
gchar * basename = g_path_get_basename(path);
gchar * dirname = g_path_get_dirname(path);
gchar ** frags = g_strsplit(dirname, "/", -1);
gint offset = 0;
GString * new_path = g_string_new(NULL);
while (*(frags + offset))
{
if ((strcmp(*(frags + offset), "..") == 0) || (strcmp(*(frags + offset), ".") == 0))
{
offset++;
continue;
}
g_string_append(new_path, *(frags + offset));
g_string_append_c(new_path, G_DIR_SEPARATOR);
offset++;
}
g_string_append(new_path, basename);
g_free(basename);
g_free(dirname);
g_strfreev(frags);
return new_path;
}
int js_check_num_args(JSContextRef ctx, gchar const * func_name, int argc, int num_expected, JSValueRef * exception)
{
GString * buf;
if (argc < num_expected)
{
buf = g_string_new(NULL);
g_string_printf(buf, "%s: Not enough args, expected %d got %d", func_name, num_expected, argc);
*exception = JSValueMakeString(ctx, JS_STR(buf->str));
g_string_free(buf, TRUE);
return 0;
}
return 1;
}
GString * js_get_cstr_from_val(JSContextRef ctx, JSValueRef val)
{
gchar * buf;
GString * ret;
size_t len;
JSStringRef str;
if (!JSValueIsString(ctx, val))
return NULL;
str = JSValueToStringCopy(ctx, val, NULL);
len = JSStringGetMaximumUTF8CStringSize(str);
buf = (gchar *)g_malloc(len);
ret = g_string_sized_new(JSStringGetUTF8CString(str, buf, len));
g_string_assign(ret, buf);
g_free(buf);
return ret;
}
JSValueRef js_log(JSContextRef ctx, JSObjectRef func, JSObjectRef this,
size_t argc, const JSValueRef argv[], JSValueRef * exception)
{
if (!js_check_num_args(ctx, "log", argc, 1, exception))
return JS_NULL;
GString * buffer;
buffer = js_get_cstr_from_val(ctx, argv[0]);
if (!buffer)
return JS_NULL;
g_message("JS Log: %s", buffer->str);
g_string_free(buffer, TRUE);
return JS_NULL;
}
JSValueRef js_exists_in_mod(JSContextRef ctx, JSObjectRef func,
JSObjectRef this, size_t argc,
const JSValueRef argv[], JSValueRef * exception)
{
GString * mod_buf, * file_buf, * search_path, * search_path_sanitized;
JSValueRef return_value = JS_FALSE;
FILE * f;
if (!js_check_num_args(ctx, "existsInMod", argc, 2, exception))
return JS_NULL;
file_buf = js_get_cstr_from_val(ctx, argv[0]);
if (!file_buf)
return JS_NULL;
mod_buf = js_get_cstr_from_val(ctx, argv[1]);
if (!mod_buf)
return JS_NULL;
search_path = g_string_new(NULL);
g_string_printf(search_path, "mods/%s/%s", mod_buf->str, file_buf->str);
search_path_sanitized = sanitize_path(search_path->str);
g_string_free(search_path, TRUE);
g_string_free(file_buf, TRUE);
g_string_free(mod_buf, TRUE);
g_message("JS ExistsInMod: Looking for %s", search_path_sanitized->str);
f = fopen(search_path_sanitized->str, "r");
g_string_free(search_path_sanitized, TRUE);
if (f != NULL)
{
g_message("JS ExistsInMod: Found");
fclose(f);
return_value = JS_TRUE;
}
else
g_message("JS ExistsInMod: Not found");
return return_value;
}
JSValueRef js_launch_mod(JSContextRef ctx, JSObjectRef func,
JSObjectRef this, size_t argc,
const JSValueRef argv[], JSValueRef * exception)
{
GString * mod_key, * mod_list;
mod_t * mod;
if (!js_check_num_args(ctx, "launchMod", argc, 1, exception))
return JS_NULL;
if (!JSValueIsString(ctx, argv[0]))
{
*exception = JSValueMakeString(ctx, JS_STR("One or more args are incorrect types."));
return JS_NULL;
}
mod_key = js_get_cstr_from_val(ctx, argv[0]);
g_message("JS LaunchMod: %s", mod_key->str);
mod = get_mod(mod_key->str);
mod_list = g_string_new(mod_key->str);
g_string_free(mod_key, TRUE);
while (strlen(mod->requires) > 0)
{
gchar * r = g_strdup(mod->requires), * comma;
if (NULL != (comma = g_strstr_len(r, -1, ",")))
{
*comma = '\0';
}
mod = get_mod(r);
if (mod == NULL)
{
GString * exception_msg = g_string_new(NULL);
g_string_printf(exception_msg, "The mod %s is missing, cannot launch.", r);
*exception = JSValueMakeString(ctx, JS_STR(exception_msg->str));
g_string_free(exception_msg, TRUE);
g_string_free(mod_list, TRUE);
return JS_NULL;
}
g_string_append_printf(mod_list, ",%s", r);
g_free(r);
}
{
gchar * launch_args[] = { "mono", "OpenRA.Game.exe", NULL, NULL, "SupportDir=~/.openra", NULL };
GString * game_mods_arg = g_string_new(NULL);
GString * renderer_arg = g_string_new(NULL);
g_string_printf(game_mods_arg, "Game.Mods=%s", mod_list->str);
g_string_printf(renderer_arg, "Graphics.Renderer=%s", get_renderer() == RENDERER_GL ? "Gl" : "Cg");
launch_args[2] = game_mods_arg->str;
launch_args[3] = renderer_arg->str;
g_spawn_async(NULL, launch_args, NULL, G_SPAWN_SEARCH_PATH, NULL, NULL, NULL, NULL);
g_string_free(game_mods_arg, TRUE);
g_string_free(renderer_arg, TRUE);
}
g_string_free(mod_list, TRUE);
return JS_NULL;
}
typedef struct js_callback
{
JSObjectRef func;
JSContextGroupRef ctx_group;
} js_callback;
typedef struct download_t
{
gchar * key;
gchar * url;
gchar * dest;
int current_bytes;
int total_bytes;
js_callback * download_progressed_cb;
js_callback * extraction_progressed_cb;
JSValueRef status;
JSValueRef error;
GIOChannel * output_channel;
GPid pid;
} download_t;
#define MAX_DOWNLOADS 16
static download_t downloads[MAX_DOWNLOADS];
static int num_downloads = 0;
download_t * find_download(gchar const * key)
{
int i;
for (i = 0; i < num_downloads; i++)
{
if (0 == strcmp(downloads[i].key, key))
return downloads + i;
}
return NULL;
}
void set_download_status(JSContextRef ctx, download_t * download, const char * status)
{
JSValueUnprotect(ctx, download->status);
download->status = JSValueMakeString(ctx, JS_STR(status));
JSValueProtect(ctx, download->status);
}
void set_download_error(JSContextRef ctx, download_t * download, const char * error)
{
JSValueUnprotect(ctx, download->error);
download->error = JSValueMakeString(ctx, JS_STR(error));
JSValueProtect(ctx, download->error);
}
void free_download(download_t * download)
{
g_free(download->key);
g_free(download->url);
g_free(download->dest);
g_free(download->download_progressed_cb);
g_free(download->extraction_progressed_cb);
}
JSValueRef js_register_download(JSContextRef ctx, JSObjectRef func, JSObjectRef this,
size_t argc, const JSValueRef argv[], JSValueRef * exception)
{
GString * key, * url, * filename;
download_t * download;
JSValueRef o;
FILE * f;
if (!js_check_num_args(ctx, "registerDownload", argc, 3, exception))
return JS_NULL;
key = js_get_cstr_from_val(ctx, argv[0]);
g_message("JS RegisterDownload: Registering %s", key->str);
if (NULL == (download = find_download(key->str)))
{
download = downloads + num_downloads++;
if (num_downloads >= MAX_DOWNLOADS)
{
num_downloads = MAX_DOWNLOADS - 1;
return JS_NULL;
}
}
free_download(download);
memset(download, 0, sizeof(download_t));
download->download_progressed_cb = (js_callback *)g_malloc(sizeof(js_callback));
o = JSObjectGetProperty(ctx, JSContextGetGlobalObject(ctx), JS_STR("downloadProgressed"), NULL);
download->download_progressed_cb->ctx_group = JSContextGetGroup(ctx);
download->download_progressed_cb->func = JSValueToObject(ctx, o, NULL);
download->extraction_progressed_cb = (js_callback *)g_malloc(sizeof(js_callback));
o = JSObjectGetProperty(ctx, JSContextGetGlobalObject(ctx), JS_STR("extractProgressed"), NULL);
download->extraction_progressed_cb->ctx_group = JSContextGetGroup(ctx);
download->extraction_progressed_cb->func = JSValueToObject(ctx, o, NULL);
download->key = g_strdup(key->str);
g_string_free(key, TRUE);
url = js_get_cstr_from_val(ctx, argv[1]);
download->url = g_strdup(url->str);
g_string_free(url, TRUE);
filename = js_get_cstr_from_val(ctx, argv[2]);
{
GString * path = g_string_new(NULL), * sanitized_path;
g_string_printf(path, "/tmp/%s", filename->str);
g_string_free(filename, TRUE);
sanitized_path = sanitize_path(path->str);
g_string_free(path, TRUE);
download->dest = g_strdup(sanitized_path->str);
g_string_free(sanitized_path, TRUE);
}
f = fopen(download->dest, "r");
if (NULL != f)
{
fclose(f);
set_download_status(ctx, download, "DOWNLOADED");
}
else
set_download_status(ctx, download, "AVAILABLE");
return JS_NULL;
}
gboolean update_download_stats(GIOChannel * source, GIOCondition condition, gpointer data)
{
int ret = TRUE;
download_t * download = (download_t *)data;
gchar * line;
gsize line_length;
GIOStatus io_status;
JSValueRef args[1];
JSContextRef ctx;
ctx = JSGlobalContextCreateInGroup(download->download_progressed_cb->ctx_group, NULL);
switch(condition)
{
case G_IO_IN:
io_status = g_io_channel_read_line(source, &line, &line_length, NULL, NULL);
if (G_IO_STATUS_NORMAL == io_status)
{
if (g_str_has_prefix(line, "Error:"))
{
set_download_status(ctx, download, "ERROR");
set_download_error(ctx, download, line + 7);
}
else
{
set_download_status(ctx, download, "DOWNLOADING");
GRegex * pattern = g_regex_new("(\\d{1,3})% (\\d+)/(\\d+) bytes", 0, 0, NULL);
GMatchInfo * match;
if (g_regex_match(pattern, line, 0, &match))
{
gchar * current = g_match_info_fetch(match, 2), * total = g_match_info_fetch(match, 3);
download->current_bytes = atoi(current);
download->total_bytes = atoi(total);
g_free(current);
g_free(total);
}
g_free(match);
}
}
g_free(line);
break;
case G_IO_HUP:
if (!JSStringIsEqualToUTF8CString(JSValueToStringCopy(ctx, download->status, NULL), "ERROR"))
set_download_status(ctx, download, "DOWNLOADED");
g_io_channel_shutdown(source, FALSE, NULL);
ret = FALSE;
break;
default:
break;
}
args[0] = JSValueMakeString(ctx, JS_STR(download->key));
JSObjectCallAsFunction(ctx, download->download_progressed_cb->func, NULL, 1, args, NULL);
return ret;
}
JSValueRef js_start_download(JSContextRef ctx, JSObjectRef func, JSObjectRef this,
size_t argc, const JSValueRef argv[], JSValueRef * exception)
{
GString * key;
download_t * download;
int fd;
GPid pid;
if (!js_check_num_args(ctx, "startDownload", argc, 1, exception))
return JS_NULL;
key = js_get_cstr_from_val(ctx, argv[0]);
if (NULL == (download = find_download(key->str)))
{
g_string_free(key, TRUE);
return JS_FALSE;
}
g_string_free(key, TRUE);
g_message("Starting download %s", download->key);
set_download_status(ctx, download, "DOWNLOADING");
fd = util_do_download(download->url, download->dest, &pid);
if (!fd)
return JS_FALSE;
download->pid = pid;
download->output_channel = g_io_channel_unix_new(fd);
g_io_add_watch(download->output_channel, G_IO_IN | G_IO_HUP, update_download_stats, download);
return JS_TRUE;
}
JSValueRef js_cancel_download(JSContextRef ctx, JSObjectRef func, JSObjectRef this,
size_t argc, const JSValueRef argv[], JSValueRef * exception)
{
GString * key;
download_t * download;
if (!js_check_num_args(ctx, "cancelDownload", argc, 1, exception))
return JS_NULL;
key = js_get_cstr_from_val(ctx, argv[0]);
if (NULL == (download = find_download(key->str)))
{
g_string_free(key, TRUE);
return JS_FALSE;
}
if (download->pid)
{
set_download_status(ctx, download, "ERROR");
set_download_error(ctx, download, "Download Cancelled");
kill(download->pid, SIGTERM);
remove(download->dest);
}
g_string_free(key, TRUE);
return JS_TRUE;
}
JSValueRef js_download_status(JSContextRef ctx, JSObjectRef func, JSObjectRef this,
size_t argc, const JSValueRef argv[], JSValueRef * exception)
{
GString * key;
download_t * download;
if (!js_check_num_args(ctx, "downloadStatus", argc, 1, exception))
return JS_NULL;
key = js_get_cstr_from_val(ctx, argv[0]);
if (NULL == (download = find_download(key->str)))
{
g_string_free(key, TRUE);
return JSValueMakeString(ctx, JS_STR("NOT_REGISTERED"));
}
g_string_free(key, TRUE);
return download->status;
}
JSValueRef js_download_error(JSContextRef ctx, JSObjectRef func, JSObjectRef this,
size_t argc, const JSValueRef argv[], JSValueRef * exception)
{
GString * key;
download_t * download;
if (!js_check_num_args(ctx, "downloadError", argc, 1, exception))
return JS_NULL;
key = js_get_cstr_from_val(ctx, argv[0]);
g_message("JS DownloadError: Retrieving error message for %s", key->str);
if (NULL == (download = find_download(key->str)))
{
g_string_free(key, TRUE);
return JSValueMakeString(ctx, JS_STR(""));
}
g_string_free(key, TRUE);
return download->error;
}
JSValueRef js_bytes_completed(JSContextRef ctx, JSObjectRef func, JSObjectRef this,
size_t argc, const JSValueRef argv[], JSValueRef * exception)
{
GString * key;
download_t * download;
if (!js_check_num_args(ctx, "bytesCompleted", argc, 1, exception))
return JSValueMakeNull(ctx);
key = js_get_cstr_from_val(ctx, argv[0]);
if (NULL == (download = find_download(key->str)))
{
g_string_free(key, TRUE);
return JSValueMakeNumber(ctx, -1);
}
g_string_free(key, TRUE);
return JSValueMakeNumber(ctx, download->current_bytes);
}
JSValueRef js_bytes_total(JSContextRef ctx, JSObjectRef func, JSObjectRef this,
size_t argc, const JSValueRef argv[], JSValueRef * exception)
{
GString * key;
download_t * download;
if (!js_check_num_args(ctx, "bytesTotal", argc, 1, exception))
return JS_NULL;
key = js_get_cstr_from_val(ctx, argv[0]);
if (NULL == (download = find_download(key->str)))
{
g_string_free(key, TRUE);
return JSValueMakeNumber(ctx, -1);
}
g_string_free(key, TRUE);
return JSValueMakeNumber(ctx, download->total_bytes);
}
gboolean update_extraction_progress(GIOChannel * source, GIOCondition condition, gpointer data)
{
int ret = TRUE;
download_t * download = (download_t *)data;
gchar * line;
gsize line_length;
GIOStatus io_status;
JSValueRef args[1];
JSContextRef ctx;
ctx = JSGlobalContextCreateInGroup(download->extraction_progressed_cb->ctx_group, NULL);
switch(condition)
{
case G_IO_IN:
io_status = g_io_channel_read_line(source, &line, &line_length, NULL, NULL);
if ((G_IO_STATUS_NORMAL == io_status) && (g_str_has_prefix(line, "Error:")))
{
set_download_status(ctx, download, "ERROR");
set_download_error(ctx, download, line + 7);
}
free(line);
break;
case G_IO_HUP:
if (!JSStringIsEqualToUTF8CString(JSValueToStringCopy(ctx, download->status, NULL), "ERROR"))
set_download_status(ctx, download, "EXTRACTED");
g_io_channel_shutdown(source, FALSE, NULL);
ret = FALSE;
break;
default:
break;
}
args[0] = JSValueMakeString(ctx, JS_STR(download->key));
JSObjectCallAsFunction(ctx, download->extraction_progressed_cb->func, NULL, 1, args, NULL);
return ret;
}
JSValueRef js_extract_download(JSContextRef ctx, JSObjectRef func, JSObjectRef this,
size_t argc, const JSValueRef argv[], JSValueRef * exception)
{
GString * key, * dir, * mod, * status, * dest_path, * sanitized_dest_path;
download_t * download;
int fd;
GPid pid;
if (!js_check_num_args(ctx, "extractDownload", argc, 3, exception))
return JS_NULL;
key = js_get_cstr_from_val(ctx, argv[0]);
if (NULL == (download = find_download(key->str)))
{
g_string_free(key, TRUE);
return JS_FALSE;
}
g_string_free(key, TRUE);
status = js_get_cstr_from_val(ctx, download->status);
if (0 != strcmp(status->str, "DOWNLOADED"))
{
g_string_free(status, TRUE);
return JSValueMakeBoolean(ctx, 0);
}
g_string_free(status, TRUE);
set_download_status(ctx, download, "EXTRACTING");
dir = js_get_cstr_from_val(ctx, argv[1]);
mod = js_get_cstr_from_val(ctx, argv[2]);
dest_path = g_string_new(NULL);
g_string_printf(dest_path, "%s/%s", mod->str, dir->str);
sanitized_dest_path = sanitize_path(dest_path->str);
g_string_free(dest_path, TRUE);
g_string_free(mod, TRUE);
g_string_free(dir, TRUE);
fd = util_do_extract(download->dest, sanitized_dest_path->str, &pid);
g_string_free(sanitized_dest_path, TRUE);
if (!fd)
return JS_FALSE;
download->pid = pid;
download->output_channel = g_io_channel_unix_new(fd);
g_io_add_watch(download->output_channel, G_IO_IN | G_IO_HUP, update_extraction_progress, download);
return JS_TRUE;
}
JSValueRef js_metadata(JSContextRef ctx, JSObjectRef func, JSObjectRef this,
size_t argc, const JSValueRef argv[], JSValueRef * exception)
{
GString * field, * mod;
if (!js_check_num_args(ctx, "metadata", argc, 2, exception))
return JS_NULL;
field = js_get_cstr_from_val(ctx, argv[0]);
if (!field)
return JS_NULL;
mod = js_get_cstr_from_val(ctx, argv[1]);
if (!mod)
{
g_string_free(field, TRUE);
return JS_NULL;
}
if (0 == strcmp(field->str, "VERSION"))
{
mod_t * m = get_mod(mod->str);
if (m)
{
g_string_free(mod, TRUE);
g_string_free(field, TRUE);
return JSValueMakeString(ctx, JS_STR(m->version));
}
}
g_string_free(mod, TRUE);
g_string_free(field, TRUE);
return JS_NULL;
}
void request_finished(GPid pid, gint status, gpointer data)
{
GString * msg;
JSValueRef args[1];
callback_data * d = (callback_data *)data;
js_callback * cb = (js_callback *)d->user_data;
JSContextRef ctx = JSGlobalContextCreateInGroup(cb->ctx_group, NULL);
msg = util_get_output(d->output_fd);
close(d->output_fd);
if (!msg)
{
g_free(cb);
g_free(d);
return;
}
args[0] = JSValueMakeString(ctx, JS_STR(msg->str));
g_message("Request response: %s", msg->str);
JSObjectCallAsFunction(ctx, cb->func, NULL, 1, args, NULL);
g_string_free(msg, TRUE);
g_free(cb);
g_free(d);
g_spawn_close_pid(pid);
}
JSValueRef js_http_request(JSContextRef ctx, JSObjectRef func, JSObjectRef this,
size_t argc, const JSValueRef argv[], JSValueRef * exception)
{
GString * url, * callbackname;
js_callback * cb = (js_callback *)g_malloc(sizeof(js_callback));
JSValueRef o;
if (!js_check_num_args(ctx, "httpRequest", argc, 2, exception))
return JS_NULL;
url = js_get_cstr_from_val(ctx, argv[0]);
if (!url)
return JS_NULL;
callbackname = js_get_cstr_from_val(ctx, argv[1]);
if (!callbackname)
{
g_string_free(url, TRUE);
return JS_NULL;
}
o = JSObjectGetProperty(ctx, JSContextGetGlobalObject(ctx), JS_STR(callbackname->str), NULL);
cb->func = JSValueToObject(ctx, o, NULL);
cb->ctx_group = JSContextGetGroup(ctx);
util_do_http_request(url->str, request_finished, cb);
g_string_free(url, TRUE);
g_string_free(callbackname, TRUE);
return JS_NULL;
}
void js_add_functions(JSGlobalContextRef ctx, JSObjectRef target, char ** names,
JSObjectCallAsFunctionCallback * callbacks, size_t count)
{
int i;
for (i = 0; i < count; i++)
{
JSObjectRef func = JS_FUNC(ctx, callbacks[i]);
JSObjectSetProperty(ctx, target, JS_STR(names[i]), func, kJSPropertyAttributeNone, NULL);
}
}
void bind_js_bridge(WebKitWebView * view, WebKitWebFrame * frame,
gpointer context, gpointer window_object,
gpointer user_data)
{
JSGlobalContextRef js_ctx;
JSObjectRef window_obj, external_obj;
int func_count = 13;
char * names[] = { "log", "existsInMod", "launchMod", "registerDownload",
"startDownload", "cancelDownload", "downloadStatus",
"downloadError", "bytesCompleted", "bytesTotal",
"extractDownload", "metadata", "httpRequest"};
JSObjectCallAsFunctionCallback callbacks[] = { js_log, js_exists_in_mod, js_launch_mod,
js_register_download, js_start_download,
js_cancel_download, js_download_status,
js_download_error, js_bytes_completed,
js_bytes_total, js_extract_download, js_metadata, js_http_request };
js_ctx = (JSGlobalContextRef)context;
external_obj = JSObjectMake(js_ctx, NULL, NULL);
window_obj = (JSObjectRef)window_object;
JSObjectSetProperty(js_ctx, window_obj, JS_STR("external"),
external_obj, 0, NULL);
js_add_functions(js_ctx, external_obj, names, callbacks, func_count);
}