From 907d837c99c944b936650b396d32488b8ffdfeff Mon Sep 17 00:00:00 2001 From: Kishikawa Katsumi Date: Mon, 9 Sep 2024 12:20:58 +0900 Subject: [PATCH] Refactor --- Sources/SMBClient/DCERPC/NetShareEnum.swift | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/Sources/SMBClient/DCERPC/NetShareEnum.swift b/Sources/SMBClient/DCERPC/NetShareEnum.swift index d3b2341..db08929 100644 --- a/Sources/SMBClient/DCERPC/NetShareEnum.swift +++ b/Sources/SMBClient/DCERPC/NetShareEnum.swift @@ -71,9 +71,11 @@ struct WStr { init(referentID: UInt32, byteReader: ByteReader) { self.referentID = referentID + maxCount = byteReader.read() offset = byteReader.read() actualCount = byteReader.read() + let valueCount = Int(actualCount) * 2 valueData = byteReader.read(count: valueCount) if valueCount % 4 != 0 { @@ -81,6 +83,7 @@ struct WStr { } else { terminator = 0 } + let valueData = Data(valueData)[0..