gambas-source-code/main/gbx/gbx_jit.c

/***************************************************************************

  gbx_jit.c

  (c) 2018 Benoît Minisini <g4mba5@gmail.com>

  This program is free software; you can redistribute it and/or modify
  it under the terms of the GNU General Public License as published by
  the Free Software Foundation; either version 2, or (at your option)
  any later version.

  This program is distributed in the hope that it will be useful,
  but WITHOUT ANY WARRANTY; without even the implied warranty of
  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  GNU General Public License for more details.

  You should have received a copy of the GNU General Public License
  along with this program; if not, write to the Free Software
  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
  MA 02110-1301, USA.

***************************************************************************/

#define __GBX_JIT_C

#include "gb_common.h"
#include "gb_common_buffer.h"
#include "gb_common_case.h"
#include "gbx_component.h"
#include "gbx_api.h"
#include "gbx_jit.h"

static bool _component_loaded = FALSE;
static GB_FUNCTION _jit_func;

static bool _no_jit = FALSE;
static void *_jit_library = NULL;

bool JIT_compile(ARCHIVE *arch)
{
	GB_VALUE *ret;
	char *path;
	void *lib;
	void **iface;
	
	if (_no_jit)
		return TRUE;
	
	if (arch)
	{
		if (arch->jit_library)
			return FALSE;
	}
	else
	{
		if (_jit_library)
			return FALSE;
	}
	
	if (!_component_loaded)
	{
		char *var = getenv("GB_NO_JIT");
		
		_component_loaded = TRUE;
		
		if (var && var[0] && !(var[0] == '0' && var[1] == 0))
		{
			_no_jit = TRUE;
			return TRUE;
		}
		
		fprintf(stderr, "gbx3: loading gb.jit component\n");
		COMPONENT_load(COMPONENT_create("gb.jit"));
		if (GB_GetFunction(&_jit_func, CLASS_find_global("_Jit"), "Compile", "", "s"))
			ERROR_panic("Unable to find _Jit.Compile method");
	}
	
	ret = GB_Call(&_jit_func, 0, FALSE);
	path = GB_ToZeroString((GB_STRING *)ret);
	if (!*path)
		ERROR_panic("Unable to compile jit source file");
	
	fprintf(stderr, "gbx3: shared jit library is: %s\n", path);

	lib = dlopen(path, RTLD_NOW);
	if (!lib)
		ERROR_panic("Unable to load jit library: %s", dlerror());
	
	if (arch)
		arch->jit_library = lib;
	else
		_jit_library = lib;
	
  iface = dlsym(lib, "JIT_PTR");
	if (iface)
		*((void **)iface) = &GAMBAS_JitApi;
	
	return FALSE;
}

void *JIT_get_function(ARCHIVE *arch, CLASS *class, int index)
{
	void *lib;
	void *addr;
	int i;
	int len;
	
	if (!arch)
		lib = _jit_library;
	else
		lib = arch->jit_library;
	
	len = sprintf(COMMON_buffer, "%s_%d", class->name, index);
	
	for (i = 0; i < len; i++)
		COMMON_buffer[i] = tolower(COMMON_buffer[i]);
	
	addr = dlsym(lib, COMMON_buffer);
	/*if (!addr)
		ERROR_panic("Unable to find jit function %s", COMMON_buffer);*/
	return addr;
}

void JIT_hello(void)
{
	fprintf(stderr, "Hello JIT!\n");
}