调用 PAM 进行认证

1// gcc main.c -lpam
2
3#include <stdio.h>
4#include <stdlib.h>
5#include <string.h>
6#include <unistd.h>
7#include <termio.h>
8#include <security/pam_appl.h>
9
10/***************************************
11 * @brief 开关回显 
12 * @param[in] fd 文件描述符
13 * @param[in] off 1-关闭回显，0-开启回显
14 * *************************************/
15static void echoOff(int fd, int off)
16{
17    struct termio tty;
18    (void) ioctl(fd, TCGETA, &tty);
19    if (off) 
20    {
21        tty.c_lflag &= ~(ECHO | ECHOE | ECHOK | ECHONL);
22        (void) ioctl(fd, TCSETAF, &tty);
23    }
24    else
25    {
26        tty.c_lflag |= (ECHO | ECHOE | ECHOK | ECHONL);
27        (void) ioctl(fd, TCSETAW, &tty);
28    }
29}
30
31/***************************************
32 * @brief 关闭标准输入的回显 
33 * *************************************/
34static void echoOffStdin()
35{
36    echoOff(fileno(stdin), 1);
37}
38
39/***************************************
40 * @brief 开启标准输入的回显 
41 * *************************************/
42static void echoOnStdin()
43{
44    echoOff(fileno(stdin), 0);
45}
46
47/***************************************
48 * @brief 读取一行输入
49 * @return 输入的字符串
50 * *************************************/
51static char* readline()
52{
53    struct termio tty;
54    char input[PAM_MAX_RESP_SIZE];
55
56    /* 读取字符直到回车 */
57    flockfile(stdin);
58    int i = 0;
59    for (; i < PAM_MAX_RESP_SIZE; i++)
60    {
61        int ch = getchar_unlocked();
62        if (ch == '\n' || ch == '\r' ||ch == EOF)
63            break;
64        input[i] = ch;
65    }
66    funlockfile(stdin);
67    input[i] = '\0';
68
69    return (strdup(input));
70}
71
72/**************************************************
73 * @brief PAM对话回调函数
74 * @param[in] num_msg PAM发送过来的消息数量
75 * @param[in] msg PAM发送过来的消息数据 
76 * @param[out] resp 发回给PAM的应答
77 * @param[in] appdata_ptr 用户参数 
78 * @return 状态 
79 * ************************************************/
80static int conversation(int num_msg, const struct pam_message** msg, struct pam_response **resp, void *appdata_ptr)
81{
82    // 检查参数
83    if (num_msg <= 0 || num_msg >= PAM_MAX_MSG_SIZE)
84    {
85        fprintf(stderr, "invalid num_msg(%d)\n", num_msg);
86        return PAM_CONV_ERR;
87    }
88
89    // 给回复消息分配内存，这里分配的内存，由外层的 PAM 框架释放
90    // TODO: 发生错误的时候需要手动释放
91    if ((resp[0] = malloc(num_msg * sizeof(struct pam_response))) == NULL)
92    {
93        fprintf(stderr, "bad alloc\n");
94        return PAM_BUF_ERR;
95    }
96
97    // 处理PAM发来的消息并应答
98    for(int i = 0; i < num_msg; i++)
99    {
100        const struct pam_message* m = *msg + i;
101        struct pam_response* r = *resp + i;
102        r->resp_retcode = 0;    // 这个是保留属性，固定为0
103        switch (m->msg_style)
104        {
105        case PAM_PROMPT_ECHO_OFF:   // 请求输入，不回显，例如请求输入用户名
106            printf("%s", m->msg);
107            echoOffStdin();         // 关闭回显
108            r->resp = readline();   // 读取密码
109            echoOnStdin();          // 开启回显
110            printf("\n");           // 补个换行
111            break;
112
113        case PAM_PROMPT_ECHO_ON:    // 请求输入，回显，例如请求输入密码
114            printf("%s", m->msg);
115            r->resp = readline();
116            break;
117
118        case PAM_TEXT_INFO:         // 打印普通消息
119            printf("%s\n", m->msg);
120            break;
121
122        case PAM_ERROR_MSG:         // 打印错误消息
123            fprintf(stderr, "%s\n", m->msg);
124            break;
125
126        default:
127            printf("DEFAULT\n");
128            break;
129        }
130    }
131    return PAM_SUCCESS;
132}
133
134int main()
135{
136    atexit(echoOnStdin); // 退出时开启回显
137
138    struct pam_conv pam_conv = {conversation, NULL};
139    pam_handle_t *pamh;
140    if (PAM_SUCCESS != pam_start("login", NULL, &pam_conv, &pamh))
141    {
142        fprintf(stderr, "pam_start failed\n");
143        return EXIT_FAILURE;
144    }
145
146    if (PAM_SUCCESS != pam_authenticate(pamh, 0))
147    {
148        fprintf(stderr, "pam_authenticate failed\n");
149        pam_end(pamh, 0);
150        return EXIT_FAILURE;
151    }
152    
153    pam_end(pamh, 0);
154    return EXIT_SUCCESS;
155}