forked from MacRuby/MacRuby
-
Notifications
You must be signed in to change notification settings - Fork 0
/
blockimp.c
117 lines (99 loc) · 3.59 KB
/
blockimp.c
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
/*
* Author: Landon Fuller <landonf@plausible.coop>
*
* Copyright 2010-2011 Plausible Labs Cooperative, Inc.
* All rights reserved.
*
* Permission is hereby granted, free of charge,
* to any person obtaining a copy of this software and associated documentation
* files (the "Software"), to deal in the Software without restriction,
* including without limitation the rights to use, copy, modify, merge, publish,
* distribute, sublicense, and/or sell copies of the Software, and to permit
* persons to whom the Software is furnished to do so, subject to the following
* conditions:
*
* The above copyright notice and this permission notice shall be included in
* all copies or substantial portions of the Software.
*
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
* SOFTWARE.
*/
#include "blockimp.h"
#include "blockimp_private.h"
#include "trampoline_table.h"
#include <stdio.h>
#include <Block.h>
#pragma mark Trampolines
/* Global lock for our mutable state. Must be held when accessing the trampoline tables. */
static pthread_mutex_t blockimp_lock = PTHREAD_MUTEX_INITIALIZER;
/* Trampoline tables for objc_msgSend_stret() dispatch. */
static pl_trampoline_table *blockimp_table_stret = NULL;
/* Trampoline tables for objc_msgSend() dispatch. */
static pl_trampoline_table *blockimp_table = NULL;
/**
*
*/
IMP pl_imp_implementationWithBlock (void *block) {
#if SUPPORT_APPLE_FALLBACK
/* Prefer Apple's implementation */
if (&imp_implementationWithBlock != NULL)
return imp_implementationWithBlock(block);
#endif
/* Allocate the appropriate trampoline type. */
pl_trampoline *tramp;
struct Block_layout *bl = block;
if (bl->flags & BLOCK_USE_STRET) {
tramp = pl_trampoline_alloc(&pl_blockimp_table_stret_page_config, &blockimp_lock, &blockimp_table_stret);
} else {
tramp = pl_trampoline_alloc(&pl_blockimp_table_page_config, &blockimp_lock, &blockimp_table);
}
/* Configure the trampoline */
void **config = pl_trampoline_data_ptr(tramp->trampoline);
config[0] = Block_copy(block);
config[1] = tramp;
/* Return the function pointer. */
return (IMP) tramp->trampoline;
}
/**
*
*/
void *pl_imp_getBlock(IMP anImp) {
#if SUPPORT_APPLE_FALLBACK
/* Prefer Apple's implementation */
if (&imp_getBlock != NULL) {
return imp_getBlock(anImp);
}
#endif
/* Fetch the config data and return the block reference. */
void **config = pl_trampoline_data_ptr(anImp);
return config[0];
}
/**
*
*/
BOOL pl_imp_removeBlock(IMP anImp) {
#if SUPPORT_APPLE_FALLBACK
/* Prefer Apple's implementation */
if (&imp_removeBlock != NULL)
return imp_removeBlock(anImp);
#endif
/* Fetch the config data */
void **config = pl_trampoline_data_ptr(anImp);
struct Block_layout *bl = config[0];
pl_trampoline *tramp = config[1];
/* Drop the trampoline allocation */
if (bl->flags & BLOCK_USE_STRET) {
pl_trampoline_free(&blockimp_lock, &blockimp_table_stret, tramp);
} else {
pl_trampoline_free(&blockimp_lock, &blockimp_table, tramp);
}
/* Release the block */
Block_release(config[0]);
// TODO - what does this return value mean?
return YES;
}