File: pam_tmpdir.c

package info (click to toggle)
pam-tmpdir 0.11
  • links: PTS
  • area: main
  • in suites: sid, trixie
  • size: 128 kB
  • sloc: ansic: 445; sh: 171; makefile: 28
file content (247 lines) | stat: -rw-r--r-- 5,231 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
/* pam_tmpdir module, based on pam_env which in turn was based on
   pam_mail */

/*
 * $Id: pam_env.c,v 1.3 1999/11/08 05:46:53 morgan Exp $
 * 
 * Written by Tollef Fog Heen <tfheen@err.no> 2001-02-03
 *
 * Based on pam_env
 * Written by Dave Kinchlea <kinch@kinch.ark.com> 1997/01/31
 * Inspired by Andrew Morgan <morgan@parc.power.net, who also supplied the 
 * template for this file (via pam_mail)
 */

#define _GNU_SOURCE

#include <ctype.h>
#include <errno.h>
#include <pwd.h>
#include <signal.h>
#include <stdarg.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <syslog.h>
#include <sys/stat.h>
#include <sys/types.h>
#include <sys/wait.h>
#include <unistd.h>
#include "util.h"

/*
 * here, we make a definition for the externally accessible function
 * in this file (this definition is required for static a module
 * but strongly encouraged generally) it is used to instruct the
 * modules include file to define the function prototypes.
 */

#define PAM_SM_SESSION      /* This is primarily a AUTH_SESSION module */
#define PAM_SM_AUTH         /* But I like to be friendly */
#define PAM_SM_PASSWORD     /*        ""                 */
#define PAM_SM_ACCOUNT      /*        ""                 */

#include <security/pam_modules.h>
#include <security/_pam_macros.h>

#define PAM_TMPDIR_HELPER "/usr/libexec/pam-tmpdir/pam-tmpdir-helper"

static int set_environment(pam_handle_t *pamh);
static int make_tmp_directory(pam_handle_t *pamh);

int get_user_id(pam_handle_t *pamh, uid_t *uid) {
  const void *username;
  struct passwd* user_entry;
  int r;

  r = pam_get_item(pamh, PAM_USER, &username);
  if (r != PAM_SUCCESS) {
    _log_err(LOG_WARNING, "could not get PAM user");
    return -1;
  }
  user_entry = getpwnam(username);
  *uid = user_entry->pw_uid;
  return 0;
}

/* argument parsing */

#define PAM_DEBUG_ARG       0x01
#define PAM_NEW_CONF_FILE   0x02
#define PAM_ENV_SILENT      0x04
#define PAM_NEW_ENV_FILE    0x10

static int set_environment(pam_handle_t *pamh) {
  char *buf;
  char *tmpdir = get_tmp_dir();
  char *envs[] = { "TMP", "TMPDIR", "TEMP", "TEMPDIR", NULL };
  uid_t u;
  if (get_user_id(pamh, &u) != 0) {
    free(tmpdir);
    return -1;
  }

  for (int i = 0; envs[i]; i++) {
    asprintf(&buf, "%s=%s/%d", envs[i], tmpdir, u);
    pam_putenv(pamh, buf);
    free(buf);
  }

  free(tmpdir);
  return 0;
}

/* --- authentication management functions (only) --- */

static int make_tmp_directory(pam_handle_t *pamh) {
  int ret = 0;
  int status;
  pid_t pid;
  struct sigaction old, new;

  new.sa_handler = SIG_DFL;
  sigemptyset(&new.sa_mask);
  new.sa_flags = 0;

  /* save old signal handler */
  ret = sigaction(SIGCHLD, &new, &old);
  if (ret == -1) {
    _log_err(LOG_WARNING, "could not save SIGCHLD handler");
    return 1;
  }

  /* Do fork and exec by hand.. */
  pid = fork();
  if (!pid) {
    /* child */
    if (geteuid() == 0) {
      uid_t u;
      if (get_user_id(pamh, &u) != 0) {
	exit(1);
      }
      if (setuid(u) != 0) {
	exit(1);
      }
    }
    execl(PAM_TMPDIR_HELPER,PAM_TMPDIR_HELPER,NULL);
    exit(1);
  }

  ret = waitpid(pid, &status, 0);
  if (ret == -1) {
    _log_err(LOG_ERR, "error waiting for helper binary (%d)\n", errno);
    return 1;
  }

  /* restore old signal handler */
  ret = sigaction(SIGCHLD, &old, NULL);
  if (ret == -1) {
    _log_err(LOG_WARNING, "could not restore SIGCHLD handler");
    return 1;
  }

  return WEXITSTATUS(status);
}


PAM_EXTERN
int pam_sm_authenticate(pam_handle_t *pamh, int flags, int argc,
			const char **argv)
{ 
  (void)pamh;
  (void)flags;
  (void)argc;
  (void)argv;
  return PAM_IGNORE;
}

PAM_EXTERN 
int pam_sm_setcred(pam_handle_t *pamh, int flags, int argc, 
		   const char **argv)
{
  (void)pamh;
  (void)flags;
  (void)argc;
  (void)argv;

  if (make_tmp_directory(pamh) == 0) {
    set_environment(pamh);
    return PAM_SUCCESS;
  } else {
    return PAM_ABORT;
  }
}

PAM_EXTERN 
int pam_sm_acct_mgmt(pam_handle_t *pamh, int flags, int argc, 
		     const char **argv)
{
  (void)pamh;
  (void)flags;
  (void)argc;
  (void)argv;

  _log_err(LOG_NOTICE, "pam_sm_acct_mgmt called inappropriatly");
  return PAM_SERVICE_ERR;
}
 
PAM_EXTERN
int pam_sm_open_session(pam_handle_t *pamh,int flags,int argc, const char **argv)
{
  (void)pamh;
  (void)flags;
  (void)argc;
  (void)argv;

  if (make_tmp_directory(pamh) == 0) {
    set_environment(pamh);
    return PAM_SUCCESS;
  } else {
    return PAM_ABORT;
  }
  
}

PAM_EXTERN
int pam_sm_close_session(pam_handle_t *pamh,int flags,int argc,
			 const char **argv)
{
  (void)pamh;
  (void)flags;
  (void)argc;
  (void)argv;

  D(("Called and Exit"));
  return PAM_SUCCESS;
}

PAM_EXTERN 
int pam_sm_chauthtok(pam_handle_t *pamh, int flags, int argc, 
		     const char **argv)
{
  (void)pamh;
  (void)flags;
  (void)argc;
  (void)argv;

  _log_err(LOG_NOTICE, "pam_sm_chauthtok called inappropriatly");
  return PAM_SERVICE_ERR;
}

#ifdef PAM_STATIC

/* static module data */

struct pam_module _pam_env_modstruct = {
     "pam_tmpdir",
     pam_sm_authenticate,
     pam_sm_setcred,
     pam_sm_acct_mgmt,
     pam_sm_open_session,
     pam_sm_close_session,
     pam_sm_chauthtok,
};

#endif

/* end of module definition */