From db8062c0b1147f88362d8d84bcf9e7458d8d89de Mon Sep 17 00:00:00 2001
From: midipix <writeonce@midipix.org>
Date: Wed, 14 Dec 2016 06:25:59 +0000
Subject: __ntapi_pty_query(): properly validate pty_info_class.

---
 src/pty/ntapi_pty_query.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/pty/ntapi_pty_query.c b/src/pty/ntapi_pty_query.c
index 6754c77..7550ad3 100644
--- a/src/pty/ntapi_pty_query.c
+++ b/src/pty/ntapi_pty_query.c
@@ -63,6 +63,8 @@ int32_t	__stdcall __ntapi_pty_query(
 		msg.data.ctlinfo.ctxarg[3] = inherit->any[3];
 
 		hport = pty ? pty : __ntapi_internals()->hport_tty_session;
+	} else {
+		return NT_STATUS_INVALID_INFO_CLASS;
 	}
 
 	if ((status = __ntapi->zw_request_wait_reply_port(hport,&msg,&msg)))
-- 
cgit v1.2.3