lmh188
/
code_scanner
mirror of http://cd.holdtecs.net:8099/scanner/code_scanner
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

添加枚举进程及模块函数

This commit is contained in:
gb 2023-02-08 14:15:50 +08:00
parent 9be3ce881d
commit af5c96ec2e
2 changed files with 122 additions and 82 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

111
common/referer.cpp
View File

@ -159,7 +159,7 @@ void chronograph::reset()
namespace sys_util namespace sys_util
{ {
static bool find_module(const char* path, void* param) static bool find_module(const char* path, bool is_dir, void* param)
{ {
std::string* para = (std::string*)param; std::string* para = (std::string*)param;
@ -184,8 +184,40 @@ namespace sys_util
return true; return true;
} }
} }
typedef struct _enum_proc_cb
{
bool(*on_found)(uint64_t pid, const char* path_name, void* param);
void* param;
}ENPROCCB, * LPENPROCCB;
static bool found_process(const char* path, bool is_dir, void* param)
{
LPENPROCCB cb = (LPENPROCCB)param;
const char* id = strstr(path + 2, "/");
uint64_t pid = 0;
std::string path_name("");
int32_t enum_modules(bool(*on_found)(const char* path_module_name, void* param),// return false to stop enumeratin if (id++ == nullptr)
{
id = path;
}
while (*id)
{
if (*id < '0' || *id > '9')
break;
pid *= 10;
pid += *id - '0';
id++;
}
if (*id)
return true;
path_name = get_module_path(nullptr, pid);
return cb->on_found(pid, path_name.c_str(), cb->param);
}
int32_t enum_modules(bool(*on_found)(const char* path_module_name, bool is_dir, void* param),// return false to stop enumeratin
void* param, // user defined data, passed into callback on_found void* param, // user defined data, passed into callback on_found
unsigned pid // process id, -1 is self unsigned pid // process id, -1 is self
) // return errno ) // return errno
@ -200,7 +232,7 @@ namespace sys_util
} }
int32_t enum_files(const char* dir, // dir path int32_t enum_files(const char* dir, // dir path
bool(*on_found)(const char* path_name, void* param), // return false to stop enumeratin bool(*on_found)(const char* path_name, bool is_dir, void* param), // return false to stop enumeratin
void* param // user defined data, passed into callback on_found void* param // user defined data, passed into callback on_found
, bool recursive , bool recursive
) // return errno ) // return errno
@ -215,11 +247,22 @@ namespace sys_util
while ((ent = readdir(pdir))) while ((ent = readdir(pdir)))
{ {
if (ent->d_type & DT_DIR) if (strcmp(ent->d_name, ".") == 0 || strcmp(ent->d_name, "..") == 0)
continue;
std::string file(dir);
file += "/";
file += ent->d_name;
if ((ent->d_type & DT_DIR) == 0)
file = read_link(file.c_str());
if (!on_found(file.c_str(), ent->d_type & DT_DIR, param))
{ {
if (recursive) ret = 0x5e17;
{ break;
if (strcmp(ent->d_name, ".") && strcmp(ent->d_name, "..")) }
if ((ent->d_type & DT_DIR) && recursive)
{ {
std::string sub(dir); std::string sub(dir);
sub += "/"; sub += "/";
@ -229,25 +272,21 @@ namespace sys_util
break; break;
} }
} }
}
else
{
std::string file(dir);
file += "/";
file += ent->d_name;
if (!on_found(read_link(file.c_str()).c_str(), param))
{
ret = 0x5e17;
break;
}
}
}
closedir(pdir); closedir(pdir);
return ret == 0x5e17 ? 0 : ret; return ret == 0x5e17 ? 0 : ret;
} }
int32_t enum_processes(bool(*on_found)(uint64_t pid, const char* path_name, void* param), void* param)
{
ENPROCCB cb;
cb.on_found = on_found;
cb.param = param;
return enum_files("/proc", found_process, &cb, false);
}
std::string get_module_path(const char* module_name, unsigned pid) // get module full path, nullptr is for main-exe std::string get_module_path(const char* module_name, unsigned pid) // get module full path, nullptr is for main-exe
{ {
std::string param[] = { module_name ? module_name : "", "" }; std::string param[] = { module_name ? module_name : "", "" };
@ -301,34 +340,34 @@ namespace sys_util
int32_t get_memory_info(uint64_t* total, uint64_t* available) int32_t get_memory_info(uint64_t* total, uint64_t* available)
{ {
if(!total && !available) if (!total && !available)
return 0; return 0;
char line[128] = {0}; char line[128] = { 0 };
FILE *src = fopen("/proc/meminfo", "rb"); FILE* src = fopen("/proc/meminfo", "rb");
int32_t count = total && available ? 2 : 1; int32_t count = total && available ? 2 : 1;
unsigned long val = 0; unsigned long val = 0;
if(!src) if (!src)
return log_cls::log_when_err(-1, "fopen('/proc/meminfo', 'rb')", LOG_LEVEL_FATAL); return log_cls::log_when_err(-1, "fopen('/proc/meminfo', 'rb')", LOG_LEVEL_FATAL);
while(fgets(line, sizeof(line) - 1, src)) while (fgets(line, sizeof(line) - 1, src))
{ {
if(sscanf(line, "MemTotal: %ld", &val)) if (sscanf(line, "MemTotal: %ld", &val))
{ {
if(total) if (total)
{ {
*total = val * 1024; *total = val * 1024;
if(--count == 0) if (--count == 0)
break; break;
} }
} }
else if(sscanf(line, "MemFree: %ld", &val)) else if (sscanf(line, "MemFree: %ld", &val))
{ {
if(available) if (available)
{ {
*available = val * 1024; *available = val * 1024;
if(--count == 0) if (--count == 0)
break; break;
} }
} }
@ -342,25 +381,25 @@ namespace sys_util
{ {
std::string str("\0", 80); std::string str("\0", 80);
if(bytes >= SIZE_GB(1)) if (bytes >= SIZE_GB(1))
{ {
double v = bytes * 1.0f / (SIZE_GB(1)); double v = bytes * 1.0f / (SIZE_GB(1));
size_t pos = 0; size_t pos = 0;
sprintf(&str[0], "%.2fGB", v); sprintf(&str[0], "%.2fGB", v);
pos = str.find("."); pos = str.find(".");
while(pos > 3) while (pos > 3)
{ {
pos -= 3; pos -= 3;
str.insert(pos, ","); str.insert(pos, ",");
} }
} }
else if(bytes >= SIZE_MB(1)) else if (bytes >= SIZE_MB(1))
{ {
double v = bytes * 1.0f / (SIZE_MB(1)); double v = bytes * 1.0f / (SIZE_MB(1));
sprintf(&str[0], "%.2fMB", v); sprintf(&str[0], "%.2fMB", v);
} }
else if(bytes >= SIZE_KB(1)) else if (bytes >= SIZE_KB(1))
{ {
double v = bytes * 1.0f / (SIZE_KB(1)); double v = bytes * 1.0f / (SIZE_KB(1));
sprintf(&str[0], "%.2fKB", v); sprintf(&str[0], "%.2fKB", v);

17
common/referer.h
View File

@ -112,16 +112,17 @@ public:
#include <string> #include <string>
namespace sys_util namespace sys_util
{ {
int32_t enum_modules(bool(*on_found)(const char* path_module_name, void* param),// return false to stop enumeratin int32_t enum_modules(bool(*on_found)(const char* path_module_name, bool is_dir, void* param),// return false to stop enumeratin
void* param, // user defined data, passed into callback on_found void* param, // user defined data, passed into callback on_found
unsigned pid = -1 // process id, -1 is self unsigned pid = -1 // process id, -1 is self
); // return errno ); // return errno
int32_t enum_files(const char* dir, // dir path int32_t enum_files(const char* dir // dir path
bool(*on_found)(const char* path_name, void* param), // return false to stop enumeratin , bool(*on_found)(const char* path_name, bool is_dir, void* param)// return false to stop enumeratin
void* param // user defined data, passed into callback on_found , void* param // user defined data, passed into callback on_found
, bool recursive = true // walk recursive , bool recursive = true // walk recursive
); // return errno ); // return errno
int32_t enum_processes(bool(*on_found)(uint64_t pid, const char* path_name, void* param), void* param);
std::string get_module_path(const char* module_name = nullptr std::string get_module_path(const char* module_name = nullptr
, unsigned pid = -1); // get module full path, nullptr is for main-exe , unsigned pid = -1); // get module full path, nullptr is for main-exe
@ -144,16 +145,16 @@ namespace sys_util
int32_t get_inf_file_data(const char* file, size_t line_max, const char* fmt, Args ... args) int32_t get_inf_file_data(const char* file, size_t line_max, const char* fmt, Args ... args)
{ {
std::string buf("\0", line_max + 8); std::string buf("\0", line_max + 8);
FILE *src = fopen(file, "rb"); FILE* src = fopen(file, "rb");
int32_t count = 0; int32_t count = 0;
if(!src) if (!src)
return 0; return 0;
while(fgets(&buf[0], line_max, src)) while (fgets(&buf[0], line_max, src))
{ {
count = sscanf(&buf[0], fmt, args ...); count = sscanf(&buf[0], fmt, args ...);
if(count > 0) if (count > 0)
break; break;
} }
fclose(src); fclose(src);