RFC - Added the apache_connection_stream() function for CGI WebSockets

sapi/apache2handler/php_functions.c

@@ -205,6 +205,49 @@ PHP_FUNCTION(apache_response_headers)
 }
 /* }}} */
 
+/* {{{ The allows for direct select calls on Apache's connection socket */
+PHP_FUNCTION(apache_connection_stream)
+{
+	php_struct *ctx = SG(server_context);
+	request_rec *r;
+	apr_socket_t *apr_sock;
+	int fd;
+	php_stream *stream;
+
+	ZEND_PARSE_PARAMETERS_NONE();
+
+	if (!ctx) {
+		php_error_docref(NULL, E_WARNING, "Server context is not available");
+		RETURN_FALSE;
+	}
+
+	r = ctx->r;
+	if (!r) {
+		php_error_docref(NULL, E_WARNING, "Request record is not available");
+		RETURN_FALSE;
+	}
+
+	apr_sock = ap_get_conn_socket(r->connection);
+	if (!apr_sock) {
+		php_error_docref(NULL, E_WARNING, "Failed to obtain connection socket");
+		RETURN_FALSE;
+	}
+
+	if (apr_os_sock_get(&fd, apr_sock) != APR_SUCCESS) {
+		php_error_docref(NULL, E_WARNING, "Failed to get native socket descriptor");
+		RETURN_FALSE;
+	}
+
+	stream = php_stream_sock_open_from_socket(fd, NULL);
+	if (!stream) {
+		php_error_docref(NULL, E_WARNING, "Failed to open stream from socket");
+		RETURN_FALSE;
+	}
+
+	php_stream_to_zval(stream, return_value);
+}
+/* }}} */
+
 /* {{{ Get and set Apache request notes */
 PHP_FUNCTION(apache_note)
 {

sapi/cli/php_cli_server.c

@@ -447,6 +447,28 @@ PHP_FUNCTION(apache_response_headers) /* {{{ */
 }
 /* }}} */
 
+PHP_FUNCTION(apache_connection_stream) /* {{{ */
+{
+	php_cli_server_client *client = SG(server_context);
+	php_stream *stream;
+
+	ZEND_PARSE_PARAMETERS_NONE();
+
+	if (!client) {
+		php_error_docref(NULL, E_WARNING, "Server context is not available");
+		RETURN_FALSE;
+	}
+
+	stream = php_stream_sock_open_from_socket(client->sock, NULL);
+	if (!stream) {
+		php_error_docref(NULL, E_WARNING, "Failed to open stream from socket in cli server");
+		RETURN_FALSE;
+	}
+
+	php_stream_to_zval(stream, return_value);
+}
+/* }}} */
+
 /* {{{ cli_server module */
 
 static void cli_server_init_globals(zend_cli_server_globals *cg)
@@ -495,6 +517,7 @@ const zend_function_entry server_additional_functions[] = {
 	PHP_FE(cli_get_process_title,	arginfo_cli_get_process_title)
 	PHP_FE(apache_request_headers,	arginfo_apache_request_headers)
 	PHP_FE(apache_response_headers,	arginfo_apache_response_headers)
+	PHP_FE(apache_connection_stream,	NULL)
 	PHP_FALIAS(getallheaders, apache_request_headers, arginfo_getallheaders)
 	PHP_FE_END
 };