main.cpp

#include <fstream>
#include <iostream>
#include <sstream>
#include <string>

std::string GetSectionType(const std::string& details, const std::string& perms)
{
    if(details == "[heap]")  return "HEAP";
    if(details == "[stack]") return "STACK";
    if(details == "[vdso]")  return "VDSO";
    if(details == "[vvar]")  return "VVAR";
    if(perms[2] == 'x')      return "TEXT";
    if(perms[1] == 'w')      return "DATA";
    return "OTHER";
}

void ShowMemoryLayout(int pid)
{
    std::ifstream fp("/proc/" + std::to_string(pid) + "/maps");
    if(!fp)
    {
        std::cout << "Unable to open file\n";
        return;
    }

    std::cout << "--------------------------------------------------------\n";
    std::cout << "-------------------- Memory Layout ---------------------\n";
    std::cout << "--------------------------------------------------------\n";
    std::cout << "----- Header -----\n";

    std::string line;
    while(std::getline(fp,line))
    {
        std::istringstream iss(line);
        std::string address, perms, offset, dev, details;
        unsigned long inode = 0;

        iss >> address >> perms >> offset >> dev >> inode >> details;

        std::istringstream addrStream(address);
        unsigned long start = 0, end = 0;
        addrStream >> std::hex >> start;
        addrStream.ignore();  // skip '-'
        addrStream >> std::hex >> end;

        unsigned long sizekb = (end - start) / 1024;

        std::string section = GetSectionType(details,perms);
        if(details.empty())
        {
            details = "Anonymous";
        }

        std::cout << std::hex << start << " " << end << std::dec
                  << " " << sizekb << " " << perms << " " << section << " " << details << "\n";
    }

    std::cout << "--------------------------------------------------------\n";
}

void ShowProcessInformation(int pid)
{
    std::ifstream fp("/proc/" + std::to_string(pid) + "/status");
    if(!fp)
    {
        std::cout << "Unable to open file\n";
        return;
    }

    std::cout << "--------------------------------------------------------\n";
    std::cout << "------------------ Process Information -----------------\n";
    std::cout << "--------------------------------------------------------\n";

    std::string line;
    while(std::getline(fp,line))
    {
        if(line.rfind("Name:",0) == 0 || line.rfind("Pid:",0) == 0 ||
           line.rfind("State:",0) == 0 || line.rfind("Threads:",0) == 0)
        {
            std::cout << line << "\n";
        }
    }

    std::cout << "--------------------------------------------------------\n";
}

int main()
{
    std::cout << "-----------------------------------------------------------\n";
    std::cout << "--------------- Kundan Process Inspector --------------\n";
    std::cout << "-----------------------------------------------------------\n";

    std::cout << "Enter PID of a process that you want to inspect\n";

    int pid = 0;
    std::cin >> pid;

    ShowProcessInformation(pid);
    ShowMemoryLayout(pid);

    return 0;
}