pam: avoid warning messages
[openafs.git] / src / pam / test_pam.c
1 /*
2  * Copyright 2000, International Business Machines Corporation and others.
3  * All Rights Reserved.
4  *
5  * This software has been released under the terms of the IBM Public
6  * License.  For details, see the LICENSE file in the top-level source
7  * directory or online at http://www.openafs.org/dl/license10.html
8  */
9
10 #include <afsconfig.h>
11 #include <afs/param.h>
12
13 #include <roken.h>
14
15 #include <security/pam_appl.h>
16
17 static int my_conv(int num_msg, PAM_CONST struct pam_message **msg,
18                    struct pam_response **response, void *appdata_ptr);
19
20
21 static struct pam_conv pam_conv = { &my_conv, NULL };
22
23
24 static pam_handle_t *pamh;
25
26
27 static const char *service = "afstest";
28 static const char *new_envstring = "GOTHEREVIATESTPAM=1";
29 static const char *new_homestring = "HOME=/tmp";
30
31 #if defined(AFS_LINUX20_ENV) || defined(AFS_FBSD_ENV) || defined(AFS_DFBSD_ENV) || defined(AFS_NBSD_ENV) || defined(AFS_DARWIN_ENV)
32 #define getpassphrase getpass
33 #endif
34
35
36 int
37 main(int argc, char *argv[])
38 {
39     int authenticated = 0;
40     int retcode;
41     char *username;
42     int setcred = 1;
43
44     if (argc < 2 || argc > 3) {
45         fprintf(stderr, "Usage: %s [-u] <user>\n", argv[0]);
46         exit(1);
47     }
48     if (argc == 3) {
49         if (strcmp(argv[1], "-u") != 0) {
50             fprintf(stderr, "Usage: %s [-u] <user>\n", argv[0]);
51             exit(1);
52         }
53         /* service = "unixtest"; */
54         setcred = 0;
55         username = argv[2];
56     } else {
57         username = argv[1];
58     }
59
60     if ((retcode =
61          pam_start(service, username, &pam_conv, &pamh)) != PAM_SUCCESS) {
62         fprintf(stderr, "PAM error %d\n", retcode);
63         exit(1);
64     }
65
66     authenticated = ((retcode = pam_authenticate(pamh, 0)) == PAM_SUCCESS);
67
68     if (!authenticated) {
69         fprintf(stderr, "PAM couldn't authenticate you.\n");
70         pam_end(pamh, PAM_ABORT);
71         exit(1);
72     }
73
74     if ((retcode = pam_acct_mgmt(pamh, 0)) != PAM_SUCCESS) {
75         fprintf(stderr, "pam_acct_mgmt returned %d.\n", retcode);
76         pam_end(pamh, PAM_ABORT);
77         exit(1);
78     }
79
80     /* pam_open_session */
81
82     if (setcred)
83         if ((retcode = pam_setcred(pamh, PAM_ESTABLISH_CRED)) != PAM_SUCCESS) {
84             fprintf(stderr, "pam_setcred returned %d.\n", retcode);
85             pam_end(pamh, PAM_ABORT);
86             exit(1);
87         }
88
89     if ((retcode = pam_open_session(pamh, PAM_SILENT)) != PAM_SUCCESS) {
90         fprintf(stderr, "pam_open_session returned %d.\n", retcode);
91         pam_end(pamh, PAM_ABORT);
92         exit(1);
93     }
94     pam_end(pamh, PAM_SUCCESS);
95
96     putenv((char *)new_envstring);
97     putenv((char *)new_homestring);
98
99     if ((retcode = chdir("/tmp")) != 0) {
100         fprintf(stderr, "chdir returned %d.\n", retcode);
101         exit(1);
102     }
103
104     printf("Type exit to back out.\n");
105     return execl("/bin/csh", "/bin/csh", NULL);
106 }
107
108
109 static int
110 my_conv(int num_msg, PAM_CONST struct pam_message **msg, struct pam_response **response,
111         void *appdata_ptr)
112 {
113     PAM_CONST struct pam_message *m;
114     struct pam_response *r;
115     char *p;
116
117     m = *msg;
118     if (response) {
119         *response = calloc(num_msg, sizeof(struct pam_response));
120         if (*response == NULL)
121             return PAM_BUF_ERR;
122         r = *response;
123     } else {
124         r = NULL;
125     }
126
127     while (num_msg--) {
128         switch (m->msg_style) {
129         case PAM_PROMPT_ECHO_OFF:
130 #ifdef __hpux
131             /* ON HP's we still read 8 chars */
132             if (r)
133                 r->resp = strdup(getpass(m->msg));
134 #else
135             if (r)
136                 r->resp = strdup(getpassphrase(m->msg));
137 #endif
138             break;
139         case PAM_PROMPT_ECHO_ON:
140             fputs(m->msg, stdout);
141             if (r) {
142                 r->resp = malloc(PAM_MAX_RESP_SIZE);
143                 if (fgets(r->resp, PAM_MAX_RESP_SIZE, stdin) == NULL) {
144                     fprintf(stderr, "fgets did not work as expected\n");
145                     exit(1);
146                 }
147                 r->resp[PAM_MAX_RESP_SIZE - 1] = '\0';
148                 p = &r->resp[strlen(r->resp) - 1];
149                 while (*p == '\n' && p >= r->resp)
150                     *(p--) = '\0';
151             }
152             break;
153         case PAM_ERROR_MSG:
154             fputs(m->msg, stderr);
155             break;
156         case PAM_TEXT_INFO:
157             fputs(m->msg, stdout);
158             break;
159         default:
160             break;
161         }
162         m++;
163         if (r)
164             r++;
165     }
166     return PAM_SUCCESS;
167 }