且构网

分享程序员开发的那些事...
且构网 - 分享程序员编程开发的那些事

windows和linux下获取当前程序路径以及cpu数

更新时间:2022-09-16 19:33:12

[cpp] view plaincopy
 
  1. #ifdef WIN32  
  2. #include <Windows.h>  
  3. #else  
  4. #include <stdio.h>  
  5. #include <unistd.h>  
  6. #endif  
  7.   
  8. #include <assert.h>  
  9.   
  10.     std::string getCurrentAppPath()  
  11.     {  
  12. #ifdef WIN32  
  13.         char path[MAX_PATH + 1] = {0};  
  14.         if (GetModuleFileName(NULL, path, MAX_PATH) != 0)  
  15.             return std::string(path);  
  16. #else  
  17.         char path[256] = {0};  
  18.         char filepath[256] = {0};  
  19.         char cmd[256] = {0};  
  20.         FILE* fp = NULL;  
  21.   
  22.         // 设置进程所在proc路径  
  23.         sprintf(filepath, "/proc/%d", getpid());  
  24.         // 将当前路径设为进程路径  
  25.         if(chdir(filepath) != -1)  
  26.         {  
  27.             //指定待执行的shell 命令  
  28.             snprintf(cmd, 256, "ls -l | grep exe | awk '{print $10}'");  
  29.             if((fp = popen(cmd,"r")) == NULL)  
  30.             {  
  31.                 return std::string();  
  32.             }  
  33.             //读取shell命令执行结果到字符串path中  
  34.             if (fgets(path, sizeof(path)/sizeof(path[0]), fp) == NULL)  
  35.             {  
  36.                 pclose(fp);  
  37.                 return std::string();  
  38.             }  
  39.   
  40.             //popen开启的fd必须要pclose关闭  
  41.             pclose(fp);  
  42.             return std::string(path);  
  43.         }  
  44. #endif  
  45.         return std::string();  
  46.     }  
  47.   
  48.     std::size_t getCpuCount()  
  49.     {  
  50. #ifdef WIN32  
  51.         SYSTEM_INFO sysInfo;  
  52.         GetSystemInfo(&sysInfo);  
  53.         return sysInfo.dwNumberOfProcessors;  
  54. #else  
  55.         long cpu_num = sysconf(_SC_NPROCESSORS_ONLN);  
  56.         if (cpu_num == -1)  
  57.         {  
  58.             assert(false);  
  59.             return 0;  
  60.         }  
  61.         // 看两者是否相等  
  62.         assert(cpu_num == sysconf(_SC_NPROCESSORS_CONF));  
  63.         return cpu_num;  
  64. #endif  
  65.     }