who命令的编写——登录与注销__who, whoami, who am i

最新推荐文章于 2022-10-31 18:05:51 发布

谁不小心的

最新推荐文章于 2022-10-31 18:05:51 发布

阅读量2.9k

点赞数

分类专栏： C语言文章标签： who命令

本文链接：https://blog.csdn.net/trochiluses/article/details/10825061

版权

C语言专栏收录该内容

38 篇文章 1 订阅

订阅专栏

摘要:who是一个linux下重要的系统内建指令，编写who命令，需要了解相关的数据结构struct utmp。同时进一步熟悉结构化数据的读取和访问方法。另外总结了系统调用相关的知识。

本文来源：http://blog.csdn.net/trochiluses/article/details/10825061

1.who程序的编写

#include<stdio.h>
#include<unistd.h>
#include<stdlib.h>
#include<fcntl.h>
#include<utmp.h>
#include<time.h>

#define SHOWHOST
int show_info(struct utmp *utbufp);
void showtime(long);

int main(int arv, char *arg[])
{
  struct utmp current_record;
  int utmpfd;
  int reclen=sizeof(struct utmp);

  if((utmpfd = open(UTMP_FILE,O_RDONLY))==-1)
  {
    perror(UTMP_FILE);
    exit(1);
  }
  
  while(read(utmpfd,¤t_record,reclen)==reclen)
  {
    show_info(¤t_record);
    
  }
  close(utmpfd);
  return 0;

}

int show_info(struct utmp *utbufp)
{
  if(utbufp->ut_type!=USER_PROCESS)
    return;
  printf("%-8.8s",utbufp->ut_name);
  printf(" ");
  printf("%-8.8s",utbufp->ut_line);
  printf(" ");
  showtime(utbufp->ut_time);
  printf(" ");
#ifdef SHOWHOST
    printf("%-8.8s",utbufp->ut_host);
#endif
  printf("\n");
}


void showtime(long timeeval)
{
  char *p;
  p=ctime(&timeeval );
  printf("%12.12s",p+4);
}

1）从who命令的man文档可以查看到它与utmp这个文件有关，进而man utmp这个文件，可以找到这个头文件和utmp这个结构体的一些信息。

2）open，write，close，seek，creat是与文件读写有关的系统调用，我们在下一篇博客里总结它们。

3）与时间有关的函数ctime，我们将用一篇文章总结它们。

4）同样，知道了登录信息是存在UTMPFILE,那么我们可以进一步编写注销信息，实际上就是将改写该文件中的数据而已。下面是注销的代码：

2.注销程序的编写

#include<stdio.h>
#include<unistd.h>
#include<stdlib.h>
#include<utmp.h>
#include<fcntl.h>
#include<sys/types.h>
#include<string.h>
#include<time.h>


int main(int argc, char *argv[])
{
  char *termname=argv[1];
  logout(termname);
  return 0;
}


int logout(const char *ttyname)
{
  int fd; 
  struct utmp utmpunit;
  int utmpunitlen=sizeof(struct utmp);

  if ((fd=open(UTMP_FILE,O_RDWR))==-1)
  {
    perror("can not open file:");
    return -1; 

  }
  
  while(read(fd,&utmpunit,utmpunitlen )==utmpunitlen)
  {
    printf("%s      %s\n",ttyname,utmpunit.ut_line );
    if(strstr(ttyname,utmpunit.ut_line)!=NULL)
    {   
      printf("find the login account\n");
      utmpunit.ut_type=DEAD_PROCESS;
      utmpunit.ut_time=time(NULL);
      if(lseek(fd,-utmpunitlen,SEEK_CUR)!=-1)
      {   
        if(write(fd,&utmpunit,utmpunitlen)==utmpunitlen)
          ;   
      }
      break;
    }
  }
  if(close(fd)==-1)
  {
    perror("can not close file:");
    return -1;
  }

注意，注销的是当前终端，当前终端的名称可以用tty获得，作为命令行参数

3.析who，whoami，who am i

先看看这三个命令的输出信息：

[rocrocket@rocrocket ~]$ whoami

rocrocket

[rocrocket@rocrocket ~]$ who am i

rocrocket pts/3 2008-12-30 13:17 (:0.0)

[rocrocket@rocrocket ~]$ who

rocrocket :0 2008-12-30 09:54

rocrocket pts/0 2008-12-30 09:55 (:0.0)

rocrocket pts/1 2008-12-30 09:57 (:0.0)

rocrocket pts/3 2008-12-30 13:17 (:0.0)

当我用sudo su（或者sudo su -）更换到root用户之后，你再看看：

www.2cto.com

[rocrocket@rocrocket ~]$ sudo su

[root@rocrocket rocrocket]# whoami

root

[root@rocrocket rocrocket]# who am i

rocrocket pts/3 2008-12-30 13:17 (:0.0)

[root@rocrocket rocrocket]# who

rocrocket :0 2008-12-30 09:54

rocrocket pts/0 2008-12-30 09:55 (:0.0)

rocrocket pts/1 2008-12-30 09:57 (:0.0)

rocrocket pts/3 2008-12-30 13:17 (:0.0)

看出区别来了吧，whoami显示的是当前“操作用户”的用户名，而who am i显示的是“登录用户”的用户名。

4.结构化提取utmp数据：

4.1.getutent函数

头文件：#include <utmp.h>
定义函数：struct utmp *getutent(void);
函数说明：getutent()用来从utmp 文件(/var/run/utmp)中读取一项登录数据, 该数据以utmp 结构返回. 第一次调用时会取得第一位用户数据, 之后每调用一次就会返回下一项数据, 直到已无任何数据时返回NULL。

4.2.getutmp函数

#define _GNU_SOURCE /* See feature_test_macros(7) */

#include <utmpx.h>

void getutmp(const struct utmpx *ux, struct utmp *u);
void getutmpx(const struct utmp *u, struct utmpx *ux);

DESCRIPTION
The getutmp() function copies the fields of the utmpx structure pointed
to by ux to the corresponding fields of the utmp structure pointed to
by u. The getutmpx() function performs the converse operation.

5.另外一个命令w：

显示当前用户和他们正在进行的操作

$ w
14:57:22 up 3 days, 18:35, 6 users, load average: 0.38, 0.25, 0.22
USER TTY FROM LOGIN@ IDLE JCPU PCPU WHAT
hyk tty7 Sun20 3days 1:18m 2.17s gnome-session -
hyk pts/1 :0.0 Mon18 17:33m 5.06s 4.77s vncviewer 10.10
hyk pts/2 :0.0 Mon10 16:57m 0.48s 0.01s ssh huangyukun@
hyk pts/3 :0.0 10:43 3:44m 0.83s 2:01 gnome-terminal
hyk pts/4 :0.0 10:45 2.00s 0.91s 0.01s w