C 编程获取正在运行的程序进程号

程序实现思想：
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 <filename>\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（）函数获取），然后通过上述的程序代码，获取当前程序的进程号，如果存在两个，则退出当前的程序，从而保证在一个特定的时间内，扎帐程序只有一个运行。

阅读(153) | 评论(0) | 转发(0) |

0
上一篇：Linux程序前台后台切换

下一篇：time命令含义

相关热门文章

承接自动化测试培训、外包、实...

Solaris PowerTOP 1.0 发布

For STKMonitor

busybox的httpd使用CGI脚本(Bu...

项目小体会

初学UNIX环境高级编程的，关于...

chinaunix博客什么时候可以设...

一个访问量较大网站的服务器。...

收音机驱动与v4l2架构的关系，...

如何将linux驱动改为裸机驱动(...

给主人留下些什么吧！~~

评论热议