PHPC-190: Provide __wakeUp() to disable unserialize()

Author: bjori

php_phongo_classes.h

@@ -223,6 +223,9 @@ extern PHONGO_API zend_class_entry *php_phongo_regex_ce;
 extern PHONGO_API zend_class_entry *php_phongo_timestamp_ce;
 extern PHONGO_API zend_class_entry *php_phongo_utcdatetime_ce;
 
+/* Shared accross all MongoDB\Driver objects to disable unserialize() */
+PHP_METHOD(Manager, __wakeUp);
+
 PHP_MINIT_FUNCTION(Command);
 PHP_MINIT_FUNCTION(Cursor);
 PHP_MINIT_FUNCTION(CursorId);

src/MongoDB/BulkWrite.c

@@ -254,6 +254,7 @@ static zend_function_entry php_phongo_bulkwrite_me[] = {
 	PHP_ME(BulkWrite, update, ai_BulkWrite_update, ZEND_ACC_PUBLIC)
 	PHP_ME(BulkWrite, delete, ai_BulkWrite_delete, ZEND_ACC_PUBLIC)
 	PHP_ME(BulkWrite, count, ai_BulkWrite_count, ZEND_ACC_PUBLIC)
+	PHP_ME(Manager, __wakeUp, NULL, ZEND_ACC_PUBLIC)
 	PHP_FE_END
 };
 

src/MongoDB/Command.c

@@ -84,6 +84,7 @@ ZEND_END_ARG_INFO();
 
 static zend_function_entry php_phongo_command_me[] = {
 	PHP_ME(Command, __construct, ai_Command___construct, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
+	PHP_ME(Manager, __wakeUp, NULL, ZEND_ACC_PUBLIC)
 	PHP_FE_END
 };
 

src/MongoDB/CursorId.c

@@ -105,6 +105,7 @@ ZEND_END_ARG_INFO();
 static zend_function_entry php_phongo_cursorid_me[] = {
 	PHP_ME(CursorId, __construct, ai_CursorId___construct, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
 	PHP_ME(CursorId, __toString, ai_CursorId___toString, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
+	PHP_ME(Manager, __wakeUp, NULL, ZEND_ACC_PUBLIC)
 	PHP_FE_END
 };
 

src/MongoDB/Manager.c

@@ -277,6 +277,19 @@ PHP_METHOD(Manager, getServers)
 	}
 }
 /* }}} */
+/* {{{ proto void MongoDB\Driver\Manager::__wakeUp()
+ * Throws MongoDB\Driver\RuntimeException as it cannot be serialized */
+PHP_METHOD(Manager, __wakeUp)
+{
+	(void)return_value_ptr; (void)return_value_used; (void)return_value; (void)this_ptr;
+
+	if (zend_parse_parameters_none() == FAILURE) {
+		return;
+	}
+
+	phongo_throw_exception(PHONGO_ERROR_RUNTIME TSRMLS_CC, "%s", "MongoDB\\Driver objects cannot be serialized");
+}
+/* }}} */
 
 /**
  * Manager abstracts a cluster of Server objects (i.e. socket connections).
@@ -350,6 +363,7 @@ static zend_function_entry php_phongo_manager_me[] = {
 	PHP_ME(Manager, executeUpdate, ai_Manager_executeUpdate, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
 	PHP_ME(Manager, executeDelete, ai_Manager_executeDelete, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
 	PHP_ME(Manager, getServers, ai_Manager_getServers, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
+	PHP_ME(Manager, __wakeUp, NULL, ZEND_ACC_PUBLIC)
 	PHP_FE_END
 };
 

src/MongoDB/Query.c

@@ -81,6 +81,7 @@ ZEND_END_ARG_INFO();
 
 static zend_function_entry php_phongo_query_me[] = {
 	PHP_ME(Query, __construct, ai_Query___construct, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
+	PHP_ME(Manager, __wakeUp, NULL, ZEND_ACC_PUBLIC)
 	PHP_FE_END
 };
 

src/MongoDB/Result.c

@@ -190,6 +190,7 @@ static zend_function_entry php_phongo_result_me[] = {
 	PHP_ME(Result, getIterator, ai_Result_getIterator, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
 	PHP_ME(Result, toArray, ai_Result_toArray, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
 	PHP_ME(Result, getServer, ai_Result_getServer, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
+	PHP_ME(Manager, __wakeUp, NULL, ZEND_ACC_PUBLIC)
 	PHP_FE_END
 };
 

src/MongoDB/Server.c

@@ -455,6 +455,7 @@ static zend_function_entry php_phongo_server_me[] = {
 	PHP_ME(Server, isArbiter, ai_Server_isArbiter, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
 	PHP_ME(Server, isHidden, ai_Server_isHidden, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
 	PHP_ME(Server, isPassive, ai_Server_isPassive, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
+	PHP_ME(Manager, __wakeUp, NULL, ZEND_ACC_PUBLIC)
 	PHP_FE_END
 };
 

src/MongoDB/WriteConcernError.c

@@ -118,6 +118,7 @@ static zend_function_entry php_phongo_writeconcernerror_me[] = {
 	PHP_ME(WriteConcernError, getCode, ai_WriteConcernError_getCode, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
 	PHP_ME(WriteConcernError, getInfo, ai_WriteConcernError_getInfo, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
 	PHP_ME(WriteConcernError, getMessage, ai_WriteConcernError_getMessage, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
+	PHP_ME(Manager, __wakeUp, NULL, ZEND_ACC_PUBLIC)
 	PHP_FE_END
 };
 

src/MongoDB/WriteError.c

@@ -140,6 +140,7 @@ static zend_function_entry php_phongo_writeerror_me[] = {
 	PHP_ME(WriteError, getIndex, ai_WriteError_getIndex, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
 	PHP_ME(WriteError, getMessage, ai_WriteError_getMessage, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
 	PHP_ME(WriteError, getInfo, ai_WriteError_getInfo, ZEND_ACC_PUBLIC|ZEND_ACC_FINAL)
+	PHP_ME(Manager, __wakeUp, NULL, ZEND_ACC_PUBLIC)
 	PHP_FE_END
 };