feat: add writev() function

Author: ace411

README.md

@@ -55,6 +55,7 @@ class Mrloop
     callable $callback,
   ): void
   public tcpServer(int $port, callable $callback): void
+  public writev(int $fd, string $message): void
   public static parseHttpRequest(string $request, int $headerlimit = 100): iterable
   public static parseHttpResponse(string $response, int $headerlimit = 100): iterable
   public addTimer(float $interval, callable $callback): void
@@ -69,6 +70,7 @@ class Mrloop
 - [`Mrloop::addReadStream`](#mrloopaddreadstream)
 - [`Mrloop::addWriteStream`](#mrloopaddwritestream)
 - [`Mrloop::tcpServer`](#mrlooptcpserver)
+- [`Mrloop::writev`](#mrloopwritev)
 - [`Mrloop::parseHttpRequest`](#mrloopparsehttprequest)
 - [`Mrloop::parseHttpResponse`](#mrloopparsehttpresponse)
 - [`Mrloop::addTimer`](#mrloopaddtimer)
@@ -242,6 +244,7 @@ Instantiates a simple TCP server.
     - **client** (iterable) - An array containing client socket information.
       - **client_addr** (string) - The client IP address.
       - **client_port** (integer) - The client socket port.
+      - **client_fd** (integer) - The client socket file descriptor.
 
 **Return value(s)**
 
@@ -282,6 +285,60 @@ The example above will produce output similar to that in the snippet to follow.
 
 ```
 
+### `Mrloop::writev`
+
+```php
+public Mrloop::writev(int $fd, string $contents): void
+```
+
+Performs vectorized non-blocking write operation on a specified file descriptor.
+
+**Parameter(s)**
+
+- **fd** (integer) - The file descriptor to write to.
+- **contents** (string) - The arbitrary contents to write.
+
+**Return value(s)**
+
+The parser will throw an exception in the event that an invalid file descriptor is encountered and will not return anything otherwise.
+
+```php
+use ringphp\Mrloop;
+
+$loop = Mrloop::init();
+
+$loop->tcpServer(
+  8080,
+  function (string $message, iterable $client) use ($loop) {
+    [
+      'client_addr' => $addr,
+      'client_port' => $port,
+      'client_fd'   => $fd,
+    ] = $client;
+
+    $loop->writev(
+      $fd,
+      \sprintf(
+        "Hello, %s:%d\r\n",
+        $addr,
+        $port,
+      ),
+    );
+  },
+);
+
+echo "Listening on port 8080\n";
+
+$loop->run();
+```
+
+The example above will produce output similar to that in the snippet to follow.
+
+```
+Listening on port 8080
+
+```
+
 ### `Mrloop::parseHttpRequest`
 
 ```php

mrloop_arginfo.h

@@ -50,6 +50,11 @@ ZEND_ARG_TYPE_INFO(0, contents, IS_STRING, 0)
 ZEND_ARG_TYPE_INFO(0, callback, IS_CALLABLE, 0)
 ZEND_END_ARG_INFO()
 
+ZEND_BEGIN_ARG_INFO_EX(arginfo_class_Mrloop_writev, 0, 0, 2)
+ZEND_ARG_TYPE_INFO(0, fd, IS_LONG, 0)
+ZEND_ARG_TYPE_INFO(0, contents, IS_STRING, 0)
+ZEND_END_ARG_INFO()
+
 ZEND_METHOD(Mrloop, init);
 ZEND_METHOD(Mrloop, stop);
 ZEND_METHOD(Mrloop, run);
@@ -61,6 +66,7 @@ ZEND_METHOD(Mrloop, addSignal);
 ZEND_METHOD(Mrloop, addReadStream);
 ZEND_METHOD(Mrloop, addWriteStream);
 ZEND_METHOD(Mrloop, parseHttpResponse);
+ZEND_METHOD(Mrloop, writev);
 
 static const zend_function_entry class_Mrloop_methods[] = {
   PHP_ME(Mrloop, init, arginfo_class_Mrloop_init, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
@@ -74,4 +80,5 @@ static const zend_function_entry class_Mrloop_methods[] = {
                   PHP_ME(Mrloop, addReadStream, arginfo_class_Mrloop_addReadStream, ZEND_ACC_PUBLIC)
                     PHP_ME(Mrloop, addWriteStream, arginfo_class_Mrloop_addWriteStream, ZEND_ACC_PUBLIC)
                       PHP_ME(Mrloop, parseHttpResponse, arginfo_class_Mrloop_parseHttpResponse, ZEND_ACC_PUBLIC | ZEND_ACC_STATIC)
-                        PHP_FE_END};
+                        PHP_ME(Mrloop, writev, arginfo_class_Mrloop_writev, ZEND_ACC_PUBLIC)
+                          PHP_FE_END};

php_mrloop.c

@@ -84,6 +84,13 @@ PHP_METHOD(Mrloop, addWriteStream)
 }
 /* }}} */
 
+/* {{{ proto void Mrloop::writev( int fd [, string contents ] ) */
+PHP_METHOD(Mrloop, writev)
+{
+  php_mrloop_writev(INTERNAL_FUNCTION_PARAM_PASSTHRU);
+}
+/* }}} */
+
 /* {{{ PHP_MINIT_FUNCTION */
 PHP_MINIT_FUNCTION(mrloop)
 {

src/loop.c

@@ -261,25 +261,29 @@ static int php_mrloop_tcp_server_recv(void *conn, int fd, ssize_t nbytes, char *
   array_init(&args[1]);
   add_assoc_string(&args[1], "client_addr", (char *)client->addr);
   add_assoc_long(&args[1], "client_port", client->port);
+  add_assoc_long(&args[1], "client_fd", client->fd);
 
   MRLOOP_G(tcp_cb)->fci.retval = &result;
   MRLOOP_G(tcp_cb)->fci.param_count = 2;
   MRLOOP_G(tcp_cb)->fci.params = args;
 
-  if (zend_call_function(&MRLOOP_G(tcp_cb)->fci, &MRLOOP_G(tcp_cb)->fci_cache) == FAILURE ||
-      Z_TYPE(result) != IS_STRING)
+  if (zend_call_function(&MRLOOP_G(tcp_cb)->fci, &MRLOOP_G(tcp_cb)->fci_cache) == FAILURE)
   {
     PHP_MRLOOP_THROW("There is an error in your callback");
     zval_ptr_dtor(&result);
 
     return 1;
   }
 
-  client->iov.iov_base = Z_STRVAL(result);
-  client->iov.iov_len = Z_STRLEN(result);
+  if (Z_TYPE(result) == IS_STRING)
+  {
+    client->iov.iov_base = Z_STRVAL(result);
+    client->iov.iov_len = Z_STRLEN(result);
+
+    mr_writev(loop, client->fd, &(client->iov), 1);
+    mr_flush(loop);
+  }
 
-  mr_writev(loop, client->fd, &(client->iov), 1);
-  mr_flush(loop);
   zval_ptr_dtor(&result);
 
   return 1;
@@ -629,3 +633,36 @@ static void php_mrloop_add_write_stream(INTERNAL_FUNCTION_PARAMETERS)
 
   return;
 }
+static void php_mrloop_writev(INTERNAL_FUNCTION_PARAMETERS)
+{
+  zend_long fd;
+  zend_string *contents;
+  php_iovec_t iov;
+  php_mrloop_t *this;
+  zval *obj;
+  size_t nbytes;
+
+  obj = getThis();
+
+  ZEND_PARSE_PARAMETERS_START(2, 2)
+  Z_PARAM_LONG(fd)
+  Z_PARAM_STR(contents)
+  ZEND_PARSE_PARAMETERS_END();
+
+  this = PHP_MRLOOP_OBJ(obj);
+
+  if (fcntl((int)fd, F_GETFD) < 0)
+  {
+    PHP_MRLOOP_THROW("Detected invalid file descriptor");
+    mr_stop(this->loop);
+
+    return;
+  }
+
+  nbytes = ZSTR_LEN(contents);
+  iov.iov_base = ZSTR_VAL(contents);
+  iov.iov_len = nbytes;
+
+  mr_writev(this->loop, fd, &iov, 1);
+  mr_flush(this->loop);
+}

src/loop.h

@@ -156,6 +156,8 @@ static void php_mrloop_add_signal(INTERNAL_FUNCTION_PARAMETERS);
 static void php_mrloop_add_read_stream(INTERNAL_FUNCTION_PARAMETERS);
 /* funnels file descriptor in writable stream into event loop and thence executes a non-blocking write operation */
 static void php_mrloop_add_write_stream(INTERNAL_FUNCTION_PARAMETERS);
+/* performs vectorized non-blocking write operation on a specified file descriptor */
+static void php_mrloop_writev(INTERNAL_FUNCTION_PARAMETERS);
 
 zend_class_entry *php_mrloop_ce, *php_mrloop_exception_ce;
 

tests/011.phpt

@@ -0,0 +1,17 @@
+--TEST--
+writev() performs vectorized non-blocking write operation on file descriptor
+--FILE--
+<?php
+
+use ringphp\Mrloop;
+
+$loop = Mrloop::init();
+
+$loop->writev(1, "Hello, user");
+$loop->stop();
+
+$loop->run();
+
+?>
+--EXPECT--
+Hello, user

tests/012.phpt

@@ -0,0 +1,21 @@
+--TEST--
+writev() throws exception on detection of invalid file descriptor
+--FILE--
+<?php
+
+use ringphp\Mrloop;
+
+$loop = Mrloop::init();
+
+try {
+  $loop->writev(987874, "Hello, user");
+} catch (\Throwable $err) {
+  $loop->writev(1, $err->getMessage());
+}
+$loop->stop();
+
+$loop->run();
+
+?>
+--EXPECT--
+Detected invalid file descriptor