Added method Socket::shutdown()

classes/socket.h

@@ -42,6 +42,9 @@ PHP_METHOD(Socket, getLastError);
 PHP_METHOD(Socket, clearError);
 PHP_METHOD(Socket, strerror);
 
+#ifdef HAVE_SHUTDOWN
+PHP_METHOD(Socket, shutdown);
+#endif
 PHP_METHOD(Socket, close);
 
 ZEND_BEGIN_ARG_INFO_EX(Socket___construct, 0, 0, 3)
@@ -137,6 +140,12 @@ ZEND_BEGIN_ARG_WITH_RETURN_TYPE_INFO_EX(Socket_strerror, 0, 1, IS_STRING, 1)
 	ZEND_ARG_TYPE_INFO(0, error, IS_LONG, 0)
 ZEND_END_ARG_INFO()
 
+#ifdef HAVE_SHUTDOWN
+ZEND_BEGIN_ARG_WITH_RETURN_TYPE_INFO_EX(Socket_shutdown, 0, 0, _IS_BOOL, 0)
+	ZEND_ARG_TYPE_INFO(0, how, IS_LONG, 0)
+ZEND_END_ARG_INFO()
+#endif
+
 ZEND_BEGIN_ARG_INFO_EX(Socket_void, 0, 0, 0)
 ZEND_END_ARG_INFO()
 
@@ -162,6 +171,9 @@ zend_function_entry pthreads_socket_methods[] = {
 	PHP_ME(Socket, getPeerName, Socket_getHost, ZEND_ACC_PUBLIC)
 	PHP_ME(Socket, getSockName, Socket_getHost, ZEND_ACC_PUBLIC)
 	PHP_ME(Socket, close, Socket_void, ZEND_ACC_PUBLIC)
+#ifdef HAVE_SHUTDOWN
+	PHP_ME(Socket, shutdown, Socket_shutdown, ZEND_ACC_PUBLIC)
+#endif
 	PHP_ME(Socket, getLastError, Socket_getLastError, ZEND_ACC_PUBLIC)
 	PHP_ME(Socket, clearError, Socket_void, ZEND_ACC_PUBLIC)
 	PHP_ME(Socket, strerror, Socket_strerror, ZEND_ACC_PUBLIC|ZEND_ACC_STATIC)
@@ -399,6 +411,24 @@ PHP_METHOD(Socket, strerror) {
 	pthreads_socket_strerror(error, return_value);
 } /* }}} */
 
+#ifdef HAVE_SHUTDOWN
+/* {{{ proto bool Socket::shutdown(int how = Socket::SHUTDOWN_BOTH) */
+PHP_METHOD(Socket, shutdown) {
+	zend_long how_shutdown = SHUT_RDWR;
+
+	if (zend_parse_parameters(ZEND_NUM_ARGS(), "|l", &how_shutdown) != SUCCESS) {
+		RETURN_FALSE;
+	}
+
+	if (how_shutdown != SHUT_RD && how_shutdown != SHUT_WR && how_shutdown != SHUT_RDWR) {
+		zend_throw_exception(spl_ce_InvalidArgumentException, "Invalid shutdown type", 0);
+		return;
+	}
+
+	pthreads_socket_shutdown(getThis(), how_shutdown, return_value);
+} /* }}} */
+#endif
+
 /* {{{ proto bool Socket::close(void) */
 PHP_METHOD(Socket, close) {
 	if (zend_parse_parameters_none() != SUCCESS) {

examples/stub.php

@@ -689,6 +689,8 @@ public function getPeerName(bool $port = true) : array{}
     public function getSockName(bool $port = true) : array{}
 
     public function close(){}
+
+    public function shutdown(int $how) : bool{}
 
     public function getLastError(bool $clear = false){}
 

php_pthreads.c

@@ -743,6 +743,11 @@ PHP_MINIT_FUNCTION(pthreads)
 	zend_declare_class_constant_long(pthreads_socket_entry,  ZEND_STRL("NO_DATA"), WSANO_DATA);
 	zend_declare_class_constant_long(pthreads_socket_entry,  ZEND_STRL("NO_ADDRESS"), WSANO_ADDRESS);
 #endif
+#ifdef HAVE_SHUTDOWN
+	zend_declare_class_constant_long(pthreads_socket_entry, ZEND_STRL("SHUTDOWN_READ"), SHUT_RD);
+	zend_declare_class_constant_long(pthreads_socket_entry, ZEND_STRL("SHUTDOWN_WRITE"), SHUT_WR);
+	zend_declare_class_constant_long(pthreads_socket_entry, ZEND_STRL("SHUTDOWN_BOTH"), SHUT_RDWR);
+#endif
 
 	/*
 	* Setup object handlers

src/socket.c

@@ -591,6 +591,23 @@ void pthreads_socket_close(zval *object, zval *return_value) {
 	threaded->store.sock->fd = PTHREADS_INVALID_SOCKET;
 }
 
+#ifdef HAVE_SHUTDOWN
+void pthreads_socket_shutdown(zval *object, zend_long how_shutdown, zval *return_value) {
+	pthreads_object_t *threaded =
+		PTHREADS_FETCH_FROM(Z_OBJ_P(object));
+
+	PTHREADS_SOCKET_CHECK(threaded->store.sock);
+
+	if (shutdown(threaded->store.sock->fd, how_shutdown) != SUCCESS) {
+		PTHREADS_SOCKET_ERROR(threaded->store.sock, "Unable to shutdown socket", errno);
+
+		RETURN_FALSE;
+	}
+
+	RETURN_TRUE;
+}
+#endif
+
 void pthreads_socket_set_blocking(zval *object, zend_bool blocking, zval *return_value) {
 	pthreads_object_t *threaded =
 		PTHREADS_FETCH_FROM(Z_OBJ_P(object));

src/socket.h

@@ -46,6 +46,9 @@ void pthreads_socket_read(zval *object, zend_long length, zend_long flags, zend_
 void pthreads_socket_write(zval *object, zend_string *buf, zend_long length, zval *return_value);
 void pthreads_socket_send(zval *object, zend_string *buf, zend_long length, zend_long flags, zval *return_value);
 void pthreads_socket_close(zval *object, zval *return_value);
+#ifdef HAVE_SHUTDOWN
+void pthreads_socket_shutdown(zval *object, zend_long how_shutdown, zval *return_value);
+#endif
 void pthreads_socket_set_blocking(zval *object, zend_bool blocking, zval *return_value);
 void pthreads_socket_get_peer_name(zval *object, zend_bool port, zval *return_value);
 void pthreads_socket_get_sock_name(zval *object, zend_bool port, zval *return_value);

tests/socket-shutdown.phpt

@@ -0,0 +1,34 @@
+--TEST--
+Test of Socket::shutdown() - testing params and separate shutdown of read and write channel
+--SKIPIF--
+<?php
+    if(getenv("SKIP_ONLINE_TESTS")) die("skip online test");
+    if(!method_exists(\Socket::class, 'shutdown')) {
+        die('skip.. Socket::shutdown() doesn\'t exist');
+    }
+--FILE--
+<?php
+    $socket = new \Socket(\Socket::AF_INET, \Socket::SOCK_STREAM, \Socket::SOL_TCP);
+
+    try{
+        $socket->shutdown(4); // invalid
+    }catch(\InvalidArgumentException $e){
+        var_dump($e->getMessage());
+    }
+
+    try {
+        var_dump($socket->shutdown(\Socket::SHUTDOWN_READ)); // close reading
+    } catch(Exception $exception) { var_dump('not connected'); }
+
+    $socket->connect('www.php.net', 80);
+
+    var_dump($socket->shutdown(\Socket::SHUTDOWN_READ)); // close reading
+    var_dump($socket->shutdown(\Socket::SHUTDOWN_WRITE)); // close writing
+
+    $socket->close();
+?>
+--EXPECT--
+string(21) "Invalid shutdown type"
+string(13) "not connected"
+bool(true)
+bool(true)