mirror of
https://github.com/kerberos-io/openalpr-base.git
synced 2025-10-06 11:07:00 +08:00
Added return value checking for readlink
This commit is contained in:
@@ -32,19 +32,24 @@ namespace alpr
|
|||||||
char buffer[2048];
|
char buffer[2048];
|
||||||
memset(buffer, 0, sizeof(buffer));
|
memset(buffer, 0, sizeof(buffer));
|
||||||
|
|
||||||
readlink("/proc/self/exe", buffer, sizeof(buffer));
|
if (readlink("/proc/self/exe", buffer, sizeof(buffer)) > 0)
|
||||||
|
|
||||||
std::stringstream ss;
|
|
||||||
ss << buffer;
|
|
||||||
std::string exeFile = ss.str();
|
|
||||||
std::string directory;
|
|
||||||
|
|
||||||
const size_t last_slash_idx = exeFile.rfind('/');
|
|
||||||
if (std::string::npos != last_slash_idx)
|
|
||||||
{
|
{
|
||||||
directory = exeFile.substr(0, last_slash_idx);
|
std::stringstream ss;
|
||||||
|
ss << buffer;
|
||||||
|
std::string exeFile = ss.str();
|
||||||
|
std::string directory;
|
||||||
|
|
||||||
|
const size_t last_slash_idx = exeFile.rfind('/');
|
||||||
|
if (std::string::npos != last_slash_idx)
|
||||||
|
{
|
||||||
|
directory = exeFile.substr(0, last_slash_idx);
|
||||||
|
}
|
||||||
|
return directory;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return "/";
|
||||||
}
|
}
|
||||||
return directory;
|
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Reference in New Issue
Block a user