X-Git-Url: http://git.iain.cx/?a=blobdiff_plain;f=imports.cpp;h=d0970bdc8a894d5e28c87921c5e03bb1ed731e1f;hb=73dbba7a2e6150dea8f58dd2c25ab914a0636308;hp=9ffd486467036de55cb52bfb6a544321064fbe83;hpb=d3b4f286fa65c5a8881d595038505229e467bf49;p=nssm.git

diff --git a/imports.cpp b/imports.cpp
index 9ffd486..d0970bd 100644
--- a/imports.cpp
+++ b/imports.cpp
@@ -32,9 +32,9 @@ FARPROC get_import(HMODULE library, const char *function, unsigned long *error)
     size_t buflen;
     mbstowcs_s(&buflen, NULL, 0, function, _TRUNCATE);
     function_name = (TCHAR *) HeapAlloc(GetProcessHeap(), 0, buflen * sizeof(TCHAR));
-    if (function_name) mbstowcs_s(&buflen, function_name, buflen, function, _TRUNCATE);
+    if (function_name) mbstowcs_s(&buflen, function_name, buflen * sizeof(TCHAR), function, _TRUNCATE);
 #else
-    function_name = function;
+    function_name = (TCHAR *) function;
 #endif
     log_event(EVENTLOG_WARNING_TYPE, NSSM_EVENT_GETPROCADDRESS_FAILED, function_name, error_string(*error), 0);
 #ifdef UNICODE