程序实现思想:
Linux 系统的进程号是保存在 /proc 目录下的,一个在 linux 系统上运行的程序,只要是在运行状态下,就必然会在 /proc 目录下写入一批文件,其文件写入的方式如下:
/proc/<进程ID>/文件组
这里面的文件组是指一批文件,不是指一个文件,大致的文件名信息如下:
- dr-xr-xr-x 2 work work 0 09-21 00:51 attr
-
-r-------- 1 work work 0 09-21 00:51 auxv
-
-r--r--r-- 1 work work 0 09-21 00:51 cmdline
-
-rw-r--r-- 1 work work 0 09-21 00:51 coredump_filter
-
-r--r--r-- 1 work work 0 09-21 00:51 cpuset
-
lrwxrwxrwx 1 work work 0 09-21 00:51 cwd -> /home/work/tmp
-
-r-------- 1 work work 0 09-21 00:51 environ
-
lrwxrwxrwx 1 work work 0 09-21 00:51 exe -> /home/work/tmp/tt
-
dr-x------ 2 work work 0 09-21 00:51 fd
-
dr-x------ 2 work work 0 09-21 00:51 fdinfo
-
-r-------- 1 work work 0 09-21 00:51 io
-
-r--r--r-- 1 work work 0 09-21 00:51 limits
-
-rw-r--r-- 1 work work 0 09-21 00:51 loginuid
-
-r--r--r-- 1 work work 0 09-21 00:51 maps
-
-rw------- 1 work work 0 09-21 00:51 mem
-
-r--r--r-- 1 work work 0 09-21 00:51 mounts
-
-r-------- 1 work work 0 09-21 00:51 mountstats
-
-r--r--r-- 1 work work 0 09-21 00:51 numa_maps
-
-rw-r--r-- 1 work work 0 09-21 00:51 oom_adj
-
-r--r--r-- 1 work work 0 09-21 00:51 oom_score
-
lrwxrwxrwx 1 work work 0 09-21 00:51 root -> /
-
-r--r--r-- 1 work work 0 09-21 00:51 schedstat
-
-r--r--r-- 1 work work 0 09-21 00:51 smaps
-
-r--r--r-- 1 work work 0 09-21 00:51 stat
-
-r--r--r-- 1 work work 0 09-21 00:51 statm
-
-r--r--r-- 1 work work 0 09-21 00:51 status
-
dr-xr-xr-x 3 work work 0 09-21 00:51 task
-
-r--r--r-- 1 work work 0 09-21 00:51 wchan
比如,我们运行了一个程序 tt ,假设其进程号为 31673 ,则在 /proc 目录下会形成 /proc/31673/ 的目录,且在其目录下存在上述的文件组。这个里面,我们需要的文件为 status ,该文件保存的是 tt 当前运行得状态,我们来看看这个文件的内容:
- Name: tt
-
State: S (sleeping)
-
SleepAVG: 98%
-
Tgid: 31673
-
Pid: 31673
-
PPid: 7977
-
TracerPid: 0
-
Uid: 500 500 500 500
-
Gid: 500 500 500 500
-
FDSize: 256
-
Groups: 500
-
VmPeak: 3828 kB
-
VmSize: 3664 kB
-
VmLck: 0 kB
-
VmHWM: 300 kB
-
VmRSS: 300 kB
-
VmData: 44 kB
-
VmStk: 88 kB
-
VmExe: 4 kB
-
VmLib: 1448 kB
-
VmPTE: 32 kB
-
StaBrk: 09e04000 kB
-
Brk: 09e04000 kB
-
StaStk: 7ffff0e895d0 kB
-
Threads: 1
-
SigQ: 0/8192
-
SigPnd: 0000000000000000
-
ShdPnd: 0000000000000000
-
SigBlk: 0000000000000000
-
SigIgn: 0000000000000000
-
SigCgt: 0000000000000000
-
CapInh: 0000000000000000
-
CapPrm: 0000000000000000
-
CapEff: 0000000000000000
-
Cpus_allowed: 00000000,00000000,00000000,00000000,00000000,00000000,00000000,ffffffff
-
Mems_allowed: 00000000,00000001
我们发现,这个文件的第一行就是文件名,有了这个东西,那我们就有了获取当前正在运行的特定程序的进程号的办法了。
基本的解决思路是——我们通过遍历 /proc 目录下所有用数字作为目录名的子目录,依次打开这些目录下的 status 文件,获取其中的第一行内容,并从中获取对应的 Name 节的信息,与我们传入的需要查询的程序名字进行比较,如果一致,则该目录的数字就是该程序的进程号。
- /* find_pid_by_name()
-
*
-
* This finds the pid of the specified process.
-
* Currently, it's implemented by rummaging through
-
* the proc filesystem.
-
* Returns a list of all matching PIDs
-
*/
-
-
#include <unistd.h>
-
#include <sys/types.h>
-
#include <sys/stat.h>
-
#include <unistd.h>
-
#include <dirent.h>
-
#include <string.h>
-
#include <stdlib.h>
-
#include <stdio.h>
-
-
#define READ_BUF_SIZE 1024
-
-
-
long* find_pid_by_name( char* pidName)
-
{
-
DIR *dir;
-
struct dirent *next;
-
long* pidList=NULL;
-
int i=0;
-
-
-
dir = opendir("/proc");
-
if (!dir)
-
{
-
//perror_msg_and_die("Cannot open /proc");
-
fprintf(stderr, "Cannot open /proc\n");
-
return pidList;
-
}
-
-
while ((next = readdir(dir)) != NULL)
-
{
-
FILE *status;
-
char filename[READ_BUF_SIZE];
-
char buffer[READ_BUF_SIZE];
-
char name[READ_BUF_SIZE];
-
-
/* Must skip ".." since that is outside /proc */
-
if (strcmp(next->d_name, "..") == 0)
-
continue;
-
-
/* If it isn't a number, we don't want it */
-
if (!isdigit(*next->d_name))
-
continue;
-
sprintf(filename, "/proc/%s/status", next->d_name);
-
-
if (! (status = fopen(filename, "r")) )
-
{
-
continue;
-
}
-
-
//Read first line in /proc/?pid?/status
-
-
if (fgets(buffer, READ_BUF_SIZE-1, status) == NULL)
-
{
-
fclose(status);
-
continue;
-
}
-
-
fclose(status);
-
-
// Buffer should contain a string like "Name: binary_name"a
-
-
sscanf(buffer, "%*s %s", name);
-
-
if ( name != NULL && name[0] != '\0')
-
{
-
if (strcmp(name, pidName) == 0)
-
{
-
pidList=realloc( pidList, sizeof(long) * (i+2));
-
pidList[i++]=strtol(next->d_name, NULL, 0);
-
}
-
-
}
-
-
}
-
-
if (pidList)
-
{
-
pidList[i]=0;
-
}
-
return pidList;
-
}
-
-
-
-
int main ( int argc , char **argv)
-
{
-
long *pid_t=NULL;
-
if ( argc != 2 )
-
{
-
fprintf(stdout,"Usage %s \n",argv[0]);
-
return 0;
-
}
-
-
pid_t = find_pid_by_name( argv[1]);
-
-
while ( pid_t != 0 && *pid_t != 0)
-
{
-
fprintf(stdout,"\n%s is [ %d]\n",argv[1],*pid_t);
-
pid_t++;
-
}
-
return 0;
-
}
这段程序中特别需要注意的是:
main 函数中的 while(pid_t
!= 0
&& *pid_t
!= 0) 这一行,开始的时候俺没有注意,一直报段错误,仔细观察后发现,不仅需要判断这个指针的值是否是等于 0 , 同时还得判断这个指针本身是否等于 0 ,否则会出现查询的程序根本就没有运行时出现段错误,或者是查询的程序运行了,也会出现段错误的情况。
出现段错误是由于 pid_t 指针的值是空值,传给 fprintf 函数时会出现段错误。
==============================================
此程序在需要限制某些程序在特定的时间内只能一次运行的情况下很有好处,这种情况多数出现在金融系统、保险系统、收费系统等核心业务领域,通常在日终账务处理时,绝对不允许在同一个时间内连续两次运行扎帐程序,否则会造成业务数据紊乱。
因此,我们可以在程序运行得时候,首先获取自身的进程号 pid (用 getpid()函数获取),然后通过上述的程序代码,获取当前程序的进程号,如果存在两个,则退出当前的程序,从而保证在一个特定的时间内,扎帐程序只有一个运行。
阅读(597) | 评论(0) | 转发(0) |