From 9cdf2042bd3e1a34c15e120cba85a76ef54eb854 Mon Sep 17 00:00:00 2001
From: Adam Baratz <adambaratz@php.net>
Date: Tue, 13 Sep 2016 15:24:28 -0400
Subject: [PATCH] Allow \PDO::setAttribute() to set pdo_dblib query timeouts

---
 ext/pdo_dblib/dblib_driver.c     |  7 +++----
 ext/pdo_dblib/tests/timeout.phpt | 30 ++++++++++++++++++++++++++++++
 2 files changed, 33 insertions(+), 4 deletions(-)

diff --git a/ext/pdo_dblib/dblib_driver.c b/ext/pdo_dblib/dblib_driver.c
index 64a3646b32..13ddd8a472 100644
--- a/ext/pdo_dblib/dblib_driver.c
+++ b/ext/pdo_dblib/dblib_driver.c
@@ -253,14 +253,13 @@ static int dblib_set_attr(pdo_dbh_t *dbh, zend_long attr, zval *val)
 {
 	switch(attr) {
 		case PDO_ATTR_TIMEOUT:
-			return 0;
-
+		case PDO_DBLIB_ATTR_QUERY_TIMEOUT:
+			return SUCCEED == dbsettime(zval_get_long(val)) ? 1 : 0;
 		case PDO_DBLIB_ATTR_STRINGIFY_UNIQUEIDENTIFIER:
 			((pdo_dblib_db_handle *)dbh->driver_data)->stringify_uniqueidentifier = zval_get_long(val);
 			return 1;
-
 		default:
-			return 1;
+			return 0;
 	}
 }
 
diff --git a/ext/pdo_dblib/tests/timeout.phpt b/ext/pdo_dblib/tests/timeout.phpt
index f92a7da72f..d711f2a3a0 100644
--- a/ext/pdo_dblib/tests/timeout.phpt
+++ b/ext/pdo_dblib/tests/timeout.phpt
@@ -18,6 +18,32 @@ if ($stmt->execute()) {
 	echo "OK\n";
 }
 
+// regular timeout attribute, set after instance created, will affect query timeout, causing this query to fail
+$db = new PDO($dsn, $user, $pass);
+$db->setAttribute(PDO::ATTR_TIMEOUT, 1);
+$stmt = $db->prepare($sql);
+if (!$stmt->execute()) {
+	echo "OK\n";
+
+	// expect some kind of error code
+	if ($stmt->errorCode() != '00000') {
+		echo "OK\n";
+	}
+}
+
+// pdo_dblib-specific timeout attribute, set after instance created, will control query timeout, causing this query to fail
+$db = new PDO($dsn, $user, $pass);
+$db->setAttribute(PDO::DBLIB_ATTR_QUERY_TIMEOUT, 1);
+$stmt = $db->prepare($sql);
+if (!$stmt->execute()) {
+	echo "OK\n";
+
+	// expect some kind of error code
+	if ($stmt->errorCode() != '00000') {
+		echo "OK\n";
+	}
+}
+
 // regular timeout attribute will affect query timeout, causing this query to fail
 $db = new PDO($dsn, $user, $pass, [PDO::ATTR_TIMEOUT => 1]);
 $stmt = $db->prepare($sql);
@@ -49,3 +75,7 @@ OK
 OK
 OK
 OK
+OK
+OK
+OK
+OK
-- 
2.40.0