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

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