#!/bin/bash
#****************************************************************#
# ScriptName: pm
# Author: osdba
# Create Date: 2010-01-31
# Modify Author:
# Modify Date: 2010-01-31
# Function: show processes memory used
#***************************************************************#

function getpidmem()
{
cat /proc/$1/smaps 2>/dev/null |awk  /
'BEGIN{
   total_size=0;
   total_rss=0;
   total_shared_clean=0;
   total_shared_dirty=0;
   total_private_clean=0;
   total_private_dirty=0;
   total_swap=0;

   mem_size=0;
   mem_rss=0;
   mem_shared_clean=0;
   mem_shared_dirty=0;
   mem_private_clean=0;
   mem_private_dirty=0;
   mem_swap=0;
}
{
   if($1 ~ /^Size:/)
   {
       mem_size=$2;
   }
   else if($1 ~ /^Rss:/)
   {
       mem_rss=$2;
   }
   else if($1 ~ /^Shared_Clean:/)
   {
       mem_shared_clean=$2;
   }
   else if($1 ~ /^Shared_Dirty:/)
   {
       mem_shared_dirty=$2;
   }
   else if($1 ~ /^Private_Clean:/)
   {
       mem_private_clean=$2;
   }
   else if($1 ~ /^Private_Dirty:/)
   {
       mem_private_dirty=$2;
   }
   else if($1 ~ /^Swap:/)
   {
       mem_swap=$2;
   }
   else
   {
       if(ipcmem_start==1)
       {
           ipc_size=mem_size;
           ipc_rss=mem_rss;
           ipc_shared_clean=mem_shared_clean;
           ipc_shared_dirty=mem_shared_dirty;
           ipc_private_clean=mem_private_clean;
           ipc_private_dirty=mem_private_dirty;
           ipc_swap=mem_swap;
       }
       else
       {
           total_size=total_size+mem_size;
           total_rss=total_rss+mem_rss;
           total_shared_clean=total_shared_clean+mem_shared_clean;
           total_shared_dirty=total_shared_dirty+mem_shared_dirty;
           total_private_clean=total_private_clean+mem_private_clean;
           total_private_dirty=total_private_dirty+mem_private_dirty;
           total_swap=total_swap+mem_swap;
       };
       if($6 ~ /^//SYSV/)
       {
           ipcmem_start=1;            
       }
       else
       {
           ipcmem_start=0;       
       };
   };
}
END{
   printf("%15d %15d %15d %15d %15d %15d/n",total_rss,total_private_clean,total_private_dirty,total_shared_clean,total_shared_dirty,total_swap);
}
'
}

function show_all_process_mem()
{
   TMPFILE=`/bin/mktemp -q`
   echo "   pid     RSS(KB)   PRIV_DIRTY PRIV_CLEAN SHR_DIRTY  SHR_CLEAN    SWAP     PAGETABLE      CMDLINE"
   echo "--------- ---------- ---------- ---------- ---------- ---------- ---------- ---------- ----------------"
   ls -ld /proc/[0-9]* |awk '{printf("%s/n",$NF)}' > $TMPFILE
   while read LINE
   do
       PID=`basename $LINE`
       cmdline=`ps -o command -p $PID |tail -1  |cut -b 1-26`
       pagetablesize=`cat /proc/$PID/status |grep "VmPTE:"|awk '{print $2}'`
       if [ "$pagetablesize" = "" ] ; then
           pagetablesize=0
       fi

       meminfoline=`getpidmem $PID`
       rss=`echo $meminfoline |awk '{print $1}'`
       private_clean=`echo $meminfoline |awk '{print $2}'`
       private_dirty=`echo $meminfoline |awk '{print $3}'`
       shared_dirty=`echo $meminfoline |awk '{print $4}'`
       shared_clean=`echo $meminfoline |awk '{print $5}'`
       swap=`echo $meminfoline |awk '{print $6}'`
       total_rss=$((total_rss+rss))
       total_private_clean=$((total_private_clean+private_clean))
       total_private_dirty=$((total_private_dirty+private_dirty))
       total_shared_dirty=$((total_shared_dirty+shared_dirty))
       total_shared_clean=$((total_rss+shared_clean))
       total_swap=$((total_swap+swap))
       total_pagetablesize=$((total_pagetablesize+pagetablesize))
       if [ "$1" = "-a" ] ;then
           echo $PID $meminfoline "$pagetablesize" "$cmdline"|awk /
           '{
             printf("%9d %10d %10d %10d %10d %10d %10d %10d %-26s/n",$1,$2,$3,$4,$5,$6,$7,$8,$9);
           }'
       fi
   done <$TMPFILE
   /bin/rm $TMPFILE

   echo $total_rss $total_private_clean $total_private_dirty $total_shared_dirty $total_shared_clean $total_swap $total_pagetablesize |awk /
   '{
       printf("   ALL    %10d %10d %10d %10d %10d %10d %10d/n",$1,$2,$3,$4,$5,$6,$7);
   }'
}

function print_usage()
{
   echo "usage:"
   echo "  pm.sh -a : display all process memory info"
   echo "  pm.sh -s : display total memory info"
   echo "  pm.sh -p <pid> : display pid memory info"
}

total_rss=0
total_private_clean=0
total_private_dirty=0
total_shared_dirty=0
total_shared_clean=0
total_swap=0
total_pagetablesize=0

if [ $# -lt 1 ];then
  print_usage;
  exit 1
fi

if [ "$1" = "-a" ];then
   show_all_process_mem $1;
   exit 0
fi
if [ "$1" = "-s" ];then
   show_all_process_mem $1;
   exit 0
fi

if [ "$1" = "-p" ];then
   if [ $# -ne 2 ];then
       print_usage;
       exit 1
   fi

   meminfoline=`getpidmem $2`
   cmdline=`ps -o command -p $2 |tail -1  |cut -b 1-26`
   pagetablesize=`cat /proc/$2/status |grep "VmPTE:"|awk '{print $2}'`
   if [ "$pagetablesize" = "" ]; then
       pagetablesize=0
   fi

   echo "   pid     RSS(KB)   PRIV_DIRTY PRIV_CLEAN SHR_DIRTY  SHR_CLEAN    SWAP     PAGETABLE     CMDLINE"
   echo "--------- ---------- ---------- ---------- ---------- ---------- ---------- ---------- ----------------"
   echo $2 $meminfoline "$pagetablesize" "$cmdline"|awk /
   '{
       printf("%9d %10d %10d %10d %10d %10d %10d %10d %-26s/n",$1,$2,$3,$4,$5,$6,$7,$8,$9);
   }'
   exit 0
fi