lib/libc/gen/getlogin.c

   1 /*
   2  * Copyright (c) 1988, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer.
  10  * 2. Redistributions in binary form must reproduce the above copyright
  11  *    notice, this list of conditions and the following disclaimer in the
  12  *    documentation and/or other materials provided with the distribution.
  13  * 3. All advertising materials mentioning features or use of this software
  14  *    must display the following acknowledgement:
  15  *      This product includes software developed by the University of
  16  *      California, Berkeley and its contributors.
  17  * 4. Neither the name of the University nor the names of its contributors
  18  *    may be used to endorse or promote products derived from this software
  19  *    without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31  * SUCH DAMAGE.
  32  *
  33  * $FreeBSD: src/lib/libc/gen/getlogin.c,v 1.4.2.1 2001/03/05 09:06:50 obrien Exp $
  34  * $DragonFly: src/lib/libc/gen/getlogin.c,v 1.2 2003/06/17 04:26:42 dillon Exp $
  35  *
  36  * @(#)getlogin.c       8.1 (Berkeley) 6/4/93
  37  */
  38
  39 #include <sys/param.h>
  40 #include <errno.h>
  41 #include <pwd.h>
  42 #include <utmp.h>
  43 #include <stdio.h>
  44 #include <string.h>
  45 #include <unistd.h>
  46
  47 #include <libc_private.h>
  48
  49 #ifndef _THREAD_SAFE
  50 #define THREAD_LOCK()
  51 #define THREAD_UNLOCK()
  52 #else
  53 #include <pthread.h>
  54 #include "pthread_private.h"
  55 static struct pthread_mutex logname_lock = PTHREAD_MUTEX_STATIC_INITIALIZER;
  56 static pthread_mutex_t logname_mutex = &logname_lock;
  57 #define THREAD_LOCK()   if (__isthreaded) pthread_mutex_lock(&logname_mutex)
  58 #define THREAD_UNLOCK() if (__isthreaded) pthread_mutex_unlock(&logname_mutex)
  59 #endif /* _THREAD_SAFE */
  60
  61 int             _logname_valid;         /* known to setlogin() */
  62
  63 static char *
  64 getlogin_basic(int *status)
  65 {
  66         static char logname[MAXLOGNAME];
  67
  68         if (_logname_valid == 0) {
  69 #ifdef __NETBSD_SYSCALLS
  70                 if (__getlogin(logname, sizeof(logname) - 1) < 0) {
  71 #else
  72                 if (_getlogin(logname, sizeof(logname)) < 0) {
  73 #endif
  74                         *status = errno;
  75                         return (NULL);
  76                 }
  77                 _logname_valid = 1;
  78         }
  79         *status = 0;
  80         return (*logname ? logname : NULL);
  81 }
  82
  83 char *
  84 getlogin(void)
  85 {
  86         char    *result;
  87         int     status;
  88
  89         THREAD_LOCK();
  90         result = getlogin_basic(&status);
  91         THREAD_UNLOCK();
  92         return (result);
  93 }
  94
  95 int
  96 getlogin_r(char *logname, int namelen)
  97 {
  98         char    *result;
  99         int     len;
 100         int     status;
 101
 102         THREAD_LOCK();
 103         result = getlogin_basic(&status);
 104         if (status == 0) {
 105                 if ((len = strlen(result) + 1) > namelen)
 106                         status = ERANGE;
 107                 else
 108                         strncpy(logname, result, len);
 109         }
 110         THREAD_UNLOCK();
 111         return (status);
 112 }