zend_observer.c 10 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305
  1. /*
  2. +----------------------------------------------------------------------+
  3. | Zend Engine |
  4. +----------------------------------------------------------------------+
  5. | Copyright (c) Zend Technologies Ltd. (http://www.zend.com) |
  6. +----------------------------------------------------------------------+
  7. | This source file is subject to version 2.00 of the Zend license, |
  8. | that is bundled with this package in the file LICENSE, and is |
  9. | available through the world-wide-web at the following url: |
  10. | http://www.zend.com/license/2_00.txt. |
  11. | If you did not receive a copy of the Zend license and are unable to |
  12. | obtain it through the world-wide-web, please send a note to |
  13. | license@zend.com so we can mail you a copy immediately. |
  14. +----------------------------------------------------------------------+
  15. | Authors: Levi Morrison <levim@php.net> |
  16. | Sammy Kaye Powers <sammyk@php.net> |
  17. +----------------------------------------------------------------------+
  18. */
  19. #include "zend_observer.h"
  20. #include "zend_extensions.h"
  21. #include "zend_llist.h"
  22. #include "zend_vm.h"
  23. #define ZEND_OBSERVER_DATA(op_array) \
  24. ZEND_OP_ARRAY_EXTENSION(op_array, zend_observer_fcall_op_array_extension)
  25. #define ZEND_OBSERVER_NOT_OBSERVED ((void *) 2)
  26. #define ZEND_OBSERVABLE_FN(fn_flags) \
  27. (!(fn_flags & ZEND_ACC_CALL_VIA_TRAMPOLINE))
  28. zend_llist zend_observers_fcall_list;
  29. zend_llist zend_observer_error_callbacks;
  30. zend_llist zend_observer_fiber_init;
  31. zend_llist zend_observer_fiber_switch;
  32. zend_llist zend_observer_fiber_destroy;
  33. int zend_observer_fcall_op_array_extension;
  34. ZEND_TLS zend_execute_data *first_observed_frame;
  35. ZEND_TLS zend_execute_data *current_observed_frame;
  36. // Call during minit/startup ONLY
  37. ZEND_API void zend_observer_fcall_register(zend_observer_fcall_init init)
  38. {
  39. zend_llist_add_element(&zend_observers_fcall_list, &init);
  40. }
  41. // Called by engine before MINITs
  42. ZEND_API void zend_observer_startup(void)
  43. {
  44. zend_llist_init(&zend_observers_fcall_list, sizeof(zend_observer_fcall_init), NULL, 1);
  45. zend_llist_init(&zend_observer_error_callbacks, sizeof(zend_observer_error_cb), NULL, 1);
  46. zend_llist_init(&zend_observer_fiber_init, sizeof(zend_observer_fiber_init_handler), NULL, 1);
  47. zend_llist_init(&zend_observer_fiber_switch, sizeof(zend_observer_fiber_switch_handler), NULL, 1);
  48. zend_llist_init(&zend_observer_fiber_destroy, sizeof(zend_observer_fiber_destroy_handler), NULL, 1);
  49. zend_observer_fcall_op_array_extension = -1;
  50. }
  51. ZEND_API void zend_observer_post_startup(void)
  52. {
  53. if (zend_observers_fcall_list.count) {
  54. /* We don't want to get an extension handle unless an ext installs an observer
  55. * Allocate each a begin and an end pointer */
  56. zend_observer_fcall_op_array_extension =
  57. zend_get_op_array_extension_handles("Zend Observer", (int) zend_observers_fcall_list.count * 2);
  58. /* ZEND_CALL_TRAMPOLINE has SPEC(OBSERVER) but zend_init_call_trampoline_op()
  59. * is called before any extensions have registered as an observer. So we
  60. * adjust the offset to the observed handler when we know we need to observe. */
  61. ZEND_VM_SET_OPCODE_HANDLER(&EG(call_trampoline_op));
  62. /* ZEND_HANDLE_EXCEPTION also has SPEC(OBSERVER) and no observer extensions
  63. * exist when zend_init_exception_op() is called. */
  64. ZEND_VM_SET_OPCODE_HANDLER(EG(exception_op));
  65. ZEND_VM_SET_OPCODE_HANDLER(EG(exception_op) + 1);
  66. ZEND_VM_SET_OPCODE_HANDLER(EG(exception_op) + 2);
  67. }
  68. }
  69. ZEND_API void zend_observer_activate(void)
  70. {
  71. first_observed_frame = NULL;
  72. current_observed_frame = NULL;
  73. }
  74. ZEND_API void zend_observer_deactivate(void)
  75. {
  76. // now empty and unused, but kept for ABI compatibility
  77. }
  78. ZEND_API void zend_observer_shutdown(void)
  79. {
  80. zend_llist_destroy(&zend_observers_fcall_list);
  81. zend_llist_destroy(&zend_observer_error_callbacks);
  82. zend_llist_destroy(&zend_observer_fiber_switch);
  83. }
  84. static void zend_observer_fcall_install(zend_execute_data *execute_data)
  85. {
  86. zend_llist *list = &zend_observers_fcall_list;
  87. zend_function *function = execute_data->func;
  88. zend_op_array *op_array = &function->op_array;
  89. ZEND_ASSERT(function->type != ZEND_INTERNAL_FUNCTION);
  90. ZEND_ASSERT(RUN_TIME_CACHE(op_array));
  91. zend_observer_fcall_begin_handler *begin_handlers = (zend_observer_fcall_begin_handler *)&ZEND_OBSERVER_DATA(op_array);
  92. zend_observer_fcall_end_handler *end_handlers = (zend_observer_fcall_end_handler *)begin_handlers + list->count, *end_handlers_start = end_handlers;
  93. *begin_handlers = ZEND_OBSERVER_NOT_OBSERVED;
  94. *end_handlers = ZEND_OBSERVER_NOT_OBSERVED;
  95. for (zend_llist_element *element = list->head; element; element = element->next) {
  96. zend_observer_fcall_init init;
  97. memcpy(&init, element->data, sizeof init);
  98. zend_observer_fcall_handlers handlers = init(execute_data);
  99. if (handlers.begin) {
  100. *(begin_handlers++) = handlers.begin;
  101. }
  102. if (handlers.end) {
  103. *(end_handlers++) = handlers.end;
  104. }
  105. }
  106. // end handlers are executed in reverse order
  107. for (--end_handlers; end_handlers_start < end_handlers; --end_handlers, ++end_handlers_start) {
  108. zend_observer_fcall_end_handler tmp = *end_handlers;
  109. *end_handlers = *end_handlers_start;
  110. *end_handlers_start = tmp;
  111. }
  112. }
  113. static void ZEND_FASTCALL _zend_observe_fcall_begin(zend_execute_data *execute_data)
  114. {
  115. if (!ZEND_OBSERVER_ENABLED) {
  116. return;
  117. }
  118. zend_op_array *op_array = &execute_data->func->op_array;
  119. uint32_t fn_flags = op_array->fn_flags;
  120. if (!ZEND_OBSERVABLE_FN(fn_flags)) {
  121. return;
  122. }
  123. zend_observer_fcall_begin_handler *handler = (zend_observer_fcall_begin_handler *)&ZEND_OBSERVER_DATA(op_array);
  124. if (!*handler) {
  125. zend_observer_fcall_install(execute_data);
  126. }
  127. zend_observer_fcall_begin_handler *possible_handlers_end = handler + zend_observers_fcall_list.count;
  128. zend_observer_fcall_end_handler *end_handler = (zend_observer_fcall_end_handler *)possible_handlers_end;
  129. if (*end_handler != ZEND_OBSERVER_NOT_OBSERVED) {
  130. if (first_observed_frame == NULL) {
  131. first_observed_frame = execute_data;
  132. }
  133. current_observed_frame = execute_data;
  134. }
  135. if (*handler == ZEND_OBSERVER_NOT_OBSERVED) {
  136. return;
  137. }
  138. do {
  139. (*handler)(execute_data);
  140. } while (++handler != possible_handlers_end && *handler != NULL);
  141. }
  142. ZEND_API void ZEND_FASTCALL zend_observer_generator_resume(zend_execute_data *execute_data)
  143. {
  144. _zend_observe_fcall_begin(execute_data);
  145. }
  146. ZEND_API void ZEND_FASTCALL zend_observer_fcall_begin(zend_execute_data *execute_data)
  147. {
  148. ZEND_ASSUME(execute_data->func);
  149. if (!(execute_data->func->common.fn_flags & ZEND_ACC_GENERATOR)) {
  150. _zend_observe_fcall_begin(execute_data);
  151. }
  152. }
  153. static inline bool zend_observer_is_skipped_frame(zend_execute_data *execute_data) {
  154. zend_function *func = execute_data->func;
  155. if (!func || func->type == ZEND_INTERNAL_FUNCTION || !ZEND_OBSERVABLE_FN(func->common.fn_flags)) {
  156. return true;
  157. }
  158. zend_observer_fcall_end_handler end_handler = (&ZEND_OBSERVER_DATA(&func->op_array))[zend_observers_fcall_list.count];
  159. if (end_handler == NULL || end_handler == ZEND_OBSERVER_NOT_OBSERVED) {
  160. return true;
  161. }
  162. return false;
  163. }
  164. ZEND_API void ZEND_FASTCALL zend_observer_fcall_end(zend_execute_data *execute_data, zval *return_value)
  165. {
  166. zend_function *func = execute_data->func;
  167. if (!ZEND_OBSERVER_ENABLED
  168. || !ZEND_OBSERVABLE_FN(func->common.fn_flags)) {
  169. return;
  170. }
  171. zend_observer_fcall_end_handler *handler = (zend_observer_fcall_end_handler *)&ZEND_OBSERVER_DATA(&func->op_array) + zend_observers_fcall_list.count;
  172. // TODO: Fix exceptions from generators
  173. // ZEND_ASSERT(fcall_data);
  174. if (!*handler || *handler == ZEND_OBSERVER_NOT_OBSERVED) {
  175. return;
  176. }
  177. zend_observer_fcall_end_handler *possible_handlers_end = handler + zend_observers_fcall_list.count;
  178. do {
  179. (*handler)(execute_data, return_value);
  180. } while (++handler != possible_handlers_end && *handler != NULL);
  181. if (first_observed_frame == execute_data) {
  182. first_observed_frame = NULL;
  183. current_observed_frame = NULL;
  184. } else {
  185. zend_execute_data *ex = execute_data->prev_execute_data;
  186. while (ex && zend_observer_is_skipped_frame(ex)) {
  187. ex = ex->prev_execute_data;
  188. }
  189. current_observed_frame = ex;
  190. }
  191. }
  192. ZEND_API void zend_observer_fcall_end_all(void)
  193. {
  194. zend_execute_data *ex = current_observed_frame;
  195. while (ex != NULL) {
  196. if (ex->func && ex->func->type != ZEND_INTERNAL_FUNCTION) {
  197. zend_observer_fcall_end(ex, NULL);
  198. }
  199. ex = ex->prev_execute_data;
  200. }
  201. }
  202. ZEND_API void zend_observer_error_register(zend_observer_error_cb cb)
  203. {
  204. zend_llist_add_element(&zend_observer_error_callbacks, &cb);
  205. }
  206. void zend_observer_error_notify(int type, zend_string *error_filename, uint32_t error_lineno, zend_string *message)
  207. {
  208. for (zend_llist_element *element = zend_observer_error_callbacks.head; element; element = element->next) {
  209. zend_observer_error_cb callback = *(zend_observer_error_cb *) (element->data);
  210. callback(type, error_filename, error_lineno, message);
  211. }
  212. }
  213. ZEND_API void zend_observer_fiber_init_register(zend_observer_fiber_init_handler handler)
  214. {
  215. zend_llist_add_element(&zend_observer_fiber_init, &handler);
  216. }
  217. ZEND_API void zend_observer_fiber_switch_register(zend_observer_fiber_switch_handler handler)
  218. {
  219. zend_llist_add_element(&zend_observer_fiber_switch, &handler);
  220. }
  221. ZEND_API void zend_observer_fiber_destroy_register(zend_observer_fiber_destroy_handler handler)
  222. {
  223. zend_llist_add_element(&zend_observer_fiber_destroy, &handler);
  224. }
  225. ZEND_API void ZEND_FASTCALL zend_observer_fiber_init_notify(zend_fiber_context *initializing)
  226. {
  227. zend_llist_element *element;
  228. zend_observer_fiber_init_handler callback;
  229. for (element = zend_observer_fiber_init.head; element; element = element->next) {
  230. callback = *(zend_observer_fiber_init_handler *) element->data;
  231. callback(initializing);
  232. }
  233. }
  234. ZEND_API void ZEND_FASTCALL zend_observer_fiber_switch_notify(zend_fiber_context *from, zend_fiber_context *to)
  235. {
  236. zend_llist_element *element;
  237. zend_observer_fiber_switch_handler callback;
  238. for (element = zend_observer_fiber_switch.head; element; element = element->next) {
  239. callback = *(zend_observer_fiber_switch_handler *) element->data;
  240. callback(from, to);
  241. }
  242. }
  243. ZEND_API void ZEND_FASTCALL zend_observer_fiber_destroy_notify(zend_fiber_context *destroying)
  244. {
  245. zend_llist_element *element;
  246. zend_observer_fiber_destroy_handler callback;
  247. for (element = zend_observer_fiber_destroy.head; element; element = element->next) {
  248. callback = *(zend_observer_fiber_destroy_handler *) element->data;
  249. callback(destroying);
  250. }
  251. }