added Php::define() function to create constants at run time

1 files changed, 71 insertions, 0 deletions

diff --git a/zend/constantfuncs.cpp b/zend/constantfuncs.cpp
index c350f91..40ea20e 100644
--- a/zend/constantfuncs.cpp
+++ b/zend/constantfuncs.cpp
@@ -61,6 +61,77 @@ Value constant(const std::string &name)
 }
 
 /**
+ *  Define a new constant
+ *  @param  name            Name of the constant
+ *  @param  size            Size of the name
+ *  @param  value           Value of the constant
+ *  @return bool
+ */
+bool define(const char *name, size_t size, const Value &value)
+{
+    // we need the tsrm_ls variable
+    TSRMLS_FETCH();
+    
+    // the constant structure from the zend engine
+    zend_constant constant;
+    
+    // copy the name (note that name_len also includes the end-of-string '\0' byte)
+    constant.name = zend_strndup(name, size);
+    constant.name_len = size + 1;
+    
+    // only scalar values can be used for constants
+    if (value.isScalar())
+    {
+        // make a full copy of the passed in zval
+        constant.value = *value._val;
+        zval_copy_ctor(&constant.value);
+    }
+    else
+    {
+        // we're going to convert the value object into a string, and use that
+        Value str = value.clone(Type::String);
+        
+        // use the copied value
+        constant.value = *str._val;
+        zval_copy_ctor(&constant.value);
+    }
+    
+    // constants are case sensitive (but not persistent, because this is a user
+    // space constant!)
+    constant.flags = CONST_CS;
+
+    // as module number we use a fake module number
+    constant.module_number = PHP_USER_CONSTANT;
+    
+    // register the constant
+    return zend_register_constant(&constant TSRMLS_CC) == SUCCESS;
+}
+
+/**
+ *  Define a new constant
+ *  @param  name
+ *  @param  value
+ *  @return bool
+ */
+bool define(const char *name, const Value &value)
+{
+    // define a constant
+    return define(name, ::strlen(name), value);
+}
+
+/**
+ *  Define a new constant
+ *  @param  name
+ *  @param  value
+ *  @return bool
+ */
+bool define(const std::string &name, const Value &value)
+{
+    // pass on to the other define function
+    return define(name.c_str(), name.size(), value);
+}
+
+/**
  *  Check whether a constant exists
  *  @param  name
  *  @param  size