X-Git-Url: http://git.iain.cx/?a=blobdiff_plain;f=imports.cpp;h=f50591cc77a4777084f6fe061d32a60ac6cfb867;hb=d1c0d356f6ea58980a33f2fa2da1b6971dd9f909;hp=230bf437e0ad6f4febffdfda50eddc99e555a186;hpb=b013e3bce4a3abd1e0c02297a2ac8d6c6d66c169;p=nssm.git diff --git a/imports.cpp b/imports.cpp index 230bf43..f50591c 100644 --- a/imports.cpp +++ b/imports.cpp @@ -59,14 +59,19 @@ int get_imports() { if (error != ERROR_PROC_NOT_FOUND) return 2; } + imports.QueryFullProcessImageName = (QueryFullProcessImageName_ptr) get_import(imports.kernel32, QUERYFULLPROCESSIMAGENAME_EXPORT, &error); + if (! imports.QueryFullProcessImageName) { + if (error != ERROR_PROC_NOT_FOUND) return 3; + } + imports.SleepConditionVariableCS = (SleepConditionVariableCS_ptr) get_import(imports.kernel32, "SleepConditionVariableCS", &error); if (! imports.SleepConditionVariableCS) { - if (error != ERROR_PROC_NOT_FOUND) return 3; + if (error != ERROR_PROC_NOT_FOUND) return 4; } imports.WakeConditionVariable = (WakeConditionVariable_ptr) get_import(imports.kernel32, "WakeConditionVariable", &error); if (! imports.WakeConditionVariable) { - if (error != ERROR_PROC_NOT_FOUND) return 4; + if (error != ERROR_PROC_NOT_FOUND) return 5; } } else if (error != ERROR_MOD_NOT_FOUND) return 1; @@ -75,14 +80,14 @@ int get_imports() { if (imports.advapi32) { imports.CreateWellKnownSid = (CreateWellKnownSid_ptr) get_import(imports.advapi32, "CreateWellKnownSid", &error); if (! imports.CreateWellKnownSid) { - if (error != ERROR_PROC_NOT_FOUND) return 6; + if (error != ERROR_PROC_NOT_FOUND) return 7; } imports.IsWellKnownSid = (IsWellKnownSid_ptr) get_import(imports.advapi32, "IsWellKnownSid", &error); if (! imports.IsWellKnownSid) { - if (error != ERROR_PROC_NOT_FOUND) return 7; + if (error != ERROR_PROC_NOT_FOUND) return 8; } } - else if (error != ERROR_MOD_NOT_FOUND) return 5; + else if (error != ERROR_MOD_NOT_FOUND) return 6; return 0; }