JIT support for FFI calls (incomplete)

Author: dstogov

ext/opcache/jit/zend_jit.c

@@ -67,6 +67,36 @@ static bool zend_jit_ffi_supported_type(zend_ffi_type *type)
 	return true;
 }
 
+static bool zend_jit_ffi_supported_func(zend_ffi_type *type)
+{
+	zend_ffi_type *t;
+
+	type = ZEND_FFI_TYPE(type);
+
+	if (type->func.abi != ZEND_FFI_ABI_CDECL
+	 && type->func.abi != ZEND_FFI_ABI_FASTCALL) {
+		return false;
+	}
+
+	t = ZEND_FFI_TYPE(type->func.ret_type);
+	if (t->kind == ZEND_FFI_TYPE_STRUCT
+	 || !zend_jit_ffi_supported_type(t)) {
+		return false;
+	}
+
+	if (type->func.args) {
+		ZEND_HASH_PACKED_FOREACH_PTR(type->func.args, t) {
+			t = ZEND_FFI_TYPE(t);
+			if (t->kind == ZEND_FFI_TYPE_STRUCT
+			 || !zend_jit_ffi_supported_type(t)) {
+				return false;
+			}
+		} ZEND_HASH_FOREACH_END();
+	}
+
+	return true;
+}
+
 static bool zend_jit_ffi_compatible(zend_ffi_type *dst_type, uint32_t src_info, zend_ffi_type *src_type)
 {
 	dst_type = ZEND_FFI_TYPE(dst_type);

ext/opcache/jit/zend_jit_helpers.c

@@ -3271,3 +3271,37 @@ static zend_string* ZEND_FASTCALL zend_jit_rope_end(zend_string **rope, uint32_t
 	*target = '\0';
 	return ret;
 }
+
+#ifdef HAVE_FFI
+static void ZEND_FASTCALL zend_jit_zval_string(zval *zv, const char *str)
+{
+	if (str) {
+		ZVAL_STRING(zv, str);
+	} else {
+		ZVAL_NULL(zv);
+	}
+}
+
+static void ZEND_FASTCALL zend_jit_zval_ffi_ptr(zval *zv, zend_ffi_type *type, void *ptr)
+{
+	ZEND_ASSERT(type->kind == ZEND_FFI_TYPE_POINTER);
+	if (ptr) {
+		zend_ffi_cdata *cdata = emalloc(sizeof(zend_ffi_cdata));
+
+		// inlined zend_ffi_object_init()
+		GC_SET_REFCOUNT(&cdata->std, 1);
+		GC_TYPE_INFO(&cdata->std) = GC_OBJECT | (IS_OBJ_DESTRUCTOR_CALLED << GC_FLAGS_SHIFT);
+		cdata->std.ce = zend_ffi_cdata_ce;
+		cdata->std.handlers = zend_ffi_cdata_ce->default_object_handlers; /* zend_ffi_cdata_handlers */
+		cdata->std.properties = NULL;
+		zend_objects_store_put(&cdata->std);
+		cdata->type = type;
+		cdata->flags = 0;
+		cdata->ptr = (void*)&cdata->ptr_holder;
+		cdata->ptr_holder = ptr;
+		ZVAL_OBJ(zv, &cdata->std);
+	} else {
+		ZVAL_NULL(zv);
+	}
+}
+#endif

ext/opcache/jit/zend_jit_internal.h

@@ -381,6 +381,12 @@ typedef enum _zend_jit_trace_op {
 #define ZEND_JIT_TRACE_FAKE_INFO(level) \
 	(((level) << ZEND_JIT_TRACE_FAKE_LEVEL_SHIFT) | ZEND_JIT_TRACE_FAKE_INIT_CALL)
 
+#define ZEND_JIT_TRACE_NUM_ARGS_INFO(count) \
+	((count) << ZEND_JIT_TRACE_FAKE_LEVEL_SHIFT)
+
+#define ZEND_JIT_TRACE_NUM_ARGS(info) \
+	(((info) & ZEND_JIT_TRACE_FAKE_LEVEL_MASK) >> ZEND_JIT_TRACE_FAKE_LEVEL_SHIFT)
+
 #define ZEND_JIT_TRACE_SET_FIRST_SSA_VAR(_info, var) do { \
 		_info |= (var << ZEND_JIT_TRACE_SSA_VAR_SHIFT); \
 	} while (0)
@@ -564,6 +570,8 @@ struct _zend_jit_trace_stack_frame {
 #define TRACE_FRAME_MASK_CLOSURE_CALL         0x00000200
 #define TRACE_FRAME_MASK_ALWAYS_RELEASE_THIS  0x00000400
 
+#define TRACE_FRAME_MASK_FFI                  0x00000800
+
 
 #define TRACE_FRAME_INIT(frame, _func, _flags, num_args) do { \
 		zend_jit_trace_stack_frame *_frame = (frame); \
@@ -602,6 +610,8 @@ struct _zend_jit_trace_stack_frame {
 	((frame)->_info & TRACE_FRAME_MASK_CLOSURE_CALL)
 #define TRACE_FRAME_ALWAYS_RELEASE_THIS(frame) \
 	((frame)->_info & TRACE_FRAME_MASK_ALWAYS_RELEASE_THIS)
+#define TRACE_FRAME_FFI(frame) \
+	((frame)->_info & TRACE_FRAME_MASK_FFI)
 
 #define TRACE_FRAME_SET_UNKNOWN_NUM_ARGS(frame) do { \
 		(frame)->_info |= (0xffffu << TRACE_FRAME_SHIFT_NUM_ARGS); \