linux命令：su的实现

使用su命令可以切换用户，不加参数时默认切换到root用户。

#include<stdio.h>
#include<stdlib.h>
#include<unistd.h>
#include<string.h>
#include<assert.h>
#include <shadow.h>
#include <pwd.h>
#include <sys/types.h>
#include<termios.h>
#include<errno.h>


int main(int argc, char *argv[])
{
	char *user = "root";
	if(argv[1] != NULL)
	{
		user = argv[1];
	}

	printf("Password: ");
	fflush(stdout);
	
/
/设置终端不回显的代码
	struct termios oldflag, newflag;//保存原有终端属性和新设置的终端属性

	tcgetattr(fileno(stdin), &oldflag);//获得终端原有属性并保存在结构体oldflag
	newflag = oldflag;
	newflag.c_lflag &= ~ECHO;
	newflag.c_lflag |= ECHONL;
	if(tcsetattr(fileno(stdin), TCSANOW, &newflag) != 0)
	{
		perror("tcsetattr");
		return -1;
	}

	//
	char passwd[128] = {0}; 
	fgets(passwd, 128, stdin);
	passwd[strlen(passwd) - 1] = 0;
	/
	if(tcsetattr(fileno(stdin), TCSANOW, &oldflag) != 0)
	{
		perror("tcsetattr");
		return -1;
	}
	
	//根据用户名获取系统存储用户的密码信息
	struct spwd *pwd = getspnam(user);
	assert(pwd != NULL);
//
//根据用户密码信息获取加密算法iD以及密钥
	char *p = pwd->sp_pwdp;
	char salt[128] = {0};
	int count = 0, i = 0;
	while(1)
	{
		salt[i] = *p;
		if(salt[i] == '$')
		{
			count++;
		}

		if(count == 3)
		{
			break;
		}

		i++,p++;
	}


	//根据获取的算法ID和密钥，对输入的明文加密
    char *s = crypt(passwd, salt);

	if(strcmp(s, pwd->sp_pwdp) != 0)
	{
		printf("password is error\n");
		exit(0);
	}

	pid_t n = fork();
	assert(n != -1);

	if(n == 0)
	{
		struct passwd *pw = getpwnam(user);
		assert(pw != NULL);
	
		setenv("HOME", pw->pw_dir, 1); // 设置环境变量
		setuid(pw->pw_uid);     // 修改用户
		execl(pw->pw_shell, pw->pw_shell, (char*)0);

		printf("su is fail\n");
		exit(0);
	}
	else
	{
		wait(NULL);
	}
}

 

系统中的su命令，在你输入密码时不会回显，这样更加安全，

下面是实现代码：

#include<termios.h>
#include<errno.h>


/设置终端不回显的代码
	struct termios oldflag, newflag;//保存原有终端属性和新设置的终端属性

	tcgetattr(fileno(stdin), &oldflag);//获得终端原有属性并保存在结构体oldflag
	newflag = oldflag;
	newflag.c_lflag &= ~ECHO;
	newflag.c_lflag |= ECHONL;
	if(tcsetattr(fileno(stdin), TCSANOW, &newflag) != 0)
	{
		perror("tcsetattr");
		return -1;
	}

	//
	char passwd[128] = {0}; 
	fgets(passwd, 128, stdin);
	passwd[strlen(passwd) - 1] = 0;
	/
	if(tcsetattr(fileno(stdin), TCSANOW, &oldflag) != 0)
	{
		perror("tcsetattr");
		return -1;
	}
	

在该命令中，我们要判断用户输入的密码是否正确，在判断密码正确后，要重新设置环境变量，并且修改用户，将终端切换到当前新用户的默认终端下。这才真正完成了su命令。

执行结果如下：