From 7bb8ac5bd6ed78105d214554713ba1e46e5ddd23 Mon Sep 17 00:00:00 2001
From: midipix <writeonce@midipix.org>
Date: Tue, 11 Jun 2019 23:51:36 +0000
Subject: __ntapi_tt_{i}stat{fs}(): top-level UNC shares might not have a
 trailing slash.

---
 src/fs/ntapi_tt_istat.c | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

(limited to 'src/fs/ntapi_tt_istat.c')

diff --git a/src/fs/ntapi_tt_istat.c b/src/fs/ntapi_tt_istat.c
index 51f1d43..18840fb 100644
--- a/src/fs/ntapi_tt_istat.c
+++ b/src/fs/ntapi_tt_istat.c
@@ -25,6 +25,7 @@ int32_t __stdcall __ntapi_tt_istat(
 	nt_unicode_string *	sdev;
 	wchar16_t *		wch;
 	wchar16_t *		wch_mark;
+	wchar16_t *		wch_cap;
 	uint32_t		hash;
 	uint32_t		written;
 
@@ -108,7 +109,9 @@ int32_t __stdcall __ntapi_tt_istat(
 		hash     = __DEVICE_PATH_PREFIX_HASH;
 	}
 
-	for (; *wch!='\\'; )
+	wch_cap = sdev->buffer + (sdev->strlen / sizeof(wchar16_t));
+
+	for (; wch<wch_cap && *wch!='\\'; )
 		wch++;
 
 	istat->obj_name_strlen = sdev->strlen;
-- 
cgit v1.2.3