Fix issue 3988 (swoole#3990)

huanghantao · web-flow · commit 36a1dc2045b4 · 2021-01-10T12:43:54.000+08:00
* Fix issue 3988

* Add test

* Don't create socket in enableSSL

* Fix test

* Enable ssl in PHP_METHOD(swoole_client_coro, enableSSL)

* Enable ssl before php_swoole_socket_set_ssl
diff --git a/ext-src/swoole_client_coro.cc b/ext-src/swoole_client_coro.cc
@@ -867,9 +867,11 @@ static PHP_METHOD(swoole_client_coro, close) {
 #ifdef SW_USE_OPENSSL
 static PHP_METHOD(swoole_client_coro, enableSSL) {
     Socket *cli = client_get_ptr(ZEND_THIS);
+
     if (!cli) {
         RETURN_FALSE;
     }
+
     if (cli->get_type() != SW_SOCK_TCP && cli->get_type() != SW_SOCK_TCP6) {
         php_swoole_fatal_error(E_WARNING, "cannot use enableSSL");
         RETURN_FALSE;
@@ -878,6 +880,9 @@ static PHP_METHOD(swoole_client_coro, enableSSL) {
         php_swoole_fatal_error(E_WARNING, "SSL has been enabled");
         RETURN_FALSE;
     }
+
+    cli->enable_ssl_encrypt();
+
     zval *zset = sw_zend_read_property_ex(swoole_client_coro_ce, ZEND_THIS, SW_ZSTR_KNOWN(SW_ZEND_STR_SETTING), 0);
     if (php_swoole_array_length_safe(zset) > 0) {
         php_swoole_socket_set_ssl(cli, zset);
diff --git a/tests/swoole_client_coro/enableSSL.phpt b/tests/swoole_client_coro/enableSSL.phpt
@@ -0,0 +1,35 @@
+--TEST--
+swoole_client_coro: enableSSL
+--SKIPIF--
+<?php require __DIR__ . '/../include/skipif.inc'; ?>
+--FILE--
+<?php
+require __DIR__ . '/../include/bootstrap.php';
+
+use Swoole\Coroutine\Client;
+
+use function Swoole\Coroutine\run;
+
+run(function () {
+    $client = new Client(SWOOLE_SOCK_TCP);
+    $client->connect('www.baidu.com', 443);
+    $client->enableSSL();
+
+    $http = "GET / HTTP/1.0\r\nAccept: */*User-Agent: Lowell-Agent\r\nHost: www.baidu.com\r\nConnection: Close\r\n\r\n";
+    if (!$client->send($http)) {
+        echo "ERROR\n";
+    }
+
+    $content = '';
+    while (true) {
+        $read = $client->recv();
+        if (empty($read)) {
+            break;
+        }
+        $content .= $read;
+    }
+    $client->close();
+    Assert::assert(strpos($content, 'map.baidu.com') !== false);
+});
+?>
+--EXPECT--
