process_prctl.cpp

/*!
  @author goldenhawking@163.com
  @date 2017-02-11
  */
#include "process_prctl.h"
#include <QFile>
#include <QTextStream>
#include <QDebug>
#ifdef WIN32
#include <windows.h>
#include <psapi.h>
#endif
#ifdef linux
#include <sys/resource.h>
#include <unistd.h>
#endif

namespace TASKBUS {

#ifdef WIN32
	const int pnice_min = 0;
	const int pnice_max = 5;
	const int pnice_norm = 2;
#endif

#ifdef linux
	const int pnice_min = PRIO_MIN;
	const int pnice_max = PRIO_MAX;
	const int pnice_norm = 0;

#endif

	void set_proc_nice(QProcess * p, int nice)
	{
#ifdef WIN32
		qint64 id = p->processId();
		HANDLE hProcess=OpenProcess(PROCESS_ALL_ACCESS,
									FALSE,id);
		if (hProcess)
		{
			const DWORD dw_pn []= {
				IDLE_PRIORITY_CLASS,
				BELOW_NORMAL_PRIORITY_CLASS,
				NORMAL_PRIORITY_CLASS,
				ABOVE_NORMAL_PRIORITY_CLASS,
				HIGH_PRIORITY_CLASS,
				REALTIME_PRIORITY_CLASS
			};
			if (nice<0)		nice = 0;
			if (nice>5)		nice = 5;
			SetPriorityClass(hProcess,dw_pn[nice]);
			CloseHandle(hProcess);
		}

#endif

#ifdef linux
		Q_PID id = p->pid();
		if (nice<-PRIO_MIN)	nice = PRIO_MIN;
		if (nice>PRIO_MAX)	nice = PRIO_MAX;
		setpriority(PRIO_PROCESS,id,nice);
#endif


	}

	void set_proc_nice (int nice)
	{
#ifdef WIN32
		const DWORD dw_pn []= {
			IDLE_PRIORITY_CLASS,
			BELOW_NORMAL_PRIORITY_CLASS,
			NORMAL_PRIORITY_CLASS,
			ABOVE_NORMAL_PRIORITY_CLASS,
			HIGH_PRIORITY_CLASS,
			REALTIME_PRIORITY_CLASS
		};
		if (nice<0)		nice = 0;
		if (nice>5)		nice = 5;
		SetPriorityClass(GetCurrentProcess(),dw_pn[nice]);
#endif

#ifdef linux
		Q_PID id = getpid();
		if (nice<-PRIO_MIN)	nice = PRIO_MIN;
		if (nice>PRIO_MAX)	nice = PRIO_MAX;
		setpriority(PRIO_PROCESS,id,nice);
#endif
	}

	bool get_memory (qint64 p ,tagMemoryInfo * info)
	{
#ifdef linux
		info->pid = p;
		info->phandle = p;
		QString strFm;
		strFm.sprintf("/proc/%lld/status",p);
		QFile fin(strFm);
		if (fin.open(QIODevice::ReadOnly)==false)
			return false;
		QTextStream s(&fin);
		int hit = 0;
		QString l = s.readLine();
		while (l.size() && hit<2)
		{
			QStringList lst = l.split(":",Qt::SkipEmptyParts);
			if (lst.size()>=2)
			{
				QString keystr = lst.first().trimmed();
				lst.pop_front();
				QString v = lst.first().trimmed();
				if (keystr=="Name")
				{
					info->m_name = v;
					++hit;
				}
				if (keystr=="VmRSS")
				{
					v = v.toUpper();
					++hit;
					if (v.indexOf("K"))
					{
						QString digiga = v.left(v.indexOf("K"));
						info->m_memsize = digiga.toDouble()*1024;
					}
					else if (v.indexOf("M"))
					{
						QString digiga = v.left(v.indexOf("M"));
						info->m_memsize = digiga.toDouble()*1024*1024;
					}
					else if (v.indexOf("G"))
					{
						QString digiga = v.left(v.indexOf("G"));
						info->m_memsize = digiga.toDouble()*1024*1024*1024;
					}
					else if (v.indexOf("B"))
					{
						QString digiga = v.left(v.indexOf("B"));
						info->m_memsize = digiga.toDouble();
					}
					else if (v.toDouble()>0)
						info->m_memsize = v.toDouble();
					else
						--hit;
				}

			}
			l = s.readLine();
		}
		fin.close();
		if (hit>=2)
			return true;
#endif

#ifdef WIN32
		char ch_module[MAX_PATH] = {0,0,0,0};
		if (p==-1)
		{
			GetModuleFileNameA(0,ch_module,MAX_PATH);
			PROCESS_MEMORY_COUNTERS pmc;
			GetProcessMemoryInfo(GetCurrentProcess(),&pmc,sizeof(pmc));
			info->m_memsize  = pmc.WorkingSetSize;
			info->pid = GetProcessId(GetCurrentProcess());
			info->phandle = (qint64)GetCurrentProcess();
		}
		else if (p)
		{

			HANDLE hProcess=OpenProcess(PROCESS_QUERY_INFORMATION|
										PROCESS_VM_READ,
										FALSE,p);
			info->pid = 0;
			if (hProcess)
			{
				HMODULE hMod;
				DWORD cbNeeded;

				if ( EnumProcessModules( hProcess, &hMod, sizeof(hMod),
										 &cbNeeded) )
				{
					GetModuleFileNameExA(hProcess,hMod,ch_module,MAX_PATH);
					PROCESS_MEMORY_COUNTERS pmc;
					GetProcessMemoryInfo(hProcess,&pmc,sizeof(pmc));
					info->m_memsize  = pmc.WorkingSetSize;
				}
				info->pid = p;

			}
			//printf( "%s (Process ID: %u)\n", szProcessName, processID );

			CloseHandle( hProcess );
			if (info->pid==0)
				return false;
			info->phandle = (qint64)p;

		}
		char * pt =ch_module + strlen(ch_module)-1;
		while (pt > ch_module)
		{
			if (*pt == '\\' || *pt=='/'  || *pt==':')
				break;
			--pt;
		}
		++pt;
		info->m_name = QString::fromLocal8Bit(pt);

		return true;
#endif
		return false;
	}

	qint64 get_procid(QProcess * p)
	{
#ifdef linux
		if (p)
			return p->pid();
		else
			return getpid();
#endif
#ifdef WIN32
		if (p)
			return (qint64)p->processId();
		else
			return (qint64)GetCurrentProcess();
#endif
		return 0;
	}
}