c++ 对特定进程的内存监控

在工具实现的过程中，遇到了内存爆了的问题，部分模型的规模可以达到10的100次方方甚至1000次方。
（工具的主要算法涉及到了递归，递归深度会很深，所以也用到了ulimit修改栈空间来缓解爆栈的问题，治标不治本。）
有一个循环，这个循环迭代16次，但是可能程序在某一次迭代过程中，已经耗光了内存，再申请空间的时候，出现异常，导致将整个进程杀死。而我的想法是，当某一个迭代过程中，出现无法再申请内存的情况，就输出内存申请失败的提示信息，并跳过它，进行下一次的迭代，而不是将整个进程kill。

解决方法：
创建一个子线程，让它每隔1秒去检查当前进程的内存占用，如果超过一定的比例，则跳过此次的迭代。
检查当前进程的内存占用，是用shell脚本实现的，使用shell脚本进行对内存的查询只需要两三行即可。
当执行shell脚本的时候，要获取其输出结果，也就是该进程的内存占用。
具体代码如下

 1 void detect_memory()
 2 {
 3 
 4     for(;;)
 5     {
 6         FILE *pf;
 7         char buffer[20];
 8 
 9         //执行内存检测脚本
10         pf = popen("sh detect_memory.sh", "r");
11 
12         if(pf!=NULL){
13 
14             //buffer：enPAC进程占用的内存(KB)
15             //fread 将pf指向的内容，读sizeof(buffer)个字节，读一次，读到buffer中
16             fread(buffer, sizeof(buffer), 1, pf);
17 
18             //used:将buffer转成整型，单位为MB
19             short int used = atoi(buffer)/1024;
20             pclose(pf);
21 
22             //判断enPAC占用的内存，超过一定比例，则将memory_flag=false
23             //同时break，让dfs（）进行return，结束此公式的search
24             if(100*used/total_mem > 70)
25             {
26                 memory_flag = false;
27                 cout<<"detect memory over the size  given"<<endl;
28                 break;
29             }
30         }else{
31             cout<<"未能检测到enPAC进程所占内存"<<endl;
32             pclose(pf);
33         }
34 
35         //每隔5秒，对enPAC占用内存进行查询
36         sleep(1);
37     }
38 
39 
40 }
41 
42 detect_mem_thread = thread(&Product_Automata::detect_memory,this);  //创建一个子线程进行内存监控

View Code

注：popen总是和pclose一起出现被使用的。popen() 创建一个管道，通过fork或者invoke一个子进程,然后执行command。popen返回该FIFO数据流的指针。fread(buffer, sizeof(buffer), 1, pf)，fread 将pf指向的内容，读sizeof(buffer)个字节，读一次，读到buffer中。

shell脚本如下：
#!/bin/bash
process="enPAC" #进程名字
PID=$(ps x | grep $process | grep -v grep | awk '{print $1}')
size=` cat /proc/${PID}/status | grep RSS | tr -cd "[0-9]"`
echo $size