1165 lines
38 KiB
C
1165 lines
38 KiB
C
/* GNU Objective C Runtime message lookup
|
|
Copyright (C) 1993-2016 Free Software Foundation, Inc.
|
|
Contributed by Kresten Krab Thorup
|
|
|
|
This file is part of GCC.
|
|
|
|
GCC 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 3, or (at your option) any later version.
|
|
|
|
GCC 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.
|
|
|
|
Under Section 7 of GPL version 3, you are granted additional
|
|
permissions described in the GCC Runtime Library Exception, version
|
|
3.1, as published by the Free Software Foundation.
|
|
|
|
You should have received a copy of the GNU General Public License and
|
|
a copy of the GCC Runtime Library Exception along with this program;
|
|
see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
|
|
<http://www.gnu.org/licenses/>. */
|
|
|
|
/* Uncommented the following line to enable debug logging. Use this
|
|
only while debugging the runtime. */
|
|
/* #define DEBUG 1 */
|
|
|
|
/* FIXME: This should be using libffi instead of __builtin_apply
|
|
and friends. */
|
|
|
|
#include "objc-private/common.h"
|
|
#include "objc-private/error.h"
|
|
#include "tconfig.h"
|
|
#include "coretypes.h"
|
|
#include "objc/runtime.h"
|
|
#include "objc/message.h" /* For objc_msg_lookup(), objc_msg_lookup_super(). */
|
|
#include "objc/thr.h"
|
|
#include "objc-private/module-abi-8.h"
|
|
#include "objc-private/runtime.h"
|
|
#include "objc-private/hash.h"
|
|
#include "objc-private/sarray.h"
|
|
#include "objc-private/selector.h" /* For sel_is_mapped() */
|
|
#include "runtime-info.h"
|
|
#include <assert.h> /* For assert */
|
|
#include <string.h> /* For strlen */
|
|
|
|
#define INVISIBLE_STRUCT_RETURN 1
|
|
|
|
/* The uninstalled dispatch table. If a class' dispatch table points
|
|
to __objc_uninstalled_dtable then that means it needs its dispatch
|
|
table to be installed. */
|
|
struct sarray *__objc_uninstalled_dtable = 0; /* !T:MUTEX */
|
|
|
|
/* Two hooks for method forwarding. If either is set, it is invoked to
|
|
* return a function that performs the real forwarding. If both are
|
|
* set, the result of __objc_msg_forward2 will be preferred over that
|
|
* of __objc_msg_forward. If both return NULL or are unset, the
|
|
* libgcc based functions (__builtin_apply and friends) are used. */
|
|
IMP (*__objc_msg_forward) (SEL) = NULL;
|
|
IMP (*__objc_msg_forward2) (id, SEL) = NULL;
|
|
|
|
/* Send +initialize to class. */
|
|
static void __objc_send_initialize (Class);
|
|
|
|
/* Forward declare some functions */
|
|
static void __objc_install_dtable_for_class (Class cls);
|
|
static void __objc_prepare_dtable_for_class (Class cls);
|
|
static void __objc_install_prepared_dtable_for_class (Class cls);
|
|
|
|
static struct sarray *__objc_prepared_dtable_for_class (Class cls);
|
|
static IMP __objc_get_prepared_imp (Class cls,SEL sel);
|
|
|
|
|
|
/* Various forwarding functions that are used based upon the
|
|
return type for the selector.
|
|
__objc_block_forward for structures.
|
|
__objc_double_forward for floats/doubles.
|
|
__objc_word_forward for pointers or types that fit in registers. */
|
|
static double __objc_double_forward (id, SEL, ...);
|
|
static id __objc_word_forward (id, SEL, ...);
|
|
typedef struct { id many[8]; } __big;
|
|
#if INVISIBLE_STRUCT_RETURN
|
|
static __big
|
|
#else
|
|
static id
|
|
#endif
|
|
__objc_block_forward (id, SEL, ...);
|
|
static struct objc_method * search_for_method_in_hierarchy (Class class, SEL sel);
|
|
struct objc_method * search_for_method_in_list (struct objc_method_list * list, SEL op);
|
|
id nil_method (id, SEL);
|
|
|
|
/* Make sure this inline function is exported regardless of GNU89 or C99
|
|
inlining semantics as it is part of the libobjc ABI. */
|
|
extern IMP __objc_get_forward_imp (id, SEL);
|
|
|
|
/* Given a selector, return the proper forwarding implementation. */
|
|
inline
|
|
IMP
|
|
__objc_get_forward_imp (id rcv, SEL sel)
|
|
{
|
|
/* If a custom forwarding hook was registered, try getting a
|
|
forwarding function from it. There are two forward routine hooks,
|
|
one that takes the receiver as an argument and one that does
|
|
not. */
|
|
if (__objc_msg_forward2)
|
|
{
|
|
IMP result;
|
|
if ((result = __objc_msg_forward2 (rcv, sel)) != NULL)
|
|
return result;
|
|
}
|
|
if (__objc_msg_forward)
|
|
{
|
|
IMP result;
|
|
if ((result = __objc_msg_forward (sel)) != NULL)
|
|
return result;
|
|
}
|
|
|
|
/* In all other cases, use the default forwarding functions built
|
|
using __builtin_apply and friends. */
|
|
{
|
|
const char *t = sel->sel_types;
|
|
|
|
if (t && (*t == '[' || *t == '(' || *t == '{')
|
|
#ifdef OBJC_MAX_STRUCT_BY_VALUE
|
|
&& objc_sizeof_type (t) > OBJC_MAX_STRUCT_BY_VALUE
|
|
#endif
|
|
)
|
|
return (IMP)__objc_block_forward;
|
|
else if (t && (*t == 'f' || *t == 'd'))
|
|
return (IMP)__objc_double_forward;
|
|
else
|
|
return (IMP)__objc_word_forward;
|
|
}
|
|
}
|
|
|
|
/* Selectors for +resolveClassMethod: and +resolveInstanceMethod:.
|
|
These are set up at startup. */
|
|
static SEL selector_resolveClassMethod = NULL;
|
|
static SEL selector_resolveInstanceMethod = NULL;
|
|
|
|
/* Internal routines use to resolve a class method using
|
|
+resolveClassMethod:. 'class' is always a non-Nil class (*not* a
|
|
meta-class), and 'sel' is the selector that we are trying to
|
|
resolve. This must be called when class is not Nil, and the
|
|
dispatch table for class methods has already been installed.
|
|
|
|
This routine tries to call +resolveClassMethod: to give an
|
|
opportunity to resolve the method. If +resolveClassMethod: returns
|
|
YES, it tries looking up the method again, and if found, it returns
|
|
it. Else, it returns NULL. */
|
|
static inline
|
|
IMP
|
|
__objc_resolve_class_method (Class class, SEL sel)
|
|
{
|
|
/* We need to lookup +resolveClassMethod:. */
|
|
BOOL (*resolveMethodIMP) (id, SEL, SEL);
|
|
|
|
/* The dispatch table for class methods is already installed and we
|
|
don't want any forwarding to happen when looking up this method,
|
|
so we just look it up directly. Note that if 'sel' is precisely
|
|
+resolveClassMethod:, this would look it up yet again and find
|
|
nothing. That's no problem and there's no recursion. */
|
|
resolveMethodIMP = (BOOL (*) (id, SEL, SEL))sarray_get_safe
|
|
(class->class_pointer->dtable, (size_t) selector_resolveClassMethod->sel_id);
|
|
|
|
if (resolveMethodIMP && resolveMethodIMP ((id)class, selector_resolveClassMethod, sel))
|
|
{
|
|
/* +resolveClassMethod: returned YES. Look the method up again.
|
|
We already know the dtable is installed. */
|
|
|
|
/* TODO: There is the case where +resolveClassMethod: is buggy
|
|
and returned YES without actually adding the method. We
|
|
could maybe print an error message. */
|
|
return sarray_get_safe (class->class_pointer->dtable, (size_t) sel->sel_id);
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
/* Internal routines use to resolve a instance method using
|
|
+resolveInstanceMethod:. 'class' is always a non-Nil class, and
|
|
'sel' is the selector that we are trying to resolve. This must be
|
|
called when class is not Nil, and the dispatch table for instance
|
|
methods has already been installed.
|
|
|
|
This routine tries to call +resolveInstanceMethod: to give an
|
|
opportunity to resolve the method. If +resolveInstanceMethod:
|
|
returns YES, it tries looking up the method again, and if found, it
|
|
returns it. Else, it returns NULL. */
|
|
static inline
|
|
IMP
|
|
__objc_resolve_instance_method (Class class, SEL sel)
|
|
{
|
|
/* We need to lookup +resolveInstanceMethod:. */
|
|
BOOL (*resolveMethodIMP) (id, SEL, SEL);
|
|
|
|
/* The dispatch table for class methods may not be already installed
|
|
so we have to install it if needed. */
|
|
resolveMethodIMP = sarray_get_safe (class->class_pointer->dtable,
|
|
(size_t) selector_resolveInstanceMethod->sel_id);
|
|
if (resolveMethodIMP == 0)
|
|
{
|
|
/* Try again after installing the dtable. */
|
|
if (class->class_pointer->dtable == __objc_uninstalled_dtable)
|
|
{
|
|
objc_mutex_lock (__objc_runtime_mutex);
|
|
if (class->class_pointer->dtable == __objc_uninstalled_dtable)
|
|
__objc_install_dtable_for_class (class->class_pointer);
|
|
objc_mutex_unlock (__objc_runtime_mutex);
|
|
}
|
|
resolveMethodIMP = sarray_get_safe (class->class_pointer->dtable,
|
|
(size_t) selector_resolveInstanceMethod->sel_id);
|
|
}
|
|
|
|
if (resolveMethodIMP && resolveMethodIMP ((id)class, selector_resolveInstanceMethod, sel))
|
|
{
|
|
/* +resolveInstanceMethod: returned YES. Look the method up
|
|
again. We already know the dtable is installed. */
|
|
|
|
/* TODO: There is the case where +resolveInstanceMethod: is
|
|
buggy and returned YES without actually adding the method.
|
|
We could maybe print an error message. */
|
|
return sarray_get_safe (class->dtable, (size_t) sel->sel_id);
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
/* Given a CLASS and selector, return the implementation corresponding
|
|
to the method of the selector.
|
|
|
|
If CLASS is a class, the instance method is returned.
|
|
If CLASS is a meta class, the class method is returned.
|
|
|
|
Since this requires the dispatch table to be installed, this function
|
|
will implicitly invoke +initialize for CLASS if it hasn't been
|
|
invoked yet. This also insures that +initialize has been invoked
|
|
when the returned implementation is called directly.
|
|
|
|
The forwarding hooks require the receiver as an argument (if they are to
|
|
perform dynamic lookup in proxy objects etc), so this function has a
|
|
receiver argument to be used with those hooks. */
|
|
static inline
|
|
IMP
|
|
get_implementation (id receiver, Class class, SEL sel)
|
|
{
|
|
void *res;
|
|
|
|
if (class->dtable == __objc_uninstalled_dtable)
|
|
{
|
|
/* The dispatch table needs to be installed. */
|
|
objc_mutex_lock (__objc_runtime_mutex);
|
|
|
|
/* Double-checked locking pattern: Check
|
|
__objc_uninstalled_dtable again in case another thread
|
|
installed the dtable while we were waiting for the lock to be
|
|
released. */
|
|
if (class->dtable == __objc_uninstalled_dtable)
|
|
__objc_install_dtable_for_class (class);
|
|
|
|
/* If the dispatch table is not yet installed, we are still in
|
|
the process of executing +initialize. But the implementation
|
|
pointer should be available in the prepared ispatch table if
|
|
it exists at all. */
|
|
if (class->dtable == __objc_uninstalled_dtable)
|
|
{
|
|
assert (__objc_prepared_dtable_for_class (class) != 0);
|
|
res = __objc_get_prepared_imp (class, sel);
|
|
}
|
|
else
|
|
res = 0;
|
|
|
|
objc_mutex_unlock (__objc_runtime_mutex);
|
|
/* Call ourselves with the installed dispatch table and get the
|
|
real method. */
|
|
if (!res)
|
|
res = get_implementation (receiver, class, sel);
|
|
}
|
|
else
|
|
{
|
|
/* The dispatch table has been installed. */
|
|
res = sarray_get_safe (class->dtable, (size_t) sel->sel_id);
|
|
if (res == 0)
|
|
{
|
|
/* The dispatch table has been installed, and the method is
|
|
not in the dispatch table. So the method just doesn't
|
|
exist for the class. */
|
|
|
|
/* Try going through the +resolveClassMethod: or
|
|
+resolveInstanceMethod: process. */
|
|
if (CLS_ISMETA (class))
|
|
{
|
|
/* We have the meta class, but we need to invoke the
|
|
+resolveClassMethod: method on the class. So, we
|
|
need to obtain the class from the meta class, which
|
|
we do using the fact that both the class and the
|
|
meta-class have the same name. */
|
|
Class realClass = objc_lookUpClass (class->name);
|
|
if (realClass)
|
|
res = __objc_resolve_class_method (realClass, sel);
|
|
}
|
|
else
|
|
res = __objc_resolve_instance_method (class, sel);
|
|
|
|
if (res == 0)
|
|
res = __objc_get_forward_imp (receiver, sel);
|
|
}
|
|
}
|
|
return res;
|
|
}
|
|
|
|
/* Make sure this inline function is exported regardless of GNU89 or C99
|
|
inlining semantics as it is part of the libobjc ABI. */
|
|
extern IMP get_imp (Class, SEL);
|
|
|
|
inline
|
|
IMP
|
|
get_imp (Class class, SEL sel)
|
|
{
|
|
/* In a vanilla implementation we would first check if the dispatch
|
|
table is installed. Here instead, to get more speed in the
|
|
standard case (that the dispatch table is installed) we first try
|
|
to get the imp using brute force. Only if that fails, we do what
|
|
we should have been doing from the very beginning, that is, check
|
|
if the dispatch table needs to be installed, install it if it's
|
|
not installed, and retrieve the imp from the table if it's
|
|
installed. */
|
|
void *res = sarray_get_safe (class->dtable, (size_t) sel->sel_id);
|
|
if (res == 0)
|
|
{
|
|
res = get_implementation(nil, class, sel);
|
|
}
|
|
return res;
|
|
}
|
|
|
|
/* The new name of get_imp(). */
|
|
IMP
|
|
class_getMethodImplementation (Class class_, SEL selector)
|
|
{
|
|
if (class_ == Nil || selector == NULL)
|
|
return NULL;
|
|
|
|
/* get_imp is inlined, so we're good. */
|
|
return get_imp (class_, selector);
|
|
}
|
|
|
|
/* Given a method, return its implementation. This has been replaced
|
|
by method_getImplementation() in the modern API. */
|
|
IMP
|
|
method_get_imp (struct objc_method * method)
|
|
{
|
|
return (method != (struct objc_method *)0) ? method->method_imp : (IMP)0;
|
|
}
|
|
|
|
/* Query if an object can respond to a selector, returns YES if the
|
|
object implements the selector otherwise NO. Does not check if the
|
|
method can be forwarded. Since this requires the dispatch table to
|
|
installed, this function will implicitly invoke +initialize for the
|
|
class of OBJECT if it hasn't been invoked yet. */
|
|
inline
|
|
BOOL
|
|
__objc_responds_to (id object, SEL sel)
|
|
{
|
|
void *res;
|
|
struct sarray *dtable;
|
|
|
|
/* Install dispatch table if need be */
|
|
dtable = object->class_pointer->dtable;
|
|
if (dtable == __objc_uninstalled_dtable)
|
|
{
|
|
objc_mutex_lock (__objc_runtime_mutex);
|
|
if (object->class_pointer->dtable == __objc_uninstalled_dtable)
|
|
__objc_install_dtable_for_class (object->class_pointer);
|
|
|
|
/* If the dispatch table is not yet installed, we are still in
|
|
the process of executing +initialize. Yet the dispatch table
|
|
should be available. */
|
|
if (object->class_pointer->dtable == __objc_uninstalled_dtable)
|
|
{
|
|
dtable = __objc_prepared_dtable_for_class (object->class_pointer);
|
|
assert (dtable);
|
|
}
|
|
else
|
|
dtable = object->class_pointer->dtable;
|
|
|
|
objc_mutex_unlock (__objc_runtime_mutex);
|
|
}
|
|
|
|
/* Get the method from the dispatch table. */
|
|
res = sarray_get_safe (dtable, (size_t) sel->sel_id);
|
|
return (res != 0) ? YES : NO;
|
|
}
|
|
|
|
BOOL
|
|
class_respondsToSelector (Class class_, SEL selector)
|
|
{
|
|
struct sarray *dtable;
|
|
void *res;
|
|
|
|
if (class_ == Nil || selector == NULL)
|
|
return NO;
|
|
|
|
/* Install dispatch table if need be. */
|
|
dtable = class_->dtable;
|
|
if (dtable == __objc_uninstalled_dtable)
|
|
{
|
|
objc_mutex_lock (__objc_runtime_mutex);
|
|
if (class_->dtable == __objc_uninstalled_dtable)
|
|
__objc_install_dtable_for_class (class_);
|
|
|
|
/* If the dispatch table is not yet installed,
|
|
we are still in the process of executing +initialize.
|
|
Yet the dispatch table should be available. */
|
|
if (class_->dtable == __objc_uninstalled_dtable)
|
|
{
|
|
dtable = __objc_prepared_dtable_for_class (class_);
|
|
assert (dtable);
|
|
}
|
|
else
|
|
dtable = class_->dtable;
|
|
|
|
objc_mutex_unlock (__objc_runtime_mutex);
|
|
}
|
|
|
|
/* Get the method from the dispatch table. */
|
|
res = sarray_get_safe (dtable, (size_t) selector->sel_id);
|
|
return (res != 0) ? YES : NO;
|
|
}
|
|
|
|
/* This is the lookup function. All entries in the table are either a
|
|
valid method *or* zero. If zero then either the dispatch table
|
|
needs to be installed or it doesn't exist and forwarding is
|
|
attempted. */
|
|
IMP
|
|
objc_msg_lookup (id receiver, SEL op)
|
|
{
|
|
IMP result;
|
|
if (receiver)
|
|
{
|
|
/* First try a quick lookup assuming the dispatch table exists. */
|
|
result = sarray_get_safe (receiver->class_pointer->dtable,
|
|
(sidx)op->sel_id);
|
|
if (result == 0)
|
|
{
|
|
/* Not found ... call get_implementation () to install the
|
|
dispatch table and call +initialize as required,
|
|
providing the method implementation or a forwarding
|
|
function. */
|
|
result = get_implementation (receiver, receiver->class_pointer, op);
|
|
}
|
|
return result;
|
|
}
|
|
else
|
|
return (IMP)nil_method;
|
|
}
|
|
|
|
IMP
|
|
objc_msg_lookup_super (struct objc_super *super, SEL sel)
|
|
{
|
|
if (super->self)
|
|
return get_imp (super->super_class, sel);
|
|
else
|
|
return (IMP)nil_method;
|
|
}
|
|
|
|
void
|
|
__objc_init_dispatch_tables ()
|
|
{
|
|
__objc_uninstalled_dtable = sarray_new (200, 0);
|
|
|
|
/* TODO: It would be cool to register typed selectors here. */
|
|
selector_resolveClassMethod = sel_registerName ("resolveClassMethod:");
|
|
selector_resolveInstanceMethod = sel_registerName ("resolveInstanceMethod:");
|
|
}
|
|
|
|
|
|
/* Install dummy table for class which causes the first message to
|
|
that class (or instances hereof) to be initialized properly. */
|
|
void
|
|
__objc_install_premature_dtable (Class class)
|
|
{
|
|
assert (__objc_uninstalled_dtable);
|
|
class->dtable = __objc_uninstalled_dtable;
|
|
}
|
|
|
|
/* Send +initialize to class if not already done. */
|
|
static void
|
|
__objc_send_initialize (Class class)
|
|
{
|
|
/* This *must* be a class object. */
|
|
assert (CLS_ISCLASS (class));
|
|
assert (! CLS_ISMETA (class));
|
|
|
|
/* class_add_method_list/__objc_update_dispatch_table_for_class may
|
|
have reset the dispatch table. The canonical way to insure that
|
|
we send +initialize just once, is this flag. */
|
|
if (! CLS_ISINITIALIZED (class))
|
|
{
|
|
DEBUG_PRINTF ("+initialize: need to initialize class '%s'\n", class->name);
|
|
CLS_SETINITIALIZED (class);
|
|
CLS_SETINITIALIZED (class->class_pointer);
|
|
|
|
/* Create the garbage collector type memory description. */
|
|
__objc_generate_gc_type_description (class);
|
|
|
|
if (class->super_class)
|
|
__objc_send_initialize (class->super_class);
|
|
|
|
{
|
|
SEL op = sel_registerName ("initialize");
|
|
struct objc_method *method = search_for_method_in_hierarchy (class->class_pointer,
|
|
op);
|
|
|
|
if (method)
|
|
{
|
|
DEBUG_PRINTF (" begin of [%s +initialize]\n", class->name);
|
|
(*method->method_imp) ((id)class, op);
|
|
DEBUG_PRINTF (" end of [%s +initialize]\n", class->name);
|
|
}
|
|
#ifdef DEBUG
|
|
else
|
|
{
|
|
DEBUG_PRINTF (" class '%s' has no +initialize method\n", class->name);
|
|
}
|
|
#endif
|
|
}
|
|
}
|
|
}
|
|
|
|
/* Walk on the methods list of class and install the methods in the
|
|
reverse order of the lists. Since methods added by categories are
|
|
before the methods of class in the methods list, this allows
|
|
categories to substitute methods declared in class. However if
|
|
more than one category replaces the same method nothing is
|
|
guaranteed about what method will be used. Assumes that
|
|
__objc_runtime_mutex is locked down. */
|
|
static void
|
|
__objc_install_methods_in_dtable (struct sarray *dtable, struct objc_method_list * method_list)
|
|
{
|
|
int i;
|
|
|
|
if (! method_list)
|
|
return;
|
|
|
|
if (method_list->method_next)
|
|
__objc_install_methods_in_dtable (dtable, method_list->method_next);
|
|
|
|
for (i = 0; i < method_list->method_count; i++)
|
|
{
|
|
struct objc_method * method = &(method_list->method_list[i]);
|
|
sarray_at_put_safe (dtable,
|
|
(sidx) method->method_name->sel_id,
|
|
method->method_imp);
|
|
}
|
|
}
|
|
|
|
void
|
|
__objc_update_dispatch_table_for_class (Class class)
|
|
{
|
|
Class next;
|
|
struct sarray *arr;
|
|
|
|
DEBUG_PRINTF (" _objc_update_dtable_for_class (%s)\n", class->name);
|
|
|
|
objc_mutex_lock (__objc_runtime_mutex);
|
|
|
|
/* Not yet installed -- skip it unless in +initialize. */
|
|
if (class->dtable == __objc_uninstalled_dtable)
|
|
{
|
|
if (__objc_prepared_dtable_for_class (class))
|
|
{
|
|
/* There is a prepared table so we must be initialising this
|
|
class ... we must re-do the table preparation. */
|
|
__objc_prepare_dtable_for_class (class);
|
|
}
|
|
objc_mutex_unlock (__objc_runtime_mutex);
|
|
return;
|
|
}
|
|
|
|
arr = class->dtable;
|
|
__objc_install_premature_dtable (class); /* someone might require it... */
|
|
sarray_free (arr); /* release memory */
|
|
|
|
/* Could have been lazy... */
|
|
__objc_install_dtable_for_class (class);
|
|
|
|
if (class->subclass_list) /* Traverse subclasses. */
|
|
for (next = class->subclass_list; next; next = next->sibling_class)
|
|
__objc_update_dispatch_table_for_class (next);
|
|
|
|
objc_mutex_unlock (__objc_runtime_mutex);
|
|
}
|
|
|
|
/* This function adds a method list to a class. This function is
|
|
typically called by another function specific to the run-time. As
|
|
such this function does not worry about thread safe issues.
|
|
|
|
This one is only called for categories. Class objects have their
|
|
methods installed right away, and their selectors are made into
|
|
SEL's by the function __objc_register_selectors_from_class. */
|
|
void
|
|
class_add_method_list (Class class, struct objc_method_list * list)
|
|
{
|
|
/* Passing of a linked list is not allowed. Do multiple calls. */
|
|
assert (! list->method_next);
|
|
|
|
__objc_register_selectors_from_list(list);
|
|
|
|
/* Add the methods to the class's method list. */
|
|
list->method_next = class->methods;
|
|
class->methods = list;
|
|
|
|
/* Update the dispatch table of class. */
|
|
__objc_update_dispatch_table_for_class (class);
|
|
}
|
|
|
|
struct objc_method *
|
|
class_getInstanceMethod (Class class_, SEL selector)
|
|
{
|
|
struct objc_method *m;
|
|
|
|
if (class_ == Nil || selector == NULL)
|
|
return NULL;
|
|
|
|
m = search_for_method_in_hierarchy (class_, selector);
|
|
if (m)
|
|
return m;
|
|
|
|
/* Try going through +resolveInstanceMethod:, and do the search
|
|
again if successful. */
|
|
if (__objc_resolve_instance_method (class_, selector))
|
|
return search_for_method_in_hierarchy (class_, selector);
|
|
|
|
return NULL;
|
|
}
|
|
|
|
struct objc_method *
|
|
class_getClassMethod (Class class_, SEL selector)
|
|
{
|
|
struct objc_method *m;
|
|
|
|
if (class_ == Nil || selector == NULL)
|
|
return NULL;
|
|
|
|
m = search_for_method_in_hierarchy (class_->class_pointer,
|
|
selector);
|
|
if (m)
|
|
return m;
|
|
|
|
/* Try going through +resolveClassMethod:, and do the search again
|
|
if successful. */
|
|
if (__objc_resolve_class_method (class_, selector))
|
|
return search_for_method_in_hierarchy (class_->class_pointer,
|
|
selector);
|
|
|
|
return NULL;
|
|
}
|
|
|
|
BOOL
|
|
class_addMethod (Class class_, SEL selector, IMP implementation,
|
|
const char *method_types)
|
|
{
|
|
struct objc_method_list *method_list;
|
|
struct objc_method *method;
|
|
const char *method_name;
|
|
|
|
if (class_ == Nil || selector == NULL || implementation == NULL
|
|
|| method_types == NULL || (strcmp (method_types, "") == 0))
|
|
return NO;
|
|
|
|
method_name = sel_getName (selector);
|
|
if (method_name == NULL)
|
|
return NO;
|
|
|
|
/* If the method already exists in the class, return NO. It is fine
|
|
if the method already exists in the superclass; in that case, we
|
|
are overriding it. */
|
|
if (CLS_IS_IN_CONSTRUCTION (class_))
|
|
{
|
|
/* The class only contains a list of methods; they have not been
|
|
registered yet, ie, the method_name of each of them is still
|
|
a string, not a selector. Iterate manually over them to
|
|
check if we have already added the method. */
|
|
struct objc_method_list * method_list = class_->methods;
|
|
while (method_list)
|
|
{
|
|
int i;
|
|
|
|
/* Search the method list. */
|
|
for (i = 0; i < method_list->method_count; ++i)
|
|
{
|
|
struct objc_method * method = &method_list->method_list[i];
|
|
|
|
if (method->method_name
|
|
&& strcmp ((char *)method->method_name, method_name) == 0)
|
|
return NO;
|
|
}
|
|
|
|
/* The method wasn't found. Follow the link to the next list of
|
|
methods. */
|
|
method_list = method_list->method_next;
|
|
}
|
|
/* The method wasn't found. It's a new one. Go ahead and add
|
|
it. */
|
|
}
|
|
else
|
|
{
|
|
/* Do the standard lookup. This assumes the selectors are
|
|
mapped. */
|
|
if (search_for_method_in_list (class_->methods, selector))
|
|
return NO;
|
|
}
|
|
|
|
method_list = (struct objc_method_list *)objc_calloc (1, sizeof (struct objc_method_list));
|
|
method_list->method_count = 1;
|
|
|
|
method = &(method_list->method_list[0]);
|
|
method->method_name = objc_malloc (strlen (method_name) + 1);
|
|
strcpy ((char *)method->method_name, method_name);
|
|
|
|
method->method_types = objc_malloc (strlen (method_types) + 1);
|
|
strcpy ((char *)method->method_types, method_types);
|
|
|
|
method->method_imp = implementation;
|
|
|
|
if (CLS_IS_IN_CONSTRUCTION (class_))
|
|
{
|
|
/* We only need to add the method to the list. It will be
|
|
registered with the runtime when the class pair is registered
|
|
(if ever). */
|
|
method_list->method_next = class_->methods;
|
|
class_->methods = method_list;
|
|
}
|
|
else
|
|
{
|
|
/* Add the method to a live class. */
|
|
objc_mutex_lock (__objc_runtime_mutex);
|
|
class_add_method_list (class_, method_list);
|
|
objc_mutex_unlock (__objc_runtime_mutex);
|
|
}
|
|
|
|
return YES;
|
|
}
|
|
|
|
IMP
|
|
class_replaceMethod (Class class_, SEL selector, IMP implementation,
|
|
const char *method_types)
|
|
{
|
|
struct objc_method * method;
|
|
|
|
if (class_ == Nil || selector == NULL || implementation == NULL
|
|
|| method_types == NULL)
|
|
return NULL;
|
|
|
|
method = search_for_method_in_hierarchy (class_, selector);
|
|
|
|
if (method)
|
|
{
|
|
return method_setImplementation (method, implementation);
|
|
}
|
|
else
|
|
{
|
|
class_addMethod (class_, selector, implementation, method_types);
|
|
return NULL;
|
|
}
|
|
}
|
|
|
|
/* Search for a method starting from the current class up its
|
|
hierarchy. Return a pointer to the method's method structure if
|
|
found. NULL otherwise. */
|
|
static struct objc_method *
|
|
search_for_method_in_hierarchy (Class cls, SEL sel)
|
|
{
|
|
struct objc_method * method = NULL;
|
|
Class class;
|
|
|
|
if (! sel_is_mapped (sel))
|
|
return NULL;
|
|
|
|
/* Scan the method list of the class. If the method isn't found in
|
|
the list then step to its super class. */
|
|
for (class = cls; ((! method) && class); class = class->super_class)
|
|
method = search_for_method_in_list (class->methods, sel);
|
|
|
|
return method;
|
|
}
|
|
|
|
|
|
|
|
/* Given a linked list of method and a method's name. Search for the
|
|
named method's method structure. Return a pointer to the method's
|
|
method structure if found. NULL otherwise. */
|
|
struct objc_method *
|
|
search_for_method_in_list (struct objc_method_list * list, SEL op)
|
|
{
|
|
struct objc_method_list * method_list = list;
|
|
|
|
if (! sel_is_mapped (op))
|
|
return NULL;
|
|
|
|
/* If not found then we'll search the list. */
|
|
while (method_list)
|
|
{
|
|
int i;
|
|
|
|
/* Search the method list. */
|
|
for (i = 0; i < method_list->method_count; ++i)
|
|
{
|
|
struct objc_method * method = &method_list->method_list[i];
|
|
|
|
if (method->method_name)
|
|
if (method->method_name->sel_id == op->sel_id)
|
|
return method;
|
|
}
|
|
|
|
/* The method wasn't found. Follow the link to the next list of
|
|
methods. */
|
|
method_list = method_list->method_next;
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
typedef void * retval_t;
|
|
typedef void * arglist_t;
|
|
|
|
static retval_t __objc_forward (id object, SEL sel, arglist_t args);
|
|
|
|
/* Forwarding pointers/integers through the normal registers. */
|
|
static id
|
|
__objc_word_forward (id rcv, SEL op, ...)
|
|
{
|
|
void *args, *res;
|
|
|
|
args = __builtin_apply_args ();
|
|
res = __objc_forward (rcv, op, args);
|
|
if (res)
|
|
__builtin_return (res);
|
|
else
|
|
return res;
|
|
}
|
|
|
|
/* Specific routine for forwarding floats/double because of
|
|
architectural differences on some processors. i386s for example
|
|
which uses a floating point stack versus general registers for
|
|
floating point numbers. This forward routine makes sure that GCC
|
|
restores the proper return values. */
|
|
static double
|
|
__objc_double_forward (id rcv, SEL op, ...)
|
|
{
|
|
void *args, *res;
|
|
|
|
args = __builtin_apply_args ();
|
|
res = __objc_forward (rcv, op, args);
|
|
__builtin_return (res);
|
|
}
|
|
|
|
#if INVISIBLE_STRUCT_RETURN
|
|
static __big
|
|
#else
|
|
static id
|
|
#endif
|
|
__objc_block_forward (id rcv, SEL op, ...)
|
|
{
|
|
void *args, *res;
|
|
|
|
args = __builtin_apply_args ();
|
|
res = __objc_forward (rcv, op, args);
|
|
if (res)
|
|
__builtin_return (res);
|
|
else
|
|
#if INVISIBLE_STRUCT_RETURN
|
|
return (__big) {{0, 0, 0, 0, 0, 0, 0, 0}};
|
|
#else
|
|
return nil;
|
|
#endif
|
|
}
|
|
|
|
|
|
/* This function is called for methods which are not implemented,
|
|
unless a custom forwarding routine has been installed. Please note
|
|
that most serious users of libobjc (eg, GNUstep base) do install
|
|
their own forwarding routines, and hence this is never actually
|
|
used. But, if no custom forwarding routine is installed, this is
|
|
called when a selector is not recognized. */
|
|
static retval_t
|
|
__objc_forward (id object, SEL sel, arglist_t args)
|
|
{
|
|
IMP imp;
|
|
static SEL frwd_sel = 0; /* !T:SAFE2 */
|
|
SEL err_sel;
|
|
|
|
/* First try if the object understands forward::. */
|
|
if (! frwd_sel)
|
|
frwd_sel = sel_get_any_uid ("forward::");
|
|
|
|
if (__objc_responds_to (object, frwd_sel))
|
|
{
|
|
imp = get_implementation (object, object->class_pointer, frwd_sel);
|
|
return (*imp) (object, frwd_sel, sel, args);
|
|
}
|
|
|
|
/* If the object recognizes the doesNotRecognize: method then we're
|
|
going to send it. */
|
|
err_sel = sel_get_any_uid ("doesNotRecognize:");
|
|
if (__objc_responds_to (object, err_sel))
|
|
{
|
|
imp = get_implementation (object, object->class_pointer, err_sel);
|
|
return (*imp) (object, err_sel, sel);
|
|
}
|
|
|
|
/* The object doesn't recognize the method. Check for responding to
|
|
error:. If it does then sent it. */
|
|
{
|
|
char msg[256 + strlen ((const char *) sel_getName (sel))
|
|
+ strlen ((const char *) object->class_pointer->name)];
|
|
|
|
sprintf (msg, "(%s) %s does not recognize %s",
|
|
(CLS_ISMETA (object->class_pointer)
|
|
? "class"
|
|
: "instance" ),
|
|
object->class_pointer->name, sel_getName (sel));
|
|
|
|
/* The object doesn't respond to doesNotRecognize:. Therefore, a
|
|
default action is taken. */
|
|
_objc_abort ("%s\n", msg);
|
|
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
void
|
|
__objc_print_dtable_stats (void)
|
|
{
|
|
int total = 0;
|
|
|
|
objc_mutex_lock (__objc_runtime_mutex);
|
|
|
|
#ifdef OBJC_SPARSE2
|
|
printf ("memory usage: (%s)\n", "2-level sparse arrays");
|
|
#else
|
|
printf ("memory usage: (%s)\n", "3-level sparse arrays");
|
|
#endif
|
|
|
|
printf ("arrays: %d = %ld bytes\n", narrays,
|
|
(long) ((size_t) narrays * sizeof (struct sarray)));
|
|
total += narrays * sizeof (struct sarray);
|
|
printf ("buckets: %d = %ld bytes\n", nbuckets,
|
|
(long) ((size_t) nbuckets * sizeof (struct sbucket)));
|
|
total += nbuckets * sizeof (struct sbucket);
|
|
|
|
printf ("idxtables: %d = %ld bytes\n",
|
|
idxsize, (long) ((size_t) idxsize * sizeof (void *)));
|
|
total += idxsize * sizeof (void *);
|
|
printf ("-----------------------------------\n");
|
|
printf ("total: %d bytes\n", total);
|
|
printf ("===================================\n");
|
|
|
|
objc_mutex_unlock (__objc_runtime_mutex);
|
|
}
|
|
|
|
static cache_ptr prepared_dtable_table = 0;
|
|
|
|
/* This function is called by: objc_msg_lookup, get_imp and
|
|
__objc_responds_to (and the dispatch table installation functions
|
|
themselves) to install a dispatch table for a class.
|
|
|
|
If CLS is a class, it installs instance methods.
|
|
If CLS is a meta class, it installs class methods.
|
|
|
|
In either case +initialize is invoked for the corresponding class.
|
|
|
|
The implementation must insure that the dispatch table is not
|
|
installed until +initialize completes. Otherwise it opens a
|
|
potential race since the installation of the dispatch table is used
|
|
as gate in regular method dispatch and we need to guarantee that
|
|
+initialize is the first method invoked an that no other thread my
|
|
dispatch messages to the class before +initialize completes. */
|
|
static void
|
|
__objc_install_dtable_for_class (Class cls)
|
|
{
|
|
/* If the class has not yet had its class links resolved, we must
|
|
re-compute all class links. */
|
|
if (! CLS_ISRESOLV (cls))
|
|
__objc_resolve_class_links ();
|
|
|
|
/* Make sure the super class has its dispatch table installed or is
|
|
at least preparing. We do not need to send initialize for the
|
|
super class since __objc_send_initialize will insure that. */
|
|
if (cls->super_class
|
|
&& cls->super_class->dtable == __objc_uninstalled_dtable
|
|
&& !__objc_prepared_dtable_for_class (cls->super_class))
|
|
{
|
|
__objc_install_dtable_for_class (cls->super_class);
|
|
/* The superclass initialisation may have also initialised the
|
|
current class, in which case there is no more to do. */
|
|
if (cls->dtable != __objc_uninstalled_dtable)
|
|
return;
|
|
}
|
|
|
|
/* We have already been prepared but +initialize hasn't completed.
|
|
The +initialize implementation is probably sending 'self'
|
|
messages. We rely on _objc_get_prepared_imp to retrieve the
|
|
implementation pointers. */
|
|
if (__objc_prepared_dtable_for_class (cls))
|
|
return;
|
|
|
|
/* We have this function cache the implementation pointers for
|
|
_objc_get_prepared_imp but the dispatch table won't be initilized
|
|
until __objc_send_initialize completes. */
|
|
__objc_prepare_dtable_for_class (cls);
|
|
|
|
/* We may have already invoked +initialize but
|
|
__objc_update_dispatch_table_for_class invoked by
|
|
class_add_method_list may have reset dispatch table. */
|
|
|
|
/* Call +initialize. If we are a real class, we are installing
|
|
instance methods. If we are a meta class, we are installing
|
|
class methods. The __objc_send_initialize itself will insure
|
|
that the message is called only once per class. */
|
|
if (CLS_ISCLASS (cls))
|
|
__objc_send_initialize (cls);
|
|
else
|
|
{
|
|
/* Retrieve the class from the meta class. */
|
|
Class c = objc_getClass (cls->name);
|
|
assert (CLS_ISMETA (cls));
|
|
assert (c);
|
|
__objc_send_initialize (c);
|
|
}
|
|
|
|
/* We install the dispatch table correctly when +initialize completed. */
|
|
__objc_install_prepared_dtable_for_class (cls);
|
|
}
|
|
|
|
/* Builds the dispatch table for the class CLS and stores it in a
|
|
place where it can be retrieved by __objc_get_prepared_imp until
|
|
__objc_install_prepared_dtable_for_class installs it into the
|
|
class. The dispatch table should not be installed into the class
|
|
until +initialize has completed. */
|
|
static void
|
|
__objc_prepare_dtable_for_class (Class cls)
|
|
{
|
|
struct sarray *dtable;
|
|
struct sarray *super_dtable;
|
|
|
|
/* This table could be initialized in init.c. We can not use the
|
|
class name since the class maintains the instance methods and the
|
|
meta class maintains the the class methods yet both share the
|
|
same name. Classes should be unique in any program. */
|
|
if (! prepared_dtable_table)
|
|
prepared_dtable_table
|
|
= objc_hash_new (32,
|
|
(hash_func_type) objc_hash_ptr,
|
|
(compare_func_type) objc_compare_ptrs);
|
|
|
|
/* If the class has not yet had its class links resolved, we must
|
|
re-compute all class links. */
|
|
if (! CLS_ISRESOLV (cls))
|
|
__objc_resolve_class_links ();
|
|
|
|
assert (cls);
|
|
assert (cls->dtable == __objc_uninstalled_dtable);
|
|
|
|
/* If there is already a prepared dtable for this class, we must
|
|
replace it with a new version (since there must have been methods
|
|
added to or otherwise modified in the class while executing
|
|
+initialize, and the table needs to be recomputed. */
|
|
dtable = __objc_prepared_dtable_for_class (cls);
|
|
if (dtable != 0)
|
|
{
|
|
objc_hash_remove (prepared_dtable_table, cls);
|
|
sarray_free (dtable);
|
|
}
|
|
|
|
/* Now prepare the dtable for population. */
|
|
assert (cls != cls->super_class);
|
|
if (cls->super_class)
|
|
{
|
|
/* Inherit the method list from the super class. Yet the super
|
|
class may still be initializing in the case when a class
|
|
cluster sub class initializes its super classes. */
|
|
if (cls->super_class->dtable == __objc_uninstalled_dtable)
|
|
__objc_install_dtable_for_class (cls->super_class);
|
|
|
|
super_dtable = cls->super_class->dtable;
|
|
/* If the dispatch table is not yet installed, we are still in
|
|
the process of executing +initialize. Yet the dispatch table
|
|
should be available. */
|
|
if (super_dtable == __objc_uninstalled_dtable)
|
|
super_dtable = __objc_prepared_dtable_for_class (cls->super_class);
|
|
|
|
assert (super_dtable);
|
|
dtable = sarray_lazy_copy (super_dtable);
|
|
}
|
|
else
|
|
dtable = sarray_new (__objc_selector_max_index, 0);
|
|
|
|
__objc_install_methods_in_dtable (dtable, cls->methods);
|
|
|
|
objc_hash_add (&prepared_dtable_table,
|
|
cls,
|
|
dtable);
|
|
}
|
|
|
|
/* This wrapper only exists to allow an easy replacement of the lookup
|
|
implementation and it is expected that the compiler will optimize
|
|
it away. */
|
|
static struct sarray *
|
|
__objc_prepared_dtable_for_class (Class cls)
|
|
{
|
|
struct sarray *dtable = 0;
|
|
assert (cls);
|
|
if (prepared_dtable_table)
|
|
dtable = objc_hash_value_for_key (prepared_dtable_table, cls);
|
|
/* dtable my be nil, since we call this to check whether we are
|
|
currently preparing before we start preparing. */
|
|
return dtable;
|
|
}
|
|
|
|
/* Helper function for messages sent to CLS or implementation pointers
|
|
retrieved from CLS during +initialize before the dtable is
|
|
installed. When a class implicitly initializes another class which
|
|
in turn implicitly invokes methods in this class, before the
|
|
implementation of +initialize of CLS completes, this returns the
|
|
expected implementation. Forwarding remains the responsibility of
|
|
objc_msg_lookup. This function should only be called under the
|
|
global lock. */
|
|
static IMP
|
|
__objc_get_prepared_imp (Class cls,SEL sel)
|
|
{
|
|
struct sarray *dtable;
|
|
IMP imp;
|
|
|
|
assert (cls);
|
|
assert (sel);
|
|
assert (cls->dtable == __objc_uninstalled_dtable);
|
|
dtable = __objc_prepared_dtable_for_class (cls);
|
|
|
|
assert (dtable);
|
|
assert (dtable != __objc_uninstalled_dtable);
|
|
imp = sarray_get_safe (dtable, (size_t) sel->sel_id);
|
|
|
|
/* imp may be Nil if the method does not exist and we may fallback
|
|
to the forwarding implementation later. */
|
|
return imp;
|
|
}
|
|
|
|
/* When this function is called +initialize should be completed. So
|
|
now we are safe to install the dispatch table for the class so that
|
|
they become available for other threads that may be waiting in the
|
|
lock. */
|
|
static void
|
|
__objc_install_prepared_dtable_for_class (Class cls)
|
|
{
|
|
assert (cls);
|
|
assert (cls->dtable == __objc_uninstalled_dtable);
|
|
cls->dtable = __objc_prepared_dtable_for_class (cls);
|
|
|
|
assert (cls->dtable);
|
|
assert (cls->dtable != __objc_uninstalled_dtable);
|
|
objc_hash_remove (prepared_dtable_table, cls);
|
|
}
|