From 0b70517ba9860360328f88572cb158fb1ee4e1a3 Mon Sep 17 00:00:00 2001
From: Simon Rozman <simon@rozman.si>
Date: Tue, 16 May 2017 13:03:05 +0200
Subject: [PATCH] WlanReasonCodeToString() tested and fixed

---
 include/WinStd/WLAN.h | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/include/WinStd/WLAN.h b/include/WinStd/WLAN.h
index a5f75f1f..112519fa 100644
--- a/include/WinStd/WLAN.h
+++ b/include/WinStd/WLAN.h
@@ -174,8 +174,6 @@ namespace winstd
 template<class _Elem, class _Traits, class _Ax>
 inline DWORD WlanReasonCodeToString(_In_ DWORD dwReasonCode, _Out_ std::basic_string<_Elem, _Traits, _Ax> &sValue, __reserved PVOID pReserved)
 {
-    assert(0); // TODO: Test this code.
-
     DWORD dwSize = 0;
 
     if (!::pfnWlanReasonCodeToString)
@@ -189,7 +187,7 @@ inline DWORD WlanReasonCodeToString(_In_ DWORD dwReasonCode, _Out_ std::basic_st
         DWORD dwResult = ::pfnWlanReasonCodeToString(dwReasonCode, dwSize, szBuffer.get(), pReserved);
         if (dwResult == ERROR_SUCCESS) {
             DWORD dwLength = (DWORD)wcsnlen(szBuffer.get(), dwSize);
-            if (dwLength < dwSize) {
+            if (dwLength < dwSize - 1) {
                 // Buffer was long enough.
                 sValue.assign(szBuffer.get(), dwLength);
                 return ERROR_SUCCESS;