Initial import

1 files changed, 77 insertions, 0 deletions

diff --git a/src/sp_harden_rand.c b/src/sp_harden_rand.c
new file mode 100644
index 0000000..e0e35ff
--- /dev/null
+++ b/src/sp_harden_rand.c
@@ -0,0 +1,77 @@
+#include "php_snuffleupagus.h"
+
+extern ZEND_API zend_class_entry *zend_ce_error;
+
+ZEND_DECLARE_MODULE_GLOBALS(snuffleupagus)
+
+/* This function is needed because `rand` and `mt_rand` parameters
+ * are optional, while the ones from `random_int` aren't. */
+static void random_int_wrapper(INTERNAL_FUNCTION_PARAMETERS) {
+  zend_long min, max, result;
+
+  switch (EX_NUM_ARGS()) {
+  case 0:
+    min = 0;
+    max = PHP_MT_RAND_MAX;
+    break;
+  case 1:
+    ZEND_PARSE_PARAMETERS_START_EX(ZEND_PARSE_PARAMS_QUIET, 1, 1);
+    Z_PARAM_LONG(min);
+    ZEND_PARSE_PARAMETERS_END();
+    max = PHP_MT_RAND_MAX;
+    break;
+  case 2:
+  default:
+    ZEND_PARSE_PARAMETERS_START_EX(ZEND_PARSE_PARAMS_QUIET, 0, 2);
+    Z_PARAM_LONG(min);
+    Z_PARAM_LONG(max);
+    ZEND_PARSE_PARAMETERS_END();
+  }
+
+  if (min > max) {
+    if (php_random_int_throw(max, min, &result) == FAILURE) {
+      return;
+    }
+  } else {
+    if (php_random_int_throw(min, max, &result) == FAILURE) {
+      return;
+    }
+  }
+
+  RETURN_LONG(result);
+}
+
+PHP_FUNCTION(sp_rand) {
+  void (*orig_handler)(INTERNAL_FUNCTION_PARAMETERS);
+
+  if ((orig_handler = zend_hash_str_find_ptr(SNUFFLEUPAGUS_G(sp_internal_functions_hook), "rand",
+                                             strlen("rand")))) {
+    /* call the original `rand` function,
+     * since we might no be the only ones to hook it*/
+    orig_handler(INTERNAL_FUNCTION_PARAM_PASSTHRU);
+  }
+
+  random_int_wrapper(INTERNAL_FUNCTION_PARAM_PASSTHRU);
+}
+
+PHP_FUNCTION(sp_mt_rand) {
+  void (*orig_handler)(INTERNAL_FUNCTION_PARAMETERS);
+
+  if ((orig_handler = zend_hash_str_find_ptr(SNUFFLEUPAGUS_G(sp_internal_functions_hook),
+                                             "mt_rand", strlen("mt_rand")))) {
+    /* call the original `mt_rand` function,
+     * since we might no be the only ones to hook it*/
+    orig_handler(INTERNAL_FUNCTION_PARAM_PASSTHRU);
+  }
+
+  random_int_wrapper(INTERNAL_FUNCTION_PARAM_PASSTHRU);
+}
+
+int hook_rand() {
+  TSRMLS_FETCH();
+
+  HOOK_FUNCTION("rand", sp_internal_functions_hook, PHP_FN(sp_rand), false);
+  HOOK_FUNCTION("mt_rand", sp_internal_functions_hook, PHP_FN(sp_mt_rand), false);
+
+  return SUCCESS;
+}