summaryrefslogtreecommitdiff
path: root/libc/pwd_grp/lckpwdf.c
blob: 41987ca5d325bf0478e4fbbd044db0420efd6a5f (plain)
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
/* vi: set sw=4 ts=4: */
/* Handle locking of password file.
   Copyright (C) 1996,98,2000,02 Free Software Foundation, Inc.
   This file is part of the GNU C Library.
   Contributed by Ulrich Drepper <drepper@cygnus.com>, 1996.

   The GNU C Library is free software; you can redistribute it and/or
   modify it under the terms of the GNU Lesser General Public
   License as published by the Free Software Foundation; either
   version 2.1 of the License, or (at your option) any later version.

   The GNU C Library is distributed in the hope that it will be useful,
   but WITHOUT ANY WARRANTY; without even the implied warranty of
   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
   Lesser General Public License for more details.

   You should have received a copy of the GNU Lesser General Public
   License along with the GNU C Library; if not, write to the Free
   Software Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA
   02111-1307 USA.  */

#include <features.h>
#include <fcntl.h>
#include <signal.h>
#include <string.h>
#include <unistd.h>
#include <sys/file.h>
#include <paths.h>
#include <shadow.h>

/* Experimentally off - libc_hidden_proto(memset) */
/* libc_hidden_proto(open) */
/* libc_hidden_proto(fcntl) */
/* libc_hidden_proto(close) */
/* libc_hidden_proto(sigfillset) */
/* libc_hidden_proto(sigaction) */
/* libc_hidden_proto(sigprocmask) */
/* libc_hidden_proto(sigaddset) */
/* libc_hidden_proto(sigemptyset) */
/* libc_hidden_proto(alarm) */

/* How long to wait for getting the lock before returning with an error.  */
#define TIMEOUT 15 /* sec */

/* File descriptor for lock file.  */
static int lock_fd = -1;

/* Prevent problems in multithreaded program by using mutex.  */
#include <bits/uClibc_mutex.h>
__UCLIBC_MUTEX_STATIC(mylock, PTHREAD_MUTEX_INITIALIZER);

/* Prototypes for local functions.  */
static void noop_handler (int __sig);


int
lckpwdf (void)
{
  int flags;
  sigset_t saved_set;			/* Saved set of caught signals.  */
  struct sigaction saved_act;		/* Saved signal action.  */
  sigset_t new_set;			/* New set of caught signals.  */
  struct sigaction new_act;		/* New signal action.  */
  struct flock fl;			/* Information struct for locking.  */
  int result;
  int rv = -1;

  if (lock_fd != -1)
    /* Still locked by own process.  */
    return -1;

  /* Prevent problems caused by multiple threads.  */
  __UCLIBC_MUTEX_LOCK(mylock);

  lock_fd = open (_PATH_PASSWD, O_WRONLY);
  if (lock_fd == -1) {
    /* Cannot create lock file.  */
	goto DONE;
  }

  /* Make sure file gets correctly closed when process finished.  */
  flags = fcntl (lock_fd, F_GETFD);
#if 0 /* never fails */
  if (flags == -1) {
    /* Cannot get file flags.  */
    close(lock_fd);
    lock_fd = -1;
	goto DONE;
  }
#endif
  flags |= FD_CLOEXEC;
#if 1
  fcntl (lock_fd, F_SETFD, flags);
#else /* never fails */
  if (fcntl (lock_fd, F_SETFD, flags) < 0) {
    /* Cannot set new flags.  */
    close(lock_fd);
    lock_fd = -1;
	goto DONE;
  }
#endif
  /* Now we have to get exclusive write access.  Since multiple
     process could try this we won't stop when it first fails.
     Instead we set a timeout for the system call.  Once the timer
     expires it is likely that there are some problems which cannot be
     resolved by waiting. (sa_flags have no SA_RESTART. Thus SIGALRM
     will EINTR fcntl(F_SETLKW)

     It is important that we don't change the signal state.  We must
     restore the old signal behaviour.  */
  memset (&new_act, '\0', sizeof (struct sigaction));
  new_act.sa_handler = noop_handler;
  __sigfillset (&new_act.sa_mask);
  /* new_act.sa_flags = 0; */

  /* Install new action handler for alarm and save old.  */
  if (sigaction (SIGALRM, &new_act, &saved_act) < 0) {
    /* Cannot install signal handler.  */
    close(lock_fd);
    lock_fd = -1;
	goto DONE;
  }

  /* Now make sure the alarm signal is not blocked.  */
  __sigemptyset (&new_set);
  sigaddset (&new_set, SIGALRM);
#if 1
  sigprocmask (SIG_UNBLOCK, &new_set, &saved_set);
#else /* never fails */
  if (sigprocmask (SIG_UNBLOCK, &new_set, &saved_set) < 0) {
    sigaction (SIGALRM, &saved_act, NULL);
    close(lock_fd);
    lock_fd = -1;
	goto DONE;
  }
#endif

  /* Start timer.  If we cannot get the lock in the specified time we
     get a signal.  */
  alarm (TIMEOUT);

  /* Try to get the lock.  */
  memset (&fl, '\0', sizeof (fl));
  if (F_WRLCK)
    fl.l_type = F_WRLCK;
  if (SEEK_SET)
    fl.l_whence = SEEK_SET;
  result = fcntl (lock_fd, F_SETLKW, &fl);

  /* Clear alarm.  */
  alarm (0);

  sigprocmask (SIG_SETMASK, &saved_set, NULL);
  sigaction (SIGALRM, &saved_act, NULL);

  if (result < 0) {
    close(lock_fd);
    lock_fd = -1;
	goto DONE;
  }
  rv = 0;

DONE:
  __UCLIBC_MUTEX_UNLOCK(mylock);
  return 0;
}


int
ulckpwdf (void)
{
  int result;

  if (lock_fd == -1)
    /* There is no lock set.  */
    result = -1;
  else
    {
      /* Prevent problems caused by multiple threads.  */
	  __UCLIBC_MUTEX_LOCK(mylock);

      result = close (lock_fd);

      /* Mark descriptor as unused.  */
      lock_fd = -1;

      /* Clear mutex.  */
	  __UCLIBC_MUTEX_UNLOCK(mylock);
    }

  return result;
}


static void
noop_handler (int sig)
{
  /* We simply return which makes the `fcntl' call return with an error.  */
}