diff --git a/src/pty/ntapi_pty_query.c b/src/pty/ntapi_pty_query.c index 21ade12..6754c77 100644 --- a/src/pty/ntapi_pty_query.c +++ b/src/pty/ntapi_pty_query.c @@ -24,7 +24,7 @@ int32_t __stdcall __ntapi_pty_query( uintptr_t * info; nt_pty_inherit_info * inherit; - if ((pty_info_class=NT_PTY_INFORMATION_CAP)) + if (pty_info_class>=NT_PTY_INFORMATION_CAP) return NT_STATUS_INVALID_INFO_CLASS; else if (pty_info_class == NT_PTY_BASIC_INFORMATION) return NT_STATUS_NOT_IMPLEMENTED; diff --git a/src/pty/ntapi_pty_set.c b/src/pty/ntapi_pty_set.c index 9c5af44..27f022f 100644 --- a/src/pty/ntapi_pty_set.c +++ b/src/pty/ntapi_pty_set.c @@ -22,7 +22,7 @@ int32_t __stdcall __ntapi_pty_set( nt_pty_sigctl_msg msg; uintptr_t * info; - if ((pty_info_class=NT_PTY_INFORMATION_CAP)) + if (pty_info_class>=NT_PTY_INFORMATION_CAP) return NT_STATUS_INVALID_INFO_CLASS; else if (pty_info_class == NT_PTY_BASIC_INFORMATION) return NT_STATUS_NOT_IMPLEMENTED;